#pragma once

#ifndef TCG_PIXEL_H
#define TCG_PIXEL_H

// tcg includes
#include "pixel_ops.h"

// STD includes
#include <limits>

namespace tcg {

//***********************************************************
//    Tcg Pixel Types  definition
//***********************************************************

template <typename Chan, typename pixel_category>
struct Pixel;

template <typename Chan>
struct Pixel<Chan, grayscale_pixel_tag> {
  typedef grayscale_pixel_tag pixel_category;
  typedef Chan channel_type;

public:
  channel_type l;

public:
  Pixel() : l() {}
  Pixel(channel_type _l) : l(_l) {}

  bool operator==(const Pixel &other) const { return l == other.l; }
  bool operator!=(const Pixel &other) const { return l != other.l; }
};

template <typename Chan>
struct Pixel<Chan, rgb_pixel_tag> {
  typedef rgb_pixel_tag pixel_category;
  typedef Chan channel_type;

public:
  channel_type r, g, b;

public:
  Pixel() : r(), g(), b() {}
  Pixel(channel_type _r, channel_type _g, channel_type _b)
      : r(_r), g(_g), b(_b) {}

  bool operator==(const Pixel &other) const {
    return (r == other.r) && (g == other.g) && (g == other.g);
  }
  bool operator!=(const Pixel &other) const {
    return (r != other.r) || (g != other.g) || (g != other.g);
  }
};

template <typename Chan>
struct Pixel<Chan, rgbm_pixel_tag> {
  typedef rgbm_pixel_tag pixel_category;
  typedef Chan channel_type;

public:
  channel_type r, g, b, m;

public:
  Pixel() : r(), g(), b(), m() {}
  Pixel(channel_type _r, channel_type _g, channel_type _b, channel_type _m)
      : r(_r), g(_g), b(_b), m(_m) {}

  bool operator==(const Pixel &other) const {
    return (r == other.r) && (g == other.g) && (g == other.g) && (m == other.m);
  }
  bool operator!=(const Pixel &other) const {
    return (r != other.r) || (g != other.g) || (g != other.g) || (m != other.m);
  }
};

//***********************************************************
//    Tcg Pixel  traits
//***********************************************************

namespace pixel_ops {

template <typename Chan>
struct pixel_traits<Pixel<Chan, grayscale_pixel_tag>, grayscale_pixel_tag> {
  typedef Pixel<Chan, grayscale_pixel_tag> pixel_type;
  typedef grayscale_pixel_tag pixel_category;
  typedef Chan channel_type;

  enum { channels_count = 1 };

  static channel_type max_channel_value() {
    return (std::numeric_limits<channel_type>::max)();
  }

  static channel_type l(const pixel_type &pix) { return pix.l; }
  static channel_type &l(pixel_type &pix) { return pix.l; }
};

template <typename Chan>
struct pixel_traits<Pixel<Chan, rgb_pixel_tag>, rgb_pixel_tag> {
  typedef Pixel<Chan, rgb_pixel_tag> pixel_type;
  typedef rgb_pixel_tag pixel_category;
  typedef Chan channel_type;

  enum { channels_count = 3 };

  static channel_type max_channel_value() {
    return (std::numeric_limits<channel_type>::max)();
  }

  static channel_type r(const pixel_type &pix) { return pix.r; }
  static channel_type &r(pixel_type &pix) { return pix.r; }

  static channel_type g(const pixel_type &pix) { return pix.g; }
  static channel_type &g(pixel_type &pix) { return pix.g; }

  static channel_type b(const pixel_type &pix) { return pix.b; }
  static channel_type &b(pixel_type &pix) { return pix.b; }
};

template <typename Chan>
struct pixel_traits<Pixel<Chan, rgbm_pixel_tag>, rgbm_pixel_tag>
    : public pixel_traits<Pixel<Chan, rgb_pixel_tag>, rgb_pixel_tag> {
  typedef Pixel<Chan, rgbm_pixel_tag> pixel_type;
  typedef rgbm_pixel_tag pixel_category;
  typedef Chan channel_type;

  enum { channels_count = 4 };

  static channel_type m(const pixel_type &pix) { return pix.m; }
  static channel_type &m(pixel_type &pix) { return pix.m; }
};
}
}  // namespace tcg::pixel_ops

#endif  // TCG_PIXEL_H