\c@ sdZddlmZmZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZidd 6d d 6d d 6d d 6d d 6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd 6d!d!6d"d"6d#d#6d$d$6d%d%6Zd&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9gZd:e fd;YZd>d<Zd=Zd>S(?s R code printer The RCodePrinter converts single sympy expressions into single R expressions, using the functions defined in math.h where possible. i(tprint_functiontdivision(t Assignment(t string_typestrange(t CodePrinter(t precedencet PRECEDENCE(tRangetabstAbstsintcosttantasintacostatantatan2texptlogterftsinhtcoshttanhtasinhtacoshtatanhtfloortceilingtsigntmaxtMaxtmintMint factorialtgammatdigammattrigammatbetatiftelsetrepeattwhiletfunctiontfortintnexttbreaktTRUEtFALSEtNULLtInftNaNtNAt NA_integer_tNA_real_t NA_complex_t NA_character_tvolatilet RCodePrintercB speZdZdZdZi d/d6dd6dd6id6ed 6ed 6ed 6ed 6d d6Z idd6dd6dd6Z iZ idZ dZ dZdZdZdZdZdZdZdZdZd Zd!Zd"Zd#Zd$Zd%Zd&Zd'Zd(Zd)Z d*Z!d+Z"d,Z#d-Z$d.Z%RS(0s<A printer to convert python expressions to strings of R codet_rcodetRtordertautot full_precit precisiontuser_functionsthumantcontractt dereferenceterror_on_reservedt_treserved_word_suffixt&tandt|tort!tnotcC sotj||tt|_|jdi}|jj|t|jdg|_tt|_dS(NRBRE( Rt__init__tdicttknown_functionstgettupdatetsett _dereferencetreserved_words(tselftsettingst userfuncs((s3lib/python2.7/site-packages/sympy/printing/rcode.pyROfs cC s|dS(Ni((RWtp((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_rate_index_positionnscC sd|S(Ns%s;((RWt codestring((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_get_statementqscC s dj|S(Ns// {0}(tformat(RWttext((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _get_commenttscC sdj||S(Ns {0} = {1};(R^(RWtnametvalue((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_declare_number_constwscC s |j|S(N(t indent_code(RWtlines((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _format_codezsc s)|j\}fdt|DS(Nc3 s.|]$}tD]}||fVqqdS(N(R(t.0titj(tcols(s3lib/python2.7/site-packages/sympy/printing/rcode.pys s(tshapeR(RWtmattrows((Rjs3lib/python2.7/site-packages/sympy/printing/rcode.pyt_traverse_matrix_indices}scC sg}g}d}xm|D]e}|j|i|j|jd6|j|jdd6|j|jdd6|jdqW||fS(sPReturns a tuple (open_lines, close_lines) containing lists of codelines s#for (%(var)s in %(start)s:%(end)s){tvaritstarttendt}(tappendt_printtlabeltlowertupper(RWtindicest open_linest close_linest loopstartRh((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_get_loop_opening_endings  cC sd|jkr|j|St|}|jdkrNd|j|j|S|jdkrqd|j|jSd|j|j||j|j|fSdS(NtPowis1.0/%sg?ssqrt(%s)s%s^%s(RQt_print_FunctionRRt parenthesizetbaseRt(RWtexprtPREC((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_Pows  cC s-t|jt|j}}d||fS(Ns %d.0/%d.0(tintRZtq(RWRRZR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_RationalscC sKg|jD]}|j|^q }d|j|jjdj|fS(Ns%s[%s]s, (RxRtRRutjoin(RWRRhtinds((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_Indexeds%cC s|j|jS(N(RtRu(RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_IdxscC sdS(Nsexp(1)((RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_Exp1scC sdS(Ntpi((RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_PiscC sdS(NR3((RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_InfinityscC sdS(Ns-Inf((RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_NegativeInfinitysc C s'ddlm}ddlm}|j}|j}t||rg}x\|j|D]K\}}t|||f|||f} |j | } |j | qWWdj |S|j dr|j |s|j |r|j||S|j |} |j |} |jd| | fSdS(Ni(t MatrixSymbol(t IndexedBases RDs%s = %s(t"sympy.matrices.expressions.matexprRtsympy.tensor.indexedRtlhstrhst isinstanceRnRRtRsRt _settingsthast_doprint_loopsR]( RWRRRRRReRhRittemptcode0tlhs_codetrhs_code((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_Assignments"  # cC s|jdjtkr6d|j|jdj}n6d|j|jdj|j|jdjf}|}xKt|jd D]6\}}d|j||j|f|d}qW|S(Nis%ssifelse(%s,%s,NA)s ifelse(%s,%s,t)(targstcondtTrueRtRtreversed(RWRt last_linetcodetetc((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_Piecewises 6 .cC sMddlm}||jd|jdf|jdtf}|j|S(Ni(t Piecewiseiii(tsympy.functionsRRRRt(RWRRt _piecewise((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_ITEs0cC sAdj|j|jtddt|j|j|jjdS(Ns{0}[{1}]tAtomtstricti(R^RtparentRRRiRhRk(RWR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_MatrixElementscC s<tt|j|}||jkr4dj|S|SdS(Ns(*{0})(tsuperR;t _print_SymbolRUR^(RWRRa((s3lib/python2.7/site-packages/sympy/printing/rcode.pyRs cC s@|j|j}|j|j}|j}dj|||S(Ns {0} {1} {2}(RtRRtrel_opR^(RWRRRtop((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_Relationals cC sddlm}ddlm}ddlm}|||jd|jd||jddfdtf}|j|S(Ni(R (tNe(Rii( t(sympy.functions.elementary.trigonometricR tsympy.core.relationalRRRRRRt(RWRR RRR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_sincs @cC s@|j|j}|j}|j|j}dj|||S(Ns {0} {1} {2};(RtRRRR^(RWRRRR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt_print_AugmentedAssignments c C s|j|j}t|jtr<|jj\}}}n td|j|j}djd|d|d|d|d|S(Ns*Only iterable currently supported is RangesLfor ({target} = {start}; {target} < {stop}; {target} += {step}) {{ {body} }}ttargetRptstoptsteptbody( RtRRtiterableRRtNotImplementedErrorRR^(RWRRRpRRR((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt _print_For s   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 linests t{t(s{ s( RrRs is s%s%s(RRs{ s( (RrR(RRRdt splitlinesRRtlstripRtanytmaptendswitht startswitht enumerateRs( RWRt code_linesttabt inc_tokent dec_tokentlinetincreasetdecreasetprettytleveltn((s3lib/python2.7/site-packages/sympy/printing/rcode.pyRds( "1. N(&t__name__t __module__t__doc__t printmethodtlanguagetNoneRRTtFalset_default_settingst _operatorst _relationalsROR[R]R`RcRfRnR|RRRRRRRRRRRRRRRRRRd(((s3lib/python2.7/site-packages/sympy/printing/rcode.pyR;MsX                   %       cK st|j||S(sConverts 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;tdoprint(Rt assign_toRX((s3lib/python2.7/site-packages/sympy/printing/rcode.pytrcode2skcK stt||dS(s0Prints R representation of the given expression.N(tprintR(RRX((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt print_rcodesN(Rt __future__RRtsympy.codegen.astRtsympy.core.compatibilityRRtsympy.printing.codeprinterRtsympy.printing.precedenceRRtsympy.sets.fancysetsRRQRVR;RRR(((s3lib/python2.7/site-packages/sympy/printing/rcode.pyt sr   n