1 files changed, 66 insertions, 0 deletions
diff --git a/imageutil.hpp b/imageutil.hpp
new file mode 100644
index 0000000..438c06b
--- /dev/null
+++ b/imageutil.hpp
@@ -0,0 +1,66 @@
+#ifndef IMAGEUTIL_HPP
+#define IMAGEUTIL_HPP
+
+#include <cstdlib>
+#include <vector>
+#include <opencv2/core.hpp>
+
+#include "compressed_vector.hpp"
+
+class image_util
+{
+public:
+    static cv::Mat crop(cv::InputArray s, double contrast_threshold, double max_crop_ratio);
+    static cv::Range crop_axis(cv::InputArray s, int axis, double contrast_threshold, double max_crop_ratio);
+    static double median(std::vector<double> &v);
+    static cv::Mat blend_white(cv::Mat m);
+
+    template<class T, int B>
+    static double length(const compressed_vector<T, B> &v, T center)
+    {
+        double ret = 0;
+        for (size_t i = 0; i < v.size(); ++i)
+        {
+            ret += (double)(v.get(i) - center) * (v.get(i) - center);
+        }
+        return sqrt(ret);
+    }
+    template<class T, int B>
+    static double distance(const compressed_vector<T, B> &v1, const compressed_vector<T, B> &v2)
+    {
+        //assert(v1.size() == v2.size())
+        double ret = 0;
+        for (size_t i = 0; i < v1.size(); ++i)
+        {
+            if (abs((int)v1.get(i) - (int)v2.get(i)) == 2 && (v1.get(i) == 2 || v2.get(i) == 2))
+            ret += 9;
+            else
+            ret += (double)(v1.get(i) - v2.get(i)) * (v1.get(i) - v2.get(i));
+        }
+        return sqrt(ret);
+    }
+    static double length(const std::vector<uint8_t> &v, uint8_t center)
+    {
+        double ret = 0;
+        for (size_t i = 0; i < v.size(); ++i)
+        {
+            ret += (double)(v[i] - center) * (v[i] - center);
+        }
+        return sqrt(ret);
+    }
+    static double distance(const std::vector<uint8_t> &v1, const std::vector<uint8_t> &v2)
+    {
+        //assert(v1.size() == v2.size())
+        double ret = 0;
+        for (size_t i = 0; i < v1.size(); ++i)
+        {
+            if (abs((int)v1[i] - (int)v2[i]) == 2 && (v1[i] == 2 || v2[i] == 2))
+            ret += 9;
+            else
+            ret += (double)(v1[i] - v2[i]) * (v1[i] - v2[i]);
+        }
+        return sqrt(ret);
+    }
+};
+
+#endif