#include <iostream></iostream>

#include <vector></vector>

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

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

#include "igs_color_rgb_hsv.h"

#include "igs_math_random.h"

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

#include "igs_hsv_noise_in_camera.h"

namespace igs

{

namespace hsv_noise_in_camera

{

/* バッファをまとめて確保(return or throwで自動解放) */

class noise_reference

{

public:

	noise_reference(

		const int ww, const int hh, const double hue_range, const double sat_range, const double val_range, const double alp_range, const unsigned long random_seed, const double near_blur

		,

		const int camera_x, const int camera_y, const int camera_w, const int camera_h) : w_(ww), h_(hh), nblur_(near_blur)

	{

		if (0 == ww) {

			return;

		}

		if (0 == hh) {

			return;

		}

		/* 枚数ゼロ(rangeが全部ゼロ)なのでノイズかけない */

		if ((0.0 == hue_range) && (0.0 == sat_range) && (0.0 == val_range) && (0.0 == alp_range)) {

			return;

		}

		/* memory確保 */

		const int sz = ww * hh;

		if (0.0 != hue_range) {

			this->hue_array_.resize(sz);

		}

		if (0.0 != sat_range) {

			this->sat_array_.resize(sz);

		}

		if (0.0 != val_range) {

			this->val_array_.resize(sz);

		}

		if (0.0 != alp_range) {

			this->alp_array_.resize(sz);

		}

		/* タネを設定 */

		igs::math::random hue_rand, sat_rand, val_rand, alp_rand;

		unsigned long step = 0;

		if (0.0 != hue_range) {

			hue_rand.seed(random_seed + step++);

		}

		if (0.0 != sat_range) {

			sat_rand.seed(random_seed + step++);

		}

		if (0.0 != val_range) {

			val_rand.seed(random_seed + step++);

		}

		if (0.0 != alp_range) {

			alp_rand.seed(random_seed + step++);

		}

		igs::math::random hue_rand_ext, sat_rand_ext, val_rand_ext, alp_rand_ext;

		if (0.0 != hue_range) {

			hue_rand_ext.seed(random_seed + step++);

		}

		if (0.0 != sat_range) {

			sat_rand_ext.seed(random_seed + step++);

		}

		if (0.0 != val_range) {

			val_rand_ext.seed(random_seed + step++);

		}

		if (0.0 != alp_range) {

			alp_rand_ext.seed(random_seed + step++);

		}

		/* ノイズを書く */

		/* -???_range/2〜???_range/2 */

		int x1 = camera_x;

		int y1 = camera_y;

		int x2 = camera_x + camera_w - 1;

		int y2 = camera_y + camera_h - 1;

		/* マージンあるかないかでノイズパターンが変わってしまうため、

	near_blur用に広げてはいけない */

		//if (0.0 != near_blur) { x1-=1; x2+=1; y1-=1; y2+=1; }

		if (0.0 != hue_range) {

			int pos = 0;

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, ++pos) {

					if ((xx < x1) || (x2 < xx) || (yy < y1) || (y2 < yy)) {

						this->hue_array_[pos] = hue_range * (hue_rand_ext.next_d() - 0.5);

					} else {

						this->hue_array_[pos] = hue_range * (hue_rand.next_d() - 0.5);

					}

				}

			}

		}

		if (0.0 != sat_range) {

			int pos = 0;

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, ++pos) {

					if ((xx < x1) || (x2 < xx) || (yy < y1) || (y2 < yy)) {

						this->sat_array_[pos] = sat_range * (sat_rand_ext.next_d() - 0.5);

					} else {

						this->sat_array_[pos] = sat_range * (sat_rand.next_d() - 0.5);

					}

				}

			}

		}

		if (0.0 != val_range) {

			int pos = 0;

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, ++pos) {

					if ((xx < x1) || (x2 < xx) || (yy < y1) || (y2 < yy)) {

						this->val_array_[pos] = val_range * (val_rand_ext.next_d() - 0.5);

					} else {

						this->val_array_[pos] = val_range * (val_rand.next_d() - 0.5);

					}

				}

			}

		}

		if (0.0 != alp_range) {

			int pos = 0;

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, ++pos) {

					if ((xx < x1) || (x2 < xx) || (yy < y1) || (y2 < yy)) {

						this->alp_array_[pos] = alp_range * (alp_rand_ext.next_d() - 0.5);

					} else {

						this->alp_array_[pos] = alp_range * (alp_rand.next_d() - 0.5);

					}

				}

			}

		}

	}

	double hue_value(const int xx, const int yy)

	{

		return this->noise_value_(

			this->hue_array_, this->w_, this->h_, xx, yy, this->nblur_);

	}

	double sat_value(const int xx, const int yy)

	{

		return this->noise_value_(

			this->sat_array_, this->w_, this->h_, xx, yy, this->nblur_);

	}

	double val_value(const int xx, const int yy)

	{

		return this->noise_value_(

			this->val_array_, this->w_, this->h_, xx, yy, this->nblur_);

	}

	double alp_value(const int xx, const int yy)

	{

		return this->noise_value_(

			this->alp_array_, this->w_, this->h_, xx, yy, this->nblur_);

	}

	void clear()

	{

		this->alp_array_.clear();

		this->val_array_.clear();

		this->sat_array_.clear();

		this->hue_array_.clear();

	}

	~noise_reference()

	{

		this->clear();

	}

