Ed7 dZddlmZmZddlmZddlmZm Z ddl m Z iddddd d d d d d d d d d ddddddddddddddddddddddddddddd d!d"d# Z gd$Z Gd%d&eZd*d(Zd)Zd'S)+z R code printer The RCodePrinter converts single SymPy expressions into single R expressions, using the functions defined in math.h where possible. )AnyDict) CodePrinter) precedence PRECEDENCE)RangeAbsabssincostanasinacosatanatan2explogerfsinhcoshtanhasinhacoshatanhfloorceilingsignmaxmin factorialgammadigammatrigammabetasqrt) rrrMaxMinr r!r"r#r$r%)ifelserepeatwhilefunctionforinnextbreakTRUEFALSENULLInfNaNNA NA_integer_NA_real_ NA_complex_ NA_character_volatilec eZdZdZdZdZdddiddedd d Zd d d dZiZ ifdZ dZ dZ dZ dZdZdZdZdZdZdZdZdZdZdZdZdZd Zd!Zd"Zfd#Zd$Zd%Z d&Z!d'Z"xZ#S)( RCodePrinterz;A printer to convert SymPy expressions to strings of R code_rcodeRNautoTF_) order full_prec precisionuser_functionshumancontract dereferenceerror_on_reservedreserved_word_suffix&|!)andornotcDtj||tt|_|di}|j|t |dg|_t t|_dS)NrFrI) r__init__dictknown_functionsgetupdateset _dereferencereserved_words)selfsettings userfuncss 4lib/python3.11/site-packages/sympy/printing/rcode.pyrSzRCodePrinter.__init__esT8,,,#O44LL!1266  ##I... ]B ? ?@@!.11c |dzS)N)r[ps r^_rate_index_positionz!RCodePrinter._rate_index_positionms s r_c d|zS)Nz%s;rb)r[ codestrings r^_get_statementzRCodePrinter._get_statementps z!!r_c,d|S)Nz// {}format)r[texts r^ _get_commentzRCodePrinter._get_commentss~~d###r_c.d||S)Nz{} = {};ri)r[namevalues r^_declare_number_constz"RCodePrinter._declare_number_constvs  u---r_c,||SN) indent_code)r[liness r^ _format_codezRCodePrinter._format_codeys&&&r_cN|j\}fdt|DS)Nc3DK|]}tD]}||fV dSrr)range).0ijcolss r^ z8RCodePrinter._traverse_matrix_indices..~s:AA1U4[[AAAAAAAAAAr_)shaperx)r[matrowsr|s @r^_traverse_matrix_indicesz%RCodePrinter._traverse_matrix_indices|s.Y dAAAAd AAAAr_c g}g}d}|D]}||||j||jdz||jdzdz|d||fS)zPReturns a tuple (open_lines, close_lines) containing lists of codelines z#for (%(var)s in %(start)s:%(end)s){)varstartend})append_printlabellowerupper)r[indices open_lines close_lines loopstartrzs r^_get_loop_opening_endingz%RCodePrinter._get_loop_opening_endings  9  $ $A   i{{17++QWQY//{{17Q;//+1+11 2 2 2   s # # # #;&&r_cpd|jvr||St|}|jdkrd||j|zS|jdkrd||jzS||j|d||j|S)NPowz1.0/%sg?zsqrt(%s)^)rU_print_Functionrr parenthesizebaser)r[exprPRECs r^ _print_PowzRCodePrinter._print_Pows D( ( .''-- -$ 8r> Dt00DAAB B X_ D DI 6 66 6"// 4@@@@!%!2!248T!B!B!BD Dr_c`t|jt|j}}d||fzS)Nz %d.0/%d.0)intrcq)r[rrcrs r^_print_RationalzRCodePrinter._print_Rationals*46{{CKK1aV##r_cfd|jD}|jjdd|dS)Nc:g|]}|Srb)r)ryrzr[s r^ z/RCodePrinter._print_Indexed..s#777AQ777r_[z, ])rrrrjoin)r[rindss` r^_print_IndexedzRCodePrinter._print_IndexedsM7777777;;ty77774IIr_c6||jSrr)rrr[rs r^ _print_IdxzRCodePrinter._print_Idxs{{4:&&&r_cdS)Nzexp(1)rbrs r^ _print_Exp1zRCodePrinter._print_Exp1sxr_cdS)Npirbrs r^ _print_PizRCodePrinter._print_Pistr_cdS)Nr4rbrs r^_print_InfinityzRCodePrinter._print_Infinitysur_cdS)Nz-Infrbrs r^_print_NegativeInfinityz$RCodePrinter._print_NegativeInfinitysvr_ctddlm}ddlm}ddlm}|j}|j}t||rxg}| |D]K\}} |||| f||| f} | | } | | Ld |S|j dr@||s||r|||S| |} | |} || d| S)Nr) Assignment) MatrixSymbol) IndexedBase rHz = )sympy.codegen.astr"sympy.matrices.expressions.matexprrsympy.tensor.indexedrlhsrhs isinstancerrrr _settingshas_doprint_loopsrg)r[rrrrrrrtrzr{tempcode0lhs_coderhs_codes r^_print_AssignmentzRCodePrinter._print_Assignmentsn000000CCCCCC444444hh c< ( ( IE77<< $ $A!z#ad)SAY77 D)) U####99U## # ^J ' ISWW[-A-A I $$ I&&sC00 0{{3''H{{3''H&&HHHhh'GHH Hr_c|jdjdkr)d||jdjz}nOd||jdjd||jdjd}|}t |jddD]:\}}d||d||d|zdz};|S)NrTz%szifelse(,z,NA)))argscondrrreversed)r[r last_linecodeecs r^_print_PiecewisezRCodePrinter._print_Piecewises 9R=  % pt{{49R=+=>>>II p/3kk$)B-:L.M.M.M.Mt{{[_[deg[h[mOnOnOnOnoI TYss^,, N NDAqq%)[[^^^^T[[^^^^DTI#MDD r_c^ddlm}|||S)Nr) Piecewise)sympy.functionsrrrewrite)r[rrs r^ _print_ITEzRCodePrinter._print_ITEs2------{{4<< 22333r_cd||jtdd|j|j|jjdzzS)Nz{}[{}]AtomT)strictr)rjrparentrr{rzr~rs r^_print_MatrixElementz!RCodePrinter._print_MatrixElements[t00j>P 1  &46$+*;A*>#>>@@ @r_ct|}||jvrd|S|S)Nz(*{}))super _print_SymbolrYrj)r[rrn __class__s r^rzRCodePrinter._print_SymbolsAww$$T** 4$ $ >>$'' 'Kr_c||j}||j}|j}d|||S)Nz{} {} {})rrrrel_oprj)r[rrrops r^_print_RelationalzRCodePrinter._print_RelationalsG;;tx((;;tx(( [  2x888r_c||j}|j}||j}d|||S)Nz {} {} {};)rrrrrj)r[rrrrs r^_print_AugmentedAssignmentz'RCodePrinter._print_AugmentedAssignmentsG;;tx(( W;;tx((!!(B999r_c||j}t|jtr|jj\}}}nt d||j}d|||||S)Nz*Only iterable currently supported is RangezLfor ({target} = {start}; {target} < {stop}; {target} += {step}) {{ {body} }})targetrstopstepbody) rrriterablerrNotImplementedErrorrrj)r[rrrrrrs r^ _print_ForzRCodePrinter._print_ForsT[)) dmU + + T $ 2 E4%&RSS S{{49%%)*0&e4+1+1+1 1r_c 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 linesTz ){(z{ z( )rrc8g|]}|dS)z )lstrip)rylines r^rz,RCodePrinter.indent_code..s$666U##666r_c ng|]1}ttt|j2Srb)ranymapendswith)ryr inc_tokens r^rz,RCodePrinter.indent_code..s5OOOSS ::;;<<OOOr_c ng|]1}ttt|j2Srb)rrr startswith)ryr dec_tokens r^rz,RCodePrinter.indent_code..sE(((S)<<==>>(((r_r)rr)rstrrs splitlinesr enumerater) r[r code_linestabincreasedecreaseprettylevelnrrrs @@r^rszRCodePrinter.indent_code s5 dC  '))$//$*?*?@@J77:&& &,  66666OOOOOOO((((!%(((  ! !GAtz!  d### Xa[ E MMCIItt4 5 5 5 Xa[ EE r_)$__name__ __module__ __qualname____doc__ printmethodlanguagerX_default_settings _operators _relationalsrSrdrgrlrprurrrrrrrrrrrrrrrrrrrs __classcell__)rs@r^r=r=LsEEKHsuu" #  J L!#2222"""$$$...'''BBB ' ' ' D D D$$$JJJ'''%I%I%IN   444@@@999 ::: 1 1 1r_r=Nc Ht|||S)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_tor\s r^rcoder*s#V  ! ! ) )$ : ::r_c :tt|fi|dS)z0Prints R representation of the given expression.N)printr)rr\s r^ print_rcoders& % ! ! ! !"""""r_rr)r typingrrtDictsympy.printing.codeprinterrsympy.printing.precedencerrsympy.sets.fancysetsrrUrZr=rrrbr_r^rs&%%%%%%%222222<<<<<<<<&&&&&& 5 5 5   5   F  F F W 5 5 5 F F F  W!" W#$ W%&      ;F,[[[[[;[[[|k;k;k;k;\#####r_