// File: crn_color.h
// See Copyright Notice and license at the end of inc/crnlib.h
#pragma once
#include "crn_core.h"

namespace crnlib {
template <typename component_type>
struct color_quad_component_traits {
  enum {
    cSigned = false,
    cFloat = false,
    cMin = cUINT8_MIN,
    cMax = cUINT8_MAX
  };
};

template <>
struct color_quad_component_traits<int8> {
  enum {
    cSigned = true,
    cFloat = false,
    cMin = cINT8_MIN,
    cMax = cINT8_MAX
  };
};

template <>
struct color_quad_component_traits<int16> {
  enum {
    cSigned = true,
    cFloat = false,
    cMin = cINT16_MIN,
    cMax = cINT16_MAX
  };
};

template <>
struct color_quad_component_traits<uint16> {
  enum {
    cSigned = false,
    cFloat = false,
    cMin = cUINT16_MIN,
    cMax = cUINT16_MAX
  };
};

template <>
struct color_quad_component_traits<int32> {
  enum {
    cSigned = true,
    cFloat = false,
    cMin = cINT32_MIN,
    cMax = cINT32_MAX
  };
};

template <>
struct color_quad_component_traits<uint32> {
  enum {
    cSigned = false,
    cFloat = false,
    cMin = cUINT32_MIN,
    cMax = cUINT32_MAX
  };
};

template <>
struct color_quad_component_traits<float> {
  enum {
    cSigned = false,
    cFloat = true,
    cMin = cINT32_MIN,
    cMax = cINT32_MAX
  };
};

template <>
struct color_quad_component_traits<double> {
  enum {
    cSigned = false,
    cFloat = true,
    cMin = cINT32_MIN,
    cMax = cINT32_MAX
  };
};

template <typename component_type, typename parameter_type>
class color_quad : public helpers::rel_ops<color_quad<component_type, parameter_type> > {
  template <typename T>
  static inline parameter_type clamp(T v) {
    parameter_type result = static_cast<parameter_type>(v);
    if (!component_traits::cFloat) {
      if (v < component_traits::cMin)
        result = static_cast<parameter_type>(component_traits::cMin);
      else if (v > component_traits::cMax)
        result = static_cast<parameter_type>(component_traits::cMax);
    }
    return result;
  }

#ifdef _MSC_VER
  template <>
  static inline parameter_type clamp(int v) {
    if (!component_traits::cFloat) {
      if ((!component_traits::cSigned) && (component_traits::cMin == 0) && (component_traits::cMax == 0xFF)) {
        if (v & 0xFFFFFF00U)
          v = (~(static_cast<int>(v) >> 31)) & 0xFF;
      } else {
        if (v < component_traits::cMin)
          v = component_traits::cMin;
        else if (v > component_traits::cMax)
          v = component_traits::cMax;
      }
    }
    return static_cast<parameter_type>(v);
  }
#endif

 public:
  typedef component_type component_t;
  typedef parameter_type parameter_t;
  typedef color_quad_component_traits<component_type> component_traits;

  enum { cNumComps = 4 };

  union {
    struct
    {
      component_type r;
      component_type g;
      component_type b;
      component_type a;
    };

    component_type c[cNumComps];

    uint32 m_u32;
  };

  inline color_quad() {
  }

  inline color_quad(eClear)
      : r(0), g(0), b(0), a(0) {
  }

  inline color_quad(const color_quad& other)
      : r(other.r), g(other.g), b(other.b), a(other.a) {
  }

  explicit inline color_quad(parameter_type y, parameter_type alpha = component_traits::cMax) {
    set(y, alpha);
  }

  inline color_quad(parameter_type red, parameter_type green, parameter_type blue, parameter_type alpha = component_traits::cMax) {
    set(red, green, blue, alpha);
  }

  explicit inline color_quad(eNoClamp, parameter_type y, parameter_type alpha = component_traits::cMax) {
    set_noclamp_y_alpha(y, alpha);
  }

  inline color_quad(eNoClamp, parameter_type red, parameter_type green, parameter_type blue, parameter_type alpha = component_traits::cMax) {
    set_noclamp_rgba(red, green, blue, alpha);
  }

