Add get joint modes (stepjam#217)

* get joint modes * Update robot_component.py
imitrob · Nov 2, 2020 · 7b2a75c · 7b2a75c
1 parent 5958765
commit 7b2a75c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pyrep/robots/robot_component.py b/pyrep/robots/robot_component.py
@@ -246,6 +246,13 @@ def set_joint_mode(self, value: JointMode) -> None:
         [j.set_joint_mode(value)  # type: ignore
          for j in self.joints]
 
+    def get_joint_modes(self) -> List[JointMode]:
+        """Gets the operation mode of the joint group.
+
+        :return: A list of joint modes.
+        """
+        return [j.get_joint_mode() for j in self.joints]
+
     def get_visuals(self) -> List[Object]:
         """Gets a list of the visual elements of this component.