Update xr import logic, update FrankaLeap

StanfordVL · Nov 20, 2023 · e2295a4 · e2295a4
1 parent ca024ce
commit e2295a4
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 5 deletions.
diff --git a/omnigibson/__init__.py b/omnigibson/__init__.py
@@ -86,7 +86,6 @@ def create_app():
     from omni.isaac.core.utils.extensions import enable_extension
     enable_extension("omni.flowusd")
     enable_extension("omni.particle.system.bundle")
-    enable_extension("omni.kit.xr.profile.vr")
 
     # Additional import for windows
     if os.name == "nt":

diff --git a/omnigibson/robots/__init__.py b/omnigibson/robots/__init__.py
@@ -11,5 +11,5 @@
 from omnigibson.robots.two_wheel_robot import TwoWheelRobot
 from omnigibson.robots.franka import FrankaPanda
 from omnigibson.robots.franka_allegro import FrankaAllegro
-from omnigibson.robots.franka_leap import FrankaLeap
+from omnigibson.robots.franka_leap import FrankaLeap, FrankaLeapLeft
 from omnigibson.robots.behaviorbot import Behaviorbot
diff --git a/omnigibson/robots/franka_leap.py b/omnigibson/robots/franka_leap.py
@@ -9,7 +9,7 @@
 
 class FrankaLeap(ManipulationRobot):
     """
-    Franka Robot with Leap hand
+    Franka Robot with Leap right hand
     """
 
     def __init__(
@@ -193,15 +193,15 @@ def finger_joint_names(self):
 
     @property
     def usd_path(self):
-        return os.path.join(gm.ASSET_PATH, "models/franka/franka_leap.usd")
+        return os.path.join(gm.ASSET_PATH, "models/franka/franka_panda_leap.usd")
 
     @property
     def robot_arm_descriptor_yamls(self):
         return {self.default_arm: os.path.join(gm.ASSET_PATH, "models/franka/franka_leap_description.yaml")}
 
     @property
     def urdf_path(self):
-        return os.path.join(gm.ASSET_PATH, "models/franka/franka_leap.urdf")
+        return os.path.join(gm.ASSET_PATH, "models/franka/franka_panda_leap.urdf")
 
     @property
     def gripper_urdf_path(self):
@@ -217,3 +217,24 @@ def gen_action_from_vr_data(self, vr_data: dict):
         action[:6] = action_from_controller[:6]
         action[6:] = action_from_controller[6]
         return action
+
+
+class FrankaLeapLeft(FrankaLeap):
+    """
+    Franka Robot with Leap left hand
+    """
+    @property
+    def model_name(self):
+        return "FrankaLeapLeft"    
+
+    @property
+    def usd_path(self):
+        return os.path.join(gm.ASSET_PATH, "models/franka/franka_panda_leap_left.usd")
+
+    @property
+    def robot_arm_descriptor_yamls(self):
+        return {self.default_arm: os.path.join(gm.ASSET_PATH, "models/franka/franka_leap_description.yaml")}
+
+    @property
+    def urdf_path(self):
+        return os.path.join(gm.ASSET_PATH, "models/franka/franka_panda_leap_left.urdf")
diff --git a/omnigibson/utils/xr_utils.py b/omnigibson/utils/xr_utils.py
@@ -33,6 +33,10 @@ def __init__(
         NOTE: enable_touchpad_movement and align_anchor_to_robot_base cannot be enabled at the same time. 
             The former is to enable free movement of the VR system (i.e. the user), while the latter is constraining the VR system to the robot pose.
         """
+        # enable xr extension
+        from omni.isaac.core.utils.extensions import enable_extension
+        enable_extension("omni.kit.xr.profile.vr")
+        # get xr core and profile
         self.xr_core = XRCore.get_singleton()
         self.vr_profile = self.xr_core.get_profile("vr")
         self.disable_display_output = disable_display_output