jaxsim 0.5.1.dev139__tar.gz → 0.5.1.dev147__tar.gz

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: jaxsim
-Version: 0.5.1.dev139
+Version: 0.5.1.dev147
 Summary: A differentiable physics engine and multibody dynamics library for control and robot learning.
 Author-email: Diego Ferigo <dgferigo@gmail.com>, Filippo Luca Ferretti <filippoluca.ferretti@outlook.com>
 Maintainer-email: Filippo Luca Ferretti <filippo.ferretti@iit.it>, Alessandro Croci <alessandro.croci@iit.it>

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/examples/jaxsim_for_robot_controllers.ipynb

@@ -111,10 +111,12 @@
    "source": [
     "# @title Load the URDF model\n",
     "import pathlib\n",
+    "import urllib\n",
     "\n",
-    "model_urdf_string = pathlib.Path(\n",
-    "    os.path.abspath(\"\") + \"/assets/cartpole.urdf\"\n",
-    ").read_text()"
+    "# Retrieve the file\n",
+    "url = \"https://raw.githubusercontent.com/ami-iit/jaxsim/refs/heads/main/examples/assets/cartpole.urdf\"\n",
+    "model_path, _ = urllib.request.urlretrieve(url)\n",
+    "model_urdf_string = pathlib.Path(model_path).read_text()"
    ]
   },
   {
@@ -435,6 +437,11 @@
     "\n",
     "Additionally, if your controllers or planners require the derivatives of the dynamics with respect to the state or inputs, you can obtain them using automatic differentiation directly through JaxSim's API."
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": []
   }
  ],
  "metadata": {
@@ -459,7 +466,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.7"
+   "version": "3.13.1"
   }
  },
  "nbformat": 4,

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/src/jaxsim/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.5.1.dev139'
-__version_tuple__ = version_tuple = (0, 5, 1, 'dev139')
+__version__ = version = '0.5.1.dev147'
+__version_tuple__ = version_tuple = (0, 5, 1, 'dev147')

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/src/jaxsim/api/model.py

