B dQl@sdZddlZddlZddlmZddlmZddlmZddlmZddlmZdd lm Z dd lm Z dd l m Z d d dgZ ejdfejdfejddfejddfejdfejdfejdfejejddhfejejddfejdfejdfejdfejdfejejdejdd fejejdd!d"hfejejdejdd#d$hfejd%fejd&d'd(hfejejd)ejdd*d+hfejd,d-d.hfejd/d0hfejejd1d2fejdejd3fejejdd4fejdejd5fejejejd6d7hfejejdejd8d9hfejejd:fejejdd;fejejdd?hfejdd@fejd)dAfejdBfejdCfejdDfejdEdFdGhfejejdHfej dIfej ejdJfej!dKfejejdLfejejddMfejejdNfejejddOdPdQhfejejddRfej!ejdSfej"dTfej#dUfejejdVfejdejdWfej$ejdXdYhfej$dZfej%d[fej&d\fej'd]d^hfejejd_fej%ejdd`fejd1ejddafej(dbfejejdejddcfejejd1ejddddehfej)ejej*dejdffej)ej+ej*dejdgfej,dhfe j-dife j-ejdjfej.dkfej/dlfej0dmfej1dnfejejd1dodphfejejdqdrdshfejejdtdufejejdvdwdxhfejejdyfejejdzfejejd1ejdd{fejejdd|fejejd}fejejdejdd~fejejdfejejdfejejddfejdejdfejd)dfejddfej2ejd1dfej2ejd)ejddfejejddfejejdejdfejejdfejejdfej$ddfejejdfejejdfejejejdfejejdfej ejdejdfejdejejdfej3ejdfejdejddhfejejd)ejdddhfejd1dfejd)ejddfgjZ4iZ5iZ6iZ7ddZ8ddZ9ddZ:GdddZ;dd Z
Z?ee?qbWedk rdddddddgZ@xdeAe7BD]TZCe7eCZ?e@DdeCdde?jEFddddddGdde?DgqWeddGe@7Z[>[?dS)z>Defines the physical types that correspond to different units.N)core)si) astrophys)cgs)imperial)misc)quantity)AstropyDeprecationWarningdef_physical_typeget_physical_type PhysicalTypeZ dimensionlesslengthZareavolumetimeZanglez solid anglespeedvelocityZ accelerationZ frequencyZmasszamount of substanceZ temperaturezthermal conductivityz heat capacityZentropyzspecific heat capacityzspecific entropyforceZenergyZworkZtorquez energy fluxZ irradianceZpressurezenergy densityZstresspowerz radiant fluxz mass densityzspecific volumezmolar concentrationz molar volumeZmomentumZimpulsezangular momentumactionz angular speedzangular accelerationz plate scalezdynamic viscosityZ diffusivityzkinematic viscosityZ wavenumberzcolumn densityzelectrical currentzelectrical chargezelectrical potentialzelectrical resistancezelectrical impedancezelectrical reactancezelectrical resistivityzelectrical conductancezelectrical conductivityzelectrical capacitancezelectrical dipole momentzelectrical current densityzelectrical field strengthzelectrical flux densityzsurface charge densityzpolarization densityzelectrical charge densityZ permittivityz magnetic fluxzmagnetic flux densityzmagnetic field strengthzmagnetic momentzelectromagnetic field strengthZ permeabilityZ inductancezluminous intensityz luminous fluxzluminous emittanceZ illuminancezradiant intensityZ luminancezvolumetric ratezspectral flux densityzsurface tensionzspectral flux density wavz power densityzphoton flux densityzphoton flux density wavz photon fluxz data quantity bandwidthzelectrical charge (ESU)zelectrical current (ESU)zelectrical current (EMU)zelectrical charge (EMU)ZjerkZjoltZsnapZjounceZcrackleipopZpounceztemperature gradientzspecific energyz reaction ratezmoment of inertiazcatalytic activityzmolar heat capacityZmolalityZabsementZabsityzvolumetric flow ratezfrequency driftZcompressibilityzelectron densityz electron fluxzsurface mass densityZradiancezchemical potentialzlinear densityzmagnetic reluctancezthermal conductancezthermal resistancezthermal resistivityZyankzmolar conductivityzelectrical mobilityzluminous efficacyZopacityzmass attenuation coefficientz mass fluxzmomentum densityznumber densityz particle fluxcCs2|dkrtd|tkr t|St|ddS)z] Return the `PhysicalType` instance associated with the name of a physical type. unknownz4cannot uniquely identify an 'unknown' physical type.z is not a known physical type.N) ValueError_name_physical_mapping)namer#c/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/astropy/units/physical.py_physical_type_from_strs r%cCs\|}g}d}x.|D]&\}}|dkr.d}d}|||fqW|rTtjt|S|SdS)u If a unit contains a temperature unit besides kelvin, then replace that unit with kelvin. Temperatures cannot be converted directly between K, °F, °C, and °Ra, in particular since there would be different conversions for T and ΔT. However, each of these temperatures each represents the physical type. Replacing the different temperature units with kelvin allows the physical type to be treated consistently. F)Zdeg_FZdeg_CZdeg_RKTN)_get_physical_type_idappendrZUnitZ_from_physical_type_idtuple)unitphysical_type_idZphysical_type_id_componentsZsubstitution_was_madebaserr#r#r$!_replace_temperatures_with_kelvins r-cCsdt|tr|h}t}xH|D]@}t|ts8td|t|d}dd|D}||O}qW|S)ai Convert a string or `set` of strings into a `set` containing string representations of physical types. The strings provided in ``physical_type_input`` can each contain multiple physical types that are separated by a regular slash. Underscores are treated as spaces so that variable names could be identical to physical type names. zexpecting a string, but got /cSsh|]}|ddqS)_ )stripreplace).0sr#r#r$ sz3_standardize_physical_type_names..) isinstancestrsetr split)Zphysical_type_inputZstandardized_physical_typesZ ptype_inputZ input_setZ processed_setr#r#r$ _standardize_physical_type_namess    r:cseZdZdZddZddZfddZdd Zd d Zd d Z ddZ ddZ e ddZ ddZddZddZddZddZddZd d!Zd"d#Zd$ZZS)%r a Represents the physical type(s) that are dimensionally compatible with a set of units. Instances of this class should be accessed through either `get_physical_type` or by using the `~astropy.units.core.UnitBase.physical_type` attribute of units. This class is not intended to be instantiated directly in user code. Parameters ---------- unit : `~astropy.units.Unit` The unit to be represented by the physical type. physical_types : `str` or `set` of `str` A `str` representing the name of the physical type of the unit, or a `set` containing strings that represent one or more names of physical types. Notes ----- A physical type will be considered equal to an equivalent `PhysicalType` instance (recommended) or a string that contains a name of the physical type. The latter method is not recommended in packages, as the names of some physical types may change in the future. To maintain backwards compatibility, two physical type names may be included in one string if they are separated with a slash (e.g., ``"momentum/impulse"``). String representations of physical types may include underscores instead of spaces. Examples -------- `PhysicalType` instances may be accessed via the `~astropy.units.core.UnitBase.physical_type` attribute of units. >>> import astropy.units as u >>> u.meter.physical_type PhysicalType('length') `PhysicalType` instances may also be accessed by calling `get_physical_type`. This function will accept a unit, a string containing the name of a physical type, or the number one. >>> u.get_physical_type(u.m ** -3) PhysicalType('number density') >>> u.get_physical_type("volume") PhysicalType('volume') >>> u.get_physical_type(1) PhysicalType('dimensionless') Some units are dimensionally compatible with multiple physical types. A pascal is intended to represent pressure and stress, but the unit decomposition is equivalent to that of energy density. >>> pressure = u.get_physical_type("pressure") >>> pressure PhysicalType({'energy density', 'pressure', 'stress'}) >>> 'energy density' in pressure True Physical types can be tested for equality against other physical type objects or against strings that may contain the name of a physical type. >>> area = (u.m ** 2).physical_type >>> area == u.barn.physical_type True >>> area == "area" True Multiplication, division, and exponentiation are enabled so that physical types may be used for dimensional analysis. >>> length = u.pc.physical_type >>> area = (u.cm ** 2).physical_type >>> length * area PhysicalType('volume') >>> area / length PhysicalType('length') >>> length ** 3 PhysicalType('volume') may also be performed using a string that contains the name of a physical type. >>> "length" * area PhysicalType('volume') >>> "area" / length PhysicalType('length') Unknown physical types are labelled as ``"unknown"``. >>> (u.s ** 13).physical_type PhysicalType('unknown') Dimensional analysis may be performed for unknown physical types too. >>> length_to_19th_power = (u.m ** 19).physical_type >>> length_to_20th_power = (u.m ** 20).physical_type >>> length_to_20th_power / length_to_19th_power PhysicalType('length') cCs0t||_|j|_t||_t|j|_dS)N)r-_unitr'_physical_type_idr:_physical_typesorted_physical_type_list)selfr*Zphysical_typesr#r#r$__init__4s   zPhysicalType.__init__ccs|jEdHdS)N)r?)r@r#r#r$__iter__:szPhysicalType.__iter__cs^|dkrt|tt||d}tt||rNd|d}t|t|St|dS)Nr?z-support for accessing str attributes such as za from PhysicalType instances is deprecated since 4.3 and will be removed in a subsequent release.)super__getattribute__getattrr7hasattrwarningswarnr )r@attrZ self_str_attrZwarning_message) __class__r#r$ __getattr__=s   zPhysicalType.__getattr__cCs<t|tr|j|jkSt|tr4t|}||jStSdS)z Return `True` if ``other`` represents a physical type that is consistent with the physical type of the `PhysicalType` instance. N)r6r r<r7r:issubsetr=NotImplemented)r@otherr#r#r$__eq__Ps     zPhysicalType.__eq__cCs||}t|tr| StS)N)rOr6boolrM)r@rNZequalityr#r#r$__ne__]s zPhysicalType.__ne__cCsdt|jdddS)N{rr})r7r?)r@r#r#r$_name_string_as_ordered_setasz(PhysicalType._name_string_as_ordered_setcCs6t|jdkr"d|jdd}n|}d|dS)Nr'rz PhysicalType())lenr=r?rT)r@namesr#r#r$__repr__dszPhysicalType.__repr__cCs d|jS)Nr.)joinr?)r@r#r#r$__str__kszPhysicalType.__str__cCsZt|tjrt|St|tr$|jSt|tjr>|dkr>tjSt|t rRt |jSt SdS)a Return a unit that corresponds to the provided argument. If a unit is passed in, return that unit. If a physical type (or a `str` with the name of a physical type) is passed in, return a unit that corresponds to that physical type. If the number equal to ``1`` is passed in, return a dimensionless unit. Otherwise, return `NotImplemented`. rN) r6rUnitBaser-r r;numbersRealdimensionless_unscaledr7r%rM)objr#r#r$_dimensionally_compatible_unitns    z+PhysicalType._dimensionally_compatible_unitcCs4||}|tkrtSt|}t|j||}|jS)N)rarMr-rEr; physical_type)r@rNZ operationZ other_unitZnew_unitr#r#r$_dimensional_analysiss  z"PhysicalType._dimensional_analysiscCs ||dS)N__mul__)rc)r@rNr#r#r$rdszPhysicalType.__mul__cCs ||S)N)rd)r@rNr#r#r$__rmul__szPhysicalType.__rmul__cCs ||dS)N __truediv__)rc)r@rNr#r#r$rfszPhysicalType.__truediv__cCs$||}|tkrtS|j|dS)Nrf)rarMrbrc)r@rNr#r#r$ __rtruediv__s zPhysicalType.__rtruediv__cCs |j|jS)N)r;rb)r@rr#r#r$__pow__szPhysicalType.__pow__cCs t|jS)N)hashr<)r@r#r#r$__hash__szPhysicalType.__hash__cCs t|jS)N)rWr=)r@r#r#r$__len__szPhysicalType.__len__N)__name__ __module__ __qualname____doc__rArBrKrOrQrTrYr[ staticmethodrarcrdrerfrgrhrjrkZ __array__ __classcell__r#r#)rJr$r s&h   c Cs|}t|}d|kr tdttt|i}||@}|rVtd|d|tk}|rt|}|t|O}| ||nt ||}|t|<x|D] }|t|<qWx|D] } |t | <qWdS)a Add a mapping between a unit and the corresponding physical type(s). If a physical type already exists for a unit, add new physical type names so long as those names are not already in use for other physical types. Parameters ---------- unit : `~astropy.units.Unit` The unit to be represented by the physical type. name : `str` or `set` of `str` A `str` representing the name of the physical type of the unit, or a `set` containing strings that represent one or more names of physical types. Raises ------ ValueError If a physical type name is already in use for another unit, or if attempting to name a unit as ``"unknown"``. rz/cannot uniquely define an unknown physical typez6the following physical type names are already in use: .N) r'r:r r8_unit_physical_mappingkeys difference_physical_unit_mappinggetrAr r!) r*r"r+Zphysical_type_namesZnames_for_other_unitsZnames_already_in_useZunit_already_in_userbptypeZ ptype_namer#r#r$r s*      c Cst|tr|St|tr t|Sy$t|tjr2|ntj|ddj}Wn2t k rv}zt |d|Wdd}~XYnXt |}| }|t k}|rt |St|dSdS)a@ Return the physical type that corresponds to a unit (or another physical type representation). Parameters ---------- obj : quantity-like or `~astropy.units.PhysicalType`-like An object that (implicitly or explicitly) has a corresponding physical type. This object may be a unit, a `~astropy.units.Quantity`, an object that can be converted to a `~astropy.units.Quantity` (such as a number or array), a string that contains a name of a physical type, or a `~astropy.units.PhysicalType` instance. Returns ------- `~astropy.units.PhysicalType` A representation of the physical type(s) of the unit. Examples -------- The physical type may be retrieved from a unit or a `~astropy.units.Quantity`. >>> import astropy.units as u >>> u.get_physical_type(u.meter ** -2) PhysicalType('column density') >>> u.get_physical_type(0.62 * u.barn * u.Mpc) PhysicalType('volume') The physical type may also be retrieved by providing a `str` that contains the name of a physical type. >>> u.get_physical_type("energy") PhysicalType({'energy', 'torque', 'work'}) Numbers and arrays of numbers correspond to a dimensionless physical type. >>> u.get_physical_type(1) PhysicalType('dimensionless') F)copyz( does not correspond to a physical type.Nr) r6r r7r%rr\r ZQuantityr* TypeErrorr-r'rv)r`r*excr+Zunit_has_known_physical_typer#r#r$r s+  $"z&.. list-table:: Defined Physical Typesz :header-rows: 1z :widths: 30 10 50z * - Physical typez - Unitz- - Other physical type(s) with same unitz * - _``z - :math:`latexz - z, cCsg|]}|tkr|qSr#)r")r3nr#r#r$ 7srz  )Hror]rGr|rrrrrrr Zastropy.utils.exceptionsr __all__r_mr4ZradsrZHzgZmolr&WJZkgNZPaACVZOhmSFZWbTHcdZlmlxZJyZphotoncmZAARbitZFranklinZ statampereZBiotZ abcoulombZelectronZlumenZ_units_and_physical_typesrvrsr!r%r-r:r r r r*rbZdoclinesr>rtr"extendr;Z to_stringrZr#r#r#r$s$                        c6@  $