o jg&@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZddlmZddlmZddl m!Z!ddl"m#Z#ddZ$dddZ%d de&fddZ'dddZ(dS)!zA Several methods to simplify expressions involving unit objects. )reduce)Iterable)Optional)default_sort_key)Add)Tuple)Mul)Pow)ordered)sympify)Function)NonInvertibleMatrixError) DimensionDimensionSystem)Prefix)Quantity UnitSystem)siftc sddlm}t|}j|ddfdd|DfddD}t}|t|s7dStfdd|D}|fd d|D}|fd d|D}z||} W| St ylYdSw) Nr)MatrixTmark_dimensionlesscsg|] }t|qS)rget_dimensional_expr).0x) unit_systemrP/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/physics/units/util.py sz3_get_conversion_matrix_for_expr..cs$g|]}j|ddD]}|q qS)Tr)get_dimensional_dependencies)rri)dimension_systemrrr s$cs"g|] }|vs|s|qSr)addrr )seenrrr'"cs g|] fddDqS)cs"g|] }j|dddqS)Trr)rgetr#)r!jrrr)r%z>_get_conversion_matrix_for_expr...r)r)r! target_dims)r'rr)s csg|]}|dqSr)r&)rk)dim_dependenciesrrr*) sympy.matrices.denserget_dimension_systemrrrsetissubsetsolver ) expr target_unitsrrexpr_dimcanon_dim_unitscanon_expr_unitscamatexprmat res_exponentsr)r+r!r$r(rr_get_conversion_matrix_for_exprs&   r:SIcsddlm}|tttfsgfdd}t|tr%||St|tr8t|jtr8||j|j St |}t t|t rI| }t|t s`|t r`|ddfdd}fdd t|}|d urs|S|}|tfd d t|DS) a Convert ``expr`` to the same expression with all of its units and quantities represented as factors of ``target_units``, whenever the dimension is compatible. ``target_units`` may be a single unit/quantity, or a collection of units/quantities. Examples ======== >>> from sympy.physics.units import speed_of_light, meter, gram, second, day >>> from sympy.physics.units import mile, newton, kilogram, atomic_mass_constant >>> from sympy.physics.units import kilometer, centimeter >>> from sympy.physics.units import gravitational_constant, hbar >>> from sympy.physics.units import convert_to >>> convert_to(mile, kilometer) 25146*kilometer/15625 >>> convert_to(mile, kilometer).n() 1.609344*kilometer >>> convert_to(speed_of_light, meter/second) 299792458*meter/second >>> convert_to(day, second) 86400*second >>> 3*newton 3*newton >>> convert_to(3*newton, kilogram*meter/second**2) 3*kilogram*meter/second**2 >>> convert_to(atomic_mass_constant, gram) 1.660539060e-24*gram Conversion to multiple units: >>> convert_to(speed_of_light, [meter, second]) 299792458*meter/second >>> convert_to(3*newton, [centimeter, gram, second]) 300000*centimeter*gram/second**2 Conversion to Planck units: >>> convert_to(atomic_mass_constant, [gravitational_constant, speed_of_light, hbar]).n() 7.62963087839509e-20*hbar**0.5*speed_of_light**0.5/gravitational_constant**0.5 rrcstfdd|jDS)Nc3s|] }t|VqdSN convert_tor#r3rrr gsz2convert_to..handle_Adds..)rfromiterargsr2r?rr handle_Addsfszconvert_to..handle_AddscSs t|tSr<) isinstancerrrrrvs zconvert_to..cs |Sr<r=rFr?rrrGws csZt|trtddfdd|jDSt|tr!|j|jSt|tr+|S|S)NcSs||Sr<r)ryrrrrG{sz.get_total_scale_factor..csg|]}|qSrrr#get_total_scale_factorrrr|z>convert_to..get_total_scale_factor..) rErrrBr baseexprget_quantity_scale_factorrC)rJrrrrJys    z*convert_to..get_total_scale_factorNc3s(|]\}}d|||VqdS)Nr)ruprIrrr@s zconvert_to..)sympy.physics.unitsrget_unit_systemrErrrr rLrMr r togetherrhasreplacer:rrAzip)r2r3rrrDdepmatexpr_scale_factorr)rJr3rrr>4s2 ,       r>FNacross_dimensionscs8|js |tts |S|t}|dd|D}t|tdd}|D],}t||dkr1q&tt ||}|d|dj |fdd|ddD}q&|r|dur]t d t |}|}||}|j|d d } d} |jD] \} } | | kr| } nqy| dur|S|j| } | rt|| |}|S) aReturn an equivalent expression in which prefixes are replaced with numerical values and all units of a given dimension are the unified in a canonical manner by default. `across_dimensions` allows for units of different dimensions to be simplified together. `unit_system` must be specified if `across_dimensions` is True. Examples ======== >>> from sympy.physics.units.util import quantity_simplify >>> from sympy.physics.units.prefixes import kilo >>> from sympy.physics.units import foot, inch, joule, coulomb >>> quantity_simplify(kilo*foot*inch) 250*foot**2/3 >>> quantity_simplify(foot - 6*inch) foot/2 >>> quantity_simplify(5*joule/coulomb, across_dimensions=True, unit_system="SI") 5*volt cSsi|]}||jqSr scale_factor)rrQrrr rKz%quantity_simplify..cSs|jSr<) dimension)r rrrrGsz#quantity_simplify..rOrcsi|]}||jqSrr[)rvirefrrr]r,Nz:unit_system must be specified if across_dimensions is TrueTr)is_AtomrUrratomsxreplacerlenlistr r\ ValueErrorrrSr.rrdimensional_dependenciesitems derived_unitsr&r>)r2rZrrQdr*vr!dim_exprdim_depstarget_dimensionds_dim ds_dim_deps target_unitrr`rquantity_simplifys< "    rsc CsJddlm}||}dd}|t}|j}|D]c}t}|jD]Z}|j r.| dq#g} d} i} t |D]"} | trGt|| } | trT|| || } q9| jr[d} nq9| | | s}| tt| tdt|d kr}td |q#qi} |t D]}td d |jDr|jd d|jD| |<q|| S)z[Return expr if units in addends have the same base dimensions, else raise a ValueError.rrcSsPi||}|D]\}}||vr||vr|||||<q dd|DS)z]Merge dictionaries by adding values of common keys and removing keys with value of 0.cSsi|] \}}|dkr||qSr)r)rkeyvalrrrr]sz5check_dimensions..addDict..)ri)dict1dict2dict3rtvaluerrraddDicts z!check_dimensions..addDictrFT)rtrOz(addends have incompatible dimensions: {}css|]}t|tVqdSr<)rErr#rrrr@sz#check_dimensions..cSsg|]}|js|qSr) is_numberr#rrrrs z$check_dimensions..)rRrrSrcrr.rr/rBr{r"r make_argsrUrrr free_symbolsextendrituplesortedrrergformatanyfuncrd)r2rrrzaddsDIM_OFadesetaidimsskipdimdictr repsmrrrcheck_dimensionssP          r)r;)FN))__doc__ functoolsrcollections.abcrtypingrsympyrsympy.core.addrsympy.core.containersrsympy.core.mulrsympy.core.powerr sympy.core.sortingr sympy.core.sympifyr sympy.core.functionr sympy.matrices.exceptionsr sympy.physics.units.dimensionsrrsympy.physics.units.prefixesrsympy.physics.units.quantitiesrsympy.physics.units.unitsystemrsympy.utilities.iterablesrr:r>boolrsrrrrrs,                 YD