improve image rendering performance by less waited locking

Author: christianrauch

image_view/include/image_view/image_view_node.hpp

@@ -31,18 +31,6 @@
 namespace image_view
 {
 
-class ThreadSafeImage
-{
-  std::mutex mutex_;
-  std::condition_variable condition_;
-  cv_bridge::CvImageConstPtr image_;
-
-public:
-  void set(cv_bridge::CvImageConstPtr image);
-  cv_bridge::CvImageConstPtr get();
-  cv_bridge::CvImageConstPtr pop();
-};
-
 class ImageViewNode
   : public rclcpp::Node
 {
@@ -55,7 +43,7 @@ class ImageViewNode
   ~ImageViewNode();
 
 private:
-  ThreadSafeImage queued_image_, shown_image_;
+  cv_bridge::CvImageConstPtr queued_image_, shown_image_;
   bool autosize_;
   int window_height_, window_width_;
   bool g_gui;
@@ -75,6 +63,8 @@ class ImageViewNode
   void windowThread();
   rcl_interfaces::msg::SetParametersResult paramCallback(const std::vector<rclcpp::Parameter> &);
   std::mutex param_mutex_;
+  std::mutex image_mutex_;
+  std::mutex new_data_mutex_;
 };
 
 }  // namespace image_view

image_view/src/image_view_node.cpp

@@ -70,38 +70,6 @@
 namespace image_view
 {
 
-void ThreadSafeImage::set(cv_bridge::CvImageConstPtr image)
-{
-  std::lock_guard<std::mutex> lock(mutex_);
-  image_ = image;
-  condition_.notify_one();
-}
-
-cv_bridge::CvImageConstPtr ThreadSafeImage::get()
-{
-  std::lock_guard<std::mutex> lock(mutex_);
-  return image_;
-}
-
-cv_bridge::CvImageConstPtr ThreadSafeImage::pop()
-{
-  cv_bridge::CvImageConstPtr image;
-
-  {
-    std::unique_lock<std::mutex> lock(mutex_);
-
-    condition_.wait_for(
-      lock, std::chrono::milliseconds(100),
-      [this] {
-        return !image_;
-      });
-
-    image = std::move(image_);
-  }
-
-  return image;
-}
-
 ImageViewNode::ImageViewNode(const rclcpp::NodeOptions & options)
 : rclcpp::Node("image_view_node", options)
 {
@@ -177,6 +145,9 @@ ImageViewNode::ImageViewNode(const rclcpp::NodeOptions & options)
 
 ImageViewNode::~ImageViewNode()
 {
+  image_mutex_.unlock();
+  new_data_mutex_.unlock();
+
   if (window_thread_.joinable()) {
     window_thread_.join();
   }
@@ -231,9 +202,10 @@ void ImageViewNode::imageCb(const sensor_msgs::msg::Image::ConstSharedPtr & msg)
       encoding = "bgr8";
     }
 
-    queued_image_.set(
-      cv_bridge::cvtColorForDisplay(
-        cv_bridge::toCvShare(msg), encoding, options));
+    image_mutex_.lock();
+    queued_image_ = cv_bridge::cvtColorForDisplay(cv_bridge::toCvShare(msg), encoding, options);
+    image_mutex_.unlock();
+    new_data_mutex_.unlock();
   } catch (cv_bridge::Exception & e) {
     RCLCPP_ERROR_EXPRESSION(
       this->get_logger(), (static_cast<int>(this->now().seconds()) % 30 == 0),
@@ -261,16 +233,23 @@ void ImageViewNode::mouseCb(int event, int /* x */, int /* y */, int /* flags */
     return;
   }
 
-  cv_bridge::CvImageConstPtr image(this_->shown_image_.get());
+  const std::string filename = string_format(this_->filename_format_, this_->count_);
 
-  if (!image) {
+  this_->image_mutex_.lock();
+
+  if (!this_->shown_image_) {
+    this_->image_mutex_.unlock();
     RCLCPP_WARN(this_->get_logger(), "Couldn't save image, no data!");
     return;
   }
 
-  std::string filename = string_format(this_->filename_format_, this_->count_);
+  const cv::Mat image = this_->shown_image_->image;
+
+  this_->image_mutex_.unlock();
+
+  const bool suc = cv::imwrite(filename, image);
 
-  if (cv::imwrite(filename, image->image)) {
+  if (suc) {
     RCLCPP_INFO(this_->get_logger(), "Saved image %s", filename.c_str());
     this_->count_++;
   } else {
@@ -291,19 +270,20 @@ void ImageViewNode::windowThread()
   }
 
   while (rclcpp::ok()) {
-    cv_bridge::CvImageConstPtr image(queued_image_.pop());
-
     if (cv::getWindowProperty(window_name_, 1) < 0) {
       break;
     }
 
-    if (image) {
-      cv::imshow(window_name_, image->image);
-      shown_image_.set(image);
-      cv::waitKey(1);
-    } else {
-      rclcpp::sleep_for(std::chrono::milliseconds(20));
+    // wait for new image data
+    new_data_mutex_.lock();
+
+    image_mutex_.lock();
+    if (queued_image_) {
+      cv::imshow(window_name_, queued_image_->image);
+      shown_image_ = queued_image_;
     }
+    image_mutex_.unlock();
+    cv::waitKey(1);
   }
 
   cv::destroyAllWindows();