[example] Load in MuJoCo

Author: stephane-caron

CHANGELOG.md

@@ -13,6 +13,7 @@ This release adds 2 robot descriptions:
 
 ### Added
 
+- Example: load in MuJoCo
 - Example: load in Pinocchio
 - Example: load in PyBullet
 - Example: show in MeshCat

README.md

@@ -106,9 +106,16 @@ New robot descriptions are welcome! Check out the [guidelines](CONTRIBUTING.md)
 
 ## Examples
 
-- [MeshCat](examples/show_in_meshcat.py)
+Load a robot description in:
+
+- [MuJoCo](examples/load_in_mujoco.py)
 - [Pinocchio](examples/load_in_pinocchio.py)
-- [PyBullet](examples/show_in_bullet.py)
+- [PyBullet](examples/load_in_pybullet.py)
+
+Visualize a robot description usingf:
+
+- [MeshCat](examples/show_in_meshcat.py)
+- [PyBullet](examples/show_in_pybullet.py)
 - [yourdfpy](examples/show_in_yourdfpy.py)
 
 ## Thanks

examples/load_in_mujoco.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright 2022 Stéphane Caron
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Show a robot descriptions specified from the command line, using yourdfpy.
+
+This example requires MuJoCo, which is installed by ``pip install mujoco``.
+"""
+
+import argparse
+from importlib import import_module  # type: ignore
+
+try:
+    import mujoco
+except ImportError as e:
+    raise ImportError("MuJoCo not found, try ``pip install mujoco``") from e
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("name", help="name of the robot description")
+    args = parser.parse_args()
+
+    try:
+        module = import_module(f"robot_descriptions.{args.name}")
+    except ModuleNotFoundError:
+        module = import_module(f"robot_descriptions.{args.name}_description")
+
+    if not hasattr(module, "MJCF_PATH"):
+        raise ValueError(f"{args.name} is not an MJCF description")
+
+    model = mujoco.MjModel.from_xml_path(module.MJCF_PATH)
+    data = mujoco.MjData(model)
+
+    print(f"Robot successfully loaded with model={model} and data={data}")