#include "igs_color_rgb_hls.h"

namespace {

void pixel_rgba_(const double red_in, const double gre_in, const double blu_in,

                 double &red_out, double &gre_out, double &blu_out,

                 const double hue_pivot  // 0.0  ...0...360...

                 ,

                 const double hue_scale  // 1.0  ...1...

                 ,

                 const double hue_shift  // 0.0  ...0...360...

                 ,

                 const double lig_pivot  // 0.0  ...0...1...

                 ,

                 const double lig_scale  // 1.0  ...1...

                 ,

                 const double lig_shift  // 0.0  ...0...1...

                 ,

                 const double sat_pivot  // 0.0  ...0...1...

                 ,

                 const double sat_scale  // 1.0  ...1...

                 ,

                 const double sat_shift  // 0.0  ...0...1...

                 ) {

  double hue, lig, sat;

  igs::color::rgb_to_hls(red_in, gre_in, blu_in, hue, lig, sat);

  if ((1.0 != hue_scale) || (0.0 != hue_shift)) {

    hue -= hue_pivot;

    while (hue < -180.0) {

      hue += 360.0;

    }

    while (180.0 <= hue) {

      hue -= 360.0;

    }

    hue *= hue_scale;

    hue += hue_pivot;

    hue += hue_shift;

    while (hue < 0.0) {

      hue += 360.0;

    }

    while (360.0 <= hue) {

      hue -= 360.0;

    }

  }

  if ((1.0 != lig_scale) || (0.0 != lig_shift)) {

    lig -= lig_pivot;

    lig *= lig_scale;

    lig += lig_pivot;

    lig += lig_shift;

    lig = (lig < 0.0) ? 0.0 : ((1.0 < lig) ? 1.0 : lig);

  }

  if ((1.0 != sat_scale) || (0.0 != sat_shift)) {

    sat -= sat_pivot;

    sat *= sat_scale;

    sat += sat_pivot;

    sat += sat_shift;

    sat = (sat < 0.0) ? 0.0 : ((1.0 < sat) ? 1.0 : sat);

  }

  igs::color::hls_to_rgb(hue, lig, sat, red_out, gre_out, blu_out);

}

}

//------------------------------------------------------------

#include <limits>           /* std::numeric_limits */</limits>

#include "igs_ifx_common.h" /* igs::image::rgba */

#include "igs_hls_adjust.h"

namespace {

/* raster画像にノイズをのせるtemplate */

template <class class="" it,="" rt=""></class>

void change_template_(

    IT *image_array, const int height, const int width, const int channels

    ,

    const RT *ref /* 求める画像(out)と同じ高さ、幅、チャンネル数 */

    ,

    const int ref_mode /* 0=R,1=G,2=B,3=A,4=Luminance,5=Nothing */

    ,

    const double hue_pivot, const double hue_scale, const double hue_shift,

    const double lig_pivot, const double lig_scale, const double lig_shift,

    const double sat_pivot, const double sat_scale, const double sat_shift

    ,

    const bool add_blend_sw) {

  const int t_max      = std::numeric_limits<it>::max();</it>

  const double div_val = static_cast<double>(t_max);</double>

  const double mul_val = static_cast<double>(t_max) + 0.999999;</double>

  const int pixsize    = height * width;

  const int r_max      = std::numeric_limits<rt>::max();</rt>

  if (igs::image::rgba::siz == channels) {

    using namespace igs::image::rgba;

    for (int ii = 0; ii < pixsize; ++ii, image_array += channels) {

      /* 変化量初期値 */

      double refv = 1.0;

      /* 参照画像あればピクセル単位の画像変化量を得る */

      if (ref != 0) {

        refv *= igs::color::ref_value(ref, channels, r_max, ref_mode);

        ref += channels; /* continue;の前に行うこと */

      }

      /* 加算合成で、Alpha値ゼロならRGB値を計算する必要はない */

      if (add_blend_sw && (0 == image_array[alp])) {

        continue;

      }

      /* 加算合成でなくAlpha合成の時は、

      Alpha値がゼロでもRGB値は存在する(してもよい) */

      /* RGB値を正規化 */

      const IT *const ia = image_array;

      const double rr1   = static_cast<double>(ia[red]) / div_val;</double>

      const double gg1   = static_cast<double>(ia[gre]) / div_val;</double>

      const double bb1   = static_cast<double>(ia[blu]) / div_val;</double>

      /* pivot,scale,shiftによってRGB値を変化する */

      double rr2, gg2, bb2;

      pixel_rgba_(rr1, gg1, bb1, rr2, gg2, bb2, hue_pivot, hue_scale, hue_shift,

                  lig_pivot, lig_scale, lig_shift, sat_pivot, sat_scale,

                  sat_shift);

      /* 加算合成で、その値がMaxでなければ変化量に乗算 */

      if (add_blend_sw && (ia[alp] < t_max)) {

        refv *= static_cast<double>(ia[alp]) / div_val;</double>

      }

      /* ピクセル単位の変化量があれば、RGBを調整する */

      if ((ref != 0) || (add_blend_sw && (ia[alp] < t_max))) {

        rr2 = rr1 + (rr2 - rr1) * refv;

        gg2 = gg1 + (gg2 - gg1) * refv;

        bb2 = bb1 + (bb2 - bb1) * refv;

      }

      /* 結果をRGBに戻す */

      image_array[red] = static_cast<it>(rr2 * mul_val);</it>

      image_array[gre] = static_cast<it>(gg2 * mul_val);</it>

      image_array[blu] = static_cast<it>(bb2 * mul_val);</it>

    }

  } else if (igs::image::rgb::siz == channels) {

    using namespace igs::image::rgb;

    for (int ii = 0; ii < pixsize; ++ii, image_array += channels) {

      double refv = 1.0;

      if (ref != 0) {

        refv *= igs::color::ref_value(ref, channels, r_max, ref_mode);

        ref += channels;

      }

      const IT *const ia = image_array;

      const double rr1   = static_cast<double>(ia[red]) / div_val;</double>

      const double gg1   = static_cast<double>(ia[gre]) / div_val;</double>

      const double bb1   = static_cast<double>(ia[blu]) / div_val;</double>

      double rr2, gg2, bb2;

      pixel_rgba_(rr1, gg1, bb1, rr2, gg2, bb2, hue_pivot, hue_scale, hue_shift,

                  lig_pivot, lig_scale, lig_shift, sat_pivot, sat_scale,

                  sat_shift);

      if (ref != 0) {

        rr2 = rr1 + (rr2 - rr1) * refv;

        gg2 = gg1 + (gg2 - gg1) * refv;

        bb2 = bb1 + (bb2 - bb1) * refv;

      }

      image_array[red] = static_cast<it>(rr2 * mul_val);</it>

      image_array[gre] = static_cast<it>(gg2 * mul_val);</it>

      image_array[blu] = static_cast<it>(bb2 * mul_val);</it>

    }

  } else if (1 == channels) { /* grayscale */

    for (int ii = 0; ii < pixsize; ++ii, image_array += channels) {

      double refv = 1.0;

      if (ref != 0) {

        refv *= igs::color::ref_value(ref, channels, r_max, ref_mode);

        ref += channels;

      }

      double li1 = static_cast<double>(image_array[0]) / div_val,</double>

             li2 = (li1 - lig_pivot) * lig_scale + lig_pivot + lig_shift;

      li2        = (li2 < 0.0) ? 0.0 : ((1.0 < li2) ? 1.0 : li2);

      if (ref != 0) {

        li2 = li1 + (li2 - li1) * refv;

      }

      image_array[0] = static_cast<it>(li2 * mul_val);</it>

    }

  }

}

}

