B Afxa8 @sUdZddlZddlmZmZddlmZmZddlm Z e dZ dd Z d Z d d Zd dddddedd d ddddddddd d dddddd ddd d!ddd dd"dd#ddd$Zeeefed%<iZeeefed&<d'd(Zed)d*d+Zed,d-d.Zed/d0d1Zed2d3d4Zed5d6d7Zed8d9d:Zd ddd d!d;dd?d f d@dAZedBZdCdDZdZdEdFZdGdHZ dIdJZ!dKdLZ"dMd=fdNdOZ#dPd=fdQdRZ$dSd=dTfdUdVZ%dWd=dTfdXdYZ&dS)[z pint.formatter ~~~~~~~~~~~~~~ Format units for pint. :copyright: 2016 by Pint Authors, see AUTHORS for more details. :license: BSD, see LICENSE for more details. N)CallableDict)_babel_lengths _babel_units) babel_parsez{\d*}cCsN|sdSt|s||St|}t|}x|D]}|||}|}q2W|S)a1Join an iterable with the format specified in fmt. The format can be specified in two ways: - PEP3101 format with two replacement fields (eg. '{} * {}') - The concatenating string (eg. ' * ') Parameters ---------- fmt : str iterable : Returns ------- str )__JOIN_REG_EXPsearchjoiniternextformat)fmtiterableZmiterfirstvalretr.lib/python3.7/site-packages/pint/formatting.py_joins    ru⁰¹²³⁴⁵⁶⁷⁸⁹cCsB|ddddd}x$tdD]}|t|t|}q"W|S)zFormat an number into a pretty printed exponent. Parameters ---------- num : int Returns ------- str n-u⁻.u⋅ )replacerangestr_PRETTY_EXPONENTS)Znumrrrrr_pretty_fmt_exponent6s rTF·/z{}{}z({}))as_ratiosingle_denominator product_fmt division_fmt power_fmtparentheses_fmtexp_callz \cdot z \frac[{}][{}]z{}^[{}]z\left({}\right))r"r#r$r%r&r'rz +- )ZsioptsZpm_fmt z{}/{}z{}{}z * z / z{} ** {}*z{}**{})PLLxHrC_FORMATS _FORMATTERScsfdd}|S)aregister a function as a new format for units The registered function must have a signature of: .. code:: python def new_format(unit, registry, **options): pass Parameters ---------- name : str The name of the new format (to be used in the format mini-language). A error is raised if the new format would overwrite a existing format. Examples -------- .. code:: python @pint.register_unit_format("custom") def format_custom(unit, registry, **options): result = "" # do the formatting return result ureg = pint.UnitRegistry() u = ureg.m / ureg.s ** 2 f"{u:custom}" cs&tkrtddd|t<dS)Nzformat z!rz already exists)r1 ValueError)func)namerrwrappersz%register_unit_format..wrapperr)r4r5r)r4rregister_unit_format|s r6r+c Ks$t|fddddddtd|S)NTFr r!z{}{}z({}))r"r#r$r%r&r'r() formatteritemsr)unitregistryoptionsrrr format_prettysr<r,c KsHdd|D}t|fddddddd|}|d d d d S) Nc Ss$i|]\}}|d|ddqS)z \mathrm{{{}}}_z\_)rr).0uprrr sz format_latex..Tz \cdot z \frac[{}][{}]z{}^[{}]z\left({}\right))r"r#r$r%r&r'[{]})r8r7r)r9r:r;Z preprocessed formattedrrr format_latexs rGr-cKs&|dkrtdt||}d|dS)NzCan't format as siunitx without a registry. This is usually triggered when formatting a instance of the internal `UnitsContainer` with a spec of `"Lx"` and might indicate a bug in `pint`.z\si[]{rE)r2siunitx_format_unit)r9r:r;rFrrrformat_latex_siunitxs  rIr.c Ks"t|fddddddd|S)NTr)z{}/{}z{}{}z({}))r"r#r$r%r&r')r7r8)r9r:r;rrr format_htmlsrJDc Ks"t|fddddddd|S)NTFz * z / z{} ** {}z({}))r"r#r$r%r&r')r7r8)r9r:r;rrrformat_defaultsrLr/c Ks"t|fddddddd|S)NTFr*r!z{}**{}z({}))r"r#r$r%r&r')r7r8)r9r:r;rrrformat_compactsrMz({0})cCs|dS)Nrr)xrrrrOZlongonec s|sdS|rfdd} n} gg} }| r4t|}xX|D]N\}}|r"r"| r"|tkr"t|}t|}|jd}|jd}t|dkrdn| }tkrfdd ttD}ng}xFg|D]8}||i|i|}|d k r|d d }PqW|d i|}d }t |dkr8| |q<|dkrZ| | || |q<|dkrv|rv| |q<| | || |q. unit_patternscompound_unit_patternsrrQcsg|]}|kr|qSrr)r> _babel_length) babel_lengthrr Dszformatter..Nz{0}Zperz{}{}r1)sortedrr_datarRrreversedgetrstriprappendrrlen)r8r"r#r$r%r&r'r(localerVZbabel_plural_formsortZfunZ pos_termsZ neg_termskeyvalueZ_keyrSrTpluralZ other_lengthsrUZpatZpos_retZneg_retr)rVr(rr7s\3            r7zbcdeEfFgGnosxX%uScCspd}xft|D]Z}|dks|tkr&qq|ttdgkrP|rJtdqh|}q|rftd|qPqW|S)Nr~z#expected ':' after format specifierzUnknown conversion specified )r\ _BASIC_TYPESlistr1keysr2isalpha)specresultZchrrr _parse_specxs rmcKsR|s|drdSdS|sd}t|}|dkr>td|||fd|i|S)N%rZ dimensionlessrKzUnknown conversion specified: r:)endswithr1r]r2)r9rkr:r;rrrr format_units  rpc Csdd}g}g}xt|D]\}}|dkr2|n|}d}xF|jD]8} t| } t| dkrF|| dkrF| }||dd}qFW|dkr|d|dk r|d ||d ||d |t |qWd |d |S) zHReturns LaTeX code for the unit that can be put into an siunitx command.cSsdt|tst|trP|rP|dkr(dS|dkr4dS|dkr@dSdt|Snd|d SdS) Nrrz\squaredz\cubedz\tothe{{{:d}}}z\tothe{{{:.3f}}}0) isinstanceintfloat is_integerrrstrip)powerrrr_tothesz#siunitx_format_unit.._totherNrrz\perz\{}z{}) rZr8Z _prefixesvaluesrr`findrr_rrRr ) Zunitsr:rzlposZlnegr9ryZlpickprefixr@rrrrHs$ rHcCsBddl}|ddttdgd}||}d|S)Nr(|rf)r)recompiler rhr1rifindall)rkrZflag_reZ custom_flagsrrrextract_custom_flagss& rcCs0x*ttdgD]}|r||d}qW|S)Nrfr)rhr1rir)rkflagrrrremove_custom_flagssrcCs t|dS)Nz.2f)r)rNrrrrOrPcCs t|g|S)N)matrix_to_latex)Zvecfmtfunrrrvector_to_latexsrcCs t|dS)Nz.2f)r)rNrrrrOrPcs>g}x*|D]"}|dfdd|Dg7}q Wdd|S)Nz & c3s|]}|VqdS)Nr)r>f)rrr sz"matrix_to_latex..z\begin{pmatrix}%s\end{pmatrix}z\\ )r )Zmatrixrrrowr)rrrs "rcCs t|dS)Nz.2f)r)rNrrrrOrPrcst|tr|fdd}|jdkr:|d}t||gS|jdkrPt||gS|jdkrft||gSg}|jdkrddd d |Dd }x^t|D]$\}}|||d t||g7}qWn,x*t|D]\}}|t||||f7}qW|SdS) Ncs t|S)N)r)rN)rrrrOrPz(ndarray_to_latex_parts..rrrqrrzarr[%s,,css|]}d|VqdS)z%dNr)r>drrrrsz)ndarray_to_latex_parts..z%d,:,:]z = ) rtrndimZreshaperrr enumeratendarray_to_latex_parts)ndarrrdimZ_ndarrrheaderZelnoelr)rrrs$          "rcCs t|dS)Nz.2f)r)rNrrrrOrPcCsdt|||S)N )r r)rrrrrrndarray_to_latexsr)N)'__doc__rtypingrrZ babel_namesrrcompatrrr rrrr0rdict__annotations__r1r6r<rGrIrJrLrMr7 frozensetrgrmrprHrrrrrrrrrr s  ' m *