source_capture.h

#ifndef INCLUDE_GAZE_PIPELINE_STEPS_SOURCE_CAPTURE_H_
#define INCLUDE_GAZE_PIPELINE_STEPS_SOURCE_CAPTURE_H_

#include <memory>
#include <string>

#include "opencv2/opencv.hpp"

#include "gaze/gui/visualizeable.h"
#include "gaze/pipeline_step.h"
#include "gaze/util/data.h"


namespace gaze {

namespace pipeline {

class SourceCapture final
    : public PipelineStep,
      public gui::ImageVisualizeable {
  cv::VideoCapture video_capture;
  cv::Mat last_frame;

 public:
    SourceCapture();

    ~SourceCapture();

 public:
    void process(util::Data& data) override;

    void visualize(util::Data& data) override;
};

}  // namespace pipeline

}  // namespace gaze

#endif  // INCLUDE_GAZE_PIPELINE_STEPS_SOURCE_CAPTURE_H_