B q\/ @sJddlZddlmZmZmZmZddlmZm Z ddl m Z m Z m Z dddd d d d d ddddg ZGdddeZGdddeZGdddeZGdd d eZGdd d e ZGdd d eZGdd d eZGdd d eZee _ee _ee _eejZde_eejZde_eejZdejj e_eej!Z"dej!j e"_dS)N) CompositeUnit UnitsErrordimensionless_unscaled photometric)FunctionUnitBaseFunctionQuantity)dexdBmagLogUnitMagUnitDexUnit DecibelUnit LogQuantity MagnitudeDecibelDexSTmagABmagM_bolm_bolc@sheZdZdZeddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZdS)r aLogarithmic unit containing a physical one Usually, logarithmic units are instantiated via specific subclasses such `MagUnit`, `DecibelUnit`, and `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. cCstS)N)r )selfrAlib/python3.7/site-packages/astropy/units/function/logarithmic.py_default_function_unit szLogUnit._default_function_unitcCstS)N)r)rrrr_quantity_class$szLogUnit._quantity_classcCst|jt|S)zaTransformation from value in physical to value in logarithmic units. Used in equivalency.)r to_function_unitnpZlog10)rxrrr from_physical(szLogUnit.from_physicalcCsd|jt|S)zaTransformation from value in logarithmic to value in physical units. Used in equivalency. )rrr )rr rrr to_physical-szLogUnit.to_physicalcCstyt|d||jWn.tk r,tStk rFtdYnXt|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 ofof compatible type. physical_unitr) getattrZ_torAttributeErrorNotImplementedrrrZ_physical_unit_copy)rotherZ sign_selfZ sign_otherZother_physical_unitr%rrr_add_and_adjust_physical_unit5s  z%LogUnit._add_and_adjust_physical_unitcCs||jdS)N)r)r%)rrrr__neg__VszLogUnit.__neg__cCs||ddS)Nr)r+)rr*rrr__add__YszLogUnit.__add__cCs||ddS)Nr)r+)rr*rrr__radd__^szLogUnit.__radd__cCs||ddS)Nrr,)r+)rr*rrr__sub__aszLogUnit.__sub__cCs||ddS)Nr,r)r+)rr*rrr__rsub__dszLogUnit.__rsub__N)__name__ __module__ __qualname____doc__propertyrrr!r#r+r-r.r/r0r1rrrrr s  !c@s(eZdZdZeddZeddZdS)r aLogarithmic 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``. cCstS)N)r )rrrrrwszMagUnit._default_function_unitcCstS)N)r)rrrrr{szMagUnit._quantity_classN)r2r3r4r5r6rrrrrrr js  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 ``dex`, but this allows one to use an equivalent unit such as ``0.5 dex``. cCstS)N)r )rrrrrszDexUnit._default_function_unitcCstS)N)r)rrrrrszDexUnit._quantity_classN)r2r3r4r5r6rrrrrrrs  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``. cCstS)N)r )rrrrrsz"DecibelUnit._default_function_unitcCstS)N)r)rrrrrszDecibelUnit._quantity_classN)r2r3r4r5r6rrrrrrrs  c@seZdZdZeZddZddZddZdd Z d d Z d d Z dddZ d ddZ d!ddZd"ddZd#ddZejedddDBZdS)$raA 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 convertible items. 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 : string, `~astropy.units.UnitBase` or `~astropy.units.function.FunctionUnitBase` instance, optional For an `~astropy.units.function.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.function.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|}|||S)Nunit_function_view)r7r&rr8 _new_view)rr*new_unitresultrrrr.szLogQuantity.__add__cCs ||S)N)r.)rr*rrrr/szLogQuantity.__radd__cCs6|jt|dt}|j}|t|d|7}|||S)Nr7r8)r7r&rr8 _set_unit)rr*r: function_viewrrr__iadd__s  zLogQuantity.__iadd__cCs0|jt|dt}|jt|d|}|||S)Nr7r8)r7r&rr8r9)rr*r:r;rrrr0szLogQuantity.__sub__cCs@|jt|dt}|jt|d|}||j}|||S)Nr7r8)r7r1r&rr8rr$r9)rr*r:r;rrrr1s  zLogQuantity.__rsub__cCs6|jt|dt}|j}|t|d|8}|||S)Nr7r8)r7r&rr8r<)rr*r:r=rrr__isub__s  zLogQuantity.__isub__NrcCs |jtj|||||jjddS)N)outddofr7)_wrap_functionrvarr7r$)raxisdtyperArBrrrrDszLogQuantity.varc Cs |jtj|||||jtdS)N)rArBr7)rCrstdr7r)r)rrErFrArBrrrrGszLogQuantity.stdcCs|jtj|||jtdS)N)rAr7)rCrptpr7r)r)rrErArrrrHs zLogQuantity.ptprr,cCs|jtj|||jtdS)N)r7)rCrdiffr7r)r)rnrErrrrIs zLogQuantity.diffcCs|jtj|||jtdS)N)r7)rCrediff1dr7r)r)rZto_endZto_beginrrrrK"s zLogQuantity.ediff1dccs|]}tt|VqdS)N)r&r).0Zfunctionrrr 'szLogQuantity.)rDrGrHrIrK)NNNr)NNNr)NN)rr,)NN)r2r3r4r5r _unit_classr.r/r>r0r1r?rDrGrHrIrKrZ_supported_functionssetrrrrrs ,       c@seZdZeZdS)rN)r2r3r4rrNrrrrr+sc@seZdZeZdS)rN)r2r3r4rrNrrrrr/sc@seZdZeZdS)rN)r2r3r4r rNrrrrr3sz6ST 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={0}z@Apparent bolometric magnitude: m_bol=0 corresponds to f_bol0={0})#ZnumpyrZ astropy.unitsrrrrZcorerrZunitsr r r __all__r r rrrrrrZ_function_unit_classZSTfluxrr5ZABfluxrZBolrformatZsiZbolrrrrrs6 [}