B Afxa/ @sJdZddlZddlmZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZdd lmZdd lmZmZerdd lmZdd lmZe d ZddZd ddZd!ddZddZd"de ede ee ddfe ede ee ddfeeedefgedeeffdddZ de eeddfee ge fdddZ!dS)#z pint.registry_helpers ~~~~~~~~~~~~~~~~~~~~~ Miscellaneous methods of the registry written as separate functions. :copyright: 2016 by Pint Authors, see AUTHORS for more details.. :license: BSD, see LICENSE for more details. N) signature) zip_longest) TYPE_CHECKINGCallableIterableTypeVarUnion)F)DimensionalityError)Quantity)UnitsContainerto_units_container) UnitRegistry)UnitTcCs:d}x$|D]\}}||||}qWt|dtiS)zConvert a unit compatible type to a UnitsContainer. Parameters ---------- original_units : a UnitsContainer instance. values_by_name : a map between original names and the new values. Returns ------- r _units)itemsgetattrr )Zoriginal_unitsvalues_by_nameqZarg_nameZexponentr4lib/python3.7/site-packages/pint/registry_helpers.py_replace_unitssrcCs8t|tr*d|kr*t|ddddfSt||dfS)a#Convert a unit compatible type to a UnitsContainer, checking if it is string field prefixed with an equal (which is considered a reference) Parameters ---------- a : registry : (Default value = None) Returns ------- UnitsContainer, bool =r TF) isinstancestrrsplit)aregistryrrr_to_units_container1sr c st}tttfdd|DxtD]\}\}}|dkrLq4q4|rt|dkr|\\}}|dkr||kr||||df|<q|q|q4|q4WxDD]<}|\}}t|tsqt||kstd||qWfdd}|S)Ncsg|]}t|qSr)r ).0arg)rrr Vsz$_parse_wrap_args..r TzfFound a missing token while wrapping a function: Not all variable referenced in %s are defined using !c s\tdd|D}i}x4D],}||}|||d<t|d|||<qWxbD]Z}||}t|d|dk sxt|t|d|t|dtit|d|||<qRWxD]}t|||jr|||j||j |d||<q|rt||t r4| ||}||j|j |d||<qt d |d||qW||fS)Ncss|] }|VqdS)Nr)r!valuerrr zsz7_parse_wrap_args.._converter..r _magnituderzA wrapped function using strict=True requires quantity or a string for all arguments with not None units. (error found for {}, {}))listrrAssertionError_convertr rr r&rrZparse_expression ValueErrorformat)uregvaluesstrict new_valuesrndxr$Z tmp_value) args_as_uc defs_args_ndxdependent_args_ndx unit_args_ndxrr _converterys4    $z$_parse_wrap_args.._converter) set enumeratelenraddrdictkeysr*) argsrZ defs_argsr0r"is_refkeyr$r5r)r1r2r3rr4r_parse_wrap_argsHs8        /r?csbt|}|j||x,|jD]}|jjkr |jj|j<q Wfdd|jD}|ifS)zApply default keyword arguments. Named keywords may have been left blank. This function applies the default values so that every argument is defined. csg|]}j|qSr) arguments)r!r>)bound_argumentsrrr#sz#_apply_defaults..)rZbind parametersr-namer@defaultr;)funcr<kwargsZsigZparamr)rAr_apply_defaultss  rGTrr.)r,retr<r.returncs(tttfsfx8D]0}|dk rt|jtfstdt||fqWttttfrx8D]0}|dk rnt|jtfsntdt||fqnWfddDn6dk rtjtfstdtft t dt ft dt t fdfdd }|S) aWraps a function to become pint-aware. Use it when a function requires a numerical value but in some specific units. The wrapper function will take a pint quantity, convert to the units specified in `args` and then call the wrapped function with the resulting magnitude. The value returned by the wrapped function will be converted to the units specified in `ret`. Parameters ---------- ureg : pint.UnitRegistry a UnitRegistry instance. ret : str, pint.Unit, iterable of str, or iterable of pint.Unit Units of each of the return values. Use `None` to skip argument conversion. args : str, pint.Unit, iterable of str, or iterable of pint.Unit Units of each of the input arguments. Use `None` to skip argument conversion. strict : bool Indicates that only quantities are accepted. (Default value = True) Returns ------- callable the wrapper function. Raises ------ TypeError if the number of given arguments does not match the number of function parameters. if the any of the provided arguments is not a unit a string or Quantity Nz8wraps arguments must by of type str or Unit, not %s (%s)z=wraps 'ret' argument must by of type str or Unit, not %s (%s)csg|]}t|qSr)r )r!r")r,rrr#szwraps...)rErIcsttj}t|kr2tdj|tftfddtjD}tfddtjD}tj ||dt t dfdd }|S) Nz0%s takes %i parameters, but %i units were passedc3s|]}t|r|VqdS)N)hasattr)r!attr)rErrr% sz+wraps..decorator..c3s|]}t|r|VqdS)N)rJ)r!rK)rErrr% s)assignedupdated)rIcst||\}}|\}||}r^fddD}fddt||DSddkrn|S|drtdndS)Nc3s$|]\}}|rt|n|VqdS)N)r)r!rr=)rrrr%sz.decorator..wrapper..c3s*|]"\}}|dkr|n ||VqdS)N)r )r!unitZres)r,rrr%!srr )rG __class__rr r)r-kwr/resultZ out_units) converterrEis_ret_containerrHr.r,)rrwrappers    z)wraps..decorator..wrapper) r8rrB TypeError__name__tuple functoolsWRAPPER_ASSIGNMENTSWRAPPER_UPDATESwrapsr r)rE count_paramsrLrMrU)r<rSrTrHr.r,)rEr decorators "zwraps..decorator) rr'rXrrrVtyper?rPr rrr )r,rHr<r.r"r^r)r<rSrTrHr.r,rr\s.(   2.r\)r,r<rIcs$fdd|Dfdd}|S)a"Decorator to for quantity type checking for function inputs. Use it to ensure that the decorated function input parameters match the expected dimension of pint quantity. The wrapper function raises: - `pint.DimensionalityError` if an argument doesn't match the required dimensions. ureg : UnitRegistry a UnitRegistry instance. args : str or UnitContainer or None Dimensions of each of the input arguments. Use `None` to skip argument conversion. Returns ------- callable the wrapped function. Raises ------ TypeError If the number of given dimensions does not match the number of function parameters. ValueError If the any of the provided dimensions cannot be parsed as a dimension. cs"g|]}|dk r|ndqS)N)get_dimensionality)r!dim)r,rrr#Pszcheck..csttj}t|kr2tdj|tftfddtjD}tfddtjD}tj ||dfdd}|S)Nz5%s takes %i parameters, but %i dimensions were passedc3s|]}t|r|VqdS)N)rJ)r!rK)rErrr%]sz+check..decorator..c3s|]}t|r|VqdS)N)rJ)r!rK)rErrr%`s)rLrMcsdt||\}}xHt|D]:\}}|dkr.q||s|}t|d||qW||S)Nz a quantity of)rGzipr checkr`r )r<rFZ list_argsemptyrar$Zval_dim) dimensionsrEr,rrrUcs z)check..decorator..wrapper) r8rrBrVrWrXrYrZr[r\)rEr]rLrMrU)rer,)rErr^Ss  zcheck..decoratorr)r,r<r^r)rer,rrc1s rc)N)N)T)"__doc__rYinspectr itertoolsrtypingrrrrrZ_typingr errorsr Zquantityr utilr rrrrOrrrr r?rGrboolr\rcrrrr s(         c6,r