61 lines
1.5 KiB
C++
61 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <array>
|
|
#include <cstdint>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
|
|
#include "intmath.h"
|
|
|
|
constexpr int32_t kMinColor = 0;
|
|
constexpr int32_t kMaxColor = UINT16_MAX;
|
|
|
|
// 32-bit for compiler convenience, but values are 16-bit
|
|
template <uint32_t C>
|
|
struct Color : public std::array<int32_t, C> {
|
|
constexpr uint32_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> {};
|
|
|
|
template <uint32_t C>
|
|
constexpr uint32_t Color<C>::AbsDiff(const Color<C>& other) const {
|
|
uint32_t diff = 0;
|
|
for (uint32_t c = 0; c < C; ++c) {
|
|
diff += static_cast<uint32_t>(::AbsDiff(this->at(c), other.at(c)));
|
|
}
|
|
return diff;
|
|
}
|
|
|
|
template <uint32_t C>
|
|
constexpr Color<C> Color<C>::Interpolate(const Color<C>& other, int32_t mul, int32_t div) const {
|
|
Color<C> ret;
|
|
for (uint32_t c = 0; c < C; ++c) {
|
|
ret.at(c) = ::Interpolate(this->at(c), other.at(c), mul, div);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
template <uint32_t C>
|
|
constexpr Color<C> Color<C>::Crop() const {
|
|
Color<C> ret;
|
|
for (uint32_t c = 0; c < C; ++c) {
|
|
ret.at(c) = std::max(kMinColor, std::min(kMaxColor, this->at(c)));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
template <uint32_t C>
|
|
std::ostream& operator<<(std::ostream& os, const Color<C>& color) {
|
|
os << std::hex << std::setfill('0') << "Color(";
|
|
for (uint32_t c = 0; c < C; ++c) {
|
|
os << "0x" << std::setw(4) << color.at(0);
|
|
if (c < C - 1) {
|
|
os << ", ";
|
|
}
|
|
}
|
|
return os << ")" << std::dec;
|
|
}
|