88 lines
2.8 KiB
C++
88 lines
2.8 KiB
C++
#pragma once
|
|
|
|
#include <filesystem>
|
|
#include <opencv2/core/core.hpp>
|
|
#include <opencv2/highgui/highgui.hpp>
|
|
#include <opencv2/imgproc.hpp>
|
|
#include <spdlog/spdlog.h>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <tuple>
|
|
#include <vector>
|
|
|
|
class ImageManipulator {
|
|
public:
|
|
ImageManipulator() = delete;
|
|
ImageManipulator(const ImageManipulator& other);
|
|
ImageManipulator(ImageManipulator&& other) noexcept;
|
|
[[nodiscard]] auto operator=(const ImageManipulator& other)
|
|
-> ImageManipulator;
|
|
[[nodiscard]] auto operator=(ImageManipulator&& other) noexcept
|
|
-> ImageManipulator;
|
|
|
|
// Load image from input, and prepare for output
|
|
ImageManipulator(std::filesystem::path const t_input_path,
|
|
std::filesystem::path const t_output_path,
|
|
int const iterations);
|
|
ImageManipulator(cv::Mat const& t_origin_image,
|
|
int const iterations,
|
|
int const t_x,
|
|
int const t_y,
|
|
int const t_width,
|
|
int const t_height);
|
|
|
|
void exec_method(int const t_nb_method,
|
|
bool const t_controlled_size,
|
|
int const t_cols,
|
|
int const t_rows,
|
|
int const t_submethod);
|
|
void write_file() const;
|
|
[[nodiscard]] auto const& get_generated_image() const noexcept
|
|
{
|
|
return generated_image_;
|
|
}
|
|
|
|
//! Destructor
|
|
virtual ~ImageManipulator() noexcept = default;
|
|
|
|
protected:
|
|
private:
|
|
[[nodiscard]] auto euclidian_distance(cv::Mat const& t_img) const noexcept
|
|
-> double;
|
|
[[nodiscard]] auto random_color() const noexcept;
|
|
[[nodiscard]] auto get_square_values() const noexcept;
|
|
[[nodiscard]] auto get_controlled_square_values() const noexcept;
|
|
[[nodiscard]] auto create_candidate(bool const t_controlled_size) const;
|
|
|
|
void get_color_set();
|
|
void threaded_get_color(int t_h);
|
|
void adjust_size(cv::Point& t_top_left, int const size) noexcept;
|
|
void draw_square(cv::Mat& t_img,
|
|
cv::Point const& t_top_left,
|
|
int const t_size,
|
|
cv::Scalar const& t_color) const;
|
|
void update_gen_image(cv::Mat const& t_img, double const t_diff);
|
|
void method1();
|
|
void method2();
|
|
void method3();
|
|
void method4(bool const t_controlled_size);
|
|
void method5(bool const t_controlled_size,
|
|
int cols,
|
|
int const rows,
|
|
int const submethod);
|
|
|
|
std::vector<std::array<uchar, 3>> colors_
|
|
= std::vector<std::array<uchar, 3>>{};
|
|
cv::Mat const reference_;
|
|
cv::Mat generated_image_
|
|
= cv::Mat{reference_.size().height, reference_.size().width, CV_8UC3,
|
|
cv::Scalar(0, 0, 0)};
|
|
mutable std::mutex colors_mutex_ = std::mutex{};
|
|
std::string const output_path_{""};
|
|
double diff_ = 0.0;
|
|
int const total_iterations_ = 0;
|
|
int remaining_iter_ = total_iterations_;
|
|
int const width_ = reference_.size().width;
|
|
int const height_ = reference_.size().height;
|
|
};
|