placo 0.3.6__0-cp310-cp310-manylinux_2_35_x86_64.whl → 0.3.8__0-cp310-cp310-manylinux_2_35_x86_64.whl

cmeel.prefix/lib/python3.10/site-packages/placo.pyi

@@ -3615,6 +3615,19 @@ class HumanoidRobot:
     :return: the joint current (inner state) value (e.g rad for revolute or meters for prismatic)"""
     ...
 
+  def get_joint_acceleration(
+    self: HumanoidRobot,
+    name: str, # const std::string &
+
+  ) -> float:
+    """Gets the joint acceleration from state.qd. 
+        
+
+    :param name: joint name 
+
+    :return: joint acceleration"""
+    ...
+
   def get_joint_offset(
     self: HumanoidRobot,
     name: str, # const std::string &
@@ -3797,6 +3810,20 @@ class HumanoidRobot:
     :param value: joint value (e.g rad for revolute or meters for prismatic)"""
     ...
 
+  def set_joint_acceleration(
+    self: HumanoidRobot,
+    name: str, # const std::string &
+    value: float, # double
+
+  ) -> None:
+    """Sets the joint acceleration in state.qd. 
+        
+
+    :param name: joint name 
+
+    :param value: joint acceleration"""
+    ...
+
   def set_joint_limits(
     self: HumanoidRobot,
     name: str, # const std::string &
@@ -3926,7 +3953,7 @@ class HumanoidRobot:
     self: HumanoidRobot,
 
   ) -> None:
