#include <cmath>	 // pow(-)</cmath>

#include <stdexcept> // std::domain_error(-)</stdexcept>

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

#include <vector></vector>

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

#include "igs_levels.h"

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

namespace

{

void levels_(

	double &val // 0...1

	,

	const double in_min // 0...1

	,

	const double in_max // 0...1

	,

	const double gamma // 0.1 ... 10.0

	,

	const double out_min // 0...1

	,

	const double out_max // 0...1

	,

	const bool clamp_sw)

{

	/* 1 入力範囲を正規化 */

	val = (in_max == in_min) ? in_max : (val - in_min) / (in_max - in_min);

	/* 2 (出力範囲に)clamp */

	if (clamp_sw) {

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

	}

	/* 3 正規化の範囲でgamma変換 */

	if ((1.0 != gamma) && (0.0 != gamma)) {

		if ((0.0 < val) && (val < 1.0)) {

			val = pow(val, 1.0 / gamma);

		}

	}

	/* 4 正規化範囲を出力範囲に */

	val = out_min + val * (out_max - out_min);

	/* 5 clamp */

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

}

double refchk_(const int src, const int tgt, const double refv)

{

	return (src < tgt) ? (tgt - src + 0.999999) * refv + src : (src - tgt + 0.999999) * (1.0 - refv) + tgt;

}

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

/*

	std::vector< std::vector<t> > &tables</t>

 とするとvc2005mdで他プログラムとのリンク時、

	allocatorの２重定義でエラーとなる。

	std::vector< std::vector<int> >&tables</int>

 ならOK

 2009-01-27

 */

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

void change_(

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

	,

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

	,

	const int ref_mode // R,G,B,A,luminance

	,

	const double r_in_min, const double r_in_max // 0...1

	,

	const double g_in_min, const double g_in_max // 0...1

	,

	const double b_in_min, const double b_in_max // 0...1

	,

	const double a_in_min, const double a_in_max // 0...1

	,

	const double r_gamma // 0.1 ... 10.0

	,

	const double g_gamma // 0.1 ... 10.0

	,

	const double b_gamma // 0.1 ... 10.0

	,

	const double a_gamma // 0.1 ... 10.0

	,

	const double r_out_min, const double r_out_max // 0...1

	,

	const double g_out_min, const double g_out_max // 0...1

	,

	const double b_out_min, const double b_out_max // 0...1

	,

	const double a_out_min, const double a_out_max // 0...1

	,

	const bool clamp_sw, const bool alpha_sw, const bool add_blend_sw)

{

	/* 1 最大値、最小値から変換テーブルを求める */

	std::vector<std::vector<int>> tables;</std::vector<int>

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

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

	const double mul_val = div_val + 0.999999;

	{

		using namespace igs::image::rgba;

		tables.resize(siz);

		tables[red].resize(val_max + 1);

		tables[gre].resize(val_max + 1);

		tables[blu].resize(val_max + 1);

		tables[alp].resize(val_max + 1);

		for (unsigned int yy = 0; yy <= val_max; ++yy) {

			double rr, gg, bb, aa;

			rr = gg = bb = aa = yy / div_val;

			levels_(rr, r_in_min, r_in_max, r_gamma, r_out_min, r_out_max, clamp_sw);

			levels_(gg, g_in_min, g_in_max, g_gamma, g_out_min, g_out_max, clamp_sw);

			levels_(bb, b_in_min, b_in_max, b_gamma, b_out_min, b_out_max, clamp_sw);

			levels_(aa, a_in_min, a_in_max, a_gamma, a_out_min, a_out_max, clamp_sw);

			/* 0〜1.0 --> 0〜mul_maxスケール変換し、整数値化 */

			tables[red][yy] = static_cast<int>(rr * mul_val);</int>

			tables[gre][yy] = static_cast<int>(gg * mul_val);</int>

			tables[blu][yy] = static_cast<int>(bb * mul_val);</int>

			tables[alp][yy] = static_cast<int>(aa * mul_val);</int>

		}

	}

	/* 2 変換テーブルを使ってlevel変換する */

	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のLevel処理した値をrefvの変化量比で加える */

			if (alpha_sw) {

				image_array[alp] = static_cast<it>(refchk_(</it>

					static_cast<int>(image_array[alp]), tables[alp][image_array[alp]], refv));</int>

			}

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

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

				continue;

			}

			/* 加算合成でなくAlpha合成の時は、Alpha値がゼロでも

			RGB値は存在する(してもよい)の計算する */

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