private:

	const int w_;

	const int h_;

	const double nblur_;

	std::vector<double> hue_array_;</double>

	std::vector<double> sat_array_;</double>

	std::vector<double> val_array_;</double>

	std::vector<double> alp_array_;</double>

	void accum_in_(

		const double *noise_array, const int ww, const int hh, const int xx, const int yy, double &accum_val, int &accum_count)

	{

		if ((0 <= xx) && (xx < ww) && (0 <= yy) && (yy < hh)) {

			accum_val += noise_array[yy * ww + xx];

			++accum_count;

		}

	}

	double noise_value_(

		const std::vector<double> &noise_vector, const int ww, const int hh, const int xx, const int yy, const double near_blur)</double>

	{

		if (noise_vector.size() <= 0) {

			return 0.0;

		}

		const double *noise_array = &noise_vector.at(0);

		if (0.0 == near_blur) {

			return noise_array[yy * ww + xx];

		}

		double accum_val = 0.0;

		int accum_count = 0;

		this->accum_in_(noise_array, ww, hh, xx - 1, yy - 1, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx, yy - 1, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx + 1, yy - 1, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx - 1, yy, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx + 1, yy, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx - 1, yy + 1, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx, yy + 1, accum_val, accum_count);

		this->accum_in_(noise_array, ww, hh, xx + 1, yy + 1, accum_val, accum_count);

		if (accum_count <= 0) {

			return noise_array[yy * ww + xx];

		}

		accum_val /= static_cast<double>(accum_count);</double>

		/* 中心Pixelと廻り(1Pixel幅)の平均値とのバランスを返す */

		return (near_blur * accum_val) +

			   ((1.0 - near_blur) * noise_array[yy * ww + xx]);

	}

	/* copy constructorを無効化 */

	noise_reference(const noise_reference &);

	/* 代入演算子を無効化 */

	noise_reference &operator=(const noise_reference &);

};

}

}

namespace igs

{

namespace hsv_noise_in_camera

{

/* 端値を適度に調整する */

class control_term_within_limits

{

public:

	control_term_within_limits(

		const double effective_low = 0.0, const double effective_high = 0.0, const double center = 0.5, const int type = 0, const double noise_range = 0.0) : effective_low_(effective_low),

																																							  effective_high_(effective_high),

																																							  center_(center),

																																							  type_(static_cast<term_type_>(type)),</term_type_>

																																							  noise_range_(noise_range)

	{

	}

	void exec(

		const double current_value /* 0...1 */

		,

		double &noise /* -noise_range/2...noise_range/2 */

		,

		double &shift_value)

	{

		if ((0.0 < this->effective_low_) &&

			(current_value < this->center_)) {

			const double cen = this->center_;

			const double val = current_value;

			const double ran = this->noise_range_;

			const double eff = this->effective_low_;

			switch (this->type_) {

			case shift_all_:

				shift_value = ((cen - val) / cen) * (ran / 2.0) * eff;

				break;

			case shift_term_:

				if (val < ran) {

					shift_value = (((cen < ran) ? cen : ran) - val) / 2.0 * eff;

				}

				break;

			case decrease_all_: {

				const double tmp = (cen - val) / cen * eff;

				if (0.0 < tmp) {

					noise *= 1.0 - tmp;

				}

			} break;

			case decrease_term_:

				if (val < (ran / 2.0)) {

					const double stop = (cen < (ran / 2.0)) ? cen : ran / 2.0;

					const double tmp = (stop - val) / stop * eff;

					if (0.0 < tmp) {

						noise *= 1.0 - tmp;

					}

				}

				break;

			}

		}

		if ((0.0 < this->effective_high_) &&

			(this->center_ < current_value)) {

			const double cen = this->center_;

			const double val = current_value;

			const double ran = this->noise_range_;

			const double eff = this->effective_high_;

			switch (this->type_) {

			case shift_all_:

				shift_value = ((cen - val) / (1.0 - cen)) * (ran / 2.0) * eff;

				break;

			case shift_term_:

				if ((1.0 - ran) < val) {

					const double ira = 1.0 - ran;

					shift_value = (((cen < ira) ? ira : cen) - val) / 2.0 * eff;

				}

				break;

			case decrease_all_: {

				const double tmp = (val - cen) / (1.0 - cen) * eff;

				if (0.0 < tmp) {

					noise *= 1.0 - tmp;

				}

			} break;

			case decrease_term_:

				if ((1.0 - (ran / 2.0)) < val) {

					const double rpos = 1.0 - (ran / 2.0);

					const double stop = (cen < rpos) ? rpos : cen;

					const double tmp = (val - stop) / (1.0 - stop) * eff;

					if (0.0 < tmp) {

						noise *= 1.0 - tmp;

					}

				}

				break;

			}

		}

	}

