B ~d;!@sdZddlZddlZddlmZddlmZddlmZddl Z ddl m Z ddl Z ddl mZddlm Zd Ze jd e jd e jd e jd e jde jde jde jde jde jdi ZddeDZe je je j e je j!e je j"e je j#e je j$e je j%e je j&e je j'e je j(e ji Z)dde)DZ*dde*DZ+e,dZ-ddZ.e fddZ/ddZ0e j1e j2e j3e j4e j5e 6de j7gZ8d d!Z9d"d#Z:d$d%Z;ed&d'e j= 6.14.0 N) OrderedDict)Fraction)reduce)units)to_gpsz(Duncan Macleod ZINT2ZINT4ZINT8ZUINT2ZUINT4ZUINT8ZREAL4ZREAL8ZCOMPLEX8Z COMPLEX16cCsi|]\}}||qSr).0kvrr[/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/gwpy/utils/lal.py :sr cCsi|]\}}t||qSr) LAL_TYPE_STR)r r r rrr r IscCsi|]\}}||qSrr)r r r rrr r Ksz(U?INT|REAL|COMPLEX)\d+c CsV|tkr |S|tkrt|Syt|}t|jSttfk rPtdYnXdS)aConvert the input python type to a LAL type string Examples -------- To convert a python type: >>> from gwpy.utils.lal import to_lal_type_str >>> to_lal_type_str(float) 'REAL8' To convert a `numpy.dtype`: >>> import numpy >>> to_lal_type_str(numpy.dtype('uint32')) 'UINT4' To convert a LAL type code: >>> to_lal_type_str(11) 'REAL8' Raises ------ KeyError if the input doesn't map to a LAL type string z%Failed to map {!r} to LAL type stringN) LAL_TYPE_FROM_STRrnumpydtypeLAL_TYPE_STR_FROM_NUMPYtype TypeErrorKeyError ValueError)pytyperrrr to_lal_type_strPs  rcCst|}t|d|||S)a7Returns the lal method for the correct type Parameters ---------- pytype : `type`, `numpy.dtype` the python type, or dtype, to map prefix : `str` the function name prefix (before the type tag) suffix : `str` the function name suffix (after the type tag) Raises ------ AttributeError if the function is not found Examples -------- >>> from gwpy.utils.lal import find_typed_function >>> find_typed_function(float, 'Create', 'Sequence') z {0}{1}{2})rgetattrformat)rprefixsuffixmodulelaltyperrr find_typed_function{srcCsLt|tst|}|j}t|}|r2|dtkr@t|dt|dS)aConvert the data type of a LAL instance or type into a numpy data type. Parameters ---------- laltype : `SwigPyObject` or `type` the input LAL instance or type Returns ------- dtype : `type` the numpy data type, such as `numpy.uint32`, `numpy.float64`, etc. Raises ------ ValueError if the numpy data type cannot be inferred from the LAL object rz# has no known numpy type equivalent) isinstancer__name__LAL_TYPE_REGEXmatchLAL_NUMPY_FROM_TYPE_STRr)rnamer#rrr from_lal_types  r&Zstrainc Cst|trt|}|}t|j}|rN|rNt dt |}d}nt }|j}xt |j |j D]p\}}yt|}Wn4tk r}zd|df|_Wdd}~XYnXt|}|j|j|<|jd|j|<qlW||fS)auConvert the input unit into a `lal.Unit` and a scaling factor. Parameters ---------- aunit : `~astropy.units.Unit`, `str` the input unit Returns ------- unit : `lal.Unit` the LAL representation of the base unit. scale : `float` the linear scaling factor that should be applied to any associated data, see _Notes_ below. Notes ----- Astropy supports 'scaled' units of the form `` `` ere ```` is a `float` and ```` the base `astropy.units.Unit`, e.g. ``'123 m'``, e.g: >>> from astropy.units import Quantity >>> x = Quantity(4, '123m') >>> print(x) 4.0 123 m >>> print(x.decompose()) 492.0 m LAL doesn't support scaled units in this way, so this function simply returns the scaling factor of the unit so that it may be applied manually to any associated data. Examples -------- >>> print(to_lal_unit('m**2 / kg ** 4')) (m^2 kg^-4, 1.0) >>> print(to_lal_unit('123 m')) (m, 123.0) Raises ------ ValueError if LAL doesn't understand the base units for the input z10^z"LAL has no unit corresponding to ''N)r strrUnitZ decomposerlog10scale is_integerlalintzipbasesZpowersLAL_UNIT_INDEXindexrargsr numerator unitNumerator denominatorunitDenominatorMinusOne) ZaunitZpow10lunitr,basepoweriexcfracrrr to_lal_units&/     r?cCs.ttjddtt|j|jDd|jS)aConvert a LALUnit` into a `~astropy.units.Unit` Parameters ---------- lunit : `lal.Unit` the input unit Returns ------- unit : `~astropy.units.Unit` the Astropy representation of the input Raises ------ TypeError if ``lunit`` cannot be converted to `lal.Unit` ValueError if Astropy doesn't understand the base units for the input css4|],\}\}}t|tt|t|dVqdS)r'N)r2rr/)r r<numZdenrrr #sz from_lal_unit.. )roperatormul enumerater0r6r8Z powerOfTen)r9rrr from_lal_unit srFcCst|}t|j|jS)aeConvert the given GPS time to a `lal.LIGOTimeGPS` object Parameters ---------- gps : `~gwpy.time.LIGOTimeGPS`, `float`, `str` input GPS time, can be anything parsable by :meth:`~gwpy.time.to_gps` Returns ------- ligotimegps : `lal.LIGOTimeGPS` a SWIG-LAL `~lal.LIGOTimeGPS` representation of the given GPS time )rr.Z LIGOTimeGPSZ gpsSecondsZgpsNanoSeconds)Zgpsrrr to_lal_ligotimegps,s rGccs|]}|jj|jfVqdS)N)Z frDetectorr)r Ziforrr rA?srA)>__doc__rCre collectionsr fractionsr functoolsrrZastropyrr.timerdetectorZ gwpy_units __author__Z I2_TYPE_CODEZ I4_TYPE_CODEZ I8_TYPE_CODEZ U2_TYPE_CODEZ U4_TYPE_CODEZ U8_TYPE_CODEZ S_TYPE_CODEZ D_TYPE_CODEZ C_TYPE_CODEZ Z_TYPE_CODEritemsrZint16Zint32Zint64Zuint16Zuint32Zuint64Zfloat32Zfloat64Z complex64Z complex128ZLAL_TYPE_FROM_NUMPYrr$compiler"rrr&ZmeterZkilogramsecondZampereZKelvinr*countr2r?rFrGZCachedDetectorsZ LAL_DETECTORSrrrr sf          + L