  template <typename other_component_type, typename other_parameter_type>
  inline color_quad(const color_quad<other_component_type, other_parameter_type>& other)
      : r(static_cast<component_type>(clamp(other.r))), g(static_cast<component_type>(clamp(other.g))), b(static_cast<component_type>(clamp(other.b))), a(static_cast<component_type>(clamp(other.a))) {
  }

  inline void clear() {
    r = 0;
    g = 0;
    b = 0;
    a = 0;
  }

  inline color_quad& operator=(const color_quad& other) {
    r = other.r;
    g = other.g;
    b = other.b;
    a = other.a;
    return *this;
  }

  inline color_quad& set_rgb(const color_quad& other) {
    r = other.r;
    g = other.g;
    b = other.b;
    return *this;
  }

  template <typename other_component_type, typename other_parameter_type>
  inline color_quad& operator=(const color_quad<other_component_type, other_parameter_type>& other) {
    r = static_cast<component_type>(clamp(other.r));
    g = static_cast<component_type>(clamp(other.g));
    b = static_cast<component_type>(clamp(other.b));
    a = static_cast<component_type>(clamp(other.a));
    return *this;
  }

  inline color_quad& operator=(parameter_type y) {
    set(y, component_traits::cMax);
    return *this;
  }

  inline color_quad& set(parameter_type y, parameter_type alpha = component_traits::cMax) {
    y = clamp(y);
    alpha = clamp(alpha);
    r = static_cast<component_type>(y);
    g = static_cast<component_type>(y);
    b = static_cast<component_type>(y);
    a = static_cast<component_type>(alpha);
    return *this;
  }

  inline color_quad& set_noclamp_y_alpha(parameter_type y, parameter_type alpha = component_traits::cMax) {
    CRNLIB_ASSERT((y >= component_traits::cMin) && (y <= component_traits::cMax));
    CRNLIB_ASSERT((alpha >= component_traits::cMin) && (alpha <= component_traits::cMax));

    r = static_cast<component_type>(y);
    g = static_cast<component_type>(y);
    b = static_cast<component_type>(y);
    a = static_cast<component_type>(alpha);
    return *this;
  }

  inline color_quad& set(parameter_type red, parameter_type green, parameter_type blue, parameter_type alpha = component_traits::cMax) {
    r = static_cast<component_type>(clamp(red));
    g = static_cast<component_type>(clamp(green));
    b = static_cast<component_type>(clamp(blue));
    a = static_cast<component_type>(clamp(alpha));
    return *this;
  }

  inline color_quad& set_noclamp_rgba(parameter_type red, parameter_type green, parameter_type blue, parameter_type alpha) {
    CRNLIB_ASSERT((red >= component_traits::cMin) && (red <= component_traits::cMax));
    CRNLIB_ASSERT((green >= component_traits::cMin) && (green <= component_traits::cMax));
    CRNLIB_ASSERT((blue >= component_traits::cMin) && (blue <= component_traits::cMax));
    CRNLIB_ASSERT((alpha >= component_traits::cMin) && (alpha <= component_traits::cMax));

    r = static_cast<component_type>(red);
    g = static_cast<component_type>(green);
    b = static_cast<component_type>(blue);
    a = static_cast<component_type>(alpha);
    return *this;
  }

  inline color_quad& set_noclamp_rgb(parameter_type red, parameter_type green, parameter_type blue) {
    CRNLIB_ASSERT((red >= component_traits::cMin) && (red <= component_traits::cMax));
    CRNLIB_ASSERT((green >= component_traits::cMin) && (green <= component_traits::cMax));
    CRNLIB_ASSERT((blue >= component_traits::cMin) && (blue <= component_traits::cMax));

    r = static_cast<component_type>(red);
    g = static_cast<component_type>(green);
    b = static_cast<component_type>(blue);
    return *this;
  }

  static inline parameter_type get_min_comp() { return component_traits::cMin; }
  static inline parameter_type get_max_comp() { return component_traits::cMax; }
  static inline bool get_comps_are_signed() { return component_traits::cSigned; }

