rgb.hpp

 
#ifndef UFO_VISION_COLOR_RGB_HPP
#define UFO_VISION_COLOR_RGB_HPP
 
// UFO
#include <ufo/vision/color/flags.hpp>
 
// STL
#include <concepts>
#include <cstdint>
#include <format>
#include <ostream>
 
namespace ufo
{
template <class T = float, ColorFlags Flags = ColorFlags::None>
  requires(std::integral<T> || std::floating_point<T>)
struct Rgb;
 
template <class T>
  requires(std::integral<T> || std::floating_point<T>)
struct Rgb<T, ColorFlags::None> {
    T red;
 
    T green;
 
    T blue;
 
    [[nodiscard]] friend constexpr bool operator==(Rgb const& a,
                                                   Rgb const& b) noexcept = default;
};
 
template <class T>
  requires(std::integral<T> || std::floating_point<T>)
struct Rgb<T, ColorFlags::Alpha> {
    T red;
 
    T green;
 
    T blue;
 
    T alpha;
 
    [[nodiscard]] friend constexpr bool operator==(Rgb const& a,
                                                   Rgb const& b) noexcept = default;
};
 
template <class T>
  requires(std::integral<T> || std::floating_point<T>)
struct Rgb<T, ColorFlags::Weight> {
    T red;
 
    T green;
 
    T blue;
 
    float weight;
 
    [[nodiscard]] friend constexpr bool operator==(Rgb const& a,
                                                   Rgb const& b) noexcept = default;
};
 
template <class T>
  requires(std::integral<T> || std::floating_point<T>)
struct Rgb<T, ColorFlags::Alpha | ColorFlags::Weight> {
    T red;
 
    T green;
 
    T blue;
 
    T alpha;
 
    float weight;
 
    [[nodiscard]] friend constexpr bool operator==(Rgb const& a,
                                                   Rgb const& b) noexcept = default;
};
 
template <class T, ColorFlags Flags>
[[nodiscard]] constexpr T red(Rgb<T, Flags> color) noexcept
{
    if constexpr (weightset(Flags) && std::floating_point<T>) {
        return color.red / static_cast<T>(color.weight);
    } else {
        return color.red;
    }
}
 
template <class T, ColorFlags Flags>
[[nodiscard]] constexpr T green(Rgb<T, Flags> color) noexcept
{
    if constexpr (weightset(Flags) && std::floating_point<T>) {
        return color.green / static_cast<T>(color.weight);
    } else {
        return color.green;
    }
}
 
template <class T, ColorFlags Flags>
[[nodiscard]] constexpr T blue(Rgb<T, Flags> color) noexcept
{
    if constexpr (weightset(Flags) && std::floating_point<T>) {
        return color.blue / static_cast<T>(color.weight);
    } else {
        return color.blue;
    }
}
 
using SmallRgb = Rgb<std::uint8_t>;
 
using SmallRgbA = Rgb<std::uint8_t, ColorFlags::Alpha>;
 
using SmallRgbW = Rgb<std::uint8_t, ColorFlags::Weight>;
 
using SmallRgbAW = Rgb<std::uint8_t, ColorFlags::Alpha | ColorFlags::Weight>;
 
using FineRgb = Rgb<float>;
 
using FineRgbA = Rgb<float, ColorFlags::Alpha>;
 
using FineRgbW = Rgb<float, ColorFlags::Weight>;
 
using FineRgbAW = Rgb<float, ColorFlags::Alpha | ColorFlags::Weight>;
 
template <class T, ColorFlags Flags>
inline std::ostream& operator<<(std::ostream& out, Rgb<T, Flags> const& c)
{
    out << "Red: " << +c.red << " Green: " << +c.green << " Blue: " << +c.blue;
    if constexpr (alphaset(Flags)) {
        out << " Alpha: " << +c.alpha;
    }
    if constexpr (weightset(Flags)) {
        out << " Weight: " << c.weight;
    }
    return out;
}
 
}  // namespace ufo
 
template <class T, ufo::ColorFlags Flags>
struct std::formatter<ufo::Rgb<T, Flags>> {
    constexpr auto parse(std::format_parse_context& ctx) { return ctx.begin(); }
 
    auto format(ufo::Rgb<T, Flags> const& c, std::format_context& ctx) const
    {
        auto out = std::format_to(ctx.out(), "Red: {} Green: {} Blue: {}", +c.red, +c.green,
                                  +c.blue);
        if constexpr (ufo::alphaset(Flags)) {
            out = std::format_to(out, " Alpha: {}", +c.alpha);
        }
        if constexpr (ufo::weightset(Flags)) {
            out = std::format_to(out, " Weight: {}", c.weight);
        }
        return out;
    }
};
 
#endif  // UFO_VISION_COLOR_RGB_HPP