ó ~9­\c@sqddlmZddlmZmZddlmZddlmZddl m Z ddl m Z m Z ddlmZddlmZdd lmZd d d d ddgZdefd„ƒYZdefd„ƒYZde fd„ƒYZdefd„ƒYZd„Zd„Zd„Zd„Zd„Zd„Zejj dƒdZ!ejj"e!ƒe_d S(!iÿÿÿÿ(t Derivative(tUndefinedFunctiont AppliedUndef(tSymbol(t init_printing(tsplit_super_sub(t LatexPrintert translate(t PrettyPrinter(t center_accent(t StrPrintertvprintt vsstrreprtvsprinttvpprinttvlatextinit_vprintingtVectorStrPrintercBs eZdZd„Zd„ZRS(s'String Printer for vector expressions. cCs¼ddlm}|j}ttg|jD]}||k^q)ƒƒtt|jdƒt ƒ@r¨t |jdj ƒ}x*t |jƒD]\}}||j 7}q‡W|Stƒj|ƒSdS(Niÿÿÿÿ(tdynamicsymbolsi(tsympy.physics.vector.functionsRt_ttbooltsumt variablest isinstancettypetargsRtstrtfunct enumeratet_strR tdoprint(tselfteRtttitoltv((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyt_print_Derivatives +cCsoddlm}|j}tt|ƒtƒrNtƒj|ƒjd|dƒS|j j d|j |j dƒS(Niÿÿÿÿ(Rs(%s)ts, ( RRRRRRR RtreplaceRt__name__t stringifyR(R R!RR"((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyt_print_Function s   (R)t __module__t__doc__R&R+(((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyRs tVectorStrReprPrintercBseZdZd„ZRS(s+String repr printer for vector expressions.cCs t|ƒS(N(trepr(R ts((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyt _print_str*s(R)R,R-R1(((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR.(stVectorLatexPrintercBs#eZdZdd„Zd„ZRS(s&Latex Printer for vector expressions. cCsddlm}|jj}|j}t|d|ƒrhtt|ƒtƒ rht |d|ƒ||ƒStt|ƒtƒrz|j |fkrzt |ƒ\}}}t |ƒ}g|D]} t | ƒ^q·}g|D]} t | ƒ^qÖ}t |ƒdkrddj|ƒ}nd}t |ƒdkrDddj|ƒ}nd}|rj|d|j|ƒ7}nd|||Sg|j D]} t|j| ƒƒ^q„} |jd } t}|jd oët | ƒd koë|j|j dƒ }d d ddg}||kro| dkrqo| dkr8d|d }qo| dkro|d }t}|dk rlt}qlqon|r‚d|}n)|dk r¡d||f}n d|}|r¾|d7}n |d7}|rë|dk rë|d|7}n|dj| ƒSdS(Niÿÿÿÿ(Rt_print_is^{%s}R's_{%s}s%stinv_trig_styletfold_func_bracketsitasintacostatantacott abbreviatedtfulltarctpowers\operatorname{%s}^{-1}s\operatorname{%s}^{%s}s\operatorname{%s}s\left(%s\right)t,(RRRR)RthasattrRRRtgetattrRRRtlentjoint_printRt _settingstFalset_needs_function_bracketstTruetNone(R texprtexpRRR"tnametsuperstsubstsuptsubtargRR4tinv_trig_power_casetcan_fold_bracketstinv_trig_table((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR+1s`  ' +             c Cs°ddlm}|jƒ}t|tƒs<d|j|ƒS|j}|j}|jt ƒ}|j }t g|D]}|j |hkrpt ^qpƒ }t g|D]}||k^q¢ƒ } |sÊ| rÚtƒj|ƒSt|ƒ} |j|ƒ} | jddƒ} | d} | dkr*d| } n[| dkrCd | } nB| d kr\d | } n)| d krud | } ntƒj|ƒSt| ƒdk r¬| d| d7} n| S(Niÿÿÿÿ(Rs\left(%s\right)t_iis\dot{%s}is \ddot{%s}is \dddot{%s}is \ddddot{%s}(RRtdoitRRRRRItatomsRRtallt free_symbolsRGRRAR+tsplit( R tder_exprRR"RItredtsymsR#ttest1ttest2tdotstbaset base_split((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR&{s8    2&           N(R)R,R-RHR+R&(((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR2.s JtVectorPrettyPrintercBs eZdZd„Zd„ZRS(s)Pretty Printer for vectorialexpressions. c Csíddlm}|j}d}tt|jƒƒ}xSt|ƒdkr‰|d|krp|jƒ|d7}q7tt |ƒj |ƒSq7Wt t |j ƒtƒo´|j j|fksÍtt |ƒj |ƒS|j|j ƒ}t|jƒdkr tt |ƒj |ƒS|dkr,tt |ƒj |ƒSidd6dd6dd 6d d 6d d 6}|j}|js¬d} x!td|ƒD]} | d7} q}W|ddc| d7        "cCsddlm}|j}|j}|j}|j}|jt|ƒƒ}t|t ƒod||fks}t t |ƒj |ƒS|S(Niÿÿÿÿ(R( RRRRRR)t _print_SymbolRRRRiRbR+(R R!RR"RRt func_nameRo((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR+Ós    (R)R,R-R&R+(((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyRb s 0cKs@t||}ddlm}|dkr<||_|GHndS(sBFunction for printing of expressions generated in the sympy.physics vector package. Extends SymPy's StrPrinter, takes the same setting accepted by SymPy's `sstr()`, and is equivalent to `print(sstr(foo))`. Parameters ========== expr : valid SymPy object SymPy expression to print. settings : args Same as the settings accepted by SymPy's sstr(). Examples ======== >>> from sympy.physics.vector import vprint, dynamicsymbols >>> u1 = dynamicsymbols('u1') >>> print(u1) u1(t) >>> vprint(u1) u1 iÿÿÿÿ(tbuiltinsRHN(R tsympy.core.compatibilityRtRT(RItsettingstoutstrRt((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR ãs   cKst|ƒ}|j|ƒS(s Function for displaying expression representation's with vector printing enabled. Parameters ========== expr : valid SymPy object SymPy expression to print. settings : args Same as the settings accepted by SymPy's sstrrepr(). (R.R(RIRvtp((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR s cKst|ƒ}|j|ƒS(sFunction for displaying expressions generated in the sympy.physics vector package. Returns the output of vprint() as a string. Parameters ========== expr : valid SymPy object SymPy expression to print settings : args Same as the settings accepted by SymPy's sstr(). Examples ======== >>> from sympy.physics.vector import vsprint, dynamicsymbols >>> u1, u2 = dynamicsymbols('u1 u2') >>> u2d = dynamicsymbols('u2', level=1) >>> print("%s = %s" % (u1, u2 + u2d)) u1(t) = u2(t) + Derivative(u2(t), t) >>> print("%s = %s" % (vsprint(u1), vsprint(u2 + u2d))) u1 = u2 + u2' (RR(RIRvtstring_printer((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR s cKsXt|ƒ}|jd}ddlm}||ƒ}z|j|ƒSWd||ƒXdS(sòFunction for pretty printing of expressions generated in the sympy.physics vector package. Mainly used for expressions not inside a vector; the output of running scripts and generating equations of motion. Takes the same options as SymPy's pretty_print(); see that function for more information. Parameters ========== expr : valid SymPy object SymPy expression to pretty print settings : args Same as those accepted by SymPy's pretty_print. t use_unicodeiÿÿÿÿ(tpretty_use_unicodeN(RbRDt&sympy.printing.pretty.pretty_symbologyR{R(RIRvtppRzR{tuflag((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR6s   cKst|ƒ}|j|ƒS(sÅFunction for printing latex representation of sympy.physics.vector objects. For latex representation of Vectors, Dyadics, and dynamicsymbols. Takes the same options as SymPy's latex(); see that function for more information; Parameters ========== expr : valid SymPy object SymPy expression to represent in LaTeX form settings : args Same as latex() Examples ======== >>> from sympy.physics.vector import vlatex, ReferenceFrame, dynamicsymbols >>> N = ReferenceFrame('N') >>> q1, q2 = dynamicsymbols('q1 q2') >>> q1d, q2d = dynamicsymbols('q1 q2', 1) >>> q1dd, q2dd = dynamicsymbols('q1 q2', 2) >>> vlatex(N.x + N.y) '\\mathbf{\\hat{n}_x} + \\mathbf{\\hat{n}_y}' >>> vlatex(q1 + q2) 'q_{1} + q_{2}' >>> vlatex(q1d) '\\dot{q}_{1}' >>> vlatex(q1 * q2d) 'q_{1} \\dot{q}_{2}' >>> vlatex(q1dd * q1 / q1d) '\\frac{q_{1} \\ddot{q}_{1}}{\\dot{q}_{1}}' (R2R(RIRvt latex_printer((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyRXs# cKs,t|d>> from sympy import Function, symbols >>> from sympy.physics.vector import init_vprinting >>> t, x = symbols('t, x') >>> omega = Function('omega') >>> omega(x).diff() Derivative(omega(x), x) >>> omega(t).diff() Derivative(omega(t), t) Now use the string printer: >>> init_vprinting(pretty_print=False) >>> omega(x).diff() Derivative(omega(x), x) >>> omega(t).diff() omega' t str_printertpretty_printerRN(R RRR(tkwargs((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyR€s!   sExamples ========iN(#tsympyRtsympy.core.functionRRtsympy.core.symbolRtsympy.interactive.printingRtsympy.printing.conventionsRtsympy.printing.latexRRtsympy.printing.pretty.prettyRR|R tsympy.printing.strR t__all__RR.R2RbR R R RRRR-RYtparamstformat(((s<lib/python2.7/site-packages/sympy/physics/vector/printing.pyts, rC #   " ( &