  inline component_type operator[](uint i) const {
    CRNLIB_ASSERT(i < cNumComps);
    return c[i];
  }
  inline component_type& operator[](uint i) {
    CRNLIB_ASSERT(i < cNumComps);
    return c[i];
  }

  inline color_quad& set_component(uint i, parameter_type f) {
    CRNLIB_ASSERT(i < cNumComps);

    c[i] = static_cast<component_type>(clamp(f));

    return *this;
  }

  inline color_quad& set_grayscale(parameter_t l) {
    component_t x = static_cast<component_t>(clamp(l));
    c[0] = x;
    c[1] = x;
    c[2] = x;
    return *this;
  }

  inline color_quad& clamp(const color_quad& l, const color_quad& h) {
    for (uint i = 0; i < cNumComps; i++)
      c[i] = static_cast<component_type>(math::clamp<parameter_type>(c[i], l[i], h[i]));
    return *this;
  }

  inline color_quad& clamp(parameter_type l, parameter_type h) {
    for (uint i = 0; i < cNumComps; i++)
      c[i] = static_cast<component_type>(math::clamp<parameter_type>(c[i], l, h));
    return *this;
  }

  // Returns CCIR 601 luma (consistent with color_utils::RGB_To_Y).
  inline parameter_type get_luma() const {
    return static_cast<parameter_type>((19595U * r + 38470U * g + 7471U * b + 32768U) >> 16U);
  }

  // Returns REC 709 luma.
  inline parameter_type get_luma_rec709() const {
    return static_cast<parameter_type>((13938U * r + 46869U * g + 4729U * b + 32768U) >> 16U);
  }

  // Beware of endianness!
  inline uint32 get_uint32() const {
    CRNLIB_ASSERT(sizeof(*this) == sizeof(uint32));
    return *reinterpret_cast<const uint32*>(this);
  }

  // Beware of endianness!
  inline uint64 get_uint64() const {
    CRNLIB_ASSERT(sizeof(*this) == sizeof(uint64));
    return *reinterpret_cast<const uint64*>(this);
  }

  inline uint squared_distance(const color_quad& c, bool alpha = true) const {
    return math::square(r - c.r) + math::square(g - c.g) + math::square(b - c.b) + (alpha ? math::square(a - c.a) : 0);
  }

  inline bool rgb_equals(const color_quad& rhs) const {
    return (r == rhs.r) && (g == rhs.g) && (b == rhs.b);
  }

  inline bool operator==(const color_quad& rhs) const {
    if (sizeof(color_quad) == sizeof(uint32))
      return m_u32 == rhs.m_u32;
    else
      return (r == rhs.r) && (g == rhs.g) && (b == rhs.b) && (a == rhs.a);
  }

  inline bool operator<(const color_quad& rhs) const {
    for (uint i = 0; i < cNumComps; i++) {
      if (c[i] < rhs.c[i])
        return true;
      else if (!(c[i] == rhs.c[i]))
        return false;
    }
    return false;
  }

  color_quad& operator+=(const color_quad& other) {
    for (uint i = 0; i < 4; i++)
      c[i] = static_cast<component_type>(clamp(c[i] + other.c[i]));
    return *this;
  }

  color_quad& operator-=(const color_quad& other) {
    for (uint i = 0; i < 4; i++)
      c[i] = static_cast<component_type>(clamp(c[i] - other.c[i]));
    return *this;
  }

  color_quad& operator*=(parameter_type v) {
    for (uint i = 0; i < 4; i++)
      c[i] = static_cast<component_type>(clamp(c[i] * v));
    return *this;
  }

  color_quad& operator/=(parameter_type v) {
    for (uint i = 0; i < 4; i++)
      c[i] = static_cast<component_type>(c[i] / v);
    return *this;
  }

  color_quad get_swizzled(uint x, uint y, uint z, uint w) const {
    CRNLIB_ASSERT((x | y | z | w) < 4);
    return color_quad(c[x], c[y], c[z], c[w]);
  }

  friend color_quad operator+(const color_quad& lhs, const color_quad& rhs) {
    color_quad result(lhs);
    result += rhs;
    return result;
  }

  friend color_quad operator-(const color_quad& lhs, const color_quad& rhs) {
    color_quad result(lhs);
    result -= rhs;
    return result;
  }

