B ™‘[gJã@sdZddlmZddlmZddlmZddlmZm Z ddl m Z ddlmZd d ddd ddddddddddddddddddd d!d"d#d$d%d&hZ d'd(hZd)d*d+hZd,d-iZd.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdAdBdCdDdEdFdGdHdIdJœZdKdLdKdMœZdNdO„ZdPdQ„ZGdRdS„dSeƒZx ejD]ZeedTeeƒqWxeD]ZeedTeeƒq>WdUdV„ZdW ¡ZdXdY„e ¡DƒZeefdZdZiŽZd[dLiZ Gd\d]„d]eƒZ!x e!jD]Zee!dTeeƒqªWxe D]Zee!dTeeƒqÊWd^ ¡Z"d_dY„e ¡DƒZ#ee#fd`dadbdcdddedfdgdZdhœ ŽZ$Gdidj„djeƒZ%x e%jD]Zee%dTeeƒq6Wx e%j&D]Zee%dTeeƒqXWd8d9dkdldmdnddodpdqdrœZ'dsdLdKdtœZ(Gdudv„dve%ƒZ)x e)jD]Zee)dTeeƒq²Wx e)j&D]Zee)dTeeƒqÔWGdwdx„dxeƒZ*dyS)zzy Python code printers This module contains python code printers for plain python as well as NumPy & SciPy enabled code. é)Údefaultdict)Úwraps)Úchain)ÚsympifyÚSé)Ú precedence)ÚCodePrinterÚandÚasÚassertÚbreakÚclassÚcontinueÚdefÚdelÚelifÚelseÚexceptÚfinallyÚforÚfromÚglobalÚifÚimportÚinÚisÚlambdaÚnotÚorÚpassÚraiseÚreturnÚtryÚwhileÚwithÚyieldÚNoneÚexecÚprintÚFalseÚnonlocalÚTrueZAbsÚabsÚacosÚacoshÚasinÚasinhÚatanÚatan2ÚatanhZceilÚcosÚcoshÚerfÚerfcÚexpÚexpm1Ú factorialÚfloorÚgammaÚhypotZlgammaÚlogÚlog10Úlog1pÚlog2ÚsinÚsinhZsqrtÚtanÚtanh)r.r/r0r1r2r3r4Zceilingr5r6r7r8r9r:r;r<r=r>Úloggammar?Zlnr@rArBrCrDZSqrtrErFÚeZpi)ZExp1ÚPiÚEcs8ˆj|jj}djˆ |¡d t‡fdd„|jƒ¡dS)Nz{name}({args})z, cs ˆ |¡S)N)Ú_print)Úarg)Úself©ú4lib/python3.7/site-packages/sympy/printing/pycode.pyÚGsz#_print_known_func..)ÚnameÚargs)Úknown_functionsÚ __class__Ú__name__ÚformatÚ_module_formatÚjoinÚmaprR)rMÚexprÚknownrN)rMrOÚ_print_known_funcDsr\cCs|j|jj}| |¡S)N)Úknown_constantsrTrUrW)rMrZr[rNrNrOÚ_print_known_constJsr^csŠeZdZdZdZdZe e¡Z dZ dZee e ¡dd„e ¡DƒƒƒZdd „e ¡DƒZd ddd œZeejidddddZdI‡fdd„ Zdd„Zdd„ZdJdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Z d'd(„Z!d)d*„Z"d+d,„Z#‡fd-d.„Z$d/d0„Z%d1d2„Z&d3d4„Z'd5d6„Z(d7d8„Z)Z*Z+Z,Z-Z.Z/Z0d9d:„Z1d;d<„Z2d=d>„Z3d?d@„Z4dAdB„Z5dCdD„Z6dEdF„Z7dGdH„Z8‡Z9S)KÚPythonCodePrinterZ_pythoncodeÚPythonZpython3Nz cCsg|]\}}|d|f‘qS)zmath.rN)Ú.0ÚkÚvrNrNrOú XszPythonCodePrinter.cCsi|]\}}d||“qS)zmath.rN)rarbrcrNrNrOú ZszPythonCodePrinter.r rr)r rréTF)Úuser_functionsZ precisionZinlineÚfully_qualified_modulesZcontractcsVtt|ƒ |¡ttƒ|_t|jf|p(i di¡Ž|_ t|j f|pDi di¡Ž|_dS)NrgZuser_constants)Úsuperr_Ú__init__rÚsetÚmodule_importsÚdictÚ_kfÚgetrSÚ_kcr])rMÚsettings)rTrNrOrjes zPythonCodePrinter.__init__cCs|S)NrN)rMÚ codestringrNrNrOÚ_get_statementmsz PythonCodePrinter._get_statementcCsd||fS)Nz%s = %srN)rMrQÚvaluerNrNrOÚ_declare_number_constpsz'PythonCodePrinter._declare_number_constcCsp| d¡}|r.)rrVrYrR)rMrZrN)rŠrMrOÚ _print_Mod–szPythonCodePrinter._print_ModcCsÊg}d}x||jD]r}|j}|j}|dkr2| d¡| d¡| | |¡¡| d¡| d¡| | |¡¡| d¡|d7}qW|dd…}|ddkr¶|dd …}| d¡n | d ¡d |¡S)Nrrxú)z if z else rrwr,éþÿÿÿz else None)Ú)rRrZÚcondÚappendrKrX)rMrZÚresultÚirLrHÚcrNrNrOÚ_print_Piecewisešs( z"PythonCodePrinter._print_PiecewisecsVdddddddœ}|j|krF| |j¡}| |j¡}dj|j||d Stt|ƒ |¡S) z.Relational printer for Equality and UnequalityÚequalÚ not_equalÚlessÚ less_equalÚgreaterÚ greater_equal)z==z!=úz>=z({lhs} {op} {rhs}))ÚopÚlhsÚrhs)Úrel_oprKržrŸrVrir_Ú_print_Relational)rMrZrržrŸ)rTrNrOr¡±s z#PythonCodePrinter._print_RelationalcCsddlm}| | |¡¡S)Nr)Ú Piecewise)Z$sympy.functions.elementary.piecewiser¢rKZrewrite)rMrZr¢rNrNrOÚ _print_ITEÁszPythonCodePrinter._print_ITEcs0‡fdd„|jDƒ}djˆ |j¡d |¡dS)Nc3s6|].\}}}djˆ |¡ˆ |¡ˆ |¡dVqdS)zfor {i} in range({a}, {b}+1))r’ÚaÚbN)rVrK)rar’r¤r¥)rMrNrOú Çsz/PythonCodePrinter._print_Sum..z"(builtins.sum({function} {loops}))ú )ÚfunctionÚloops)ZlimitsrVrKr¨rX)rMrZr©rN)rMrOÚ _print_SumÅs zPythonCodePrinter._print_SumcCsdS)NZ1jrN)rMrZrNrNrOÚ_print_ImaginaryUnitÐsz&PythonCodePrinter._print_ImaginaryUnitcCs,|jj}|j ||¡}d|| | ¡¡fS)Nz%s(%s))rTrUrSrorKÚtolist)rMrZrQÚfuncrNrNrOÚ_print_MatrixBaseÓsz#PythonCodePrinter._print_MatrixBasecCs | |¡S)N)r®)rMrZrNrNrOrPàszPythonCodePrinter.csd ‡fdd„| d¡Dƒ¡S)NÚ csg|]}ˆj|‘qSrN)Útab)raÚline)rMrNrOrdãsz8PythonCodePrinter._indent_codestring..)rXrz)rMrrrN)rMrOÚ_indent_codestringâsz$PythonCodePrinter._indent_codestringcsNd t‡fdd„|jƒ¡}djˆ |j¡d ‡fdd„|jDƒ¡ˆ |¡dS) Nr¯cs ˆ |¡S)N)rK)rL)rMrNrOrPæsz=PythonCodePrinter._print_FunctionDefinition..z def {name}({parameters}): {body}z, csg|]}ˆ |j¡‘qSrN)rKÚsymbol)raÚvar)rMrNrOrdész?PythonCodePrinter._print_FunctionDefinition..)rQÚ parametersÚbody)rXrYr¶rVrKrQrµr²)rMÚfdr¶rN)rMrOÚ_print_FunctionDefinitionås z+PythonCodePrinter._print_FunctionDefinitioncs6d t‡fdd„|jƒ¡}djˆ |j¡ˆ |¡dS)Nr¯cs ˆ |¡S)N)rK)rL)rMrNrOrPîsz0PythonCodePrinter._print_While..zwhile {cond}: {body})rr¶)rXrYr¶rVrKZ conditionr²)rMZwhlr¶rN)rMrOÚ_print_Whileís zPythonCodePrinter._print_WhilecCs d| |jj¡| |jj¡fS)Nz%s = %s)rKZvariabler³rt)rMZdeclrNrNrOÚ_print_Declarationôsz$PythonCodePrinter._print_DeclarationcCs|j\}d| |¡S)Nz return %s)rRrK)rMZretrLrNrNrOÚ _print_ReturnúszPythonCodePrinter._print_Returncs^d t‡fdd„|jƒ¡}|jdkr8d ˆ |j¡|¡}|jdkrV|dˆ |j¡7}d|S)Nz, cs ˆ |¡S)N)rK)rL)rMrNrOrPÿsz0PythonCodePrinter._print_Print..z{0} % ({1})z , file=%sz print(%s))rXrYÚ print_argsÚ format_stringrVrKÚfile)rMZprntr¼rN)rMrOÚ_print_Printþs zPythonCodePrinter._print_PrintcCs@t|jƒdkr| d¡St|jƒdkr0| d¡S| |j¡SdS)NÚstdoutz sys.stdoutÚstderrz sys.stderr)ÚstrrQrWrK)rMZstrmrNrNrOÚ _print_Streams zPythonCodePrinter._print_StreamcCsdS)Nr'rN)rMrLrNrNrOÚ_print_NoneTokensz"PythonCodePrinter._print_NoneToken)N)T):rUÚ __module__Ú__qualname__ÚprintmethodZlanguageZstandardÚ_kw_py2and3ÚunionÚ_kw_only_py3Zreserved_wordsÚmodulesr°rmrÚ_known_functionsÚitemsÚ_known_functions_mathrnÚ_known_constants_mathrpZ _operatorsr Z_default_settingsrjrsrurWr€r‚rƒr„r†r‡rˆr‹r”r¡r£rªr«r®Ú_print_SparseMatrixZ_print_MutableSparseMatrixÚ_print_ImmutableSparseMatrixZ _print_MatrixZ_print_DenseMatrixZ_print_MutableDenseMatrixZ_print_ImmutableMatrixZ_print_ImmutableDenseMatrixr²r¸r¹rºr»r¿rÃrÄÚ __classcell__rNrN)rTrOr_Os\ $ r_z _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``. Examples ======== >>> from sympy import tan, Symbol >>> from sympy.printing.pycode import pycode >>> pycode(tan(Symbol('x')) + 1) 'math.tan(x) + 1' )r_Zdoprint)rZrqrNrNrOÚpycodesrÓz log1p log2cCs g|]\}}|tkr||f‘qSrN)Ú_not_in_mpmath)rarbrcrNrNrOrd3srdÚsignrIc@sdeZdZdZdZeee ¡dd„e ¡DƒƒƒZ dd„Zdd„Zd d „Z dd„Zd d„Zdd„ZdS)Ú MpmathPrinterzH Lambda printer for mpmath which maintains precision for floats Z_mpmathcodecCsg|]\}}|d|f‘qS)zmpmath.rN)rarbrcrNrNrOrdDszMpmathPrinter.cCs(tttt|jƒƒƒ}dj| d¡|dS)Nz{func}({args})z mpmath.mpf)rrR)rÂÚtuplerYÚintZ_mpf_rVrW)rMrHrRrNrNrOÚ_print_FloatGszMpmathPrinter._print_FloatcCsd | d¡|j|j¡S)Nz{0}({1})/{0}({2})z mpmath.mpf)rVrWÚpÚq)rMrHrNrNrOÚ_print_RationalQszMpmathPrinter._print_RationalcCs4d | d¡| |jd¡| |jd¡| d¡¡S)Nz{0}({1}, {2}, {3})zmpmath.gammaincrrz mpmath.inf)rVrWrKrR)rMrHrNrNrOÚ_print_uppergammaXs zMpmathPrinter._print_uppergammacCs,d | d¡| |jd¡| |jd¡¡S)Nz{0}({1}, 0, {2})zmpmath.gammaincrr)rVrWrKrR)rMrHrNrNrOÚ_print_lowergamma_szMpmathPrinter._print_lowergammacCsd | d¡| |jd¡¡S)Nz{0}({1})/{0}(2)z mpmath.logr)rVrWrKrR)rMrHrNrNrOÚ_print_log2eszMpmathPrinter._print_log2cCsd | d¡| |jd¡¡S)Nz {0}({1}+1)z mpmath.logr)rVrWrKrR)rMrHrNrNrOÚ_print_log1piszMpmathPrinter._print_log1pN)rUrÅrÆÚ__doc__rÇrmrrÌrÍÚ_known_functions_mpmathrnrÙrÜrÝrÞrßràrNrNrNrOrÖ<s rÖz!erf erfc factorial gamma loggammacCs g|]\}}|tkr||f‘qSrN)Ú _not_in_numpy)rarbrcrNrNrOrdusZarccosZarccoshZarcsinZarcsinhZarctanZarctan2ZarctanhÚexp2) r.r/r0r1r2r3r4rärÕcsÜeZdZdZdZeeej ¡dd„e ¡DƒƒƒZdd„e ¡DƒZdd„Z d d „Zdd„Zd d„Z‡fdd„Zdd„Zdd„Zdd„Zdd„Zdd„Z‡fdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Z‡ZS))ÚNumPyPrinterza Numpy printer which handles vectorized piecewise functions, logical operators, etc. Z _numpycodecCsg|]\}}|d|f‘qS)znumpy.rN)rarbrcrNrNrOrdŒszNumPyPrinter.cCsi|]\}}d||“qS)znumpy.rN)rarbrcrNrNrOreŽszNumPyPrinter.cs*t dd¡d t ‡fdd„|Dƒ¡¡S)z+General sequence printer: converts to tupleÚ delimiterz, z({},)c3s|]}ˆ |¡VqdS)N)rK)raÚitem)rMrNrOr¦–sz*NumPyPrinter._print_seq..)ZdelimiterorVrærX)rMÚseqrN)rMrOÚ _print_seq‘szNumPyPrinter._print_seqcs d d ‡fdd„|jDƒ¡¡S)zMatrix multiplication printerz({0})z).dot(c3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦šsz-NumPyPrinter._print_MatMul..)rVrXrR)rMrZrN)rMrOÚ _print_MatMul˜szNumPyPrinter._print_MatMulcCsR|j\}}|jddkr|j}|jddkr2|j}d| d¡| |¡| |¡fS)Nrrz %s(%s, %s)z numpy.dot)rRÚshapeÚTrWrK)rMrZZarg1Zarg2rNrNrOÚ_print_DotProductœs zNumPyPrinter._print_DotProductcsTd d ‡fdd„|jDƒ¡¡}d d ‡fdd„|jDƒ¡¡}d ˆ d¡||¡S)zPiecewise function printerz[{0}]ú,c3s|]}ˆ |j¡VqdS)N)rKrZ)rarL)rMrNrOr¦«sz0NumPyPrinter._print_Piecewise..c3s|]}ˆ |j¡VqdS)N)rKr)rarL)rMrNrOr¦¬sz {0}({1}, {2}, default=numpy.nan)znumpy.select)rVrXrRrW)rMrZZexprsZcondsrN)rMrOr”©s zNumPyPrinter._print_Piecewisecsddddddddœ}|j|krT| |j¡}| |j¡}dj| d ||j¡||d Stt|ƒ |¡S)z.Relational printer for Equality and Unequalityr•r–r—r˜r™rš)z==z!=r›z<=rœz>=z{op}({lhs}, {rhs})znumpy.)rržrŸ) r rKržrŸrVrWrirår¡)rMrZrržrŸ)rTrNrOr¡³s zNumPyPrinter._print_Relationalcs(d ˆ d¡d ‡fdd„|jDƒ¡¡S)zLogical And printerz{0}.reduce(({1}))znumpy.logical_andrîc3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦Ész*NumPyPrinter._print_And..)rVrWrXrR)rMrZrN)rMrOÚ _print_AndÄszNumPyPrinter._print_Andcs(d ˆ d¡d ‡fdd„|jDƒ¡¡S)zLogical Or printerz{0}.reduce(({1}))znumpy.logical_orrîc3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦Ðsz)NumPyPrinter._print_Or..)rVrWrXrR)rMrZrN)rMrOÚ _print_OrËszNumPyPrinter._print_Orcs(d ˆ d¡d ‡fdd„|jDƒ¡¡S)zLogical Not printerz{0}({1})znumpy.logical_notrîc3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦×sz*NumPyPrinter._print_Not..)rVrWrXrR)rMrZrN)rMrOÚ _print_NotÒszNumPyPrinter._print_Notcs(d ˆ d¡d ‡fdd„|jDƒ¡¡S)Nz {0}(({1}))z numpy.aminrîc3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦Úsz*NumPyPrinter._print_Min..)rVrWrXrR)rMrZrN)rMrOÚ _print_MinÙszNumPyPrinter._print_Mincs(d ˆ d¡d ‡fdd„|jDƒ¡¡S)Nz {0}(({1}))z numpy.amaxrîc3s|]}ˆ |¡VqdS)N)rK)rar’)rMrNrOr¦Ýsz*NumPyPrinter._print_Max..)rVrWrXrR)rMrZrN)rMrOÚ _print_MaxÜszNumPyPrinter._print_Maxcs8|jdkr$d | d¡| |j¡¡Stt|ƒ |¡SdS)Ngà?z{0}({1})z numpy.sqrt)r9rVrWrKÚbaseriråÚ _print_Pow)rMrZ)rTrNrOrõßs zNumPyPrinter._print_PowcCsd| d¡| |jd¡fS)Nz%s(%s)znumpy.angler)rWrKrR)rMrZrNrNrOÚ _print_argåszNumPyPrinter._print_argcCsd| d| |jd¡¡S)Nz%s(%s)z numpy.imagr)rWrKrR)rMrZrNrNrOÚ _print_imèszNumPyPrinter._print_imcs(dˆ d¡d t‡fdd„|jƒ¡fS)Nz%s(%s)z numpy.modz, cs ˆ |¡S)N)rK)rL)rMrNrOrPísz)NumPyPrinter._print_Mod..)rWrXrYrR)rMrZrN)rMrOr‹ëszNumPyPrinter._print_ModcCsd| d¡| |jd¡fS)Nz%s(%s)z numpy.realr)rWrKrR)rMrZrNrNrOÚ _print_reïszNumPyPrinter._print_recCs$d| d¡| |jdtj¡fS)Nz%s(%s)z numpy.sincr)rWrKrRrrI)rMrZrNrNrOÚ_print_sincòszNumPyPrinter._print_sinccCs:|j |jjd¡}|dkr$| d¡}d|| | ¡¡fS)Nznumpy.arrayz%s(%s))rSrorTrUrWrKr¬)rMrZrrNrNrOr®õs zNumPyPrinter._print_MatrixBase)rUrÅrÆrárÇrmrr_rnrÍÚ_known_functions_numpyrÏrprérêrír”r¡rïrðrñròrórõrör÷r‹rørùr®rÒrNrN)rTrOråƒs. råZjnZynZivZknZgammalnZpsiZpoch)r7r8ZbesseljZbesselyZbesseliZbesselkr;r=rGZdigammaZRisingFactorialZgolden_ratio)ZGoldenRatiorIrJc@sLeZdZeeej ¡dd„e ¡DƒƒƒZdd„e ¡DƒZ dd„ZeZdS)ÚSciPyPrintercCsg|]\}}|d|f‘qS)zscipy.special.rN)rarbrcrNrNrOrdszSciPyPrinter.cCsi|]\}}d||“qS)zscipy.constants.rN)rarbrcrNrNrOreszSciPyPrinter.cCsfggg}}}x8|j ¡D]*\\}}}| |¡| |¡| |¡qWdj| d¡||||jdS)Nz){name}({data}, ({i}, {j}), shape={shape})zscipy.sparse.coo_matrix)rQÚdatar’Újrë)Z_smatrÍrrVrWrë)rMrZr’rýrüÚrr“rcrNrNrOrÐs z SciPyPrinter._print_SparseMatrixN) rUrÅrÆrmrrårnrÍÚ_known_functions_scipy_specialÚ _known_constants_scipy_constantsrprÐrÑrNrNrNrOrûsrûc@s4eZdZedd„ee ¡e ¡ƒDƒƒZdd„Z dS)ÚSymPyPrintercCsg|]\}}|d|f‘qS)zsympy.rN)rarbrcrNrNrOrd6szSymPyPrinter.csH|jjp d}dˆ ||rdnd|jj¡d t‡fdd„|jƒ¡fS)NrŽz%s(%s)rvz, cs ˆ |¡S)N)rK)rL)rMrNrOrP>sz.SymPyPrinter._print_Function..)rrÅrWrUrXrYrR)rMrZÚmodrN)rMrOÚ_print_Function;szSymPyPrinter._print_FunctionN) rUrÅrÆrmrrÌrÍrÎrnrrNrNrNrOr4s rN)+ráÚcollectionsrÚ functoolsrÚ itertoolsrZ sympy.corerrrZcodeprinterr rÈZ_kw_only_py2rÊrÌrÎrÏr\r^r_rnrbÚsetattrrÓrzrÔrÍZ _in_mpmathrmrâZ_known_constants_mpmathrÖrãZ _in_numpyrúrårprÿrrûrrNrNrNrOÚsÊ E 1 y