example-robot-data-loaders 6.0.0__2-py313-none-any.whl

cmeel.prefix/lib/python3.13/site-packages/example_robot_data/panda.py

@@ -0,0 +1,70 @@
+from os.path import join
+
+import numpy as np
+import pinocchio as pin
+
+try:
+    import coal
+except ImportError:
+    coal = None
+
+from .utils import RobotLoader, getModelPath
+
+
+class PandaLoader(RobotLoader):
+    path = "panda_description"
+    urdf_filename = "panda.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "panda.srdf"
+    ref_posture = "default"
+
+
+class PandaLoaderCollision(PandaLoader):
+    urdf_filename = "panda_collision.urdf"
+
+    def __init__(self, verbose=False):
+        super().__init__(verbose=verbose)
+
+        self.srdf_path = None
+        self.robot.q0 = pin.neutral(self.robot.model)
+        root = getModelPath(self.path)
+        self.robot.urdf = join(root, self.path, self.urdf_subpath, self.urdf_filename)
+
+        # If coal is not available, gracefully skip collision edits
+        if coal is None or not getattr(pin, "WITH_COLLISION", True):
+            print(
+                "[PandaLoaderCollision] coal not available - skipping collision geometry processing."
+            )
+            return
+
+        cmodel = self.robot.collision_model.copy()
+        list_names_capsules = []
+        # Iterate through geometry objects in the collision model
+        for geom_object in cmodel.geometryObjects:
+            geometry = geom_object.geometry
+            # Remove superfluous suffix from the name
+            base_name = "_".join(geom_object.name.split("_")[:-1])
+
+            # Convert cylinders to capsules
+            if isinstance(geometry, coal.Cylinder):
+                name = self.generate_capsule_name(base_name, list_names_capsules)
+                list_names_capsules.append(name)
+                capsule = pin.GeometryObject(
+                    name=name,
+                    parent_frame=int(geom_object.parentFrame),
+                    parent_joint=int(geom_object.parentJoint),
+                    collision_geometry=coal.Capsule(
+                        geometry.radius, geometry.halfLength
+                    ),
+                    placement=geom_object.placement,
+                )
+                capsule.meshColor = np.array([249, 136, 126, 125]) / 255  # Red color
+                self.robot.collision_model.addGeometryObject(capsule)
+                self.robot.collision_model.removeGeometryObject(geom_object.name)
+
+            # Remove spheres associated with links
+            elif isinstance(geometry, coal.Sphere) and "link" in geom_object.name:
+                self.robot.collision_model.removeGeometryObject(geom_object.name)
+
+        # Recreate collision data since the collision pairs changed
+        self.robot.collision_data = self.robot.collision_model.createData()

cmeel.prefix/lib/python3.13/site-packages/example_robot_data/path.py

@@ -0,0 +1,2 @@
+EXAMPLE_ROBOT_DATA_MODEL_DIR = "/tmp/pip-build-env-wl0om5xf/overlay/lib/python3.13/site-packages/cmeel.prefix/share/example-robot-data/robots"
+EXAMPLE_ROBOT_DATA_SOURCE_DIR = "/home/runner/work/example-robot-data-loaders/example-robot-data-loaders/robots"

cmeel.prefix/lib/python3.13/site-packages/example_robot_data/robots_loader.py