  friend color_quad operator*(const color_quad& lhs, parameter_type v) {
    color_quad result(lhs);
    result *= v;
    return result;
  }

  friend color_quad operator/(const color_quad& lhs, parameter_type v) {
    color_quad result(lhs);
    result /= v;
    return result;
  }

  friend color_quad operator*(parameter_type v, const color_quad& rhs) {
    color_quad result(rhs);
    result *= v;
    return result;
  }

  inline bool is_grayscale() const {
    return (c[0] == c[1]) && (c[1] == c[2]);
  }

  uint get_min_component_index(bool alpha = true) const {
    uint index = 0;
    uint limit = alpha ? cNumComps : (cNumComps - 1);
    for (uint i = 1; i < limit; i++)
      if (c[i] < c[index])
        index = i;
    return index;
  }

  uint get_max_component_index(bool alpha = true) const {
    uint index = 0;
    uint limit = alpha ? cNumComps : (cNumComps - 1);
    for (uint i = 1; i < limit; i++)
      if (c[i] > c[index])
        index = i;
    return index;
  }

  operator size_t() const {
    return (size_t)fast_hash(this, sizeof(*this));
  }

  void get_float4(float* pDst) {
    for (uint i = 0; i < 4; i++)
      pDst[i] = ((*this)[i] - component_traits::cMin) / float(component_traits::cMax - component_traits::cMin);
  }

  void get_float3(float* pDst) {
    for (uint i = 0; i < 3; i++)
      pDst[i] = ((*this)[i] - component_traits::cMin) / float(component_traits::cMax - component_traits::cMin);
  }

  static color_quad component_min(const color_quad& a, const color_quad& b) {
    color_quad result;
    for (uint i = 0; i < 4; i++)
      result[i] = static_cast<component_type>(math::minimum(a[i], b[i]));
    return result;
  }

  static color_quad component_max(const color_quad& a, const color_quad& b) {
    color_quad result;
    for (uint i = 0; i < 4; i++)
      result[i] = static_cast<component_type>(math::maximum(a[i], b[i]));
    return result;
  }

  static color_quad make_black() {
    return color_quad(0, 0, 0, component_traits::cMax);
  }

