U ;qLe^!@s,dZddlmZddlmZddlmZddlmZddl m Z ddl m Z d d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)h!Z d*d+d,d-Z d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdAdBdCdDdEdFdGdHdIdJZdKdLdKdMdNdNdOZdPdQZdRdSZGdTdUdUe ZGdVdWdWZGdXdYdYeZejD]ZeedZeeq4eD]ZeedZeeqPd[d\Zd]Zd^d_eDZeefd`dadbdcdddedfdgdhdidjdkdl ZdKdLdmdndodNdMdpdqZdrdsZ GdtdudueZ!e!jD]Zee!dZeeqeD]Zee!dZeeqGdvdwdweZ"dxS)yzy Python code printers This module contains Python code printers for plain Python as well as NumPy & SciPy enabled code. ) defaultdict)chain)S)Mod) precedence) CodePrinterandasassertbreakclasscontinuedefdelelifelseexceptfinallyforfromglobalifimportinislambdanotorpassraisereturntrywhilewithyieldNoneFalsenonlocalTrueabsminmax)AbsMinMaxacosacoshasinasinhatanatan2atanhceilcoscosherferfcexpexpm1 factorialfloorgammahypotlgammaloglog10log1plog2sinsinhsqrttantanh)r0r1r2r3r4r5r6ceilingr8r9r:r;r<r=r>r?r@rAloggammarClnrDrErFrGrHSqrtrJrKepiinfnan)Exp1PiEInfinityNaNComplexInfinitycs8j|jj}dj|dfdd|jDdS)Nz{name}({args}), c3s|]}|VqdSN_print.0argselfl/home/p21-0144/sympy/latex2sympy2solve-back-end/sympyEq/lib/python3.8/site-packages/sympy/printing/pycode.py Esz$_print_known_func..)nameargs)known_functions __class____name__format_module_formatjoinrgrbexprknownrcrard_print_known_funcBs rqcCs|j|jj}||Sr[)known_constantsrirjrlrnrcrcrd_print_known_constHsrsc seZdZdZdZeZdZdZe e e dde DZdde DZd d d d Ze ejid dddddZdOfdd ZddZdPddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Z d,d-Z!fd.d/Z"d0d1Z#d2d3Z$d4d5Z%d6d7Z&d8d9Z'd:d;Z(Z)Z*Z+Z,Z-Z.Z/dd?Z1d@dAZ2dBdCZ3dDdEZ4dFdGZ5dHdIZ6dJdKZ7dQdMdNZ8Z9S)RAbstractPythonCodePrinterZ _pythoncodePythonNz cCsg|]\}}|d|fqSzmath.rcr_kvrcrcrd Usz$AbstractPythonCodePrinter.cCsi|]\}}|d|qSrvrcrwrcrcrd Wsz$AbstractPythonCodePrinter.r rr)r rrTFpython3)user_functions precisioninlinefully_qualified_modulesZcontractstandardcst||jd}|dkr4ddl}d|jj}|dkrDtd||_t t |_ t |j f|pbidi|_t |jf|p~idi|_dS)Nrrzpython{}r}zOnly Python 3 is supported.r~Zuser_constants)super__init__ _settingssysrk version_infomajor ValueErrorrrsetmodule_importsdict_kfgetrh_kcrr)rbsettingsZstdrrircrdrcs"   z"AbstractPythonCodePrinter.__init__cCs d||fSNz%s = %src)rbrfvaluercrcrd_declare_number_constwsz/AbstractPythonCodePrinter._declare_number_constcCsp|d}|r.)rrkrgrrcrrd _print_Modsz$AbstractPythonCodePrinter._print_ModcCsg}d}|jD]r}|j}|j}|dkr0|d|d||||d|d||||d|d7}q|dd}|ddkr|dd }|dn |d d |S) Nrr)z if z else rrr)z else None))rgrocondappendr]rm)rbroresultir`rPcrcrcrd_print_Piecewises(            z*AbstractPythonCodePrinter._print_PiecewisecsRddddddd}|j|krF||j}||j}dj|j||d St|S) z.Relational printer for Equality and UnequalityequalZ not_equallessZ less_equalZgreaterZ greater_equal)z==z!=z>=z({lhs} {op} {rhs}))rlhsrhs)rel_opr]rrrkr_print_Relational)rbrorrrrrcrdrs   z+AbstractPythonCodePrinter._print_RelationalcCsddlm}|||S)Nr) Piecewise)$sympy.functions.elementary.piecewiserr]rewrite)rbrorrcrcrd _print_ITEs z$AbstractPythonCodePrinter._print_ITEcs0fdd|jD}dj|jd|dS)Nc3s6|].\}}}dj|||dVqdS)zfor {i} in range({a}, {b}+1))rabN)rkr])r_rrrrarcrdres z7AbstractPythonCodePrinter._print_Sum..z"(builtins.sum({function} {loops})) )functionloops)limitsrkr]rrm)rbrorrcrard _print_Sums  z$AbstractPythonCodePrinter._print_SumcCsdS)NZ1jrcrrcrcrd_print_ImaginaryUnitsz.AbstractPythonCodePrinter._print_ImaginaryUnitcCs$|j\}}dj||||dS)Nz(1 if {a} == {b} else 0))rr)rgrkr])rbrorrrcrcrd_print_KroneckerDeltas  z/AbstractPythonCodePrinter._print_KroneckerDeltacCs,|jj}|j||}d|||fS)N%s(%s))rirjrhrr]tolist)rbrorffuncrcrcrd_print_MatrixBasesz+AbstractPythonCodePrinter._print_MatrixBasecCs ||Sr[)rrrcrcrdz"AbstractPythonCodePrinter.csdfdd|dDS)N csg|]}j|qSrc)tab)r_linerarcrdrzsz@AbstractPythonCodePrinter._indent_codestring..)rmrrrcrard_indent_codestringsz,AbstractPythonCodePrinter._indent_codestringcsNdfdd|jD}dj|jdfdd|jD|dS) Nrc3s|]}|VqdSr[r\r^rarcrdre!szFAbstractPythonCodePrinter._print_FunctionDefinition..z def {name}({parameters}): {body}rZcsg|]}|jqSrc)r]symbol)r_varrarcrdrz$szGAbstractPythonCodePrinter._print_FunctionDefinition..)rf parametersbody)rmrrkr]rfrr)rbfdrrcrard_print_FunctionDefinition s  z3AbstractPythonCodePrinter._print_FunctionDefinitioncs6dfdd|jD}dj|j|dS)Nrc3s|]}|VqdSr[r\r^rarcrdre)sz9AbstractPythonCodePrinter._print_While..zwhile {cond}: {body})rr)rmrrkr] conditionr)rbZwhlrrcrard _print_While(s  z&AbstractPythonCodePrinter._print_WhilecCs d||jj||jjfSr)r]variablerr)rbZdeclrcrcrd_print_Declaration/s  z,AbstractPythonCodePrinter._print_DeclarationcCs|j\}d||S)Nz return %s)rgr])rbretr`rcrcrd _print_Return5sz'AbstractPythonCodePrinter._print_Returncs^dfdd|jD}|jdkr8d|j|}|jdkrV|d|j7}d|S)NrZc3s|]}|VqdSr[r\r^rarcrdre:sz9AbstractPythonCodePrinter._print_Print..z {} % ({})z , file=%sz print(%s))rm print_args format_stringrkr]file)rbZprntrrcrard _print_Print9s   z&AbstractPythonCodePrinter._print_PrintcCs@t|jdkr|dSt|jdkr0|dS||jSdS)Nstdoutz sys.stdoutstderrz sys.stderr)strrfrlr])rbZstrmrcrcrd _print_StreamCs   z'AbstractPythonCodePrinter._print_StreamcCsdS)Nr&rc)rbr`rcrcrd_print_NoneTokenKsz*AbstractPythonCodePrinter._print_NoneToken math.sqrtc Cst|}|jtjkr<|s<||}||j}dj||dS|jr|s|j tjkr||}|tj }||j}|d|d|dS|jtj kr|tj }|j |j|dd}|d|S|j |j|dd}|j |j|dd} d|| S) aPrinting helper function for ``Pow`` Notes ===== This preprocesses the ``sqrt`` as math formatter and prints division 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)' >>> printer._hprint_Pow(1/x, rational=False) '1/x' >>> printer._hprint_Pow(1/x, rational=True) 'x**(-1)' 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}))rr`/rrF)strictz{}**{}) rr<rHalfrlr]baserkis_commutativeOne NegativeOner) rbrorationalrIrrr`numbase_strZexp_strrcrcrd _hprint_PowNs$+        z%AbstractPythonCodePrinter._hprint_Pow)N)T)Fr):rj __module__ __qualname__ printmethodlanguage_kwreserved_wordsmodulesrrr_known_functionsitems_known_functions_mathr_known_constants_mathr _operatorsr_default_settingsrrrlrrrrrrrrrrrrrrrrrZ_print_SparseRepMatrixZ_print_MutableSparseMatrixZ_print_ImmutableSparseMatrixZ _print_MatrixZ_print_DenseMatrixZ_print_MutableDenseMatrixZ_print_ImmutableMatrixZ_print_ImmutableDenseMatrixrrrrrrrrr __classcell__rcrcrrdrtMs~       rtc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS) ArrayPrintercCs4ddlm}z ||WStk r.|YSXdS)Nr)convert_indexed_to_array)Z4sympy.tensor.array.expressions.from_indexed_to_arrayr  Exception)rbindexedr rcrcrd _arrayifys   zArrayPrinter._arrayifycCs|}d}d}dd|D}g}|D]J}g} t|D].} ||krR| ||n | ||d7}q6|| q&i} g} g} |D]f} | D]T}|| krt|}|| |<n| |}||7}||kr|| kr| |q| |q|d7}q|dd}|| | fS)NrrcSs i|]}|D]}|t|q qSrc)r+)r_rjrcrcrdr{s z3ArrayPrinter._get_einsum_string..r,r) _get_letter_generator_for_einsumrangernext)rbsubrankscontraction_indicesletterscontraction_stringcounterdindicesZrank_argZlindicesrmapping letters_free letters_dumrlrcrcrd_get_einsum_strings<         zArrayPrinter._get_einsum_stringccs@tddD]}t|Vq tddD]}t|Vq$tddS)Na{A[zout of letters)rchrr)rbrrcrcrdrs   z-ArrayPrinter._get_letter_generator_for_einsumcsXdfdd|jD}djdj|dfdd|jDfS)Nrcs(g|] }dfddt|DqS)rcsg|] }tqSrc)r)r_rrrcrdrzszEArrayPrinter._print_ArrayTensorProduct...)rmrr_rr'rcrdrzsz:ArrayPrinter._print_ArrayTensorProduct.. %s("%s", %s)rrZcsg|]}|qSrcr\r^rarcrdrzs)rrmrrl_module_einsumrg)rbrorrc)rrbrd_print_ArrayTensorProductsz&ArrayPrinter._print_ArrayTensorProductc sddlm}|j}|j}t||rDdfdd|jD}|j}n|}t |j g} ||\}}} |sz|St||rdfdd|jD}n |}d j djd |d t||fS) NrArrayTensorProductrcsg|]}d|qSz%sr\r^rarcrdrzsz8ArrayPrinter._print_ArrayContraction..csg|]}d|qSr/r\r^rarcrdrzsr)r{}->{}r)0sympy.tensor.array.expressions.array_expressionsr.ror isinstancermrgrr]rshaper!rlr*r+rksorted) rbror.rrelemsZranksrrrrcrard_print_ArrayContractions&       z$ArrayPrinter._print_ArrayContractionc sddlm}t|j}t|j|r4|jj}|jj}n|j}|jg}||\}}}fdd|D}d j dj d |d ||d |fS) Nrr-csg|]}|qSrcr\r(rarcrdrzsz5ArrayPrinter._print_ArrayDiagonal..r)rr0rrZ)r1r.listdiagonal_indicesr2rorrgr!rlr*r+rkrm) rbror.r8rr5Zdiagonal_stringrrrcrard_print_ArrayDiagonals    z!ArrayPrinter._print_ArrayDiagonalcCs2d||jd|j||j||jjfS)Nrr)rlr*Z _transposer]roZ permutation array_formrrcrcrd_print_PermuteDimss   zArrayPrinter._print_PermuteDimscCs||jd|j|jS)Nr)rr*_addrgrrcrcrd_print_ArrayAddszArrayPrinter._print_ArrayAddcCs.d||jd|jdt|j|jfSNz %s((%s,))rr)rlr*Z_onesrmmapr]rgrrcrcrd_print_OneArrayszArrayPrinter._print_OneArraycCs.d||jd|jdt|j|jfSr>)rlr*Z_zerosrmr?r]rgrrcrcrd_print_ZeroArrayszArrayPrinter._print_ZeroArraycCs0|||j}|||j}d||fSr)r]rrr)rbrorrrcrcrd_print_AssignmentszArrayPrinter._print_AssignmentcCs ||Sr[)_print_ArraySymbolrrcrcrd_print_IndexedBaseszArrayPrinter._print_IndexedBaseN)rjrrrr!rr,r6r9r;r=r@rArBrDrcrcrcrdr s# r csneZdZddZddZddZddd Zd d Zd d ZddZ fddZ e j Z e j Ze j Ze j ZZS)PythonCodePrintercCs dj|d||jddS)Nz"(0.0 if {e} == 0 else {f}(1, {e}))z math.copysignr)frPrkrlr]rgrbrPrcrcrd _print_signszPythonCodePrinter._print_signcCs$t|}|jd||jd|S)Nrr)rr rrg)rbrorrcrcrd _print_NotszPythonCodePrinter._print_Notcs<|jd}|jdd}dt|dfdd|DS)Nrrz{}[{}]rZcsg|]}|qSrcr\)r_indrarcrdrz&sz4PythonCodePrinter._print_Indexed..)rgrkrrm)rbrorindexrcrard_print_Indexed#s z PythonCodePrinter._print_IndexedFcCs|j||dS)N)rrrbrorrcrcrd _print_Pow(szPythonCodePrinter._print_PowcCsd|j|jS)Nz{}/{})rkpqrrcrcrd_print_Rational+sz!PythonCodePrinter._print_RationalcCs ||Sr[rSrrcrcrd _print_Half.szPythonCodePrinter._print_HalfcCs|t|jddS)Nrr)rrrgrrcrcrd _print_frac1szPythonCodePrinter._print_fraccsdt|}||jkr@|jdr2d}t||||jdSd|kr\|ddddS|SdS)NZerror_on_reservedzVThis expression includes the symbol "{}" which is a reserved keyword in this language.Zreserved_word_suffix{r})r _print_Symbolrrrrkreplace)rbrorfmsgrrcrdrY4s   zPythonCodePrinter._print_Symbol)F)rjrrrIrJrMrPrSrUrVrYrZ_print_not_supported_print_lowergamma_print_uppergamma_print_fresnelc_print_fresnelsr rcrcrrdrEs  rEz _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' )rEdoprint)rorrcrcrdpycodePsraz log1p log2cCs g|]\}}|tkr||fqSrc)_not_in_mpmathrwrcrcrdrzksrzbetafracfresnelcfresnelssignrMhypermeijergbesseljbesselybesselibesselk) rcrdrerfrgrMrhrirjrkrlrmphieulercatalanninf)rTrU GoldenRatio EulerGammaCatalanrXrWNegativeInfinitycCsTg}g}|jD]<}t|dkr*|\}}}ntd|||||fq||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)rrNotImplementedErrorr)Z integral_exprintegration_varsrZintegration_rangeZintegration_varZ lower_limitZ upper_limitrcrcrd_unpack_integral_limitss    ryc@seZdZdZdZdZeee dde DZ dde DZ dd Zd d Zd d ZddZddZddZddZdddZddZdS) MpmathPrinterzH Lambda printer for mpmath which maintains precision for floats Z _mpmathcodezPython with mpmathcCsg|]\}}|d|fqSzmpmath.rcrwrcrcrdrzszMpmathPrinter.cCsi|]\}}|d|qSr{rcrwrcrcrdr{szMpmathPrinter.cCs(tttt|j}dj|d|dS)Nz{func}({args}) mpmath.mpf)rrg)rtupler?int_mpf_rkrl)rbrPrgrcrcrd _print_FloatszMpmathPrinter._print_FloatcCs&dj|d||j||jdS)Nz{func}({p})/{func}({q})r|)rrRrQ)rkrlr]rRrQrHrcrcrdrSs   zMpmathPrinter._print_RationalcCs ||Sr[rTrHrcrcrdrUszMpmathPrinter._print_HalfcCs4d|d||jd||jd|dS)Nz{}({}, {}, {})mpmath.gammaincrrz mpmath.infrGrHrcrcrdr]s zMpmathPrinter._print_uppergammacCs,d|d||jd||jdS)Nz {}({}, 0, {})rrrrGrHrcrcrdr\s zMpmathPrinter._print_lowergammacCsd|d||jdS)Nz{0}({1})/{0}(2)z mpmath.logrrGrHrcrcrd _print_log2szMpmathPrinter._print_log2cCsd|d||jdS)Nz{}({})z mpmath.log1prrGrHrcrcrd _print_log1pszMpmathPrinter._print_log1pFcCs|j||ddS)Nz mpmath.sqrtrrIrNrOrcrcrdrPszMpmathPrinter._print_Powc sPt|\}}dddtj||jddfdd|DS)Nz{}(lambda {}: {}, {})z mpmath.quadrZrc3s"|]}dttj|VqdS)z(%s, %s)N)r}r?r])r_r rarcrdresz0MpmathPrinter._print_Integral..)ryrkrlrmr?r]rg)rbrPrxrrcrard_print_Integrals zMpmathPrinter._print_IntegralN)F)rjrr__doc__rrrrrr_known_functions_mpmathr_known_constants_mpmathrrrSrUr]r\rrrPrrcrcrcrdrzs"  rzc@s"eZdZdZddZdddZdS) SymPyPrinterzPython with SymPycsH|jjp d}d||rdnd|jjdfdd|jDfS)NrrrrZc3s|]}|VqdSr[r\r^rarcrdresz/SymPyPrinter._print_Function..)rrrlrjrmrg)rbromodrcrard_print_Functions zSymPyPrinter._print_FunctionFcCs|j||ddS)Nz sympy.sqrtrrNrOrcrcrdrPszSymPyPrinter._print_PowN)F)rjrrrrrPrcrcrcrdrsrN)#r collectionsr itertoolsr sympy.corerZsympy.core.modrr codeprinterrrrrrrqrsrtr rErrxsetattrrarrbrZ _in_mpmathrrrryrzrrcrcrcrds      ! F 0  D