a SzAJavascriptCodePrinter._traverse_matrix_indices..)shaper6)r$matrowsr&r:r'_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++){)Zvarblestartend})append_printlabellowerupper)r$indicesZ open_linesZ close_linesZ loopstartr8r&r&r'_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 parenthesizebaserGrOne)r$exprPRECr&r&r' _print_Powbs    z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr)q)r$rRr)rVr&r&r'_print_Rationalnsz%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr8|jsD|jrR|jrR|d|Sd|d|d|d|S)Ncsg|]}|qSr&)rO)r7argrSr$r&r' ur=z4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsrZis_nonnegativeZis_nonpositive)r$rRnumZdenZsnumZsdenr&rYr' _print_Modrs  z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rGlhsrhsZrel_opr,)r$rRZlhs_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>rZZerorQreversedr6ZrankrKrGrPrH)r$rRdimselemoffsetr8r&r&r'_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jSr2)rGrHr$rRr&r&r' _print_Idxsz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er&rgr&r&r' _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr&rgr&r&r' _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr&rgr&r&r'_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr&rgr&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.astrmr[Zcond ValueErrorZhas enumeraterFrGlenjoinrR) r$rRrmr4r8rorpZcode0Zecpairs last_liner&rqr'_print_Piecewises(         z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]ZAtomT)strictrB)r,rOparentrr9r8r>rgr&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 vr|| qr|||8}|d ||| f|||7}qr|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )rErscSsg|]}|dqS)z )lstripr7liner&r&r'rZr=z5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr&)rUanymapendswithr) inc_tokenr&r'rZr=cs g|]}ttt|jqSr&)rUrr startswithr) dec_tokenr&r'rZsr)r}rnz%s%s) isinstancestrr3 splitlinesrwrurF) r$codeZ code_linestabZincreaseZdecreaseprettylevelnrr&)rrr'r3s*     z!JavascriptCodePrinter.indent_code)__name__ __module__ __qualname____doc__Z printmethodlanguageZ_default_settingsrr*r+r.r1r5rArLrTrWr]rarfrhrirjrkrlryr|r3r&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)rRZ assign_tor%r&r&r'jscodesircKstt|fi|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rRr%r&r&r' print_jscodeLsr)N)rtypingrrZtDictZ sympy.corerZsympy.printing.codeprinterrZsympy.printing.precedencerrr!rrrr&r&r&r's<   5 l