mirror of
https://git.lynn.is/Gwen/pretty-automata.git
synced 2024-05-18 15:21:07 +02:00
59 lines
1.5 KiB
C++
59 lines
1.5 KiB
C++
#ifndef SHADER_AUTOMATON_COLORMAPS_H
|
|
#define SHADER_AUTOMATON_COLORMAPS_H
|
|
#include "../color.h"
|
|
#include <map>
|
|
|
|
struct ColorMapData {
|
|
private:
|
|
std::vector<float> data_{};
|
|
size_t size_{};
|
|
ColorSpace color_space_{};
|
|
|
|
public:
|
|
ColorMapData() : color_space_(ColorSpace::CIELAB) {}
|
|
explicit ColorMapData(const std::vector<Color>& colors) {
|
|
size_ = colors.size();
|
|
data_ = std::vector<float>(size_ * 3);
|
|
size_t i = 0;
|
|
for (const auto &item : colors) {
|
|
data_[i++] = static_cast<float>(item.v1);
|
|
data_[i++] = static_cast<float>(item.v2);
|
|
data_[i++] = static_cast<float>(item.v3);
|
|
}
|
|
}
|
|
|
|
static ColorMapData fromArray(const float *data, size_t num_colors, std::optional<ColorSpace> color_space) {
|
|
ColorMapData cmd{};
|
|
if (color_space.has_value()) {
|
|
cmd.color_space_ = color_space.value();
|
|
cmd.data_.assign(data, data + num_colors * 3);
|
|
} else {
|
|
cmd.data_ = std::vector<float>(num_colors * 3);
|
|
for (size_t i = 0; i < num_colors; ++i) {
|
|
auto c = Color::from_normalized_rgb(data[i * 3], data[i * 3 + 1], data[i * 3 + 2]);
|
|
cmd.data_[i * 3] = static_cast<float>(c.v1);
|
|
cmd.data_[i * 3 + 1] = static_cast<float>(c.v2);
|
|
cmd.data_[i * 3 + 2] = static_cast<float>(c.v3);
|
|
}
|
|
}
|
|
cmd.size_ = num_colors;
|
|
return cmd;
|
|
}
|
|
|
|
public:
|
|
[[nodiscard]] size_t size() const {
|
|
return size_;
|
|
}
|
|
|
|
[[nodiscard]] const float *data() const {
|
|
return data_.data();
|
|
}
|
|
};
|
|
|
|
|
|
|
|
extern const std::map<std::string, ColorMapData> colormaps;
|
|
|
|
|
|
#endif // SHADER_AUTOMATON_COLORMAPS_H
|