EdAS <dZddlmZmZddlmZmZmZm Z ddl m Z idddddd d d d d ddfdddfdddfdddfdddfdddfdddfgddddfgd d!d"d#d$d%d&d&d'd'd(d(d)d)d*d*d+d+d,d,d-d-d.d/d0d1d2dd3Z gd4Z Gd5d6e Zd:d8Zd9Zd7S);a 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. )AnyDict)SRationalFloatLambda) CodePrinterfloorceilingceilAbsabssignsignumPowc$|tj kSNrOnebaseexps 3lib/python3.11/site-packages/sympy/printing/rust.pyrGssquf}recipc"|tjkSrrHalfrs rrrHs saf}rsqrtc$|tj kSrrrs rrrIssqvg~rz sqrt().recipc*|tddkS)N)rrs rrrJsshq!nn4rcbrtc(|tjdzkS)Nrrrs rrrKstquQwrexp2r%c|jSr is_integerrs rrrLss~rpowir$c|j Srr*rs rrrMs 3>1rpowfrcdS)NT)rs rrrNsrloglnMaxmaxMinminsincostanasinacosatanatan2sinhcoshtanhasinhacoshatanh)r?r@rArBrCr!)4abstractalignofasbecomeboxbreakconstcontinuecratedoelseenumexternfalsefinalfnforifimplinletloopmacromatchmodmovemutoffsetofoverrideprivprocpubpurerefreturnSelfselfsizeofstaticstructsupertraittruetypetypeofunsafeunsizedusevirtualwherewhileyieldc :eZdZdZdZdZdddiddedd dd Zifd Zd Z d Z dZ dZ dZ dZdZdZdZdZd-fd Zd-fd ZdZdZdZdZdZd-dZd-dZd-dZd-d Zd-d!Zd-d"Zd-d#Z d-d$Z!d%Z"d&Z#d'Z$d(Z%d)Z&fd*Z'd+Z(d,Z)xZ*S).RustCodePrinterz>A printer to convert SymPy expressions to strings of Rust code _rust_codeRustNautoTF_) order full_prec precisionuser_functionshumancontract dereferenceerror_on_reservedreserved_word_suffixinlinecDtj||tt|_|di}|j|t |dg|_t t|_dS)Nrr) r __init__dictknown_functionsgetupdateset _dereferencereserved_words)rhsettings userfuncss rrzRustCodePrinter.__init__sT8,,,#O44LL!1266  ##I... ]B ? ?@@!.11rc |dzS)Nr0)rhps r_rate_index_positionz$RustCodePrinter._rate_index_positions s rc d|zS)Nz%s;r0)rh codestrings r_get_statementzRustCodePrinter._get_statements z!!rc d|zS)Nz// %sr0)rhtexts r _get_commentzRustCodePrinter._get_comments ~rcd|d|dS)Nzconst z: f64 = ;r0)rhnamevalues r_declare_number_constz%RustCodePrinter._declare_number_consts(,eee44rc,||Sr) indent_code)rhliness r _format_codezRustCodePrinter._format_codes&&&rcN|j\}fdt|DS)Nc3DK|]}tD]}||fV dSr)range).0ijcolss r z;RustCodePrinter._traverse_matrix_indices..s:AA1U4[[AAAAAAAAAAr)shaper)rhmatrowsrs @r_traverse_matrix_indicesz(RustCodePrinter._traverse_matrix_indicess.Y dAAAAd AAAArc g}g}d}|D]y}||||||j||jdzdz|dz||fS)Nz#for %(var)s in %(start)s..%(end)s {r$)varstartend})append_printlowerupper)rhindices open_lines close_lines loopstartrs r_get_loop_opening_endingz(RustCodePrinter._get_loop_opening_endings  9  $ $A   i{{1~~QW--{{17Q;//+1+11 2 2 2   s # # # #;&&rct|jdkrd||zdzS|jr||dS||S)Nr$()T)_type)lenargsr is_numberrhexprs r_print_caller_varz!RustCodePrinter._print_caller_varsg ty>>A  %T***S0 0 ^ %;;t4;00 0;;t$$ $rcd|jjjvr2j|jj}d}d}t|tr|}n|D]\}}}||jrn||dkrcd|jd|t|jdkr#|jdddnddz}n{|dkr'd |jd|d z}nN|d kr'd |jd|d z}n!d ||jdd z}|SdSt|dr 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_c3BK|]}|VdSr) _can_print)rfrhs rrz2RustCodePrinter._print_Function..Qs/0Y0Y1C1C0Y0Y0Y0Y0Y0Yr)r__name__r isinstancestrrrr stringifyhasattrrrr_rewriteable_functionsrallrewrite_print_not_supported) rhr cond_funcrstylecondrettarget_f required_fss ` r_print_FunctionzRustCodePrinter._print_Functions 9 !5 5* 3,TY-?@IDE)S))  )2%D$tTY' A:8#55dilCC"&GJ49~~XYGY aty}d C C C_a<<CC aZ0#55dilCC"&44CCaZ 0#55dilCC"&44CC / $ $ty$ ? ?22C -  .T7 # # 3 4:v(F(F 3;;ztz49566 6 Y 4#> > 3$($? @R$S !Hkx(( ;S0Y0Y0Y0Y[0Y0Y0Y-Y-Y ;{{4<<#9#9::: ; ; ; ;,,T22 2rc|jjrQ|jjsEt|t |j|j}||S||Sr)rr+rrorrrrs r _print_PowzRustCodePrinter._print_PowVsd 9  %(; %4::eDI..99D;;t$$ $##D)))rcXt|}|r|dzS|S)N_f64)rl _print_Floatrhrrr __class__s rrzRustCodePrinter._print_Float\s1gg""4((  < JrcXt|}|r|dzS|S)N_i32)rl_print_Integerrs rrzRustCodePrinter._print_Integercs1gg$$T**  < Jrc`t|jt|j}}d||fzS)Nz %d_f64/%d.0)intrq)rhrrrs r_print_RationalzRustCodePrinter._print_Rationaljs*46{{CKK11v%%rc||j}||j}|j}d|||S)Nz{} {} {})rlhsrhsrel_opformat)rhrlhs_coderhs_codeops r_print_Relationalz!RustCodePrinter._print_RelationalnsG;;tx((;;tx(( [  2x888rc4|j}tj}tj}t t |jD] }||j||zz }|||z}!||j j d||dS)N[]) rrZerorreversedrrankrrrlabel)rhrdimselemoffsetrs r_print_IndexedzRustCodePrinter._print_Indexedtszv% **++  A DLOF* *D d1g FF;;ty7777T9J9J9J9JKKrc|jjSr)rrrs r _print_IdxzRustCodePrinter._print_Idx~s zrc|jSr)rrs r _print_DummyzRustCodePrinter._print_Dummys yrcdS)NEr0rhrrs r _print_Exp1zRustCodePrinter._print_Exp1ssrcdS)NPIr0r s r _print_PizRustCodePrinter._print_PistrcdS)NINFINITYr0r s r_print_InfinityzRustCodePrinter._print_InfinityszrcdS)N NEG_INFINITYr0r s r_print_NegativeInfinityz'RustCodePrinter._print_NegativeInfinitys~rcdS)Nrnr0r s r_print_BooleanTruez"RustCodePrinter._print_BooleanTruesvrcdS)NrQr0r s r_print_BooleanFalsez#RustCodePrinter._print_BooleanFalseswrcDt|Sr)rrr s r _print_boolzRustCodePrinter._print_bools4yy   rcdS)NNANr0r s r _print_NaNzRustCodePrinter._print_NaNsurc|jdjdkrtdg}t|jD]\}\}}|dkr,|d||znX|t |jdz kr|dkr|dxxdz cc<n&|dxxd||zz cc<||}|||d |jd rd |Sd |S) NTzAll 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) {rr  ) rr ValueError enumeraterrr _settingsjoin)rhrrreccode0s r_print_Piecewisez RustCodePrinter._print_PiecewisesV 9R=  % 0/00 0 "49--  IAv1Av @ [4;;q>>9::::c$)nnq(( @Q$Y @b Y& b .Q?? KKNNE LL    LL     >( # $88E?? "99U## #rcbddlm}|||dS)Nr) PiecewiseF)deep)sympy.functionsr-rr)rhrr-s r _print_ITEzRustCodePrinter._print_ITEs6------{{4<< <>>???rc|jdkr$ddfd|DzStd)Nr$z[%s]rc3BK|]}|VdSr)r)rarhs rrz4RustCodePrinter._print_MatrixBase..s-%@%@dkk!nn%@%@%@%@%@%@rzLFull Matrix Support in Rust need Crates (https://crates.io/keywords/matrix).)rr'r$)rhAs` r_print_MatrixBasez!RustCodePrinter._print_MatrixBasesM 6Q; mDII%@%@%@%@a%@%@%@@@@ @kll lrc,||Sr)r)rhrs r_print_SparseRepMatrixz&RustCodePrinter._print_SparseRepMatrixs((---rc\|jd|j|j|jjdzzdS)Nrr$r)parentrrrrs r_print_MatrixElementz$RustCodePrinter._print_MatrixElements8;;;6DF4;+ !KrcLddlm}|j}|j}|jdr@||s||r|||S||}||}||d|S)Nr) IndexedBaserz = ) sympy.tensor.indexedr>rrr&has_doprint_loopsrr)rhrr>rrrrs r_print_Assignmentz!RustCodePrinter._print_Assignments444444hh >* % I377;+?+? I $$ I&&sC00 0{{3''H{{3''H&&HHHhh'GHH Hrc 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 linesTrz ){rz{ z( )rrc8g|]}|dS)z )lstrip)rlines r z/RustCodePrinter.indent_code..s$666U##666rc ng|]1}ttt|j2Sr0)ranymapendswith)rrG inc_tokens rrHz/RustCodePrinter.indent_code..s5OOOSS ::;;<<OOOrc ng|]1}ttt|j2Sr0)rrJrK startswith)rrG dec_tokens rrHz/RustCodePrinter.indent_code..sE(((S)<<==>>(((rr)rr#)rrr splitlinesr'r%r) rhcode code_linestabincreasedecreaseprettylevelnrGrPrMs @@rrzRustCodePrinter.indent_codes5 dC  '))$//$*?*?@@J77:&& &,  66666OOOOOOO((((!%(((  ! !GAtz!  d### Xa[ E MMCIItt4 5 5 5 Xa[ EE r)F)+r __module__ __qualname____doc__ printmethodlanguager_default_settingsrrrrrrrrrrrrrrrrrr r rrrrrrrr+r0r5r7r:r<rBr __classcell__)rs@rryrysHHKHsuu" #  !#2222"""555'''BBB ' ' ' % % %636363p*** &&&999 LLL!!!!$$$6@@@mmm ...AAA I I IrryNc Ht|||S)aConverts 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()]; )rydoprint)r assign_tors r rust_coderds#Z 8 $ $ , ,T9 = ==rc :tt|fi|dS)z3Prints Rust representation of the given expression.N)printrd)rrs rprint_rust_codergms& )D % %H % %&&&&&rr)r\typingrrtDict sympy.corerrrrsympy.printing.codeprinterr rrryrdrgr0rrrlsL  B&%%%%%%%111111111111222222$2  W 2v2 52 H2  , ,gq 9 , ,fa 8 - -~q A 4 4fa @ . . : - -vq 9 1 161 =  ?!2.  ua ( )/20 412< 5=2> 5?2B 5C2D 5E2F 5G2H FI2J FK2L FM2N WO2V FW2X      c222X333laaaaakaaaH m>m>m>m>`'''''r