g/dZddlmZddlmZddlmZddlmZddl 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/d0e Zd4d2Zd3Zd1S)5z 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)sinsinhtantanhceZdZUdZdZdZeejfididdZde d<ifd Z d Z d Z d Z d ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdS)JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code _javascript JavaScriptT) precisionuser_functionscontractzdict[str, Any]_default_settingsctj||tt|_|di}|j|dS)Nr%)r__init__dictknown_functionsgetupdate)selfsettings userfuncss Q/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/printing/jscode.pyr)zJavascriptCodePrinter.__init__:sVT8,,,#O44LL!1266  ##I.....c |dzS)N)r.ps r1_rate_index_positionz*JavascriptCodePrinter._rate_index_position@s s r2c d|zS)Nz%s;r5)r. codestrings r1_get_statementz$JavascriptCodePrinter._get_statementCs z!!r2c,d|S)Nz// {})format)r.texts r1 _get_commentz"JavascriptCodePrinter._get_commentFs~~d###r2cjd|||jdS)Nz var {} = {};r$)r<evalf _settings)r.namevalues r1_declare_number_constz+JavascriptCodePrinter._declare_number_constIs+$$T5;;t~k7R+S+STTTr2c,||SN) indent_code)r.liness r1 _format_codez"JavascriptCodePrinter._format_codeLs&&&r2cN|j\}fdt|DS)Nc3DK|]}tD]}||fV dSrF)range).0ijcolss r1 zAJavascriptCodePrinter._traverse_matrix_indices..Qs:AA1U4[[AAAAAAAAAAr2)shaperL)r.matrowsrPs @r1_traverse_matrix_indicesz.JavascriptCodePrinter._traverse_matrix_indicesOs.Y dAAAAd AAAAr2c g}g}d}|D]~}||||j||j||jdzdz|d||fS)NzAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){)varblestartend})append_printlabellowerupper)r.indices open_lines close_lines loopstartrNs r1_get_loop_opening_endingz.JavascriptCodePrinter._get_loop_opening_endingSs  W  $ $A   i++ag..QW--{{17Q;//+1+11 2 2 2   s # # # #;&&r2ct|}t|jdrd||j|zSt|jdrd||jzS|jt jdz krd||jzSd||jd||jd S) Nz1/%sg?z Math.sqrt(%s)z Math.cbrt(%s)z Math.pow(z, ))rrr parenthesizebaser]rOne)r.exprPRECs r1 _print_Powz JavascriptCodePrinter._print_Pow`s$ " % % 8T..ty$??@ @ $(C ( ( 8"T[[%;%;; ; Xq "T[[%;%;; ;!*.TY)?)?)?)?!%TX!6!6!6!68 8r2c`t|jt|j}}d||fzS)Nz%d/%d)intr6q)r.rmr6rrs r1_print_Rationalz%JavascriptCodePrinter._print_Rationalls*46{{CKK1!Qr2c|j\}}t|fd|jD\}}|jr|js|jr|jr|d|Sd|d|d|d|S)Nc<g|]}|Sr5)rj)rMargrnr.s r1 z4JavascriptCodePrinter._print_Mod..ss)HHHsd''T22HHHr2z % ((z) + z) % )argsris_nonnegativeis_nonpositive)r.rmnumdensnumsdenrns` @r1 _print_Modz JavascriptCodePrinter._print_Modps9S$HHHHHdiHHH d   &3#5 &   &# &%%t%% %7D77T77t77777r2c||j}||j}|j}d|||S)Nz{} {} {})r]lhsrhsrel_opr<)r.rmlhs_coderhs_codeops r1_print_Relationalz'JavascriptCodePrinter._print_Relational|sG;;tx((;;tx(( [  2x888r2c4|j}tj}tj}t t |jD] }||j||zz }|||z}!||j j d||dS)N[]) rRrZerorlreversedrLrankrar]rkr^)r.rmdimselemoffsetrNs r1_print_Indexedz$JavascriptCodePrinter._print_Indexedszv% **++  A DLOF* *D d1g FF;;ty7777T9J9J9J9JKKr2c6||jSrF)r]r^r.rms r1 _print_Idxz JavascriptCodePrinter._print_Idxs{{4:&&&r2cdS)NzMath.Er5rs r1 _print_Exp1z!JavascriptCodePrinter._print_Exp1sxr2cdS)NzMath.PIr5rs r1 _print_PizJavascriptCodePrinter._print_Pisyr2cdS)NzNumber.POSITIVE_INFINITYr5rs r1_print_Infinityz%JavascriptCodePrinter._print_Infinity))r2cdS)NzNumber.NEGATIVE_INFINITYr5rs r1_print_NegativeInfinityz-JavascriptCodePrinter._print_NegativeInfinityrr2cddlm}|jdjdkrt dg}||rt |jD]\}\}}|dkr,|d|znb|t|jdz kr|dkr|dn+|d |z|}|||d d |Sfd |jddD}d |jdj z} d || zd dt|zgzS)Nr) AssignmentrgTzAll 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) {rWzelse {zelse if (%s) {r[ ctg|]4\}}d|d|d5S)rxz) ? ( z ) )r])rMecr.s r1rwz:JavascriptCodePrinter._print_Piecewise..sP0001104{{1~~~~t{{1~~~~N000r2z: ( %s )z:  ri) sympy.codegen.astrrycond ValueErrorhas enumerater\r]lenjoinrm) r.rmrrHrNrrcode0ecpairs last_lines ` r1_print_Piecewisez&JavascriptCodePrinter._print_Piecewises000000 9R=  % %/00 0  88J   Q&ty11 " " 6Aq66LLt{{1~~!=>>>>#di..1,,,dLL****LL!1DKKNN!BCCC A U### S!!!!99U## # 0000 $ #2#000G$t{{49R=3E'F'FFI99W%% 1CHHc#g,,>N=O4P4PP Pr2cd||jtdd|j|j|jjdzzS)Nz{}[{}]AtomT)strictrW)r<rjparentr rOrNrRrs r1_print_MatrixElementz*JavascriptCodePrinter._print_MatrixElementsXt00 v t 1 - - FTVDK-a00 022 2r2c t|tr=||d}d|Sd}d d d|D} fd|D} fd|D}g}d }t |D]Q\}} | d vr|| |||z}|||z| |||z }R|S) z0Accepts a string of code or a list of code linesTz ){(z{ z( )r[ric8g|]}|dS)z )lstrip)rMlines r1rwz5JavascriptCodePrinter.indent_code..s$666U##666r2c ng|]1}ttt|j2Sr5)rqanymapendswith)rMr inc_tokens r1rwz5JavascriptCodePrinter.indent_code..s5OOOSS ::;;<<OOOr2c ng|]1}ttt|j2Sr5)rqrr startswith)rMr dec_tokens r1rwz5JavascriptCodePrinter.indent_code..sE(((S)<<==>>(((r2r)rr) isinstancestrrG splitlinesrrr\) r.code code_linestabincreasedecreaseprettylevelnrrrs @@r1rGz!JavascriptCodePrinter.indent_codes3 dC  '))$//$*?*?@@J77:&& &,  66666OOOOOOO((((!%(((  ! !GAtz!! d### Xa[ E MMCIItt4 5 5 5 Xa[ EE r2N)__name__ __module__ __qualname____doc__ printmethodlanguager*rr'__annotations__r)r7r:r>rDrIrUrerorsrrrrrrrrrrrGr5r2r1r r .sKH(,[-J))OO)) !#//// """$$$UUU'''BBB ' ' ' 8 8 8    8 8 8999 LLL'''******QQQB222 r2r Nc Ht|||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 doprint)rm assign_tor/s r1jscoders#R ! * * 2 24 C CCr2c :tt|fi|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rmr/s r1 print_jscoderJs(  & " " " "#####r2rF)r __future__rtypingr sympy.corersympy.core.numbersrsympy.printing.codeprinterrsympy.printing.precedencerr r+r rrr5r2r1rs#"""""++++++222222<<<<<<<<  : K \ K   \   K  \ \{ : K : \ : :  :!" K#$    +2mmmmmKmmm`iDiDiDiDX$$$$$r2