U Zʹj.@sddlmZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd d d d ddgZGdddeZGdddeZGddde ZGddde Zdd Zdd Zdd Zdd ZddZddZejddZej ee_dS) ) Derivative)UndefinedFunction AppliedUndef)Symbol) init_printing) LatexPrinter) PrettyPrinter) center_accent) StrPrinter) PRECEDENCEvprint vsstrreprvsprintvpprintvlatexinit_vprintingc@s eZdZdZddZddZdS)VectorStrPrinterz'String Printer for vector expressions. csddlm}|jttfdd|jDtt|jdt @rtt |jdj }t |jD]\}}||j 7}q\|St|SdS)Nrdynamicsymbolsc3s|]}|kVqdSN.0itr\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/physics/vector/printing.py sz5VectorStrPrinter._print_Derivative..)sympy.physics.vector.functionsr_tboolsum variables isinstancetypeargsrstrfunc enumerate_strr doprint)selferolrvrrr_print_Derivatives  z"VectorStrPrinter._print_DerivativecCsRddlm}|j}tt|tr8t|d|dS|j j d| |j dS)Nrrz(%s)z, ) rrrr#r$rr r*replacer'__name__ stringifyr%)r+r,rrrrr_print_Functions  z VectorStrPrinter._print_FunctionN)r2 __module__ __qualname____doc__r/r4rrrrrs rc@seZdZdZddZdS)VectorStrReprPrinterz+String repr printer for vector expressions.cCst|Sr)repr)r+srrr _print_str(szVectorStrReprPrinter._print_strN)r2r5r6r7r;rrrrr8&sr8cs.eZdZdZdfdd ZfddZZS)VectorLatexPrinterz&Latex Printer for vector expressions. Ncsddlm}|jj}|j}t|d|rJtt|tsJt |d|||Stt|tr|j |fkrt |}|dk r| |t d}||}d||fSt|Snt||SdS)NrrZ_print_Powz%s^{%s})rrr'r2rhasattrr#r$rgetattrr%rZ parenthesizer Zparenthesize_supersuperZ_printr4)r+exprexprr'rbase __class__rrr4/s    z"VectorLatexPrinter._print_Functionc s(ddlm}|}t|ts,d||S|j|j}|t }|j }t fdd|D }t fdd|D }|s|rt |St|}||} | dd} | d} |dkrd | } nB|d krd | } n0|d krd | } n|dkrd| } n t |St| dkr$| d| d7} | S)Nrrz\left(%s\right)c3s|]}|jhkrdVqdS)TN)Z free_symbolsrrrrrQs z7VectorLatexPrinter._print_Derivative..c3s|]}|kVqdSrrrrrrrRs_z\dot{%s}z \ddot{%s}z \dddot{%s}z \ddddot{%s})rrZdoitr#rr*rrAZatomsrr"allr@r/lenr4split) r+Zder_exprrrAredsymsZtest1Ztest2dotsrCZ base_splitrDrrr/Es8           z$VectorLatexPrinter._print_Derivative)N)r2r5r6r7r4r/ __classcell__rrrDrr<,sr<cs,eZdZdZfddZfddZZS)VectorPrettyPrinterz)Pretty Printer for vectorialexpressions. c s4ddlm}|j}d}tt|j}t|dkr\|d|krN||d7}q$t |Sq$t t |j t rz|j j|fkst |S||j }t|jdkrt |S|dkrt |Sdddd d d }|j}|jsd} td|D] } | d 7} q|d d| d7<nt|d d||g|d <|S)NrrrGr0u̇üu⃛u⃜)rrGrHrIrJ'picturez(t))rrrlistreversedr"rLpopr@r/r#r$rArr%r4rV__dict__Z _use_unicoderanger ) r+ZderivrrZdot_irOpformrPdZ apostrophesrrDrrr/ms@          z%VectorPrettyPrinter._print_DerivativecsVddlm}|j}|j}|j}|j}|t|}t|t rF||fksRt |S|S)Nrr) rrrr'r%r2Z _print_Symbolrr#rr@r4)r+r,rrr'r% func_namer\rDrrr4s  z#VectorPrettyPrinter._print_Function)r2r5r6r7r/r4rQrrrDrrRjs 0rRcKs.t|f|}ddl}|dkr*||_t|dS)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)rbuiltinsrFprint)rAsettingsZoutstrr`rrrr s  cKst|}||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(). )r8r*)rArbprrrr s cKst|}||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*)rArbZstring_printerrrrrscKsDt|}|jd}ddlm}||}z||WS||XdS)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_unicodeN)rRZ _settings&sympy.printing.pretty.pretty_symbologyrer*)rArbpprdreZuflagrrrrs  cKst|}||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}}' )r<r*)rArb latex_printerrrrr"s$cKs&t|d<t|d<t|d<tf|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_printerrhN)r rrr)kwargsrrrrKs!zExamples ========N)!Zsympy.core.functionrrrZsympy.core.symbolrZsympy.interactive.printingrZsympy.printing.latexrZsympy.printing.pretty.prettyrrfr Zsympy.printing.strr Zsympy.printing.precedencer __all__rr8r<rRr r rrrrr7rMparamsformatrrrrs.         >C#")'