imageprofile.h

 
#ifndef IMAGE_PROFILE_H
#define IMAGE_PROFILE_H
 
#include "iniparser.h" // Assuming declarations for IniReader, Saver, distributionBox
#include "imghelpers.h"
#include "saver.h"
#include "generic.h"
#include <kll_sketch.hpp>
#include <vector>
#include <string>
#include <unordered_map>
#include <functional>
 
 
// Typedef for distribution box data structure (assuming datasketches::kll_sketch<float>)
typedef datasketches::kll_sketch<float> distributionBox;
 
class ImageProfile {
public:
    ~ImageProfile();
    
    ImageProfile(const std::string& conf_path, int save_interval, int channels = 1);
 
    int profile(cv::Mat& img, bool save_sample = false);
 
#ifndef TEST
private:
#endif
    Saver* saver;
    std::string statSavepath;
    std::string dataSavepath;
    std::map<std::string, std::vector<std::string>> imageConfig;
    int channels;
  distributionBox contrastBox;
 
  distributionBox brightnessBox;
  distributionBox sharpnessBox;
 
  std::vector<distributionBox *> pixelBox;
  std::vector<distributionBox *> meanBox;
 
  distributionBox noiseBox;
 
 
    void registerStatistics(const std::string& name);
 
    float computeStatistic(const std::string& name, cv::Mat& img);
 
    bool isThresholdExceeded(const std::string& name, float stat_score, const std::vector<std::string>& config);
 
    void iterateImage(const cv::Mat& img, const std::function<void(const std::vector<int>&)>& callback);
 
    void updatePixelValues(const std::vector<int>& pixelValues);
};
 
#endif // IMAGEPROFILE_H