gaze_capture.h

#ifndef INCLUDE_GAZE_PIPELINE_STEPS_GAZE_CAPTURE_H_
#define INCLUDE_GAZE_PIPELINE_STEPS_GAZE_CAPTURE_H_

#ifdef HAS_CAFFE
  #include "itracker/itracker.h"

  #include "opencv2/opencv.hpp"
#endif

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


namespace gaze {

namespace pipeline {

class GazeCapture final
    : public PipelineStep,
      public gui::ImageVisualizeable {
  #ifdef HAS_CAFFE
    std::unique_ptr<itracker::ITracker> itracker;

    cv::Vec2d camera_offset;
    cv::Vec2d screen_size_m;
    cv::Vec2d target_size;
  #endif

 public:
  GazeCapture();

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

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

}  // namespace pipeline

}  // namespace gaze

#endif  // INCLUDE_GAZE_PIPELINE_STEPS_GAZE_CAPTURE_H_