diff options
Diffstat (limited to 'xsig/src/imageutil.hpp')
-rw-r--r-- | xsig/src/imageutil.hpp | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/xsig/src/imageutil.hpp b/xsig/src/imageutil.hpp new file mode 100644 index 0000000..f3831b0 --- /dev/null +++ b/xsig/src/imageutil.hpp @@ -0,0 +1,72 @@ +//Chris Xiong 2022 +//License: MPL-2.0 +#ifndef IMAGEUTIL_HPP +#define IMAGEUTIL_HPP + +#include <cmath> +#include <cstdlib> +#include <filesystem> +#include <vector> +#include <opencv2/core.hpp> + +#include "compressed_vector.hpp" + +#define sqr(x) ((x) * (x)) + +namespace image_util +{ + cv::Mat crop(cv::InputArray s, double contrast_threshold, double max_crop_ratio); + cv::Range crop_axis(cv::InputArray s, int axis, double contrast_threshold, double max_crop_ratio); + double median(std::vector<double> &v); + cv::Mat blend_white(cv::Mat m); + cv::Mat imread_path(const std::filesystem::path &p, int flags); + + 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 += sqr(1. * 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 += sqr(1. * 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 += sqr(1. * 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 += sqr(1. * v1[i] - v2[i]); + } + return sqrt(ret); + } +}; + +#endif |