	double noise_range(void) const { return this->noise_range_; }

private:

	/* low,high両方ゼロ =exec()内処理せず =端値はカット =default */

	const double effective_low_;

	const double effective_high_;

	const double center_;

	enum term_type_ {				  /* 端値の調整方法 */

					  shift_all_ = 0, /* 0.全体的にノイズ位置がずれる */

					  shift_term_,	/* 1.端のみでノイズ位置がずれる */

					  decrease_all_,  /* 2.全体的にノイズ幅が減る */

					  decrease_term_, /* 3.端のみでノイズ幅が減る */

	};

	const term_type_ type_;

	const double noise_range_;

	/* copy constructorを無効化 */

	control_term_within_limits(const control_term_within_limits &);

	/* 代入演算子を無効化 */

	control_term_within_limits &operator=(const control_term_within_limits &);

};

}

}

namespace igs

{

namespace hsv_noise_in_camera

{

/* RGB値にノイズをのせる */

void pixel_rgb(

	const double red_in, const double gre_in, const double blu_in, const double alp_in, const double hue_noise, const double sat_noise, const double val_noise, control_term_within_limits &sat_term, control_term_within_limits &val_term, double &red_out, double &gre_out, double &blu_out);

/* Alpha値にノイズをのせる */

void pixel_a(

	const double alp_in, const double alp_noise, control_term_within_limits &alp_term, double &alp_out);

}

}

void igs::hsv_noise_in_camera::pixel_rgb(

	const double red_in, const double gre_in, const double blu_in, const double alp_in, const double hue_noise, const double sat_noise, const double val_noise, control_term_within_limits &sat_term, control_term_within_limits &val_term, double &red_out, double &gre_out, double &blu_out)

{

	if (0.0 == alp_in) {

		red_out = red_in;

		gre_out = gre_in;

		blu_out = blu_in;

		return;

	}

	double hue, sat, val;

	igs::color::rgb_to_hsv(red_in, gre_in, blu_in, hue, sat, val);

	if (0.0 != hue_noise) {

		hue += 360.0 * hue_noise * alp_in;

		while (hue < 0.0) {

			hue += 360.0;

		}

		while (360.0 <= hue) {

			hue -= 360.0;

		}

	}

	if (0.0 != sat_term.noise_range()) {

		double shift_value = 0;

		double satnoise = sat_noise;

		sat_term.exec(sat, satnoise, shift_value);

		sat += shift_value * alp_in;

		sat += satnoise * alp_in;

		if (sat < 0.0) {

			sat = 0.0;

		} else if (1.0 < sat) {

			sat = 1.0;

		}

		// if( 0.0 == sat ) hue = -1.0; // hsv_to_rgb(-)

	}

	if (0.0 != val_term.noise_range()) {

		double shift_value = 0;

		double valnoise = val_noise;

		val_term.exec(val, valnoise, shift_value);

		val += shift_value * alp_in;

		val += valnoise * alp_in;

		if (val < 0.0) {

			val = 0.0;

		} else if (1.0 < val) {

			val = 1.0;

		}

	}

	igs::color::hsv_to_rgb(hue, sat, val, red_out, gre_out, blu_out);

}

void igs::hsv_noise_in_camera::pixel_a(

	const double alp_in, const double alp_noise, control_term_within_limits &alp_term, double &alp_out)

{

	// if (0.0 == alp_in) { return; }

	double alpin = alp_in;

	if (0.0 != alp_term.noise_range()) {

		double shift_value = 0.0;

		double alpnoise = alp_noise;

		alp_term.exec(alpin, alpnoise, shift_value);

		const double mask = alpin;

		alpin += shift_value * mask;

		alpin += alpnoise * mask;

		if (alpin < 0.0) {

			alpin = 0.0;

		} else if (1.0 < alpin) {

			alpin = 1.0;

		}

	}

	alp_out = alpin;

}

namespace igs

