fix: subcategories for arm_on_rail and base picknik behaviors

PickNikRobotics · Nov 14, 2024 · 669c5a2 · 669c5a2
1 parent da2e327
commit 669c5a2
Show file tree

Hide file tree

Showing 37 changed files with 202 additions and 22 deletions.
diff --git a/src/arm_on_rail_sim/objectives/3_waypoint_pick_and_place.xml b/src/arm_on_rail_sim/objectives/3_waypoint_pick_and_place.xml
@@ -1,3 +1,4 @@
+<?xml version="1.0" encoding="UTF-8" ?>
 <root BTCPP_format="4" main_tree_to_execute="3 Waypoints Pick and Place">
   <!--//////////-->
   <BehaviorTree
@@ -119,4 +120,11 @@
       </Decorator>
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="3 Waypoints Pick and Place">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/apriltag_pick_object.xml b/src/arm_on_rail_sim/objectives/apriltag_pick_object.xml
@@ -9,7 +9,7 @@
     <Control ID="Sequence" name="TopLevelSequence">
       <SubTree ID="Look at table" _collapsed="true" />
       <Action ID="ClearSnapshot" />
-      <SubTree ID="Wrist camera snapshot" _collapsed="true" />
+      <SubTree ID="Take wrist camera snapshot" _collapsed="true" />
       <Control ID="Sequence" name="Setup">
         <SubTree ID="Open Gripper" />
       </Control>
@@ -69,4 +69,11 @@
       <Action ID="ExecuteMTCTask" solution="{pick_object_solution}" />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Pick April Tag Labeled Object">
+      <MetadataFields>
+        <Metadata subcategory="Application - Advanced Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/constrained_pick_place.xml b/src/arm_on_rail_sim/objectives/constrained_pick_place.xml
@@ -148,4 +148,11 @@
       </Decorator>
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Constrained Pick and Place">
+      <MetadataFields>
+        <Metadata subcategory="Application - Advanced Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/cycle_between_waypoints.xml b/src/arm_on_rail_sim/objectives/cycle_between_waypoints.xml
@@ -25,4 +25,11 @@
       </Control>
     </Decorator>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Cycle Between Waypoints">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/force_relaxation.xml b/src/arm_on_rail_sim/objectives/force_relaxation.xml
@@ -50,6 +50,9 @@
       <input_port name="wrench_gain" default="0.01" />
       <input_port name="relaxation_time_ms" default="500" />
       <input_port name="tip_link" default="grasp_link" />
+      <MetadataFields>
+        <Metadata subcategory="Motion - Controls" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/joint_diagnostic.xml b/src/arm_on_rail_sim/objectives/joint_diagnostic.xml
@@ -35,4 +35,11 @@
       </Decorator>
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Joint Diagnostic">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/load_mesh_as_green_poincloud.xml b/src/arm_on_rail_sim/objectives/load_mesh_as_green_poincloud.xml
@@ -31,6 +31,9 @@
       <output_port name="point_cloud" default="{green_cloud}" />
       <input_port name="_collapsed" default="true" />
       <input_port name="initial_pose" default="{stamped_pose}" />
+      <MetadataFields>
+        <Metadata subcategory="Perception - 3D Point Cloud" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/load_mesh_as_red_pointcloud.xml b/src/arm_on_rail_sim/objectives/load_mesh_as_red_pointcloud.xml
@@ -31,6 +31,9 @@
       <output_port name="point_cloud" default="{red_cloud}" />
       <input_port name="_collapsed" default="false" />
       <input_port name="initial_pose" default="{stamped_pose}" />
+      <MetadataFields>
+        <Metadata subcategory="Perception - 3D Point Cloud" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/look_at_table.xml b/src/arm_on_rail_sim/objectives/look_at_table.xml
@@ -13,4 +13,11 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Look at table">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/move_with_velocity_and_force.xml b/src/arm_on_rail_sim/objectives/move_with_velocity_and_force.xml
@@ -73,4 +73,11 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Move until Force Contact">
+      <MetadataFields>
+        <Metadata subcategory="Application - Advanced Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/pick_april_tag_object_with_approval.xml b/src/arm_on_rail_sim/objectives/pick_april_tag_object_with_approval.xml
@@ -10,7 +10,7 @@
     _favorite="false"
   >
     <Control ID="Sequence" name="TopLevelSequence">
-      <SubTree ID="Wrist camera snapshot" _collapsed="true" />
+      <SubTree ID="Take wrist camera snapshot" _collapsed="true" />
       <Control ID="Sequence" name="Setup">
         <SubTree ID="Open Gripper" />
       </Control>
