U
e{j6 @ sf d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z G dd de j
ZdS ) z
Handles units in `Office of Guest Investigator Programs (OGIP)
FITS files
`__.
N)Fraction)parsing )coregenericutilsc @ s e Zd ZdZdZedd Zedd Zedd Z edd
d Z
edd d
Z ed ddZ edd Z
edd Zedd Zedd ZdS ) OGIPz
Support the units in `Office of Guest Investigator Programs (OGIP)
FITS files
`__.
) DIVISIONZ
OPEN_PARENZ CLOSE_PAREN
WHITESPACEZSTARSTARSTARZSIGNZUFLOATZLIT10ZUINTUNKNOWNZUNITc C s ddl m} i }t }ddddddd d
d d d
ddddddddddddddd d g}g }d d d d!d"d#d$dd%d&d'd(d)d*d+d,dd-d.d/d0g}|| D ]0}|D ]&}|| }t|rqt| |||< qq|D ] }|D ]}||| qqd1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCg} | D ]}
t| |
||
< q| jdDgdEdFdG} | dH| } | |dD< | |dI< dDdIg}
|
D ]}
||
qbdJdKdLdMdNdOdPdQdRdSdTdUdVg
}|D ]}|||< q|||fS )WNr )unitsACcdZeVFgHZHzJZJyKZlmlxmZmolNZohmZPaZpcZradsSsrTVWZWbyzafpnucd dahkMGPEZYZangstromZarcminZarcsecZAUZbarnbinbyteZchancountdaydegZergZlyrmagminZphotonZpixelZvoxelyrCrabFzCrab (X-ray flux))prefixesdocgMbP?mCrabloglnexpsqrtsincostanasinacosatansinhcoshtanh) Zastropyr
setkeyword iskeywordgetattraddZdef_unitUnit)r% namesZdeprecated_namesbasesZdeprecated_basesr; baseprefixkeyZ simple_unitsunitr: r= Zdeprecated_unitsZ functionsname rX X/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/units/format/ogip.py_generate_unit_names4 s
zOGIP._generate_unit_namesc st j }d}d}d}d}d}d}dd }d d
} d d }
d
d } dd } dd }
fdd}d}dd }tjdddS )N/z\(z\)z[ ]+z\*\*z\*c S s t | j| _| S )zK(((\d+\.?\d*)|(\.\d+))([eE][+-]?\d+))|(((\d+\.\d*)|(\.\d+))([eE][+-]?\d+)?)floatvaluetrX rX rY t_UFLOATw s z"OGIP._make_lexer..t_UFLOATc S s t | j| _| S )z\d+)intr^ r_ rX rX rY t_UINT| s z OGIP._make_lexer..t_UINTc S s t | jd | _| S )z
[+-](?=\d)1r\ r_ rX rX rY t_SIGN s z OGIP._make_lexer..t_SIGNc S s | S )u [x×]rX r_ rX rX rY t_X s z OGIP._make_lexer..t_Xc S s dS )10
rX r_ rX rX rY t_LIT10 s z!OGIP._make_lexer..t_LIT10c S s dS )z [Uu][Nn][Kk][Nn][Oo][Ww][Nn]NrX r_ rX rX rY t_UNKNOWN s z#OGIP._make_lexer..t_UNKNOWNc s | | _| S )z[a-zA-Z][a-zA-Z_]*)Z _get_unitr^ r_ clsrX rY t_UNIT s z OGIP._make_lexer..t_UNITr( c S s t d| j d S )NzInvalid character at col )
ValueErrorZlexposr_ rX rX rY t_error s z!OGIP._make_lexer..t_errorZ ogip_lextab
astropy/units)Zlextabpackage)_tokensr lex)rl tokensZ
t_DIVISIONZ t_OPEN_PARENZ
t_CLOSE_PARENZ t_WHITESPACEZ
t_STARSTARZt_STARra rc re rf ri rj rm Zt_ignorero rX rk rY _make_lexerj s" zOGIP._make_lexerc s j }dd }dd }dd } fdd}d d
}d d }d
d }dd } dd }
dd } dd } dd }
dd }dd }tjd d d S ) a
The grammar here is based on the description in the
`Specification of Physical Units within OGIP FITS files
`__,
which is not terribly precise. The exact grammar is here is
based on the YACC grammar in the `unity library
`_.
c S sT t | dkr"| d | d | d< n.t | dkrD| d | d | d< n | d | d< dS )z
main : UNKNOWN
| complete_expression
| scale_factor complete_expression
| scale_factor WHITESPACE complete_expression
r r Nlenr# rX rX rY p_main s
z!OGIP._make_parser..p_mainc S s | d | d< dS )z@
complete_expression : product_of_units
r r NrX r{ rX rX rY p_complete_expression s z0OGIP._make_parser..p_complete_expressionc S sr t | dkrD| d dkr.| d | d | d< qn| d | d | d< n*t | dkrb| d d | d< n | d | d< dS ) a
product_of_units : unit_expression
| division unit_expression
| product_of_units product unit_expression
| product_of_units division unit_expression
rv rx r r rw r Nry r{ rX rX rY p_product_of_units s z-OGIP._make_parser..p_product_of_unitsc s t | d }| jkr2|dkr2td| d dt| dkr|dkrh| d | d d| d | d < n | d | d | d | d < nt| dkr| d
| d | d < nbt| d kr|dkr| d d | d < n| d | d | d < n&t| d kr| d
| d < n | d | d < d
S )a
unit_expression : unit
| UNIT OPEN_PAREN complete_expression CLOSE_PAREN
| OPEN_PAREN complete_expression CLOSE_PAREN
| UNIT OPEN_PAREN complete_expression CLOSE_PAREN power numeric_power
| OPEN_PAREN complete_expression CLOSE_PAREN power numeric_power
r rA zThe function 'z8' is valid in OGIP, but not understood by astropy.units. rw g ? r rx rv N)strZ
_functionsrn rz )r# Zp1_strrk rX rY p_unit_expression s$
" z,OGIP._make_parser..p_unit_expressionc S sh t | dkr d| d | d< n | d | d< t| d d dkrdddlm} td | d d
| d S ) a
scale_factor : LIT10 power numeric_power
| LIT10
| signed_float
| signed_float power numeric_power
| signed_int power numeric_power
rv rh rw r r ? ) UnitsWarning'.' scale should be a power of 10 in OGIP formatN)rz mathlog10Zastropy.units.corer warningswarn)r# r rX rX rY p_scale_factor s z)OGIP._make_parser..p_scale_factorc S s d| d< dS )z
division : DIVISION
| WHITESPACE DIVISION
| WHITESPACE DIVISION WHITESPACE
| DIVISION WHITESPACE
r r NrX r{ rX rX rY
p_division
s z%OGIP._make_parser..p_divisionc S s d| d< dS )z
product : WHITESPACE
| STAR
| WHITESPACE STAR
| WHITESPACE STAR WHITESPACE
| STAR WHITESPACE
ZPRODUCTr NrX r{ rX rX rY p_product s z$OGIP._make_parser..p_productc S s d| d< dS )z*
power : STARSTAR
ZPOWERr NrX r{ rX rX rY p_power s z"OGIP._make_parser..p_powerc S s2 t | dkr"| d | d | d< n | d | d< dS )zQ
unit : UNIT
| UNIT power numeric_power
rv r rw r Nry r{ rX rX rY p_unit& s z!OGIP._make_parser..p_unitc S sV t | dkr,tt| d t| d | d< n&t | dkrF| d | d< n | d | d< dS )a#
numeric_power : UINT
| signed_float
| OPEN_PAREN signed_int CLOSE_PAREN
| OPEN_PAREN signed_float CLOSE_PAREN
| OPEN_PAREN signed_float division UINT CLOSE_PAREN
r rx rv r r N)rz r rb r{ rX rX rY p_numeric_power0 s
z*OGIP._make_parser..p_numeric_powerc S s& t | dkr| d | d< nd| d< dS )z8
sign : SIGN
|
rx r r r Nry r{ rX rX rY p_sign? s z!OGIP._make_parser..p_signc S s | d | d | d< dS )z0
signed_int : SIGN UINT
r rx r NrX r{ rX rX rY p_signed_intI s z'OGIP._make_parser..p_signed_intc S s | d | d | d< dS )zY
signed_float : sign UINT
| sign UFLOAT
r rx r NrX r{ rX rX rY p_signed_floatO s z)OGIP._make_parser..p_signed_floatc S s
t d S )N)rn r{ rX rX rY p_errorV s z"OGIP._make_parser..p_errorZ
ogip_parsetabrp )Z tabmodulerq )rr r Zyacc)rl rt r| r} r r r r r r r r r r r r rX rk rY _make_parser s &
zOGIP._make_parserTc C sb || j kr<|r6td| dt|| j | j| j nt || jkr^t|| j | d| j d S )NzUnit 'z&' not supported by the OGIP standard. r )_unitsrn r Zdid_you_mean_unitsZ_deprecated_units_to_decomposed_alternativeZunit_deprecation_warningrl rV detailed_exceptionrX rX rY _validate_unit[ s(
zOGIP._validate_unitc C s | j ||d | j| S )Nr )r r r rX rX rY _parse_unitp s zOGIP._parse_unitFc
C s | }z| j|ddW S tk
r z t| jj|| j|dW Y S tk
r } z t|rf ntd| dW 5 d }~X Y nX Y nX d S )NFr )lexerdebugzSyntax error parsing unit 'r ) stripr rn r rP _parserparse_lexerr )rl r r erX rX rY r u s z
OGIP.parsec C s | d}| | |S )NZogip)Zget_format_namer )rl rV rW rX rX rY _get_unit_name s
zOGIP._get_unit_namec s g }|j fddd |D ]l\}}|dkr>| | q t|}d|krn| | d| d q | | d| q d |S )
Nc s | d S )Nr )r lower)xrk rX rY z(OGIP._format_unit_list..)rU r r[ z**()z** )sortappendr r Z format_powerjoin)rl r
outrS powerrX rk rY _format_unit_list s
zOGIP._format_unit_listc C sR t || j}t|tjrFt|jd dkrFt
d|j dtj t
| |S )Nr r r r )r decompose_to_known_unitsr
isinstancer
CompositeUnitr r scaler r r r
_to_string)rl rV rX rX rY to_string s zOGIP.to_stringc C sf t || j}t|tjr\t|jd dkr\|j}t |}d|_
t | | d| dS t |S )Nr r z (with data multiplied by r )
r r r r r r r r r copyZ_scaler r )rl rV r rX rX rY r s
z OGIP._to_decomposed_alternativeN)T)T)F)__name__
__module__ __qualname____doc__rr staticmethodrZ classmethodru r r r r r r r r rX rX rX rY r s.
5
5
;
r )r r rL r r fractionsr Z
astropy.utilsr r( r r r Genericr rX rX rX rY
s