capture.h

Go to the documentation of this file.

#ifndef CV_CAMERA_CAPTURE_H
#define CV_CAMERA_CAPTURE_H

#include "cv_camera/exception.h"

#include <ros/ros.h>
#include <cv_bridge/cv_bridge.h>
#include <image_transport/image_transport.h>
#include <sensor_msgs/CameraInfo.h>
#include <sensor_msgs/image_encodings.h>
#include <opencv2/highgui/highgui.hpp>
#include <camera_info_manager/camera_info_manager.h>

namespace cv_camera {

class Capture {
 public:
  Capture(ros::NodeHandle& node,
          const std::string& topic_name,
          int32_t buffer_size,
          const std::string& frame_id);
  
  void open(int32_t device_id);
  
  void open();

  void openFile(const std::string& file_path);

  bool capture();

  void publish();

  const sensor_msgs::CameraInfo& getInfo() const
  {
    return info_;
  }

  const cv::Mat& getCvImage() const
  {
    return bridge_.image;
  }
  
  const sensor_msgs::ImagePtr getImageMsgPtr() const
  {
    return bridge_.toImageMsg();
  }

  bool setWidth(int32_t width) {
    return cap_.set(CV_CAP_PROP_FRAME_WIDTH, width);
  }

  bool setHeight(int32_t height) {
    return cap_.set(CV_CAP_PROP_FRAME_HEIGHT, height);
  }

 private:

  ros::NodeHandle node_;

  image_transport::ImageTransport it_;

  std::string topic_name_;

  std::string frame_id_;
  int32_t buffer_size_;

  image_transport::CameraPublisher pub_;

  cv::VideoCapture cap_;

  cv_bridge::CvImage bridge_;

  sensor_msgs::CameraInfo info_;

  camera_info_manager::CameraInfoManager info_manager_;
};

}  // end namespace

#endif  // end CV_CAMERA_CAPTURE_H