B ed/@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)ZAbsacosacoshasinasinhatanatan2atanhZceilingcoscoshexpfloorlogZMaxZMinsignsinsinhtantanhc@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)orderZ full_prec precisionuser_functionsZhumanZallow_unknown_functionsZcontractcCs2t||tt|_|di}|j|dS)Nr)r__init__dictknown_functionsgetupdate)selfsettingsZ userfuncsr'b/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/sympy/printing/jscode.pyr <s   zJavascriptCodePrinter.__init__cCs|dS)Nr')r%pr'r'r(_rate_index_positionBsz*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r')r%Z codestringr'r'r(_get_statementEsz$JavascriptCodePrinter._get_statementcCs d|S)Nz// {})format)r%textr'r'r( _get_commentHsz"JavascriptCodePrinter._get_commentcCsd|||jdS)Nz var {} = {};r)r-ZevalfZ _settings)r%namevaluer'r'r(_declare_number_constKsz+JavascriptCodePrinter._declare_number_constcCs ||S)N) indent_code)r%linesr'r'r( _format_codeNsz"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdS)N)range).0ij)colsr'r( SszAJavascriptCodePrinter._traverse_matrix_indices..)shaper6)r%matrowsr')r:r(_traverse_matrix_indicesQs 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)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 parenthesizebaserErOne)r%exprPRECr'r'r( _print_Powbs   z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr*q)r%rPr*rTr'r'r(_print_Rationalnsz%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr8|jsD|jrR|jrR|d|Sd|d|d|d|S)Ncsg|]}|qSr')rM)r7arg)rQr%r'r( usz4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsrZis_nonnegativeZis_nonpositive)r%rPnumZdenZsnumZsdenr')rQr%r( _print_Modrs   z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rElhsrhsZrel_opr-)r%rPZlhs_codeZrhs_codeopr'r'r(_print_Relational~s  z'JavascriptCodePrinter._print_RelationalcCsd|j}tj}tj}x4tt|jD]"}||j||7}|||9}q"Wd||j j ||fS)Nz%s[%s]) r<rZZerorOreversedr6ZrankrIrErNrF)r%rPZdimselemoffsetr8r'r'r(_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jS)N)rErF)r%rPr'r'r( _print_Idxsz JavascriptCodePrinter._print_IdxcCsdS)NzMath.Er')r%rPr'r'r( _print_Exp1sz!JavascriptCodePrinter._print_Exp1cCsdS)NzMath.PIr')r%rPr'r'r( _print_PiszJavascriptCodePrinter._print_PicCsdS)NzNumber.POSITIVE_INFINITYr')r%rPr'r'r(_print_Infinitysz%JavascriptCodePrinter._print_InfinitycCsdS)NzNumber.NEGATIVE_INFINITYr')r%rPr'r'r(_print_NegativeInfinitysz-JavascriptCodePrinter._print_NegativeInfinityc s&ddlm}|jdjdkr$tdg}||rxt|jD]\}\}}|dkrh|d|n:|t |jdkr|dkr|dn|d ||}|||d q>Wd |Sfd d |jddD}d|jdj } d || d dt |gSdS)Nr) AssignmentrKTzAll 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.z if (%s) {r@zelse {zelse if (%s) {rC cs(g|] \}}d||fqS)z((%s) ? ( %s ) )rE)r7ec)r%r'r(rWsz:JavascriptCodePrinter._print_Piecewise..z: ( %s )z:  )) Zsympy.codegen.astrhrXZcond ValueErrorhas enumeraterDrElenjoinrP) r%rPrhr4r8rjrkZcode0Zecpairs last_liner')r%r(_print_Piecewises&       z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]ZAtomT)strictr@)r-rMparentrr9r8r<)r%rPr'r'r(_print_MatrixElements  z*JavascriptCodePrinter._print_MatrixElementc st|tr$||d}d|Sd}dddd|D}fdd|D}fd d|D}g}d }xVt|D]J\}} | d kr|| qt|||8}|d ||| f|||7}qtW|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )rCrmcSsg|]}|dqS)z )lstrip)r7liner'r'r(rWsz5JavascriptCodePrinter.indent_code..cs g|]}ttt|jqSr')rSanymapendswith)r7r|) inc_tokenr'r(rWscs g|]}ttt|jqSr')rSr}r~ startswith)r7r|) dec_tokenr'r(rWsr)rxriz%s%s) isinstancestrr3 splitlinesrrrprD) r%codeZ code_linestabZincreaseZdecreaseprettylevelnr|r')rrr(r3s(     z!JavascriptCodePrinter.indent_code)__name__ __module__ __qualname____doc__Z printmethodlanguageZ_default_settingsr r+r,r/r2r5r?rJrRrUrZr^rbrcrdrerfrgrtrwr3r'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)rPZ 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)rPr&r'r'r( print_jscodeLsr)N)rtypingrrZtDictZ sympy.corerZsympy.printing.codeprinterrZsympy.printing.precedencerrr"rrrr'r'r'r(s:  5 l