Ë
    7^(h¯(  ã                   óp   — d dl mZmZmZmZmZmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ dgZ G d„ de
«      Zy)	é    )ÚBodyÚ
LagrangianÚKanesMethodÚLagrangesMethodÚ	RigidBodyÚParticle)ÚBodyBase)Ú_Methods)ÚMatrix)Úsympy_deprecation_warningÚJointsMethodc                   óì   — e Zd ZdZd„ Zed„ «       Zed„ «       Zed„ «       Zed„ «       Z	ed„ «       Z
ed„ «       Zed	„ «       Zed
„ «       Zed„ «       Zed„ «       Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zefd„Zdd„Zy)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.

    c                 ó`  — t        ddd¬«       t        |t        «      r|j                  | _        n|| _        || _        | j                  «       | _        | j                  «       | _        | j                  «       | _
        | j                  «       | _        | j                  «       | _        d | _        y )Nzˆ
            The JointsMethod class is deprecated.
            Its functionality has been replaced by the new System class.
            z1.13z!deprecated-mechanics-jointsmethod)Údeprecated_since_versionÚactive_deprecations_target)r   Ú
isinstancer	   ÚframeÚ_jointsÚ_generate_bodylistÚ_bodiesÚ_generate_loadlistÚ_loadsÚ_generate_qÚ_qÚ_generate_uÚ_uÚ_generate_kdesÚ_kdesÚ_method)ÚselfÚ	newtonionÚjointss      úb/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sympy/physics/mechanics/jointsmethod.pyÚ__init__zJointsMethod.__init__^   s“   € Ü!ðð &,Ø'Jõ	
ô �i¤Ô*Ø"Ÿ™ˆD�Jà"ˆDŒJàˆŒØ×.Ñ.Ó0ˆŒØ×-Ñ-Ó/ˆŒØ×"Ñ"Ó$ˆŒØ×"Ñ"Ó$ˆŒØ×(Ñ(Ó*ˆŒ
àˆ�ó    c                 ó   — | j                   S )zList of bodies in they system.)r   ©r    s    r#   ÚbodieszJointsMethod.bodiesu   ó   € ð �|‰|Ðr%   c                 ó   — | j                   S )zList of loads on the system.)r   r'   s    r#   ÚloadszJointsMethod.loadsz   s   € ð �{‰{Ðr%   c                 ó   — | j                   S ©z$List of the generalized coordinates.)r   r'   s    r#   ÚqzJointsMethod.q   ó   € ð �w‰wˆr%   c                 ó   — | j                   S )zList of the generalized speeds.)r   r'   s    r#   ÚuzJointsMethod.u„   r/   r%   c                 ó   — | j                   S r-   )r   r'   s    r#   ÚkdeszJointsMethod.kdes‰   s   € ð �z‰zÐr%   c                 ó.   — | j                   j                  S )z)The "forcing vector" for the u's and q's.)ÚmethodÚforcing_fullr'   s    r#   r6   zJointsMethod.forcing_fullŽ   s   € ð �{‰{×'Ñ'Ð'r%   c                 ó.   — | j                   j                  S )z&The "mass matrix" for the u's and q's.)r5   Úmass_matrix_fullr'   s    r#   r8   zJointsMethod.mass_matrix_full“   s   € ð �{‰{×+Ñ+Ð+r%   c                 ó.   — | j                   j                  S )zThe system's mass matrix.)r5   Úmass_matrixr'   s    r#   r:   zJointsMethod.mass_matrix˜   s   € ð �{‰{×&Ñ&Ð&r%   c                 ó.   — | j                   j                  S )zThe system's forcing vector.)r5   Úforcingr'   s    r#   r<   zJointsMethod.forcing�   s   € ð �{‰{×"Ñ"Ð"r%   c                 ó   — | j                   S )z3Object of method used to form equations of systems.)r   r'   s    r#   r5   zJointsMethod.method¢   r)   r%   c                 óÒ   — g }| j                   D ]U  }|j                  |vr|j                  |j                  «       |j                  |vsŒ;|j                  |j                  «       ŒW |S ©N)r   ÚchildÚappendÚparent)r    r(   Újoints      r#   r   zJointsMethod._generate_bodylist§   sY   € ØˆØ—\‘\ò 	,ˆEØ�{‰{ &Ñ(Ø—‘˜eŸk™kÔ*Ø�|‰| 6Ò)Ø—‘˜eŸl™lÕ+ð		,ð
 ˆr%   c                 ó„   — g }| j                   D ].  }t        |t        «      sŒ|j                  |j                  «       Œ0 |S r?   )r(   r   r   Úextendr+   )r    Ú	load_listÚbodys      r#   r   zJointsMethod._generate_loadlist°   s>   € Øˆ	Ø—K‘Kò 	-ˆDÜ˜$¤Õ%Ø× Ñ  §¡Õ,ð	-ð Ðr%   c                 ó    — g }| j                   D ]3  }|j                  D ]"  }||v rt        d«      ‚|j                  |«       Œ$ Œ5 t	        |«      S )Nz'Coordinates of joints should be unique.)r   ÚcoordinatesÚ
ValueErrorrA   r   )r    Úq_indrC   Ú
coordinates       r#   r   zJointsMethod._generate_q·   s[   € ØˆØ—\‘\ò 	)ˆEØ#×/Ñ/ò )�
Ø Ñ&Ü$Ð%NÓOÐOØ—‘˜ZÕ(ñ)ð	)ô
 �e‹}Ðr%   c                 ó    — g }| j                   D ]3  }|j                  D ]"  }||v rt        d«      ‚|j                  |«       Œ$ Œ5 t	        |«      S )Nz"Speeds of joints should be unique.)r   ÚspeedsrJ   rA   r   )r    Úu_indrC   Úspeeds       r#   r   zJointsMethod._generate_uÀ   sX   € ØˆØ—\‘\ò 	$ˆEØŸ™ò $�Ø˜E‘>Ü$Ð%IÓJÐJØ—‘˜UÕ#ñ$ð	$ô
 �e‹}Ðr%   c                 óŒ   — t        ddg «      j                  }| j                  D ]  }|j                  |j                  «      }Œ |S )Né   r   )r   ÚTr   Úcol_joinr3   )r    Úkd_indrC   s      r#   r   zJointsMethod._generate_kdesÉ   s@   € Ü˜˜1˜bÓ!×#Ñ#ˆØ—\‘\ò 	1ˆEØ—_‘_ U§Z¡ZÓ0‰Fð	1àˆr%   c           	      ó  — g }| j                   D ]í  }t        |t        «      s|j                  |«       Œ%|j                  rpt        |j                  |j                  |j                  |j                  |j                  |j                  f«      }|j                  |_        |j                  |«       Œ¡t        |j                  |j                  |j                  «      }|j                  |_        |j                  |«       Œï |S r?   )r(   r   r   rA   Úis_rigidbodyr   ÚnameÚ
masscenterr   ÚmassÚcentral_inertiaÚpotential_energyr   )r    ÚbodylistrG   ÚrbÚparts        r#   Ú_convert_bodieszJointsMethod._convert_bodiesÏ   sÅ   € àˆØ—K‘Kò 	&ˆDÜ˜d¤DÔ)Ø—‘ Ô%ØØ× Ò Ü˜tŸy™y¨$¯/©/¸4¿:¹:ÀtÇyÁyØ×)Ñ)¨4¯?©?Ð;ó=�à&*×&;Ñ&;�Ô#Ø—‘ Õ#ä §	¡	¨4¯?©?¸D¿I¹IÓF�Ø(,×(=Ñ(=�Ô%Ø—‘ Õ%ð	&ð ˆr%   c                 ó�  — | j                  «       }t        |t        «      rFt        | j                  g|¢­Ž } ||| j
                  | j                  || j                  «      | _        nE || j                  | j
                  | j                  | j                  | j                  |¬«      | _        | j                  j                  «       }|S )aŠ  Method 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]])

        )rK   rO   Úkd_eqsÚ	forcelistr(   )r`   Ú
