From c96b772e89237ea55877da424e8e699691ecbc2d Mon Sep 17 00:00:00 2001
From: Phuntsok Drak-pa <phundrak@phundrak.fr>
Date: Tue, 19 Mar 2019 15:26:03 +0100
Subject: [PATCH] added helper function for drawing squares

---
 include/genimg/drawing.hh | 10 ++++++++++
 src/drawing.cc            | 13 +++++++++++++
 2 files changed, 23 insertions(+)
 create mode 100644 include/genimg/drawing.hh
 create mode 100644 src/drawing.cc

diff --git a/include/genimg/drawing.hh b/include/genimg/drawing.hh
new file mode 100644
index 0000000..4a1b838
--- /dev/null
+++ b/include/genimg/drawing.hh
@@ -0,0 +1,10 @@
+#ifndef GENETIC_IMAGE_INCLUDE_GENIMG_DRAWING_HH_
+#define GENETIC_IMAGE_INCLUDE_GENIMG_DRAWING_HH_
+
+#include <opencv2/core/core.hpp>
+#include <opencv2/highgui/highgui.hpp>
+
+void draw_square(cv::Mat &t_img, cv::Point const &t_top_left, int const size,
+                 cv::Scalar const &t_color);
+
+#endif /* GENETIC_IMAGE_INCLUDE_GENIMG_DRAWING_HH_ */
diff --git a/src/drawing.cc b/src/drawing.cc
new file mode 100644
index 0000000..c4b9c25
--- /dev/null
+++ b/src/drawing.cc
@@ -0,0 +1,13 @@
+#include "drawing.hh"
+#include <memory>
+#include <opencv2/imgproc.hpp>
+
+void draw_square(cv::Mat &t_img, cv::Point const &t_top_left, int const size,
+                 cv::Scalar const &t_color) {
+  std::unique_ptr<cv::Point> points(new cv::Point[4]);
+  points.get()[0] = t_top_left;
+  points.get()[1] = cv::Point{t_top_left.x,        t_top_left.y + size};
+  points.get()[2] = cv::Point{t_top_left.x + size, t_top_left.y + size};
+  points.get()[3] = cv::Point{t_top_left.x + size, t_top_left.y};
+  fillConvexPoly(t_img, points.get(), 4, t_color);
+}