IR-e5 | d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z G d de j
ZdS ) z
Handles units in `Office of Guest Investigator Programs (OGIP)
FITS files
`__.
N)Fraction)parsing )coregenericutilsc e Zd ZdZdZed Zed Zed Z edd Z
edd Z edd
Z e fd Z
ed Zed fd Ze fd Z xZS )OGIPz
Support the units in `Office of Guest Investigator Programs (OGIP)
FITS files
`__.
) DIVISION
OPEN_PAREN CLOSE_PAREN
WHITESPACESTARSTARSTARSIGNUFLOATLIT10UINTUNKNOWNUNITc
ddl m} i }t }g d}g }g d}||z D ]4}|D ]/}||z }t j | r t | | ||< 05|D ] }|D ]}| ||z g d} | D ]}
t | |
||
< | dgdd } | d
| z } | |d< | |d < dd g}
|
D ]}
| |
g d }|D ]}|||< |||fS )
Nr )units)ACcdeVFgHHzJJyKlmlxmmolNohmPapcradsSsrTVWWb)yzafpnur& cd dahkMGr0 PEZY)angstromarcminarcsecAUbarnbinbytechancountdaydegergrB r? lyrmagminphotonpixelvoxelyrCrabFzCrab (X-ray flux))prefixesdocgMbP?mCrab)
loglnexpsqrtsincostanasinacosatansinhcoshtanh) astropyr setkeyword iskeywordgetattradddef_unitUnit)r: namesdeprecated_namesbasesdeprecated_basesr[ baseprefixkey simple_unitsunitrZ r] deprecated_units functionsnames 9lib/python3.11/site-packages/astropy/units/format/ogip.py_generate_unit_nameszOGIP._generate_unit_names4 s &&&&&& 55
,, - -D"
-
- tm$S)) $Q__c
-
% 4 4D"
4
4 $$Vd]3333
4
! + +D!!T**E$KK zz6(U8KzLLv}%% f
g "G,$ ' 'D & & & &
D E$KK& 11 c j }d}d}d}d}d}d}d }d } d }
d
} d } d }
fd
}d}d }t j dd S )N/z\(z\)z[ ]+z\*\*z\*c 8 t | j | _ | S )zK(((\d+\.?\d*)|(\.\d+))([eE][+-]?\d+))|(((\d+\.\d*)|(\.\d+))([eE][+-]?\d+)?)floatvaluets r t_UFLOATz"OGIP._make_lexer..t_UFLOATw s AGnnAGHr c 8 t | j | _ | S )z\d+)intr r s r t_UINTz OGIP._make_lexer..t_UINT| s !'llAGHr c > t | j dz | _ | S )z
[+-](?=\d)1r r s r t_SIGNz OGIP._make_lexer..t_SIGN s AGcM**AGHr c | S )u [x×] r s r t_Xz OGIP._make_lexer..t_X s Hr c dS )10
r r s r t_LIT10z!OGIP._make_lexer..t_LIT10 s 2r c dS )z [Uu][Nn][Kk][Nn][Oo][Ww][Nn]Nr r s r t_UNKNOWNz#OGIP._make_lexer..t_UNKNOWN s 4r c < | | _ | S )z[a-zA-Z][a-zA-Z_]*) _get_unitr )r clss r t_UNITz OGIP._make_lexer..t_UNIT s mmA&&AGHr r= c 0 t d| j )NzInvalid character at col )
ValueErrorlexposr s r t_errorz!OGIP._make_lexer..t_error s C C CDD Dr ogip_lextab
astropy/units)lextabpackage)_tokensr lex)r tokens
t_DIVISION t_OPEN_PAREN
t_CLOSE_PAREN t_WHITESPACE
t_STARSTARt_STARr r r r r r r t_ignorer s ` r _make_lexerzOGIP._make_lexerj s
E E E {- IIIIr c j }d }d }d } fd}d }d }d }d } d }
d
} d } d }
d
}d }t j dd 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 t | dk r| d | d z | d< dS t | dk r| d | d z | d< dS | d | d< dS )z
main : UNKNOWN
| complete_expression
| scale_factor complete_expression
| scale_factor WHITESPACE complete_expression
r r Nlenr8 s r p_mainz!OGIP._make_parser..p_main se 1vv{{tad{!Q1tad{!t!r c | d | d< dS )z@
complete_expression : product_of_units
r r Nr r s r p_complete_expressionz0OGIP._make_parser..p_complete_expression s Q4AaDDDr c t | dk r8| d dk r| d | d z | d< dS | d | d z | d< dS t | dk r| d dz | d< dS | d | d< dS ) a
product_of_units : unit_expression
| division unit_expression
| product_of_units product unit_expression
| product_of_units division unit_expression
r r r r r r Nr r s r p_product_of_unitsz-OGIP._make_parser..p_product_of_units s 1vv{{Q4:%% Q4!A$;AaDDD Q4!A$;AaDDDQ1trz!t!r c \ t | d }|j v r |dk rt d| d d t | dk rG|dk r"| d | d d| d z z z | d < d
S | d | d | d z z | d < d
S t | dk r| d
| d z | d < d
S t | d k r,|dk r| d dz | d < d
S | d | d z | d < d
S t | d k r
| d
| d < d
S | 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. r g ? r r r N)str
_functionsr r )r8 p1_strr s r p_unit_expressionz,OGIP._make_parser..p_unit_expression s] 1YYF ''Ff,<,< (QqT ( ( (
1vv{{V## Q4!A$31:"66AaDDD Q4!A$!A$,.AaDDDQ1tqt|!Q1V## Q43;AaDDD Q4!A$;AaDDDQ1t!t!r c t | dk rd| d z | d< n | d | d< t j | d dz dk r'ddlm} t j d | d d
| d S d S ) a
scale_factor : LIT10 power numeric_power
| LIT10
| signed_float
| signed_float power numeric_power
| signed_int power numeric_power
r r r r r ? ) UnitsWarning'.' scale should be a power of 10 in OGIP formatN)r mathlog10astropy.units.corer warningswarn)r8 r s r p_scale_factorz)OGIP._make_parser..p_scale_factor s 1vv{{QqTz!t!z!A$ #%,,;;;;;;
L!LLL -,r c d| d< dS )z
division : DIVISION
| WHITESPACE DIVISION
| WHITESPACE DIVISION WHITESPACE
| DIVISION WHITESPACE
r r Nr r s r
p_divisionz%OGIP._make_parser..p_division s AaDDDr c d| d< dS )z
product : WHITESPACE
| STAR
| WHITESPACE STAR
| WHITESPACE STAR WHITESPACE
| STAR WHITESPACE
PRODUCTr Nr r s r p_productz$OGIP._make_parser..p_product s AaDDDr c d| d< dS )z*
power : STARSTAR
POWERr Nr r s r p_powerz"OGIP._make_parser..p_power s AaDDDr c n t | dk r| d | d z | d< dS | d | d< dS )zQ
unit : UNIT
| UNIT power numeric_power
r r r r Nr r s r p_unitz!OGIP._make_parser..p_unit$ s=
1vv{{tqt|!t!r c t | dk r;t t | d t | d | d< dS t | dk r
| d | d< dS | 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 r r r r N)r r r r s r p_numeric_powerz*OGIP._make_parser..p_numeric_power. so 1vv{{ AaD 3qt9955!Q1t!t!r c P t | dk r
| d | d< dS d| d< dS )z8
sign : SIGN
|
r r r r Nr r s r p_signz!OGIP._make_parser..p_sign= s1
1vv{{t!!r c . | d | d z | d< dS )z0
signed_int : SIGN UINT
r r r Nr r s r p_signed_intz'OGIP._make_parser..p_signed_intG s Q4!A$;AaDDDr c . | d | d z | d< dS )zY
signed_float : sign UINT
| sign UFLOAT
r r r Nr r s r p_signed_floatz)OGIP._make_parser..p_signed_floatM s
Q4!A$;AaDDDr c t N)r r s r p_errorz"OGIP._make_parser..p_errorT s ,, r
ogip_parsetabr ) tabmoduler )r r yacc)r r r r r r r r r r r r r r r r s ` r _make_parserzOGIP._make_parser s "# # # # # J *
|oOOOOr Tc || j vrJ|r:t d| dt j || j | j | j z t || j v r)t j || j | d| j d S d S )NzUnit 'z&' not supported by the OGIP standard. r
)_unitsr r did_you_mean_units_deprecated_units_to_decomposed_alternativeunit_deprecation_warningr r{ detailed_exceptions r _validate_unitzOGIP._validate_unitY s sz ! !!
# ITIII.
-6 !ll" 3( ( ( *cj&0N
) (r c J | || | j | S )Nr )r r r s r _parse_unitzOGIP._parse_unitn s) 44FGGGz$ r Fc N | } | |d S # t $ rm t j | j || j | cY S # t $ r(}t | r t d| d d }~ww xY ww xY w)NFr )lexerdebugzSyntax error parsing unit 'r ) stripr r r rr _parserparse_lexerr )r r- r es r r z
OGIP.parses s
GGII I ??1??? ? I I I
Iy!2!21CJe!2!T!TUUUUU
I
I
Iq66 I$%G1%G%G%GHHH
I Is, -
B$3A.+B$.
B 8#BB B$c t t | }| | |S r )super_get_unit_namer )r r{ r~ __class__s r r zOGIP._get_unit_name s3 ww%%d++ 4 r c d|v rd| dnd| S )Nr z**()z**r )r numbers r _format_superscriptzOGIP._format_superscript s" "%--V]&]]Br inlinec B t j || j }t |t j rHt
j |j dz dk r(t j
d|j dt j t
|| S )Nr r r r )fraction)r decompose_to_known_unitsr
isinstancer
CompositeUnitr r scaler r r r to_string)r r{ r
r s r r zOGIP.to_string s -dC4FGG dD . / / z$*%%+s22
R
RRR%
ww 999r c h t j || j }t |t j r]t
j |j dz dk r=|j }t j | }d|_
t j | | d| dS t | S )Nr r z (with data multiplied by r )r r r r r r
r r r copy_scaler
_to_stringr r )r r{ r r s r r z OGIP._to_decomposed_alternative s -dC4FGG dD . / / z$*%%+s22
y ! )#t44XXPUXXX ww &&&r )T)F)r )__name__
__module__ __qualname____doc__r staticmethodr classmethodr r r r r r r r r
__classcell__)r s @r r
r
s
G 32 32 \32j 3J 3J [3Jj vP vP [vPp [( [
I
I
I [
I [
C C [C : : : : : [ : ' ' ' ' [' ' ' ' 'r r
)r r rm r r fractionsr
astropy.utilsr r= r r r Genericr
r r r r s ! ! ! ! ! ! " " " " " " " " " "M' M' M' M' M'7? M' M' M' M' M'r