Ed/dZddlmZmZddlmZddlmZddl m Z m Z idddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,Z Gd-d.eZ d2d0Zd1Zd/S)3z 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 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)sinsinhtantanhceZdZdZdZdZdddidddd Zifd Zd Zd Z d Z dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZdZdS)JavascriptCodePrinterzK"A Printer to convert Python expressions to strings of JavaScript code _javascript JavaScriptNautoTF)order full_prec precisionuser_functionshumanallow_unknown_functionscontractctj||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)r1ps r4_rate_index_positionz*JavascriptCodePrinter._rate_index_positionBs s r5c d|zS)Nz%s;r8)r1 codestrings r4_get_statementz$JavascriptCodePrinter._get_statementEs z!!r5c,d|S)Nz// {})format)r1texts r4 _get_commentz"JavascriptCodePrinter._get_commentHs~~d###r5cjd|||jdS)Nz var {} = {};r&)r?evalf _settings)r1namevalues r4_declare_number_constz+JavascriptCodePrinter._declare_number_constKs+$$T5;;t~k7R+S+STTTr5c,||SN) indent_code)r1liness r4 _format_codez"JavascriptCodePrinter._format_codeNs&&&r5cN|j\}fdt|DS)Nc3DK|]}tD]}||fV dSrI)range).0ijcolss r4 zAJavascriptCodePrinter._traverse_matrix_indices..Ss:AA1U4[[AAAAAAAAAAr5)shaperO)r1matrowsrSs @r4_traverse_matrix_indicesz.JavascriptCodePrinter._traverse_matrix_indicesQs.Y dAAAAd AAAAr5c g}g}d}|D]~}||||j||j||jdzdz|d||fS)NzAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){)varblestartend})append_printlabellowerupper)r1indices open_lines close_lines loopstartrQs r4_get_loop_opening_endingz.JavascriptCodePrinter._get_loop_opening_endingUs  W  $ $A   i++ag..QW--{{17Q;//+1+11 2 2 2   s # # # #;&&r5ct|}|jdkrd||j|zS|jdkrd||jzS|jt jdz krd||jzSd||jd||jd S) Nz1/%sg?z Math.sqrt(%s)z Math.cbrt(%s)z Math.pow(z, ))rr parenthesizebaser`rOne)r1exprPRECs r4 _print_Powz JavascriptCodePrinter._print_Powbs$ 8r> 8T..ty$??@ @ X_ 8"T[[%;%;; ; Xq  8"T[[%;%;; ; 8*.TY)?)?)?)?!%TX!6!6!6!68 8r5c`t|jt|j}}d||fzS)Nz%d/%d)intr9q)r1rpr9rus r4_print_Rationalz%JavascriptCodePrinter._print_Rationalns*46{{CKK1!Qr5c|j\}}t|fd|jD\}}|jr|js|jr|jr|d|Sd|d|d|d|S)Nc<g|]}|Sr8)rm)rPargrqr1s r4 z4JavascriptCodePrinter._print_Mod..us)HHHsd''T22HHHr5z % ((z) + z) % )argsris_nonnegativeis_nonpositive)r1rpnumdensnumsdenrqs` @r4 _print_Modz JavascriptCodePrinter._print_Modrs9S$HHHHHdiHHH d   &3#5 &   &# &%%t%% %7D77T77t77777r5c||j}||j}|j}d|||S)Nz{} {} {})r`lhsrhsrel_opr?)r1rplhs_coderhs_codeops r4_print_Relationalz'JavascriptCodePrinter._print_Relational~sG;;tx((;;tx(( [  2x888r5c4|j}tj}tj}t t |jD] }||j||zz }|||z}!||j j d||dS)N[]) rUrZerororeversedrOrankrdr`rnra)r1rpdimselemoffsetrQs r4_print_Indexedz$JavascriptCodePrinter._print_Indexedszv% **++  A DLOF* *D d1g FF;;ty7777T9J9J9J9JKKr5c6||jSrI)r`rar1rps r4 _print_Idxz JavascriptCodePrinter._print_Idxs{{4:&&&r5cdS)NzMath.Er8rs r4 _print_Exp1z!JavascriptCodePrinter._print_Exp1sxr5cdS)NzMath.PIr8rs r4 _print_PizJavascriptCodePrinter._print_Pisyr5cdS)NzNumber.POSITIVE_INFINITYr8rs r4_print_Infinityz%JavascriptCodePrinter._print_Infinity))r5cdS)NzNumber.NEGATIVE_INFINITYr8rs r4_print_NegativeInfinityz-JavascriptCodePrinter._print_NegativeInfinityrr5cddlm}|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) AssignmentrjTzAll 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) {rZzelse {zelse if (%s) {r^ ctg|]4\}}d|d|d5S)r{z) ? ( z ) )r`)rPecr1s r4rzz:JavascriptCodePrinter._print_Piecewise..sP0001104{{1~~~~t{{1~~~~N000r5z: ( %s )z:  rl) sympy.codegen.astrr|cond ValueErrorhas enumerater_r`lenjoinrp) r1rprrKrQrrcode0ecpairs last_lines ` r4_print_Piecewisez&JavascriptCodePrinter._print_Piecewises000000 9R=  % 0/00 0  88J   Q&ty11 " " 6Aq6DLLt{{1~~!=>>>>#di..1,,DdDLL****LL!1DKKNN!BCCC A U### S!!!!99U## # 0000 $ #2#000G$t{{49R=3E'F'FFI99W%% 1CHHc#g,,>N=O4P4PP Pr5cd||jtdd|j|j|jjdzzS)Nz{}[{}]AtomT)strictrZ)r?rmparentrrRrQrUrs r4_print_MatrixElementz*JavascriptCodePrinter._print_MatrixElementsXt00 v t 1 - - FTVDK-a00 022 2r5c 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^rlc8g|]}|dS)z )lstrip)rPlines r4rzz5JavascriptCodePrinter.indent_code..s$666U##666r5c ng|]1}ttt|j2Sr8)rtanymapendswith)rPr inc_tokens r4rzz5JavascriptCodePrinter.indent_code..s5OOOSS ::;;<<OOOr5c ng|]1}ttt|j2Sr8)rtrr startswith)rPr dec_tokens r4rzz5JavascriptCodePrinter.indent_code..sE(((S)<<==>>(((r5r)rr) isinstancestrrJ splitlinesrrr_) r1code code_linestabincreasedecreaseprettylevelnrrrs @@r4rJz!JavascriptCodePrinter.indent_codes5 dC  '))$//$*?*?@@J77:&& &,  66666OOOOOOO((((!%(((  ! !GAtz!  d### Xa[ E MMCIItt4 5 5 5 Xa[ EE r5)__name__ __module__ __qualname____doc__ printmethodlanguage_default_settingsr,r:r=rArGrLrXrhrrrvrrrrrrrrrrrJr8r5r4rr,sKH#(!#//// """$$$UUU'''BBB ' ' ' 8 8 8    8 8 8999 LLL'''******QQQB222 r5rNc 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); )rdoprint)rp assign_tor2s r4jscoders#R ! * * 2 24 C CCr5c :tt|fi|dS)zPrints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N)printr)rpr2s r4 print_jscoderLs(  & " " " "#####r5rI)rtypingrrtDict sympy.corersympy.printing.codeprinterrsympy.printing.precedencerrr.rrrr8r5r4rs&%%%%%%%222222<<<<<<<<  : K \ K   \   K  \ \{ : K : \ : :  :!" K#$    +2qqqqqKqqqhiDiDiDiDX$$$$$r5