Added exclude-topic-types to record (#1582)

Signed-off-by: Alejandro Hernández Cordero <[email protected]>

Author: ahcorde

ros2bag/ros2bag/verb/record.py

@@ -84,6 +84,10 @@ def add_arguments(self, parser, cli_name):  # noqa: D102
             '--exclude-regex', default='',
             help='Exclude topics and services containing provided regular expression. '
                  'Works on top of --all, --all-topics, or --regex.')
+        parser.add_argument(
+            '--exclude-topic-types', type=str, default=[], metavar='ExcludeTypes', nargs='+',
+            help='List of topic types not being recorded. '
+                 'Works on top of --all, --all-topics, or --regex.')
         parser.add_argument(
             '--exclude-topics', type=str, metavar='Topic', nargs='+',
             help='List of topics not being recorded. '
@@ -230,6 +234,10 @@ def main(self, *, args):  # noqa: D102
             return print_error('--exclude-topics argument requires either --all, --all-topics '
                                'or --regex')
 
+        if args.exclude_topic_types and not (args.regex or args.all or args.all_topics):
+            return print_error('--exclude-topic-types argument requires either --all, '
+                               '--all-topics or --regex')
+
         if args.exclude_services and not (args.regex or args.all or args.all_services):
             return print_error('--exclude-services argument requires either --all, --all-services '
                                'or --regex')
@@ -288,6 +296,7 @@ def main(self, *, args):  # noqa: D102
         record_options.is_discovery_disabled = args.no_discovery
         record_options.topics = args.topics
         record_options.topic_types = args.topic_types
+        record_options.exclude_topic_types = args.exclude_topic_types
         record_options.rmw_serialization_format = args.serialization_format
         record_options.topic_polling_interval = datetime.timedelta(
             milliseconds=args.polling_interval)

rosbag2_py/rosbag2_py/_transport.pyi

@@ -40,6 +40,7 @@ class RecordOptions:
     compression_threads: int
     exclude_regex: str
     exclude_service_events: List[str]
+    exclude_topic_types: List[str]
     exclude_topics: List[str]
     ignore_leaf_topics: bool
     include_hidden_topics: bool

rosbag2_py/src/rosbag2_py/_transport.cpp

@@ -371,6 +371,7 @@ PYBIND11_MODULE(_transport, m) {
   .def_readwrite("is_discovery_disabled", &RecordOptions::is_discovery_disabled)
   .def_readwrite("topics", &RecordOptions::topics)
   .def_readwrite("topic_types", &RecordOptions::topic_types)
+  .def_readwrite("exclude_topic_types", &RecordOptions::exclude_topic_types)
   .def_readwrite("rmw_serialization_format", &RecordOptions::rmw_serialization_format)
   .def_readwrite("topic_polling_interval", &RecordOptions::topic_polling_interval)
   .def_readwrite("regex", &RecordOptions::regex)

rosbag2_tests/test/rosbag2_tests/test_rosbag2_cpp_get_service_info.cpp

@@ -189,7 +189,7 @@ TEST_P(Rosbag2CPPGetServiceInfoTest, get_service_info_for_bag_with_services_only
   storage_options.storage_id = storage_id;
   storage_options.uri = bag_path_str;
   rosbag2_transport::RecordOptions record_options =
-  {false, true, false, {}, {}, {}, {"/rosout"}, {}, "cdr", 100ms};
+  {false, true, false, {}, {}, {}, {"/rosout"}, {}, {}, "cdr", 100ms};
   auto recorder = std::make_shared<rosbag2_transport::Recorder>(
     std::move(writer), storage_options, record_options);
   recorder->record();
@@ -267,7 +267,7 @@ TEST_P(Rosbag2CPPGetServiceInfoTest, get_service_info_for_bag_with_topics_and_se
   storage_options.storage_id = storage_id;
   storage_options.uri = bag_path_str;
   rosbag2_transport::RecordOptions record_options =
-  {true, true, false, {}, {}, {}, {"/rosout"}, {}, "cdr", 100ms};
+  {true, true, false, {}, {}, {}, {"/rosout"}, {}, {}, "cdr", 100ms};
   auto recorder = std::make_shared<rosbag2_transport::Recorder>(
     std::move(writer), storage_options, record_options);
   recorder->record();

rosbag2_transport/include/rosbag2_transport/record_options.hpp

@@ -37,6 +37,7 @@ struct RecordOptions
   std::vector<std::string> topic_types;
   std::vector<std::string> services;  // service event topic
   std::vector<std::string> exclude_topics;
+  std::vector<std::string> exclude_topic_types;
   std::vector<std::string> exclude_service_events;  // service event topic
   std::string rmw_serialization_format;
   std::chrono::milliseconds topic_polling_interval{100};

rosbag2_transport/src/rosbag2_transport/config_options_from_node_params.cpp

@@ -228,6 +228,9 @@ RecordOptions get_record_options_from_node_params(rclcpp::Node & node)
   record_options.exclude_topics = node.declare_parameter<std::vector<std::string>>(
     "record.exclude_topics", std::vector<std::string>());
 
+  record_options.exclude_topic_types = node.declare_parameter<std::vector<std::string>>(
+    "record.exclude_topic_types", std::vector<std::string>());
+
   // Convert service name to service event topic name
   auto exclude_service_list = node.declare_parameter<std::vector<std::string>>(
     "record.exclude_services", std::vector<std::string>());

rosbag2_transport/src/rosbag2_transport/record_options.cpp

@@ -32,6 +32,7 @@ Node convert<rosbag2_transport::RecordOptions>::encode(
   node["is_discovery_disabled"] = record_options.is_discovery_disabled;
   node["topics"] = record_options.topics;
   node["topic_types"] = record_options.topic_types;
+  node["exclude_topic_types"] = record_options.exclude_topic_types;
   node["services"] = record_options.services;
   node["rmw_serialization_format"] = record_options.rmw_serialization_format;
   node["topic_polling_interval"] = record_options.topic_polling_interval;
@@ -71,6 +72,9 @@ bool convert<rosbag2_transport::RecordOptions>::decode(
   optional_assign<std::string>(node, "regex", record_options.regex);
   optional_assign<std::string>(node, "exclude_regex", record_options.exclude_regex);
   optional_assign<std::vector<std::string>>(node, "exclude_topics", record_options.exclude_topics);
+  optional_assign<std::vector<std::string>>(
+    node, "exclude_topic_types",
+    record_options.exclude_topic_types);
   optional_assign<std::vector<std::string>>(
     node, "exclude_services", record_options.exclude_service_events);
   optional_assign<std::string>(node, "node_prefix", record_options.node_prefix);

rosbag2_transport/src/rosbag2_transport/topic_filter.cpp

@@ -170,6 +170,10 @@ bool TopicFilter::take_topic(
       }
     }
 
+    if (topic_type_in_list(topic_type, record_options_.exclude_topic_types)) {
+      return false;
+    }
+
     if (topic_in_list(topic_name, record_options_.exclude_topics)) {
       return false;
     }

rosbag2_transport/test/resources/recorder_node_params.yaml

@@ -9,6 +9,7 @@ recorder_params_node:
       is_discovery_disabled: true
       topics: ["topic", "other_topic"]
       topic_types: ["std_msgs/msg/Header", "geometry_msgs/msg/Pose"]
+      exclude_topic_types: ["sensor_msgs/msg/Image"]
       services: ["service", "other_service"]
       rmw_serialization_format: "cdr"
       topic_polling_interval:
@@ -41,4 +42,4 @@ recorder_params_node:
       snapshot_mode: false
       custom_data: ["key1=value1", "key2=value2"]
       start_time_ns: 0
-      end_time_ns: 100000
+      end_time_ns: 100000

rosbag2_transport/test/rosbag2_transport/test_composable_recorder.cpp

@@ -216,6 +216,8 @@ TEST_P(ComposableRecorderTests, recorder_can_parse_parameters_from_file) {
   EXPECT_EQ(record_options.topics, topics);
   std::vector<std::string> topic_types {"std_msgs/msg/Header", "geometry_msgs/msg/Pose"};
   EXPECT_EQ(record_options.topic_types, topic_types);
+  std::vector<std::string> exclude_topic_types {"sensor_msgs/msg/Image"};
+  EXPECT_EQ(record_options.exclude_topic_types, exclude_topic_types);
   std::vector<std::string> services {"/service/_service_event", "/other_service/_service_event"};
   EXPECT_EQ(record_options.services, services);
   EXPECT_EQ(record_options.rmw_serialization_format, "cdr");