visualizeable.h

#ifndef INCLUDE_GAZE_GUI_VISUALIZEABLE_H_
#define INCLUDE_GAZE_GUI_VISUALIZEABLE_H_

#include <memory>
#include <string>

#include "dlib/gui_widgets.h"

#include "gaze/util/data.h"


namespace gaze {

namespace gui {

namespace traits {


struct sizeable {};

struct textable {};


template<typename widget_type>
struct widget_trait;

template<> struct widget_trait<dlib::label> {
  typedef textable tag;
};

template<> struct widget_trait<dlib::image_display> {
  typedef sizeable tag;
};

template<> struct widget_trait<dlib::perspective_display> {
  typedef sizeable tag;
};

}  // namespace traits


template<typename widget_type>
void adjust_widget(std::shared_ptr<widget_type> widget,
    int width, int height, std::string, traits::sizeable) {
  widget->set_size(width, height);
}

template<typename widget_type>
void adjust_widget(std::shared_ptr<widget_type> widget,
    int, int, std::string text, traits::textable) {
  widget->set_text(text);
}

template<typename widget_type>
void adjust_widget(std::shared_ptr<widget_type> widget,
    int width, int height, std::string text) {
  adjust_widget(widget, width, height, text,
      typename traits::widget_trait<widget_type>::tag());
}

template<typename widget_type>
// NOLINTNEXTLINE
std::shared_ptr<widget_type> create_widget(dlib::drawable_window& parent) {
  return std::shared_ptr<widget_type>(new widget_type(parent));
}

class VisualizeableBase {
 public:
  virtual ~VisualizeableBase() = default;

  virtual std::shared_ptr<dlib::drawable> init(
      dlib::drawable_window& parent,  // NOLINT
      int width, int height,
      std::string text = "No visualization.") = 0;

  virtual void remove_widget() {}

  virtual void visualize(util::Data& data) = 0;  // NOLINT
};

template<typename widget_type = dlib::label>
class Visualizeable : public VisualizeableBase {
 protected:
  std::shared_ptr<widget_type> widget;

 public:
  virtual ~Visualizeable() = default;

  std::shared_ptr<dlib::drawable> init(
      dlib::drawable_window& parent,  // NOLINT
      int width, int height,
      std::string text = "No visualization.") final {
    widget = create_widget<widget_type>(parent);
    adjust_widget<widget_type>(widget, width, height, text);
    return widget;
  }

  void remove_widget() override {
    if (this->widget && this->widget.get()) {
      this->widget->disable();
    }
    this->widget = std::shared_ptr<widget_type>(nullptr);
  }
};

typedef Visualizeable<dlib::label> LabelVisualizeable;

typedef Visualizeable<dlib::image_display> ImageVisualizeable;

typedef Visualizeable<dlib::perspective_display> PerspectiveVisualizeable;

}  // namespace gui

}  // namespace gaze

#endif  // INCLUDE_GAZE_GUI_VISUALIZEABLE_H_