  static color_quad make_white() {
    return color_quad(component_traits::cMax, component_traits::cMax, component_traits::cMax, component_traits::cMax);
  }
};  // class color_quad

template <typename c, typename q>
struct scalar_type<color_quad<c, q> > {
  enum { cFlag = true };
  static inline void construct(color_quad<c, q>* p) {}
  static inline void construct(color_quad<c, q>* p, const color_quad<c, q>& init) { memcpy(p, &init, sizeof(color_quad<c, q>)); }
  static inline void construct_array(color_quad<c, q>* p, uint n) { p, n; }
  static inline void destruct(color_quad<c, q>* p) { p; }
  static inline void destruct_array(color_quad<c, q>* p, uint n) { p, n; }
};

typedef color_quad<uint8, int> color_quad_u8;
typedef color_quad<int8, int> color_quad_i8;
typedef color_quad<int16, int> color_quad_i16;
typedef color_quad<uint16, int> color_quad_u16;
typedef color_quad<int32, int> color_quad_i32;
typedef color_quad<uint32, uint> color_quad_u32;
typedef color_quad<float, float> color_quad_f;
typedef color_quad<double, double> color_quad_d;

namespace color {
inline uint elucidian_distance(uint r0, uint g0, uint b0, uint r1, uint g1, uint b1) {
  int dr = (int)r0 - (int)r1;
  int dg = (int)g0 - (int)g1;
  int db = (int)b0 - (int)b1;

  return static_cast<uint>(dr * dr + dg * dg + db * db);
}

inline uint elucidian_distance(uint r0, uint g0, uint b0, uint a0, uint r1, uint g1, uint b1, uint a1) {
  int dr = (int)r0 - (int)r1;
  int dg = (int)g0 - (int)g1;
  int db = (int)b0 - (int)b1;
  int da = (int)a0 - (int)a1;

  return static_cast<uint>(dr * dr + dg * dg + db * db + da * da);
}

inline uint elucidian_distance(const color_quad_u8& c0, const color_quad_u8& c1, bool alpha) {
  if (alpha)
    return elucidian_distance(c0.r, c0.g, c0.b, c0.a, c1.r, c1.g, c1.b, c1.a);
  else
    return elucidian_distance(c0.r, c0.g, c0.b, c1.r, c1.g, c1.b);
}

inline uint weighted_elucidian_distance(uint r0, uint g0, uint b0, uint r1, uint g1, uint b1, uint wr, uint wg, uint wb) {
  int dr = (int)r0 - (int)r1;
  int dg = (int)g0 - (int)g1;
  int db = (int)b0 - (int)b1;

  return static_cast<uint>((wr * dr * dr) + (wg * dg * dg) + (wb * db * db));
}

inline uint weighted_elucidian_distance(
    uint r0, uint g0, uint b0, uint a0,
    uint r1, uint g1, uint b1, uint a1,
    uint wr, uint wg, uint wb, uint wa) {
  int dr = (int)r0 - (int)r1;
  int dg = (int)g0 - (int)g1;
  int db = (int)b0 - (int)b1;
  int da = (int)a0 - (int)a1;

  return static_cast<uint>((wr * dr * dr) + (wg * dg * dg) + (wb * db * db) + (wa * da * da));
}

inline uint weighted_elucidian_distance(const color_quad_u8& c0, const color_quad_u8& c1, uint wr, uint wg, uint wb, uint wa) {
  return weighted_elucidian_distance(c0.r, c0.g, c0.b, c0.a, c1.r, c1.g, c1.b, c1.a, wr, wg, wb, wa);
}

//const uint cRWeight = 8;//24;
//const uint cGWeight = 24;//73;
//const uint cBWeight = 1;//3;

const uint cRWeight = 8;   //24;
const uint cGWeight = 25;  //73;
const uint cBWeight = 1;   //3;

inline uint color_distance(bool perceptual, const color_quad_u8& e1, const color_quad_u8& e2, bool alpha) {
  if (perceptual) {
    if (alpha)
      return weighted_elucidian_distance(e1, e2, cRWeight, cGWeight, cBWeight, cRWeight + cGWeight + cBWeight);
    else
      return weighted_elucidian_distance(e1, e2, cRWeight, cGWeight, cBWeight, 0);
  } else
    return elucidian_distance(e1, e2, alpha);
}

inline uint peak_color_error(const color_quad_u8& e1, const color_quad_u8& e2) {
  return math::maximum<uint>(labs(e1[0] - e2[0]), labs(e1[1] - e2[1]), labs(e1[2] - e2[2]));
  //return math::square<int>(e1[0] - e2[0]) + math::square<int>(e1[1] - e2[1]) + math::square<int>(e1[2] - e2[2]);
}

// y - [0,255]
// co - [-127,127]
// cg - [-126,127]
inline void RGB_to_YCoCg(int r, int g, int b, int& y, int& co, int& cg) {
  y = (r >> 2) + (g >> 1) + (b >> 2);
  co = (r >> 1) - (b >> 1);
  cg = -(r >> 2) + (g >> 1) - (b >> 2);
}

inline void YCoCg_to_RGB(int y, int co, int cg, int& r, int& g, int& b) {
  int tmp = y - cg;
  g = y + cg;
  r = tmp + co;
  b = tmp - co;
}

static inline uint8 clamp_component(int i) {
  if (static_cast<uint>(i) > 255U) {
    if (i < 0)
      i = 0;
    else if (i > 255)
      i = 255;
  }
  return static_cast<uint8>(i);
}

// RGB->YCbCr constants, scaled by 2^16
const int YR = 19595, YG = 38470, YB = 7471, CB_R = -11059, CB_G = -21709, CB_B = 32768, CR_R = 32768, CR_G = -27439, CR_B = -5329;
// YCbCr->RGB constants, scaled by 2^16
const int R_CR = 91881, B_CB = 116130, G_CR = -46802, G_CB = -22554;

inline int RGB_to_Y(const color_quad_u8& rgb) {
  const int r = rgb[0], g = rgb[1], b = rgb[2];
  return (r * YR + g * YG + b * YB + 32768) >> 16;
}

// RGB to YCbCr (same as JFIF JPEG).
// Odd default biases account for 565 endpoint packing.
inline void RGB_to_YCC(color_quad_u8& ycc, const color_quad_u8& rgb, int cb_bias = 123, int cr_bias = 125) {
  const int r = rgb[0], g = rgb[1], b = rgb[2];
  ycc.a = static_cast<uint8>((r * YR + g * YG + b * YB + 32768) >> 16);
  ycc.r = clamp_component(cb_bias + ((r * CB_R + g * CB_G + b * CB_B + 32768) >> 16));
  ycc.g = clamp_component(cr_bias + ((r * CR_R + g * CR_G + b * CR_B + 32768) >> 16));
  ycc.b = 0;
}

// YCbCr to RGB.
// Odd biases account for 565 endpoint packing.
inline void YCC_to_RGB(color_quad_u8& rgb, const color_quad_u8& ycc, int cb_bias = 123, int cr_bias = 125) {
  const int y = ycc.a;
  const int cb = ycc.r - cb_bias;
  const int cr = ycc.g - cr_bias;
  rgb.r = clamp_component(y + ((R_CR * cr + 32768) >> 16));
  rgb.g = clamp_component(y + ((G_CR * cr + G_CB * cb + 32768) >> 16));
  rgb.b = clamp_component(y + ((B_CB * cb + 32768) >> 16));
  rgb.a = 255;
}

// Float RGB->YCbCr constants
const float S = 1.0f / 65536.0f;
const float F_YR = S * YR, F_YG = S * YG, F_YB = S * YB, F_CB_R = S * CB_R, F_CB_G = S * CB_G, F_CB_B = S * CB_B, F_CR_R = S * CR_R, F_CR_G = S * CR_G, F_CR_B = S * CR_B;
// Float YCbCr->RGB constants
const float F_R_CR = S * R_CR, F_B_CB = S * B_CB, F_G_CR = S * G_CR, F_G_CB = S * G_CB;

inline void RGB_to_YCC_float(color_quad_f& ycc, const color_quad_u8& rgb) {
  const int r = rgb[0], g = rgb[1], b = rgb[2];
  ycc.a = r * F_YR + g * F_YG + b * F_YB;
  ycc.r = r * F_CB_R + g * F_CB_G + b * F_CB_B;
  ycc.g = r * F_CR_R + g * F_CR_G + b * F_CR_B;
  ycc.b = 0;
}

inline void YCC_float_to_RGB(color_quad_u8& rgb, const color_quad_f& ycc) {
  float y = ycc.a, cb = ycc.r, cr = ycc.g;
  rgb.r = color::clamp_component(static_cast<int>(.5f + y + F_R_CR * cr));
  rgb.g = color::clamp_component(static_cast<int>(.5f + y + F_G_CR * cr + F_G_CB * cb));
  rgb.b = color::clamp_component(static_cast<int>(.5f + y + F_B_CB * cb));
  rgb.a = 255;
}

}  // namespace color

// This class purposely trades off speed for extremely flexibility. It can handle any component swizzle, any pixel type from 1-4 components and 1-32 bits/component,
// any pixel size between 1-16 bytes/pixel, any pixel stride, any color_quad data type (signed/unsigned/float 8/16/32 bits/component), and scaled/non-scaled components.
// On the downside, it's freaking slow.
class pixel_packer {
 public:
  pixel_packer() {
    clear();
  }

