o o heK@sddlmZmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZddlmZddlmZgd ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZdS))ABCabstractmethod)pi DerivativeMatrix) AppliedUndef)BodyBase)_validate_coordinates)VectordynamicsymbolscrossPointReferenceFrame)iterable)sympy_deprecation_warning)JointPinJointPrismaticJointCylindricalJoint PlanarJointSphericalJoint WeldJointc@s^eZdZdZ    d>ddZddZddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZeddZeddZeddZeddZedd Zed!d"Zed#d$Zed%d&Zed'd(Zed)d*Zed+d,Zed-d.Zed/d0Ze  d?d1d2Zd3d4Zd@d5d6Z d@d7d8Z! : ;dAdrQz Joint.parentcCrK)zChild body of Joint.)r$rMrHrHrIr?rQz Joint.childcCrK)z.Matrix of the joint's generalized coordinates.)r3rMrHrHrIr6rQzJoint.coordinatescCrK)z)Matrix of the joint's generalized speeds.)r5rMrHrHrIr7rQz Joint.speedscCrK)z0Kinematical differential equations of the joint.)r9rMrHrHrIkdesrQz Joint.kdescCrK)zThe axis of parent frame.)r-rMrHrHrIrDzJoint.parent_axiscCrK)zThe axis of child frame.)r.rMrHrHrIrErSzJoint.child_axiscCrK)z=Attachment point where the joint is fixed to the parent body.)r0rMrHrHrIr@ rQzJoint.parent_pointcCrK)z>> from sympy.physics.mechanics.joint import Joint >>> from sympy.physics.mechanics import RigidBody >>> parent = RigidBody('parent') >>> parent_interframe = Joint._create_aligned_interframe( ... parent, parent.y + parent.z) >>> parent_interframe parent_int_frame >>> parent.frame.dcm(parent_interframe) Matrix([ [ 0, -sqrt(2)/2, -sqrt(2)/2], [sqrt(2)/2, 1/2, -1/2], [sqrt(2)/2, -1/2, 1/2]]) >>> (parent.y + parent.z).express(parent_interframe) sqrt(2)*parent_int_frame.x Notes ===== The direction cosine matrix between the given frame and intermediate frame is formed using a simple rotation about an axis that is normal to both ``align_axis`` and ``frame_axis``. In general, the normal axis is formed by crossing the ``frame_axis`` with the ``align_axis``. The exception is if the axes are parallel with opposite directions, in which case the rotation vector is chosen using the rules in the following table with the vectors expressed in the given frame: .. list-table:: :header-rows: 1 * - ``align_axis`` - ``frame_axis`` - ``rotation_axis`` * - ``-x`` - ``x`` - ``z`` * - ``-y`` - ``y`` - ``x`` * - ``-z`` - ``z`` - ``y`` * - ``-x-y`` - ``x+y`` - ``z`` * - ``-y-z`` - ``y+z`` - ``x`` * - ``-x-z`` - ``x+z`` - ``y`` * - ``-x-y-z`` - ``x+y+z`` - ``(x+y+z) × x`` Nir _int_framer)rrrr[r' angle_betweenr r rrrhr orient_axis set_ang_vel)r align_axis frame_axis frame_nameangle rotation_axis int_framerHrHrI_create_aligned_interframefs$ U    z Joint._create_aligned_interframecCsHg}tj}tt|jD]}||j|| |j|q t|S)z,Generate kinematical differential equations.) r _trangelenr6appenddiffr7r)r=rRtirHrHrIr8s $zJoint._generate_kdescCs|dur|j}|dur|jSt|ttfstdt|tr/|jd|jd}|j||}||j |dks>t d|S)z'Returns the attachment point of a body.Nz+Attachment point must be a Point or Vector.r_jointrz6Attachment point must be fixed to the associated body.) r masscenterrr r r!r' locatenewpos_fromr^r])r=body joint_pos body_frame point_namerHrHrIr/s zJoint._locate_joint_poscCs|dur|j}|dur |St|tr#tj|||jd|jdd}n t|ts,td||dks>t d|d|d |j |d|S) z'Returns the attachment frame of a body.Nrri)roz$Interframe must be a ReferenceFrame.rz Interframe z is not fixed to body .) rrr rrsr'rr! ang_vel_inr]r|set_vel)r=r interframerrHrHrIr)s"    zJoint._locate_joint_frameqrFc sfdd}dkrdnd}g}|durg}nt|s!|g}t|dks@t|ks@tdd |d t|d |d t|D])\} } | durV||| |qDt| ttfrc|| qDtd |d | d t t|||D] } ||| qyt |S)aHelper method for _generate_coordinates and _generate_speeds. Parameters ========== coordinates : iterable Iterable of coordinates or speeds that have been provided. n_coords : Integer Number of coordinates that should be returned. label : String, optional Coordinate type either 'q' (coordinates) or 'u' (speeds). The Default is 'q'. offset : Integer Count offset when creating new dynamicsymbols. The default is 0. number_single : Boolean Boolean whether if n_coords == 1, number should still be used. The default is False. cs8dkrstdjSt|djS)Nrbr)r r')numberlabeln_coords number_singler=rHrI create_symbols z2Joint._fill_coordinate_list..create_symbolrzgeneralized coordinatezgeneralized speedNrz Expected  zs, instead got zs.zThe z" should have been a dynamicsymbol.) rrvr] enumeraterwrrrr!rur) r=r6rroffsetrrr'generated_coordinatesrzcoordrHrrI_fill_coordinate_lists,  zJoint._fill_coordinate_list) NNNNNNNNNN)NNrL)rrF)#__name__ __module__ __qualname____doc__rJrNrPpropertyr'r>r?r6r7rRrDrEr@rArBrCrr2r4r:r;r< staticmethodrZr,rhrsr8r/r)rrHrHrHrIrsvt U                     i  rcfeZdZdZ    dfdd ZddZeddZd d Zd d Z d dZ ddZ ddZ Z S)ru!Pin (Revolute) Joint. .. raw:: html :file: ../../../doc/src/modules/physics/mechanics/api/PinJoint.svg Explanation =========== A pin joint is defined such that the joint rotation axis is fixed in both the child and parent and the location of the joint is relative to the mass center of each body. The child rotates an angle, θ, from the parent about the rotation axis and has a simple angular speed, ω, relative to the parent. The direction cosine matrix between the child interframe and parent interframe is formed using a simple rotation about the joint axis. The page on the joints framework gives a more detailed explanation of the intermediate frames. Parameters ========== name : string A unique name for the joint. parent : Particle or RigidBody or Body The parent body of joint. child : Particle or RigidBody or Body The child body of joint. coordinates : dynamicsymbol, optional Generalized coordinates of the joint. speeds : dynamicsymbol, optional Generalized speeds of joint. parent_point : Point or Vector, optional Attachment point where the joint is fixed to the parent body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the parent's mass center. child_point : Point or Vector, optional Attachment point where the joint is fixed to the child body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the child's mass center. parent_axis : Vector, optional .. deprecated:: 1.12 Axis fixed in the parent body which aligns with an axis fixed in the child body. The default is the x axis of parent's reference frame. For more information on this deprecation, see :ref:`deprecated-mechanics-joint-axis`. child_axis : Vector, optional .. deprecated:: 1.12 Axis fixed in the child body which aligns with an axis fixed in the parent body. The default is the x axis of child's reference frame. For more information on this deprecation, see :ref:`deprecated-mechanics-joint-axis`. parent_interframe : ReferenceFrame, optional Intermediate frame of the parent body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the parent's own frame. child_interframe : ReferenceFrame, optional Intermediate frame of the child body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the child's own frame. joint_axis : Vector The axis about which the rotation occurs. Note that the components of this axis are the same in the parent_interframe and child_interframe. parent_joint_pos : Point or Vector, optional .. deprecated:: 1.12 This argument is replaced by parent_point and will be removed in a future version. See :ref:`deprecated-mechanics-joint-pos` for more information. child_joint_pos : Point or Vector, optional .. deprecated:: 1.12 This argument is replaced by child_point and will be removed in a future version. See :ref:`deprecated-mechanics-joint-pos` for more information. Attributes ========== name : string The joint's name. parent : Particle or RigidBody or Body The joint's parent body. child : Particle or RigidBody or Body The joint's child body. coordinates : Matrix Matrix of the joint's generalized coordinates. The default value is ``dynamicsymbols(f'q_{joint.name}')``. speeds : Matrix Matrix of the joint's generalized speeds. The default value is ``dynamicsymbols(f'u_{joint.name}')``. parent_point : Point Attachment point where the joint is fixed to the parent body. child_point : Point Attachment point where the joint is fixed to the child body. parent_axis : Vector The axis fixed in the parent frame that represents the joint. child_axis : Vector The axis fixed in the child frame that represents the joint. parent_interframe : ReferenceFrame Intermediate frame of the parent body with respect to which the joint transformation is formulated. child_interframe : ReferenceFrame Intermediate frame of the child body with respect to which the joint transformation is formulated. joint_axis : Vector The axis about which the rotation occurs. Note that the components of this axis are the same in the parent_interframe and child_interframe. kdes : Matrix Kinematical differential equations of the joint. Examples ========= A single pin joint is created from two bodies and has the following basic attributes: >>> from sympy.physics.mechanics import RigidBody, PinJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = PinJoint('PC', parent, child) >>> joint PinJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.parent_axis P_frame.x >>> joint.child_axis C_frame.x >>> joint.coordinates Matrix([[q_PC(t)]]) >>> joint.speeds Matrix([[u_PC(t)]]) >>> child.frame.ang_vel_in(parent.frame) u_PC(t)*P_frame.x >>> child.frame.dcm(parent.frame) Matrix([ [1, 0, 0], [0, cos(q_PC(t)), sin(q_PC(t))], [0, -sin(q_PC(t)), cos(q_PC(t))]]) >>> joint.child_point.pos_from(joint.parent_point) 0 To further demonstrate the use of the pin joint, the kinematics of simple double pendulum that rotates about the Z axis of each connected body can be created as follows. >>> from sympy import symbols, trigsimp >>> from sympy.physics.mechanics import RigidBody, PinJoint >>> l1, l2 = symbols('l1 l2') First create bodies to represent the fixed ceiling and one to represent each pendulum bob. >>> ceiling = RigidBody('C') >>> upper_bob = RigidBody('U') >>> lower_bob = RigidBody('L') The first joint will connect the upper bob to the ceiling by a distance of ``l1`` and the joint axis will be about the Z axis for each body. >>> ceiling_joint = PinJoint('P1', ceiling, upper_bob, ... child_point=-l1*upper_bob.frame.x, ... joint_axis=ceiling.frame.z) The second joint will connect the lower bob to the upper bob by a distance of ``l2`` and the joint axis will also be about the Z axis for each body. >>> pendulum_joint = PinJoint('P2', upper_bob, lower_bob, ... child_point=-l2*lower_bob.frame.x, ... joint_axis=upper_bob.frame.z) Once the joints are established the kinematics of the connected bodies can be accessed. First the direction cosine matrices of pendulum link relative to the ceiling are found: >>> upper_bob.frame.dcm(ceiling.frame) Matrix([ [ cos(q_P1(t)), sin(q_P1(t)), 0], [-sin(q_P1(t)), cos(q_P1(t)), 0], [ 0, 0, 1]]) >>> trigsimp(lower_bob.frame.dcm(ceiling.frame)) Matrix([ [ cos(q_P1(t) + q_P2(t)), sin(q_P1(t) + q_P2(t)), 0], [-sin(q_P1(t) + q_P2(t)), cos(q_P1(t) + q_P2(t)), 0], [ 0, 0, 1]]) The position of the lower bob's masscenter is found with: >>> lower_bob.masscenter.pos_from(ceiling.masscenter) l1*U_frame.x + l2*L_frame.x The angular velocities of the two pendulum links can be computed with respect to the ceiling. >>> upper_bob.frame.ang_vel_in(ceiling.frame) u_P1(t)*C_frame.z >>> lower_bob.frame.ang_vel_in(ceiling.frame) u_P1(t)*C_frame.z + u_P2(t)*U_frame.z And finally, the linear velocities of the two pendulum bobs can be computed with respect to the ceiling. >>> upper_bob.masscenter.vel(ceiling.frame) l1*u_P1(t)*U_frame.y >>> lower_bob.masscenter.vel(ceiling.frame) l1*u_P1(t)*U_frame.y + l2*(u_P1(t) + u_P2(t))*L_frame.y Nc.| |_t||||||||| | | | | dSrL _joint_axissuperrJr=r'r>r?r6r7r@rArBrCrDrE joint_axisrFrG __class__rHrIrJ  zPinJoint.__init__cCd|jd|jd|jS)Nz PinJoint: parent: child: r'r>r?rMrHrHrIrNzPinJoint.__str__cCrK)z>Axis about which the child rotates with respect to the parent.rrMrHrHrIrrQzPinJoint.joint_axiscC||ddSNrbrrr= coordinaterHrHrIr2zPinJoint._generate_coordinatescCrNrburr=speedrHrHrIr4"rzPinJoint._generate_speedscCs0||j|j|_|j|j|j|jddSNr)r,rrBrrCrkr6rMrHrHrIr:%szPinJoint._orient_framescCs$|j|j|jd|jdSr)rCrlrBr7r normalizerMrHrHrIr;*s  zPinJoint._set_angular_velocitycCL|j|jd|j|jd|j|jd|jj|j|j|jdSr rAset_posr@rr&r(r?r| v2pt_theoryrMrHrHrIr<.s  zPinJoint._set_linear_velocity NNNNNNNNNNNrrrrrJrNrrr2r4r:r;r< __classcell__rHrHrrIr,s _  rcr)raPrismatic (Sliding) Joint. .. image:: PrismaticJoint.svg Explanation =========== It is defined such that the child body translates with respect to the parent body along the body-fixed joint axis. The location of the joint is defined by two points, one in each body, which coincide when the generalized coordinate is zero. The direction cosine matrix between the parent_interframe and child_interframe is the identity matrix. Therefore, the direction cosine matrix between the parent and child frames is fully defined by the definition of the intermediate frames. The page on the joints framework gives a more detailed explanation of the intermediate frames. Parameters ========== name : string A unique name for the joint. parent : Particle or RigidBody or Body The parent body of joint. child : Particle or RigidBody or Body The child body of joint. coordinates : dynamicsymbol, optional Generalized coordinates of the joint. The default value is ``dynamicsymbols(f'q_{joint.name}')``. speeds : dynamicsymbol, optional Generalized speeds of joint. The default value is ``dynamicsymbols(f'u_{joint.name}')``. parent_point : Point or Vector, optional Attachment point where the joint is fixed to the parent body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the parent's mass center. child_point : Point or Vector, optional Attachment point where the joint is fixed to the child body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the child's mass center. parent_axis : Vector, optional .. deprecated:: 1.12 Axis fixed in the parent body which aligns with an axis fixed in the child body. The default is the x axis of parent's reference frame. For more information on this deprecation, see :ref:`deprecated-mechanics-joint-axis`. child_axis : Vector, optional .. deprecated:: 1.12 Axis fixed in the child body which aligns with an axis fixed in the parent body. The default is the x axis of child's reference frame. For more information on this deprecation, see :ref:`deprecated-mechanics-joint-axis`. parent_interframe : ReferenceFrame, optional Intermediate frame of the parent body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the parent's own frame. child_interframe : ReferenceFrame, optional Intermediate frame of the child body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the child's own frame. joint_axis : Vector The axis along which the translation occurs. Note that the components of this axis are the same in the parent_interframe and child_interframe. parent_joint_pos : Point or Vector, optional .. deprecated:: 1.12 This argument is replaced by parent_point and will be removed in a future version. See :ref:`deprecated-mechanics-joint-pos` for more information. child_joint_pos : Point or Vector, optional .. deprecated:: 1.12 This argument is replaced by child_point and will be removed in a future version. See :ref:`deprecated-mechanics-joint-pos` for more information. Attributes ========== name : string The joint's name. parent : Particle or RigidBody or Body The joint's parent body. child : Particle or RigidBody or Body The joint's child body. coordinates : Matrix Matrix of the joint's generalized coordinates. speeds : Matrix Matrix of the joint's generalized speeds. parent_point : Point Attachment point where the joint is fixed to the parent body. child_point : Point Attachment point where the joint is fixed to the child body. parent_axis : Vector The axis fixed in the parent frame that represents the joint. child_axis : Vector The axis fixed in the child frame that represents the joint. parent_interframe : ReferenceFrame Intermediate frame of the parent body with respect to which the joint transformation is formulated. child_interframe : ReferenceFrame Intermediate frame of the child body with respect to which the joint transformation is formulated. kdes : Matrix Kinematical differential equations of the joint. Examples ========= A single prismatic joint is created from two bodies and has the following basic attributes: >>> from sympy.physics.mechanics import RigidBody, PrismaticJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = PrismaticJoint('PC', parent, child) >>> joint PrismaticJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.parent_axis P_frame.x >>> joint.child_axis C_frame.x >>> joint.coordinates Matrix([[q_PC(t)]]) >>> joint.speeds Matrix([[u_PC(t)]]) >>> child.frame.ang_vel_in(parent.frame) 0 >>> child.frame.dcm(parent.frame) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) >>> joint.child_point.pos_from(joint.parent_point) q_PC(t)*P_frame.x To further demonstrate the use of the prismatic joint, the kinematics of two masses sliding, one moving relative to a fixed body and the other relative to the moving body. about the X axis of each connected body can be created as follows. >>> from sympy.physics.mechanics import PrismaticJoint, RigidBody First create bodies to represent the fixed ceiling and one to represent a particle. >>> wall = RigidBody('W') >>> Part1 = RigidBody('P1') >>> Part2 = RigidBody('P2') The first joint will connect the particle to the ceiling and the joint axis will be about the X axis for each body. >>> J1 = PrismaticJoint('J1', wall, Part1) The second joint will connect the second particle to the first particle and the joint axis will also be about the X axis for each body. >>> J2 = PrismaticJoint('J2', Part1, Part2) Once the joint is established the kinematics of the connected bodies can be accessed. First the direction cosine matrices of Part relative to the ceiling are found: >>> Part1.frame.dcm(wall.frame) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) >>> Part2.frame.dcm(wall.frame) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) The position of the particles' masscenter is found with: >>> Part1.masscenter.pos_from(wall.masscenter) q_J1(t)*W_frame.x >>> Part2.masscenter.pos_from(wall.masscenter) q_J1(t)*W_frame.x + q_J2(t)*P1_frame.x The angular velocities of the two particle links can be computed with respect to the ceiling. >>> Part1.frame.ang_vel_in(wall.frame) 0 >>> Part2.frame.ang_vel_in(wall.frame) 0 And finally, the linear velocities of the two particles can be computed with respect to the ceiling. >>> Part1.masscenter.vel(wall.frame) u_J1(t)*W_frame.x >>> Part2.masscenter.vel(wall.frame) u_J1(t)*W_frame.x + Derivative(q_J2(t), t)*P1_frame.x NcrrLrrrrHrIrJrzPrismaticJoint.__init__cCr)NzPrismaticJoint: rrrrMrHrHrIrNrzPrismaticJoint.__str__cCrK)zAAxis along which the child translates with respect to the parent.rrMrHrHrIr!rQzPrismaticJoint.joint_axiscCrrrrrHrHrIr2&rz$PrismaticJoint._generate_coordinatescCrrrrrHrHrIr4)rzPrismaticJoint._generate_speedscCs*||j|j|_|j|j|jddSr)r,rrBrrCrkrMrHrHrIr:,s zPrismaticJoint._orient_framescC|j|jddSrrCrlrBrMrHrHrIr;1z$PrismaticJoint._set_angular_velocitycCs~|j}|j|j|jd||j|jd|j|jd|j|j|j d||j j |j|j d|dSr) rrrArr@r6rr&r(r7r?r|)r=rfrHrHrIr<4s  z#PrismaticJoint._set_linear_velocityrrrHrHrrIr6s \  rcseZdZdZ     dfdd ZddZeddZed d Zed d Z ed dZ eddZ ddZ ddZ ddZddZddZZS)ra'#Cylindrical Joint. .. image:: CylindricalJoint.svg :align: center :width: 600 Explanation =========== A cylindrical joint is defined such that the child body both rotates about and translates along the body-fixed joint axis with respect to the parent body. The joint axis is both the rotation axis and translation axis. The location of the joint is defined by two points, one in each body, which coincide when the generalized coordinate corresponding to the translation is zero. The direction cosine matrix between the child interframe and parent interframe is formed using a simple rotation about the joint axis. The page on the joints framework gives a more detailed explanation of the intermediate frames. Parameters ========== name : string A unique name for the joint. parent : Particle or RigidBody or Body The parent body of joint. child : Particle or RigidBody or Body The child body of joint. rotation_coordinate : dynamicsymbol, optional Generalized coordinate corresponding to the rotation angle. The default value is ``dynamicsymbols(f'q0_{joint.name}')``. translation_coordinate : dynamicsymbol, optional Generalized coordinate corresponding to the translation distance. The default value is ``dynamicsymbols(f'q1_{joint.name}')``. rotation_speed : dynamicsymbol, optional Generalized speed corresponding to the angular velocity. The default value is ``dynamicsymbols(f'u0_{joint.name}')``. translation_speed : dynamicsymbol, optional Generalized speed corresponding to the translation velocity. The default value is ``dynamicsymbols(f'u1_{joint.name}')``. parent_point : Point or Vector, optional Attachment point where the joint is fixed to the parent body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the parent's mass center. child_point : Point or Vector, optional Attachment point where the joint is fixed to the child body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the child's mass center. parent_interframe : ReferenceFrame, optional Intermediate frame of the parent body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the parent's own frame. child_interframe : ReferenceFrame, optional Intermediate frame of the child body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the child's own frame. joint_axis : Vector, optional The rotation as well as translation axis. Note that the components of this axis are the same in the parent_interframe and child_interframe. Attributes ========== name : string The joint's name. parent : Particle or RigidBody or Body The joint's parent body. child : Particle or RigidBody or Body The joint's child body. rotation_coordinate : dynamicsymbol Generalized coordinate corresponding to the rotation angle. translation_coordinate : dynamicsymbol Generalized coordinate corresponding to the translation distance. rotation_speed : dynamicsymbol Generalized speed corresponding to the angular velocity. translation_speed : dynamicsymbol Generalized speed corresponding to the translation velocity. coordinates : Matrix Matrix of the joint's generalized coordinates. speeds : Matrix Matrix of the joint's generalized speeds. parent_point : Point Attachment point where the joint is fixed to the parent body. child_point : Point Attachment point where the joint is fixed to the child body. parent_interframe : ReferenceFrame Intermediate frame of the parent body with respect to which the joint transformation is formulated. child_interframe : ReferenceFrame Intermediate frame of the child body with respect to which the joint transformation is formulated. kdes : Matrix Kinematical differential equations of the joint. joint_axis : Vector The axis of rotation and translation. Examples ========= A single cylindrical joint is created between two bodies and has the following basic attributes: >>> from sympy.physics.mechanics import RigidBody, CylindricalJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = CylindricalJoint('PC', parent, child) >>> joint CylindricalJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.parent_axis P_frame.x >>> joint.child_axis C_frame.x >>> joint.coordinates Matrix([ [q0_PC(t)], [q1_PC(t)]]) >>> joint.speeds Matrix([ [u0_PC(t)], [u1_PC(t)]]) >>> child.frame.ang_vel_in(parent.frame) u0_PC(t)*P_frame.x >>> child.frame.dcm(parent.frame) Matrix([ [1, 0, 0], [0, cos(q0_PC(t)), sin(q0_PC(t))], [0, -sin(q0_PC(t)), cos(q0_PC(t))]]) >>> joint.child_point.pos_from(joint.parent_point) q1_PC(t)*P_frame.x >>> child.masscenter.vel(parent.frame) u1_PC(t)*P_frame.x To further demonstrate the use of the cylindrical joint, the kinematics of two cylindrical joints perpendicular to each other can be created as follows. >>> from sympy import symbols >>> from sympy.physics.mechanics import RigidBody, CylindricalJoint >>> r, l, w = symbols('r l w') First create bodies to represent the fixed floor with a fixed pole on it. The second body represents a freely moving tube around that pole. The third body represents a solid flag freely translating along and rotating around the Y axis of the tube. >>> floor = RigidBody('floor') >>> tube = RigidBody('tube') >>> flag = RigidBody('flag') The first joint will connect the first tube to the floor with it translating along and rotating around the Z axis of both bodies. >>> floor_joint = CylindricalJoint('C1', floor, tube, joint_axis=floor.z) The second joint will connect the tube perpendicular to the flag along the Y axis of both the tube and the flag, with the joint located at a distance ``r`` from the tube's center of mass and a combination of the distances ``l`` and ``w`` from the flag's center of mass. >>> flag_joint = CylindricalJoint('C2', tube, flag, ... parent_point=r * tube.y, ... child_point=-w * flag.y + l * flag.z, ... joint_axis=tube.y) Once the joints are established the kinematics of the connected bodies can be accessed. First the direction cosine matrices of both the body and the flag relative to the floor are found: >>> tube.frame.dcm(floor.frame) Matrix([ [ cos(q0_C1(t)), sin(q0_C1(t)), 0], [-sin(q0_C1(t)), cos(q0_C1(t)), 0], [ 0, 0, 1]]) >>> flag.frame.dcm(floor.frame) Matrix([ [cos(q0_C1(t))*cos(q0_C2(t)), sin(q0_C1(t))*cos(q0_C2(t)), -sin(q0_C2(t))], [ -sin(q0_C1(t)), cos(q0_C1(t)), 0], [sin(q0_C2(t))*cos(q0_C1(t)), sin(q0_C1(t))*sin(q0_C2(t)), cos(q0_C2(t))]]) The position of the flag's center of mass is found with: >>> flag.masscenter.pos_from(floor.masscenter) q1_C1(t)*floor_frame.z + (r + q1_C2(t))*tube_frame.y + w*flag_frame.y - l*flag_frame.z The angular velocities of the two tubes can be computed with respect to the floor. >>> tube.frame.ang_vel_in(floor.frame) u0_C1(t)*floor_frame.z >>> flag.frame.ang_vel_in(floor.frame) u0_C1(t)*floor_frame.z + u0_C2(t)*tube_frame.y Finally, the linear velocities of the two tube centers of mass can be computed with respect to the floor, while expressed in the tube's frame. >>> tube.masscenter.vel(floor.frame).to_matrix(tube.frame) Matrix([ [ 0], [ 0], [u1_C1(t)]]) >>> flag.masscenter.vel(floor.frame).to_matrix(tube.frame).simplify() Matrix([ [-l*u0_C2(t)*cos(q0_C2(t)) - r*u0_C1(t) - w*u0_C1(t) - q1_C2(t)*u0_C1(t)], [ -l*u0_C1(t)*sin(q0_C2(t)) + Derivative(q1_C2(t), t)], [ l*u0_C2(t)*sin(q0_C2(t)) + u1_C1(t)]]) Nc  s8| |_||f} ||f}tj|||| ||| | | d dSNrBrCr)r=r'r>r?rotation_coordinatetranslation_coordinaterotation_speedtranslation_speedr@rArBrCrr6r7rrHrIrJs zCylindricalJoint.__init__cCr)NzCylindricalJoint: rrrrMrHrHrIrN,rzCylindricalJoint.__str__cCrK)z?Axis about and along which the rotation and translation occurs.rrMrHrHrIr0rQzCylindricalJoint.joint_axiscC |jdSz;Generalized coordinate corresponding to the rotation angle.rr6rMrHrHrIr5 z$CylindricalJoint.rotation_coordinatecCr)zAGeneralized coordinate corresponding to the translation distance.rbrrMrHrHrIr:rz'CylindricalJoint.translation_coordinatecCrz8Generalized speed corresponding to the angular velocity.rr7rMrHrHrIr?rzCylindricalJoint.rotation_speedcCr)z>> from sympy.physics.mechanics import RigidBody, PlanarJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = PlanarJoint('PC', parent, child) >>> joint PlanarJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.rotation_axis P_frame.x >>> joint.planar_vectors [P_frame.y, P_frame.z] >>> joint.rotation_coordinate q0_PC(t) >>> joint.planar_coordinates Matrix([ [q1_PC(t)], [q2_PC(t)]]) >>> joint.coordinates Matrix([ [q0_PC(t)], [q1_PC(t)], [q2_PC(t)]]) >>> joint.rotation_speed u0_PC(t) >>> joint.planar_speeds Matrix([ [u1_PC(t)], [u2_PC(t)]]) >>> joint.speeds Matrix([ [u0_PC(t)], [u1_PC(t)], [u2_PC(t)]]) >>> child.frame.ang_vel_in(parent.frame) u0_PC(t)*P_frame.x >>> child.frame.dcm(parent.frame) Matrix([ [1, 0, 0], [0, cos(q0_PC(t)), sin(q0_PC(t))], [0, -sin(q0_PC(t)), cos(q0_PC(t))]]) >>> joint.child_point.pos_from(joint.parent_point) q1_PC(t)*P_frame.y + q2_PC(t)*P_frame.z >>> child.masscenter.vel(parent.frame) u1_PC(t)*P_frame.y + u2_PC(t)*P_frame.z To further demonstrate the use of the planar joint, the kinematics of a block sliding on a slope, can be created as follows. >>> from sympy import symbols >>> from sympy.physics.mechanics import PlanarJoint, RigidBody, ReferenceFrame >>> a, d, h = symbols('a d h') First create bodies to represent the slope and the block. >>> ground = RigidBody('G') >>> block = RigidBody('B') To define the slope you can either define the plane by specifying the ``planar_vectors`` or/and the ``rotation_axis``. However it is advisable to create a rotated intermediate frame, so that the ``parent_vectors`` and ``rotation_axis`` will be the unit vectors of this intermediate frame. >>> slope = ReferenceFrame('A') >>> slope.orient_axis(ground.frame, ground.y, a) The planar joint can be created using these bodies and intermediate frame. We can specify the origin of the slope to be ``d`` above the slope's center of mass and the block's center of mass to be a distance ``h`` above the slope's surface. Note that we can specify the normal of the plane using the rotation axis argument. >>> joint = PlanarJoint('PC', ground, block, parent_point=d * ground.x, ... child_point=-h * block.x, parent_interframe=slope) Once the joint is established the kinematics of the bodies can be accessed. First the ``rotation_axis``, which is normal to the plane and the ``plane_vectors``, can be found. >>> joint.rotation_axis A.x >>> joint.planar_vectors [A.y, A.z] The direction cosine matrix of the block with respect to the ground can be found with: >>> block.frame.dcm(ground.frame) Matrix([ [ cos(a), 0, -sin(a)], [sin(a)*sin(q0_PC(t)), cos(q0_PC(t)), sin(q0_PC(t))*cos(a)], [sin(a)*cos(q0_PC(t)), -sin(q0_PC(t)), cos(a)*cos(q0_PC(t))]]) The angular velocity of the block can be computed with respect to the ground. >>> block.frame.ang_vel_in(ground.frame) u0_PC(t)*A.x The position of the block's center of mass can be found with: >>> block.masscenter.pos_from(ground.masscenter) d*G_frame.x + h*B_frame.x + q1_PC(t)*A.y + q2_PC(t)*A.z Finally, the linear velocity of the block's center of mass can be computed with respect to the ground. >>> block.masscenter.vel(ground.frame) u1_PC(t)*A.y + u2_PC(t)*A.z In some cases it could be your preference to only define the normals of the plane with respect to both bodies. This can most easily be done by supplying vectors to the ``interframe`` arguments. What will happen in this case is that an interframe will be created with its ``x`` axis aligned with the provided vector. For a further explanation of how this is done see the notes of the ``Joint`` class. In the code below, the above example (with the block on the slope) is recreated by supplying vectors to the interframe arguments. Note that the previously described option is however more computationally efficient, because the algorithm now has to compute the rotation angle between the provided vector and the 'x' axis. >>> from sympy import symbols, cos, sin >>> from sympy.physics.mechanics import PlanarJoint, RigidBody >>> a, d, h = symbols('a d h') >>> ground = RigidBody('G') >>> block = RigidBody('B') >>> joint = PlanarJoint( ... 'PC', ground, block, parent_point=d * ground.x, ... child_point=-h * block.x, child_interframe=block.x, ... parent_interframe=cos(a) * ground.x + sin(a) * ground.z) >>> block.frame.dcm(ground.frame).simplify() Matrix([ [ cos(a), 0, sin(a)], [-sin(a)*sin(q0_PC(t)), cos(q0_PC(t)), sin(q0_PC(t))*cos(a)], [-sin(a)*cos(q0_PC(t)), -sin(q0_PC(t)), cos(a)*cos(q0_PC(t))]]) Nc  s2||f} ||f} tj|||| | || | | d dSr)rrJ)r=r'r>r?rplanar_coordinatesr planar_speedsr@rArBrCr6r7rrHrIrJys zPlanarJoint.__init__cCr)Nz PlanarJoint: rrrrMrHrHrIrNrzPlanarJoint.__str__cCrrrrMrHrHrIrrzPlanarJoint.rotation_coordinatecC|jdddfS)z` method. The default rotation consists of three relative rotations, i.e. body-fixed rotations. Based on the direction cosine matrix following from these rotations, the angular velocity is computed based on the generalized coordinates and generalized speeds. Parameters ========== name : string A unique name for the joint. parent : Particle or RigidBody or Body The parent body of joint. child : Particle or RigidBody or Body The child body of joint. coordinates: iterable of dynamicsymbols, optional Generalized coordinates of the joint. speeds : iterable of dynamicsymbols, optional Generalized speeds of joint. parent_point : Point or Vector, optional Attachment point where the joint is fixed to the parent body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the parent's mass center. child_point : Point or Vector, optional Attachment point where the joint is fixed to the child body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the child's mass center. parent_interframe : ReferenceFrame, optional Intermediate frame of the parent body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the parent's own frame. child_interframe : ReferenceFrame, optional Intermediate frame of the child body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the child's own frame. rot_type : str, optional The method used to generate the direction cosine matrix. Supported methods are: - ``'Body'``: three successive rotations about new intermediate axes, also called "Euler and Tait-Bryan angles" - ``'Space'``: three successive rotations about the parent frames' unit vectors The default method is ``'Body'``. amounts : Expressions defining the rotation angles or direction cosine matrix. These must match the ``rot_type``. See examples below for details. The input types are: - ``'Body'``: 3-tuple of expressions, symbols, or functions - ``'Space'``: 3-tuple of expressions, symbols, or functions The default amounts are the given ``coordinates``. rot_order : str or int, optional If applicable, the order of the successive of rotations. The string ``'123'`` and integer ``123`` are equivalent, for example. Required for ``'Body'`` and ``'Space'``. The default value is ``123``. Attributes ========== name : string The joint's name. parent : Particle or RigidBody or Body The joint's parent body. child : Particle or RigidBody or Body The joint's child body. coordinates : Matrix Matrix of the joint's generalized coordinates. speeds : Matrix Matrix of the joint's generalized speeds. parent_point : Point Attachment point where the joint is fixed to the parent body. child_point : Point Attachment point where the joint is fixed to the child body. parent_interframe : ReferenceFrame Intermediate frame of the parent body with respect to which the joint transformation is formulated. child_interframe : ReferenceFrame Intermediate frame of the child body with respect to which the joint transformation is formulated. kdes : Matrix Kinematical differential equations of the joint. Examples ========= A single spherical joint is created from two bodies and has the following basic attributes: >>> from sympy.physics.mechanics import RigidBody, SphericalJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = SphericalJoint('PC', parent, child) >>> joint SphericalJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.parent_interframe P_frame >>> joint.child_interframe C_frame >>> joint.coordinates Matrix([ [q0_PC(t)], [q1_PC(t)], [q2_PC(t)]]) >>> joint.speeds Matrix([ [u0_PC(t)], [u1_PC(t)], [u2_PC(t)]]) >>> child.frame.ang_vel_in(parent.frame).to_matrix(child.frame) Matrix([ [ u0_PC(t)*cos(q1_PC(t))*cos(q2_PC(t)) + u1_PC(t)*sin(q2_PC(t))], [-u0_PC(t)*sin(q2_PC(t))*cos(q1_PC(t)) + u1_PC(t)*cos(q2_PC(t))], [ u0_PC(t)*sin(q1_PC(t)) + u2_PC(t)]]) >>> child.frame.x.to_matrix(parent.frame) Matrix([ [ cos(q1_PC(t))*cos(q2_PC(t))], [sin(q0_PC(t))*sin(q1_PC(t))*cos(q2_PC(t)) + sin(q2_PC(t))*cos(q0_PC(t))], [sin(q0_PC(t))*sin(q2_PC(t)) - sin(q1_PC(t))*cos(q0_PC(t))*cos(q2_PC(t))]]) >>> joint.child_point.pos_from(joint.parent_point) 0 To further demonstrate the use of the spherical joint, the kinematics of a spherical joint with a ZXZ rotation can be created as follows. >>> from sympy import symbols >>> from sympy.physics.mechanics import RigidBody, SphericalJoint >>> l1 = symbols('l1') First create bodies to represent the fixed floor and a pendulum bob. >>> floor = RigidBody('F') >>> bob = RigidBody('B') The joint will connect the bob to the floor, with the joint located at a distance of ``l1`` from the child's center of mass and the rotation set to a body-fixed ZXZ rotation. >>> joint = SphericalJoint('S', floor, bob, child_point=l1 * bob.y, ... rot_type='body', rot_order='ZXZ') Now that the joint is established, the kinematics of the connected body can be accessed. The position of the bob's masscenter is found with: >>> bob.masscenter.pos_from(floor.masscenter) - l1*B_frame.y The angular velocities of the pendulum link can be computed with respect to the floor. >>> bob.frame.ang_vel_in(floor.frame).to_matrix( ... floor.frame).simplify() Matrix([ [u1_S(t)*cos(q0_S(t)) + u2_S(t)*sin(q0_S(t))*sin(q1_S(t))], [u1_S(t)*sin(q0_S(t)) - u2_S(t)*sin(q1_S(t))*cos(q0_S(t))], [ u0_S(t) + u2_S(t)*cos(q1_S(t))]]) Finally, the linear velocity of the bob's center of mass can be computed. >>> bob.masscenter.vel(floor.frame).to_matrix(bob.frame) Matrix([ [ l1*(u0_S(t)*cos(q1_S(t)) + u2_S(t))], [ 0], [-l1*(u0_S(t)*sin(q1_S(t))*sin(q2_S(t)) + u1_S(t)*cos(q2_S(t)))]]) NBODY{c s4| |_| |_| |_tj||||||||| d dSr) _rot_type_amounts _rot_orderrrJ) r=r'r>r?r6r7r@rArBrCrot_typeamounts rot_orderrrHrIrJs zSphericalJoint.__init__cCr)NzSphericalJoint: rrrrMrHrHrIrNrzSphericalJoint.__str__cCr)NrrrVrHrHrIr2rz$SphericalJoint._generate_coordinatescCs||t|jdS)Nr)rrvr6rXrHrHrIr4rzSphericalJoint._generate_speedscCsZd}|j|vrtd|jd||jdur|jn|j}|j|j|j||jdS)N)rSPACEzRotation type "z6" is not implemented. Implemented rotation types are: ) rupperNotImplementedErrorrr6rCorientrBr)r=supported_rot_typesrrHrHrIr:s zSphericalJoint._orient_framescsFtj|j|jfddt|j|jD}|j |j|dS)Ncsi|] \}}||qSrH)rx).0rrryrHrI sz8SphericalJoint._set_angular_velocity..) r rtrCrrBxreplacezipr6r7rl)r=velrHrrIr;s z$SphericalJoint._set_angular_velocitycCrrrrMrHrHrIr<s z#SphericalJoint._set_linear_velocity) NNNNNNrNr rrrrrJrNr2r4r:r;r<rrHrHrrIrsL  rcsVeZdZdZ  dfdd ZddZddZd d Zd d Zd dZ ddZ Z S)raWeld Joint. .. raw:: html :file: ../../../doc/src/modules/physics/mechanics/api/WeldJoint.svg Explanation =========== A weld joint is defined such that there is no relative motion between the child and parent bodies. The direction cosine matrix between the attachment frame (``parent_interframe`` and ``child_interframe``) is the identity matrix and the attachment points (``parent_point`` and ``child_point``) are coincident. The page on the joints framework gives a more detailed explanation of the intermediate frames. Parameters ========== name : string A unique name for the joint. parent : Particle or RigidBody or Body The parent body of joint. child : Particle or RigidBody or Body The child body of joint. parent_point : Point or Vector, optional Attachment point where the joint is fixed to the parent body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the parent's mass center. child_point : Point or Vector, optional Attachment point where the joint is fixed to the child body. If a vector is provided, then the attachment point is computed by adding the vector to the body's mass center. The default value is the child's mass center. parent_interframe : ReferenceFrame, optional Intermediate frame of the parent body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the parent's own frame. child_interframe : ReferenceFrame, optional Intermediate frame of the child body with respect to which the joint transformation is formulated. If a Vector is provided then an interframe is created which aligns its X axis with the given vector. The default value is the child's own frame. Attributes ========== name : string The joint's name. parent : Particle or RigidBody or Body The joint's parent body. child : Particle or RigidBody or Body The joint's child body. coordinates : Matrix Matrix of the joint's generalized coordinates. The default value is ``dynamicsymbols(f'q_{joint.name}')``. speeds : Matrix Matrix of the joint's generalized speeds. The default value is ``dynamicsymbols(f'u_{joint.name}')``. parent_point : Point Attachment point where the joint is fixed to the parent body. child_point : Point Attachment point where the joint is fixed to the child body. parent_interframe : ReferenceFrame Intermediate frame of the parent body with respect to which the joint transformation is formulated. child_interframe : ReferenceFrame Intermediate frame of the child body with respect to which the joint transformation is formulated. kdes : Matrix Kinematical differential equations of the joint. Examples ========= A single weld joint is created from two bodies and has the following basic attributes: >>> from sympy.physics.mechanics import RigidBody, WeldJoint >>> parent = RigidBody('P') >>> parent P >>> child = RigidBody('C') >>> child C >>> joint = WeldJoint('PC', parent, child) >>> joint WeldJoint: PC parent: P child: C >>> joint.name 'PC' >>> joint.parent P >>> joint.child C >>> joint.parent_point P_masscenter >>> joint.child_point C_masscenter >>> joint.coordinates Matrix(0, 0, []) >>> joint.speeds Matrix(0, 0, []) >>> child.frame.ang_vel_in(parent.frame) 0 >>> child.frame.dcm(parent.frame) Matrix([ [1, 0, 0], [0, 1, 0], [0, 0, 1]]) >>> joint.child_point.pos_from(joint.parent_point) 0 To further demonstrate the use of the weld joint, two relatively-fixed bodies rotated by a quarter turn about the Y axis can be created as follows: >>> from sympy import symbols, pi >>> from sympy.physics.mechanics import ReferenceFrame, RigidBody, WeldJoint >>> l1, l2 = symbols('l1 l2') First create the bodies to represent the parent and rotated child body. >>> parent = RigidBody('P') >>> child = RigidBody('C') Next the intermediate frame specifying the fixed rotation with respect to the parent can be created. >>> rotated_frame = ReferenceFrame('Pr') >>> rotated_frame.orient_axis(parent.frame, parent.y, pi / 2) The weld between the parent body and child body is located at a distance ``l1`` from the parent's center of mass in the X direction and ``l2`` from the child's center of mass in the child's negative X direction. >>> weld = WeldJoint('weld', parent, child, parent_point=l1 * parent.x, ... child_point=-l2 * child.x, ... parent_interframe=rotated_frame) Now that the joint has been established, the kinematics of the bodies can be accessed. The direction cosine matrix of the child body with respect to the parent can be found: >>> child.frame.dcm(parent.frame) Matrix([ [0, 0, -1], [0, 1, 0], [1, 0, 0]]) As can also been seen from the direction cosine matrix, the parent X axis is aligned with the child's Z axis: >>> parent.x == child.z True The position of the child's center of mass with respect to the parent's center of mass can be found with: >>> child.masscenter.pos_from(parent.masscenter) l1*P_frame.x + l2*C_frame.x The angular velocity of the child with respect to the parent is 0 as one would expect. >>> child.frame.ang_vel_in(parent.frame) 0 Nc s2tj|||gg||||d tddgj|_dS)Nrrbr)rrJrTr9)r=r'r>r?r@rArBrCrrHrIrJps zWeldJoint.__init__cCr)Nz WeldJoint: rrrrMrHrHrIrNwrzWeldJoint.__str__cCtSrLrrrHrHrIr2{rOzWeldJoint._generate_coordinatescCrrLrrrHrHrIr4~rOzWeldJoint._generate_speedscCs|j|j|jjddSr)rCrkrBr[rMrHrHrIr:s zWeldJoint._orient_framescCrrrrMrHrHrIr;rzWeldJoint._set_angular_velocitycCsF|j|jd|j|jd|j|jd|jj|jddSr)rArr@rr&r(r?r|rMrHrHrIr<szWeldJoint._set_linear_velocity)NNNNrrHrHrrIrs)rN)abcrrsympyrrrsympy.core.functionr!sympy.physics.mechanics.body_baser!sympy.physics.mechanics.functionsr sympy.physics.vectorr r r r rsympy.utilities.iterablesrsympy.utilities.exceptionsr__all__rrrrrrrrHrHrHrIs:       +h|