+ j7Rt^RIHt^RIHt^RIHt^RIHt^RI H t H t ^RI H t /RRbR R bR R bR R bR R bR R bRRbRRbRRbRRbRRbRRbRRbRRbRRbRRbRRbRRRRRRRRRRR R R!R!R"R"R#R#R$R$R%R%/ Ct.R+Ot!R&R']4tR,R)ltR*tR(#)-z R code printer The RCodePrinter converts single SymPy expressions into single R expressions, using the functions defined in math.h where possible. ) annotations)Any) equal_valued) CodePrinter) precedence PRECEDENCE)RangeAbsabssincostanasinacosatanatan2explogerfsinhcoshtanhasinhacoshatanhfloorceilingsignMaxmaxMinmin factorialgammadigammatrigammabetasqrtc .a]tRt^Nt$RtRtRt]!]P3/R^R/RRR] !4/Bt R ] R &R R R RRR/t /t R] R&/3RltRtRtRtRtRtRtRtRtRtRtRtRtR tR!tR"tR#tR$tR%t V3R&lt!R't"R(t#R)t$R*t%R+t&V;t'#), RCodePrinterz;A printer to convert SymPy expressions to strings of R code_rcodeR precisionuser_functionscontractT dereferencezdict[str, Any]_default_settingsand&or|not!zdict[str, str] _relationalsc \P!W4\\4VnVP R/4pVPP V4\ VP R.44Vn\ \4VnR#)r-r/N) r__init__dictknown_functionsgetupdateset _dereferencereserved_words)selfsettings userfuncss&& U/home/luca/screen-checking/.venv/lib/python3.14/site-packages/sympy/printing/rcode.pyr9RCodePrinter.__init__asgT,#O4LL!126  ##I. ]B ?@!.1c V^,#))rAps&&rD_rate_index_position!RCodePrinter._rate_index_positionis s rFc RV,#)z%s;rI)rA codestrings&&rD_get_statementRCodePrinter._get_statementls z!!rFc $RPV4#)z// {}format)rAtexts&&rD _get_commentRCodePrinter._get_commentos~~d##rFc $RPW4#)z{} = {};rR)rAnamevalues&&&rD_declare_number_const"RCodePrinter._declare_number_constrs  --rFc $VPV4#N) indent_code)rAliness&&rD _format_codeRCodePrinter._format_codeus&&rFc LaVPwpoV3Rl\V44#)c3N<"TFp\S4Fq!V3xK K R#5ir])range).0ijcolss& rD 8RCodePrinter._traverse_matrix_indices..zsA 1U4[A[ s"%)shaperd)rAmatrowsrhs&& @rD_traverse_matrix_indices%RCodePrinter._traverse_matrix_indicesxsYY dAd AArFc  6.p.pRpVFpVPVRVPVP4RVPVP^,4RVPVP^,4/,4VPR4K W#3#)zPReturns a tuple (open_lines, close_lines) containing lists of codelines z#for (%(var)s in %(start)s:%(end)s){varstartend})append_printlabellowerupper)rAindices open_lines close_lines loopstartrfs&& rD_get_loop_opening_ending%RCodePrinter._get_loop_opening_ending|s  9 A   it{{177+QWWQY/t{{177Q;/+11 2   s # &&rFc RVP9dVPV4#\V4p\VPR4'd$RVP VP V4,#\VPR4'd#RVPVP 4,#VP VP V4: RVP VPV4: 2#)Powz1.0/%sg?zsqrt(%s)^)r;_print_Functionrrr parenthesizebaserv)rAexprPRECs&& rD _print_PowRCodePrinter._print_Pows D(( (''- -$ " % %t00DAB B $((C ( ( DII 66 6"// 4@!%!2!2488T!BD DrFc h\VP4\VP4r2RW#3,#)z %d.0/%d.0)intrJq)rArrJrs&& rD_print_RationalRCodePrinter._print_Rationals&466{CK1aV##rFc VPUu.uFq PV4NK ppVPVPP4: RRP V4: R2#uupi)[z, ])rzrvrrwjoin)rArrfindss&& rD_print_IndexedRCodePrinter._print_IndexedsI)-7AQ7;;tyy74II8sA&c R#)zexp(1)rIrArs&&rD _print_Exp1RCodePrinter._print_Exp1srFc R#)pirIrs&&rD _print_PiRCodePrinter._print_PisrFc R#)InfrIrs&&rD_print_InfinityRCodePrinter._print_InfinitysrFc R#)z-InfrIrs&&rD_print_NegativeInfinity$RCodePrinter._print_NegativeInfinitysrFc l^RIHp^RIHp^RIHpVP pVPp\WS4'di.pVPV4F@wrV!WXV 3,WhV 3,4p VPV 4p VPV 4KB RPV4#VPR,'d@VPV4'gVPV4'dVPWe4#VPV4p VPV4p VP!V : RV : 24#)) Assignment) MatrixSymbol) IndexedBase r.z = )sympy.codegen.astr"sympy.matrices.expressions.matexprrsympy.tensor.indexedrlhsrhs isinstancernrvrur _settingshas_doprint_loopsrO)rArrrrrrr_rfrgtempcode0lhs_coderhs_codes&& rD_print_AssignmentRCodePrinter._print_Assignments0C4hhhh c ( (E77<!#d)SAY7 D) U#=99U# # ^^J ' 'SWW[-A-A $$&&s0 0{{3'H{{3'H&&Hh'GH HrFc $VPR,PR8Xd5RVPVPR,P4,pM_RVPVPR,P4: RVPVPR,P4: R2pTp\ VPRR4F<wrERVPV4: RVPV4: R2V,R,pK> V#) Tz%szifelse(,z,NA)N)r)argscondrvrreversed)rAr last_linecodeecs&& rD_print_PiecewiseRCodePrinter._print_Piecewises 99R=   %t{{499R=+=+=>>I &/3kk$))B-:L:L.Mt{{[_[d[deg[h[m[mOnoI TYYs^,DAq%)[[^T[[^DTI#MD- rFc N^RIHpVPVPV44#)r) Piecewise)sympy.functionsrrvrewrite)rArrs&& rD _print_ITERCodePrinter._print_ITEs-{{4<< 233rFc RPVPVP\R,RR7VPVP VPP ^,,,4#)z{}[{}]AtomT)strict)rSrparentrrgrfrkrs&&rD_print_MatrixElement!RCodePrinter._print_MatrixElements[t00j>P 1 &&466$++*;*;A*>#>>@ @rFc h<\SV`V4pWP9dRPV4#V#)z(*{}))super _print_Symbolr?rS)rArrX __class__s&& rDrRCodePrinter._print_Symbols2w$T* $$ $>>$' 'KrFc VPVP4pVPVP4pVPpRP W$V4#)z{} {} {})rvrrrel_oprS)rArrrops&& rD_print_RelationalRCodePrinter._print_RelationalsB;;txx(;;txx( [[  x88rFc VPVP4pVPpVPVP4pRP W#V4#)z {} {} {};)rvrrrrS)rArrrrs&& rD_print_AugmentedAssignment'RCodePrinter._print_AugmentedAssignmentsB;;txx( WW;;txx(!!(99rFc .VPVP4p\VP\4'dVPP wr4pM \ R4hVPVP4pRPW#V^, WVR7#)z*Only iterable currently supported is RangezCfor({target} in seq(from={start}, to={stop}, by={step}){{ {body} }})targetrrstopstepbody) rvrriterablerrNotImplementedErrorrrS)rArrrrrrrs&& rD _print_ForRCodePrinter._print_Fors|T[[) dmmU + + $ 2 2 E%&RS S{{499%V]]ek!V$^3 3rFc  \V\4'd2VPVPR44pRP V4#RpRpRpVUu.uFqfP R4NK ppVUu.uF+p\ \\VPV444NK- ppVUu.uF+p\ \\VPV444NK- pp.p ^p \V4FZwrVR9dV PV4KWV ,,p V PW:,: V: 24WV ,, p K\ V #uupiuupiuupi)z0Accepts a string of code or a list of code linesTz z ){(z{ z( )rtr)rr) rstrr^ splitlinesrlstripranymapendswith startswith enumerateru) rAr code_linestab inc_token dec_tokenlineincreasedecreaseprettylevelns && rDr^RCodePrinter.indent_codes3 dC ))$//$*?@J77:& &,  046U#6IMOSS :;<O!%(!%S)<=>!% ( GAz! d# a[ E MMCIt4 5 a[ E ' 7O(sE 11E(1E)r?r;r@)(__name__ __module__ __qualname____firstlineno____doc__ printmethodlanguager:rr0r>__annotations__ _operatorsr7r9rKrOrUrZr`rnr~rrrrrrrrrrrrrrrr^__static_attributes__ __classcell__)rs@rDr)r)NsEKH(,[-J-J)R"Dsu O)~ c c cJ $&L.% "2"$.'B ' D$J%IN 4@9 : 3rFr)Nc 6\V4PW4#)aConverts an expr to a string of r 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 R string representations. Alternatively, the dictionary value can be a list of tuples i.e. [(argument_test, rfunction_string)] or [(argument_test, rfunction_formater)]. 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 rcode, symbols, Rational, sin, ceiling, Abs, Function >>> x, tau = symbols("x, tau") >>> rcode((2*tau)**Rational(7, 2)) '8*sqrt(2)*tau^(7.0/2.0)' >>> rcode(sin(x), assign_to="s") 's = sin(x);' 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"), ... (lambda x: x.is_integer, "ABS")], ... "func": "f" ... } >>> func = Function('func') >>> rcode(func(Abs(x) + ceiling(x)), user_functions=custom_functions) 'f(fabs(x) + CEIL(x))' or if the R-function takes a subset of the original arguments: >>> rcode(2**x + 3**x, user_functions={'Pow': [ ... (lambda b, e: b == 2, lambda b, e: 'exp2(%s)' % e), ... (lambda b, e: b != 2, 'pow')]}) 'exp2(x) + pow(3, 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(rcode(expr, assign_to=tau)) tau = ifelse(x > 0,x + 1,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])) >>> rcode(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(rcode(mat, A)) A[0] = x^2; A[1] = ifelse(x > 0,x + 1,x); A[2] = sin(x); )r)doprint)r assign_torBs&&,rDrcoder"sV  ! ) )$ ::rFc 0\\V3/VB4R#)z0Prints R representation of the given expression.N)printr)rrBs&,rD print_rcoders % ! !"rF)ifelserepeatwhilefunctionforinnextbreakTRUEFALSENULLrNaNNA NA_integer_NA_real_ NA_complex_ NA_character_volatiler])r __future__rtypingrsympy.core.numbersrsympy.printing.codeprinterrsympy.printing.precedencerrsympy.sets.fancysetsrr;r@r)rrrIrFrDr,sP#+2<& 5 5 5   5   F  F F W 5 5 5 F F F  W!" W#$ W%& W y F 5 5 W y  F F;F,Q;Qhk;\#rF