  pixel_packer(uint num_comps, uint bits_per_comp, int pixel_stride = -1, bool reversed = false) {
    init(num_comps, bits_per_comp, pixel_stride, reversed);
  }

  pixel_packer(const char* pComp_map, int pixel_stride = -1, int force_comp_size = -1) {
    init(pComp_map, pixel_stride, force_comp_size);
  }

  void clear() {
    utils::zero_this(this);
  }

  inline bool is_valid() const { return m_pixel_stride > 0; }

  inline uint get_pixel_stride() const { return m_pixel_stride; }
  void set_pixel_stride(uint n) { m_pixel_stride = n; }

  uint get_num_comps() const { return m_num_comps; }
  uint get_comp_size(uint index) const {
    CRNLIB_ASSERT(index < 4);
    return m_comp_size[index];
  }
  uint get_comp_ofs(uint index) const {
    CRNLIB_ASSERT(index < 4);
    return m_comp_ofs[index];
  }
  uint get_comp_max(uint index) const {
    CRNLIB_ASSERT(index < 4);
    return m_comp_max[index];
  }
  bool get_rgb_is_luma() const { return m_rgb_is_luma; }

  template <typename color_quad_type>
  const void* unpack(const void* p, color_quad_type& color, bool rescale = true) const {
    const uint8* pSrc = static_cast<const uint8*>(p);

    for (uint i = 0; i < 4; i++) {
      const uint comp_size = m_comp_size[i];
      if (!comp_size) {
        if (color_quad_type::component_traits::cFloat)
          color[i] = static_cast<typename color_quad_type::parameter_t>((i == 3) ? 1 : 0);
        else
          color[i] = static_cast<typename color_quad_type::parameter_t>((i == 3) ? color_quad_type::component_traits::cMax : 0);
        continue;
      }

      uint n = 0, dst_bit_ofs = 0;
      uint src_bit_ofs = m_comp_ofs[i];
      while (dst_bit_ofs < comp_size) {
        const uint byte_bit_ofs = src_bit_ofs & 7;
        n |= ((pSrc[src_bit_ofs >> 3] >> byte_bit_ofs) << dst_bit_ofs);

        const uint bits_read = 8 - byte_bit_ofs;
        src_bit_ofs += bits_read;
        dst_bit_ofs += bits_read;
      }

      const uint32 mx = m_comp_max[i];
      n &= mx;

      const uint32 h = static_cast<uint32>(color_quad_type::component_traits::cMax);

      if (color_quad_type::component_traits::cFloat)
        color.set_component(i, static_cast<typename color_quad_type::parameter_t>(n));
      else if (rescale)
        color.set_component(i, static_cast<typename color_quad_type::parameter_t>((static_cast<uint64>(n) * h + (mx >> 1U)) / mx));
      else if (color_quad_type::component_traits::cSigned)
        color.set_component(i, static_cast<typename color_quad_type::parameter_t>(math::minimum<uint32>(n, h)));
      else
        color.set_component(i, static_cast<typename color_quad_type::parameter_t>(n));
    }

    if (m_rgb_is_luma) {
      color[0] = color[1];
      color[2] = color[1];
    }

    return pSrc + m_pixel_stride;
  }