issubclassr   r   r   r.   r+   r   r1   r3   r5   Ú
_form_eoms)r    r5   r]   ÚLÚsolns        r#   Ú	form_eomszJointsMethod.form_eomsá   s˜   € ðp ×'Ñ'Ó)ˆÜ�fœoÔ.Ü˜4Ÿ:™:Ð1¨Ò1ˆAÙ! ! T§V¡V¨T¯Z©Z¸À4Ç:Á:ÓNˆD�Lá! $§*¡*°D·F±FÀ$Ç&Á&ÐQU×QZÑQZØ.2¯j©jÀôKˆDŒLà�{‰{×%Ñ%Ó'ˆØˆr%   Nc                 ó:   — | j                   j                  |¬«      S )az  Returns 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)r5   Úrhs)r    rj   s     r#   rk   zJointsMethod.rhs#  s   € ð6 �{‰{�‰¨*ˆÓ5Ð5r%   r?   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r$   Úpropertyr(   r+   r.   r1   r3   r6   r8   r:   r<   r5   r   r   r   r   r   r`   r   rh   rk   © r%   r#   r   r      s  „ ñPòdð. ñó ðð ñó ðð ñó ðð ñó ðð ñó ðð ñ(ó ð(ð ñ,ó ð,ð ñ'ó ð'ð ñ#ó ð#ð ñó ðòòòòòòð$  +ó @ôD6r%   N)Úsympy.physics.mechanicsr   r   r   r   r   r   Ú!sympy.physics.mechanics.body_baser	   Úsympy.physics.mechanics.methodr
   Úsympyr   Úsympy.utilities.exceptionsr   Ú__all__r   rq   r%   r#   ú<module>rx      s1   ð÷9÷ 9å 6Ý 3Ý Ý @àÐ
€ôs6�8õ s6r%   