~9\c4@ sdZddlmZmZddlmZmZmZmZm Z ddl m Z m Z ddl mZmZddlmZi*dd6d d6d d6d d6d d6d d 6dd6dd6dd6dd6dd6dd6dd6dd6dd6dddfdddfdd dfd!d"dfd#d$d%fd&d'd(fd)d*d(fgd+6d,d-dfgd-6d.d/6d/d6d0d6d1d6d2d6d3d6d4d56d6d76d8d6d9d96d:d:6d;d;6d<d<6d=d=6d>d>6d?d?6d@d6dAd6dBd6dCdC6dDdD6dEdE6dFdF6dGdG6dHdH6ZdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|g4Zd}efd~YZedZdZdS(s Rust code printer The `RustCodePrinter` converts SymPy expressions into Rust expressions. A complete code generator, which uses `rust_code` extensively, can be found in `sympy.utilities.codegen`. The `codegen` module can be used to generate complete source code files. i(tprint_functiontdivision(tStnumberstRationaltFloattLambda(t string_typestrange(t CodePrintert Assignment(t precedencetis_nantt is_infinitet is_finitet is_normaltclassifytfloortceiltceilingtroundttrunctfracttabstAbstsignumtsigntis_sign_positivetis_sign_negativetmul_addcC s|tj kS(N(RtOne(tbasetexp((s2lib/python2.7/site-packages/sympy/printing/rust.pytIR trecipicC s |tjkS(N(RtHalf(R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"JR tsqrtcC s|tj kS(N(RR$(R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"KR s sqrt().recipcC s|tddkS(Nii(R(R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"LR tcbrtcC s|tjdkS(Ni(RR(R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"MR texp2icC s|jS(N(t is_integer(R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"NR tpowiicC s|j S(N(R((R R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"OR tpowftPowcC stS(N(tTrue(R!((s2lib/python2.7/site-packages/sympy/printing/rust.pyR"PR R!tlntlogtlog2tlog10t to_degreest to_radianstmaxtMaxtmintMinthypottsintcosttantasintacostatantatan2tsin_costexp_m1tln_1ptsinhtcoshttanhtasinhtacoshtatanhtabstracttalignoftastbecometboxtbreaktconsttcontinuetcratetdotelsetenumtexterntfalsetfinaltfntfortiftimpltintlettlooptmacrotmatchtmodtmovetmuttoffsetoftoverridetprivtproctpubtpuretreftreturntSelftselftsizeoftstatictstructtsuperttraitttruettypettypeoftunsafetunsizedtusetvirtualtwheretwhiletyieldtRustCodePrintercB seZdZdZdZi d1d6dd6dd6id6ed 6ed 6ed 6ed 6d d6ed6Z idZ dZ dZ dZ dZdZdZdZdZdZdZedZedZdZdZdZd Zed!Zed"Zed#Zed$Zed%Zed&Z ed'Z!ed(Z"d)Z#d*Z$d+Z%d,Z&d-Z'e&Z%Z'Z(Z)Z*Z+d.Z,d/Z-d0Z.RS(2s?A printer to convert python expressions to strings of Rust codet _rust_codetRusttordertautot full_precit precisiontuser_functionsthumantcontractt dereferenceterror_on_reservedt_treserved_word_suffixtinlinecC sotj||tt|_|jdi}|jj|t|jdg|_tt|_dS(NRR( R t__init__tdicttknown_functionstgettupdatetsett _dereferencetreserved_words(Rltsettingst userfuncs((s2lib/python2.7/site-packages/sympy/printing/rust.pyRs cC s|dS(Ni((Rltp((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_rate_index_positionscC sd|S(Ns%s;((Rlt codestring((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_get_statementscC sd|S(Ns// %s((Rlttext((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _get_commentscC sd||fS(Nsconst %s: f64 = %s;((Rltnametvalue((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_declare_number_constscC s |j|S(N(t indent_code(Rltlines((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _format_codesc s)|j\}fdt|DS(Nc3 s.|]$}tD]}||fVqqdS(N(R(t.0titj(tcols(s2lib/python2.7/site-packages/sympy/printing/rust.pys s(tshapeR(Rltmattrows((Rs2lib/python2.7/site-packages/sympy/printing/rust.pyt_traverse_matrix_indicesscC sg}g}d}xf|D]^}|j|i|j|d6|j|jd6|j|jdd6|jdqW||fS(Ns#for %(var)s in %(start)s..%(end)s {tvartstartitendt}(tappendt_printtlowertupper(Rltindicest open_linest close_linest loopstartR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_get_loop_opening_endings  cC sWt|jdkr*d|j|dS|jrF|j|dtS|j|SdS(Nit(t)t_type(tlentargsRt is_numberR,(Rltexpr((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_caller_vars  cC s|jj|jkr|j|jj}d}d}t|trL|}n-x*|D]"\}}}||jrSPqSqSW|dk r|dkrdi|j|jdd6|d6t|jdkr|j |jddndd6}n|d kr(d i|j|jdd6|d6}n^|d kr_d i|j|jdd6|d6}n'd i|d 6|j |jdd6}|SnGt |drt|j t r|j |j |jS|j|SdS(s< basic function for printing `Function` Function Style : 1. args[0].func(args[1:]), method with arguments 2. args[0].func(), method without arguments 3. args[1].func(), method without arguments (e.g. (e, x) => x.exp()) 4. func(args), function with arguments is%(var)s.%(method)s(%(args)s)iRtmethods, R Ris%(var)s.%(method)s()is%(func)s(%(args)s)tfunct_imp_N(Rt__name__RtNonet isinstanceRRRRt stringifythasattrRRRt_print_not_supported(RlRt cond_funcRtstyletcondtret((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_Functions<    <  !cC sT|jjrG|jj rGt|t|j|j}|j|S|j|S(N(R R(R!RsRRR(RlR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_PowRs! cC s.tt|j|}|r&|dS|SdS(Nt_f64(RpR|t _print_Float(RlRRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyRXscC s.tt|j|}|r&|dS|SdS(Nt_i32(RpR|t_print_Integer(RlRRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyR_scC s-t|jt|j}}d||fS(Ns %d_f64/%d.0(tintRtq(RlRRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_RationalfscC s|j}tj}tj}x@tt|jD])}||j||7}|||9}q1Wd|j|j j |j|fS(Ns%s[%s]( RRtZeroRtreversedRtrankRRR tlabel(RlRtdimstelemtoffsetR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_Indexedjs   cC s |jjS(N(RR(RlR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_IdxtscC s|jS(N(R(RlR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_DummywscC sdS(NtE((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_Exp1zscC sdS(NtPI((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_Pi}scC sdS(NtINFINITY((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_InfinityscC sdS(Nt NEG_INFINITY((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_NegativeInfinityscC sdS(NRr((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_BooleanTruescC sdS(NRU((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_BooleanFalsescC st|jS(N(tstrR(RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_boolscC sdS(NtNAN((RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt _print_NaNscC s#|jdjtkr%tdng}xt|jD]\}\}}|dkrv|jd|j|nU|t|jdkr|tkr|dcd7ssLFull Matrix Support in Rust need Crates (https://crates.io/keywords/matrix).(RRR(RltA((Rls2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_MatrixBases!cC s)d|j|j|j|jjdfS(Ns%s[%s]i(RRRR(RlR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_MatrixElements cC s7tt|j|}||jkr/d|S|SdS(Ns(*%s)(RpR|t _print_SymbolR(RlRR((s2lib/python2.7/site-packages/sympy/printing/rust.pyRscC sddlm}|j}|j}|jdr]|j|sM|j|r]|j||S|j|}|j|}|jd||fSdS(Ni(t IndexedBaseRs%s = %s( tsympy.tensor.indexedRtlhstrhsRthast_doprint_loopsRR(RlRRRRtlhs_codetrhs_code((s2lib/python2.7/site-packages/sympy/printing/rust.pyt_print_Assignments  c C sYt|tr4|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 |qn| || 8} | j d || |f| || 7} qW| S(s0Accepts a string of code or a list of code linesR s t{Rs{ s( RRs is s%s%s(R Rs{ s( (RR(RRRt splitlinesR,RtlstripRtanytmaptendswitht startswithRR( Rltcodet code_linesttabt inc_tokent dec_tokentlinetincreasetdecreasetprettytleveltn((s2lib/python2.7/site-packages/sympy/printing/rust.pyRs( "1. N(/Rt __module__t__doc__t printmethodtlanguageRR,RtFalset_default_settingsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRt_print_DenseMatrixt_print_MutableDenseMatrixt_print_ImmutableMatrixt_print_ImmutableDenseMatrixRRR(((s2lib/python2.7/site-packages/sympy/printing/rust.pyR|s`          3                    cK st|j||S(sConverts an expr to a string of Rust 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 the keys are string representations of either ``FunctionClass`` or ``UndefinedFunction`` instances and the values are their desired C string representations. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, cfunction_string)]. See below for examples. dereference : iterable, optional An iterable of symbols that should be dereferenced in the printed code expression. These would be values passed by address to the function. For example, if ``dereference=[a]``, the resulting code would print ``(*a)`` instead of ``a``. 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 rust_code, symbols, Rational, sin, ceiling, Abs, Function >>> x, tau = symbols("x, tau") >>> rust_code((2*tau)**Rational(7, 2)) '8*1.4142135623731*tau.powf(7_f64/2.0)' >>> rust_code(sin(x), assign_to="s") 's = x.sin();' Simple 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, cfunction_string)]. >>> custom_functions = { ... "ceiling": "CEIL", ... "Abs": [(lambda x: not x.is_integer, "fabs", 4), ... (lambda x: x.is_integer, "ABS", 4)], ... "func": "f" ... } >>> func = Function('func') >>> rust_code(func(Abs(x) + ceiling(x)), user_functions=custom_functions) '(fabs(x) + x.CEIL()).f()' ``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(rust_code(expr, tau)) tau = if (x > 0) { x + 1 } else { 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])) >>> rust_code(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(rust_code(mat, A)) A = [x.powi(2), if (x > 0) { x + 1 } else { x }, x.sin()]; (R|tdoprint(Rt assign_toR((s2lib/python2.7/site-packages/sympy/printing/rust.pyt rust_codesmcK stt||dS(s3Prints Rust representation of the given expression.N(tprintR'(RR((s2lib/python2.7/site-packages/sympy/printing/rust.pytprint_rust_codeosN(Rt __future__RRt sympy.coreRRRRRtsympy.core.compatibilityRRtsympy.printing.codeprinterR R tsympy.printing.precedenceR RRR|RR'R)(((s2lib/python2.7/site-packages/sympy/printing/rust.pyt s( ; & p