  template <typename color_quad_type>
  void* pack(const color_quad_type& color, void* p, bool rescale = true) const {
    uint8* pDst = static_cast<uint8*>(p);

    for (uint i = 0; i < 4; i++) {
      const uint comp_size = m_comp_size[i];
      if (!comp_size)
        continue;

      uint32 mx = m_comp_max[i];

      uint32 n;
      if (color_quad_type::component_traits::cFloat) {
        typename color_quad_type::parameter_t t = color[i];
        if (t < 0.0f)
          n = 0;
        else if (t > static_cast<typename color_quad_type::parameter_t>(mx))
          n = mx;
        else
          n = math::minimum<uint32>(static_cast<uint32>(floor(t + .5f)), mx);
      } else if (rescale) {
        if (color_quad_type::component_traits::cSigned)
          n = math::maximum<int>(static_cast<int>(color[i]), 0);
        else
          n = static_cast<uint32>(color[i]);

        const uint32 h = static_cast<uint32>(color_quad_type::component_traits::cMax);
        n = static_cast<uint32>((static_cast<uint64>(n) * mx + (h >> 1)) / h);
      } else {
        if (color_quad_type::component_traits::cSigned)
          n = math::minimum<uint32>(static_cast<uint32>(math::maximum<int>(static_cast<int>(color[i]), 0)), mx);
        else
          n = math::minimum<uint32>(static_cast<uint32>(color[i]), mx);
      }

      uint src_bit_ofs = 0;
      uint dst_bit_ofs = m_comp_ofs[i];
      while (src_bit_ofs < comp_size) {
        const uint cur_byte_bit_ofs = (dst_bit_ofs & 7);
        const uint cur_byte_bits = 8 - cur_byte_bit_ofs;

        uint byte_val = pDst[dst_bit_ofs >> 3];
        uint bit_mask = (mx << cur_byte_bit_ofs) & 0xFF;
        byte_val &= ~bit_mask;
        byte_val |= (n << cur_byte_bit_ofs);
        pDst[dst_bit_ofs >> 3] = static_cast<uint8>(byte_val);

        mx >>= cur_byte_bits;
        n >>= cur_byte_bits;

        dst_bit_ofs += cur_byte_bits;
        src_bit_ofs += cur_byte_bits;
      }
    }

    return pDst + m_pixel_stride;
  }