			if (add_blend_sw && (image_array[alp] < val_max)) {

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

			}

			/* RGBのLevel処理した値をrefvの変化量比で加える */

			image_array[red] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[red]), tables[red][image_array[red]], refv));</int>

			image_array[gre] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[gre]), tables[gre][image_array[gre]], refv));</int>

			image_array[blu] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[blu]), tables[blu][image_array[blu]], refv));</int>

		}

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

			}

			/* RGBのLevel処理した値をrefvの変化量比で加える */

			image_array[red] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[red]), tables[red][image_array[red]], refv));</int>

			image_array[gre] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[gre]), tables[gre][image_array[gre]], refv));</int>

			image_array[blu] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[blu]), tables[blu][image_array[blu]], refv));</int>

		}

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

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

			double refv = 1.0;

			if (ref != 0) {

				refv *= igs::color::ref_value(

					ref, channels, r_max, ref_mode);

				ref += channels;

			}

			image_array[0] = static_cast<it>(refchk_(</it>

				static_cast<int>(image_array[0]), tables[0][image_array[0]], refv));</int>

		}

	}

	/* 1 変換テーブルのメモリ解放 */

	tables.clear();

}

}

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

void igs::levels::change(

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

	,

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

	,

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

	,

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

	,

	const double r_in_min, const double r_in_max // 0...1

	,

	const double g_in_min, const double g_in_max // 0...1

	,

	const double b_in_min, const double b_in_max // 0...1

	,

	const double a_in_min, const double a_in_max // 0...1

	,

	const double r_gamma // 0.1 ... 10.0

	,

	const double g_gamma // 0.1 ... 10.0

	,

	const double b_gamma // 0.1 ... 10.0

	,

	const double a_gamma // 0.1 ... 10.0

	,

	const double r_out_min, const double r_out_max // 0...1

	,

	const double g_out_min, const double g_out_max // 0...1

	,

	const double b_out_min, const double b_out_max // 0...1

	,

	const double a_out_min, const double a_out_max // 0...1

	,

	const bool clamp_sw, const bool alpha_sw, const bool add_blend_sw)

{

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

		(igs::image::rgb::siz != channels) &&

		(1 != channels) /* grayscale */

		) {

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

	}

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

		change_(

			image_array, height, width, channels, ref, ref_mode, r_in_min, r_in_max, g_in_min, g_in_max, b_in_min, b_in_max, a_in_min, a_in_max, r_gamma, g_gamma, b_gamma, a_gamma, r_out_min, r_out_max, g_out_min, g_out_max, b_out_min, b_out_max, a_out_min, a_out_max, clamp_sw, alpha_sw, add_blend_sw);

	} else if ((std::numeric_limits<unsigned short="">::digits == bits) && ((std::numeric_limits<unsigned char="">::digits == ref_bits) || (0 == ref_bits))) {</unsigned></unsigned>

		change_(

			reinterpret_cast<unsigned *="" short="">(image_array), height, width, channels, ref, ref_mode, r_in_min, r_in_max, g_in_min, g_in_max, b_in_min, b_in_max, a_in_min, a_in_max, r_gamma, g_gamma, b_gamma, a_gamma, r_out_min, r_out_max, g_out_min, g_out_max, b_out_min, b_out_max, a_out_min, a_out_max, clamp_sw, alpha_sw, add_blend_sw);</unsigned>

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

		change_(

			reinterpret_cast<unsigned *="" short="">(image_array), height, width, channels, reinterpret_cast<const *="" short="" unsigned="">(ref), ref_mode, r_in_min, r_in_max, g_in_min, g_in_max, b_in_min, b_in_max, a_in_min, a_in_max, r_gamma, g_gamma, b_gamma, a_gamma, r_out_min, r_out_max, g_out_min, g_out_max, b_out_min, b_out_max, a_out_min, a_out_max, clamp_sw, alpha_sw, add_blend_sw);</const></unsigned>

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

		change_(

			image_array, height, width, channels, reinterpret_cast<const *="" short="" unsigned="">(ref), ref_mode, r_in_min, r_in_max, g_in_min, g_in_max, b_in_min, b_in_max, a_in_min, a_in_max, r_gamma, g_gamma, b_gamma, a_gamma, r_out_min, r_out_max, g_out_min, g_out_max, b_out_min, b_out_max, a_out_min, a_out_max, clamp_sw, alpha_sw, add_blend_sw);</const>

	} else {

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

	}

}