#include <stdexcept> /* std::domain_error  */</stdexcept>

void igs::hls_adjust::change(

    unsigned char *image_array, const int height, const int width,

    const int channels, const int bits

    ,

    const unsigned char *ref /* 求める画像と同じ高、幅、channels数 */

    ,

    const int ref_bits /* refがゼロのときはここもゼロ */

    ,

    const int ref_mode /* 0=R,1=G,2=B,3=A,4=Luminance,5=Nothing */

    ,

    const double hue_pivot, const double hue_scale, const double hue_shift,

    const double lig_pivot, const double lig_scale, const double lig_shift,

    const double sat_pivot, const double sat_scale, const double sat_shift

    ,

    const bool add_blend_sw) {

  if ((1.0 == hue_scale) && (0.0 == hue_shift) && (1.0 == lig_scale) &&

      (0.0 == lig_shift) && (1.0 == sat_scale) && (0.0 == sat_shift)) {

    return;

  }

  if ((igs::image::rgba::siz != channels) &&

      (igs::image::rgb::siz != channels) && (1 != channels) /* grayscale */

      ) {

    throw std::domain_error("Bad channels,Not rgba/rgb/grayscale");

  }

  /* rgb(a)画像にhls(a)でドットノイズを加える */

  if ((std::numeric_limits<unsigned char="">::digits == bits) &&</unsigned>

      ((std::numeric_limits<unsigned char="">::digits == ref_bits) ||</unsigned>

       (0 == ref_bits))) {

    change_template_(image_array, height, width, channels, ref, ref_mode,

                     hue_pivot, hue_scale, hue_shift, lig_pivot, lig_scale,

                     lig_shift, sat_pivot, sat_scale, sat_shift, add_blend_sw);

  } else if ((std::numeric_limits<unsigned short="">::digits == bits) &&</unsigned>

             ((std::numeric_limits<unsigned char="">::digits == ref_bits) ||</unsigned>

              (0 == ref_bits))) {

    change_template_(reinterpret_cast<unsigned *="" short="">(image_array), height,</unsigned>

                     width, channels, ref, ref_mode, hue_pivot, hue_scale,

                     hue_shift, lig_pivot, lig_scale, lig_shift, sat_pivot,

                     sat_scale, sat_shift, add_blend_sw);

  } else if ((std::numeric_limits<unsigned short="">::digits == bits) &&</unsigned>

             (std::numeric_limits<unsigned short="">::digits == ref_bits)) {</unsigned>

    change_template_(reinterpret_cast<unsigned *="" short="">(image_array), height,</unsigned>

                     width, channels,

                     reinterpret_cast<const *="" short="" unsigned="">(ref), ref_mode,</const>

                     hue_pivot, hue_scale, hue_shift, lig_pivot, lig_scale,

                     lig_shift, sat_pivot, sat_scale, sat_shift, add_blend_sw);

  } else if ((std::numeric_limits<unsigned char="">::digits == bits) &&</unsigned>

             (std::numeric_limits<unsigned short="">::digits == ref_bits)) {</unsigned>

    change_template_(image_array, height, width, channels,

                     reinterpret_cast<const *="" short="" unsigned="">(ref), ref_mode,</const>

                     hue_pivot, hue_scale, hue_shift, lig_pivot, lig_scale,

                     lig_shift, sat_pivot, sat_scale, sat_shift, add_blend_sw);

  } else {

    throw std::domain_error("Bad bits,Not uchar/ushort");

  }

}