U e{j<@stdZddlZddlZddlmZmZmZmZdddddgZ Gd d d e Z e Z e j Zd dZd dZdd dZdS)zConverters for Quantity.N)UnitConversionError UnitsError UnitTypeErrordimensionless_unscaledcan_have_arbitrary_unitconverters_and_unit check_output UFUNC_HELPERSUNSUPPORTED_UFUNCScsDeZdZdZfddZddZddZdd Zfd d ZZ 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``. cs*i|_t|_t|_tj||dSN)modulesset UNSUPPORTED threadingRLock_locksuper__init__)selfargskwargs __class__g/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/units/quantity_helper/converters.pyr!s zUfuncHelpers.__init__c Cs&|j||d|j|<W5QRXdS)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 )rmodulerrrrrregister_module'szUfuncHelpers.register_modulec Cs2|j"|j|}||dW5QRXdS)zImport the helpers from the given module using its helper function. Parameters ---------- module : str Name of the module. Has to have been registered beforehand. rN)rr popupdate)rr module_inforrr import_module8s zUfuncHelpers.import_modulec Cs|j||kr$||W5QRS||jkr@td|jdt|jD]R\}}|j|dkrNz||Wntk rYqNX||W5QRSqNW5QRXtd|jddS)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. zCannot use ufunc 'z' with quantitiesrzunknown ufunc zm. If you believe this ufunc should be supported, please raise an issue on https://github.com/astropy/astropyN) rr TypeError__name__listr itemsr# ImportError)rZufuncrr"rrr __missing__Ds  $ zUfuncHelpers.__missing__c sZ|jJ|dkr.|j|hO_||dnt|||j|h8_W5QRXdSr )rrr r __setitem__)rkeyvaluerrrr*is zUfuncHelpers.__setitem__) r% __module__ __qualname____doc__rrr#r)r* __classcell__rrrrr s   %r cCs"ttt|dt|S)aOTest 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)r,rrrr{sc Gs t|}|dks&|dkr|jdkrdd|D}||f|\}}tdd|DrzHt|D]:\}}|dk rtqbt||rd ||<qbtd |jd qbWn6tk rtd |jd dd|DYnX|dkr|dd ks|dt krt }n|dd kr|d} n|d t j } z|d| }WnZt k r} z:z|dt |d<Wntk r~| YnXt }W5d } ~ XYnXnj|j} t|ddd } |dkr | dkr | dkr| g}n| t|ddd g}||f|\}}|dd nx|dkrP| dkrP||| | \}}|d d}|dkr|d g7}n2|dkrr| dkrrt |dtd|d| d k r|d krtd|d|jd|dd k s| d k r| |k r|| r| | dkrtd|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__outercSsg|]}t|ddqS)unitN)getattr.0argrrr sz'converters_and_unit..css|]}|dkVqdS)FNr)r< converterrrr sz&converters_and_unit..FNzCan 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 {}: '{}',cSsg|] }|jjqSr)rr%r;rrrr>srr9at>reduceat accumulatereducerD>r7rDrErFz$ 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$formatjoinrtor, ValueErrorZ_get_converterr:insertZ is_equivalentr) functionmethodrZ ufunc_helperZunits convertersZ result_unitir?pexcrGr9rrrrs             csJt|tr(tfddt||DS|dkr4dSt|dr|dkrrtddk rddjdndt|||\}}|s|t|kst d |dk rdjdndt||r|j j st j }|j st j||j d d std |j |t jS|dksB|tksBt d dk r:djdnd|SdS)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``. c3s |]\}}t||VqdSr )r)r<Zoutput_Zunit_rOinputsrrr@Aszcheck_output..N__quantity_subclass__z1Cannot store non-quantity output{} in {} instancez from z functionzNCannot store output with unit '{}'{} in {} instance. Use {} instance instead.Z same_kind)Zcastingz=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$rJr%typerWrZdtyperr1 result_typeZcan_castviewZndarrayr)outputr9rVrOZq_clsZsubokr^rrUrr%s^        )N)r/rnumpyr1Zastropy.units.corerrrr__all__dictr r rr rrrrrrrs  _