  bool init(uint num_comps, uint bits_per_comp, int pixel_stride = -1, bool reversed = false) {
    clear();

    if ((num_comps < 1) || (num_comps > 4) || (bits_per_comp < 1) || (bits_per_comp > 32)) {
      CRNLIB_ASSERT(0);
      return false;
    }

    for (uint i = 0; i < num_comps; i++) {
      m_comp_size[i] = bits_per_comp;
      m_comp_ofs[i] = i * bits_per_comp;
      if (reversed)
        m_comp_ofs[i] = ((num_comps - 1) * bits_per_comp) - m_comp_ofs[i];
    }

    for (uint i = 0; i < 4; i++)
      m_comp_max[i] = static_cast<uint32>((1ULL << m_comp_size[i]) - 1ULL);

    m_pixel_stride = (pixel_stride >= 0) ? pixel_stride : (num_comps * bits_per_comp + 7) / 8;

    return true;
  }

  // Format examples:
  // R16G16B16
  // B5G6R5
  // B5G5R5x1
  // Y8A8
  // A8R8G8B8
  // First component is at LSB in memory. Assumes unsigned integer components, 1-32bits each.
  bool init(const char* pComp_map, int pixel_stride = -1, int force_comp_size = -1) {
    clear();

    uint cur_bit_ofs = 0;

    while (*pComp_map) {
      char c = *pComp_map++;

      int comp_index = -1;
      if (c == 'R')
        comp_index = 0;
      else if (c == 'G')
        comp_index = 1;
      else if (c == 'B')
        comp_index = 2;
      else if (c == 'A')
        comp_index = 3;
      else if (c == 'Y')
        comp_index = 4;
      else if (c != 'x')
        return false;

      uint comp_size = 0;

      uint n = *pComp_map;
      if ((n >= '0') && (n <= '9')) {
        comp_size = n - '0';
        pComp_map++;

        n = *pComp_map;
        if ((n >= '0') && (n <= '9')) {
          comp_size = (comp_size * 10) + (n - '0');
          pComp_map++;
        }
      }

      if (force_comp_size != -1)
        comp_size = force_comp_size;

      if ((!comp_size) || (comp_size > 32))
        return false;

      if (comp_index == 4) {
        if (m_comp_size[0] || m_comp_size[1] || m_comp_size[2])
          return false;

        //m_comp_ofs[0] = m_comp_ofs[1] = m_comp_ofs[2] = cur_bit_ofs;
        //m_comp_size[0] = m_comp_size[1] = m_comp_size[2] = comp_size;
        m_comp_ofs[1] = cur_bit_ofs;
        m_comp_size[1] = comp_size;
        m_rgb_is_luma = true;
        m_num_comps++;
      } else if (comp_index >= 0) {
        if (m_comp_size[comp_index])
          return false;

        m_comp_ofs[comp_index] = cur_bit_ofs;
        m_comp_size[comp_index] = comp_size;
        m_num_comps++;
      }

      cur_bit_ofs += comp_size;
    }

    for (uint i = 0; i < 4; i++)
      m_comp_max[i] = static_cast<uint32>((1ULL << m_comp_size[i]) - 1ULL);

    if (pixel_stride >= 0)
      m_pixel_stride = pixel_stride;
    else
      m_pixel_stride = (cur_bit_ofs + 7) / 8;
    return true;
  }

 private:
  uint m_pixel_stride;
  uint m_num_comps;
  uint m_comp_size[4];
  uint m_comp_ofs[4];
  uint m_comp_max[4];
  bool m_rgb_is_luma;
};

}  // namespace crnlib
