img2num.h

Go to the documentation of this file.

 
#ifndef IMG2NUM_H
#define IMG2NUM_H
 
#include <cstddef>
#include <cstdint>
#include <string>
 
namespace img2num {
 
struct ImageToSvgConfig {
    struct BilateralFilterConfig {
        double sigma_spatial = 3.0;
        double sigma_range = 50.0;
    } bilateral_filter;
 
    struct KMeansConfig {
        int32_t k = 16;
        int32_t max_iter = 100;
    } kmeans;
 
    int min_cluster_area = 100;
 
    int min_thickness = 0;
 
    uint8_t color_space = 0;
};
 
void gaussian_blur_fft(uint8_t* image, size_t width, size_t height, double sigma);
 
void invert_image(uint8_t* ptr, int width, int height);
 
void threshold_image(uint8_t* ptr, const int width, const int height, const int num_thresholds);
 
void black_threshold_image(
    uint8_t* ptr, const int width, const int height, const int num_thresholds
);
 
void kmeans(
    const uint8_t* data, uint8_t* out_data, int32_t* out_labels, const int32_t width,
    const int32_t height, const int32_t k, const int32_t max_iter, const uint8_t color_space
);
 
void bilateral_filter(
    uint8_t* image, size_t width, size_t height, double sigma_spatial, double sigma_range,
    uint8_t color_space
);
 
std::string labels_to_svg(
    const uint8_t* data, const int32_t* labels, const int width, const int height,
    const int min_area, const int min_thickness
);
 
std::string image_to_svg(
    const uint8_t* data, const int width, const int height, const ImageToSvgConfig& config
);
 
} // namespace img2num
 
#endif // IMG2NUM_H