jaxsim 0.5.1.dev164__py3-none-any.whl → 0.5.1.dev166__py3-none-any.whl

jaxsim/_version.py

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

jaxsim/mujoco/__init__.py

@@ -1,4 +1,4 @@
-from .loaders import RodModelToMjcf, SdfToMjcf, UrdfToMjcf
+from .loaders import ModelToMjcf, RodModelToMjcf, SdfToMjcf, UrdfToMjcf
 from .model import MujocoModelHelper
 from .utils import mujoco_data_from_jaxsim
 from .visualizer import MujocoVideoRecorder, MujocoVisualizer

jaxsim/mujoco/loaders.py

@@ -9,6 +9,8 @@ import numpy as np
 import rod.urdf.exporter
 from lxml import etree as ET
 
+from jaxsim import logging
+
 from .utils import MujocoCamera
 
 MujocoCameraType = (
@@ -61,6 +63,59 @@ def load_rod_model(
     return models[model_name]
 
 
+class ModelToMjcf:
+    """
+    Class to convert a URDF/SDF file or a ROD model to a Mujoco MJCF string.
+    """
+
+    @staticmethod
+    def convert(
+        model: str | pathlib.Path | rod.Model,
+        considered_joints: list[str] | None = None,
+        plane_normal: tuple[float, float, float] = (0, 0, 1),
+        heightmap: bool | None = None,
+        heightmap_samples_xy: tuple[int, int] = (101, 101),
+        cameras: MujocoCameraType = (),
+    ) -> tuple[str, dict[str, Any]]:
+        """
+        Convert a model to a Mujoco MJCF string.
+
+        Args:
+            model: The URDF/SDF file or ROD model to convert.
+            considered_joints: The list of joint names to consider in the conversion.
+            plane_normal: The normal vector of the plane.
+            heightmap: Whether to generate a heightmap.
+            heightmap_samples_xy: The number of points in the heightmap grid.
+            cameras: The custom cameras to add to the scene.
+
+        Returns:
+            A tuple containing the MJCF string and the dictionary of assets.
+        """
+
+        match model:
+            case rod.Model():
+                rod_model = model
+            case str() | pathlib.Path():
+                # Convert the JaxSim model to a ROD model.
+                rod_model = load_rod_model(
+                    model_description=model,
+                    is_urdf=None,
+                    model_name=None,
+                )
+            case _:
+                raise TypeError(f"Unsupported type for 'model': {type(model)}")
+
+        # Convert the ROD model to MJCF.
+        return RodModelToMjcf.convert(
+            rod_model=rod_model,
+            considered_joints=considered_joints,
+            plane_normal=plane_normal,
+            heightmap=heightmap,
+            heightmap_samples_xy=heightmap_samples_xy,
+            cameras=cameras,
+        )
+
+
 class RodModelToMjcf:
     """
     Class to convert a ROD model to a Mujoco MJCF string.
@@ -552,6 +607,8 @@ class UrdfToMjcf:
             tuple: A tuple containing the MJCF string and the assets dictionary.
         """
 
+        logging.warning("This method is deprecated. Use 'ModelToMjcf.convert' instead.")
+
         # Get the ROD model.
         rod_model = load_rod_model(
             model_description=urdf,
@@ -598,6 +655,8 @@ class SdfToMjcf:
             tuple: A tuple containing the MJCF string and the assets dictionary.
         """
 
+        logging.warning("This method is deprecated. Use 'ModelToMjcf.convert' instead.")
+
         # Get the ROD model.
         rod_model = load_rod_model(
             model_description=sdf,

{jaxsim-0.5.1.dev164.dist-info → jaxsim-0.5.1.dev166.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jaxsim
-Version: 0.5.1.dev164
+Version: 0.5.1.dev166
 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.dev164.dist-info → jaxsim-0.5.1.dev166.dist-info}/RECORD

@@ -1,5 +1,5 @@
 jaxsim/__init__.py,sha256=_8rbKOf3bwx-2ChEbspZxs_rZY0RqUcmWAftnEw1bfM,3401
-jaxsim/_version.py,sha256=HBJFyRR4TSVLpDUSzqSMToGIY0oGpnNcNj-DP-ll5I0,428
+jaxsim/_version.py,sha256=kGOL2jvMIPzXvzoNfMwZDyAgJVTdfsfiFG-sWppA4F0,428
 jaxsim/exceptions.py,sha256=qjfTjE9lXvD3-JCPQcxxiX2XSS8QegawzQ6ZuC2tc0Y,2638
 jaxsim/logging.py,sha256=STI-D_upXZYX-ZezLrlJJ0UlD5YspST0vZ_DcIwkzO4,1553
 jaxsim/typing.py,sha256=7msl8t5Jt09RNYfKdPJtpjLfWurldcycDappb045Eso,761
@@ -30,9 +30,9 @@ jaxsim/math/rotation.py,sha256=bl9WCbYyLKg6RyRkMaEBBTmARBs8pB-FGR0JVbfbaNE,2187
 jaxsim/math/skew.py,sha256=P82yeQs9Fzb7Ri_MAikcb54_06cE_syi9yPssSg4ydw,1426
 jaxsim/math/transform.py,sha256=We0ChLajSckxGINiJsP1a5Ur3yjg3JuweQ3kK4Woix4,3332
 jaxsim/math/utils.py,sha256=2id1F6QOvkHkIF3Nuxuj_tz_kI0IYlrlgVQrETmXFfI,1058
-jaxsim/mujoco/__init__.py,sha256=fZyRWre49pIhOrYdf6yJk_hOax8qWGe8OCmoq-dMVq8,201
+jaxsim/mujoco/__init__.py,sha256=1kAWzYOS7nP29S5FGyWPqiAnPf4yPSoaPW-WBGBjVV0,214
 jaxsim/mujoco/__main__.py,sha256=GBmB7J-zj75ZnFyuAAmpSOpbxi_HhHhWJeot3ljGDJY,5291
-jaxsim/mujoco/loaders.py,sha256=lHU-Oc2hyYca2eXIWmkhQAcX7vAvxAVTmBFsEqg3fL4,21066
+jaxsim/mujoco/loaders.py,sha256=Jjb5Us9ERmLjejM4S1FcqrF12ZVkjBMZXelu9n6HGA4,23138
 jaxsim/mujoco/model.py,sha256=tZWn2gpZSpQtwS5v7O5rGdjYNcEU6rnfAS6_ZKnZagE,16478
 jaxsim/mujoco/utils.py,sha256=6vYXNXmyI7lf2cp49cq9srClrPsbHmBGgdcPP1kux8M,8396
 jaxsim/mujoco/visualizer.py,sha256=wb9Nzrlf-IAkRa4RgSNWswb3DxjnLKnOFTPfklUMpmU,7212
@@ -67,8 +67,8 @@ jaxsim/utils/__init__.py,sha256=Y5zyoRevl3EMVQadhZ4EtSwTEkDt2vcnFoRhPJjKTZ0,215
 jaxsim/utils/jaxsim_dataclass.py,sha256=Fxa555u14VUsVlKU1rBQFurrVzBp7BNsIaVoNko0lrI,11261
 jaxsim/utils/tracing.py,sha256=Btwxdfhb7fJLk3r5PlQkGYj60Y2KbFT1gANGIA697FU,530
 jaxsim/utils/wrappers.py,sha256=3IMwydqFgmSPqeuUQ3PRmdhDc1IoT6XC23jPC_LjWXs,4175
-jaxsim-0.5.1.dev164.dist-info/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
-jaxsim-0.5.1.dev164.dist-info/METADATA,sha256=ER36MgF_wHAofj6s6aJDAO8ZU7LQSe8tU_WvpzqVMsw,20201
-jaxsim-0.5.1.dev164.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-jaxsim-0.5.1.dev164.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
-jaxsim-0.5.1.dev164.dist-info/RECORD,,
+jaxsim-0.5.1.dev166.dist-info/LICENSE,sha256=eaYdFmdeMbiIoIiPzEK0MjP1S9wtFXjXNR5er49uLR0,1546
+jaxsim-0.5.1.dev166.dist-info/METADATA,sha256=4SzQHAizMPLLi6V3C01bD_C_COGWM-Ha0b8AYUlKcvY,20201
+jaxsim-0.5.1.dev166.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jaxsim-0.5.1.dev166.dist-info/top_level.txt,sha256=LxGMA8FLtXjQ6oI7N5gd_R_oSUHxpXxUEOfT1xS_ni0,7
+jaxsim-0.5.1.dev166.dist-info/RECORD,,