#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 */
#include "igs_ifx_common.h" /* igs::image::rgba */
#include "igs_hls_adjust.h"
namespace
{
/* raster画像にノイズをのせるtemplate */
template <class IT, class RT>
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();
	const double div_val = static_cast<double>(t_max);
	const double mul_val = static_cast<double>(t_max) + 0.999999;
	const int pixsize = height * width;
	const int r_max = std::numeric_limits<RT>::max();
	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;
			const double gg1 = static_cast<double>(ia[gre]) / div_val;
			const double bb1 = static_cast<double>(ia[blu]) / div_val;

			/* 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;
			}

			/* ピクセル単位の変化量があれば、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);
			image_array[gre] = static_cast<IT>(gg2 * mul_val);
			image_array[blu] = static_cast<IT>(bb2 * mul_val);
		}
	} 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;
			const double gg1 = static_cast<double>(ia[gre]) / div_val;
			const double bb1 = static_cast<double>(ia[blu]) / div_val;
			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);
			image_array[gre] = static_cast<IT>(gg2 * mul_val);
			image_array[blu] = static_cast<IT>(bb2 * mul_val);
		}
	} 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,
				   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);
		}
	}
}
}

#include <stdexcept> /* std::domain_error  */
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) && ((std::numeric_limits<unsigned char>::digits == ref_bits) || (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) && ((std::numeric_limits<unsigned char>::digits == ref_bits) || (0 == ref_bits))) {
		change_template_(
			reinterpret_cast<unsigned short *>(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) && (std::numeric_limits<unsigned short>::digits == ref_bits)) {
		change_template_(
			reinterpret_cast<unsigned short *>(image_array), height, width, channels, reinterpret_cast<const unsigned short *>(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 char>::digits == bits) && (std::numeric_limits<unsigned short>::digits == ref_bits)) {
		change_template_(
			image_array, height, width, channels, reinterpret_cast<const unsigned short *>(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 {
		throw std::domain_error("Bad bits,Not uchar/ushort");
	}
}