IR-e<dZddlZddlZddlmZmZmZmZgdZ Gdde Z e Z e j ZdZdZd d ZdS) zConverters for Quantity.N)UnitConversionError UnitsError UnitTypeErrordimensionless_unscaled)can_have_arbitrary_unitconverters_and_unit check_output UFUNC_HELPERSUNSUPPORTED_UFUNCSc>eZdZdZfdZdZdZdZfdZxZ S) UfuncHelpersaRegistry of unit conversion functions to help ufunc evaluation. Based on dict for quick access, but with a missing method to load helpers for additional modules such as scipy.special and erfa. Such modules should be registered using ``register_module``. ci|_t|_tj|_t j|i|dSN)modulesset UNSUPPORTED threadingRLock_locksuper__init__)selfargskwargs __class__s Hlib/python3.11/site-packages/astropy/units/quantity_helper/converters.pyrzUfuncHelpers.__init__!sG 55_&& $)&)))))c`|j5||d|j|<ddddS#1swxYwYdS)aRegister (but do not import) a set of ufunc helpers. Parameters ---------- module : str Name of the module with the ufuncs (e.g., 'scipy.special'). names : iterable of str Names of the module ufuncs for which helpers are available. importer : callable Function that imports the ufuncs and returns a dict of helpers keyed by those ufuncs. If the value is `None`, the ufunc is explicitly *not* supported. )namesimporterN)rr)rmodulerr s rregister_modulezUfuncHelpers.register_module'sZ J J-2#I#IDL  J J J J J J J J J J J J J J J J J Js #''c|j5|j|}||dddddS#1swxYwYdS)zImport the helpers from the given module using its helper function. Parameters ---------- module : str Name of the module. Has to have been registered beforehand. r N)rrpopupdate)rr! module_infos r import_modulezUfuncHelpers.import_module8sZ 3 3,**622K KK/ J/11 2 2 2 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s>AAAc|j5||vr||cdddS||jvrtd|jdt |jD]P\}}|j|dvr< ||||ccdddS#t$rYLwxYwQ dddn #1swxYwYtd|jd)zCalled if a ufunc is not found. Check if the ufunc is in any of the available modules, and, if so, import the helpers for that module. NzCannot use ufunc 'z' with quantitiesrzunknown ufunc zm. If you believe this ufunc should be supported, please raise an issue on https://github.com/astropy/astropy) rr TypeError__name__listritemsr' ImportError)rufuncr!r&s r __missing__zUfuncHelpers.__missing__Ds Z + +}}E{ + + + + + + + + ((( VU^ V V VWWW'+DL,>,>,@,@'A'A + +# >[%999+**6222 $E{**/ + + + + + + + +(': + + + + + + + + + + + + + + + +2 1U^ 1 1 1   s; CAC=B( C( B52C4B55CC  C c|j5|(|xj|hzc_||dn3t|||xj|hzc_ddddS#1swxYwYdSr)rrr$r __setitem__)rkeyvaluers rr1zUfuncHelpers.__setitem__is Z * *}  SE)  d######C///  SE)   * * * * * * * * * * * * * * * * * *sAA44A8;A8) r* __module__ __qualname____doc__rr"r'r/r1 __classcell__)rs@rr r s***** JJJ" 3 3 3# # # J * * * * * * * * *rr ctjtjtj|dtj|S)aPTest whether the items in value can have arbitrary units. Numbers whose value does not change upon a unit change, i.e., zero, infinity, or not-a-number Parameters ---------- value : number or array Returns ------- bool `True` if each member is either zero or not finite, `False` otherwise g)npall logical_orequalisfinite)r3s rrr{s9 6"- 4 4r{57I7I6IJJ K KKrc t|}|dks|dkr|jdkrd|D}||g|R\}}td|Dr t|D]<\}}|dur t ||rd||<%t d|jd nS#t$rFtd |jd d |DwxYw|dur|d |d tkr t}n|d |d} n%|d tj } |d | z}n#t$rG} |d t|d <t}n#t $r| wxYwYd} ~ nPd} ~ wwxYwnF|j} t|d dd} |dkrL| dkrF| dkr| g}n| t|dddg}||g|R\}}|ddn`|dvr-| dkr'||| | \}}|dd}|dkr|dgz }n/|dvr| dkrt|dtd|d| |td|d|jd|d 4| M| |urI|| r| | dkrt%d|d|jd||fS)a>Determine the required converters and the unit of the ufunc result. Converters are functions required to convert to a ufunc's expected unit, e.g., radian for np.sin; or to ensure units of two inputs are consistent, e.g., for np.add. In these examples, the unit of the result would be dimensionless_unscaled for np.sin, and the same consistent unit for np.add. Parameters ---------- function : `~numpy.ufunc` Numpy universal function method : str Method with which the function is evaluated, e.g., '__call__', 'reduce', etc. *args : `~astropy.units.Quantity` or ndarray subclass Input arguments to the function Raises ------ TypeError : when the specified function cannot be used with Quantities (e.g., np.logical_or), or when the routine does not know how to handle the specified function (in which case an issue should be raised on https://github.com/astropy/astropy). UnitTypeError : when the conversion to the required (or consistent) units is not possible. __call__outerc0g|]}t|ddS)unitN)getattr.0args r z'converters_and_unit..s$<<<fd++<<z&converters_and_unit..s'>>iyE!>>>>>>rFNzCan only apply 'zz' function to dimensionless quantities when other argument is not a quantity (unless the latter is all zero/infinity/nan).z.Unsupported operand type(s) for ufunc {}: '{}',c&g|]}|jjSrJ)rr*rEs rrHz'converters_and_unit..s!I!I!IS#-"8!I!I!IrrrCat>reducereduceat accumulaterR>r@rQrRrSz$ only supported for binary functionszUnexpected ufunc method zS. If this should work, please raise an issue on https://github.com/astropy/astropyz Cannot use 'z' method on ufunc z: with a Quantity instance as the result is not a Quantity.g?z6 with a Quantity instance as it would change the unit.)r ninany enumeraterrr*r)formatjoinrtor3 ValueError_get_converterrDinsert is_equivalentr) functionmethodr ufunc_helperunits converters result_unitirKpexcrTrCs rrrsv>!*L ' 1 1hla6G6G=>:>>> > >  %.j$9$9  LAy -- .tAw77(, 1 1Wx/@WWW     DKK )!I!ID!I!I!IJJ  %  Qx58/E#E#E4 8#QAAQ #9::@A ="'(a-KK!====(-a(?(?@V(W(W 1 '= /"""! "$ = ,ltAw-- T>>cQhhaxxwtAw==>&2l8&De&D&D&D #J    a & & & & ; ; ;q&2l8T4&H&H #J #BQBJ##tf$ FFF3RS88 F!P!P!PQQQG6GGG    3EvEE9JEEE  a= $  K'' ..t44(8Ct8L8LPS8S8S AvAA9JAAA  { ""s8A B&&AC6 E## F4.#FF/ F&&F//F4c bt|tr*tfdt||DS|dSt|dr|=t d djdndt|||\}}|sP|t|us?td| djdndt||rT|j j sHtj }|j s3tj||j d st d |j |tjS|8|t"us/td  d jdnd|S)aCheck that function output can be stored in the output array given. Parameters ---------- output : array or `~astropy.units.Quantity` or tuple Array that should hold the function output (or tuple of such arrays). unit : `~astropy.units.Unit` or None, or tuple Unit that the output will have, or `None` for pure numbers (should be tuple of same if output is a tuple of outputs). inputs : tuple Any input arguments. These should be castable to the output. function : callable The function that will be producing the output. If given, used to give a more informative error message. Returns ------- arrays : ndarray view or tuple thereof The view(s) is of ``output``. Raises ------ UnitTypeError : If ``unit`` is inconsistent with the class of ``output`` TypeError : If the ``inputs`` cannot be cast safely to ``output``. c3BK|]\}}t||VdSr)r )rFoutput_unit_r^inputss rrLzcheck_output..@sI   % : :      rN__quantity_subclass__z1Cannot store non-quantity output{} in {} instancez from z functionzNCannot store output with unit '{}'{} in {} instance. Use {} instance instead. same_kind)castingz=Arguments cannot be cast safely to inplace output with dtype=zBCannot store quantity with dimension {}in a non-Quantity instance.zresulting from z function ) isinstancetupleziphasattrr)rWr*typerlrdtyperr9 result_typecan_castviewndarrayr)outputrCrkr^q_clssubokrvs `` rr r $s@6&%        "%fd"3"3      ~tv.//> <CJJ$/>!2====LL    33D99 u $v,,..<!2====LL = =   &  &,, .&1K! ;{FL+NNN  8)/88 {{2:&&& (> > >006+Dh&7CCCC11  rr)r6rnumpyr9astropy.units.corerrrr__all__dictr r rr rrr rJrrrs   \*\*\*\*\*4\*\*\*~  ".LLL$T#T#T#nddddddr