U Zʹj/@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)ZAbsacosacoshasinasinhatanatan2atanhZceilingcoscoshexpfloorlogZMaxZMinsignsinsinhtantanhc@seZdZUdZdZdZeejfdiddZde d<ifd d Z d d Z d dZ ddZ ddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3S)4JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code Z _javascript JavaScriptT) precisionuser_functionsZcontractzdict[str, Any]_default_settingscCs2t||tt|_|di}|j|dS)Nr)r__init__dictknown_functionsgetupdate)selfsettingsZ userfuncsr'T/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/printing/jscode.pyr :s   zJavascriptCodePrinter.__init__cCs|dS)Nr')r%pr'r'r(_rate_index_position@sz*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r')r%Z codestringr'r'r(_get_statementCsz$JavascriptCodePrinter._get_statementcCs d|S)Nz// {})format)r%textr'r'r( _get_commentFsz"JavascriptCodePrinter._get_commentcCsd|||jdS)Nz var {} = {};r)r-ZevalfZ _settings)r%namevaluer'r'r(_declare_number_constIsz+JavascriptCodePrinter._declare_number_constcCs ||SN) indent_code)r%linesr'r'r( _format_codeLsz"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdSr3)range).0ijcolsr'r( Qs zAJavascriptCodePrinter._traverse_matrix_indices..)shaper7)r%Zmatrowsr'r;r(_traverse_matrix_indicesOs 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++){)Zvarblestartend})append_printlabellowerupper)r%indicesZ open_linesZ close_linesZ loopstartr9r'r'r(_get_loop_opening_endingSs    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 parenthesizebaserFrOne)r%exprPRECr'r'r( _print_Pow`s    z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr*q)r%rQr*rUr'r'r(_print_Rationallsz%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr8|jsD|jrR|jrR|d|Sd|d|d|d|S)Ncsg|]}|qSr')rN)r8argrRr%r'r( ssz4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsrZis_nonnegativeZis_nonpositive)r%rQnumZdenZsnumZsdenr'rXr( _print_Modps  z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rFlhsrhsZrel_opr-)r%rQZlhs_codeZrhs_codeopr'r'r(_print_Relational|s  z'JavascriptCodePrinter._print_RelationalcCs`|j}tj}tj}tt|jD]"}||j||7}|||9}q d||j j ||fS)Nz%s[%s]) r>rZZerorPreversedr7ZrankrJrFrOrG)r%rQZdimselemoffsetr9r'r'r(_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jSr3)rFrGr%rQr'r'r( _print_Idxsz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er'rer'r'r( _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr'rer'r'r( _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr'rer'r'r(_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr'rer'r'r(_print_NegativeInfinitysz-JavascriptCodePrinter._print_NegativeInfinityc s"ddlm}|jdjdkr$tdg}||rt|jD]\}\}}|dkrf|d|n:|t |jdkr|dkr|dn|d ||}|||d q.z: ( %s )z:  )) Zsympy.codegen.astrkrZZcond ValueErrorhas enumeraterErFlenjoinrQ) r%rQrkr5r9rmrnZcode0Zecpairs last_liner'ror(_print_Piecewises(         z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]ZAtomT)strictrA)r-rNparentrr:r9r>rer'r'r(_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( )rDrqcSsg|]}|dqS)z )lstripr8liner'r'r(rYsz5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr')rTanymapendswithr) inc_tokenr'r(rYscs g|]}ttt|jqSr')rTrr startswithr) dec_tokenr'r(rYsr)r|rlz%s%s) isinstancestrr4 splitlinesrvrtrE) r%codeZ code_linestabZincreaseZdecreaseprettylevelnrr')rrr(r4s*     z!JavascriptCodePrinter.indent_codeN)__name__ __module__ __qualname____doc__Z printmethodlanguager!rr__annotations__r r+r,r/r2r6r@rKrSrVr\r`rdrfrgrhrirjrxr{r4r'r'r'r(r.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); )rZdoprint)rQZ assign_tor&r'r'r(jscodesircKstt|f|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rQr&r'r'r( print_jscodeJsr)N)r __future__rtypingrZ sympy.corerZsympy.core.numbersrZsympy.printing.codeprinterrZsympy.printing.precedencerrr"rrrr'r'r'r(s@     1 l