Riec/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)sinsinhtantanhceZdZUdZdZdZdddidddd Zd ed <ifd Zd Z dZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZd ZdS)!JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code _javascript JavaScriptNautoTF)order full_prec precisionuser_functionshumanallow_unknown_functionscontractzdict[str, Any]_default_settingsctj||tt|_|di}|j|dS)Nr()r__init__dictknown_functionsgetupdate)selfsettings userfuncss 5lib/python3.11/site-packages/sympy/printing/jscode.pyr.zJavascriptCodePrinter.__init__>sVT8,,,#O44LL!1266  ##I.....c |dzS)N)r3ps r6_rate_index_positionz*JavascriptCodePrinter._rate_index_positionDs s r7c d|zS)Nz%s;r:)r3 codestrings r6_get_statementz$JavascriptCodePrinter._get_statementGs z!!r7c,d|S)Nz// {})format)r3texts r6 _get_commentz"JavascriptCodePrinter._get_commentJs~~d###r7cjd|||jdS)Nz var {} = {};r')rAevalf _settings)r3namevalues r6_declare_number_constz+JavascriptCodePrinter._declare_number_constMs+$$T5;;t~k7R+S+STTTr7c,||SN) indent_code)r3liness r6 _format_codez"JavascriptCodePrinter._format_codePs&&&r7cN|j\}fdt|DS)Nc3DK|]}tD]}||fV dSrK)range).0ijcolss r6 zAJavascriptCodePrinter._traverse_matrix_indices..Us:AA1U4[[AAAAAAAAAAr7)shaperQ)r3matrowsrUs @r6_traverse_matrix_indicesz.JavascriptCodePrinter._traverse_matrix_indicesSs.Y dAAAAd AAAAr7c g}g}d}|D]~}||||j||j||jdzdz|d||fS)NzAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){)varblestartend})append_printlabellowerupper)r3indices open_lines close_lines loopstartrSs r6_get_loop_opening_endingz.JavascriptCodePrinter._get_loop_opening_endingWs  W  $ $A   i++ag..QW--{{17Q;//+1+11 2 2 2   s # # # #;&&r7ct|}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 parenthesizebaserbrOne)r3exprPRECs r6 _print_Powz JavascriptCodePrinter._print_Powds$ " % % 8T..ty$??@ @ $(C ( ( 8"T[[%;%;; ; Xq "T[[%;%;; ;!*.TY)?)?)?)?!%TX!6!6!6!68 8r7c`t|jt|j}}d||fzS)Nz%d/%d)intr;q)r3rrr;rws r6_print_Rationalz%JavascriptCodePrinter._print_Rationalps*46{{CKK1!Qr7c|j\}}t|fd|jD\}}|jr|js|jr|jr|d|Sd|d|d|d|S)Nc<g|]}|Sr:)ro)rRargrsr3s r6 z4JavascriptCodePrinter._print_Mod..ws)HHHsd''T22HHHr7z % ((z) + z) % )argsris_nonnegativeis_nonpositive)r3rrnumdensnumsdenrss` @r6 _print_Modz JavascriptCodePrinter._print_Modts9S$HHHHHdiHHH d   &3#5 &   &# &%%t%% %7D77T77t77777r7c||j}||j}|j}d|||S)Nz{} {} {})rblhsrhsrel_oprA)r3rrlhs_coderhs_codeops r6_print_Relationalz'JavascriptCodePrinter._print_RelationalsG;;tx((;;tx(( [  2x888r7c4|j}tj}tj}t t |jD] }||j||zz }|||z}!||j j d||dS)N[]) rWrZerorqreversedrQrankrfrbrprc)r3rrdimselemoffsetrSs r6_print_Indexedz$JavascriptCodePrinter._print_Indexedszv% **++  A DLOF* *D d1g FF;;ty7777T9J9J9J9JKKr7c6||jSrK)rbrcr3rrs r6 _print_Idxz JavascriptCodePrinter._print_Idxs{{4:&&&r7cdS)NzMath.Er:rs r6 _print_Exp1z!JavascriptCodePrinter._print_Exp1sxr7cdS)NzMath.PIr:rs r6 _print_PizJavascriptCodePrinter._print_Pisyr7cdS)NzNumber.POSITIVE_INFINITYr:rs r6_print_Infinityz%JavascriptCodePrinter._print_Infinity))r7cdS)NzNumber.NEGATIVE_INFINITYr:rs r6_print_NegativeInfinityz-JavascriptCodePrinter._print_NegativeInfinityrr7cddlm}|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) AssignmentrlTzAll 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) {r` ctg|]4\}}d|d|d5S)r}z) ? ( z ) )rb)rRecr3s r6r|z:JavascriptCodePrinter._print_Piecewise..sP0001104{{1~~~~t{{1~~~~N000r7z: ( %s )z:  rn) sympy.codegen.astrr~cond ValueErrorhas enumeraterarblenjoinrr) r3rrrrMrSrrcode0ecpairs last_lines ` r6_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 Pr7cd||jtdd|j|j|jjdzzS)Nz{}[{}]AtomT)strictr\)rAroparentr rTrSrWrs r6_print_MatrixElementz*JavascriptCodePrinter._print_MatrixElementsXt00 v t 1 - - FTVDK-a00 022 2r7c 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`rnc8g|]}|dS)z )lstrip)rRlines r6r|z5JavascriptCodePrinter.indent_code..s$666U##666r7c ng|]1}ttt|j2Sr:)rvanymapendswith)rRr inc_tokens r6r|z5JavascriptCodePrinter.indent_code..s5OOOSS ::;;<<OOOr7c ng|]1}ttt|j2Sr:)rvrr startswith)rRr dec_tokens r6r|z5JavascriptCodePrinter.indent_code..sE(((S)<<==>>(((r7r)rr) isinstancestrrL splitlinesrrra) r3code code_linestabincreasedecreaseprettylevelnrrrs @@r6rLz!JavascriptCodePrinter.indent_codes3 dC  '))$//$*?*?@@J77:&& &,  66666OOOOOOO((((!%(((  ! !GAtz!! d### Xa[ E MMCIItt4 5 5 5 Xa[ EE r7)__name__ __module__ __qualname____doc__ printmethodlanguager,__annotations__r.r<r?rCrIrNrZrjrtrxrrrrrrrrrrrLr:r7r6r r .sKH#())!#//// """$$$UUU'''BBB ' ' ' 8 8 8    8 8 8999 LLL'''******QQQB222 r7r 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)rr assign_tor4s r6jscoders#R ! * * 2 24 C CCr7c :tt|fi|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rrr4s r6 print_jscoderNs(  & " " " "#####r7rK)r __future__rtypingr sympy.corersympy.core.numbersrsympy.printing.codeprinterrsympy.printing.precedencerr r0r rrr:r7r6rs#"""""++++++222222<<<<<<<<  : K \ K   \   K  \ \{ : K : \ : :  :!" K#$    +2qqqqqKqqqhiDiDiDiDX$$$$$r7