U e{j; @s^ddlZddlZddlmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZmZmZdddd d d d d ddddg ZGddde ZGdddeZGdddeZGdd d eZGdd d e ZGdd d eZGdd d eZGdd d eZee_ee_ee_ee jZde_ee jZ de _ee j!Z"de j!j#e"_ee j$Z%de j$j#e%_dS)N) CompositeUnitUnitUnitConversionError UnitsError UnitTypeErrordimensionless_unscaled photometric)FunctionQuantityFunctionUnitBase)dBdexmagLogUnitMagUnitDexUnit DecibelUnit LogQuantity MagnitudeDecibelDexSTmagABmagM_bolm_bolc@sheZdZdZeddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZdS)ra Logarithmic unit containing a physical one Usually, logarithmic units are instantiated via specific subclasses such `~astropy.units.MagUnit`, `~astropy.units.DecibelUnit`, and `~astropy.units.DexUnit`. Parameters ---------- physical_unit : `~astropy.units.Unit` or `string` Unit that is encapsulated within the logarithmic function unit. If not given, dimensionless. function_unit : `~astropy.units.Unit` or `string` By default, the same as the logarithmic unit set by the subclass. cCstSNr selfra/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/units/function/logarithmic.py_default_function_unit6szLogUnit._default_function_unitcCstSr)rrrrr _quantity_class:szLogUnit._quantity_classcCst|jt|S)zaTransformation from value in physical to value in logarithmic units. Used in equivalency.)r to_function_unitnplog10rxrrr from_physical>szLogUnit.from_physicalcCsd|jt|S)zaTransformation from value in logarithmic to value in physical units. Used in equivalency. )r$r#r r'rrr to_physicalCszLogUnit.to_physicalcCsxzt|d||jWn2tk r0tYStk rJtdYnXt|dt}td|j|g||g}| |S)uAdd/subtract LogUnit to/from another unit, and adjust physical unit. self and other are multiplied by sign_self and sign_other, resp. We wish to do: ±lu_1 + ±lu_2 -> lu_f (lu=logarithmic unit) and pu_1^(±1) * pu_2^(±1) -> pu_f (pu=physical unit) Raises ------ UnitsError If function units are not equivalent. function_unitz;Can only add/subtract logarithmic units of compatible type. physical_unitr ) getattr_tor$AttributeErrorNotImplementedrrrZ_physical_unit_copy)rotherZ sign_selfZ sign_otherZother_physical_unitr-rrr _add_and_adjust_physical_unitLs  z%LogUnit._add_and_adjust_physical_unitcCs||jdS)N)r2r-rrrr __neg__mszLogUnit.__neg__cCs||ddSNr r4rr3rrr __add__pszLogUnit.__add__cCs||ddSr7r8r9rrr __radd__uszLogUnit.__radd__cCs||ddS)Nr r5r8r9rrr __sub__xszLogUnit.__sub__cCs||ddS)Nr5r r8r9rrr __rsub__{szLogUnit.__rsub__N)__name__ __module__ __qualname____doc__propertyr!r"r)r+r4r6r:r;r<r=rrrr r#s   !c@s(eZdZdZeddZeddZdS)raLogarithmic physical units expressed in magnitudes Parameters ---------- physical_unit : `~astropy.units.Unit` or `string` Unit that is encapsulated within the magnitude function unit. If not given, dimensionless. function_unit : `~astropy.units.Unit` or `string` By default, this is ``mag``, but this allows one to use an equivalent unit such as ``2 mag``. cCstSr)rrrrr r!szMagUnit._default_function_unitcCstSr)rrrrr r"szMagUnit._quantity_classNr>r?r@rArBr!r"rrrr rs   cs:eZdZdZeddZeddZd fdd ZZS) raLogarithmic physical units expressed in magnitudes Parameters ---------- physical_unit : `~astropy.units.Unit` or `string` Unit that is encapsulated within the magnitude function unit. If not given, dimensionless. function_unit : `~astropy.units.Unit` or `string` By default, this is ``dex``, but this allows one to use an equivalent unit such as ``0.5 dex``. cCstSrrrrrr r!szDexUnit._default_function_unitcCstSr)rrrrr r"szDexUnit._quantity_classgenericcs<|dkr.|jtkrdSd|jj|ddSn tSdS)NZcdsz[-][)format])r-r to_stringsuper)rrF __class__rr rHs  zDexUnit.to_string)rD) r>r?r@rArBr!r"rH __classcell__rrrJr rs    c@s(eZdZdZeddZeddZdS)raLogarithmic physical units expressed in dB Parameters ---------- physical_unit : `~astropy.units.Unit` or `string` Unit that is encapsulated within the decibel function unit. If not given, dimensionless. function_unit : `~astropy.units.Unit` or `string` By default, this is ``dB``, but this allows one to use an equivalent unit such as ``2 dB``. cCstSr)r rrrr r!sz"DecibelUnit._default_function_unitcCstSr)rrrrr r"szDecibelUnit._quantity_classNrCrrrr rs   cseZdZdZeZddZddZddZdd Z d d Z d d Z fddZ ddZ fddZfddZfddZddZddZd-ddZd.d d!Zd/d"d#Zd0d&d'Zd1d(d)Zejd*d+d,DBZZS)2raA representation of a (scaled) logarithm of a number with a unit Parameters ---------- value : number, `~astropy.units.Quantity`, `~astropy.units.function.logarithmic.LogQuantity`, or sequence of quantity-like. The numerical value of the logarithmic quantity. If a number or a `~astropy.units.Quantity` with a logarithmic unit, it will be converted to ``unit`` and the physical unit will be inferred from ``unit``. If a `~astropy.units.Quantity` with just a physical unit, it will converted to the logarithmic unit, after, if necessary, converting it to the physical unit inferred from ``unit``. unit : str, `~astropy.units.UnitBase`, or `~astropy.units.FunctionUnitBase`, optional For an `~astropy.units.FunctionUnitBase` instance, the physical unit will be taken from it; for other input, it will be inferred from ``value``. By default, ``unit`` is set by the subclass. dtype : `~numpy.dtype`, optional The ``dtype`` of the resulting Numpy array or scalar that will hold the value. If not provided, is is determined automatically from the input value. copy : bool, optional If `True` (default), then the value is copied. Otherwise, a copy will only be made if ``__array__`` returns a copy, if value is a nested sequence, or if a copy is needed to satisfy an explicitly given ``dtype``. (The `False` option is intended mostly for internal use, to speed up initialization where a copy is known to have been made. Use with care.) Examples -------- Typically, use is made of an `~astropy.units.FunctionQuantity` subclass, as in:: >>> import astropy.units as u >>> u.Magnitude(-2.5) >>> u.Magnitude(10.*u.count/u.second) >>> u.Decibel(1.*u.W, u.DecibelUnit(u.mW)) # doctest: +FLOAT_CMP cCs0|jt|dt}|jt|d|}|||SNunit_function_viewrNr.rrO _new_viewrr3new_unitresultrrr r:szLogQuantity.__add__cCs ||Sr)r:r9rrr r; szLogQuantity.__radd__cCs6|jt|dt}|j}|t|d|7}|||SrMrNr.rrO _set_unitrr3rS function_viewrrr __iadd__ s  zLogQuantity.__iadd__cCs0|jt|dt}|jt|d|}|||SrMrPrRrrr r<szLogQuantity.__sub__cCs@|jt|dt}|jt|d|}||j}|||SrM)rNr=r.rrOr#r,rQrRrrr r=s zLogQuantity.__rsub__cCs6|jt|dt}|j}|t|d|8}|||SrMrUrWrrr __isub__$s  zLogQuantity.__isub__csLt|tjr<|jj|}|tj|}|||j |St |SdSr) isinstancenumbersNumberrNr-viewr%ndarrayrQr2rI__mul__rr3new_physical_unitrTrJrr r`,s   zLogQuantity.__mul__cCs ||Sr)r`r9rrr __rmul__7szLogQuantity.__rmul__csLt|tjr<|jj|}|j}||9}||j||St |SdSr) r[r\r]rNr-rOrVr2rI__imul__rr3rbrXrJrr rd:s  zLogQuantity.__imul__csPt|tjr@|jjd|}|tj|}|||j |St |SdSr7) r[r\r]rNr-r^r%r_rQr2rI __truediv__rarJrr rfDs  zLogQuantity.__truediv__csPt|tjr@|jjd|}|j}||}||j||St |SdSr7) r[r\r]rNr-rOrVr2rI __itruediv__rerJrr rgOs zLogQuantity.__itruediv__cCsLz t|}Wntk r$tYSX|j|}|tj|}|||Sr)float TypeErrorr1rNr^r%r_rQ)rr3rS new_valuerrr __pow__Ys   zLogQuantity.__pow__c Csz t|}Wntk r$tYSXt||js6tSz|jj|j}WnPtk rz| |}Wntk rtYYSX|| t j d<Yn"X| t j d|j |7<|||S)N.)rrr1r[ _unit_classrNr-r/rZ _to_valuer^r%r_r)rV)rr3factorvaluerrr __ilshift__cs"     zLogQuantity.__ilshift__NrcCs$|jjd}|jtj|||||dS)NoutddofrN)rNr,_wrap_functionr%varraxisZdtyperrrsrNrrr rus zLogQuantity.varcCs$|jt}|jtj|||||dS)Nrq)rNr2rrtr%stdrvrrr rxs zLogQuantity.stdcCs |jt}|jtj|||dS)N)rrrN)rNr2rrtr%ptp)rrwrrrNrrr rys zLogQuantity.ptpr r5cCs |jt}|jtj|||dSN)rN)rNr2rrtr%diff)rnrwrNrrr r{s zLogQuantity.diffcCs |jt}|jtj|||dSrz)rNr2rrtr%ediff1d)rZto_endZto_beginrNrrr r}s zLogQuantity.ediff1dcCsh|]}tt|qSr)r.r%).0functionrrr szLogQuantity.)rurxryr{r})NNNr)NNNr)NN)r r5)NN)r>r?r@rArrlr:r;rYr<r=rZr`rcrdrfrgrkrorurxryr{r}r Z_supported_functionsrLrrrJr rs..          c@seZdZeZdS)rN)r>r?r@rrlrrrr rsc@seZdZeZdS)rN)r>r?r@rrlrrrr rsc@seZdZeZdS)rN)r>r?r@rrlrrrr rsz6ST magnitude: STmag=-21.1 corresponds to 1 erg/s/cm2/Az7AB magnitude: ABmag=-48.6 corresponds to 1 erg/s/cm2/Hzz=Absolute bolometric magnitude: M_bol=0 corresponds to L_bol0=z=Apparent bolometric magnitude: m_bol=0 corresponds to f_bol0=)&r\numpyr%Z astropy.unitsrrrrrrrcorer r Zunitsr r r__all__rrrrrrrrZ_function_unit_classZSTfluxrrAZABfluxrZBolrsiZbolrrrrr sN$ ^ J