feat: add diagnostic handler to warn when the processing time exceed …

…a threshold Signed-off-by: ktro2828 <[email protected]>
autowarefoundation · Mar 3, 2025 · 468156f · 468156f
1 parent b7acf99
commit 468156f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/.../autoware_map_based_prediction/include/map_based_prediction/map_based_prediction_node.hpp b/.../autoware_map_based_prediction/include/map_based_prediction/map_based_prediction_node.hpp
@@ -21,6 +21,7 @@
 
 #include <autoware_utils/geometry/geometry.hpp>
 #include <autoware_utils/ros/debug_publisher.hpp>
+#include <autoware_utils/ros/diagnostics_interface.hpp>
 #include <autoware_utils/ros/polling_subscriber.hpp>
 #include <autoware_utils/ros/published_time_publisher.hpp>
 #include <autoware_utils/ros/transform_listener.hpp>
@@ -115,6 +116,9 @@ class MapBasedPredictionNode : public rclcpp::Node
   // Predictor
   std::shared_ptr<PredictorVru> predictor_vru_;
 
+  // Diagnostics handler
+  std::unique_ptr<autoware_utils::DiagnosticsInterface> diagnostics_interface_ptr_;
+
   ////// Parameters
 
   //// Vehicle Parameters

diff --git a/perception/autoware_map_based_prediction/src/map_based_prediction_node.cpp b/perception/autoware_map_based_prediction/src/map_based_prediction_node.cpp
@@ -30,6 +30,7 @@
 #include <autoware_utils/ros/uuid_helper.hpp>
 
 #include <autoware_perception_msgs/msg/detected_objects.hpp>
+#include <diagnostic_msgs/msg/detail/diagnostic_status__struct.hpp>
 #include <tf2_geometry_msgs/tf2_geometry_msgs.hpp>
 
 #include <boost/geometry.hpp>
@@ -50,6 +51,7 @@
 #include <functional>
 #include <limits>
 #include <memory>
+#include <sstream>
 #include <string>
 #include <utility>
 #include <vector>
@@ -497,6 +499,10 @@ MapBasedPredictionNode::MapBasedPredictionNode(const rclcpp::NodeOptions & node_
     time_keeper_ = std::make_shared<autoware_utils::TimeKeeper>(time_keeper);
     path_generator_->setTimeKeeper(time_keeper_);
     predictor_vru_->setTimeKeeper(time_keeper_);
+
+    // diagnostics handler
+    diagnostics_interface_ptr_ =
+      std::make_unique<autoware_utils::DiagnosticsInterface>(this, "map_based_prediction");
   }
 
   if (use_debug_marker) {
@@ -671,6 +677,19 @@ void MapBasedPredictionNode::objectsCallback(const TrackedObjects::ConstSharedPt
       "debug/cyclic_time_ms", cyclic_time_ms);
     processing_time_publisher_->publish<autoware_internal_debug_msgs::msg::Float64Stamped>(
       "debug/processing_time_ms", processing_time_ms);
+
+    diagnostics_interface_ptr_->clear();
+    constexpr double processing_time_threshold_ms = 500.0;  // TODO(ktro2828): Use parameter value
+    bool is_processing_time = processing_time_ms > processing_time_threshold_ms;
+    diagnostics_interface_ptr_->add_key_value("is_processing_time", is_processing_time);
+    if (is_processing_time) {
+      std::ostringstream oss;
+      oss << "Processing time exceeded " << processing_time_threshold_ms << "[ms] < "
+          << processing_time_ms << "[ms]";
+      diagnostics_interface_ptr_->update_level_and_message(
+        diagnostic_msgs::msg::DiagnosticStatus::WARN, oss.str());
+    }
+    diagnostics_interface_ptr_->publish(output.header.stamp);
   }
 }