@@ -50,4 +50,11 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Pick April Tag Object With Approval">
+      <MetadataFields>
+        <Metadata subcategory="Application - Advanced Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/pick_first_object_in_vector.xml b/src/arm_on_rail_sim/objectives/pick_first_object_in_vector.xml
@@ -43,6 +43,10 @@
     </Decorator>
   </BehaviorTree>
   <TreeNodesModel>
-    <SubTree ID="Pick first object in vector" />
+    <SubTree ID="Pick first object in vector">
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
+    </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/pick_object.xml b/src/arm_on_rail_sim/objectives/pick_object.xml
@@ -29,4 +29,7 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Pick object" />
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/pick_up_cube.xml b/src/arm_on_rail_sim/objectives/pick_up_cube.xml
@@ -6,8 +6,15 @@
       <SubTree ID="Clear Snapshot" _collapsed="true" />
       <SubTree ID="Look at table" _collapsed="true" />
       <!--Take a snapshot so we have a visualization of the cubes to use in grasp pose correction-->
-      <SubTree ID="Wrist camera snapshot" _collapsed="true" />
+      <SubTree ID="Take wrist camera snapshot" _collapsed="true" />
       <SubTree ID="Pick object" _collapsed="false" />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Pick up Cube">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/place_first_object_in_vector.xml b/src/arm_on_rail_sim/objectives/place_first_object_in_vector.xml
@@ -42,4 +42,11 @@
       </Control>
     </Decorator>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Place first object in vector">
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/place_object.xml b/src/arm_on_rail_sim/objectives/place_object.xml
@@ -25,6 +25,10 @@
     </Control>
   </BehaviorTree>
   <TreeNodesModel>
-    <SubTree ID="Place object" />
+    <SubTree ID="Place object">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/register_cad_part.xml b/src/arm_on_rail_sim/objectives/register_cad_part.xml
@@ -53,4 +53,11 @@
       <Action ID="SendPointCloudToUI" point_cloud="{output_cloud_aligned}" />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Register CAD Part">
+      <MetadataFields>
+        <Metadata subcategory="Perception - Planning Scene" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/take_snap.xml b/src/arm_on_rail_sim/objectives/take_snap.xml
@@ -1,15 +1,23 @@
 <?xml version="1.0" encoding="UTF-8" ?>
-<root BTCPP_format="4" main_tree_to_execute="Scene camera snapshot">
+<root BTCPP_format="4" main_tree_to_execute="Take scene camera snapshot">
   <!--//////////-->
-  <BehaviorTree ID="Scene camera snapshot" _description="" _favorite="true">
+  <BehaviorTree
+    ID="Take scene camera snapshot"
+    _description=""
+    _favorite="true"
+  >
     <Control ID="Sequence" name="TopLevelSequence">
-      <!-- Clear out old snapshot data -->
+      <!--Clear out old snapshot data-->
       <Action ID="ClearSnapshot" />
       <Action ID="GetPointCloud" topic_name="/scene_camera/points" />
       <Action ID="SendPointCloudToUI" />
     </Control>
   </BehaviorTree>
   <TreeNodesModel>
-    <SubTree ID="Scene camera snapshot" />
+    <SubTree ID="Take scene camera snapshot">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/arm_on_rail_sim/objectives/wrist_snap.xml b/src/arm_on_rail_sim/objectives/wrist_snap.xml
@@ -1,14 +1,22 @@
 <?xml version="1.0" encoding="UTF-8" ?>
-<root BTCPP_format="4" main_tree_to_execute="Wrist camera snapshot">
+<root BTCPP_format="4" main_tree_to_execute=" Take wrist camera snapshot">
   <!--//////////-->
-  <BehaviorTree ID="Wrist camera snapshot" _description="" _favorite="true">
+  <BehaviorTree
+    ID="Take wrist camera snapshot"
+    _description=""
+    _favorite="true"
+  >
     <Control ID="Sequence" name="TopLevelSequence">
       <!-- Do not clear snapshots first, so we can take multiple snapshots from different angles -->
       <Action ID="GetPointCloud" topic_name="/wrist_camera/points" />
       <Action ID="SendPointCloudToUI" />
     </Control>
   </BehaviorTree>
   <TreeNodesModel>
