68 lines
1.5 KiB
C++
68 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
|
|
#include "array.h"
|
|
#include "intmath.h"
|
|
|
|
constexpr int32_t kMinColor = 0;
|
|
constexpr int32_t kMaxColor = UINT16_MAX;
|
|
|
|
class ColorBase {};
|
|
|
|
template <int32_t C>
|
|
struct Color : public Array<int32_t, C>, public ColorBase {
|
|
constexpr int32_t AbsDiff(const Color<C>& other) const;
|
|
constexpr Color<C> Interpolate(const Color<C>& other, int32_t mul, int32_t div) const;
|
|
constexpr Color<C> Crop() const;
|
|
};
|
|
|
|
|
|
struct RgbColor : public Color<3> {
|
|
public:
|
|
RgbColor() = default;
|
|
RgbColor(const Color<3>& src);
|
|
};
|
|
|
|
|
|
template <int32_t C>
|
|
constexpr int32_t Color<C>::AbsDiff(const Color<C>& other) const {
|
|
int32_t diff = 0;
|
|
for (int32_t c = 0; c < C; ++c) {
|
|
diff += ::AbsDiff(this->at(c), other.at(c));
|
|
}
|
|
return diff;
|
|
}
|
|
|
|
template <int32_t C>
|
|
constexpr Color<C> Color<C>::Interpolate(const Color<C>& other, int32_t mul, int32_t div) const {
|
|
Color<C> ret;
|
|
for (int32_t c = 0; c < C; ++c) {
|
|
ret.at(c) = ::Interpolate(this->at(c), other.at(c), mul, div);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
template <int32_t C>
|
|
constexpr Color<C> Color<C>::Crop() const {
|
|
Color<C> ret;
|
|
for (int32_t c = 0; c < C; ++c) {
|
|
ret.at(c) = std::max(kMinColor, std::min(kMaxColor, this->at(c)));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
template <int32_t C>
|
|
std::ostream& operator<<(std::ostream& os, const Color<C>& color) {
|
|
os << std::hex << std::setfill('0') << "Color(";
|
|
for (int32_t c = 0; c < C; ++c) {
|
|
os << "0x" << std::setw(4) << color.at(0);
|
|
if (c < C - 1) {
|
|
os << ", ";
|
|
}
|
|
}
|
|
return os << ")" << std::dec;
|
|
}
|