imageColor.hpp

Go to the documentation of this file.

 
#ifndef IMAGECOLOR_HPP
#define IMAGECOLOR_HPP
 
#include <sys/stat.h>
 
#include <algorithm>
#include <cctype>
#include <unordered_map>
 
#include "boost/filesystem.hpp"
 
#include <cpl.h>
#undef I
#undef arg
 
#include "ddt/dataAcquisition.hpp"
#include "ddt/dataFile.hpp"
#include "ddt/ddtLogger.hpp"
#include "ddt/imageError.hpp"
#include "ddt/imageIO.hpp"
#include "ddt/imageProc.hpp"
#include "ddt/imageStats.hpp"
 
namespace ddt {
 
constexpr size_t MAX_SCALING_LUT_ENTRIES = 65536;
constexpr uint MAX_SCALING_LUT_VALUE = 65535;
constexpr size_t MAX_COLOR_MAP_ENTRIES = 256;
constexpr size_t MAX_COLOR_MAP_VALUE = 255;
constexpr size_t COLOR_MAP_ENTRY_SIZE = 3;

using colorMap_t = std::array<std::array<float, COLOR_MAP_ENTRY_SIZE>, MAX_COLOR_MAP_ENTRIES>;

using colorMapARGB_t = std::array<unsigned int, MAX_COLOR_MAP_ENTRIES>;

typedef struct scalingLut_t {
  std::array<uint, MAX_SCALING_LUT_ENTRIES> lut;
  double offset;
  double factor;
} scalingLut_t;

class ImageColor {
 public:
  ImageColor();

  virtual ~ImageColor();

  void set_logger(ddt::DdtLogger* const in_logger);

  void LoadColorMaps(const std::string color_map_folder);

  void AddCustomColorMaps(const std::string color_map_name,
                          ddt::colorMap_t const& colourmap,
                          ddt::colorMap_t const& colourmap_inverted,
                          ddt::colorMapARGB_t const& colourmap_argb,
                          ddt::colorMapARGB_t const& colourmap_argb_inverted);

  std::list<std::string> get_ColorMapsList() const;

  ddt::colorMap_t* get_ColorMap(const std::string color_map_name);

  ddt::colorMapARGB_t* get_ColorMapARGB(const std::string color_map_name);

  ddt::colorMap_t* get_GrayScaleColorMap();

  ddt::colorMapARGB_t* get_GrayScaleColorMapARGB();

  ddt::scalingLut_t* get_LinearScalingLut(const double cut_level_min,
                                          const double cut_level_max);

  ddt::scalingLut_t* get_LogarithmicScalingLut(const double cut_level_min,
                                               const double cut_level_max);

  ddt::scalingLut_t* get_SqrtScalingLut(const double cut_level_min,
                                        const double cut_level_max);

  void set_InvertColourmap(const bool invert);
 
 protected:
  ddt::DdtLogger* logger;
 
 private:
  void Initialize();

  void LoadColorMap(std::string color_map_name);

  void LoadGrayScaleColorMap();

  void CalculateScalingParameter(const double cut_level_min,
                                 const double cut_level_max,
                                 double* const pixval_range,
                                 double* const offset, double* const factor,
                                 int* const cut_level_min_tmp,
                                 int* const cut_level_max_tmp) const;
  std::unordered_map<std::string, ddt::colorMap_t> color_maps;

  std::unordered_map<std::string, ddt::colorMap_t> color_maps_inverted;

  std::unordered_map<std::string, ddt::colorMapARGB_t> color_maps_argb;

  std::unordered_map<std::string, ddt::colorMapARGB_t> color_maps_argb_inverted;

  scalingLut_t linear_scaling_lut;

  scalingLut_t log_scaling_lut;

  scalingLut_t sqrt_scaling_lut;

  bool invert_colourmap;
};
 
}  // namespace ddt
 
#endif /* IMAGECOLOR_HPP */