o o h/@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z idd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.ZGd/d0d0e Zd6d2d3Zd4d5Zd1S)7z 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 PRECEDENCEAbszMath.absacosz Math.acosacoshz Math.acoshasinz Math.asinasinhz Math.asinhatanz Math.atanatan2z Math.atan2atanhz Math.atanhceilingz Math.ceilcoszMath.coscoshz Math.coshexpzMath.expfloorz Math.floorlogzMath.logMaxzMath.maxMinzMath.minsignz Math.signzMath.sinz Math.sinhzMath.tanz Math.tanh)sinsinhtantanhc@seZdZUdZdZdZeejfididdZde 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 _javascript JavaScriptT) precisionuser_functionscontractzdict[str, Any]_default_settingscCs2t||tt|_|di}|j|dS)Nr#)r__init__dictknown_functionsgetupdate)selfsettings userfuncsr.i/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/printing/jscode.pyr&:s   zJavascriptCodePrinter.__init__cCs|dS)Nr.)r+pr.r.r/_rate_index_position@z*JavascriptCodePrinter._rate_index_positioncCsd|S)Nz%s;r.)r+ codestringr.r.r/_get_statementCr3z$JavascriptCodePrinter._get_statementcCs d|S)Nz// {})format)r+textr.r.r/ _get_commentF z"JavascriptCodePrinter._get_commentcCsd|||jdS)Nz var {} = {};r")r6evalf _settings)r+namevaluer.r.r/_declare_number_constIsz+JavascriptCodePrinter._declare_number_constcCs ||SN) indent_code)r+linesr.r.r/ _format_codeLr9z"JavascriptCodePrinter._format_codecs |j\}fddt|DS)Nc3s&|]}tD]}||fVqqdSr?)range).0ijcolsr.r/ Qs$zAJavascriptCodePrinter._traverse_matrix_indices..)shaperC)r+matrowsr.rGr/_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++){)varblestartend})append_printlabellowerupper)r+indices open_lines close_lines loopstartrEr.r.r/_get_loop_opening_endingSs   z.JavascriptCodePrinter._get_loop_opening_endingcCs~t|}t|jdrd||j|St|jdr!d||jS|jtjdkr1d||jSd||j||jfS)Nz1/%sg?z Math.sqrt(%s)z Math.cbrt(%s)zMath.pow(%s, %s))rrr parenthesizebaserTrOne)r+exprPRECr.r.r/ _print_Pow`s    z JavascriptCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz%d/%d)intr1q)r+rbr1rfr.r.r/_print_Rationalls z%JavascriptCodePrinter._print_Rationalcsn|j\}}t|fdd|jD\}}|jr|js"|jr)|jr)|d|Sd|d|d|d|S)Ncsg|]}|qSr.)r_)rDargrcr+r.r/ ssz4JavascriptCodePrinter._print_Mod..z % z((z) + z) % )argsris_nonnegativeis_nonpositive)r+rbnumdensnumsdenr.rir/ _print_Modps  z JavascriptCodePrinter._print_ModcCs,||j}||j}|j}d|||S)Nz{} {} {})rTlhsrhsrel_opr6)r+rblhs_coderhs_codeopr.r.r/_print_Relational|s  z'JavascriptCodePrinter._print_RelationalcCs`|j}tj}tj}tt|jD]}||j||7}|||9}qd||j j ||fS)Nz%s[%s]) rJrZerorareversedrCrankrXrTr`rU)r+rbdimselemoffsetrEr.r.r/_print_Indexedsz$JavascriptCodePrinter._print_IndexedcCs ||jSr?)rTrUr+rbr.r.r/ _print_Idxs z JavascriptCodePrinter._print_IdxcCdS)NzMath.Er.rr.r.r/ _print_Exp1z!JavascriptCodePrinter._print_Exp1cCr)NzMath.PIr.rr.r.r/ _print_PirzJavascriptCodePrinter._print_PicCr)NzNumber.POSITIVE_INFINITYr.rr.r.r/_print_Infinityrz%JavascriptCodePrinter._print_InfinitycCr)NzNumber.NEGATIVE_INFINITYr.rr.r.r/_print_NegativeInfinityrz-JavascriptCodePrinter._print_NegativeInfinityc sddlm}|jdjdkrtdg}||ret|jD]A\}\}}|dkr3|d|n|t |jdkrF|dkrF|dn |d ||}|||d qd |Sfd d |jddD}d|jdj } d || d dt |gS)Nr) Assignmentr]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.z if (%s) {rNzelse {zelse if (%s) {rR cs(g|]\}}d||fqS)z((%s) ? ( %s ) )rT)rDecr+r.r/rjsz:JavascriptCodePrinter._print_Piecewise..z: ( %s )z:  )) sympy.codegen.astrrkcond ValueErrorhas enumeraterSrTlenjoinrb) r+rbrrArErrcode0ecpairs last_liner.rr/_print_Piecewises(         "z&JavascriptCodePrinter._print_PiecewisecCs2d|j|jtddd|j|j|jjdS)Nz{}[{}]AtomT)strictrN)r6r_parentrrFrErJrr.r.r/_print_MatrixElements z*JavascriptCodePrinter._print_MatrixElementc st|tr||d}d|Sd}dddd|D}fdd|D}fd d|D}g}d }t|D]%\}} | d vrG|| q9|||8}|d ||| f|||7}q9|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )rRrcSsg|]}|dqS)z )lstriprDliner.r.r/rjsz5JavascriptCodePrinter.indent_code..c g|] }ttt|jqSr.)reanymapendswithr) inc_tokenr.r/rjs crr.)rerr startswithr) dec_tokenr.r/rjsr)rrz%s%s) isinstancestrr@ splitlinesrrrS) r+code code_linestabincreasedecreaseprettylevelnrr.)rrr/r@s*     z!JavascriptCodePrinter.indent_codeN)__name__ __module__ __qualname____doc__ printmethodlanguager'rr%__annotations__r&r2r5r8r>rBrMr\rdrgrrryrrrrrrrrr@r.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); )rdoprint)rb 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)rbr,r.r.r/ print_jscodeJsrr?)r __future__rtypingr sympy.corersympy.core.numbersrsympy.printing.codeprinterrsympy.printing.precedencerrr(rrrr.r.r.r/sf          1 l