-    """Update the current kinematics."""
+    """Update internal computation for kinematics (frames, jacobian). This method should be called when the robot state has changed."""
     ...
 
   def update_support_side(
@@ -6007,6 +6034,19 @@ class RobotWrapper:
     :return: the joint current (inner state) value (e.g rad for revolute or meters for prismatic)"""
     ...
 
+  def get_joint_acceleration(
+    self: RobotWrapper,
+    name: str, # const std::string &
+
+  ) -> float:
+    """Gets the joint acceleration from state.qd. 
+        
+
+    :param name: joint name 
+
+    :return: joint acceleration"""
+    ...
+
   def get_joint_offset(
     self: RobotWrapper,
     name: str, # const std::string &
@@ -6176,6 +6216,20 @@ class RobotWrapper:
     :param value: joint value (e.g rad for revolute or meters for prismatic)"""
     ...
 
+  def set_joint_acceleration(
+    self: RobotWrapper,
+    name: str, # const std::string &
+    value: float, # double
+
+  ) -> None:
+    """Sets the joint acceleration in state.qd. 
+        
+
+    :param name: joint name 
+
+    :param value: joint acceleration"""
+    ...
+
   def set_joint_limits(
     self: RobotWrapper,
     name: str, # const std::string &
@@ -6301,7 +6355,7 @@ class RobotWrapper:
     self: RobotWrapper,
 
   ) -> None:
-    """Update the current kinematics."""
+    """Update internal computation for kinematics (frames, jacobian). This method should be called when the robot state has changed."""
     ...
 
   visual_model: any # pinocchio::GeometryModel
@@ -7191,19 +7245,6 @@ def flatten_on_floor(
   ...
 
 
-def frame(
-  matrix: any, # Eigen::Matrix4d
-
-) -> numpy.ndarray:
-  """Makes an Affine3d from a 4x4 matrix (for python bindings) 
-        
-
-  :param matrix: the 4x4 matrix 
-
-  :return: The Affine3d"""
-  ...
-
-
 def frame_yaw(
   rotation: numpy.ndarray, # Eigen::Matrix3d
 

cmeel.prefix/lib/python3.10/site-packages/placo_utils/visualization.py

@@ -18,12 +18,15 @@ def get_viewer() -> meshcat.Visualizer:
 
     if viewer is None:
         viewer = meshcat.Visualizer()
+
         print(f"Viewer URL: {viewer.url()}")
 
     return viewer
 
 
-def robot_viz(robot: placo.RobotWrapper, name: str = "robot") -> pin.visualize.MeshcatVisualizer:
+def robot_viz(
+    robot: placo.RobotWrapper, name: str = "robot"
+) -> pin.visualize.MeshcatVisualizer:
     """
     Builds an instance of pinocchio MeshcatVisualizer, which allows to push the model to the meshcat
     visualizer passed as parameter
@@ -31,14 +34,13 @@ def robot_viz(robot: placo.RobotWrapper, name: str = "robot") -> pin.visualize.M
     The robot can further be displayed using:
 
     > viz.display(q)
-
-    :param robot: robot instance
-    :param name: name of the robot
     """
     global robot_names
 
     robot_names[robot] = name
-    viz = pin.visualize.MeshcatVisualizer(robot.model, robot.collision_model, robot.visual_model)
+    viz = pin.visualize.MeshcatVisualizer(
+        robot.model, robot.collision_model, robot.visual_model
+    )
     viz.initViewer(viewer=get_viewer())
     viz.loadViewerModel(name)
 
@@ -51,10 +53,6 @@ cylinders: dict = {}
 def frame_viz(name: str, T: np.ndarray, opacity: float = 1.0) -> None:
     """
     Visualizes a given frame
-
-    :param name: name of the frame
-    :param T: transformation of the frame to the world
-    :param opacity: opacity of the frame
     """
     global cylinders
     vis = get_viewer()
@@ -76,30 +74,60 @@ def frame_viz(name: str, T: np.ndarray, opacity: float = 1.0) -> None:
             )
         obj = cylinders[node_name]
 
-        obj.set_transform(T @ tf.rotation_matrix(*rotate) @ tf.translation_matrix([0, 0.05, 0]))
+        obj.set_transform(
+            T @ tf.rotation_matrix(*rotate) @ tf.translation_matrix([0, 0.05, 0])
+        )
 
 
-def point_viz(name: str, point: np.ndarray, radius: float = 0.01, color: float = 0xFF0000, opacity: float = 1.0) -> None:
+def point_viz(
+    name: str,
+    point: np.ndarray,
+    radius: float = 0.01,
+    color: float = 0xFF0000,
+    opacity: float = 1.0,
+) -> None:
     """
     Prints a point (sphere)
+    """
+    vis = get_viewer()
+    vis["point"][name].set_object(
+        g.Sphere(radius), g.MeshPhongMaterial(color=color, opacity=opacity)
+    )
+    vis["point"][name].set_transform(tf.translation_matrix(point))
+
 
-    :param name: name of the point
-    :param point: point to print
-    :param radius: radius of the point
-    :param color: color of the point
-    :param opacity: opacity of the point
+points_sizes = {}
+
+
+def points_viz(
+    name: str,
+    points: np.ndarray,
+    radius: float = 0.01,
+    color: float = 0xFF0000,
+    opacity: float = 1.0,
+) -> None:
+    """
+    Prints a point (sphere)
     """
+    global points_sizes
     vis = get_viewer()
-    vis["points"][name].set_object(g.Sphere(radius), g.MeshPhongMaterial(color=color, opacity=opacity))
-    vis["points"][name].set_transform(tf.translation_matrix(point))
+    k = 0
+    for point in points:
+        entry = f"{name}_{k}"
+        k += 1
+        vis["points"][name][entry].set_object(
+            g.Sphere(radius), g.MeshPhongMaterial(color=color, opacity=opacity)
+        )
+        vis["points"][name][entry].set_transform(tf.translation_matrix(point))
+    while k < points_sizes.get(name, 0):
+        vis["points"][name][f"{name}_{k}"].delete()
+        k += 1
+    points_sizes[name] = len(points)
 
 
 def robot_frame_viz(robot: placo.RobotWrapper, frame: str) -> None:
     """
-    Draw a frame attached to the robot
-
-    :param robot: robot instance
-    :param frame: frame name
+    Draw a frame from the robot
     """
     node_name = f"{robot_names[robot]}_{frame}"
     frame_viz(node_name, robot.get_T_world_frame(frame))
@@ -109,12 +137,6 @@ steps: int = 0
 
 
 def footsteps_viz(footsteps: placo.Footsteps, T: np.ndarray = np.eye(4)) -> None:
-    """
-    Visualizes humanoid footsteps
-
-    :param footsteps: footsteps to visualize
-    :param T: left transformation to apply to the footsteps
-    """    
     global steps
     vis = get_viewer()
 
@@ -132,23 +154,25 @@ def footsteps_viz(footsteps: placo.Footsteps, T: np.ndarray = np.eye(4)) -> None
             if len(footstep.footsteps) >= 2:
                 color = 0x1111AA
             else:
-                color = 0xFF3333 if str(footstep.footsteps[0].side) == "left" else 0x33FF33
+                color = (
+                    0xFF3333 if str(footstep.footsteps[0].side) == "left" else 0x33FF33
+                )
         else:
             color = 0xFF3333 if str(footstep.side) == "left" else 0x33FF33
 
-        vis["footsteps"][str(k)].set_object(g.LineLoop(g.PointsGeometry(polygon.T), g.MeshBasicMaterial(color=color)))
+        vis["footsteps"][str(k)].set_object(
+            g.LineLoop(g.PointsGeometry(polygon.T), g.MeshBasicMaterial(color=color))
+        )
 
 
 def line_viz(name: str, points: np.ndarray, color: float = 0xFF0000) -> None:
     """
     Prints a line
-
-    :param name: name of the line
-    :param points: points of the line
-    :param color: color of the line
     """
     vis = get_viewer()
-    vis["lines"][name].set_object(g.LineSegments(g.PointsGeometry(points.T), g.LineBasicMaterial(color=color)))
+    vis["lines"][name].set_object(
+        g.LineSegments(g.PointsGeometry(points.T), g.LineBasicMaterial(color=color))
+    )
 
 
 def arrow_viz(
@@ -160,14 +184,8 @@ def arrow_viz(
 ) -> None:
     """
     Prints an arrow
-
-    :param name: name of the arrow
-    :param point_from: origin of the arrow
-    :param point_to: end of the arrow
-    :param color: color of the arrow
-    :param radius: radius of the arrow
     """
-    head_length = radius*3
+    head_length = radius * 3
     vis = get_viewer()
     length = np.linalg.norm(point_to - point_from)
     length = max(1e-3, length - head_length)
@@ -189,43 +207,58 @@ def arrow_viz(
     T_cylinder = T @ tf.translation_matrix(np.array([0, length / 2.0, 0.0]))
     T_head = T @ tf.translation_matrix(np.array([0, length + head_length / 2.0, 0.0]))
 
-    vis["arrows"][name]["cylinder"].set_object(cylinder, g.MeshBasicMaterial(color=color))
+    vis["arrows"][name]["cylinder"].set_object(
+        cylinder, g.MeshBasicMaterial(color=color)
+    )
     vis["arrows"][name]["cylinder"].set_transform(T_cylinder)
     vis["arrows"][name]["head"].set_object(head, g.MeshBasicMaterial(color=color))
     vis["arrows"][name]["head"].set_transform(T_head)
 
+
 previous_contacts: int = 0
 
-def contacts_viz(solver: placo.DynamicsSolver, ratio=0.1, radius=0.005):
-    """
-    Draws the contacts produced by a dynamics solver
 
-    :param solver: solver instance
-    :param ratio: ratio of the wrench to display
-    :param radius: radius of the arrows
-    """    
+def contacts_viz(solver: placo.DynamicsSolver, ratio=0.1, radius=0.005):
     global previous_contacts
     robot = solver.robot
     frames = robot.frame_names()
-    
+
     for k in range(solver.count_contacts()):
         contact = solver.get_contact(k)
 
         if isinstance(contact, placo.PointContact):
             frame_name = frames[contact.position_task().frame_index]
             T_world_frame = robot.get_T_world_frame(frame_name)
-            arrow_viz(f"contact_{k}", T_world_frame[:3, 3], T_world_frame[:3, 3] + contact.wrench * ratio, color=0x00FF00, radius=radius)
+            arrow_viz(
+                f"contact_{k}",
+                T_world_frame[:3, 3],
+                T_world_frame[:3, 3] + contact.wrench * ratio,
+                color=0x00FF00,
+                radius=radius,
+            )
         elif isinstance(contact, placo.Contact6D):
             frame_name = frames[contact.position_task().frame_index]
             T_world_frame = robot.get_T_world_frame(frame_name)
             wrench = T_world_frame[:3, :3] @ contact.wrench[:3]
             origin = T_world_frame[:3, 3] + T_world_frame[:3, :3] @ contact.zmp()
-            arrow_viz(f"contact_{k}", origin, origin + wrench * ratio, color=0x00FFAA, radius=radius)
+            arrow_viz(
+                f"contact_{k}",
+                origin,
+                origin + wrench * ratio,
+                color=0x00FFAA,
+                radius=radius,
+            )
         elif isinstance(contact, placo.ExternalWrenchContact):
             frame_name = frames[contact.frame_index]
             T_world_frame = robot.get_T_world_frame(frame_name)
-            arrow_viz(f"contact_{k}", T_world_frame[:3, 3], T_world_frame[:3, 3] + contact.w_ext[:3] * ratio, color=0xFF2222, radius=radius)
-            
+            arrow_viz(
+                f"contact_{k}",
+                T_world_frame[:3, 3],
+                T_world_frame[:3, 3] + contact.w_ext[:3] * ratio,
+                color=0xFF2222,
+                radius=radius,
+            )
+
     vis = get_viewer()
     while k < previous_contacts:
         k += 1

{placo-0.3.6.dist-info → placo-0.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: placo
-Version: 0.3.6
+Version: 0.3.8
 Summary: PlaCo: Rhoban Planning and Control
 Requires-Python: >= 3.8
 License-Expression: MIT
@@ -15,6 +15,7 @@ Requires-Dist: pin >= 2.6.18, < 3
 Requires-Dist: rhoban-cmeel-jsoncpp
 Requires-Dist: meshcat
 Requires-Dist: numpy
+Requires-Dist: ischedule
 Provides-Extra: build
 Requires-Dist: pin[build] >= 2.6.18, < 3 ; extra == "build"
 Requires-Dist: cmeel-urdfdom[build] ; extra == "build"

placo-0.3.8.dist-info/RECORD

@@ -0,0 +1,14 @@
+cmeel.prefix/lib/liblibplaco.so,sha256=LA0wX_wjE4-eUV1gVQfvvKU5G6HzWc6-zuDYipFuMmI,2377288
+cmeel.prefix/lib/python3.10/site-packages/placo.pyi,sha256=6e0UoGP1Q5mbOZtjTu2fErtGNF4Cay4yiNHmbBceEsY,152828
+cmeel.prefix/lib/python3.10/site-packages/placo.so,sha256=AKUO5w1YL7C4_DjHeCfxPma2-I94-8sV232cGzYzSTA,9113176
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/__init__.py,sha256=UN-fc5KfBWQ-_qkm0Ajouh-T9tBGm5aUtuzBiH1tRtk,80
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/tf.py,sha256=fFRXNbeLlXzn5VOqYl7hcSuvOOtTDTiLi_Lpd9_l6wA,36
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/visualization.py,sha256=nsO5-1y0itnIOR79-HrQSvS7YEjJmV5MAROULnfX3x8,7590
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/__init__.cpython-310.pyc,sha256=tJrTajdA7hNO9tZkE1oN49rU-61lCNUED-mQXdAvzAs,261
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/tf.cpython-310.pyc,sha256=APWQ5-zKgbEcMnkv6Hxgj_Q46jpnuOyL9YbSrj6Z5Vw,204
+cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/visualization.cpython-310.pyc,sha256=Yu4t9oM-MHJCjF11e8G6LjCQJFIpqYmoAcA5_yY66y0,6560
+placo-0.3.8.dist-info/license/LICENSE,sha256=q2bBXvk4Eh7TmP11LoIOIGSUuJbR30JBI6ZZ37g52T4,1061
+placo-0.3.8.dist-info/METADATA,sha256=Gs02_iU-AAYYcGfvXydxHjAVhaa41o71W6ivl4GuOik,1643
+placo-0.3.8.dist-info/WHEEL,sha256=GG3aWrINZ6-EfOf2jaRO1SBQfOwv0eJwKe893X49tgU,115
+placo-0.3.8.dist-info/top_level.txt,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+placo-0.3.8.dist-info/RECORD,,

placo-0.3.6.dist-info/RECORD

@@ -1,14 +0,0 @@
-cmeel.prefix/lib/liblibplaco.so,sha256=urzydnb6NnMOLjsqLhEXVzxTvC-SPzr3bjVxR7gfHKc,2377112
-cmeel.prefix/lib/python3.10/site-packages/placo.pyi,sha256=HhW9bu7PlK_eo86uT8hkMIRCCeKmap1X7qZdYthSUGo,151822
-cmeel.prefix/lib/python3.10/site-packages/placo.so,sha256=vK-g0qk1bC1pTm2JlkbUwo8yeHYW8Hp-UQbtAiK5HyA,9123808
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/__init__.py,sha256=UN-fc5KfBWQ-_qkm0Ajouh-T9tBGm5aUtuzBiH1tRtk,80
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/tf.py,sha256=fFRXNbeLlXzn5VOqYl7hcSuvOOtTDTiLi_Lpd9_l6wA,36
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/visualization.py,sha256=rY-8ARc8aeOVX9KNAWNX3H9VK3v28TAISRabflxZ00I,7565
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/__init__.cpython-310.pyc,sha256=tJrTajdA7hNO9tZkE1oN49rU-61lCNUED-mQXdAvzAs,261
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/tf.cpython-310.pyc,sha256=APWQ5-zKgbEcMnkv6Hxgj_Q46jpnuOyL9YbSrj6Z5Vw,204
-cmeel.prefix/lib/python3.10/site-packages/placo_utils/__pycache__/visualization.cpython-310.pyc,sha256=dTICIIwrtr6m-s4GOALeJxKg2T9mZThg6cxmNjKbr98,7000
-placo-0.3.6.dist-info/license/LICENSE,sha256=q2bBXvk4Eh7TmP11LoIOIGSUuJbR30JBI6ZZ37g52T4,1061
-placo-0.3.6.dist-info/METADATA,sha256=MpTFWIVVogDly0Ma5BhsQmmYn-l-tB6dGGA1Bi-RYtk,1618
-placo-0.3.6.dist-info/WHEEL,sha256=GG3aWrINZ6-EfOf2jaRO1SBQfOwv0eJwKe893X49tgU,115
-placo-0.3.6.dist-info/top_level.txt,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-placo-0.3.6.dist-info/RECORD,,