a RG5d/@sdZddlmZmZddlmZddlmZddl m Z m Z dddd d d d d ddddddddddddddZ GdddeZ d#dd Zd!d"ZdS)$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. )AnyDict)S) 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@seZdZdZdZdZdddidddd Zifd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3ZdS)4JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code Z _javascript JavaScriptNautoTF)order full_prec precisionuser_functionshumanallow_unknown_functionscontractcCs2t||tt|_|di}|j|dS)Nr$)r__init__dictknown_functionsgetupdate)selfsettings userfuncsr0Q/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/printing/jscode.pyr(<s   zJavascriptCodePrinter.__init__cCs|dS)Nr0)r-pr0r0r1_rate_index_positionBsz*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r0)r- codestringr0r0r1_get_statementEsz$JavascriptCodePrinter._get_statementcCs d|S)Nz// {})format)r-textr0r0r1 _get_commentHsz"JavascriptCodePrinter._get_commentcCsd|||jdS)Nz var {} = {};r#)r7evalf _settings)r-namevaluer0r0r1_declare_number_constKsz+JavascriptCodePrinter._declare_number_constcCs ||SN) indent_code)r-linesr0r0r1 _format_codeNsz"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdSr?)range).0ijcolsr0r1 SzAJavascriptCodePrinter._traverse_matrix_indices..)shaperC)r-matrowsr0rGr1_traverse_matrix_indicesQs 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 loopstartrEr0r0r1_get_loop_opening_endingUs    z.JavascriptCodePrinter._get_loop_opening_endingcCs~t|}|jdkr$d||j|S|jdkr>d||jS|jtjdkr^d||jSd||j||jfSdS)Nz1/%sg?z Math.sqrt(%s)z Math.cbrt(%s)zMath.pow(%s, %s))rr parenthesizebaserUrOne)r-exprPRECr0r0r1 _print_Powbs    z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr3q)r-rcr3rgr0r0r1_print_Rationalnsz%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr8|jsD|jrR|jrR|d|Sd|d|d|d|S)Ncsg|]}|qSr0)r`)rDargrdr-r0r1 urJz4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsris_nonnegativeis_nonpositive)r-rcnumdensnumZsdenr0rjr1 _print_Modrs  z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rUlhsrhsrel_opr7)r-rclhs_coderhs_codeopr0r0r1_print_Relational~s  z'JavascriptCodePrinter._print_RelationalcCs`|j}tj}tj}tt|jD]"}||j||7}|||9}q d||j j ||fS)Nz%s[%s]) rKrZerorbreversedrCrankrYrUrarV)r-rcdimselemoffsetrEr0r0r1_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jSr?)rUrVr-rcr0r0r1 _print_Idxsz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er0rr0r0r1 _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr0rr0r0r1 _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr0rr0r0r1_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr0rr0r0r1_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-rcrrArErrcode0ecpairs last_liner0rr1_print_Piecewises(         z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]AtomT)strictrO)r7r`parentrrFrErKrr0r0r1_print_MatrixElements  z*JavascriptCodePrinter._print_MatrixElementc st|tr$||d}d|Sd}dddd|D}fdd|D}fd d|D}g}d }t|D]J\}} | d vr|| qr|||8}|d ||| f|||7}qr|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )rSrcSsg|]}|dqS)z )lstriprDliner0r0r1rkrJz5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr0)rfanymapendswithr) inc_tokenr0r1rkrJcs g|]}ttt|jqSr0)rfrr startswithr) dec_tokenr0r1rksr)rrz%s%s) isinstancestrr@ splitlinesrrrT) r-code code_linestabincreasedecreaseprettylevelnrr0)rrr1r@s*     z!JavascriptCodePrinter.indent_code)__name__ __module__ __qualname____doc__ printmethodlanguage_default_settingsr(r4r6r9r>rBrNr]rerhrrryrrrrrrrrr@r0r0r0r1r,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.r0r0r1jscodesircKstt|fi|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rcr.r0r0r1 print_jscodeLsr)N)rtypingrrtDict sympy.corerZsympy.printing.codeprinterrsympy.printing.precedencerrr*rrrr0r0r0r1s<   5 l