MZd.tddlmZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZgd ZGd d eZGd deZGdde ZGdde ZdZdZdZdZdZdZej:j=ddZej:jAee_y)) Derivative)UndefinedFunction AppliedUndef)Symbol) init_printing) LatexPrinter) PrettyPrinter) center_accent) StrPrinter) PRECEDENCE)vprint vsstrreprvsprintvpprintvlatexinit_vprintingceZdZdZdZdZy)VectorStrPrinterz'String Printer for vector expressions. c ddlm}|j}tt |j Dcgc]}||k( c}t t|jdtzrPt|jdj}t|j D]\}}||jz }|Stj|Scc}wNrdynamicsymbols)sympy.physics.vector.functionsr_tboolsum variables isinstancetypeargsrstrfunc enumerate_strr doprint)selfertiolvs ?/usr/lib/python3/dist-packages/sympy/physics/vector/printing.py_print_Derivativez"VectorStrPrinter._print_DerivativesA    akk2a1f23 44q ?,=> ?QVVAY^^$B!!++. *1n))) *I<''* *3s C cddlm}|j}tt |t r,t j|jd|zdS|jjd|j|jdzzS)Nrrz(%s)z, ) rrrrrrr r%replacer"__name__ stringifyr )r&r'rr(s r,_print_Functionz VectorStrPrinter._print_FunctionsjA    d1g0 1<''*226A:rB Bvv$..*F!FFFN)r1 __module__ __qualname____doc__r-r3r4r,rrs1 +Gr4rceZdZdZdZy)VectorStrReprPrinterz+String repr printer for vector expressions.ct|SN)repr)r&ss r, _print_strzVectorStrReprPrinter._print_str(s Awr4N)r1r5r6r7r?r8r4r,r:r:&s 5r4r:c.eZdZdZdfd ZfdZxZS)VectorLatexPrinterz&Latex Printer for vector expressions. cddlm}|jj}|j}t |d|zr/t t|tst|d|z||St t|tr^|j|fk(rNt|}|2|j|td}|j|}|d|dSt|A|St|E||S)Nrr_print_Powz^{})rrr"r1rhasattrrrrgetattrr r parenthesizer parenthesize_supersuper_printr3)r&exprexprr"r(base __class__s r,r3z"VectorLatexPrinter._print_Function/sAyy!!    D)d* + tDz#4 5274T!124= = T $5 6DII!z7VectorLatexPrinter._print_Derivative..QsCQ^^s-BCs  c3(K|] }|k( ywr<r8rSs r,rUz7VectorLatexPrinter._print_Derivative..Rs-1Q-s_z\dot{%s}z \ddot{%s}z \dddot{%s}z \ddddot{%s})rrdoitrrr%rrLatomsrrallrJr-lenr3split) r&der_exprrrLredsymstest1test2dotsrN base_splitr(rOs @r,r-z$VectorLatexPrinter._print_DerivativeEsEA==?(J/% X(>> >   }}jj&!!CcCCC---- E7,X6 64y##D)ZZQ' !} 19%D QY$&D QY 4'D QY!D(D7,X6 6 z?a  C*Q-' 'D r4r<)r1r5r6r7r3r- __classcell__rOs@r,rArA,s06,""r4rAc,eZdZdZfdZfdZxZS)VectorPrettyPrinterz)Pretty Printer for vectorialexpressions. cddlm}|j}d}tt |j }t |dkDr<|d|k(r|j|dz }nt |%|St |dkDr?QD(7,U3 3((4E u}}  !7,U3 3 A:7,U3 3,,32 4 NN  K1e_ #s"  # iLO{U2 2O *!I,q/4;GHAiL r4cddlm}|j}|j}|j}|j }|j t|}t|tr||fk(st|-|S|Sr) rrrr"r r1 _print_SymbolrrrrJr3) r&r'rr(r"r func_nameryrOs r,r3z#VectorPrettyPrinter._print_FunctionskA   vvvvMM ""6)#454!23!7*1- - r4)r1r5r6r7r-r3rhris@r,rkrkjs3.`  r4rkc Tt|fi|}ddl}|dk7r||_t|yy)aJFunction for printing of expressions generated in the sympy.physics vector package. Extends SymPy's StrPrinter, takes the same setting accepted by SymPy's :func:`~.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 rNNone)rbuiltinsrWprint)rLsettingsoutstrrs r,r r s16T &X &F&  f  r4c :t|}|j|S)a 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%)rLrps r,rrs X&A 99T?r4c :t|}|j|S)aFunction 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%)rLrstring_printers r,rrs6&h/N  ! !$ ''r4c t|}|jd}ddlm}||} |j |||S#||wxYw)aFunction 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 :func:`~.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. use_unicoder)pretty_use_unicode)rk _settings&sympy.printing.pretty.pretty_symbologyrr%)rLrpprruflags r,rrsN& X &B ,,}-KI { +E"zz$5!5!s A A c :t|}|j|S)aFunction for printing latex representation of sympy.physics.vector objects. For latex representation of Vectors, Dyadics, and dynamicsymbols. Takes the same options as SymPy's :func:`~.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}}' )rAr%)rLr latex_printers r,rr"s H'x0M   &&r4c Pt|d<t|d<t|d<tdi|y)aInitializes time derivative printing for all SymPy objects, i.e. any functions of time will be displayed in a more compact notation. The main benefit of this is for printing of time derivatives; instead of displaying as ``Derivative(f(t),t)``, it will display ``f'``. This is only actually needed for when derivatives are present and are not in a physics.vector.Vector or physics.vector.Dyadic object. This function is a light wrapper to :func:`~.init_printing`. Any keyword arguments for it are valid here. {0} Examples ======== >>> from sympy import Function, symbols >>> 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: >>> from sympy.physics.vector import init_vprinting >>> init_vprinting(pretty_print=False) >>> omega(x).diff() Derivative(omega(x), x) >>> omega(t).diff() omega' str_printerpretty_printerrNr8)rrrr)kwargss r,rrKs/B&F=&F $F?Fr4zExamples ========N)!sympy.core.functionrrrsympy.core.symbolrsympy.interactive.printingrsympy.printing.latexrsympy.printing.pretty.prettyr rr sympy.printing.strr sympy.printing.precedencer __all__rr:rArkr rrrrrr7r`paramsformatr8r4r,rs*?$4-6@)0 GzG.+ ;;|@-@F F"(>"D&'R$N    $ $%= >q A'//66v>r4