data.h

#ifndef INCLUDE_GAZE_UTIL_DATA_H_
#define INCLUDE_GAZE_UTIL_DATA_H_

#include <chrono>  // NOLINT
#include <iostream>
#include <map>
#include <string>
#include <utility>  // Not used, but cpplint detects swap() as part of this
#include <vector>

#include "opencv2/core.hpp"
#include "dlib/opencv.h"
#include "dlib/image_processing.h"


namespace gaze {

namespace util {

struct Data {
  Data();
  Data(const Data& data);

  Data& operator=(Data data);

  cv::Mat source_image;

  dlib::array2d<dlib::bgr_pixel> image;


  dlib::full_object_detection landmarks;

  dlib::array<dlib::array2d<double>> eyes;

  dlib::array<dlib::point> centers;

  cv::Mat head_rotation;

  cv::Mat head_translation;

  std::vector<cv::Vec3d> pupils;

  std::vector<cv::Vec3d> gaze_points;

  cv::Vec2d estimated_gaze_point;

  std::map<std::string, double> execution_times;

 private:
  void swap(Data& data);
};

inline std::ostream& operator<<(std::ostream& ostr,
                                const Data& data) {
  ostr << "Source image size: " << data.source_image.size() << std::endl
       << "dlib image size: " << data.image.size() << std::endl
       << "Face: " << data.landmarks.get_rect() << std::endl
       << "Eyes detected: " << data.eyes.size() << std::endl;
  return ostr;
}

}  // namespace util

}  // namespace gaze

#endif  // INCLUDE_GAZE_UTIL_DATA_H_