IR-e#dZddlZddlZddlmZddlZddlmZeeZ edde j zz Z edde j zzZ dZdZd Zd Zd Zd ZddZdZdZddZdS)z} Miscellaneous utilities for `astropy.units`. None of the functions in the module are meant for use outside of the package. N)Fraction)finfo?g@ctjd|}||d}|ddS)zS Get the first sentence from a string and remove any carriage returns. z .*?\S\.\sNr  )rematchgroupreplace)sxs 3lib/python3.11/site-packages/astropy/units/utils.py_get_first_sentencers= q!!A} GGAJJ 99T3  c#Kddlm}g}t}|D]|\}}t ||js||jkr't ||jr+||j j djg| |}| d|D]}t|j}d}t ||jr&d|j ddd d }d d |jD} |||| |j|vrd ndfVdS)z Generates the ``(unit, doc, represents, aliases, prefixes)`` tuple used to format the unit summary docs in `generate_unit_summary`. )corerc4|jSN)namelower)rs rz$_iter_unit_summary..>sQV\\^^r)keyz:math:`latex`z, c3"K|] }d|dV dS)z``N).0rs r z%_iter_unit_summary..Es*==1JJJJ======rYesNoN)rrsetitems isinstanceUnitBaser PrefixUnitadd _representsbasesappendsortr__doc__stripUnit to_stringjoinaliases) namespacerunits has_prefixesrvalunitdoc representsr4s r_iter_unit_summaryr<"s  E55LOO%%S#t}--   #(??  c4? + +    S_215: ; ; ; ; LL     JJ++J,,,   !$,//5577 dDI & & PO4#3#=#=g#F#Fqt#LOOOJ))== =====    Y,..EED         rctj}|dd}t|D]}||j| |S)a` Generates a summary of units from a given namespace. This is used to generate the docstring for the modules that define the actual units. Parameters ---------- namespace : dict A namespace containing units. Returns ------- docstring : str A docstring containing a summary table of the units. z .. list-table:: Available Units :header-rows: 1 :widths: 10 20 20 20 1 * - Unit - Description - Represents - Aliases - SI Prefixes z7 * - ``{}`` - {} - {} - {} - {} )ioStringIOwriter<formatgetvalue)r5 docstringtemplate unit_summarys rgenerate_unit_summaryrFPsz  I OO    H+95588 67777     rcNddlm}i}|D]1\}}t||r|jjd}|||j<2tj}d}t|D]}| |j | | S)a Generates table entries for units in a namespace that are just prefixes without the base unit. Note that this is intended to be used *after* `generate_unit_summary` and therefore does not include the table header. Parameters ---------- namespace : dict A namespace containing units that are prefixes but do *not* have the base unit in their namespace. Returns ------- docstring : str A docstring containing a summary table of the units. r)r)rzO * - Prefixes for ``{}`` - {} prefixes - {} - {} - Only ) rr)r&r'r;r,rr>r?r<r@rArB) r5r)faux_namespacenmr9 base_unitrCrDrEs r generate_prefixonly_unit_summaryrK|s"NOO%%77D dJ ' ' 7-a0I-6N9> * IH+>::88 67777     rc t|cxko tkncS#t$rDt|jcxko tknco!t|jdzcxko tknccYSwxYw)Nr)_JUST_BELOW_UNITY_JUST_ABOVE_UNITY TypeErrorrealimag)values ris_effectively_unityrSs  E>>>>->>>>>>     @ @ @ @/@ @ @ @ @ I!UZ!^HHHH7HHHHH    sA A*)A*ct|rdS|jtur|St|dr|}|jrt |jt |jkr't|j|jz dzr|jSn4t|j|jz dzrtd|jS|S|jS)Nrdtyperg) rS __class__floathasattritemrQabsrPcomplex)scales rsanitize_scaler]sE""s % ug  z  uz??S__ , ,#EJ$;a$?@@ "z! ""%*uz"9A"= > > ,3 ++ + zrdc:|dks|jtur|S|\}}||z}d\}}|d}}||krYt|||zz cxkr tkrnnt ||S||||zz }}||z}||||zz}}||||zz}}||kY|S)aSFraction very close to x with denominator at most max_denominator. The fraction has to be such that fraction/x is unity to within 4 ulp. If such a fraction does not exist, returns the float number. The algorithm is that of `fractions.Fraction.limit_denominator`, but sped up by not creating a fraction to start with. r)rrr)rVintas_integer_ratiorMrNr) pmax_denominatorndan0d0n1d1s rmaybe_simple_fractionrks Avv##    DAq QA FB B    b1f B B B B1B B B B B BB## #!a!e)1 FR!b&[BR!b&[B    Hrct|dd}| t|}ny#t$rl tj|}|jdx}|krt|}ntdn#t$rwxYwYnwxYwt|}t|tr|S|j }|dkr|j }n||dz zdkrt|}|S)aConvert a power to a floating point value, an integer, or a Fraction. If a fractional power can be represented exactly as a floating point number, convert it to a float, to make the math much faster; otherwise, retain it as a `fractions.Fraction` object to avoid losing precision. Conversely, if the value is indistinguishable from a rational number with a low-numbered denominator, convert to a Fraction object. Parameters ---------- p : float, int, Rational, Fraction Power to be converted denominatorNrz9Quantities and Units may only be raised to a scalar powerr) getattrrW Exceptionnp asanyarrayflatall ValueErrorrkr'rm numerator)rbdenomfirsts rvalidate_powerrxs4 A}d + +E } aAA    M!$$fQi'UA-2244e AA$S      A  "! $ $ a   H  zz K 519 ! # # !HH Hs'% BAB B BBBc|jtuo"|jtuot|t}|jtuo"|jtuot|t}|r|st |}n|s|rt |}||fS)z If either input is a Fraction, convert the other to a Fraction (at least if it does not have a ridiculous denominator). This ensures that any operation involving a Fraction will use rational arithmetic and preserve precision. )rVr`rWr'rrk)rfb a_is_fraction b_is_fractions rresolve_fractionsr}s 3W1;e#;W 1h@W@W 3W1;e#;W 1h@W@W%]% !! $ $ %}% !! $ $ a4Krc ddlmt|tjs.5s-33A 1h''333333r)rU) quantityrr'rpndarrayisscalaranyinexactfloat64rq)rfrUrs @rquantity_asanyarrayr/s"""""" q"* % % - A - 3333333 3 3 - x''''$rz11 u}Qe,,,,r)r^r)r/r>r fractionsrnumpyrprrW _float_finfoepsnegrMepsrNrr<rFrKrSr]rkrxr}rr rrrsJ uU|| E#l&9 99::E#l&6 6677   + + + \) ) ) X$ $ $ N    :    6/ / / d, - - - - - -r