a Ÿ¬d?d@dAdBdCdDdEdFdGœ¤ŽZd(d)dHdIdJd+d*dKdLœZdMdN„ZGdOdP„dPeƒZejD]Zeed5eeƒqˆeD]Zeed5eeƒq¤GdQdR„dReƒZdSS)Tzy Python code printers This module contains Python code printers for plain Python as well as NumPy & SciPy enabled code. é)Údefaultdict)Úchain)ÚSé)Ú precedence)ÚCodePrinter>!ÚelseÚelifÚTrueÚassertÚtryÚwithÚfromÚforÚdelÚNoneÚexceptÚimportÚFalseÚbreakÚreturnÚwhileÚcontinueÚnotÚifÚlambdaÚyieldÚisÚpassÚdefÚnonlocalÚfinallyÚasÚinÚraiseÚorÚclassÚandÚglobalZAbsÚabsÚacosÚacoshÚasinÚasinhÚatanÚatan2ÚatanhÚceilÚcosÚcoshÚerfÚerfcÚexpÚexpm1Ú factorialÚfloorÚgammaÚhypotÚlgammaÚlogÚlog10Úlog1pÚlog2ÚsinÚsinhÚsqrtÚtanÚtanh)r*r+r,r-r.r/r0Zceilingr2r3r4r5r6r7r8r9r:r;Úloggammar=Úlnr>r?r@rArBZSqrtrDrEÚeÚpiÚinfÚnan)ÚExp1ÚPiÚEÚInfinityÚNaNZComplexInfinitycs8ˆj|jj}djˆ |¡d t‡fdd„|jƒ¡dS)Nz{name}({args})ú, cs ˆ |¡S©N©Ú_print©Úarg©Úself©úe/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/printing/pycode.pyÚBóz#_print_known_func..)ÚnameÚargs)Úknown_functionsÚ __class__Ú__name__ÚformatÚ_module_formatÚjoinÚmapr^©rXÚexprZknownrYrWrZÚ_print_known_func?sÿrhcCs|j|jj}| |¡SrR)Úknown_constantsr`rarcrfrYrYrZÚ_print_known_constEsrjc sœeZdZdZdZeZdZdZe e e ¡dd„e ¡DƒƒƒZdd„e ¡DƒZd d ddœZe ejid dddddZdQ‡fdd„ Zdd„ZdRdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Z d,d-„Z!d.d/„Z"‡fd0d1„Z#d2d3„Z$d4d5„Z%d6d7„Z&d8d9„Z'd:d;„Z(dd?„Z1d@dA„Z2dBdC„Z3dDdE„Z4dFdG„Z5dHdI„Z6dJdK„Z7dLdM„Z8dSdOdP„Z9‡Z:S)TÚAbstractPythonCodePrinterZ_pythoncodeÚPythonNz cCsg|]\}}|d|f‘qS©zmath.rY©Ú.0ÚkÚvrYrYrZÚ Rr\z$AbstractPythonCodePrinter.cCsi|]\}}|d|“qSrmrYrnrYrYrZÚ Tr\z$AbstractPythonCodePrinter.r'r%r)r'r%réTFÚpython3)Úuser_functionsZ precisionÚinlineÚfully_qualified_modulesZcontractÚstandardcs˜tƒ |¡|jd}|dur4ddl}d |jj¡}|dkrDtdƒ‚||_t t ƒ|_t|j fi|pdi di¡¤Ž|_t|jfi|p„i di¡¤Ž|_dS)Nryrzpython{}ruzOnly Python 3 is supported.rvZuser_constants)ÚsuperÚ__init__Ú _settingsÚsysrbÚversion_infoÚmajorÚ ValueErrorryrÚsetÚmodule_importsÚdictÚ_kfÚgetr_Ú_kcri)rXÚsettingsZstdr}©r`rYrZr{`s ÿ ÿz"AbstractPythonCodePrinter.__init__cCsd||fS©Nz%s = %srY)rXr]ÚvaluerYrYrZÚ_declare_number_consttsz/AbstractPythonCodePrinter._declare_number_constcCsp| d¡}|r.rú,r)Z _get_letter_generator_for_einsumÚrangeÚappendÚnext)rXZsubranksZcontraction_indicesÚlettersZcontraction_stringÚcounterÚdÚindicesZrank_argZlindicesr¤ÚmappingZletters_freeZletters_dumr¥ÚlrYrYrZÚ_get_einsum_stringºs< z,AbstractPythonCodePrinter._get_einsum_stringcCsdS)Nzfloat('nan')rY©rXrgrYrYrZÚ _print_NaNÝsz$AbstractPythonCodePrinter._print_NaNcCsdS)Nzfloat('inf')rYr±rYrYrZÚ_print_Infinityàsz)AbstractPythonCodePrinter._print_InfinitycCsdS)Nz float('-inf')rYr±rYrYrZÚ_print_NegativeInfinityãsz1AbstractPythonCodePrinter._print_NegativeInfinitycCs | |¡SrR)r²r±rYrYrZÚ_print_ComplexInfinityæsz0AbstractPythonCodePrinter._print_ComplexInfinitycs$t|ƒ‰djt‡‡fdd„|jƒŽS)Nz{} % {}csˆ |ˆ¡SrR)Úparenthesize)Úx©ÚPRECrXrYrZr[ër\z6AbstractPythonCodePrinter._print_Mod..)rrbrer^r±rYr¸rZÚ _print_Modész$AbstractPythonCodePrinter._print_ModcCsÆg}d}|jD]r}|j}|j}|dkr0| d¡| d¡| | |¡¡| d¡| d¡| | |¡¡| d¡|d7}q|dd…}|ddkr²|dd …}| d¡n | d ¡d |¡S)NrrŽú)z if z else rrr éþÿÿÿz else None)r¢)r^rgÚcondr¨rTrd)rXrgÚresultr¤rVrHÚcrYrYrZÚ_print_Piecewiseís( z*AbstractPythonCodePrinter._print_PiecewisecsRdddddddœ}|j|vrF| |j¡}| |j¡}dj|j||d Stƒ |¡S) z.Relational printer for Equality and UnequalityÚequalÚ not_equalZlessZ less_equalZgreaterZ greater_equal)z==z!=úz>=z({lhs} {op} {rhs}))ržÚlhsÚrhs)Zrel_oprTrÅrÆrbrzÚ_print_Relational)rXrgržrÅrÆrˆrYrZrÇsú z+AbstractPythonCodePrinter._print_RelationalcCsddlm}| | |¡¡S)Nr)Ú Piecewise)Z$sympy.functions.elementary.piecewiserÈrTZrewrite)rXrgrÈrYrYrZÚ _print_ITEsz$AbstractPythonCodePrinter._print_ITEcs0‡fdd„|jDƒ}djˆ |j¡d |¡dS)Nc3s6|].\}}}djˆ |¡ˆ |¡ˆ |¡dVqdS)zfor {i} in range({a}, {b}+1))r¤ÚaÚbN)rbrT)ror¤rÊrËrWrYrZÚ süýz7AbstractPythonCodePrinter._print_Sum..z"(builtins.sum({function} {loops}))ú )ÚfunctionÚloops)ÚlimitsrbrTrÎrd)rXrgrÏrYrWrZÚ _print_Sums û þz$AbstractPythonCodePrinter._print_SumcCsdS)NZ1jrYr±rYrYrZÚ_print_ImaginaryUnit#sz.AbstractPythonCodePrinter._print_ImaginaryUnitcCs$|j\}}dj| |¡| |¡dS)Nz(1 if {a} == {b} else 0))rÊrË)r^rbrT)rXrgrÊrËrYrYrZÚ_print_KroneckerDelta&s þz/AbstractPythonCodePrinter._print_KroneckerDeltacCs,|jj}|j ||¡}d|| | ¡¡fS)Nú%s(%s))r`rar_r…rTÚtolist)rXrgr]ÚfuncrYrYrZÚ_print_MatrixBase.sz+AbstractPythonCodePrinter._print_MatrixBasecCs | |¡SrR)r×r±rYrYrZr[;r\z"AbstractPythonCodePrinter.csd ‡fdd„| d¡Dƒ¡S)NÚ csg|]}ˆj|‘qSrY)Útab)roÚlinerWrYrZrr>r\z@AbstractPythonCodePrinter._indent_codestring..)rdrr˜rYrWrZÚ_indent_codestring=sz,AbstractPythonCodePrinter._indent_codestringcsNd t‡fdd„|jƒ¡}djˆ |j¡d ‡fdd„|jDƒ¡ˆ |¡dS) NrØcs ˆ |¡SrRrSrUrWrYrZr[Ar\zEAbstractPythonCodePrinter._print_FunctionDefinition..z def {name}({parameters}): {body}rQcsg|]}ˆ |j¡‘qSrY)rTÚsymbol)roÚvarrWrYrZrrDr\zGAbstractPythonCodePrinter._print_FunctionDefinition..)r]Ú parametersÚbody)rdrerßrbrTr]rÞrÛ)rXÚfdrßrYrWrZÚ_print_FunctionDefinition@s ýz3AbstractPythonCodePrinter._print_FunctionDefinitioncs6d t‡fdd„|jƒ¡}djˆ |j¡ˆ |¡dS)NrØcs ˆ |¡SrRrSrUrWrYrZr[Ir\z8AbstractPythonCodePrinter._print_While..zwhile {cond}: {body})r½rß)rdrerßrbrTÚ conditionrÛ)rXZwhlrßrYrWrZÚ_print_WhileHs þz&AbstractPythonCodePrinter._print_WhilecCs d| |jj¡| |jj¡fSr‰)rTÚvariablerÜrŠ)rXÚdeclrYrYrZÚ_print_DeclarationOsþz,AbstractPythonCodePrinter._print_DeclarationcCs|j\}d| |¡S)Nz return %s)r^rT)rXÚretrVrYrYrZÚ _print_ReturnUsz'AbstractPythonCodePrinter._print_Returncs^d t‡fdd„|jƒ¡}|jdkr8d ˆ |j¡|¡}|jdkrV|dˆ |j¡7}d|S)NrQcs ˆ |¡SrRrSrUrWrYrZr[Zr\z8AbstractPythonCodePrinter._print_Print..z {} % ({})z , file=%sz print(%s))rdreÚ print_argsÚ format_stringrbrTÚfile)rXZprntrérYrWrZÚ_print_PrintYs ÿ z&AbstractPythonCodePrinter._print_PrintcCs@t|jƒdkr| d¡St|jƒdkr0| d¡S| |j¡SdS)NÚstdoutz sys.stdoutÚstderrz sys.stderr)Ústrr]rcrT)rXÚstrmrYrYrZÚ _print_Streamcs z'AbstractPythonCodePrinter._print_StreamcCsdS)NrrY)rXrVrYrYrZÚ_print_NoneTokenksz*AbstractPythonCodePrinter._print_NoneTokenú math.sqrtc Cs¶t|ƒ}|jtjkr<|s<| |¡}| |j¡}dj||dS|jr†|jtjur†|s†| |¡}| tj ¡}| |j¡}dj|||dS|j |j|dd}|j |j|dd} d || ¡S)aPrinting helper function for ``Pow`` Notes ===== This only preprocesses the ``sqrt`` as math formatter Examples ======== >>> from sympy import sqrt >>> from sympy.printing.pycode import PythonCodePrinter >>> from sympy.abc import x Python code printer automatically looks up ``math.sqrt``. >>> printer = PythonCodePrinter() >>> printer._hprint_Pow(sqrt(x), rational=True) 'x**(1/2)' >>> printer._hprint_Pow(sqrt(x), rational=False) 'math.sqrt(x)' >>> printer._hprint_Pow(1/sqrt(x), rational=True) 'x**(-1/2)' >>> printer._hprint_Pow(1/sqrt(x), rational=False) '1/math.sqrt(x)' Using sqrt from numpy or mpmath >>> printer._hprint_Pow(sqrt(x), sqrt='numpy.sqrt') 'numpy.sqrt(x)' >>> printer._hprint_Pow(sqrt(x), sqrt='mpmath.sqrt') 'mpmath.sqrt(x)' See Also ======== sympy.printing.str.StrPrinter._print_Pow z {func}({arg}))rÖrVz{num}/{func}({arg}))ÚnumrÖrVF)Ústrictz{}**{})rr6rZHalfrcrTÚbaserbZis_commutativeZOner¶) rXrgÚrationalrCr¹rÖrVrôZbase_strZexp_strrYrYrZÚ_hprint_Powns ' ÿz%AbstractPythonCodePrinter._hprint_Pow)N)T)Fró);raÚ __module__Ú__qualname__ÚprintmethodÚlanguageÚ_kwÚreserved_wordsÚmodulesrÙrƒrÚ_known_functionsÚitemsÚ_known_functions_mathr„Ú_known_constants_mathr†Ú _operatorsrZ_default_settingsr{r‹rcr–r™r›rr r°r²r³r´rµrºrÀrÇrÉrÑrÒrÓr×Z_print_SparseRepMatrixZ_print_MutableSparseMatrixZ_print_ImmutableSparseMatrixZ _print_MatrixZ_print_DenseMatrixZ_print_MutableDenseMatrixZ_print_ImmutableMatrixZ_print_ImmutableDenseMatrixrÛrárãrærèrìrñròrøÚ __classcell__rYrYrˆrZrkJs€þù # øÿþýüû rkcsneZdZdd„Zdd„Zdd„Zddd „Zd d„Zdd „Zdd„Z ‡fdd„Z ejZ ejZejZejZ‡ZS)ÚPythonCodePrintercCs dj| d¡| |jd¡dS)Nz"(0.0 if {e} == 0 else {f}(1, {e}))z math.copysignr)ÚfrH©rbrcrTr^©rXrHrYrYrZÚ_print_sign«sÿzPythonCodePrinter._print_signcCs$t|ƒ}|jd| |jd|¡S)Nrr)rrr¶r^)rXrgr¹rYrYrZÚ _print_Not¯szPythonCodePrinter._print_Notcs<|jd}|jdd…}d t|ƒd ‡fdd„|Dƒ¡¡S)Nrrz{}[{}]rQcsg|]}ˆ |¡‘qSrYrS)roÚindrWrYrZrr¶r\z4PythonCodePrinter._print_Indexed..)r^rbrïrd)rXrgröÚindexrYrWrZÚ_print_Indexed³s z PythonCodePrinter._print_IndexedFcCs|j||dS)N)r÷©rø©rXrgr÷rYrYrZÚ _print_Pow¸szPythonCodePrinter._print_PowcCsd |j|j¡S)Nz{}/{})rbÚpÚqr±rYrYrZÚ_print_Rational»sz!PythonCodePrinter._print_RationalcCs | |¡SrR©rr±rYrYrZÚ_print_Half¾szPythonCodePrinter._print_HalfcCs"ddlm}| ||jddƒ¡S)Nr)ÚModr)Zsympy.core.modrrºr^)rXrgrrYrYrZÚ_print_fracÁszPythonCodePrinter._print_fraccsdtƒ |¡}||jvr@|jdr2d}t| |¡ƒ‚||jdSd|vr\| dd¡ dd¡S|SdS)NZerror_on_reservedzVThis expression includes the symbol "{}" which is a reserved keyword in this language.Zreserved_word_suffixÚ{r¢Ú})rzÚ _print_Symbolrþr|r€rbÚreplace)rXrgr]ÚmsgrˆrYrZrÅs zPythonCodePrinter._print_Symbol)F)rarùrúr rrrrrrrrZ_print_not_supportedÚ_print_lowergammaÚ_print_uppergammaZ_print_fresnelcZ_print_fresnelsrrYrYrˆrZr©s rz _print_%scKst|ƒ |¡S)a) Converts an expr to a string of Python code Parameters ========== expr : Expr A SymPy expression. fully_qualified_modules : bool Whether or not to write out full module names of functions (``math.sin`` vs. ``sin``). default: ``True``. standard : str or None, optional Only 'python3' (default) is supported. This parameter may be removed in the future. Examples ======== >>> from sympy import pycode, tan, Symbol >>> pycode(tan(Symbol('x')) + 1) 'math.tan(x) + 1' )rZdoprint)rgr‡rYrYrZÚpycodeásr z log1p log2cCs g|]\}}|tvr||f‘qSrY)Ú_not_in_mpmathrnrYrYrZrrür\rrÚbetaÚfracÚfresnelcÚfresnelsÚsignrFÚhyperÚmeijergÚbesseljÚbesselyÚbesseliÚbesselk)r"r#r$r%r&rFr'r(r)r*r+r,ÚphiZeulerÚcatalanZninf)rLrMZGoldenRatioZ EulerGammaZCatalanrPrOÚNegativeInfinitycCsTg}g}|jD]<}t|ƒdkr*|\}}}ntdƒ‚| |¡| ||f¡q||fS)zì helper function for _print_Integral that - accepts an Integral expression - returns a tuple of - a list variables of integration - a list of tuples of the upper and lower limits of integration éz%Only definite integrals are supported)rÐr‘ÚNotImplementedErrorr¨)Z integral_exprÚintegration_varsrÐZintegration_rangeZintegration_varZlower_limitZupper_limitrYrYrZÚ_unpack_integral_limitss r3c@s”eZdZdZdZdZeee ¡dd„e ¡DƒƒƒZdd„e ¡DƒZ dd „Zd d„Zdd „Zdd„Zdd„Zdd„Zdd„Zddd„Zdd„ZdS)Ú MpmathPrinterzH Lambda printer for mpmath which maintains precision for floats Z_mpmathcodezPython with mpmathcCsg|]\}}|d|f‘qS©zmpmath.rYrnrYrYrZrr4r\zMpmathPrinter.cCsi|]\}}|d|“qSr5rYrnrYrYrZrs6r\zMpmathPrinter.cCs(tttt|jƒƒƒ}dj| d¡|dS)Nz{func}({args})ú mpmath.mpf)rÖr^)rïÚtuplereÚintZ_mpf_rbrc)rXrHr^rYrYrZÚ_print_Float8szMpmathPrinter._print_FloatcCs&dj| d¡| |j¡| |j¡dS)Nz{func}({p})/{func}({q})r6)rÖrr)rbrcrTrrr rYrYrZrBs ýzMpmathPrinter._print_RationalcCs | |¡SrRrr rYrYrZrIszMpmathPrinter._print_HalfcCs4d | d¡| |jd¡| |jd¡| d¡¡S)Nz{}({}, {}, {})úmpmath.gammaincrrz mpmath.infrr rYrYrZrLsüzMpmathPrinter._print_uppergammacCs,d | d¡| |jd¡| |jd¡¡S)Nz {}({}, 0, {})r:rrrr rYrYrZrSs ýzMpmathPrinter._print_lowergammacCsd | d¡| |jd¡¡S)Nz{0}({1})/{0}(2)ú mpmath.logrrr rYrYrZÚ_print_log2YsÿzMpmathPrinter._print_log2cCsd | d¡| |jd¡¡S)Nz{}({}+1)r;rrr rYrYrZÚ_print_log1p]sÿzMpmathPrinter._print_log1pFcCs|j||ddS)Nzmpmath.sqrt©r÷rCrrrYrYrZraszMpmathPrinter._print_Powc sPt|ƒ\}}d ˆ d¡d tˆj|ƒ¡ˆ |jd¡d ‡fdd„|Dƒ¡¡S)Nz{}(lambda {}: {}, {})zmpmath.quadrQrc3s"|]}dttˆj|ƒƒVqdS)z(%s, %s)N)r7rerT)ror¯rWrYrZrÌkr\z0MpmathPrinter._print_Integral..)r3rbrcrdrerTr^)rXrHr2rÐrYrWrZÚ_print_IntegraldsüzMpmathPrinter._print_IntegralN)F)rarùrúÚ__doc__rûrürƒrrrÚ_known_functions_mpmathr„Ú_known_constants_mpmathr†r9rrrrr<r=rr?rYrYrYrZr4*s"þ r4c@s"eZdZdZdd„Zddd„ZdS) ÚSymPyPrinterzPython with SymPycsH|jjp d}dˆ ||rdnd|jj¡d t‡fdd„|jƒ¡fS)Nr¢rÔrŒrQcs ˆ |¡SrRrSrUrWrYrZr[|r\z.SymPyPrinter._print_Function..)rÖrùrcrardrer^)rXrgÚmodrYrWrZÚ_print_FunctionysÿzSymPyPrinter._print_FunctionFcCs|j||ddS)Nz sympy.sqrtr>rrrYrYrZr~szSymPyPrinter._print_PowN)F)rarùrúrürErrYrYrYrZrCusrCN) r@ÚcollectionsrÚ itertoolsrZ sympy.corerrZcodeprinterrrýrrrrhrjrkrr„rpÚsetattrr rr!rZ _in_mpmathrƒrArBr3r4rCrYrYrYrZÚs°ÿã!ú a1 ô øD