~9\c!@ sdZddlmZmZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+dd l,m-Z-m.Z.dd l/m0Z0id d fddfgd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6dd6Z1e2e1id d 6d!d!6d"d"6d#d#6d$d%6Z3e3Z4e2e3i d&d&6d'd'6d'd'6d(d(6d)d)6d*d*6d+d,6d-d-6d.d.6d/d06d1d16d2d36d4d56Z5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVg!Z6dWdXgZ7dYZ8dZZ9d[e*fd\YZ:d]e;fd^YZ<ed_d`dadbdcdddedfdge<e:fdhYZ=die<e:fdjYZ>x+dkj?D]Z@eAe>dle@e>jBqWdme>fdnYZCie:do6e>dp6eCdq6ZDeEdpdrZFdsZGdtS(us C code printer The C89CodePrinter & C99CodePrinter converts single sympy expressions into single C expressions, using the functions defined in math.h where possible. A complete code generator, which uses ccode extensively, can be found in sympy.utilities.codegen. The codegen module can be used to generate complete source code files that are compilable without further modifications. i(tprint_functiontdivision(twraps(tchain(tS(t string_typestrange(t deprecated(t AssignmenttPointertVariablet Declarationtrealtcomplex_tintegertbool_tfloat32tfloat64tfloat80t complex64t complex128tintct value_constt pointer_consttint8tint16tint32tint64tuint8tuint16tuint32tuint64tuntyped(t CodePrintertrequires(t precedencet PRECEDENCE(tRangecC s|j S(N(t is_integer(tx((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt#ttfabscC s|jS(N(R&(R'((s3lib/python2.7/site-packages/sympy/printing/ccode.pyR(#R)tabstAbstsintcosttantasintacostatantatan2texptlogtsinhtcoshttanhtfloortceiltceilingtasinhtacoshtatanhterfttgammatgammatexp2texpm1tlog10tlog2tlog1ptcbrttCbrtthypottfmatlgammatloggammaterfctfmaxtMaxtfmintMintautotbreaktcasetchartconsttcontinuetdefaulttdotdoubletelsetenumtexterntfloattfortgototiftinttlongtregistertreturntshorttsignedtsizeoftstatictstructtentrytswitchttypedeftuniontunsignedtvoidtvolatiletwhiletinlinetrestrictcC s%ddlm}m}ddlm}ddlm}idtj6d|tj6dd|d6d |d6d |d 6d tj 6d tj d6dtj d6ddtj 6ddtj 6dd|tj 6dd|tj 6d|d6d|d6dd|d6dd|d6S(sV Returns a dictionary with math-related macros from math.h/cmath Note that these macros are not strictly required by the C/C++-standard. For MSVC they are enabled by defining "_USE_MATH_DEFINES" (preferably via a compilation flag). Returns ======= Dictionary mapping sympy expressions to strings (macro names) i(REtSqrt(R5(tsqrttM_EtM_LOG2EiitM_LN2tM_LN10i tM_PItM_PI_2tM_PI_4itM_1_PItM_2_PIt M_2_SQRTPItM_SQRT2t M_SQRT1_2( tsympy.codegen.cfunctionsRERut&sympy.functions.elementary.exponentialR5t(sympy.functions.elementary.miscellaneousRvRtExp1tPi(RERuR5Rv((s3lib/python2.7/site-packages/sympy/printing/ccode.pytget_math_macros[s(       c stfd}|S(s Decorator for printer methods When a Printer's method is decorated using this decorator the expressions printed will first be looked for in the attribute ``math_macros``, and if present it will print the macro name in ``math_macros`` followed by a type suffix for the type ``real``. e.g. printing ``sympy.pi`` would print ``M_PIl`` if real is mapped to float80. c sA||jkr-d|j||jtfS|||SdS(Ns%s%s(t math_macrost_get_math_macro_suffixR (tselftexprtkwargs(tmeth(s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _meth_wrappers(R(RR((Rs3lib/python2.7/site-packages/sympy/printing/ccode.pyt_as_macro_if_defineds tC89CodePrintercB seZdZdZdZdZeeZi dWd6dd6dd6id 6e d 6e d 6e d 6ed 6e d6dd6Z ie e 6ee6ee6Zide 6de6de6de 6de6de6de6de6de6de6de6de6de6de6Zi dhe6dhe6dhe6dhe6dhe6dhe6dhe6dhe6dhe6ZiZide6de 6de 6Z!ide6de 6de 6Z"ide 6Z#dWZ$dZ%e&Z'id Z(d!Z)d"Z*d#Z+d$Z,d%Z-d&Z.e/d'Z0e/d(Z1d)Z2d*Z3d+Z4d,Z5e/d-Z6d.Z7d/Z8d0Z9d1Z:d2Z;d3Z<d4Z=d5Z>d6Z?d7Z@d8ZAd9ZBd:ZCd;ZDd<ZEd=ZFd>ZGd?ZHd@ZIeJdAdhdBZKeJdAdhdCZLdDZMdEZNdFZOdGZPeJdAdHhdIZQdJZRdKZSdLZTdMZUdNZVdOZWdPZXdQZYdRZZdSZ[dTZ\dUZ]dVZ^e\Z_RS(Xs<A printer to convert python expressions to strings of c codet_ccodetCtC89torderRRt full_precit precisiontuser_functionsthumantallow_unknown_functionstcontractt dereferenceterror_on_reservedt_treserved_word_suffixRZRbR^tbooltint8_ttint16_ttint32_ttint64_ts stdbool.hsstdint.htfR)tltFtLcC s |jdkr*|jdt|_ntt|jj|jdij|_tt|jj|jdij|_tt|j j|jdij|_ tt|j j|jdij|_ tt|j j|jdij|_ tt|j j|jdij|_ tt|j j|jdij|_ tt|j|t|j|jd i|_t|jd g|_t|_t|_t|_dS( NRt type_aliasest type_mappingst type_headerst type_macrosttype_func_suffixesttype_literal_suffixesttype_math_macro_suffixesRR(RtNonetpopRtdictRRtitemsRRRRRRtsuperRt__init__t_kftgettknown_functionstsett _dereferencetheaderst librariestmacros(Rtsettings((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRs,!!!!!!!!  cC s|dS(Ni((Rtp((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_rate_index_positionscC s|jdr|S|dS(s@ Get code string as a statement - i.e. ending with a semicolon. t;(tendswith(Rt codestring((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_get_statementscC s dj|S(Ns// {0}(tformat(Rttext((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _get_commentscC s\|jt}t|d|d|j|jdth}t|}|j|j|S(Nttypetvaluetattrs( RR R tevalft decimal_digRR Rt_print(RtnameRttype_tvartdecl((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_declare_number_consts - cC s |j|S(N(t indent_code(Rtlines((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _format_codesc s)|j\}fdt|DS(Nc3 s.|]$}tD]}||fVqqdS(N(R(t.0titj(tcols(s3lib/python2.7/site-packages/sympy/printing/ccode.pys s(tshapeR(Rtmattrows((Rs3lib/python2.7/site-packages/sympy/printing/ccode.pyt_traverse_matrix_indices scK stt|j||S(N(RRt _print_Mul(RRR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRscC sd|jkr|j|St|}|jt}|jdkrid|j|j|j|fS|jdkrd|j ||j |jfS|jt j dkr|j dkrd|j ||j |jfSd |j ||j |j|j |jfSdS( NtPowis1.0%s/%sg?s %ssqrt%s(%s)iRs %scbrt%s(%s)s%spow%s(%s, %s)(Rt_print_FunctionR#t_get_func_suffixR R4tuppert parenthesizetbaset_nsRRtOnetstandard(RRtPRECtsuffix((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Pows  # % cC sZ|j\}}|jrC|jrCdj|j||j|S|j|ddSdS(Ns (({}) % ({}))tknowntfmod(targsR&RRt_print_math_func(RRtnumtden((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Mod#s"cC sBt|jt|j}}|jt}d||||fS(Ns %d.0%s/%d.0%s(RbRtqt_get_literal_suffixR (RRRRR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_Rational*sc C s`t|jdtj}t|jdd}|j}|dksQt|tr|j}tj }t }|dks|dkrt t |j }|ddd}n|dkrt |j }nx)|D]!} ||f7}||| 9}qW|}ntgt||D]} | d| d^q|} d|j|jj|j| fS( NtoffsettstridesRiRiis%s[%s](tgetattrRRtZeroRtindicest isinstanceRRRttupletreversedRtranktsumtzipRtlabel( RRRRRtdimstshiftttempt traversalRR't flat_index((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_Indexed/s&        8cC s|j|jS(N(RR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_IdxGscC stt|j|S(N(RRt_print_NumberSymbol(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRJscC sdS(NtHUGE_VAL((RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_InfinityNscC sdS(Ns -HUGE_VAL((RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_NegativeInfinityQsc C s|jdjtkr%tdng}|jtrxt|jD]\}\}}|dkr|jd|j|nO|t |jdkr|tkr|jdn|jd|j||j|}|j||jdqJWd j |Sg|jd D].\}}d |j||j|f^q}d |j|jdj }d j ||d j dt |gSdS(NisAll 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.is if (%s) {iselse {selse if (%s) {t}s s((%s) ? ( %s ) s: ( %s )s: t t)( RtcondtTruet ValueErrorthasRt enumeratetappendRtlentjoinR( RRRRtetctcode0tecpairst last_line((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_PiecewiseTs$" %  ?cC sMddlm}||jd|jdf|jdtf}|j|S(Ni(t Piecewiseiii(tsympy.functionsRRRR(RRRt _piecewise((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_ITEus0cC sAdj|j|jtddt|j|j|jjdS(Ns{0}[{1}]tAtomtstricti(RRtparentR$RRRR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_MatrixElementzscC s@tt|j|}||jdkr8dj|S|SdS(NRs(*{0})(RRt _print_Symbolt _settingsR(RRR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyR%~s cC s@|j|j}|j|j}|j}dj|||S(Ns {0} {1} {2}(Rtlhstrhstrel_opR(RRtlhs_codetrhs_codetop((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_Relationals cC sddlm}ddlm}ddlm}|||jd|jd||jddfdtf}|j|S(Ni(R-(tNe(Rii( t(sympy.functions.elementary.trigonometricR-tsympy.core.relationalR.RRRRR(RRR-R.RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_sincs @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} }}ttargettstarttstoptsteptbody( RR2RtiterableR%RtNotImplementedErrorR6R(RRR2R3R4R5R6((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Fors   cC sdj|j|jdS(Ns((({0}) > 0) - (({0}) < 0))i(RRR(Rtfunc((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_signsc s;djkrj|Sfd|jS(NROc s]t|dkr#j|dSt|d}di|| d6||d6S(Niiis!((%(a)s > %(b)s) ? %(a)s : %(b)s)tatb(RR(Rthalf(tinner_print_maxR(s3lib/python2.7/site-packages/sympy/printing/ccode.pyR?s (RRR(RR((R?Rs3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Maxs c s;djkrj|Sfd|jS(NRQc s]t|dkr#j|dSt|d}di|| d6||d6S(Niiis!((%(a)s < %(b)s) ? %(a)s : %(b)s)R<R=(RR(RR>(tinner_print_minR(s3lib/python2.7/site-packages/sympy/printing/ccode.pyRAs (RRR(RR((RARs3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Mins 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 linesR)s t{t(s{ s( R Rs is s%s%s(RCRDs{ s( (R R(RRRt splitlinesRRtlstripRbtanytmapRt startswithRR( Rtcodet code_linesttabt inc_tokent dec_tokentlinetincreasetdecreasetprettytleveltn((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRs& "11 cC s|j|jj||S(N(RRR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRscC s|j|jj||S(N(RRR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRscC s=|jj||}|jj|d}|jj||S(NR)(RRR(RRtaliastdflt((s3lib/python2.7/site-packages/sympy/printing/ccode.pyRscC sc|jj|jj|t|jj|jj|t|j|jj||j S(N( RtupdateRRRRRRRR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt _print_Types""c C spddlm}|j}|j}|jtkr@tdnt|trdj dt |j krmdndd|j |jd t |j krd ndd ||j krd ndd |j |j}nst|tr0dj dt |j krdndd|j |jd |j |j}ntdt||dkrl|d|j |7}n|S(Ni(Rts$C does not support untyped variabless{vc}{t} *{pc} {r}{s}tvcsconst R)tttpcs consttrs restrict tss {vc}{t} {s}sUnknown type of var: %ss = %s(tsympy.codegen.cnodesRttvariableRRR RRR RRRRRtsymbolR R8R(RRRtRtvaltresult((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt_print_Declarations*     cC s|jjtt}|jj|jj|t|j|}t|j |j }d|krd|kr|d7}n|j d}|dj d|d<|dj dr|dcd7>> from sympy import ccode, symbols, Rational, sin, ceiling, Abs, Function >>> x, tau = symbols("x, tau") >>> expr = (2*tau)**Rational(7, 2) >>> ccode(expr) '8*M_SQRT2*pow(tau, 7.0/2.0)' >>> ccode(expr, math_macros={}) '8*sqrt(2)*pow(tau, 7.0/2.0)' >>> ccode(sin(x), assign_to="s") 's = sin(x);' >>> from sympy.codegen.ast import real, float80 >>> ccode(expr, type_aliases={real: float80}) '8*M_SQRT2l*powl(tau, 7.0L/2.0L)' 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') >>> ccode(func(Abs(x) + ceiling(x)), standard='C89', user_functions=custom_functions) 'f(fabs(x) + CEIL(x))' or if the C-function takes a subset of the original arguments: >>> ccode(2**x + 3**x, standard='C99', 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(ccode(expr, tau, standard='C89')) if (x > 0) { tau = x + 1; } else { tau = 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])) >>> ccode(e.rhs, assign_to=e.lhs, contract=False, standard='C89') '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(ccode(mat, A, standard='C89')) A[0] = pow(x, 2); if (x > 0) { A[1] = x + 1; } else { A[1] = x; } A[2] = sin(x); (tc_code_printersRtdoprint(Rt assign_toRR((s3lib/python2.7/site-packages/sympy/printing/ccode.pytccodescK stt||dS(s0Prints C representation of the given expression.N(tprintR(RR((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt print_ccodezsN(HRt __future__RRt functoolsRt itertoolsRt sympy.coreRtsympy.core.compatibilityRRtsympy.core.decoratorsRtsympy.codegen.astRR R R R R RRRRRRRRRRRRRRRRRRR tsympy.printing.codeprinterR!R"tsympy.printing.precedenceR#R$tsympy.sets.fancysetsR%RRRRRRRRRRtobjectRRRRgtktsetattrRRRRRR(((s3lib/python2.7/site-packages/sympy/printing/ccode.pyt s       %   P