ó ¡¼™\c@ s@dZddlmZmZddlmZddlmZddlm Z m Z ddl m Z ddl mZmZidd 6d d 6d d 6dd6dd6dd6dd6dd6dd6dd6dd6dd6d d!6d"d#6d$d%6d&d'6d(d)6d*d+6d,d-6d.d/6d0d16Zd2e fd3„ƒYZd6d4„Zd5„Zd6S(7sÅ Javascript code printer The JavascriptCodePrinter converts single sympy expressions into single Javascript expressions, using the functions defined in the Javascript Math object where possible. iÿÿÿÿ(tprint_functiontdivision(t Assignment(tS(t string_typestrange(t CodePrinter(t precedencet PRECEDENCEsMath.abstAbss Math.acostacoss Math.acoshtacoshs Math.asintasins Math.asinhtasinhs Math.atantatans Math.atan2tatan2s Math.atanhtatanhs Math.ceiltceilingsMath.costcoss Math.coshtcoshsMath.exptexps Math.floortfloorsMath.logtlogsMath.maxtMaxsMath.mintMins Math.signtsignsMath.sintsins Math.sinhtsinhsMath.tanttans Math.tanhttanhtJavascriptCodePrintercB sÿeZdZdZdZidd6dd6dd6id6ed 6ed 6ed 6Zid „Z d „Z d„Z d„Z d„Z d„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS( sK"A Printer to convert python expressions to strings of javascript code t _javascriptt Javascripttordertautot full_precit precisiontuser_functionsthumantallow_unknown_functionstcontractcC sEtj||ƒttƒ|_|jdiƒ}|jj|ƒdS(NR%(Rt__init__tdicttknown_functionstgettupdate(tselftsettingst userfuncs((s4lib/python2.7/site-packages/sympy/printing/jscode.pyR)>scC s|dS(Ni((R.tp((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_rate_index_positionDscC sd|S(Ns%s;((R.t codestring((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_get_statementGscC s dj|ƒS(Ns// {0}(tformat(R.ttext((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _get_commentJscC s dj||j|jdƒƒS(Nsvar {0} = {1};R$(R5tevalft _settings(R.tnametvalue((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_declare_number_constMscC s |j|ƒS(N(t indent_code(R.tlines((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _format_codePsc s)|j\}‰‡fd†t|ƒDƒS(Nc3 s.|]$}tˆƒD]}||fVqqdS(N(R(t.0titj(tcols(s4lib/python2.7/site-packages/sympy/printing/jscode.pys Us(tshapeR(R.tmattrows((RCs4lib/python2.7/site-packages/sympy/printing/jscode.pyt_traverse_matrix_indicesSscC sˆg}g}d}xi|D]a}|j|i|j|jƒd6|j|jƒd6|j|jdƒd6ƒ|jdƒqW||fS(NsAfor (var %(varble)s=%(start)s; %(varble)s<%(end)s; %(varble)s++){tvarbletstartitendt}(tappendt_printtlabeltlowertupper(R.tindicest open_linest close_linest loopstartRA((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_get_loop_opening_endingWs  cC s¬t|ƒ}|jdkr2d|j|j|ƒS|jdkrUd|j|jƒS|jtdƒdkr‚d|j|jƒSd|j|jƒ|j|jƒfSdS( Niÿÿÿÿs1/%sgà?s Math.sqrt(%s)iis Math.cbrt(%s)sMath.pow(%s, %s)(RRt parenthesizetbaseRMR(R.texprtPREC((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _print_Powds cC s-t|jƒt|jƒ}}d||fS(Ns%d/%d(tintR1tq(R.RXR1R\((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_RationalpscC s„|j}tj}tj}x@tt|jƒƒD])}||j||7}|||9}q1Wd|j|j j ƒ|j|ƒfS(Ns%s[%s]( RDRtZerotOnetreversedRtrankRQRMRWRN(R.RXtdimstelemtoffsetRA((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_Indexedts   cC s|j|jƒS(N(RMRN(R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _print_Idx~scC sdS(NsMath.E((R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _print_Exp1scC sdS(NsMath.PI((R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt _print_Pi„scC sdS(NsNumber.POSITIVE_INFINITY((R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_Infinity‡scC sdS(NsNumber.NEGATIVE_INFINITY((R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_NegativeInfinityŠsc C sœ|jdjtkr%tdƒ‚ng}|jtƒrxÄt|jƒD]³\}\}}|dkr…|jd|j|ƒƒnO|t |jƒdkrº|tkrº|jdƒn|jd|j|ƒƒ|j|ƒ}|j|ƒ|jdƒqJWd j |ƒSg|jd D].\}}d |j|ƒ|j|ƒf^q}d |j|jdj ƒ}d j |ƒ|d j dt |ƒgƒSdS(Niÿÿÿÿs¼All 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.is if (%s) {iselse {selse if (%s) {RKs s((%s) ? ( %s ) s: ( %s )s: t t)( targstcondtTruet ValueErrorthasRt enumerateRLRMtlentjoinRX( R.RXR>RAtetctcode0tecpairst last_line((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_Piecewises$" %  ?cC sAdj|j|jtddtƒ|j|j|jjdƒS(Ns{0}[{1}]tAtomtstricti(R5RVtparentRRoRBRARD(R.RX((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt_print_MatrixElement­sc C sYt|tƒr4|j|jtƒƒ}dj|ƒSd}d }d}g|D]}|jd ƒ^qM}g|D]$}ttt |j |ƒƒƒ^qo}g|D]$}ttt |j |ƒƒƒ^q }g} d } x|t |ƒD]n\} }|dks|d kr| j |ƒqãn| || 8} | j d || |fƒ| || 7} qãW| S(s0Accepts a string of code or a list of code linests t{t(s{ s( RKRls is s%s%s(R€Rs{ s( (RKRl(t isinstanceRR=t splitlinesRoRttlstripR[tanytmaptendswitht startswithRrRL( R.tcodet code_linesttabt inc_tokent dec_tokentlinetincreasetdecreasetprettytleveltn((s4lib/python2.7/site-packages/sympy/printing/jscode.pyR=²s( "1. N(t__name__t __module__t__doc__t printmethodtlanguagetNoneRotFalset_default_settingsR)R2R4R7R<R?RGRURZR]ReRfRgRhRiRjRzR~R=(((s4lib/python2.7/site-packages/sympy/printing/jscode.pyR.s<               cK st|ƒj||ƒS(sÈConverts 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); (Rtdoprint(RXt assign_toR/((s4lib/python2.7/site-packages/sympy/printing/jscode.pytjscodeÏsicK stt||ƒdS(s€Prints the Javascript representation of the given expression. See jscode for the meaning of the optional arguments. N(tprintRž(RXR/((s4lib/python2.7/site-packages/sympy/printing/jscode.pyt print_jscode;sN(R–t __future__RRtsympy.codegen.astRt sympy.coreRtsympy.core.compatibilityRRtsympy.printing.codeprinterRtsympy.printing.precedenceRRR+RR™RžR (((s4lib/python2.7/site-packages/sympy/printing/jscode.pyts> ¡ l