o 8Va.@s ddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZgd ZGd d d eZGd ddeZGddde ZGddde ZddZddZddZddZddZddZejddZ ej!e e_d S)!) Derivative)UndefinedFunction AppliedUndef)Symbol) init_printing) LatexPrinter) PrettyPrinter) center_accent) StrPrinter) PRECEDENCE)vprint vsstrreprvsprintvpprintvlatexinit_vprintingc@s eZdZdZddZddZdS)VectorStrPrinterz'String Printer for vector expressions. csddlm}|jttfdd|jDtt|jdt @r:t |jdj }t |jD] \}}||j 7}q.|St|S)Nrdynamicsymbolscsg|]}|kqSr.0itr?/usr/lib/python3/dist-packages/sympy/physics/vector/printing.py z6VectorStrPrinter._print_Derivative..)sympy.physics.vector.functionsr_tboolsum variables isinstancetypeargsrstrfunc enumerate_strr doprint)selferZolrvrrr_print_Derivatives   z"VectorStrPrinter._print_DerivativecCsRddlm}|j}tt|trt|d|dS|j j d| |j dS)Nrrz(%s)z, ) rrrr#r$rr r*replacer'__name__Z stringifyr%)r+r,rrrrr_print_Functions z VectorStrPrinter._print_FunctionN)r1 __module__ __qualname____doc__r.r2rrrrrs rc@seZdZdZddZdS)VectorStrReprPrinterz+String repr printer for vector expressions.cCst|SN)repr)r+srrr _print_str(szVectorStrReprPrinter._print_strN)r1r3r4r5r:rrrrr6&s r6cs.eZdZdZdfdd ZfddZZS)VectorLatexPrinterz&Latex Printer for vector expressions. Ncsddlm}|jj}|j}t|d|r%tt|ts%t |d|||Stt|trS|j |fkrSt |}|durM| |t d}||}d||fSt|St||S)NrrZ_print_ZPowz%s^{%s})rrr'r1rhasattrr#r$rgetattrr%rZ parenthesizer Zparenthesize_supersuperZ_printr2)r+exprZexprr'rbase __class__rrr2/s     z"VectorLatexPrinter._print_Functionc s&ddlm}|}t|tsd||S|j|j}|t }|j }t fdd|D }t fdd|D }|s@|rFt |St|}||} | dd} | d} |dkrbd | } n!|d krkd | } n|d krtd | } n|dkr}d| } nt |St| dkr| d| d7} | S)Nrrz\left(%s\right)csg|] }|jhkrdqS)T)Z free_symbolsrrrrrQsz8VectorLatexPrinter._print_Derivative..csg|]}|kqSrrrrrrrRr_z\dot{%s}z \ddot{%s}z \dddot{%s}z \ddddot{%s})rrZdoitr#rr*rr?Zatomsrr"allr>r.lenr2split) r+Zder_exprrr?ZredsymsZtest1Ztest2dotsr@Z base_splitrArrr.Es8            z$VectorLatexPrinter._print_Derivativer7)r1r3r4r5r2r. __classcell__rrrArr;,sr;cs,eZdZdZfddZfddZZS)VectorPrettyPrinterz)Pretty Printer for vectorialexpressions. c s>ddlm}|j}d}tt|j}t|dkr3|d|kr'||d7}nt |St|dkst t |j t rB|j j|fksHt |S||j }t|jdkr[t |S|dkret |Sdddd d d }|j}|jsd} td|D]} | d 7} qz|d d| d7<|St|d d||g|d <|S)NrrrDr/u̇üu⃛u⃜)rrDrErFrG'picturez(t))rrrlistreversedr"rIpopr>r.r#r$r?rr%r2rR__dict__Z _use_unicoderanger ) r+ZderivrrZdot_irKpformrLdZ apostrophesrrArrr.msD            z%VectorPrettyPrinter._print_DerivativecsVddlm}|j}|j}|j}|j}|t|}t|t r#||fks)t |S|S)Nrr) rrrr'r%r1Z _print_Symbolrr#rr>r2)r+r,rrr'r%Z func_namerXrArrr2s  z#VectorPrettyPrinter._print_Function)r1r3r4r5r.r2rMrrrArrNjs /rNcKs6t|fi|}ddl}|dkr||_t|dSdS)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)rbuiltinsrCprint)r?settingsZoutstrr[rrrr s  r cKt|}||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(). )r6r*)r?r]prrrr s r cKr^)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*)r?r]Zstring_printerrrrrs rcKsFt|}|jd}ddlm}||}z ||W||S||w)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)rNZ _settings&sympy.printing.pretty.pretty_symbologyrar*)r?r]ppr`raZuflagrrrrs   rcKr^)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}}' )r;r*)r?r] latex_printerrrrr!s# rcKs*t|d<t|d<t|d<tdi|dS)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' Z str_printerZpretty_printerrdNr)r rrr)kwargsrrrrIs!rzExamples ========N)"ZsympyrZsympy.core.functionrrZsympy.core.symbolrZsympy.interactive.printingrZsympy.printing.latexrZsympy.printing.pretty.prettyrrbr Zsympy.printing.strr Zsympy.printing.precedencer __all__rr6r;rNr r rrrrr5rJparamsformatrrrrs,        >B#"(&