U ;qLec/@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z dd d d d d ddddddddddddddddZGddde Zd%d!d"Zd#d$Zd S)&z Javascript code printer The JavascriptCodePrinter converts single SymPy expressions into single Javascript expressions, using the functions defined in the Javascript Math object where possible. ) annotations)Any)S) equal_valued) CodePrinter) precedence PRECEDENCEzMath.absz Math.acosz Math.acoshz Math.asinz Math.asinhz Math.atanz Math.atan2z Math.atanhz Math.ceilzMath.cosz Math.coshzMath.expz Math.floorzMath.logzMath.maxzMath.minz Math.signzMath.sinz Math.sinhzMath.tanz Math.tanh)Absacosacoshasinasinhatanatan2atanhceilingcoscoshexpfloorlogMaxMinsignsinsinhtantanhc@seZdZUdZdZdZdddidddd Zd ed <ifd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5ZdS)6JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code Z _javascriptZ JavaScriptNautoTF)order full_prec precisionuser_functionshumanallow_unknown_functionscontractzdict[str, Any]_default_settingscCs2t||tt|_|di}|j|dS)Nr$)r__init__dictknown_functionsgetupdate)selfsettings userfuncsr1l/home/p21-0144/sympy/latex2sympy2solve-back-end/sympyEq/lib/python3.8/site-packages/sympy/printing/jscode.pyr)>s   zJavascriptCodePrinter.__init__cCs|dS)Nr1)r.pr1r1r2_rate_index_positionDsz*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r1)r. codestringr1r1r2_get_statementGsz$JavascriptCodePrinter._get_statementcCs d|S)Nz// {})format)r.textr1r1r2 _get_commentJsz"JavascriptCodePrinter._get_commentcCsd|||jdS)Nz var {} = {};r#)r8evalf _settings)r.namevaluer1r1r2_declare_number_constMsz+JavascriptCodePrinter._declare_number_constcCs ||SN) indent_code)r.linesr1r1r2 _format_codePsz"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdSr@)range).0ijcolsr1r2 Us zAJavascriptCodePrinter._traverse_matrix_indices..)shaperD)r.matrowsr1rHr2_traverse_matrix_indicesSs z.JavascriptCodePrinter._traverse_matrix_indicesc CsZg}g}d}|D]@}||||j||j||jdd|dq||fS)NzAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){)varblestartend})append_printlabellowerupper)r.indices open_lines close_lines loopstartrFr1r1r2_get_loop_opening_endingWs    z.JavascriptCodePrinter._get_loop_opening_endingcCst|}t|jdr&d||j|St|jdrBd||jS|jtjdkrbd||jSd||j||jfSdS)Nz1/%sg?z Math.sqrt(%s)z Math.cbrt(%s)zMath.pow(%s, %s))rrr parenthesizebaserUrOne)r.exprPRECr1r1r2 _print_Powds    z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr4q)r.rcr4rgr1r1r2_print_Rationalpsz%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr8|jsD|jrR|jrR|d|Sd|d|d|d|S)Ncsg|]}|qSr1)r`)rEargrdr.r1r2 wsz4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsris_nonnegativeis_nonpositive)r.rcnumdenZsnumZsdenr1rjr2 _print_Modts  z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rUlhsrhsrel_opr8)r.rclhs_coderhs_codeopr1r1r2_print_Relationals  z'JavascriptCodePrinter._print_RelationalcCs`|j}tj}tj}tt|jD]"}||j||7}|||9}q d||j j ||fS)Nz%s[%s]) rKrZerorbreversedrDrankrYrUrarV)r.rcdimselemoffsetrFr1r1r2_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jSr@)rUrVr.rcr1r1r2 _print_Idxsz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er1rr1r1r2 _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr1rr1r1r2 _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr1rr1r1r2_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr1rr1r1r2_print_NegativeInfinitysz-JavascriptCodePrinter._print_NegativeInfinityc s"ddlm}|jdjdkr$tdg}||rt|jD]\}\}}|dkrf|d|n:|t |jdkr|dkr|dn|d ||}|||d q.z: ( %s )z:  )) sympy.codegen.astrrlcond ValueErrorhas enumeraterTrUlenjoinrc) r.rcrrBrFrrcode0ecpairs last_liner1rr2_print_Piecewises(         z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]AtomT)strictrO)r8r`parentrrGrFrKrr1r1r2_print_MatrixElements  z*JavascriptCodePrinter._print_MatrixElementc st|tr$||d}d|Sd}dddd|D}fdd|D}fd d|D}g}d }t|D]J\}} | d kr|| qr|||8}|d ||| f|||7}qr|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )rSrcSsg|]}|dqS)z )lstriprEliner1r1r2rksz5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr1)rfanymapendswithr) inc_tokenr1r2rkscs g|]}ttt|jqSr1)rfrr startswithr) dec_tokenr1r2rksr)rrz%s%s) isinstancestrrA splitlinesrrrT) r.code code_linestabincreasedecreaseprettylevelnrr1)rrr2rAs*     z!JavascriptCodePrinter.indent_code)__name__ __module__ __qualname____doc__ printmethodlanguager(__annotations__r)r5r7r:r?rCrNr]rerhrqrxrrrrrrrrrAr1r1r1r2r.s@      !rNcKst|||S)aConverts an expr to a string of javascript code Parameters ========== expr : Expr A SymPy expression to be converted. assign_to : optional When given, the argument is used as the name of the variable to which the expression is assigned. Can be a string, ``Symbol``, ``MatrixSymbol``, or ``Indexed`` type. This is helpful in case of line-wrapping, or for expressions that generate multi-line statements. precision : integer, optional The precision for numbers such as pi [default=15]. user_functions : dict, optional A dictionary where keys are ``FunctionClass`` instances and values are their string representations. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, js_function_string)]. See below for examples. human : bool, optional If True, the result is a single string that may contain some constant declarations for the number symbols. If False, the same information is returned in a tuple of (symbols_to_declare, not_supported_functions, code_text). [default=True]. contract: bool, optional If True, ``Indexed`` instances are assumed to obey tensor contraction rules and the corresponding nested loops over indices are generated. Setting contract=False will not generate loops, instead the user is responsible to provide values for the indices in the code. [default=True]. Examples ======== >>> from sympy import jscode, symbols, Rational, sin, ceiling, Abs >>> x, tau = symbols("x, tau") >>> jscode((2*tau)**Rational(7, 2)) '8*Math.sqrt(2)*Math.pow(tau, 7/2)' >>> jscode(sin(x), assign_to="s") 's = Math.sin(x);' Custom printing can be defined for certain types by passing a dictionary of "type" : "function" to the ``user_functions`` kwarg. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, js_function_string)]. >>> custom_functions = { ... "ceiling": "CEIL", ... "Abs": [(lambda x: not x.is_integer, "fabs"), ... (lambda x: x.is_integer, "ABS")] ... } >>> jscode(Abs(x) + ceiling(x), user_functions=custom_functions) 'fabs(x) + CEIL(x)' ``Piecewise`` expressions are converted into conditionals. If an ``assign_to`` variable is provided an if statement is created, otherwise the ternary operator is used. Note that if the ``Piecewise`` lacks a default term, represented by ``(expr, True)`` then an error will be thrown. This is to prevent generating an expression that may not evaluate to anything. >>> from sympy import Piecewise >>> expr = Piecewise((x + 1, x > 0), (x, True)) >>> print(jscode(expr, tau)) if (x > 0) { tau = x + 1; } else { tau = x; } Support for loops is provided through ``Indexed`` types. With ``contract=True`` these expressions will be turned into loops, whereas ``contract=False`` will just print the assignment expression that should be looped over: >>> from sympy import Eq, IndexedBase, Idx >>> len_y = 5 >>> y = IndexedBase('y', shape=(len_y,)) >>> t = IndexedBase('t', shape=(len_y,)) >>> Dy = IndexedBase('Dy', shape=(len_y-1,)) >>> i = Idx('i', len_y-1) >>> e=Eq(Dy[i], (y[i+1]-y[i])/(t[i+1]-t[i])) >>> jscode(e.rhs, assign_to=e.lhs, contract=False) 'Dy[i] = (y[i + 1] - y[i])/(t[i + 1] - t[i]);' Matrices are also supported, but a ``MatrixSymbol`` of the same dimensions must be provided to ``assign_to``. Note that any expression that can be generated normally can also exist inside a Matrix: >>> from sympy import Matrix, MatrixSymbol >>> mat = Matrix([x**2, Piecewise((x + 1, x > 0), (x, True)), sin(x)]) >>> A = MatrixSymbol('A', 3, 1) >>> print(jscode(mat, A)) A[0] = Math.pow(x, 2); if (x > 0) { A[1] = x + 1; } else { A[1] = x; } A[2] = Math.sin(x); )rdoprint)rc assign_tor/r1r1r2jscodesircKstt|f|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rcr/r1r1r2 print_jscodeNsr)N)r __future__rtypingr sympy.corersympy.core.numbersrZsympy.printing.codeprinterrsympy.printing.precedencerrr+rrrr1r1r1r2s@     5 l