ЦiR JSrSSKJr SSKJr SSKJrJrJrJ r SSK J r SSK J r 0SS_SS _S S _S S _SSSS4SSS4SSS4SSS4SSS4SSS4SSS4/_S S!S S4/_S"S#_S$S%_S&S'_S(S(_S)S)_S*S*_S+S+_S,S,_S-S-_S.S._S/S/_S0S1S2S3S4SS5.Er/S6Qr"S7S8\ 5rSrIs |C4recipc[US5$)Ng?r rs rrrJs |C5rsqrtc[US5$)Ngr rs rrrKs |C6rz sqrt().recipc U[SS5:H$)N)rrs rrrLsshq!n4rcbrtc[US5$)Nrr rs rrrMs |D!4rexp2r#cUR$N is_integerrs rrrNss~~rpowir"c$UR(+$r(r)rs rrrOs 3>>1rpowfrcg)NT)rs rrrPsrloglnMaxmaxMinminsincostanasinacosatanatan2sinhcoshtanhasinhacoshatanh)r>r?r@rArBr)4abstractalignofasbecomeboxbreakconstcontinuecratedoelseenumexternfalsefinalfnforifimplinletloopmacromatchmodmovemutoffsetofoverrideprivprocpubpurerefreturnSelfselfsizeofstaticstructsupertraittruetypetypeofunsafeunsizedusevirtualwherewhileyieldc z^\rSrSr%SrSrSr\"\R40S0S\ "5S.D6r S\ S '04S jr S r S rS rSrSrSrSrSrSrSrS-U4SjjrS-U4SjjrSrSrSrSrSrS-SjrS-SjrS-Sjr S-Sjr!S-S jr"S-S!jr#S-S"jr$S-S#jr%S$r&S%r'S&r(S'r)S(r*U4S)jr+S*r,S+r-S,r.U=r/$).RustCodePrinterz>A printer to convert SymPy expressions to strings of Rust code _rust_codeRustT) precisionuser_functionscontract dereferencezdict[str, Any]_default_settingsc[R"X5 [[5UlUR S05nURR U5 [ UR S/55Ul[ [5Ulg)Nr~r) r __init__dictknown_functionsgetupdateset _dereferencereserved_words)rgsettings userfuncss rrRustCodePrinter.__init__sgT,#O4LL!126  ##I. ]B ?@!.1rc US-$)Nr/)rgps r_rate_index_position$RustCodePrinter._rate_index_positions s rc SU-$)Nz%s;r/)rg codestrings r_get_statementRustCodePrinter._get_statements z!!rc SU-$)Nz// %sr/)rgtexts r _get_commentRustCodePrinter._get_comments ~rcSU<SU<S3$)Nzconst z: f64 = ;r/)rgnamevalues r_declare_number_const%RustCodePrinter._declare_number_consts (,e44rc$URU5$r() indent_code)rgliness r _format_codeRustCodePrinter._format_codes&&rcL^URunmU4Sj[U55$)Nc3P># UHn[T5Ho!U4v M M g7fr()range).0ijcolss r ;RustCodePrinter._traverse_matrix_indices..sA 1U4[A[ s#&)shaper)rgmatrowsrs @r_traverse_matrix_indices(RustCodePrinter._traverse_matrix_indicessYY dAd AArc /n/nSnUHpnURUURU5URUR5URURS-5S.-5 URS5 Mr X#4$)Nz#for %(var)s in %(start)s..%(end)s {r")varstartend})append_printlowerupper)rgindices open_lines close_lines loopstartrs r_get_loop_opening_ending(RustCodePrinter._get_loop_opening_endings  9 A   i{{1~QWW-{{177Q;/+11 2   s # &&rc[UR5S:aSURU5-S-$UR(aURUSS9$URU5$)Nr"()T)_type)lenargsr is_numberrgexprs r_print_caller_var!RustCodePrinter._print_caller_varsV tyy>A T**S0 0 ^^;;t4;0 0;;t$ $rc^URRTR;GaFTRURRnSnSn[U[5(aUnO"UHupSnU"UR 6(dM O UbUS:Xa_STR UR S5U[UR 5S:aTRUR SSS5OSS.-nU$US:Xa&S TR UR S5US .-nU$US :Xa&S TR UR S5US .-nU$S UTRUR S5S .-nU$g[US5(aG[UR[5(a(TRUR"UR 65$URRTR;awTRURRupxTRU5(a;[U4SjU55(a TRUR!U55$ggTR#U5$)a 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 Nr"z%(var)s.%(method)s(%(args)s)r, )rmethodrrz%(var)s.%(method)s())rrr#z%(func)s(%(args)s))funcr_imp_c3F># UHnTRU5v M g7fr() _can_print)rfrgs rr2RustCodePrinter._print_Function..Ms0Y[1C1C[!)r__name__r isinstancestrrrr stringifyhasattrrrr_rewriteable_functionsrallrewrite_print_not_supported) rgr cond_funcrstylecondrettarget_f required_fss ` r_print_FunctionRustCodePrinter._print_FunctionsB 99  !5!5 5,,TYY-?-?@IDE)S)) )2%DTYY''*3A:8#55diilC"&GJ499~XYGYtyy}d C_a<C( aZ0#55diilC"&4C aZ0#55diilC"&4C / $ $tyy$ ?2C - .T7 # # 4::v(F(F;;tzz49956 6 YY  4#>#> >$($?$? @R@R$S !Hx((S0Y[0Y-Y-Y{{4<<#9::.Z(,,T2 2rcURR(a[URR(d@[U5"[ UR5UR5nUR U5$UR U5$r()rr*rrnrrrrs r _print_PowRustCodePrinter._print_PowRsX 99  (;(;:eDII.9D;;t$ $##D))rc>>[TU]U5nU(aUS-$U$)N_f64)rk _print_Floatrgrrr __class__s rrRustCodePrinter._print_FloatXs$g"4( < Jrc>>[TU]U5nU(aUS-$U$)N_i32)rk_print_Integerrs rrRustCodePrinter._print_Integer_s$g$T* < Jrc`[UR5[UR5p2SX#4-$)Nz %d_f64/%d.0)intrq)rgrrrs r_print_RationalRustCodePrinter._print_Rationalfs&466{CK1v%%rcURUR5nURUR5nURnSR X$U5$)Nz{} {} {})rlhsrhsrel_opformat)rgrlhs_coderhs_codeops r_print_Relational!RustCodePrinter._print_RelationaljsB;;txx(;;txx( [[  x88rcRURn[Rn[Rn[ [ UR 55HnX1RUU-- nXBU-nM URURR5<SURU5<S3$)N[]) rrZeroOnereversedrrankrrrlabel)rgrdimselemoffsetrs r_print_IndexedRustCodePrinter._print_Indexedps}zzvv% *+A LLOF* *D 1g F, ;;tyy7T9JKKrc.URR$r()r rrs r _print_IdxRustCodePrinter._print_IdxzszzrcUR$r()rrs r _print_DummyRustCodePrinter._print_Dummy}s yyrcg)NEr/rgrrs r _print_Exp1RustCodePrinter._print_Exp1srcg)NPIr/rs r _print_PiRustCodePrinter._print_Pisrcg)NINFINITYr/rs r_print_InfinityRustCodePrinter._print_Infinitysrcg)N NEG_INFINITYr/rs r_print_NegativeInfinity'RustCodePrinter._print_NegativeInfinitysrcg)Nrmr/rs r_print_BooleanTrue"RustCodePrinter._print_BooleanTruesrcg)NrPr/rs r_print_BooleanFalse#RustCodePrinter._print_BooleanFalsesrc4[U5R5$r()rrrs r _print_boolRustCodePrinter._print_bools4y  rcg)NNANr/rs r _print_NaNRustCodePrinter._print_NaNsrc\URSRS:wa [S5e/n[UR5HunupEUS:Xa$UR SUR U5-5 OOU[ UR5S- :XaUS:XaUS==S- ss'OUS==SUR U5-- ss'UR U5nUR U5 UR S 5 M URS (aS RU5$S RU5$) NrTzAll 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.rz if (%s) {r"z else {z else if (%s) {rinline  ) rr ValueError enumeraterrr _settingsjoin)rgrrreccode0s r_print_Piecewise RustCodePrinter._print_Piecewises 99R=   %/0 0 "499-IAvAv [4;;q>9:c$))nq((Q$Yb Y& b .Q?? KKNE LL  LL . >>( #88E? "99U# #rcLSSKJn URURUSS95$)Nr) PiecewiseF)deep)sympy.functionsrErr)rgrrEs r _print_ITERustCodePrinter._print_ITEs!-{{4<< <>??rcv^URS:XaSSRU4SjU55-$[S5e)Nr"z[%s]rc3F># UHnTRU5v M g7fr()r)rargs rr4RustCodePrinter._print_MatrixBase..s%@adkk!nnarzLFull Matrix Support in Rust need Crates (https://crates.io/keywords/matrix).)rr>r;)rgAs` r_print_MatrixBase!RustCodePrinter._print_MatrixBases4 66Q;DII%@a%@@@ @kl lrc$URU5$r()r)rgrs r_print_SparseRepMatrix&RustCodePrinter._print_SparseRepMatrixs((--rcUR<SURURURRS--<S3$)Nrr"r)parentrrrrs r_print_MatrixElement$RustCodePrinter._print_MatrixElements9;;66DFF4;;+<+[TU]U5nXR;aSU-$U$)Nz(*%s))rk _print_Symbolr)rgrrrs rrYRustCodePrinter._print_Symbols-w$T* $$ $T> !KrcTSSKJn URnURnURS(a=UR U5(dUR U5(aUR XC5$URU5nURU5nURU<SU<35$)Nr) IndexedBaserz = ) sympy.tensor.indexedr\rrr=has_doprint_loopsrr)rgrr\rrrrs r_print_Assignment!RustCodePrinter._print_Assignments4hhhh >>* %377;+?+? $$&&s0 0{{3'H{{3'H&&Hh'GH Hrc [U[5(a1URURS55nSR U5$SnSnSnUVs/sHofR S5PM nnUVs/sH+n[ [[URU555PM- nnUVs/sH+n[ [[URU555PM- nn/n Sn [U5HEupUS;aU RU5 MXU -n U RX:-<U<35 XU - n MG U $s snfs snfs snf) z0Accepts a string of code or a list of code linesTrz ){rz{ z( )rrz r)rr:) rrr splitlinesr>lstripranymapendswith startswithr<r) rgcode code_linestab inc_token dec_tokenlineincreasedecreaseprettylevelns rrRustCodePrinter.indent_codes3 dC ))$//$*?@J77:& &,  046U#6IMOSS :;<O!%(!%S)<=>!% ( GAz! d# a[ E MMCIt4 5 a[ E ' 7O(sD712D<)2E)rrr)F)0r __module__ __qualname____firstlineno____doc__ printmethodlanguagerr rr__annotations__rrrrrrrrrrrrrrrrrrrr r$r(r+r.r1r5rBrHrOrRrVrYr`r__static_attributes__ __classcell__)rs@rrxrxsHKH(,[-J-J)u O)~!#2"5'B ' %63p* &9 L!$6@m .A IrrxNc 6[U5RX5$)a Converts 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()]; )rxdoprint)r assign_tors r rust_codersZ 8 $ , ,T ==rc .[[U40UD65 g)z3Prints Rust representation of the given expression.N)printr)rrs rprint_rust_coderis )D %H %&rr()ry __future__rtypingr sympy.corerrrrsympy.core.numbersr sympy.printing.codeprinterr rrrxrrr/rrrs B#11+2$2  W 2v2 52 H2  4gq A 5vq A 6 J 4fa @ 4fa @ -vq 9 161 =  ?!2. ua ( )/20 412< 5=2> 5?2B 5C2D 5E2F 5G2H FI2J FK2L FM2N WO2V FW2X      c2X3l[k[|m>`'r