U Zʹj.a!@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@dAdBdBdCdDdEdFdGdHdIdJdKZdLdMdLdNdOdOdPZdQdRZdSdTZGdUdVdVe ZGdWdXdXZGdYdZdZeZejD]Zeed[eeq6eD]Zeed[eeqRd\d]Zd^Zd_d`eDZeefdadbdcdddedfdgdhdidjdkdldm ZdLdMdndodpdOdNdqdrZdsdtZ GdudvdveZ!e!jD]Zee!d[eeqeD]Zee!d[eeqGdwdxdxeZ"dyS)zzy 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)ZAbsZMinZMaxacosacoshasinasinhatanatan2atanhceilcoscosherferfcexpexpm1 factorialfloorgammahypotisnanlgammaloglog10log1plog2sinsinhsqrttantanh)r-r.r/r0r1r2r3Zceilingr5r6r7r8r9r:r;r<r=r>r?loggammarAlnrBrCrDrErFZSqrtrHrIepiinfnan)Exp1PiEInfinityNaNZComplexInfinitycs8j|jj}dj|dfdd|jDdS)Nz{name}({args}), c3s|]}|VqdSN_print.0argselfT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/printing/pycode.py Fsz$_print_known_func..)nameargs)known_functions __class____name__format_module_formatjoinrbr]exprZknownr^r\r__print_known_funcCs rkcCs|j|jj}||SrV)known_constantsrdrergrir^r^r__print_known_constIsrmc seZdZdZdZeZdZdZe e e dde DZdde DZd d d d Ze ejid dddddZdUfdd ZddZdVddZddZddZddZddZd 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@dAZ2dBdCZ3dDdEZ4dFdGZ5dHdIZ6dJdKZ7dLdMZ8dNdOZ9dPdQZ:dWdSdTZ;ZVsz$AbstractPythonCodePrinter.cCsi|]\}}|d|qSrpr^rqr^r^r_ Xsz$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)Nr|rzpython{}rwzOnly Python 3 is supported.rxZuser_constants)super__init__ _settingssysrf version_infomajor ValueErrorr|rsetmodule_importsdict_kfgetrc_kcrl)r]settingsZstdrrdr^r_r~ds"   z"AbstractPythonCodePrinter.__init__cCs d||fSNz%s = %sr^)r]ravaluer^r^r__declare_number_constxsz/AbstractPythonCodePrinter._declare_number_constcCsp|d}|r.)rrfrbrr^rr_ _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))rbrjcondappendrXrh)r]rjresultir[rLcr^r^r__print_Piecewises(            z*AbstractPythonCodePrinter._print_PiecewisecsRddddddd}|j|krF||j}||j}dj|j||d St|S) z.Relational printer for Equality and Unequalityequal not_equallessZ less_equalZgreaterZ greater_equal)z==z!=z>=z({lhs} {op} {rhs}))rlhsrhs)Zrel_oprXrrrfr}_print_Relational)r]rjrrrrr^r_rs   z+AbstractPythonCodePrinter._print_RelationalcCsddlm}|||S)Nr) Piecewise)Z$sympy.functions.elementary.piecewiserrXZrewrite)r]rjrr^r^r_ _print_ITEs z$AbstractPythonCodePrinter._print_ITEcs0fdd|jD}dj|jd|dS)Nc3s6|].\}}}dj|||dVqdS)zfor {i} in range({a}, {b}+1))rabN)rfrX)rZrrrr\r^r_r`s z7AbstractPythonCodePrinter._print_Sum..z"(builtins.sum({function} {loops})) )functionloops)limitsrfrXrrh)r]rjrr^r\r_ _print_Sums  z$AbstractPythonCodePrinter._print_SumcCsdS)NZ1jr^rr^r^r__print_ImaginaryUnitsz.AbstractPythonCodePrinter._print_ImaginaryUnitcCs$|j\}}dj||||dS)Nz(1 if {a} == {b} else 0))rr)rbrfrX)r]rjrrr^r^r__print_KroneckerDeltas  z/AbstractPythonCodePrinter._print_KroneckerDeltacCs,|jj}|j||}d|||fS)N%s(%s))rdrercrrXtolist)r]rjrafuncr^r^r__print_MatrixBasesz+AbstractPythonCodePrinter._print_MatrixBasecCs ||SrV)rrr^r^r_z"AbstractPythonCodePrinter.csdfdd|dDS)N csg|]}j|qSr^)tab)rZliner\r^r_rtsz@AbstractPythonCodePrinter._indent_codestring..)rhrrr^r\r__indent_codestringsz,AbstractPythonCodePrinter._indent_codestringcsNdfdd|jD}dj|jdfdd|jD|dS) Nrc3s|]}|VqdSrVrWrYr\r^r_r`"szFAbstractPythonCodePrinter._print_FunctionDefinition..z def {name}({parameters}): {body}rUcsg|]}|jqSr^)rXsymbol)rZvarr\r^r_rt%szGAbstractPythonCodePrinter._print_FunctionDefinition..)ra parametersbody)rhrrfrXrarr)r]fdrr^r\r__print_FunctionDefinition!s  z3AbstractPythonCodePrinter._print_FunctionDefinitioncs6dfdd|jD}dj|j|dS)Nrc3s|]}|VqdSrVrWrYr\r^r_r`*sz9AbstractPythonCodePrinter._print_While..zwhile {cond}: {body})rr)rhrrfrX conditionr)r]Zwhlrr^r\r_ _print_While)s  z&AbstractPythonCodePrinter._print_WhilecCs d||jj||jjfSr)rXvariablerr)r]declr^r^r__print_Declaration0s  z,AbstractPythonCodePrinter._print_DeclarationcCsdS)Nr r^)r]Zbtr^r^r__print_BreakToken6sz+AbstractPythonCodePrinter._print_BreakTokencCs|j\}d||S)Nz return %srbrX)r]retr[r^r^r_ _print_Return9sz'AbstractPythonCodePrinter._print_ReturncCs|j\}d||S)Nzraise %sr)r]rsr[r^r^r_ _print_Raise=sz&AbstractPythonCodePrinter._print_RaisecCs|j\}d||S)NzRuntimeError(%s)r)r]remessager^r^r__print_RuntimeError_Asz.AbstractPythonCodePrinter._print_RuntimeError_csjdfdd|jD}ddlm}|j|krDd|j|}|jdkrb|d|j7}d|S) NrUc3s|]}|VqdSrVrWrYr\r^r_r`Fsz9AbstractPythonCodePrinter._print_Print..r)nonez{} % ({}), end=""z , file=%sz print(%s))rh print_argsZsympy.codegen.astr format_stringrfrXfile)r]Zprntrrr^r\r_ _print_PrintEs    z&AbstractPythonCodePrinter._print_PrintcCs@t|jdkr|dSt|jdkr0|dS||jSdS)Nstdoutz sys.stdoutstderrz sys.stderr)strrargrX)r]strmr^r^r_ _print_StreamQs   z'AbstractPythonCodePrinter._print_StreamcCsdS)Nr&r^)r]r[r^r^r__print_NoneTokenYsz*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[/rrFstrictz{}**{}) rr9rZHalfrgrXbaserfZis_commutativeZOneZ NegativeOner) r]rjrationalrGrrr[numZbase_strZexp_strr^r^r_ _hprint_Pow\s$+        z%AbstractPythonCodePrinter._hprint_Pow)N)T)Fr)=re __module__ __qualname__ printmethodlanguage_kwreserved_wordsmodulesrrr_known_functionsitems_known_functions_mathr_known_constants_mathr _operatorsr_default_settingsr~rrgrrrrrrrrrrrrrrrrrZ_print_SparseRepMatrixZ_print_MutableSparseMatrixZ_print_ImmutableSparseMatrixZ _print_MatrixZ_print_DenseMatrixZ_print_MutableDenseMatrixZ_print_ImmutableMatrixZ_print_ImmutableDenseMatrixrrrrrrrrrrrr __classcell__r^r^rr_rnNs       rnc@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)r]Zindexedr r^r^r_ _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 qSr^)r+)rZrjr^r^r_rus z3ArrayPrinter._get_einsum_string..r,r) _get_letter_generator_for_einsumrangernext)r]subrankscontraction_indicesletterscontraction_stringcounterdindicesZrank_argZlindicesrmapping letters_free letters_dumrlr^r^r__get_einsum_strings<         zArrayPrinter._get_einsum_stringccs@tddD]}t|Vq tddD]}t|Vq$tddS)Na{A[zout of letters)rchrr)r]rr^r^r_rs   z-ArrayPrinter._get_letter_generator_for_einsumcsXdfdd|jD}djdj|dfdd|jDfS)Nrcs(g|] }dfddt|DqS)rcsg|] }tqSr^)r)rZrrr^r_rtszEArrayPrinter._print_ArrayTensorProduct...)rhrrZrr$r^r_rtsz:ArrayPrinter._print_ArrayTensorProduct.. %s("%s", %s)rrUcsg|]}|qSr^rWrYr\r^r_rts)rrhrrg_module_einsumrb)r]rjrr^)rr]r__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%srWrYr\r^r_rtsz8ArrayPrinter._print_ArrayContraction..csg|]}d|qSr,rWrYr\r^r_rtsr&r{}->{}r)0sympy.tensor.array.expressions.array_expressionsr+rjr isinstancerhrbrrXrshaperrgr'r(rfsorted) r]rjr+rrelemsZranksrrrr^r\r__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|]}|qSr^rWr%r\r^r_rtsz5ArrayPrinter._print_ArrayDiagonal..r&rr-rrU)r.r+listdiagonal_indicesr/rjrrbrrgr'r(rfrh) r]rjr+r5rr2Zdiagonal_stringrrr^r\r__print_ArrayDiagonals    z!ArrayPrinter._print_ArrayDiagonalcCs2d||jd|j||j||jjfS)Nrr)rgr'Z _transposerXrjZ permutationZ array_formrr^r^r__print_PermuteDimss   zArrayPrinter._print_PermuteDimscCs||jd|j|jS)Nr)rr'_addrbrr^r^r__print_ArrayAdd szArrayPrinter._print_ArrayAddcCs.d||jd|jdt|j|jfSNz %s((%s,))rr)rgr'Z_onesrhmaprXrbrr^r^r__print_OneArrayszArrayPrinter._print_OneArraycCs.d||jd|jdt|j|jfSr:)rgr'Z_zerosrhr;rXrbrr^r^r__print_ZeroArrayszArrayPrinter._print_ZeroArraycCs0|||j}|||j}d||fSr)rXr rr)r]rjrrr^r^r__print_AssignmentszArrayPrinter._print_AssignmentcCs ||SrV)Z_print_ArraySymbolrr^r^r__print_IndexedBase#szArrayPrinter._print_IndexedBaseN)rerrr rrr)r3r6r7r9r<r=r>r?r^r^r^r_r s# r csveZdZddZddZddZddZdd d Zd d ZddZ ddZ fddZ e j Ze j Ze j Ze j ZZS)PythonCodePrintercCs dj|d||jddS)Nz"(0.0 if {e} == 0 else {f}(1, {e}))z math.copysignr)frLrfrgrXrbr]rLr^r^r_ _print_sign)szPythonCodePrinter._print_signcCs$t|}|jd||jd|S)Nrr)rrrrb)r]rjrr^r^r_ _print_Not-szPythonCodePrinter._print_NotcCs|jSrV)rarr^r^r_r?1sz$PythonCodePrinter._print_IndexedBasecs<|jd}|jdd}dt|dfdd|DS)Nrrz{}[{}]rUcsg|]}|qSr^rW)rZindr\r^r_rt7sz4PythonCodePrinter._print_Indexed..)rbrfrrh)r]rjrindexr^r\r__print_Indexed4s z PythonCodePrinter._print_IndexedFcCs|j||dS)N)rrr]rjrr^r^r_ _print_Pow9szPythonCodePrinter._print_PowcCsd|j|jS)Nz{}/{})rfpqrr^r^r__print_Rational<sz!PythonCodePrinter._print_RationalcCs ||SrVrNrr^r^r_ _print_Half?szPythonCodePrinter._print_HalfcCs|t|jddS)Nrr)rrrbrr^r^r_ _print_fracBszPythonCodePrinter._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_Symbolrrrrfreplace)r]rjramsgrr^r_rTEs   zPythonCodePrinter._print_Symbol)F)rerrrDrEr?rHrKrNrPrQrTrZ_print_not_supported_print_lowergamma_print_uppergammaZ_print_fresnelcZ_print_fresnelsr r^r^rr_r@'s  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``. 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' )r@Zdoprint)rjrr^r^r_pycodeasrYz log1p log2cCs g|]\}}|tkr||fqSr^)_not_in_mpmathrqr^r^r_rt|srtbetafracfresnelcfresnelssignrJhypermeijergbesseljbesselybesselibesselk) r[r\r]r^r_rJr`rarbrcrdrephiZeulercatalanZninf)rPrQZ GoldenRatioZ EulerGammaZCatalanrTrSNegativeInfinitycCsTg}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_limitr^r^r__unpack_integral_limitss    rlc@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.r^rqr^r^r_rtszMpmathPrinter.cCsi|]\}}|d|qSrnr^rqr^r^r_ruszMpmathPrinter.cCs(tttt|j}dj|d|dS)Nz{func}({args}) mpmath.mpf)rrb)rtupler;intZ_mpf_rfrg)r]rLrbr^r^r_ _print_FloatszMpmathPrinter._print_FloatcCs&dj|d||j||jdS)Nz{func}({p})/{func}({q})ro)rrMrL)rfrgrXrMrLrCr^r^r_rNs   zMpmathPrinter._print_RationalcCs ||SrVrOrCr^r^r_rPszMpmathPrinter._print_HalfcCs4d|d||jd||jd|dS)Nz{}({}, {}, {})mpmath.gammaincrrz mpmath.infrBrCr^r^r_rXs zMpmathPrinter._print_uppergammacCs,d|d||jd||jdS)Nz {}({}, 0, {})rsrrrBrCr^r^r_rWs zMpmathPrinter._print_lowergammacCsd|d||jdS)Nz{0}({1})/{0}(2)z mpmath.logrrBrCr^r^r_ _print_log2szMpmathPrinter._print_log2cCsd|d||jdS)Nz{}({})z mpmath.log1prrBrCr^r^r_ _print_log1pszMpmathPrinter._print_log1pFcCs|j||ddS)Nz mpmath.sqrtrrGrIrJr^r^r_rKszMpmathPrinter._print_Powc sPt|\}}dddtj||jddfdd|DS)Nz{}(lambda {}: {}, {})z mpmath.quadrUrc3s"|]}dttj|VqdS)z(%s, %s)N)rpr;rX)rZrr\r^r_r`sz0MpmathPrinter._print_Integral..)rlrfrgrhr;rXrb)r]rLrkrr^r\r__print_Integrals zMpmathPrinter._print_IntegralN)F)rerr__doc__rrrrrr_known_functions_mpmathr_known_constants_mpmathrrrrNrPrXrWrtrurKrwr^r^r^r_rms"  rmc@s0eZdZdZeejddZddZd ddZdS) SymPyPrinterzPython with SymPyFrcsH|jjp d}d||rdnd|jjdfdd|jDfS)NrrrrUc3s|]}|VqdSrVrWrYr\r^r_r`sz/SymPyPrinter._print_Function..)rrrgrerhrb)r]rjmodr^r\r__print_Functions zSymPyPrinter._print_FunctioncCs|j||ddS)Nz sympy.sqrtrvrIrJr^r^r_rKszSymPyPrinter._print_PowN)F) rerrrrrnrr}rKr^r^r^r_r{sr{N)#rx collectionsr itertoolsrZ sympy.corerZsympy.core.modrrZ codeprinterrrrrrrkrmrnr r@rrrsetattrrYrrZrZ _in_mpmathrryrzrlrmr{r^r^r^r_s      " S 3  D