Merge branch 'master' of github.com:shbang91/pink into shbang91-master

examples/draco3.py

@@ -0,0 +1,184 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright 2023 Inria
+#
+# 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.
+
+"""DRACO 3 humanoid standing on two feet and reaching with a hand."""
+
+import os
+import sys
+
+import meshcat_shapes
+import numpy as np
+import pinocchio as pin
+import qpsolvers
+from loop_rate_limiters import RateLimiter
+
+cwd = os.getcwd()
+sys.path.append(cwd)
+import pink
+from pink import solve_ik
+from pink.tasks import (
+    BodyTask,
+    JointCouplingTask,
+    PostureTask,
+)
+
+try:
+    from robot_descriptions.loaders.pinocchio import load_robot_description
+except ModuleNotFoundError:
+    raise ModuleNotFoundError(
+        "Examples need robot_descriptions, "
+        "try `pip install robot_descriptions`"
+    )
+
+
+class WavingPose:
+    """Moving target to the wave the right hand."""
+
+    def __init__(self, init: pin.SE3):
+        """Initialize pose.
+
+        Args:
+            init: Initial transform from the wrist frame to the world frame.
+        """
+        self.init = init
+
+    def at(self, t):
+        """Get waving pose at a given time.
+
+        Args:
+            t: Time in seconds.
+        """
+        T = self.init.copy()
+        R = T.rotation
+        R = np.dot(R, pin.utils.rpyToMatrix(0.0, 0.0, np.pi / 2))
+        R = np.dot(R, pin.utils.rpyToMatrix(0.0, -np.pi, 0.0))
+        T.rotation = R
+        T.translation[0] += 0.5
+        T.translation[1] += 0.1 + 0.05 * np.sin(8.0 * t)
+        T.translation[2] += 0.5
+        return T
+
+
+if __name__ == "__main__":
+    robot = load_robot_description(
+        "draco3_description", root_joint=pin.JointModelFreeFlyer()
+    )
+
+    # Initialize meschcat visualizer
+    viz = pin.visualize.MeshcatVisualizer(
+        robot.model, robot.collision_model, robot.visual_model
+    )
+    robot.setVisualizer(viz, init=False)
+    viz.initViewer(open=True)
+    viz.loadViewerModel()
+
+    # Set initial robot configuration
+    configuration = pink.Configuration(robot.model, robot.data, robot.q0)
+    viz.display(configuration.q)
+
+    # Tasks initialization for IK
+    left_foot_task = BodyTask(
+        "l_foot_contact", position_cost=1.0, orientation_cost=1.0
+    )
+    pelvis_task = BodyTask(
+        "torso_com_link", position_cost=1.0, orientation_cost=0.0
+    )
+    right_foot_task = BodyTask(
+        "r_foot_contact", position_cost=1.0, orientation_cost=1.0
+    )
+    right_wrist_task = BodyTask(
+        "r_hand_contact", position_cost=4.0, orientation_cost=4.0
+    )
+    posture_task = PostureTask(
+        cost=1e-1,  # [cost] / [rad]
+    )
+
+    # Joint coupling task
+    r_knee_holonomic_task = JointCouplingTask(
+        ["r_knee_fe_jp", "r_knee_fe_jd"], [1.0, -1.0], 100.0, configuration
+    )
+    l_knee_holonomic_task = JointCouplingTask(
+        ["l_knee_fe_jp", "l_knee_fe_jd"], [1.0, -1.0], 100.0, configuration
+    )
+
+    tasks = [
+        left_foot_task,
+        pelvis_task,
+        right_foot_task,
+        right_wrist_task,
+        posture_task,
+        l_knee_holonomic_task,
+        r_knee_holonomic_task,
+    ]
+
+    # Task target specifications
+    pelvis_pose = configuration.get_transform_body_to_world(
+        "torso_com_link"
+    ).copy()
+    pelvis_pose.translation[0] += 0.05
+    pelvis_task.set_target(pelvis_pose)
+
+    transform_l_ankle_target_to_init = pin.SE3(
+        np.eye(3), np.array([0.1, 0.0, 0.0])
+    )
+    transform_r_ankle_target_to_init = pin.SE3(
+        np.eye(3), np.array([-0.1, 0.0, 0.0])
+    )
+
+    left_foot_task.set_target(
+        configuration.get_transform_body_to_world("l_foot_contact")
+        * transform_l_ankle_target_to_init
+    )
+    right_foot_task.set_target(
+        configuration.get_transform_body_to_world("r_foot_contact")
+        * transform_r_ankle_target_to_init
+    )
+
+    pelvis_task.set_target(
+        configuration.get_transform_body_to_world("torso_com_link")
+    )
+
+    posture_task.set_target_from_configuration(configuration)
+
+    right_wrist_pose = WavingPose(
+        configuration.get_transform_body_to_world("r_hand_contact")
+    )
+
+    wrist_frame = viz.viewer["right_wrist_pose"]
+    meshcat_shapes.frame(wrist_frame)
+
+    # Select QP solver
+    solver = qpsolvers.available_solvers[0]
+    if "quadprog" in qpsolvers.available_solvers:
+        solver = "quadprog"
+
+    rate = RateLimiter(frequency=200.0)
+    dt = rate.period
+    t = 0.0  # [s]
+    while True:
+        # Update task targets
+        right_wrist_task.set_target(right_wrist_pose.at(t))
+        wrist_frame.set_transform(right_wrist_pose.at(t).np)
+
+        # Compute velocity and integrate it into next configuration
+        velocity = solve_ik(configuration, tasks, dt, solver=solver)
+        configuration.integrate_inplace(velocity, dt)
+
+        # Visualize result at fixed FPS
+        viz.display(configuration.q)
+        rate.sleep()
+        t += dt