@@ -0,0 +1,526 @@
+import typing
+
+import pinocchio as pin
+
+try:
+    from .path import (  # noqa: F401
+        EXAMPLE_ROBOT_DATA_MODEL_DIR,
+        EXAMPLE_ROBOT_DATA_SOURCE_DIR,
+    )
+except ImportError:
+    pass
+
+from .human import HumanLoader
+from .panda import PandaLoader, PandaLoaderCollision
+from .talos import (
+    TalosArmLoader,
+    TalosBoxLoader,
+    TalosFullBoxLoader,
+    TalosFullLoader,
+    TalosLegsLoader,
+    TalosLoader,
+)
+from .utils import RobotLoader, getModelPath, readParamsFromSrdf  # noqa: F401
+
+
+class CentauroLoader(RobotLoader):
+    path = "centauro_description"
+    urdf_filename = "centauro.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "centauro.srdf"
+    ref_posture = "homing_balanced"
+    free_flyer = True
+
+
+class B1Loader(RobotLoader):
+    path = "b1_description"
+    urdf_filename = "b1.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "b1.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class Go1Loader(RobotLoader):
+    path = "go1_description"
+    urdf_filename = "go1.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "go1.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class Go2Loader(RobotLoader):
+    path = "go2_description"
+    urdf_filename = "go2.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "go2.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class A1Loader(RobotLoader):
+    path = "a1_description"
+    urdf_filename = "a1.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "a1.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class G1Loader(RobotLoader):
+    path = "g1_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "g1_29dof_rev_1_0.urdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class G1WithHandsLoader(G1Loader):
+    path = "g1_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "g1_29dof_with_hand_rev_1_0.urdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class Z1Loader(RobotLoader):
+    path = "z1_description"
+    urdf_filename = "z1.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "z1.srdf"
+    ref_posture = "arm_up"
+
+
+class B1Z1Loader(B1Loader):
+    urdf_filename = "b1-z1.urdf"
+    srdf_filename = "b1-z1.srdf"
+    ref_posture = "standing_with_arm_home"
+
+
+class ANYmalLoader(RobotLoader):
+    path = "anymal_b_simple_description"
+    urdf_filename = "anymal.urdf"
+    srdf_filename = "anymal.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class ANYmalCLoader(RobotLoader):
+    path = "anymal_c_simple_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "anymal.urdf"
+    srdf_filename = "anymal.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class LaikagoLoader(RobotLoader):
+    path = "laikago_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "laikago.urdf"
+    free_flyer = True
+
+
+class ANYmalKinovaLoader(ANYmalLoader):
+    urdf_filename = "anymal-kinova.urdf"
+    srdf_filename = "anymal-kinova.srdf"
+    ref_posture = "standing_with_arm_up"
+
+
+class BaxterLoader(RobotLoader):
+    path = "baxter_description"
+    urdf_filename = "baxter.urdf"
+    srdf_filename = "baxter_manipulation.srdf"
+    urdf_subpath = "urdf"
+    srdf_subpath = "srdf"
+    ref_posture = "neutral"
+
+
+class CassieLoader(RobotLoader):
+    path = "cassie_description"
+    if tuple(int(i) for i in pin.__version__.split(".")) > (2, 9, 1):
+        sdf_filename = "cassie.sdf"
+    else:
+        sdf_filename = "cassie_v2.sdf"
+    sdf_subpath = "robots"
+    srdf_filename = "cassie_v2.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+    sdf_root_link_name = "pelvis"
+    sdf_parent_guidance: typing.ClassVar = [
+        "left-roll-op",
+        "left-yaw-op",
+        "left-pitch-op",
+        "left-knee-op",
+        "left-tarsus-spring-joint",
+        "left-foot-op",
+        "right-roll-op",
+        "right-yaw-op",
+        "right-pitch-op",
+        "right-knee-op",
+        "right-tarsus-spring-joint",
+        "right-foot-op",
+    ]
+
+
+class AsrTwoDofLoader(RobotLoader):
+    path = "asr_twodof_description"
+    urdf_filename = "TwoDofs.urdf"
+    urdf_subpath = "urdf"
+
+
+class HyQLoader(RobotLoader):
+    path = "hyq_description"
+    urdf_filename = "hyq_no_sensors.urdf"
+    srdf_filename = "hyq.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class BoltLoader(RobotLoader):
+    path = "bolt_description"
+    urdf_filename = "bolt.urdf"
+    srdf_filename = "bolt.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class BorinotLoader(RobotLoader):
+    path = "borinot_description"
+    urdf_subpath = "urdf"
+    srdf_subpath = "srdf"
+    urdf_filename = "borinot_flying_arm_2.urdf"
+    srdf_filename = "borinot_flying_arm_2.srdf"
+    ref_posture = "home"
+    free_flyer = True
+
+
+class Solo8Loader(RobotLoader):
+    path = "solo_description"
+    urdf_filename = "solo.urdf"
+    srdf_filename = "solo.srdf"
+    ref_posture = "standing"
+    free_flyer = True
+
+
+class Solo12Loader(Solo8Loader):
+    urdf_filename = "solo12.urdf"
+
+
+class FingerEduLoader(RobotLoader):
+    path = "finger_edu_description"
+    urdf_filename = "finger_edu.urdf"
+    srdf_filename = "finger_edu.srdf"
+    ref_posture = "hanging"
+    free_flyer = False
+
+
+class KinovaLoader(RobotLoader):
+    path = "kinova_description"
+    urdf_filename = "kinova.urdf"
+    srdf_filename = "kinova.srdf"
+    ref_posture = "arm_up"
+
+
+class TiagoLoader(RobotLoader):
+    path = "tiago_description"
+    urdf_filename = "tiago.urdf"
+
+
+class TiagoDualLoader(TiagoLoader):
+    urdf_filename = "tiago_dual.urdf"
+
+
+class TiagoNoHandLoader(TiagoLoader):
+    urdf_filename = "tiago_no_hand.urdf"
+
+
+class TiagoProLoader(RobotLoader):
+    path = "tiago_pro_description"
+    urdf_filename = "tiago_pro.urdf"
+
+
+class ICubLoader(RobotLoader):
+    path = "icub_description"
+    urdf_filename = "icub.urdf"
+    srdf_filename = "icub.srdf"
+    free_flyer = True
+
+
+class ICubReducedLoader(ICubLoader):
+    urdf_filename = "icub_reduced.urdf"
+
+
+class AlexanderLoader(RobotLoader):
+    path = "alexander_description"
+    urdf_filename = "alexander_v1.lowerBodyOnly.urdf"
+    urdf_subpath = "urdf"
+    free_flyer = True
+    ref_posture = "default"
+    srdf_filename = "alexander.srdf"
+
+
+class AlexNubHandsLoader(RobotLoader):
+    path = "alex_description"
+    urdf_filename = "alex_nub_hands.urdf"
+    urdf_subpath = "urdf"
+    ref_posture = "default"
+
+
+class AlexPsyonicHandsLoader(RobotLoader):
+    path = "alex_description"
+    urdf_filename = "alex_psyonic_hands.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "alex_psyonic_hands.srdf"
+    ref_posture = "default"
+
+
+class AlexSakeHandsLoader(RobotLoader):
+    path = "alex_description"
+    urdf_filename = "alex_sake_hands.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "alex_sake_hands.srdf"
+    ref_posture = "default"
+
+
+class AllegroRightHandLoader(RobotLoader):
+    path = "allegro_hand_description"
+    urdf_filename = "allegro_right_hand.urdf"
+    urdf_subpath = "urdf"
+
+
+class AllegroLeftHandLoader(RobotLoader):
+    path = "allegro_hand_description"
+    urdf_filename = "allegro_left_hand.urdf"
+    urdf_subpath = "urdf"
+
+
+class UR3Loader(RobotLoader):
+    path = "ur_description"
+    urdf_filename = "ur3_robot.urdf"
+    urdf_subpath = "urdf"
+    ref_posture = None
+
+
+class UR3GripperLoader(UR3Loader):
+    urdf_filename = "ur3_gripper.urdf"
+    srdf_filename = "ur3_gripper.srdf"
+
+
+class UR3LimitedLoader(UR3Loader):
+    urdf_filename = "ur3_joint_limited_robot.urdf"
+
+
+class UR5Loader(UR3Loader):
+    urdf_filename = "ur5_robot.urdf"
+    srdf_filename = "ur5.srdf"
+
+
+class UR5GripperLoader(UR5Loader):
+    urdf_filename = "ur5_gripper.urdf"
+    srdf_filename = "ur5_gripper.srdf"
+
+
+class UR5LimitedLoader(UR5Loader):
+    urdf_filename = "ur5_joint_limited_robot.urdf"
+
+
+class UR10Loader(UR3Loader):
+    urdf_filename = "ur10_robot.urdf"
+
+
+class UR10LimitedLoader(UR10Loader):
+    urdf_filename = "ur10_joint_limited_robot.urdf"
+
+
+class HectorLoader(RobotLoader):
+    path = "hector_description"
+    urdf_filename = "quadrotor_base.urdf"
+    free_flyer = True
+
+
+class HextiltLoader(RobotLoader):
+    path = "hextilt_description"
+    urdf_subpath = "urdf"
+    srdf_subpath = "srdf"
+    urdf_filename = "hextilt_flying_arm_5.urdf"
+    srdf_filename = "hextilt_flying_arm_5.srdf"
+    ref_posture = "home"
+    free_flyer = True
+
+
+class DoublePendulumLoader(RobotLoader):
+    path = "double_pendulum_description"
+    urdf_filename = "double_pendulum.urdf"
+    urdf_subpath = "urdf"
+
+
+class DoublePendulumContinuousLoader(DoublePendulumLoader):
+    urdf_filename = "double_pendulum_continuous.urdf"
+
+
+class DoublePendulumSimpleLoader(DoublePendulumLoader):
+    urdf_filename = "double_pendulum_simple.urdf"
+
+
+class QuadrupedLoader(RobotLoader):
+    path = "quadruped_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "quadruped.urdf"
+    free_flyer = True
+
+
+class RomeoLoader(RobotLoader):
+    path = "romeo_description"
+    urdf_filename = "romeo.urdf"
+    urdf_subpath = "urdf"
+    free_flyer = True
+
+
+class SimpleHumanoidLoader(RobotLoader):
+    path = "simple_humanoid_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "simple_humanoid.urdf"
+    srdf_filename = "simple_humanoid.srdf"
+    free_flyer = True
+
+
+class SimpleHumanoidClassicalLoader(SimpleHumanoidLoader):
+    urdf_filename = "simple_humanoid_classical.urdf"
+    srdf_filename = "simple_humanoid_classical.srdf"
+
+
+class IrisLoader(RobotLoader):
+    path = "iris_description"
+    urdf_filename = "iris_simple.urdf"
+    free_flyer = True
+
+
+class SO100Loader(RobotLoader):
+    path = "so_arm_description"
+    urdf_filename = "so100.urdf"
+    urdf_subpath = "urdf"
+    free_flyer = True
+
+
+class SO101Loader(RobotLoader):
+    path = "so_arm_description"
+    urdf_filename = "so101.urdf"
+    urdf_subpath = "urdf"
+    free_flyer = True
+
+
+class PR2Loader(RobotLoader):
+    path = "pr2_description"
+    urdf_filename = "pr2.urdf"
+    urdf_subpath = "urdf"
+    srdf_filename = "pr2.srdf"
+    free_flyer = True
+    ref_posture = "tuck_left_arm"
+
+
+class xArm7Loader(RobotLoader):
+    path = "xarm_description"
+    urdf_subpath = "urdf"
+    urdf_filename = "xarm7.urdf"
+    srdf_filename = "xarm7.srdf"
+    ref_posture = "home"
+
+
+ROBOTS = {
+    "centauro": CentauroLoader,
+    "b1": B1Loader,
+    "go1": Go1Loader,
+    "go2": Go2Loader,
+    "a1": A1Loader,
+    "g1": G1Loader,
+    "g1_with_hands": G1WithHandsLoader,
+    "z1": Z1Loader,
+    "b1_z1": B1Z1Loader,
+    "anymal": ANYmalLoader,
+    "anymal_c": ANYmalCLoader,
+    "anymal_kinova": ANYmalKinovaLoader,
+    "asr_twodof": AsrTwoDofLoader,
+    "baxter": BaxterLoader,
+    "cassie": CassieLoader,
+    "double_pendulum": DoublePendulumLoader,
+    "double_pendulum_continuous": DoublePendulumContinuousLoader,
+    "double_pendulum_simple": DoublePendulumSimpleLoader,
+    "hector": HectorLoader,
+    "hextilt": HextiltLoader,
+    "human": HumanLoader,
+    "hyq": HyQLoader,
+    "icub": ICubLoader,
+    "icub_reduced": ICubReducedLoader,
+    "iris": IrisLoader,
+    "kinova": KinovaLoader,
+    "laikago": LaikagoLoader,
+    "panda": PandaLoader,
+    "panda_collision": PandaLoaderCollision,
+    "alexander": AlexanderLoader,
+    "alex_nub_hands": AlexNubHandsLoader,
+    "alex_psyonic_hands": AlexPsyonicHandsLoader,
+    "alex_sake_hands": AlexSakeHandsLoader,
+    "allegro_right_hand": AllegroRightHandLoader,
+    "allegro_left_hand": AllegroLeftHandLoader,
+    "quadruped": QuadrupedLoader,
+    "romeo": RomeoLoader,
+    "simple_humanoid": SimpleHumanoidLoader,
+    "simple_humanoid_classical": SimpleHumanoidClassicalLoader,
+    "bolt": BoltLoader,
+    "borinot": BorinotLoader,
+    "so100": SO100Loader,
+    "so101": SO101Loader,
+    "solo8": Solo8Loader,
+    "solo12": Solo12Loader,
+    "finger_edu": FingerEduLoader,
+    "pr2": PR2Loader,
+    "talos": TalosLoader,
+    "talos_box": TalosBoxLoader,
+    "talos_arm": TalosArmLoader,
+    "talos_legs": TalosLegsLoader,
+    "talos_full": TalosFullLoader,
+    "talos_full_box": TalosFullBoxLoader,
+    "tiago": TiagoLoader,
+    "tiago_dual": TiagoDualLoader,
+    "tiago_no_hand": TiagoNoHandLoader,
+    "tiago_pro": TiagoProLoader,
+    "ur3": UR5Loader,
+    "ur3_gripper": UR3GripperLoader,
+    "ur3_limited": UR3LimitedLoader,
+    "ur5": UR5Loader,
+    "ur5_gripper": UR5GripperLoader,
+    "ur5_limited": UR5LimitedLoader,
+    "ur10": UR10Loader,
+    "ur10_limited": UR10LimitedLoader,
+    "xarm7": xArm7Loader,
+}
+
+
+def loader(name, display=False, rootNodeName="", verbose=False):
+    """Load a robot by its name, and optionally display it in a viewer."""
+    if name not in ROBOTS:
+        robots = ", ".join(sorted(ROBOTS.keys()))
+        raise ValueError(f"Robot '{name}' not found. Possible values are {robots}")
+    inst = ROBOTS[name](verbose=verbose)
+    if display:
+        if rootNodeName:
+            inst.robot.initViewer()
+            inst.robot.viz.loadViewerModel(rootNodeName=rootNodeName)
+        else:
+            inst.robot.initViewer(loadModel=True)
+        inst.robot.display(inst.robot.q0)
+    return inst
+
+
+def load(name, display=False, rootNodeName="", verbose=False):
+    """Load a robot by its name, and optionnaly display it in a viewer."""
+    return loader(name, display, rootNodeName, verbose).robot
+
+
+def load_full(name, display=False, rootNodeName="", verbose=False):
+    """Load a robot by its name, optionnaly display it in a viewer,
+    and provide its q0 and paths."""
+    inst = loader(name, display, rootNodeName, verbose)
+    return inst.robot, inst.robot.q0, inst.df_path, inst.srdf_path

