B }[R,@sdZddlmZmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZdd 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. )print_functiondivision)S) Assignment) CodePrinter) precedence PRECEDENCE) string_typesrangezMath.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)ZAbsZacosZacoshZasinZasinhZatanZatan2ZatanhZceilingZcosZcoshexpZfloorlogZMaxZMinZsignZsinZsinhZtanZtanhc@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/ZdS)0JavascriptCodePrinterzK"A Printer to convert python expressions to strings of javascript code Z _javascriptZ JavascriptNautoTF)orderZ full_prec precisionuser_functionsZhumanZallow_unknown_functionsZcontractcCs2t||tt|_|di}|j|dS)Nr)r__init__dictknown_functionsgetupdate)selfsettingsZ userfuncsr4lib/python3.7/site-packages/sympy/printing/jscode.pyr>s   zJavascriptCodePrinter.__init__cCs|dS)Nr)rprrr_rate_index_positionDsz*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r)rZ codestringrrr_get_statementGsz$JavascriptCodePrinter._get_statementcCs d|S)Nz// {0})format)rtextrrr _get_commentJsz"JavascriptCodePrinter._get_commentcCsd|||jdS)Nzvar {0} = {1};r)r ZevalfZ _settings)rnamevaluerrr_declare_number_constMsz+JavascriptCodePrinter._declare_number_constcCs ||S)N) indent_code)rlinesrrr _format_codePsz"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdS)N)r ).0ij)colsrr UszAJavascriptCodePrinter._traverse_matrix_indices..)shaper )rZmatZrowsr)r,r_traverse_matrix_indicesSs z.JavascriptCodePrinter._traverse_matrix_indicesc Cs^g}g}d}xH|D]@}||||j||j||jdd|dqW||fS)NzAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){)Zvarblestartend})append_printlabellowerupper)rindicesZ open_linesZ close_linesZ loopstartr*rrr_get_loop_opening_endingWs   z.JavascriptCodePrinter._get_loop_opening_endingcCst|}|jdkr$d||j|S|jdkr>d||jS|jtddkr`d||jSd||j||jfSdS) Nz1/%sg?z Math.sqrt(%s)r0z Math.cbrt(%s)zMath.pow(%s, %s))rr parenthesizebaser5r)rexprZPRECrrr _print_Powds   z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intrq)rr?rrBrrr_print_Rationalpsz%JavascriptCodePrinter._print_RationalcCsd|j}tj}tj}x4tt|jD]"}||j||7}|||9}q"Wd||j j ||fS)Nz%s[%s]) r.rZZeroZOnereversedr Zrankr9r5r>r6)rr?Zdimselemoffsetr*rrr_print_Indexedtsz$JavascriptCodePrinter._print_IndexedcCs ||jS)N)r5r6)rr?rrr _print_Idx~sz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er)rr?rrr _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr)rr?rrr _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr)rr?rrr_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr)rr?rrr_print_NegativeInfinitysz-JavascriptCodePrinter._print_NegativeInfinityc s|jdjdkrtdg}|trxt|jD]\}\}}|dkr\|d|n:|t|jdkr|dkr|dn|d||}|||d q2Wd |Sfd d |jddD}d |jdj }d ||d dt|gSdS)Nr;TzAll Piecewise expressions must contain an (expr, True) statement to be used as a default condition. Without one, the generated expression may not evaluate to anything under some condition.rz if (%s) {r0zelse {zelse if (%s) {r3 cs(g|] \}}d||fqS)z((%s) ? ( %s ) )r5)r)ec)rrr sz:JavascriptCodePrinter._print_Piecewise..z: ( %s )z:  )) argsZcond ValueErrorZhasr enumerater4r5lenjoinr?) rr?r'r*rNrOZcode0Zecpairs last_liner)rr_print_Piecewises$      z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{0}[{1}]ZAtomT)strictr0)r r=parentrr+r*r.)rr?rrr_print_MatrixElements  z*JavascriptCodePrinter._print_MatrixElementc st|tr$||d}d|Sd}dddd|D}fdd|D}fd d|D}g}d }x^t|D]R\}} | dks| d kr|| qt|||8}|d ||| f|||7}qtW|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )r3rRcSsg|]}|dqS)z )lstrip)r)linerrrrPsz5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr)rAanymapendswith)r)ra) inc_tokenrrrPscs g|]}ttt|jqSr)rArbrc startswith)r)ra) dec_tokenrrrPsrrMz%s%s) isinstancer r& splitlinesrWrUr4) rcodeZ code_linesZtabZincreaseZdecreaseZprettylevelnrar)rgrerr&s(     z!JavascriptCodePrinter.indent_code)__name__ __module__ __qualname____doc__Z printmethodZlanguageZ_default_settingsrrrr"r%r(r/r:r@rCrGrHrIrJrKrLrYr\r&rrrrr .s:     r NcKst|||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); )r Zdoprint)r?Z assign_torrrrjscodesirqcKstt|f|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printrq)r?rrrr print_jscode;srs)N)rpZ __future__rrZ sympy.corerZsympy.codegen.astrZsympy.printing.codeprinterrZsympy.printing.precedencerrZsympy.core.compatibilityr r rr rqrsrrrrs>   " l