pink/tasks/__init__.py

@@ -18,13 +18,18 @@
 """Kinematic tasks."""
 
 from .body_task import BodyTask
-from .exceptions import TargetNotSet
+from .exceptions import TargetNotSet, TaskJacobianNotSet
+from .joint_coupling_task import JointCouplingTask
+from .linear_holonomic_task import LinearHolonomicTask
 from .posture_task import PostureTask
 from .task import Task
 
 __all__ = [
     "BodyTask",
-    "PostureTask",
     "TargetNotSet",
+    "TaskJacobianNotSet",
+    "JointCouplingTask",
+    "LinearHolonomicTask",
+    "PostureTask",
     "Task",
 ]

pink/tasks/exceptions.py

@@ -22,3 +22,7 @@
 
 class TargetNotSet(PinkError):
     """Exception raised when attempting to compute with an unset target."""
+
+
+class TaskJacobianNotSet(PinkError):
+    """Exception raised when attempting to compute with a not proper Jacobian."""

pink/tasks/joint_coupling_task.py

@@ -0,0 +1,58 @@
+#!/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.
+
+"""Joint coupling task implementation."""
+
+from typing import Sequence
+
+import numpy as np
+
+from ..configuration import Configuration
+from ..utils import get_joint_idx
+from .linear_holonomic_task import LinearHolonomicTask
+
+
+class JointCouplingTask(LinearHolonomicTask):
+    r"""Child class for linear holonomic tasks.
+
+    Note:
+        This task only considers a 1-Dimensional task.
+    """
+
+    def __init__(
+        self,
+        joint_name_list: Sequence[str],
+        ratios: Sequence[float],
+        cost: float,
+        configuration: Configuration,
+    ) -> None:
+        r"""Compute Jacobian matrix of a linear holonomic constraint.
+
+        Args:
+            joint_name_list: a list of joint names consisting of a linear
+                holonomic constraint.
+            ratios: a list of ratios that each joint takes in a linear
+                holonomic constraint.
+            cost: Cost of a linear holonomic task.
+            configuration: Robot configuration.
+        """
+        assert len(joint_name_list) == len(ratios)
+        self.A = np.zeros((1, configuration.model.nv))
+        for joint, ratio in zip(joint_name_list, ratios):
+            _, joint_v_idx = get_joint_idx(configuration.model, joint)
+            self.A[:, joint_v_idx] = ratio
+        self.cost = cost