cmeel.prefix/lib/python3.13/site-packages/example_robot_data/talos.py

@@ -0,0 +1,99 @@
+import numpy as np
+import pinocchio as pin
+
+from .utils import RobotLoader, readParamsFromSrdf
+
+
+class TalosLoader(RobotLoader):
+    path = "talos_data"
+    urdf_filename = "talos_reduced.urdf"
+    srdf_filename = "talos.srdf"
+    free_flyer = True
+    has_rotor_parameters = True
+
+
+class TalosBoxLoader(TalosLoader):
+    urdf_filename = "talos_reduced_box.urdf"
+
+
+class TalosFullLoader(TalosLoader):
+    urdf_filename = "talos_full_v2.urdf"
+
+
+class TalosFullBoxLoader(TalosLoader):
+    urdf_filename = "talos_full_v2_box.urdf"
+
+
+class TalosArmLoader(TalosLoader):
+    urdf_filename = "talos_left_arm.urdf"
+    free_flyer = False
+
+
+class TalosLegsLoader(TalosLoader):
+    def __init__(self, verbose=False):
+        super().__init__(verbose=verbose)
+        legMaxId = 14
+        m1 = self.robot.model
+        m2 = pin.Model()
+        for j, M, name, parent, Y in zip(
+            m1.joints, m1.jointPlacements, m1.names, m1.parents, m1.inertias
+        ):
+            if j.id < legMaxId:
+                jid = m2.addJoint(parent, getattr(pin, j.shortname())(), M, name)
+                idx_q, idx_v = m2.joints[jid].idx_q, m2.joints[jid].idx_v
+                m2.upperPositionLimit[idx_q : idx_q + j.nq] = m1.upperPositionLimit[
+                    j.idx_q : j.idx_q + j.nq
+                ]
+                m2.lowerPositionLimit[idx_q : idx_q + j.nq] = m1.lowerPositionLimit[
+                    j.idx_q : j.idx_q + j.nq
+                ]
+                m2.velocityLimit[idx_v : idx_v + j.nv] = m1.velocityLimit[
+                    j.idx_v : j.idx_v + j.nv
+                ]
+                m2.effortLimit[idx_v : idx_v + j.nv] = m1.effortLimit[
+                    j.idx_v : j.idx_v + j.nv
+                ]
+                assert jid == j.id
+                m2.appendBodyToJoint(jid, Y, pin.SE3.Identity())
+
+        upperPos = m2.upperPositionLimit
+        upperPos[:7] = 1
+        m2.upperPositionLimit = upperPos
+        lowerPos = m2.lowerPositionLimit
+        lowerPos[:7] = -1
+        m2.lowerPositionLimit = lowerPos
+        effort = m2.effortLimit
+        effort[:6] = np.inf
+        m2.effortLimit = effort
+
+        # q2 = self.robot.q0[:19]
+        for f in m1.frames:
+            if tuple(int(i) for i in pin.__version__.split(".")) >= (3, 0, 0):
+                if f.parentJoint < legMaxId:
+                    m2.addFrame(f)
+            elif f.parent < legMaxId:
+                m2.addFrame(f)
+
+        g2 = pin.GeometryModel()
+        for g in self.robot.visual_model.geometryObjects:
+            if g.parentJoint < 14:
+                g2.addGeometryObject(g)
+
+        self.robot.model = m2
+        self.robot.data = m2.createData()
+        self.robot.visual_model = g2
+        # self.robot.q0=q2
+        self.robot.visual_data = pin.GeometryData(g2)
+
+        # Load SRDF file
+        self.robot.q0 = readParamsFromSrdf(
+            self.robot.model,
+            self.srdf_path,
+            self.verbose,
+            self.has_rotor_parameters,
+            self.ref_posture,
+        )
+
+        assert (m2.armature[:6] == 0.0).all()
+        # Add the free-flyer joint limits to the new model
+        self.addFreeFlyerJointLimits()