jaxsim 0.7.1.dev18__py3-none-any.whl → 0.7.1.dev32__py3-none-any.whl

jaxsim/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.7.1.dev18'
-__version_tuple__ = version_tuple = (0, 7, 1, 'dev18')
+__version__ = version = '0.7.1.dev32'
+__version_tuple__ = version_tuple = (0, 7, 1, 'dev32')

jaxsim/api/kin_dyn_parameters.py

@@ -910,9 +910,9 @@ class HwLinkMetadata(JaxsimDataclass):
 
     Attributes:
         shape: The shape of the link.
-            0 = box, 1 = sphere, 2 = cylinder, -1 = unsupported.
+            0 = box, 1 = cylinder, 2 = sphere, -1 = unsupported.
         dims: The dimensions of the link.
-            box: [lx,ly,lz], sphere: [r,0,0], cylinder: [r,l,0]
+            box: [lx,ly,lz], cylinder: [r,l,0], sphere: [r,0,0].
         density: The density of the link.
         L_H_G: The homogeneous transformation matrix from the link frame to the CoM frame G.
         L_H_vis: The homogeneous transformation matrix from the link frame to the visual frame.
@@ -1026,7 +1026,13 @@ class HwLinkMetadata(JaxsimDataclass):
             shape,
             branches=[
                 # Box
-                lambda: scaling_factors,
+                lambda: jnp.array(
+                    [
+                        scaling_factors[0],
+                        scaling_factors[1],
+                        scaling_factors[2],
+                    ]
+                ),
                 # Cylinder
                 lambda: jnp.array(
                     [

jaxsim/api/model.py

@@ -387,7 +387,7 @@ class JaxSimModel(JaxsimDataclass):
         L_H_Gs = []
         L_H_vises = []
         L_H_pre_masks = []
-        L_H_pres = []
+        L_H_pre = []
 
         # Process each link
         for link_description in ordered_links:
@@ -454,17 +454,17 @@ class JaxSimModel(JaxsimDataclass):
             L_H_pre_masks.append(
                 [
                     int(joint_index in child_joints_indices)
-                    for joint_index in range(self.number_of_joints())
+                    for joint_index in range(0, self.number_of_joints())
                 ]
             )
-            L_H_pres.append(
+            L_H_pre.append(
                 [
                     (
                         self.kin_dyn_parameters.joint_model.λ_H_pre[joint_index + 1]
                         if joint_index in child_joints_indices
                         else jnp.eye(4)
                     )
-                    for joint_index in range(self.number_of_joints())
+                    for joint_index in range(0, self.number_of_joints())
                 ]
             )
 
@@ -476,7 +476,7 @@ class JaxSimModel(JaxsimDataclass):
             L_H_G=jnp.array(L_H_Gs, dtype=float),
             L_H_vis=jnp.array(L_H_vises, dtype=float),
             L_H_pre_mask=jnp.array(L_H_pre_masks, dtype=bool),
-            L_H_pre=jnp.array(L_H_pres, dtype=float),
+            L_H_pre=jnp.array(L_H_pre, dtype=float),
         )
 
     def export_updated_model(self) -> str:
@@ -777,10 +777,17 @@ def reduce(
         integrator=model.integrator,
     )
 
-    # Store the origin of the model, in case downstream logic needs it.
     with reduced_model.mutable_context(mutability=Mutability.MUTABLE_NO_VALIDATION):
+        # Store the origin of the model, in case downstream logic needs it.
         reduced_model.built_from = model.built_from
 
+        # Compute the hw parametrization metadata of the reduced model
+        # TODO: move the building of the metadata to KinDynParameters.build()
+        #       and use the model_description instead of model.built_from.
+        reduced_model.kin_dyn_parameters.hw_link_metadata = (
+            reduced_model.compute_hw_link_metadata()
+        )
+
     return reduced_model
 
 
@@ -2373,15 +2380,22 @@ def update_hw_parameters(
         return jax.lax.cond(
             jnp.any(L_H_pre_mask_for_joint),
             lambda: selected_transform,
-            lambda: kin_dyn_params.joint_model.λ_H_pre[joint_index],
+            lambda: kin_dyn_params.joint_model.λ_H_pre[joint_index + 1],
         )
 
     # Apply the update function to all joint indices
     updated_λ_H_pre = jax.vmap(update_λ_H_pre)(
-        jnp.arange(kin_dyn_params.number_of_joints() + 1)
+        jnp.arange(kin_dyn_params.number_of_joints())
+    )
+    # NOTE: λ_H_pre should be of len (1+n_joints) with the 0-th element equal
+    # to identity to represent the world-to-base tree transform. See JointModel class
+    updated_λ_H_pre_with_base = jnp.concatenate(
+        (jnp.eye(4).reshape(1, 4, 4), updated_λ_H_pre), axis=0
     )
     # Replace the joint model with the updated transforms
-    updated_joint_model = kin_dyn_params.joint_model.replace(λ_H_pre=updated_λ_H_pre)
+    updated_joint_model = kin_dyn_params.joint_model.replace(
+        λ_H_pre=updated_λ_H_pre_with_base
+    )
 
     # Replace the kin_dyn_parameters with updated values
     updated_kin_dyn_params = kin_dyn_params.replace(

{jaxsim-0.7.1.dev18.dist-info → jaxsim-0.7.1.dev32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jaxsim
-Version: 0.7.1.dev18
+Version: 0.7.1.dev32
 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.7.1.dev18.dist-info → jaxsim-0.7.1.dev32.dist-info}/RECORD

@@ -1,5 +1,5 @@
 jaxsim/__init__.py,sha256=EKeysKN-7UswwJLCl7n6qIBKQIVUtYsCMYu_tCoFn7g,3628
-jaxsim/_version.py,sha256=TLs9huVO9q-y4NJCdgDEyChpVIXEhTeSUWBiVJbQ3AU,526
+jaxsim/_version.py,sha256=CRJMS8F-10dpwjKoRwLGUghZ4-YrOJ9deKAbdUr0Q9g,526
 jaxsim/exceptions.py,sha256=MQ3LRMfVMX2-g3qYj7mUVNV9OLlIA48TANJegbcQyXI,2641
 jaxsim/logging.py,sha256=STI-D_upXZYX-ZezLrlJJ0UlD5YspST0vZ_DcIwkzO4,1553
 jaxsim/typing.py,sha256=7msl8t5Jt09RNYfKdPJtpjLfWurldcycDappb045Eso,761
@@ -12,9 +12,9 @@ jaxsim/api/data.py,sha256=9pxug2gFIZPwqi9kNYXhEziA5IQBB9KNNwIfPfc_kAU,23249
 jaxsim/api/frame.py,sha256=4wg6GsyBQgYhSvc-ry_31JsaL66sZt3TtgwjB7NrHmk,14583
 jaxsim/api/integrators.py,sha256=sHdTWw2Z-Q7jggA8zRkA1KYYd4HNIozXPwNvFwt0YBs,9011
 jaxsim/api/joint.py,sha256=AnqlNWmBOay-gsoo0y4AbfFQ2OCJm-8T1E0IMhZeLoY,7457
-jaxsim/api/kin_dyn_parameters.py,sha256=r1rOjKfe7Qg7xBWyCNtlAi_Wg1WTzToQrUdbZ7PO10Q,39073
+jaxsim/api/kin_dyn_parameters.py,sha256=VrQvH5MXxyiZRkUtXrm99bAZ8bs9Ry02uWMYc-5ZL9Q,39263
 jaxsim/api/link.py,sha256=bSZOYJDY9HJMgY25VzevTTsdFZTJc6yRRpslc5FhGHE,12782
-jaxsim/api/model.py,sha256=ppdriJBHWJ-qXey9Vjqnd7IjqdEE0R6W9ZG9Y7KuS2s,85460
+jaxsim/api/model.py,sha256=Yhcf2ktTO8Ja3XLDCuh8Xc88FXI92801xiSMHCIFv1o,86126
 jaxsim/api/ode.py,sha256=fp20_LK9lXw2DfNkQgrfJmtd_iBXDNzZkOn0u5Pm8Qw,6190
 jaxsim/api/references.py,sha256=-vd50y3v-jkXAsILS432etIKV6e2EUE2oOeLHuUrfuQ,20399
 jaxsim/math/__init__.py,sha256=dNozvtm8WsB7nxw4uK29yQQKPcDUEczr2zcHoZfMItc,383
@@ -65,8 +65,8 @@ jaxsim/utils/__init__.py,sha256=Y5zyoRevl3EMVQadhZ4EtSwTEkDt2vcnFoRhPJjKTZ0,215
 jaxsim/utils/jaxsim_dataclass.py,sha256=XzmZeIibcaOzaxpprsGSxH3UrM66PAO456rFV91sNXg,11453
 jaxsim/utils/tracing.py,sha256=Btwxdfhb7fJLk3r5PlQkGYj60Y2KbFT1gANGIA697FU,530
 jaxsim/utils/wrappers.py,sha256=3IMwydqFgmSPqeuUQ3PRmdhDc1IoT6XC23jPC_LjWXs,4175
-jaxsim-0.7.1.dev18.dist-info/licenses/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
-jaxsim-0.7.1.dev18.dist-info/METADATA,sha256=I0xiLffISlyNbHD7sMWPYHeQbxaA3dHG1FCVLAGZW5I,17851
-jaxsim-0.7.1.dev18.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-jaxsim-0.7.1.dev18.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
-jaxsim-0.7.1.dev18.dist-info/RECORD,,
+jaxsim-0.7.1.dev32.dist-info/licenses/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
+jaxsim-0.7.1.dev32.dist-info/METADATA,sha256=Jf7hLqf34wgxQaT7ZNW74eNpTpYo3Hr2fXEwPro1o1w,17851
+jaxsim-0.7.1.dev32.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+jaxsim-0.7.1.dev32.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
+jaxsim-0.7.1.dev32.dist-info/RECORD,,

{jaxsim-0.7.1.dev18.dist-info → jaxsim-0.7.1.dev32.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.4.0)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any