-    <SubTree ID="Wrist camera snapshot" />
+    <SubTree ID="Take wrist camera snapshot">
+      <MetadataFields>
+        <Metadata subcategory="Application - Basic Examples" />
+      </MetadataFields>
+    </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/example_behaviors/call_my_service_solution/src/call_my_service.cpp b/src/example_behaviors/call_my_service_solution/src/call_my_service.cpp
@@ -23,7 +23,7 @@ BT::PortsList CallMyService::providedPorts()
 
 BT::KeyValueVector CallMyService::metadata()
 {
-  return { { "subcategory", "Example" },
+  return { { "subcategory", "ROS Messaging" },
            { "description", "Example of calling a ROS2 service." } };
 }
 

diff --git a/src/example_behaviors/translate_pose_solution/src/translate_pose.cpp b/src/example_behaviors/translate_pose_solution/src/translate_pose.cpp
@@ -19,7 +19,7 @@ BT::PortsList TranslatePose::providedPorts()
 
 BT::KeyValueVector TranslatePose::metadata()
 {
-  return { { "description", "Translates a PoseStamped, given x,y,z" } };
+  return { {"subcategory", "Pose Handling"}, { "description", "Translates a PoseStamped, given x,y,z" } };
 }
 
 BT::NodeStatus TranslatePose::tick()

diff --git a/src/picknik_ur_base_config/objectives/clear_snapshot.xml b/src/picknik_ur_base_config/objectives/clear_snapshot.xml
@@ -7,4 +7,10 @@
   >
     <Action ID="ClearSnapshot" />
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="ClearSnapshot" />
+    <MetadataFields>
+      <Metadata subcategory="Perception - 3D Point Cloud" />
+    </MetadataFields>
+  </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/close_gripper.xml b/src/picknik_ur_base_config/objectives/close_gripper.xml
@@ -10,4 +10,10 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Close Gripper" />
+    <MetadataFields>
+      <Metadata subcategory="Grasping" />
+    </MetadataFields>
+  </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/get_imarker_pose_from_mesh_visualization.xml b/src/picknik_ur_base_config/objectives/get_imarker_pose_from_mesh_visualization.xml
@@ -27,6 +27,9 @@
       <input_port name="mesh_path" default="{mesh_path}" />
       <input_port name="mesh_pose" default="{mesh_pose}" />
       <input_port name="_collapsed" default="true" />
+      <MetadataFields>
+        <Metadata subcategory="Visualization" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/interpolate_to_joint_state.xml b/src/picknik_ur_base_config/objectives/interpolate_to_joint_state.xml
@@ -34,6 +34,9 @@
   <TreeNodesModel>
     <SubTree ID="Interpolate to Joint State">
       <input_port name="target_joint_state" default="{target_joint_state}" />
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/move_to_joint_state.xml b/src/picknik_ur_base_config/objectives/move_to_joint_state.xml
@@ -32,6 +32,9 @@
   <TreeNodesModel>
     <SubTree ID="Move to Joint State">
       <input_port name="target_joint_state" default="{target_joint_state}" />
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/move_to_pose.xml b/src/picknik_ur_base_config/objectives/move_to_pose.xml
@@ -36,6 +36,9 @@
   <TreeNodesModel>
     <SubTree ID="Move to Pose">
       <input_port name="target_pose" default="{target_pose}" />
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/move_to_waypoint.xml b/src/picknik_ur_base_config/objectives/move_to_waypoint.xml
@@ -45,6 +45,9 @@
         name="controller_names"
         default="/joint_trajectory_controller"
       />
+      <MetadataFields>
+        <Metadata subcategory="Motion - Execute" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/open_gripper.xml b/src/picknik_ur_base_config/objectives/open_gripper.xml
@@ -11,4 +11,11 @@
       />
     </Control>
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Open Gripper">
+      <MetadataFields>
+        <Metadata subcategory="Grasping" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/request_teleoperation.xml b/src/picknik_ur_base_config/objectives/request_teleoperation.xml
@@ -164,6 +164,9 @@
       <input_port name="enable_user_interaction" default="false" />
       <input_port name="user_interaction_prompt" default="" />
       <input_port name="initial_teleop_mode" default="3" />
+      <MetadataFields>
+        <Metadata subcategory="User Input" />
+      </MetadataFields>
     </SubTree>
   </TreeNodesModel>
 </root>
diff --git a/src/picknik_ur_base_config/objectives/reset_planning_scene.xml b/src/picknik_ur_base_config/objectives/reset_planning_scene.xml
@@ -12,4 +12,11 @@
       apply_planning_scene_service="/apply_planning_scene"
     />
   </BehaviorTree>
+  <TreeNodesModel>
+    <SubTree ID="Reset Planning Scene">
+      <MetadataFields>
+        <Metadata subcategory="Perception - Planning Scene" />
+      </MetadataFields>
+    </SubTree>
+  </TreeNodesModel>
 </root>