o 8VaT@sddlmZmZddlmZmZmZmZddlmZmZm Z m Z m Z m Z ddlm Z ddlmZddlmZddlmZddlmZdd lmZGd d d eZd S) )SRational)reim conjugatesign)sqrtsincosacosexpln)trigsimp integrate)Matrixsympify) prec_to_dps)Exprc@sXeZdZdZdZdZdMddZedd Zed d Z ed d Z eddZ eddZ e ddZe ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zed0d1Zd2d3Zd4d5Z d6d7Z!d8d9Z"d:d;Z#dd?Z%d@dAZ&dBdCZ'dDdEZ(edFdGZ)dHdIZ*dNdKdLZ+dJS)O QuaternionaProvides basic quaternion operations. Quaternion objects can be instantiated as Quaternion(a, b, c, d) as in (a + b*i + c*j + d*k). Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 2, 3, 4) >>> q 1 + 2*i + 3*j + 4*k Quaternions over complex fields can be defined as : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols, I >>> x = symbols('x') >>> q1 = Quaternion(x, x**3, x, x**2, real_field = False) >>> q2 = Quaternion(3 + 4*I, 2 + 5*I, 0, 7 + 8*I, real_field = False) >>> q1 x + x**3*i + x*j + x**2*k >>> q2 (3 + 4*I) + (2 + 5*I)*i + 0*j + (7 + 8*I)*k g&@FrTcCsvt|}t|}t|}t|}tdd||||fDr!tdt|||||}||_||_||_||_||_ |S)Ncss|]}|jduVqdS)FN)is_commutative).0ir;/usr/lib/python3/dist-packages/sympy/algebras/quaternion.py 3sz%Quaternion.__new__..z arguments have to be commutative) rany ValueErrorr__new___a_b_c_d _real_field)clsabcd real_fieldobjrrrr-szQuaternion.__new__cC|jSN)r selfrrrr&>z Quaternion.acCr,r-)r!r.rrrr'Br0z Quaternion.bcCr,r-)r"r.rrrr(Fr0z Quaternion.ccCr,r-)r#r.rrrr)Jr0z Quaternion.dcCr,r-)r$r.rrrr*Nr0zQuaternion.real_fieldc Cs|\}}}t|d|d|d}||||||}}}t|tj}t|tj}||} ||} ||} ||| | | S)aReturns a rotation quaternion given the axis and the angle of rotation. Parameters ========== vector : tuple of three numbers The vector representation of the given axis. angle : number The angle by which axis is rotated (in radians). Returns ======= Quaternion The normalized rotation quaternion calculated from the given axis and the angle of rotation. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import pi, sqrt >>> q = Quaternion.from_axis_angle((sqrt(3)/3, sqrt(3)/3, sqrt(3)/3), 2*pi/3) >>> q 1/2 + 1/2*i + 1/2*j + 1/2*k )rr rZHalfr ) r%Zvectoranglexyznormsr&r'r(r)rrrfrom_axis_angleRs zQuaternion.from_axis_anglecCs|tdd}t||d|d|dd}t||d|d|dd}t||d|d|dd}t||d|d|dd}|t|d|d}|t|d |d }|t|d |d }t||||S) aReturns the equivalent quaternion of a matrix. The quaternion will be normalized only if the matrix is special orthogonal (orthogonal and det(M) = 1). Parameters ========== M : Matrix Input matrix to be converted to equivalent quaternion. M must be special orthogonal (orthogonal and det(M) = 1) for the quaternion to be normalized. Returns ======= Quaternion The quaternion equivalent to given matrix. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import Matrix, symbols, cos, sin, trigsimp >>> x = symbols('x') >>> M = Matrix([[cos(x), -sin(x), 0], [sin(x), cos(x), 0], [0, 0, 1]]) >>> q = trigsimp(Quaternion.from_rotation_matrix(M)) >>> q sqrt(2)*sqrt(cos(x) + 1)/2 + 0*i + 0*j + sqrt(2 - 2*cos(x))*sign(sin(x))/2*k )rr)r9r9)r1r1r1)r1r9)r9r1)rr1)r1r)r9r)rr9)Zdetrrrr)r%MZabsQr&r'r(r)rrrfrom_rotation_matrix}s$$$$zQuaternion.from_rotation_matrixcC ||Sr-addr/otherrrr__add__ zQuaternion.__add__cCr=r-r>r@rrr__radd__rCzQuaternion.__radd__cCs||dSNr>r@rrr__sub__szQuaternion.__sub__cC |||Sr- _generic_mulr@rrr__mul__ zQuaternion.__mul__cCs |||Sr-rIr@rrr__rmul__rLzQuaternion.__rmul__cCr=r-)pow)r/prrr__pow__rCzQuaternion.__pow__cCst|j |j |j |j Sr-)rr r!r"r)r.rrr__neg__szQuaternion.__neg__cCs|t|dSrErr@rrr __truediv__zQuaternion.__truediv__cCst||dSrErr@rrr __rtruediv__rSzQuaternion.__rtruediv__cGs |j|Sr-rr/argsrrr_eval_IntegralrCzQuaternion._eval_Integralcs(dd|jfdd|jDS)NZevaluateTcsg|] }|jiqSr)diff)rr&kwargssymbolsrr z#Quaternion.diff..) setdefaultfuncrV)r/r[rZrrYrrXs zQuaternion.diffcCs|}t|}t|ts8|jr$|jr$tt||jt||j|j |j S|j r4t|j||j|j |j St dt|j|j|j|j|j |j |j |j S)aAdds quaternions. Parameters ========== other : Quaternion The quaternion to add to current (self) quaternion. Returns ======= Quaternion The resultant quaternion after adding self to other Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols >>> q1 = Quaternion(1, 2, 3, 4) >>> q2 = Quaternion(5, 6, 7, 8) >>> q1.add(q2) 6 + 8*i + 10*j + 12*k >>> q1 + 5 6 + 2*i + 3*j + 4*k >>> x = symbols('x', real = True) >>> q1.add(x) (x + 1) + 2*i + 3*j + 4*k Quaternions over complex fields : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import I >>> q3 = Quaternion(3 + 4*I, 2 + 5*I, 0, 7 + 8*I, real_field = False) >>> q3.add(2 + 3*I) (5 + 7*I) + (2 + 5*I)*i + 0*j + (7 + 8*I)*k z>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols >>> q1 = Quaternion(1, 2, 3, 4) >>> q2 = Quaternion(5, 6, 7, 8) >>> q1.mul(q2) (-60) + 12*i + 30*j + 24*k >>> q1.mul(2) 2 + 4*i + 6*j + 8*k >>> x = symbols('x', real = True) >>> q1.mul(x) x + 2*x*i + 3*x*j + 4*x*k Quaternions over complex fields : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import I >>> q3 = Quaternion(3 + 4*I, 2 + 5*I, 0, 7 + 8*I, real_field = False) >>> q3.mul(2 + 3*I) (2 + 3*I)*(3 + 4*I) + (2 + 3*I)*(2 + 5*I)*i + 0*j + (2 + 3*I)*(7 + 8*I)*k rIr@rrrmuls 'zQuaternion.mulcCst|}t|}t|tst|ts||St|tsH|jr.|jr.tt|t|dd|S|jrDt||j||j ||j ||j St dt|tsz|jr`|jr`|tt|t|ddS|jrvt||j||j ||j ||j St dt|j |j |j |j |j |j |j|j|j |j|j |j |j |j |j|j |j |j |j |j|j |j |j|j |j |j |j |j |j |j|j|j S)aGeneric multiplication. Parameters ========== q1 : Quaternion or symbol q2 : Quaternion or symbol It's important to note that if neither q1 nor q2 is a Quaternion, this function simply returns q1 * q2. Returns ======= Quaternion The resultant quaternion after multiplying q1 and q2 Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import Symbol >>> q1 = Quaternion(1, 2, 3, 4) >>> q2 = Quaternion(5, 6, 7, 8) >>> Quaternion._generic_mul(q1, q2) (-60) + 12*i + 30*j + 24*k >>> Quaternion._generic_mul(q1, 2) 2 + 4*i + 6*j + 8*k >>> x = Symbol('x', real = True) >>> Quaternion._generic_mul(q1, x) x + 2*x*i + 3*x*j + 4*x*k Quaternions over complex fields : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import I >>> q3 = Quaternion(3 + 4*I, 2 + 5*I, 0, 7 + 8*I, real_field = False) >>> Quaternion._generic_mul(q3, 2 + 3*I) (2 + 3*I)*(3 + 4*I) + (2 + 3*I)*(2 + 5*I)*i + 0*j + (2 + 3*I)*(7 + 8*I)*k rzAOnly commutative expressions can be multiplied with a Quaternion.) rr`rr*rarrrr&r'r(r)r)rbrcrrrrJ*s*+  &  &2.0.zQuaternion._generic_mulcCs |}t|j|j |j |j S)z(Returns the conjugate of the quaternion.)rr&r'r(r)r/qrrr_eval_conjugatesszQuaternion._eval_conjugatecCs4|}tt|jd|jd|jd|jdS)z#Returns the norm of the quaternion.r1)rrr&r'r(r)rerrrr6xs0zQuaternion.normcCs|}|d|S)z.Returns the normalized form of the quaternion.r9)r6rerrr normalizeszQuaternion.normalizecCs,|}|s tdt|d|dS)z&Returns the inverse of the quaternion.z6Cannot compute inverse for a quaternion with zero normr9r1)r6rrrerrrinverseszQuaternion.inversecCszt|}|}|dkr|Sd}|jstS|dkr!|| }}|dkr;|ddkr/||}|d}||}|dks%|S)aFinds the pth power of the quaternion. Parameters ========== p : int Power to be applied on quaternion. Returns ======= Quaternion Returns the p-th power of the current quaternion. Returns the inverse if p = -1. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 2, 3, 4) >>> q.pow(4) 668 + (-224)*i + (-336)*j + (-448)*k rFr9rr1)rriZ is_IntegerNotImplemented)r/rOrfresrrrrNs  zQuaternion.powcCs|}t|jd|jd|jd}t|jt|}t|jt||j|}t|jt||j|}t|jt||j|}t||||S)aReturns the exponential of q (e^q). Returns ======= Quaternion Exponential of q (e^q). Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 2, 3, 4) >>> q.exp() E*cos(sqrt(29)) + 2*sqrt(29)*E*sin(sqrt(29))/29*i + 3*sqrt(29)*E*sin(sqrt(29))/29*j + 4*sqrt(29)*E*sin(sqrt(29))/29*k r1) rr'r(r)r r&r r r)r/rf vector_normr&r'r(r)rrrr s"zQuaternion.expcCs|}t|jd|jd|jd}|}t|}|jt|j||}|jt|j||}|jt|j||}t||||S)ayReturns the natural logarithm of the quaternion (_ln(q)). Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 2, 3, 4) >>> q._ln() log(sqrt(30)) + 2*sqrt(29)*acos(sqrt(30)/30)/29*i + 3*sqrt(29)*acos(sqrt(30)/30)/29*j + 4*sqrt(29)*acos(sqrt(30)/30)/29*k r1) rr'r(r)r6r r r&r)r/rfrlZq_normr&r'r(r)rrr_lns"zQuaternion._lncstfdd|jDS)aReturns the floating point approximations (decimal numbers) of the quaternion. Returns ======= Quaternion Floating point approximations of quaternion(self) Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import sqrt >>> q = Quaternion(1/sqrt(1), 1/sqrt(2), 1/sqrt(3), 1/sqrt(4)) >>> q.evalf() 1.00000000000000 + 0.707106781186547*i + 0.577350269189626*j + 0.500000000000000*k csg|] }|jtdqS))n)Zevalfr)rargprecrrr\ r]z*Quaternion._eval_evalf..)rrV)r/rqrrpr _eval_evalfszQuaternion._eval_evalfcCs0|}|\}}t|||}|||S)amComputes the pth power in the cos-sin form. Parameters ========== p : int Power to be applied on quaternion. Returns ======= Quaternion The p-th power in the cos-sin form. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 2, 3, 4) >>> q.pow_cos_sin(4) 900*cos(4*acos(sqrt(30)/30)) + 1800*sqrt(29)*sin(4*acos(sqrt(30)/30))/29*i + 2700*sqrt(29)*sin(4*acos(sqrt(30)/30))/29*j + 3600*sqrt(29)*sin(4*acos(sqrt(30)/30))/29*k ) to_axis_anglerr8r6)r/rOrfvr2rcrrr pow_cos_sin s zQuaternion.pow_cos_sincGsFtt|jg|Rt|jg|Rt|jg|Rt|jg|RS)aComputes integration of quaternion. Returns ======= Quaternion Integration of the quaternion(self) with the given variable. Examples ======== Indefinite Integral of quaternion : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy.abc import x >>> q = Quaternion(1, 2, 3, 4) >>> q.integrate(x) x + 2*x*i + 3*x*j + 4*x*k Definite integral of quaternion : >>> from sympy.algebras.quaternion import Quaternion >>> from sympy.abc import x >>> q = Quaternion(1, 2, 3, 4) >>> q.integrate((x, 1, 5)) 4 + 8*i + 12*j + 16*k )rrr&r'r(r)rUrrrr.s" zQuaternion.integratecCs^t|trt|d|d}n|}|td|d|d|dt|}|j|j|jfS)aReturns the coordinates of the point pin(a 3 tuple) after rotation. Parameters ========== pin : tuple A 3-element tuple of coordinates of a point which needs to be rotated. r : Quaternion or tuple Axis and angle of rotation. It's important to note that when r is a tuple, it must be of the form (axis, angle) Returns ======= tuple The coordinates of the point after rotation. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols, trigsimp, cos, sin >>> x = symbols('x') >>> q = Quaternion(cos(x/2), 0, 0, sin(x/2)) >>> trigsimp(Quaternion.rotate_point((1, 1, 1), q)) (sqrt(2)*cos(x + pi/4), sqrt(2)*sin(x + pi/4), 1) >>> (axis, angle) = q.to_axis_angle() >>> trigsimp(Quaternion.rotate_point((1, 1, 1), (axis, angle))) (sqrt(2)*cos(x + pi/4), sqrt(2)*sin(x + pi/4), 1) rr9r1) r`tuplerr8rhrr'r(r))ZpinrrfZpoutrrr rotate_pointOs $&zQuaternion.rotate_pointc Cs|}|jjr |d}|}tdt|j}td|j|j}t|j|}t|j|}t|j|}|||f}||f}|S)aReturns the axis and angle of rotation of a quaternion Returns ======= tuple Tuple of (axis, angle) Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> q = Quaternion(1, 1, 1, 1) >>> (axis, angle) = q.to_axis_angle() >>> axis (sqrt(3)/3, sqrt(3)/3, sqrt(3)/3) >>> angle 2*pi/3 rFr1r9) r&Z is_negativerhrr rr'r(r)) r/rfr2r7r3r4r5rttrrrrs|s zQuaternion.to_axis_angleNcCs|}|d}dd||jd|jd}d||j|j|j|j}d||j|j|j|j}d||j|j|j|j}dd||jd|jd}d||j|j|j|j} d||j|j|j|j} d||j|j|j|j} dd||jd|jd} |st|||g||| g| | | ggS|\} }}| | |||||}|| ||||| }|| | || || }d}}}d}t||||g||| |g| | | |g||||ggS)aReturns the equivalent rotation transformation matrix of the quaternion which represents rotation about the origin if v is not passed. Parameters ========== v : tuple or None Default value: None Returns ======= tuple Returns the equivalent rotation transformation matrix of the quaternion which represents rotation about the origin if v is not passed. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols, trigsimp, cos, sin >>> x = symbols('x') >>> q = Quaternion(cos(x/2), 0, 0, sin(x/2)) >>> trigsimp(q.to_rotation_matrix()) Matrix([ [cos(x), -sin(x), 0], [sin(x), cos(x), 0], [ 0, 0, 1]]) Generates a 4x4 transformation matrix (used for rotation about a point other than the origin) if the point(v) is passed as an argument. Examples ======== >>> from sympy.algebras.quaternion import Quaternion >>> from sympy import symbols, trigsimp, cos, sin >>> x = symbols('x') >>> q = Quaternion(cos(x/2), 0, 0, sin(x/2)) >>> trigsimp(q.to_rotation_matrix((1, 1, 1))) Matrix([ [cos(x), -sin(x), 0, sin(x) - cos(x) + 1], [sin(x), cos(x), 0, -sin(x) - cos(x) + 1], [ 0, 0, 1, 0], [ 0, 0, 0, 1]]) r9r1r)r6r(r)r'r&r)r/rtrfr7Zm00Zm01Zm02Zm10Zm11Zm12Zm20Zm21Zm22r3r4r5Zm03Zm13Zm23Zm30Zm31Zm32Zm33rrrto_rotation_matrixs,1             zQuaternion.to_rotation_matrix)rrrrTr-),__name__ __module__ __qualname____doc__Z _op_priorityrrpropertyr&r'r(r)r* classmethodr8r<rBrDrGrKrMrPrQrRrTrWrXr?rd staticmethodrJrgr6rhrirNr rmrrrurrxrsr{rrrrrs`       * +6) H.#! ,(rN)Zsympyrrrrrrrr r r r r rrrrZsympy.core.evalfrZsympy.core.exprrrrrrrs