a ߙfbZ/@sddlZddlmZddlmZddlmZddlmZm Z ddl m Z dgZ dgZ ejejZejejejejd efZejejejd ZGd ddeZdS) N)si equivalencies)Unit)SpecificTypeQuantityQuantity)quantity_inputSpectralQuantityzSpectralQuantity.*)ZradioZopticalZ relativisticcseZdZdZeZdZdfdd ZfddZdd Z fd d Z e d d Z e j eeddd Z e ddZej ddZeedgddffdd ZdddZZS)r a One or more value(s) with spectral units. The spectral units should be those for frequencies, wavelengths, energies, wavenumbers, or velocities (interpreted as Doppler velocities relative to a rest spectral value). The advantage of using this class over the regular `~astropy.units.Quantity` class is that in `SpectralQuantity`, the ``u.spectral`` equivalency is enabled by default (allowing automatic conversion between spectral units), and a preferred Doppler rest value and convention can be stored for easy conversion to/from velocities. Parameters ---------- value : ndarray or `~astropy.units.Quantity` or `SpectralQuantity` Spectral axis data values. unit : unit-like Unit for the given data. doppler_rest : `~astropy.units.Quantity` ['speed'], optional The rest value to use for conversions from/to velocities doppler_convention : str, optional The convention to use when converting the spectral data to/from velocities. TNc sRtj||fd|i|}|dur.t|dd}|durBt|dd}||_||_|S)Nunit doppler_restdoppler_convention)super__new__getattr _doppler_rest_doppler_convention)clsvaluer r r kwargsobj __class__Dlib/python3.9/site-packages/astropy/coordinates/spectral_quantity.pyr5s  zSpectralQuantity.__new__cs,t|t|dd|_t|dd|_dS)Nrr)r__array_finalize__rrr)selfrrrrrGs z#SpectralQuantity.__array_finalize__cCs||jurtdfStdfSdS)NTF)r r r)rr rrr__quantity_subclass__Ls z&SpectralQuantity.__quantity_subclass__cstj||g|Ri|}|tjus<|tjurH|d|urH|j|jksh|tjtjtjtj fvr|dvr| |j }| |n>||urt d|j j|jjdkr| tj}n | t}|S)Nr)reduceZreduceatz-Cannot store the result of this operation in b)r__array_ufunc__npZmultiplyZ true_divider ZminimumZmaximumZfmaxZfminviewrr TypeError__name__Zdtypekindndarrayr)rZfunctionmethodZinputsrresultrrrr Vs(       z SpectralQuantity.__array_ufunc__cCs|jS)z The rest value of the spectrum used for transformations to/from velocity space. Returns ------- `~astropy.units.Quantity` ['speed'] Rest value as an astropy `~astropy.units.Quantity` object. )rrrrrr is zSpectralQuantity.doppler_rest)rcCs|jdurtd||_dS)z New rest value needed for velocity-space conversions. Parameters ---------- value : `~astropy.units.Quantity` ['speed'] Rest value. Nzdoppler_rest has already been set, and cannot be changed. Use the ``to`` method to convert the spectral values(s) to use a different rest value)rAttributeErrorrrrrrr vs cCs|jS)z The defined convention for conversions to/from velocity space. Returns ------- str One of 'optical', 'radio', or 'relativistic' representing the equivalency used in the unit conversions. )rr)rrrr s z#SpectralQuantity.doppler_conventioncCsD|jdurtd|dur:|tvr:tddtt||_dS)a New velocity convention used for velocity space conversions. Parameters ---------- value Notes ----- More information on the equations dictating the transformations can be found in the astropy documentation [1]_. References ---------- .. [1] Astropy documentation: https://docs.astropy.org/en/stable/units/equivalencies.html#spectral-doppler-equivalencies Nzdoppler_convention has already been set, and cannot be changed. Use the ``to`` method to convert the spectral values(s) to use a different convention$doppler_convention should be one of /)rr*DOPPLER_CONVENTIONS ValueErrorjoinsortedr+rrrr s  )r c st|}|dur:tj|dd}||j}|||S|durH|j}|durX|j}n |tvrxt dd t t|j trd| trd|dur|jdurt d|dur|jdurt d|dur|durtj||d}||j}|||S|du|duurt dt|j|j}tjtj||d}t||}|j|||d}njt} |j ts| tr|durt d|durt d | t||} tj||| d}||j}||||_||_|S) ay Return a new `~astropy.coordinates.SpectralQuantity` object with the specified unit. By default, the ``spectral`` equivalency will be enabled, as well as one of the Doppler equivalencies if converting to/from velocities. Parameters ---------- unit : unit-like An object that represents the unit to convert to. Must be an `~astropy.units.UnitBase` object or a string parseable by the `~astropy.units` package, and should be a spectral unit. equivalencies : list of `~astropy.units.equivalencies.Equivalency`, optional A list of equivalence pairs to try if the units are not directly convertible (along with spectral). See :ref:`astropy:unit_equivalencies`. If not provided or ``[]``, spectral equivalencies will be used. If `None`, no equivalencies will be applied at all, not even any set globally or within a context. doppler_rest : `~astropy.units.Quantity` ['speed'], optional The rest value used when converting to/from velocities. This will also be set at an attribute on the output `~astropy.coordinates.SpectralQuantity`. doppler_convention : {'relativistic', 'optical', 'radio'}, optional The Doppler convention used when converting to/from velocities. This will also be set at an attribute on the output `~astropy.coordinates.SpectralQuantity`. Returns ------- `SpectralQuantity` New spectral coordinate object with data converted to the new unit. Nrr,r-z#Original doppler_convention not setzOriginal doppler_rest not setzeEither both or neither doppler_rest and doppler_convention should be defined for velocity conversionsz=doppler_convention not set, cannot convert to/from velocitiesz7doppler_rest not set, cannot convert to/from velocities)rrtor"rrrrr.r/r0r1r Z is_equivalentKMSrHzeqZspectral) rr rr r r(Z vel_equiv1ZfreqZ vel_equiv2Zadditional_equivalenciesrrrr2sR(         zSpectralQuantity.tocOs.|dur|tjS|j|g|Ri|jS)N)r"r!r&r2r)rr argsrrrrto_value,s zSpectralQuantity.to_value)NNN)N)r$ __module__ __qualname____doc__SPECTRAL_UNITSZ_equivalent_unitZ _include_easy_conversion_membersrrrr propertyr setterrr r2r7 __classcell__rrrrr s0      w)Znumpyr!Z astropy.unitsrrr5rZastropy.units.quantityrrZastropy.units.decoratorsr__all__Z__doctest_skip__Zkmsr3r4mJr;Z doppler_radioZdoppler_opticalZdoppler_relativisticr.r rrrrs