{

namespace hsv_noise_in_camera

{

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

template <class t=""></class>

void change_template_(

	T *image_array, const int ww, const int hh, const int ch, noise_reference &noise, const double hue_range, control_term_within_limits &sat_term, control_term_within_limits &val_term, control_term_within_limits &alp_term)

{

	const double div_val =

		static_cast<double>(std::numeric_limits<t>::max());</t></double>

	const double mul_val =

		static_cast<double>(std::numeric_limits<t>::max()) + 0.999999;</t></double>

	double rr, gg, bb, aa;

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

		using namespace igs::image::rgba;

		for (int yy = 0; yy < hh; ++yy) {

			for (int xx = 0; xx < ww; ++xx) {

				if (

					((0.0 != hue_range) || (0.0 != val_term.noise_range()) || (0.0 != sat_term.noise_range())) /* ノイズがhsvのどれか一つはある */

					) {

					pixel_rgb(

						static_cast<double>(image_array[red]) / div_val, static_cast<double>(image_array[gre]) / div_val, static_cast<double>(image_array[blu]) / div_val, static_cast<double>(image_array[alp]) / div_val, noise.hue_value(xx, yy), noise.sat_value(xx, yy), noise.val_value(xx, yy), sat_term, val_term, rr, gg, bb);</double></double></double></double>

					image_array[red] = static_cast<t>(rr * mul_val);</t>

					image_array[gre] = static_cast<t>(gg * mul_val);</t>

					image_array[blu] = static_cast<t>(bb * mul_val);</t>

				}

				if (0.0 != alp_term.noise_range()) {

					pixel_a(

						static_cast<double>(image_array[alp]) / div_val, noise.alp_value(xx, yy), alp_term, aa);</double>

					image_array[alp] = static_cast<t>(aa * mul_val);</t>

				}

				image_array += ch;

			}

		}

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

		using namespace igs::image::rgb;

		if (

			((0.0 != hue_range) || (0.0 != sat_term.noise_range()) || (0.0 != val_term.noise_range())) /* ノイズがhsvのどれか一つはある */

			) {

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, image_array += ch) {

					pixel_rgb(

						static_cast<double>(image_array[red]) / div_val, static_cast<double>(image_array[gre]) / div_val, static_cast<double>(image_array[blu]) / div_val, 1.0, noise.hue_value(xx, yy), noise.sat_value(xx, yy), noise.val_value(xx, yy), sat_term, val_term, rr, gg, bb);</double></double></double>

					image_array[red] = static_cast<t>(rr * mul_val);</t>

					image_array[gre] = static_cast<t>(gg * mul_val);</t>

					image_array[blu] = static_cast<t>(bb * mul_val);</t>

				}

			}

		}

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

		if (0.0 != val_term.noise_range()) {

			for (int yy = 0; yy < hh; ++yy) {

				for (int xx = 0; xx < ww; ++xx, ++image_array) {

					double val = static_cast<double>(image_array[0]) / div_val;</double>

					double shift_value = 0;

					double val_noise = noise.val_value(xx, yy);

					val_term.exec(val, val_noise, shift_value);

					val += shift_value;

					val += val_noise;

					if (val < 0.0) {

						val = 0.0;

					} else if (1.0 < val) {

						val = 1.0;

					}

					image_array[0] = static_cast<t>(val * mul_val);</t>

				}

			}

		}

	}

}

}

}

void igs::hsv_noise_in_camera::change(

	void *image_array

	,

	const int height, const int width, const int channels, const int bits

	,

	const int camera_x, const int camera_y, const int camera_w, const int camera_h

	,

	const double hue_range, const double sat_range, const double val_range, const double alp_range, const unsigned long random_seed, const double near_blur

	,

	const double sat_effective, const double sat_center, const int sat_type, const double val_effective, const double val_center, const int val_type, const double alp_effective, const double alp_center, const int alp_type)

{

	if ((0.0 == hue_range) && (0.0 == sat_range) && (0.0 == val_range) && (0.0 == alp_range)) {

		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");

	}

	/* ノイズ参照画像を作成する */

	noise_reference noise(

		width, height, hue_range, sat_range, val_range, alp_range, random_seed, near_blur, camera_x, camera_y, camera_w, camera_h);

	/* 端値を適度に調整する設定 */

	control_term_within_limits sat_term(

		sat_effective, sat_effective, sat_center, sat_type, sat_range);

	control_term_within_limits val_term(

		val_effective, val_effective, val_center, val_type, val_range);

	control_term_within_limits alp_term(

		alp_effective, alp_effective, alp_center, alp_type, alp_range);

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

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

		change_template_(

			static_cast<unsigned *="" char="">(image_array), width, height, channels, noise, hue_range, sat_term, val_term, alp_term);</unsigned>

		noise.clear(); /* ノイズ画像メモリ解放 */

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

		change_template_(

			static_cast<unsigned *="" short="">(image_array), width, height, channels, noise, hue_range, sat_term, val_term, alp_term);</unsigned>

		noise.clear(); /* ノイズ画像メモリ解放 */

	} else {

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

	}

}