@@ -2295,9 +2295,6 @@ def step(
         # Hence, here we need to reset the velocity after each impact to guarantee that
         # the linear velocity of the active collidable points is zero.
         case jaxsim.rbda.contacts.RigidContacts():
-            assert isinstance(
-                data_tf.contacts_params, jaxsim.rbda.contacts.RigidContactsParams
-            )
 
             # Raise runtime error for not supported case in which Rigid contacts and
             # Baumgarte stabilization are enabled and used with ForwardEuler integrator.
@@ -2331,12 +2328,13 @@ def step(
                     indices_of_enabled_collidable_points
                 ]
                 M = js.model.free_floating_mass_matrix(model, data_tf)
+                BW_ν_pre_impact = data_tf.generalized_velocity()
 
                 # Compute the impact velocity.
                 # It may be discontinuous in case new contacts are made.
-                BW_nu_post_impact = (
+                BW_ν_post_impact = (
                     jaxsim.rbda.contacts.RigidContacts.compute_impact_velocity(
-                        data=data_tf,
+                        generalized_velocity=BW_ν_pre_impact,
                         inactive_collidable_points=(δ <= 0),
                         M=M,
                         J_WC=J_WC,
@@ -2344,8 +2342,8 @@ def step(
                 )
 
                 # Reset the generalized velocity.
-                data_tf = data_tf.reset_base_velocity(BW_nu_post_impact[0:6])
-                data_tf = data_tf.reset_joint_velocities(BW_nu_post_impact[6:])
+                data_tf = data_tf.reset_base_velocity(BW_ν_post_impact[0:6])
+                data_tf = data_tf.reset_joint_velocities(BW_ν_post_impact[6:])
 
     # Restore the input velocity representation.
     data_tf = data_tf.replace(

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/src/jaxsim/rbda/contacts/rigid.py

@@ -173,7 +173,7 @@ class RigidContacts(ContactModel):
         inactive_collidable_points: jtp.ArrayLike,
         M: jtp.MatrixLike,
         J_WC: jtp.MatrixLike,
-        data: js.data.JaxSimModelData,
+        generalized_velocity: jtp.VectorLike,
     ) -> jtp.Vector:
         """
         Return the new velocity of the system after a potential impact.
@@ -182,55 +182,37 @@ class RigidContacts(ContactModel):
             inactive_collidable_points: The activation state of the collidable points.
             M: The mass matrix of the system (in mixed representation).
             J_WC: The Jacobian matrix of the collidable points (in mixed representation).
-            data: The `JaxSimModelData` instance.
-        """
+            generalized_velocity: The generalized velocity of the system.
 
-        def impact_velocity(
-            inactive_collidable_points: jtp.ArrayLike,
-            nu_pre: jtp.ArrayLike,
-            M: jtp.MatrixLike,
-            J_WC: jtp.MatrixLike,
-            data: js.data.JaxSimModelData,
-        ):
-            # Compute system velocity after impact maintaining zero linear velocity of active points
-            with data.switch_velocity_representation(VelRepr.Mixed):
-                sl = jnp.s_[:, 0:3, :]
-                Jl_WC = J_WC[sl]
-                # Zero out the jacobian rows of inactive points
-                Jl_WC = jnp.vstack(
-                    jnp.where(
-                        inactive_collidable_points[:, jnp.newaxis, jnp.newaxis],
-                        jnp.zeros_like(Jl_WC),
-                        Jl_WC,
-                    )
-                )
+        Note:
+            The mass matrix `M`, the Jacobian `J_WC`, and the generalized velocity `generalized_velocity`
+            must be expressed in the same velocity representation.
+        """
 
-                A = jnp.vstack(
-                    [
-                        jnp.hstack([M, -Jl_WC.T]),
-                        jnp.hstack(
-                            [Jl_WC, jnp.zeros((Jl_WC.shape[0], Jl_WC.shape[0]))]
-                        ),
-                    ]
-                )
-                b = jnp.hstack([M @ nu_pre, jnp.zeros(Jl_WC.shape[0])])
-                x = jnp.linalg.lstsq(A, b)[0]
-                nu_post = x[0 : M.shape[0]]
+        # Compute system velocity after impact maintaining zero linear velocity of active points.
+        sl = jnp.s_[:, 0:3, :]
+        Jl_WC = J_WC[sl]
+
+        # Zero out the jacobian rows of inactive points.
+        Jl_WC = jnp.vstack(
+            jnp.where(
+                inactive_collidable_points[:, jnp.newaxis, jnp.newaxis],
+                jnp.zeros_like(Jl_WC),
+                Jl_WC,
+            )
+        )
 
-                return nu_post
+        A = jnp.vstack(
+            [
+                jnp.hstack([M, -Jl_WC.T]),
+                jnp.hstack([Jl_WC, jnp.zeros((Jl_WC.shape[0], Jl_WC.shape[0]))]),
+            ]
+        )
+        b = jnp.hstack([M @ generalized_velocity, jnp.zeros(Jl_WC.shape[0])])
 
-        with data.switch_velocity_representation(VelRepr.Mixed):
-            BW_ν_pre_impact = data.generalized_velocity()
-
-            BW_ν_post_impact = impact_velocity(
-                data=data,
-                inactive_collidable_points=inactive_collidable_points,
-                nu_pre=BW_ν_pre_impact,
-                M=M,
-                J_WC=J_WC,
-            )
+        BW_ν_post_impact = jnp.linalg.lstsq(A, b)[0]
 
-        return BW_ν_post_impact
+        return BW_ν_post_impact[0 : M.shape[0]]
 
     @jax.jit
     def compute_contact_forces(

{jaxsim-0.5.1.dev139 → jaxsim-0.5.1.dev147}/src/jaxsim.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: jaxsim
-Version: 0.5.1.dev139
+Version: 0.5.1.dev147
 Summary: A differentiable physics engine and multibody dynamics library for control and robot learning.
 Author-email: Diego Ferigo <dgferigo@gmail.com>, Filippo Luca Ferretti <filippoluca.ferretti@outlook.com>
 Maintainer-email: Filippo Luca Ferretti <filippo.ferretti@iit.it>, Alessandro Croci <alessandro.croci@iit.it>