o o h(@sjddlmZmZmZmZmZmZddlmZddl m Z ddl m Z ddl mZdgZGddde ZdS) )Body Lagrangian KanesMethodLagrangesMethod RigidBodyParticle)BodyBase)_Methods)Matrix)sympy_deprecation_warning JointsMethodc@seZdZdZddZeddZeddZedd Zed d Z ed d Z eddZ eddZ eddZ eddZeddZddZddZddZddZd d!Zd"d#Zefd$d%Zd)d'd(Zd&S)*r a Method for formulating the equations of motion using a set of interconnected bodies with joints. .. deprecated:: 1.13 The JointsMethod class is deprecated. Its functionality has been replaced by the new :class:`~.System` class. Parameters ========== newtonion : Body or ReferenceFrame The newtonion(inertial) frame. *joints : Joint The joints in the system Attributes ========== q, u : iterable Iterable of the generalized coordinates and speeds bodies : iterable Iterable of Body objects in the system. loads : iterable Iterable of (Point, vector) or (ReferenceFrame, vector) tuples describing the forces on the system. mass_matrix : Matrix, shape(n, n) The system's mass matrix forcing : Matrix, shape(n, 1) The system's forcing vector mass_matrix_full : Matrix, shape(2*n, 2*n) The "mass matrix" for the u's and q's forcing_full : Matrix, shape(2*n, 1) The "forcing vector" for the u's and q's method : KanesMethod or Lagrange's method Method's object. kdes : iterable Iterable of kde in they system. Examples ======== As Body and JointsMethod have been deprecated, the following examples are for illustrative purposes only. The functionality of Body is fully captured by :class:`~.RigidBody` and :class:`~.Particle` and the functionality of JointsMethod is fully captured by :class:`~.System`. To ignore the deprecation warning we can use the ignore_warnings context manager. >>> from sympy.utilities.exceptions import ignore_warnings This is a simple example for a one degree of freedom translational spring-mass-damper. >>> from sympy import symbols >>> from sympy.physics.mechanics import Body, JointsMethod, PrismaticJoint >>> from sympy.physics.vector import dynamicsymbols >>> c, k = symbols('c k') >>> x, v = dynamicsymbols('x v') >>> with ignore_warnings(DeprecationWarning): ... wall = Body('W') ... body = Body('B') >>> J = PrismaticJoint('J', wall, body, coordinates=x, speeds=v) >>> wall.apply_force(c*v*wall.x, reaction_body=body) >>> wall.apply_force(k*x*wall.x, reaction_body=body) >>> with ignore_warnings(DeprecationWarning): ... method = JointsMethod(wall, J) >>> method.form_eoms() Matrix([[-B_mass*Derivative(v(t), t) - c*v(t) - k*x(t)]]) >>> M = method.mass_matrix_full >>> F = method.forcing_full >>> rhs = M.LUsolve(F) >>> rhs Matrix([ [ v(t)], [(-c*v(t) - k*x(t))/B_mass]]) Notes ===== ``JointsMethod`` currently only works with systems that do not have any configuration or motion constraints. cGsjtddddt|tr|j|_n||_||_||_||_| |_ | |_ | |_d|_dS)Nz The JointsMethod class is deprecated. Its functionality has been replaced by the new System class. z1.13z!deprecated-mechanics-jointsmethod)deprecated_since_versionactive_deprecations_target)r isinstancerframe_joints_generate_bodylist_bodies_generate_loadlist_loads _generate_q_q _generate_u_u_generate_kdes_kdes_method)self newtonionjointsr x/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/physics/mechanics/jointsmethod.py__init__^s        zJointsMethod.__init__cC|jS)zList of bodies in they system.)rrr r r!bodiesuzJointsMethod.bodiescCr#)zList of loads on the system.)rr$r r r!loadszr&zJointsMethod.loadscCr#z$List of the generalized coordinates.)rr$r r r!qr&zJointsMethod.qcCr#)zList of the generalized speeds.)rr$r r r!ur&zJointsMethod.ucCr#r()rr$r r r!kdesr&zJointsMethod.kdescC|jjS)z)The "forcing vector" for the u's and q's.)method forcing_fullr$r r r!r.zJointsMethod.forcing_fullcCr,)z&The "mass matrix" for the u's and q's.)r-mass_matrix_fullr$r r r!r0r/zJointsMethod.mass_matrix_fullcCr,)zThe system's mass matrix.)r- mass_matrixr$r r r!r1r/zJointsMethod.mass_matrixcCr,)zThe system's forcing vector.)r-forcingr$r r r!r2r/zJointsMethod.forcingcCr#)z3Object of method used to form equations of systems.)rr$r r r!r-r&zJointsMethod.methodcCs@g}|jD]}|j|vr||j|j|vr||jq|SN)rchildappendparent)rr%jointr r r!rs     zJointsMethod._generate_bodylistcCs*g}|jD] }t|tr||jq|Sr3)r%rrextendr')r load_listbodyr r r!rs    zJointsMethod._generate_loadlistcC>g}|jD]}|jD]}||vrtd||q qt|S)Nz'Coordinates of joints should be unique.)r coordinates ValueErrorr5r )rq_indr7 coordinater r r!r   zJointsMethod._generate_qcCr;)Nz"Speeds of joints should be unique.)rspeedsr=r5r )ru_indr7speedr r r!rr@zJointsMethod._generate_ucCs*tddgj}|jD]}||j}q |S)Nr)r Trcol_joinr+)rkd_indr7r r r!rs zJointsMethod._generate_kdescCsg}|jD]<}t|ts||q|jr/t|j|j|j|j |j |jf}|j |_ ||qt |j|j|j }|j |_ ||q|Sr3) r%rrr5 is_rigidbodyrname masscenterrmasscentral_inertiapotential_energyr)rbodylistr:rbpartr r r!_convert_bodiess      zJointsMethod._convert_bodiescCsl|}t|trt|jg|R}|||j|j||j|_n||j|j|j|j |j|d|_|j }|S)aMethod to form system's equation of motions. Parameters ========== method : Class Class name of method. Returns ======== Matrix Vector of equations of motions. Examples ======== As Body and JointsMethod have been deprecated, the following examples are for illustrative purposes only. The functionality of Body is fully captured by :class:`~.RigidBody` and :class:`~.Particle` and the functionality of JointsMethod is fully captured by :class:`~.System`. To ignore the deprecation warning we can use the ignore_warnings context manager. >>> from sympy.utilities.exceptions import ignore_warnings This is a simple example for a one degree of freedom translational spring-mass-damper. >>> from sympy import S, symbols >>> from sympy.physics.mechanics import LagrangesMethod, dynamicsymbols, Body >>> from sympy.physics.mechanics import PrismaticJoint, JointsMethod >>> q = dynamicsymbols('q') >>> qd = dynamicsymbols('q', 1) >>> m, k, b = symbols('m k b') >>> with ignore_warnings(DeprecationWarning): ... wall = Body('W') ... part = Body('P', mass=m) >>> part.potential_energy = k * q**2 / S(2) >>> J = PrismaticJoint('J', wall, part, coordinates=q, speeds=qd) >>> wall.apply_force(b * qd * wall.x, reaction_body=part) >>> with ignore_warnings(DeprecationWarning): ... method = JointsMethod(wall, J) >>> method.form_eoms(LagrangesMethod) Matrix([[b*Derivative(q(t), t) + k*q(t) + m*Derivative(q(t), (t, 2))]]) We can also solve for the states using the 'rhs' method. >>> method.rhs() Matrix([ [ Derivative(q(t), t)], [(-b*Derivative(q(t), t) - k*q(t))/m]]) )r>rBkd_eqs forcelistr%) rQ issubclassrrrr)r'rr*r+r- _form_eoms)rr-rNLsolnr r r! form_eomss8  zJointsMethod.form_eomsNcCs|jj|dS)azReturns equations that can be solved numerically. Parameters ========== inv_method : str The specific sympy inverse matrix calculation method to use. For a list of valid methods, see :meth:`~sympy.matrices.matrixbase.MatrixBase.inv` Returns ======== Matrix Numerically solvable equations. See Also ======== sympy.physics.mechanics.kane.KanesMethod.rhs: KanesMethod's rhs function. sympy.physics.mechanics.lagrange.LagrangesMethod.rhs: LagrangesMethod's rhs function. ) inv_method)r-rhs)rrYr r r!rZ#szJointsMethod.rhsr3)__name__ __module__ __qualname____doc__r"propertyr%r'r)r*r+r.r0r1r2r-rrrrrrQrrXrZr r r r!r s>R              BN)sympy.physics.mechanicsrrrrrr!sympy.physics.mechanics.body_basersympy.physics.mechanics.methodr sympyr sympy.utilities.exceptionsr __all__r r r r r!s