ó šßÈ[c@`s2ddlmZmZmZmZddlZddlmZm Z m Z ddl m Z ddl Z ddlmZddlmZmZddlmZd d d d d gZdefd„ƒYZdefd„ƒYZejeƒdefd„ƒYƒZdefd„ƒYZdefd„ƒYZdS(i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tABCMetatabstractpropertytabstractmethod(tdeepcopyi(tlog(tUnittQuantity(tsixuMissingDataAssociationExceptionu"IncompatibleUncertaintiesExceptionu NDUncertaintyuStdDevUncertaintyuUnknownUncertaintyt"IncompatibleUncertaintiesExceptioncB`seZdZRS(uThis exception should be used to indicate cases in which uncertainties with two different classes can not be propagated. (t__name__t __module__t__doc__(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR stMissingDataAssociationExceptioncB`seZdZRS(u–This exception should be used to indicate that an uncertainty instance has not been associated with a parent `~astropy.nddata.NDData` object. (R RR(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRst NDUncertaintycB`sïeZdZdedd„Zed„ƒZed„ƒZ ed„ƒZ e j d„ƒZ ed„ƒZ ed„ƒZ e j d„ƒZ d „Zd „Zd „Zd „Zed „ƒZed„ƒZed„ƒZed„ƒZRS(uThis is the metaclass for uncertainty classes used with `NDData`. Parameters ---------- array : any type, optional The array or value (the parameter name is due to historical reasons) of the uncertainty. `numpy.ndarray`, `~astropy.units.Quantity` or `NDUncertainty` subclasses are recommended. If the `array` is `list`-like or `numpy.ndarray`-like it will be cast to a plain `numpy.ndarray`. Default is ``None``. unit : `~astropy.units.Unit` or str, optional Unit for the uncertainty ``array``. Strings that can be converted to a `~astropy.units.Unit` are allowed. Default is ``None``. copy : `bool`, optional Indicates whether to save the `array` as a copy. ``True`` copies it before saving, while ``False`` tries to save every parameter as reference. Note however that it is not always possible to save the input as reference. Default is ``True``. Raises ------ IncompatibleUncertaintiesException If given another `NDUncertainty`-like class as ``array`` if their ``uncertainty_type`` is different. cC`sJt|tƒr||j|jkr*t‚n|dk rU||jkrUtjdƒn|jdk rp|j}n|j }npt|t ƒrì|dk rÅ|jdk rÅ||jkrÅtjdƒn|jdk rà|j}n|j }n|dkrd|_nt |ƒ|_|r4t |ƒ}t |ƒ}n||_ d|_dS(Nu;overwriting Uncertainty's current unit with specified unit.u8overwriting Quantity's current unit with specified unit.(t isinstanceRtuncertainty_typeR tNonet_unitRtinfotunittarrayR tvalueR Rt parent_nddata(tselfRtcopyR((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyt__init__Bs.         cC`sdS(u‹`str` : Short description of the type of uncertainty. Defined as abstract property so subclasses *have* to override this. N(R(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRgscC`stS(u`bool` : Supports uncertainty propagation with correlated uncertainties? .. versionadded:: 1.2 (tFalse(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pytsupports_correlatedoscC`s|jS(u3`numpy.ndarray` : the uncertainty's value. (t_array(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRxscC`sCt|ttjfƒr6tj|dtdtƒ}n||_dS(NtsubokR(RtlisttnptndarrayRRR (RR((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR~scC`sH|jdkrA|jdks0|jjdkr4dS|jjSn|jS(u@`~astropy.units.Unit` : The unit of the uncertainty, if any. Even though it is not enforced the unit should be convertible to the ``parent_nddata`` unit. Otherwise uncertainty propagation might give wrong results. If the unit is not set the unit of the parent will be returned. N(RRt_parent_nddataRR(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR„s  cC`sd}yU|jdkr't|ƒ‚n3t|jtjƒrF|jƒStjdƒ|jSWntk rzt|ƒ‚nXdS(uù`NDData` : reference to `NDData` instance with this uncertainty. In case the reference is not set uncertainty propagation will not be possible since propagation might need the uncertain data besides the uncertainty. u3uncertainty is not associated with an NDData objectu6parent_nddata should be a weakref to an NDData object.N( R%RRRtweakreftrefRRtAttributeError(Rtmessage((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR–s    cC`s>|dk r1t|tjƒ r1tj|ƒ}n||_dS(N(RRR&R'R%(RR((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR®scC`sk|jjd}y"tj|jddd|ƒ}Wn tk rTt|jƒ}nXdj||dgƒS(Nu(t separatoru, tprefixuu)(t __class__R R#t array2stringRR(tstrtjoin(RR+tbody((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyt__repr__·s " cC`s#|j|j|d|jdtƒS(uKNormal slicing on the array, keep the unit and return a reference. RR(R,RRR(Rtitem((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyt __getitem__ÀscC`s|jsHt|tjƒs'|dkrHtdj|jjƒƒ‚qHn|j|j ƒ}|jdkr|j |||ƒ}n|jdkr¨|j |||ƒ}nZ|jdkrÏ|j |||ƒ}n3|jd krö|j |||ƒ}n tdƒ‚|j|d tƒS( uhCalculate the resulting uncertainty given an operation on the data. .. versionadded:: 1.2 Parameters ---------- operation : callable The operation that is performed on the `NDData`. Supported are `numpy.add`, `numpy.subtract`, `numpy.multiply` and `numpy.true_divide` (or `numpy.divide`). other_nddata : `NDData` instance The second operand in the arithmetic operation. result_data : `~astropy.units.Quantity` or `numpy.ndarray` The result of the arithmetic operations on the data. correlation : `numpy.ndarray` or number The correlation (rho) is defined between the uncertainties in sigma_AB = sigma_A * sigma_B * rho. A value of ``0`` means uncorrelated operands. Returns ------- resulting_uncertainty : `NDUncertainty` instance Another instance of the same `NDUncertainty` subclass containing the uncertainty of the result. Raises ------ ValueError If the ``operation`` is not supported or if correlation is not zero but the subclass does not support correlated uncertainties. Notes ----- First this method checks if a correlation is given and the subclass implements propagation with correlated uncertainties. Then the second uncertainty is converted (or an Exception is raised) to the same class in order to do the propagation. Then the appropriate propagation method is invoked and the result is returned. iu>{0} does not support uncertainty propagation with correlation.uaddusubtractumultiplyu true_divideudivideuunsupported operationR(u true_divideudivide(RRR#R$t ValueErrortformatR,R t_convert_uncertaintyt uncertaintyt_propagate_addt_propagate_subtractt_propagate_multiplyt_propagate_divideR(Rt operationt other_nddatat result_datat correlationt other_uncerttresult((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyt propagateÅs&-           cC`s?t|tƒr5|j|jkr%|S|j|ƒSnt‚dS(uChecks if the uncertainties are compatible for propagation. Checks if the other uncertainty is `NDUncertainty`-like and if so verify that the uncertainty_type is equal. If the latter is not the case try returning ``self.__class__(other_uncert)``. Parameters ---------- other_uncert : `NDUncertainty` subclass The other uncertainty. Returns ------- other_uncert : `NDUncertainty` subclass but converted to a compatible `NDUncertainty` subclass if possible and necessary. Raises ------ IncompatibleUncertaintiesException: If the other uncertainty cannot be converted to a compatible `NDUncertainty` subclass. N(RRRR,R (RR@((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR6 s cC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR8,scC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR90scC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR:4scC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR;8sN(R RRRtTrueRRRtpropertyRRtsetterRRR1R3RBR6RR8R9R:R;(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR!s"%   G tUnknownUncertaintycB`sYeZdZed„ƒZed„ƒZd„Zd„Zd„Zd„Z d„Z RS(uîThis class implements any unknown uncertainty type. The main purpose of having an unknown uncertainty class is to prevent uncertainty propagation. Parameters ---------- args, kwargs : see `NDUncertainty` cC`stS(uL`False` : Uncertainty propagation is *not* possible for this class. (R(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRIscC`sdS(uq``"unknown"`` : `UnknownUncertainty` implements any unknown uncertainty type. uunknown((R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyROscC`sd}t|ƒ‚dS(ucRaise an Exception because unknown uncertainty types cannot implement propagation. u3Uncertainties of unknown type cannot be propagated.N(R (RR@tmsg((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR6VscC`sdS(u4Not possible for unknown uncertainty types. N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR8]scC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR9bscC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR:escC`sdS(N(R(RR@R>R?((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR;hs( R RRRDRRR6R8R9R:R;(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRF=s     tStdDevUncertaintycB`sYeZdZed„ƒZed„ƒZd„Zd„Zd„Zd„Z d„Z RS(u Standard deviation uncertainty assuming first order gaussian error propagation. This class implements uncertainty propagation for ``addition``, ``subtraction``, ``multiplication`` and ``division`` with other instances of `StdDevUncertainty`. The class can handle if the uncertainty has a unit that differs from (but is convertible to) the parents `NDData` unit. The unit of the resulting uncertainty will have the same unit as the resulting data. Also support for correlation is possible but requires the correlation as input. It cannot handle correlation determination itself. Parameters ---------- args, kwargs : see `NDUncertainty` Examples -------- `StdDevUncertainty` should always be associated with an `NDData`-like instance, either by creating it during initialization:: >>> from astropy.nddata import NDData, StdDevUncertainty >>> ndd = NDData([1,2,3], ... uncertainty=StdDevUncertainty([0.1, 0.1, 0.1])) >>> ndd.uncertainty # doctest: +FLOAT_CMP StdDevUncertainty([0.1, 0.1, 0.1]) or by setting it manually on the `NDData` instance:: >>> ndd.uncertainty = StdDevUncertainty([0.2], unit='m', copy=True) >>> ndd.uncertainty # doctest: +FLOAT_CMP StdDevUncertainty([0.2]) the uncertainty ``array`` can also be set directly:: >>> ndd.uncertainty.array = 2 >>> ndd.uncertainty StdDevUncertainty(2) .. note:: The unit will not be displayed. cC`stS(uÍ`True` : `StdDevUncertainty` allows to propagate correlated uncertainties. ``correlation`` must be given, this class does not implement computing it by itself. (RC(R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR˜scC`sdS(uG``"std"`` : `StdDevUncertainty` implements standard deviation. ustd((R((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR¢scC`st|tƒr|St‚dS(N(RRHR (RR@((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR6¨scC`s”|jdkr]|jdk rM|j|jkrM|j|jj|jƒjSt|jƒSn3|jdkr¹|jdk r©|j|jjkr©|jj|j|jƒSt|jƒSn×|j|jkrî|j|j}|j|j}n|j}|j}t|tj ƒs|dkrRd|||}tj |d|d|ƒ}ntj |d|dƒ}t|t ƒrŒ|j |jƒS|SdS(Nii( RRRttoRRRRR#R$tsqrtR tto_value(RR@R>R?tthistothertcorrRA((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR8®s, $  "cC`s”|jdkr]|jdk rM|j|jkrM|j|jj|jƒjSt|jƒSn3|jdkr¹|jdk r©|j|jjkr©|jj|j|jƒSt|jƒSn×|j|jkrî|j|j}|j|j}n|j}|j}t|tj ƒs|dkrRd|||}tj |d|d|ƒ}ntj |d|dƒ}t|t ƒrŒ|j |jƒS|SdS(Nii( RRRRIRRRRR#R$RJR RK(RR@R>R?RLRMRNRA((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR9ës, $  "c C`st|tƒr|j}n|jdkr„|j|jjkrd|j|jj|jjƒj}n |j}tj |jj |ƒS|jdkrí|j|jjkrÍ|j|jj|jjƒj}n |j}tj |jj |ƒS|j|jjkr1|j|jj|jjƒj|jj }n|j|jj }|j|jjkrˆ|j|jj|jjƒj|jj }n|j|jj }t|tj ƒs¹|dkrèd|||}tj |d|d|ƒStj |d|dƒSdS(Nii( RR RRRRRRIR#tabstdataR$RJ( RR@R>R?RMRLtlefttrightRN((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR:s8    cC`st|tƒr|j}n|jdkr’|j|jjkrn|j|jj|jjƒj|jj}n|j|jj}t j ||ƒS|jdkrû|j|jjkrÛ|j|jj|jjƒj}n |j}t j ||jjƒS|j|jjkr?|j|jj|jjƒj|jj}n|j|jj}|j|jjkrš|j|jj|jjƒj|jj|}n||j|jj}t|t j ƒsÏ|dkrþd|||}t j |d|d|ƒSt j |d|dƒSdS(Nii( RR RRRRRRIRPR#ROR$RJ(RR@R>R?RRRQRN((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyR;Ms<    ( R RRRDRRR6R8R9R:R;(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyRHls*   = $ >(t __future__RRRRtnumpyR#tabcRRRRRR&tRtunitsR R texternR t__all__t ExceptionR Rt add_metaclasstobjectRRFRH(((s;lib/python2.7/site-packages/astropy/nddata/nduncertainty.pyts""   ÿ/