U -eRL@s\ddlmZmZmZddlmZddlmZddl m Z dgZ GdddeeZ ddZ d S) )sympifyAddImmutableMatrix) EvalfMixin) Printable) prec_to_dpsDyadicc@seZdZdZdZddZeddZddZd d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zd!d"Zd#d$ZeZe Zd8d&d'Zd9d(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Ze Z eZ!d4d5Z"d6d7Z#d%S):rayA Dyadic object. See: https://en.wikipedia.org/wiki/Dyadic_tensor Kane, T., Levinson, D. Dynamics Theory and Applications. 1985 McGraw-Hill A more powerful way to represent a rigid body's inertia. While it is more complex, by choosing Dyadic components to be in body fixed basis vectors, the resulting matrix is equivalent to the inertia tensor. FcCsbg|_|dkrg}t|dkrd}t|jD]\}}t|ddt|j|dkr,t|ddt|j|dkr,|j|d|dd|dd|ddf|j|<||dd}qq,|dkr|j|d||dqd}|t|jkr^|j|ddk|j|ddkB|j|ddkBrT|j|j||d8}|d7}qdS)a2 Just like Vector's init, you should not call this unless creating a zero dyadic. zd = Dyadic(0) Stores a Dyadic as a list of lists; the inner list has the measure number and the two unit vectors; the outerlist holds each unique unit vector pair. rN)argslen enumeratestrremoveappend)selfinlistaddedivr\/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/sympy/physics/vector/dyadic.py__init__s6  "  "zDyadic.__init__cCstS)zReturns the class Dyadic. )rrrrrfunc@sz Dyadic.funccCst|}t|j|jS)zThe add operator for Dyadic. ) _check_dyadicrr rotherrrr__add__EszDyadic.__add__c Csddlm}m}t|trt|}td}t|jD]P\}}t|jD]<\}}||d|d|d|d@|d|dB7}qFq4nD||}|d}t|jD](\}}||d|d|d|@7}q|S)aThe inner product operator for a Dyadic and a Dyadic or Vector. Parameters ========== other : Dyadic or Vector The other Dyadic or Vector to take the inner product with Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer >>> N = ReferenceFrame('N') >>> D1 = outer(N.x, N.y) >>> D2 = outer(N.y, N.y) >>> D1.dot(D2) (N.x|N.y) >>> D1.dot(N.y) N.x rVector _check_vectorr r )sympy.physics.vector.vectorr r! isinstancerrr r ) rrr r!olrri2Zv2rrr__and__Js :"zDyadic.__and__cCs|d|S)z0Divides the Dyadic by a sympifyable expression. r )__mul__rrrr __truediv__nszDyadic.__truediv__cCs\|dkrtd}t|}|jgkr0|jgkr0dS|jgksD|jgkrHdSt|jt|jkS)z[Tests for equality. Is currently weak; needs stronger comparison testing rTF)rrr setrrrr__eq__rsz Dyadic.__eq__cCsVt|j}t|}t|D]2\}}|||d||d||df||<qt|S)aMultiplies the Dyadic by a sympifyable expression. Parameters ========== other : Sympafiable The scalar to multiply this Dyadic with Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer >>> N = ReferenceFrame('N') >>> d = outer(N.x, N.x) >>> 5 * d 5*(N.x|N.x) rr r )listr rr r)rrZnewlistrrrrrr's  zDyadic.__mul__cCs ||k S)Nrrrrr__ne__sz Dyadic.__ne__cCs|dSNrrrrr__neg__szDyadic.__neg__c Cs|j}t|dkrtdSg}t|D]$\}}||ddkrt|d|||dd|||dq&||ddkr|d|||dd|||dq&||ddkr&|||d}t||dtrd|}|d r|dd}d}nd}||||||dd|||dq&d |}|drr|d d}n|d r|dd}|S) Nrr  + z\otimes r r. - (%s)- ) r r rr r_printr#r startswithjoin rprinterarr$rrarg_str str_startoutstrrrr_latexsH       z Dyadic._latexcs|Gfddd}|S)NcseZdZdZfddZdS)zDyadic._pretty..Fakerc sj}}t|dkrtdSjr(dnd}g}t|D].\}}||ddkr|d|||d||||dgq8||ddkr|d|||d||||dgq8||ddkr8t||dtr| ||d d} n|||d} | d r2| dd} d} nd} || | d |||d||||dgq8d |} | dr| d d} n| d r| dd} | S) Nru⊗|r r0r r.r1r3r6r4r5) r r rZ _use_unicoder extendZdoprintr#rr7parensr8r9) rr kwargsr<Zmppbarr$rrr=r>r?er;rrrendersT         z#Dyadic._pretty..Fake.renderN)__name__ __module__ __qualname__ZbaselinerHrrFrrFakesrLr)rr;rLrrFr_prettys1zDyadic._prettycCsXddlm}m}||}|d}t|jD](\}}||d|d|d|@7}q*|S)aThe inner product operator for a Vector or Dyadic, and a Dyadic This is for: Vector dot Dyadic Parameters ========== other : Vector The vector we are dotting with Examples ======== >>> from sympy.physics.vector import ReferenceFrame, dot, outer >>> N = ReferenceFrame('N') >>> d = outer(N.x, N.x) >>> dot(N.x, d) N.x rrr r )r"r r!r r )rrr r!r$rrrrr__rand__s "zDyadic.__rand__cCs d||Sr-rrrrr__rsub__szDyadic.__rsub__cCsTddlm}||}td}t|jD](\}}||d||dA|dB7}q&|S)aFor a cross product in the form: Vector x Dyadic Parameters ========== other : Vector The Vector that we are crossing this Dyadic with Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer, cross >>> N = ReferenceFrame('N') >>> d = outer(N.x, N.x) >>> cross(N.y, d) - (N.z|N.x) rr!r r r"r!rr r rrr!r$rrrrr__rxor__s  "zDyadic.__rxor__c Cs|j}t|dkr|dSg}t|D]8\}}||ddkrz|d|||dd|||ddq(||ddkr|d|||dd|||ddq(||ddkr(|||d}t||dtrd |}|dd kr |dd }d }nd }|||d|||dd|||ddq(d|}|d r|dd }n|dr|dd }|S)zPrinting method. rr z + (rAr )r.z - (r2r3Nr1r0z*(r4r5r6) r r r7r rr#rr9r8r:rrr _sympystr8sP       zDyadic._sympystrcCs||dS)zThe subtraction operator. r.)rrrrr__sub__\szDyadic.__sub__cCsTddlm}||}td}t|jD](\}}||d|d|d|AB7}q&|S)aFor a cross product in the form: Dyadic x Vector. Parameters ========== other : Vector The Vector that we are crossing this Dyadic with Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer, cross >>> N = ReferenceFrame('N') >>> d = outer(N.x, N.x) >>> cross(d, N.y) (N.x|N.z) rrPr r rQrRrrr__xor__`s  "zDyadic.__xor__NcCsddlm}||||S)aExpresses this Dyadic in alternate frame(s) The first frame is the list side expression, the second frame is the right side; if Dyadic is in form A.x|B.y, you can express it in two different frames. If no second frame is given, the Dyadic is expressed in only one frame. Calls the global express function Parameters ========== frame1 : ReferenceFrame The frame to express the left side of the Dyadic in frame2 : ReferenceFrame If provided, the frame to express the right side of the Dyadic in Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer, dynamicsymbols >>> from sympy.physics.vector import init_vprinting >>> init_vprinting(pretty_print=False) >>> N = ReferenceFrame('N') >>> q = dynamicsymbols('q') >>> B = N.orientnew('B', 'Axis', [q, N.z]) >>> d = outer(N.x, N.x) >>> d.express(B, N) cos(q)*(B.x|N.x) - sin(q)*(B.y|N.x) r)express)sympy.physics.vector.functionsrX)rZframe1Zframe2rXrrrrX~s zDyadic.expresscs,dkr |tfdd|DddS)aReturns the matrix form of the dyadic with respect to one or two reference frames. Parameters ---------- reference_frame : ReferenceFrame The reference frame that the rows and columns of the matrix correspond to. If a second reference frame is provided, this only corresponds to the rows of the matrix. second_reference_frame : ReferenceFrame, optional, default=None The reference frame that the columns of the matrix correspond to. Returns ------- matrix : ImmutableMatrix, shape(3,3) The matrix that gives the 2D tensor form. Examples ======== >>> from sympy import symbols >>> from sympy.physics.vector import ReferenceFrame, Vector >>> Vector.simp = True >>> from sympy.physics.mechanics import inertia >>> Ixx, Iyy, Izz, Ixy, Iyz, Ixz = symbols('Ixx, Iyy, Izz, Ixy, Iyz, Ixz') >>> N = ReferenceFrame('N') >>> inertia_dyadic = inertia(N, Ixx, Iyy, Izz, Ixy, Iyz, Ixz) >>> inertia_dyadic.to_matrix(N) Matrix([ [Ixx, Ixy, Ixz], [Ixy, Iyy, Iyz], [Ixz, Iyz, Izz]]) >>> beta = symbols('beta') >>> A = N.orientnew('A', 'Axis', (beta, N.x)) >>> inertia_dyadic.to_matrix(A) Matrix([ [ Ixx, Ixy*cos(beta) + Ixz*sin(beta), -Ixy*sin(beta) + Ixz*cos(beta)], [ Ixy*cos(beta) + Ixz*sin(beta), Iyy*cos(2*beta)/2 + Iyy/2 + Iyz*sin(2*beta) - Izz*cos(2*beta)/2 + Izz/2, -Iyy*sin(2*beta)/2 + Iyz*cos(2*beta) + Izz*sin(2*beta)/2], [-Ixy*sin(beta) + Ixz*cos(beta), -Iyy*sin(2*beta)/2 + Iyz*cos(2*beta) + Izz*sin(2*beta)/2, -Iyy*cos(2*beta)/2 + Iyy/2 - Iyz*sin(2*beta) + Izz*cos(2*beta)/2 + Izz/2]]) Ncs&g|]}D]}||q qSr)dot).0rjsecond_reference_framerrr sz$Dyadic.to_matrix..r5)MatrixZreshape)rZreference_framer^rr]r to_matrixs ,zDyadic.to_matrixc stfdd|jDtdS)z(Calls .doit() on each term in the Dyadiccs0g|](}t|djf|d|dfgqSrr r )rdoitr[rhintsrrr_szDyadic.doit..rsumr r)rrfrrerrcs  z Dyadic.doitcCsddlm}|||S)aTake the time derivative of this Dyadic in a frame. This function calls the global time_derivative method Parameters ========== frame : ReferenceFrame The frame to take the time derivative in Examples ======== >>> from sympy.physics.vector import ReferenceFrame, outer, dynamicsymbols >>> from sympy.physics.vector import init_vprinting >>> init_vprinting(pretty_print=False) >>> N = ReferenceFrame('N') >>> q = dynamicsymbols('q') >>> B = N.orientnew('B', 'Axis', [q, N.z]) >>> d = outer(N.x, N.x) >>> d.dt(B) - q'*(N.y|N.x) - q'*(N.x|N.y) r)time_derivative)rYri)rframerirrrdts z Dyadic.dtcCs<td}|jD](}|t|d|d|dfg7}q|S)zReturns a simplified Dyadic.rr r )rr simplify)routrrrrrls &zDyadic.simplifycs tfdd|jDtdS)a5Substitution on the Dyadic. Examples ======== >>> from sympy.physics.vector import ReferenceFrame >>> from sympy import Symbol >>> N = ReferenceFrame('N') >>> s = Symbol('s') >>> a = s*(N.x|N.x) >>> a.subs({s: 2}) 2*(N.x|N.x) cs0g|](}t|dj|d|dfgqSrb)rsubsrdr rDrrr_ szDyadic.subs..rrg)rr rDrrorrns z Dyadic.subscCsBt|stdtd}|jD]\}}}|||||B7}q|S)z/Apply a function to each component of a Dyadic.z`f` must be callable.r)callable TypeErrorrr )rfrmabcrrr applyfuncs zDyadic.applyfunccCsT|js |Sg}t|}|jD].}t|}|dj|d|d<|t|qt|S)Nr)n)r rr+Zevalfrtupler)rprecnew_argsZdpsr new_inlistrrr _eval_evalfs zDyadic._eval_evalfcCs@g}|jD],}t|}|d||d<|t|q t|S)a Replace occurrences of objects within the measure numbers of the Dyadic. Parameters ========== rule : dict-like Expresses a replacement rule. Returns ======= Dyadic Result of the replacement. Examples ======== >>> from sympy import symbols, pi >>> from sympy.physics.vector import ReferenceFrame, outer >>> N = ReferenceFrame('N') >>> D = outer(N.x, N.x) >>> x, y, z = symbols('x y z') >>> ((1 + x*y) * D).xreplace({x: pi}) (pi*y + 1)*(N.x|N.x) >>> ((1 + x*y) * D).xreplace({x: pi, y: 2}) (1 + 2*pi)*(N.x|N.x) Replacements occur only if an entire node in the expression tree is matched: >>> ((x*y + z) * D).xreplace({x*y: pi}) (z + pi)*(N.x|N.x) >>> ((x*y*z) * D).xreplace({x*y: pi}) x*y*z*(N.x|N.x) r)r r+xreplacerrxr)rrulerzrr{rrrr}&s ( zDyadic.xreplace)N)N)$rIrJrK__doc__Z is_numberrpropertyrrr&r(r*r'r,r/r@rMrNrOrSrUrVrW__radd____rmul__rXrarcrkrlrnrvrZcrossr|r}rrrrr sB & $$6$ # 2  cCst|tstd|S)NzA Dyadic must be supplied)r#rrq)rrrrrVs rN)Zsympy.core.backendrrrr`Zsympy.core.evalfrZsympy.printing.defaultsrZmpmath.libmp.libmpfr__all__rrrrrrs   O