#include "traster.h"

#include "trastercm.h"

#include "tpalette.h"

#include "tcolorstyles.h"

//#include "trop.h"

#include "tropcm.h"

#include "tpixelutils.h"

#include <memory></memory>

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

namespace

{

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

template <class class="" pixin="" pixout,=""></class>

void doConvolve_row_9_i(PIXOUT *pixout, int n, PIXIN *pixarr[], long w[])

{

	long w1, w2, w3, w4, w5, w6, w7, w8, w9;

	PIXIN *p1, *p2, *p3, *p4, *p5, *p6, *p7, *p8, *p9;

	w1 = w[0];

	w2 = w[1];

	w3 = w[2];

	w4 = w[3];

	w5 = w[4];

	w6 = w[5];

	w7 = w[6];

	w8 = w[7];

	w9 = w[8];

	p1 = pixarr[0];

	p2 = pixarr[1];

	p3 = pixarr[2];

	p4 = pixarr[3];

	p5 = pixarr[4];

	p6 = pixarr[5];

	p7 = pixarr[6];

	p8 = pixarr[7];

	p9 = pixarr[8];

	int rightShift = 16 + ((int)sizeof(typename PIXIN::Channel) - (int)sizeof(typename PIXOUT::Channel)) * 8;

	while (n-- > 0) {

		pixout->r = (typename PIXOUT::Channel)((p1->r * w1 + p2->r * w2 + p3->r * w3 +

												p4->r * w4 + p5->r * w5 + p6->r * w6 +

												p7->r * w7 + p8->r * w8 + p9->r * w9 + (1 << 15)) >>

											   rightShift);

		pixout->g = (typename PIXOUT::Channel)((p1->g * w1 + p2->g * w2 + p3->g * w3 +

												p4->g * w4 + p5->g * w5 + p6->g * w6 +

												p7->g * w7 + p8->g * w8 + p9->g * w9 + (1 << 15)) >>

											   rightShift);

		pixout->b = (typename PIXOUT::Channel)((p1->b * w1 + p2->b * w2 + p3->b * w3 +

												p4->b * w4 + p5->b * w5 + p6->b * w6 +

												p7->b * w7 + p8->b * w8 + p9->b * w9 + (1 << 15)) >>

											   rightShift);

		pixout->m = (typename PIXOUT::Channel)((p1->m * w1 + p2->m * w2 + p3->m * w3 +

												p4->m * w4 + p5->m * w5 + p6->m * w6 +

												p7->m * w7 + p8->m * w8 + p9->m * w9 + (1 << 15)) >>

											   rightShift);

		p1++;

		p2++;

		p3++;

		p4++;

		p5++;

		p6++;

		p7++;

		p8++;

		p9++;

		pixout++;

	}

}

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

template <class pixout=""></class>

void doConvolve_cm32_row_9_i(PIXOUT *pixout, int n,

							 TPixelCM32 *pixarr[], long w[],

							 const std::vector<tpixel32> &paints,</tpixel32>

							 const std::vector<tpixel32> &inks)</tpixel32>

{

	long w1, w2, w3, w4, w5, w6, w7, w8, w9;

	TPixelCM32 *p1, *p2, *p3, *p4, *p5, *p6, *p7, *p8, *p9;

	TPixel32 val[9];

	w1 = w[0];

	w2 = w[1];

	w3 = w[2];

	w4 = w[3];

	w5 = w[4];

	w6 = w[5];

	w7 = w[6];

	w8 = w[7];

	w9 = w[8];

	p1 = pixarr[0];

	p2 = pixarr[1];

	p3 = pixarr[2];

	p4 = pixarr[3];

	p5 = pixarr[4];

	p6 = pixarr[5];

	p7 = pixarr[6];

	p8 = pixarr[7];

	p9 = pixarr[8];

	while (n-- > 0) {

		for (int i = 0; i < 9; ++i) {

			int tone = p1->getTone();

			int paint = p1->getPaint();

			int ink = p1->getInk();

			if (tone == TPixelCM32::getMaxTone())

				val[i] = paints[paint];

			else if (tone == 0)

				val[i] = inks[ink];

			else

				val[i] = blend(inks[ink], paints[paint], tone, TPixelCM32::getMaxTone());

		}

		pixout->r = (typename PIXOUT::Channel)((val[1].r * w1 + val[2].r * w2 + val[3].r * w3 +

												val[4].r * w4 + val[5].r * w5 + val[6].r * w6 +

												val[7].r * w7 + val[8].r * w8 + val[9].r * w9 + (1 << 15)) >>

											   16);

		pixout->g = (typename PIXOUT::Channel)((val[1].g * w1 + val[2].g * w2 + val[3].g * w3 +

												val[4].g * w4 + val[5].g * w5 + val[6].g * w6 +

												val[7].g * w7 + val[8].g * w8 + val[9].g * w9 + (1 << 15)) >>

											   16);

		pixout->b = (typename PIXOUT::Channel)((val[1].b * w1 + val[2].b * w2 + val[3].b * w3 +

												val[4].b * w4 + val[5].b * w5 + val[6].b * w6 +

												val[7].b * w7 + val[8].b * w8 + val[9].b * w9 + (1 << 15)) >>

											   16);

		pixout->m = (typename PIXOUT::Channel)((val[1].m * w1 + val[2].m * w2 + val[3].m * w3 +

												val[4].m * w4 + val[5].m * w5 + val[6].m * w6 +

												val[7].m * w7 + val[8].m * w8 + val[9].m * w9 + (1 << 15)) >>

											   16);

		p1++;

		p2++;

		p3++;

		p4++;

		p5++;

		p6++;

		p7++;

		p8++;

		p9++;

		pixout++;

	}

}

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

template <class class="" pixin="" pixout,=""></class>

void doConvolve_row_i(PIXOUT *pixout, int n,

					  PIXIN *pixarr[], long w[], int pixn)

{

	long ar, ag, ab, am;

	int i;

	int rightShift = 16 + ((int)sizeof(typename PIXIN::Channel) - (int)sizeof(typename PIXOUT::Channel)) * 8;

	while (n-- > 0) {

		ar = ag = ab = am = 0;

		for (i = 0; i < pixn; i++) {

			ar += pixarr[i]->r * w[i];

			ag += pixarr[i]->g * w[i];

			ab += pixarr[i]->b * w[i];

			am += pixarr[i]->m * w[i];

			pixarr[i]++;

		}

		pixout->r = (typename PIXOUT::Channel)((ar + (1 << 15)) >> rightShift);

		pixout->g = (typename PIXOUT::Channel)((ag + (1 << 15)) >> rightShift);

		pixout->b = (typename PIXOUT::Channel)((ab + (1 << 15)) >> rightShift);

		pixout->m = (typename PIXOUT::Channel)((am + (1 << 15)) >> rightShift);

		pixout++;

	}

}

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

template <class pixout=""></class>

void doConvolve_cm32_row_i(PIXOUT *pixout, int n,

						   TPixelCM32 *pixarr[], long w[], int pixn,

						   const std::vector<tpixel32> &paints,</tpixel32>

						   const std::vector<tpixel32> &inks)</tpixel32>

{

	long ar, ag, ab, am;

	int i;

	while (n-- > 0) {

		ar = ag = ab = am = 0;

		for (i = 0; i < pixn; i++) {

			TPixel32 val;

			int tone = pixarr[i]->getTone();

			int paint = pixarr[i]->getPaint();

			int ink = pixarr[i]->getInk();

			if (tone == TPixelCM32::getMaxTone())

				val = paints[paint];

			else if (tone == 0)

				val = inks[ink];

			else

				val = blend(inks[ink], paints[paint], tone, TPixelCM32::getMaxTone());

			ar += val.r * w[i];

			ag += val.g * w[i];

			ab += val.b * w[i];

			am += val.m * w[i];

			pixarr[i]++;

		}

		pixout->r = (typename PIXOUT::Channel)((ar + (1 << 15)) >> 16);

		pixout->g = (typename PIXOUT::Channel)((ag + (1 << 15)) >> 16);

		pixout->b = (typename PIXOUT::Channel)((ab + (1 << 15)) >> 16);

		pixout->m = (typename PIXOUT::Channel)((am + (1 << 15)) >> 16);

		pixout++;

	}

}

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

template <class class="" pixin="" pixout,=""></class>

void doConvolve_3_i(TRasterPT<pixout> rout,</pixout>

					TRasterPT<pixin> rin,</pixin>

					int dx, int dy,

					double conv[])

{

	PIXIN *bufferin;

	PIXOUT *bufferout;

	PIXIN *pixin;

	PIXOUT *pixout;

	PIXIN *pixarr[9];

	long w[9];

	int pixn;

	int wrapin, wrapout;

	int x, y, n;

	int x1, y1, x2, y2;

	int fx1, fy1, fx2, fy2, fx, fy;

	rout->clear();

	wrapin = rin->getWrap();

	wrapout = rout->getWrap();

	/* calcolo l'area di output interessata */

	x1 = std::max(0, -dx - 1);

	y1 = std::max(0, -dy - 1);

	x2 = std::min(rout->getLx() - 1, -dx + rin->getLx());

	y2 = std::min(rout->getLy() - 1, -dy + rin->getLy());

	rin->lock();

	rout->lock();

	bufferin = rin->pixels();

	bufferout = rout->pixels();

	for (y = y1; y <= y2; y++) {

		fy1 = std::max(-1, -dy - y);

		fy2 = std::min(1, -dy + rin->getLy() - 1 - y);

		if (fy1 > fy2)

			continue;

		x = x1;

		pixout = bufferout + wrapout * y + x;

		pixin = bufferin + wrapin * (y + dy) + (x + dx);

		while (x <= x2) {

			fx1 = std::max(-1, -dx - x);

			fx2 = std::min(1, -dx + rin->getLx() - 1 - x);

			if (x > -dx && x < -dx + rin->getLx() - 1)

				n = std::min(-dx + rin->getLx() - 1 - x, x2 - x + 1);

			else

				n = 1;

			if (n < 1)

				break;

			pixn = 0;

			for (fy = fy1; fy <= fy2; fy++)

				for (fx = fx1; fx <= fx2; fx++) {

					pixarr[pixn] = pixin + fy * wrapin + fx;

					w[pixn] = (long)(conv[(fy + 1) * 3 + fx + 1] * (1 << 16));

					pixn++;

				}

			if (pixn == 9)

				doConvolve_row_9_i<pixout, pixin="">(pixout, n, pixarr, w);</pixout,>

			else

				doConvolve_row_i<pixout, pixin="">(pixout, n, pixarr, w, pixn);</pixout,>

			x += n;

			pixin += n;

			pixout += n;

		}

	}

	rin->unlock();

	rout->unlock();

}

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

template <class class="" pixin="" pixout,=""></class>

void doConvolve_i(TRasterPT<pixout> rout,</pixout>

				  TRasterPT<pixin> rin,</pixin>

				  int dx, int dy,

				  double conv[], int radius)

{

	PIXIN *bufferin;

	PIXOUT *bufferout;

	PIXIN *pixin;

	PIXOUT *pixout;

	int radiusSquare = sq(radius);

	std::unique_ptr<pixin*[]> pixarr(new PIXIN *[radiusSquare]);</pixin*[]>

	std::unique_ptr<long[]> w(new long[radiusSquare]);</long[]>

	int pixn;

	int wrapin, wrapout;

	int x, y, n;

	int x1, y1, x2, y2;

	int fx1, fy1, fx2, fy2, fx, fy;

	int radius1 = radius / 2;

	int radius0 = radius1 - radius + 1;

	rout->clear();

	wrapin = rin->getWrap();

	wrapout = rout->getWrap();

	/* calcolo l'area di output interessata */

	x1 = std::max(0, -dx - 1);

	y1 = std::max(0, -dy - 1);

	x2 = std::min(rout->getLx() - 1, -dx + rin->getLx());

	y2 = std::min(rout->getLy() - 1, -dy + rin->getLy());

	rin->lock();

	rout->lock();

	bufferin = rin->pixels();

	bufferout = rout->pixels();

	for (y = y1; y <= y2; y++) {

		fy1 = std::max(radius0, -dy - y);

		fy2 = std::min(radius1, -dy - y + rin->getLy() - 1);

		if (fy1 > fy2)

			continue;

		x = x1;

		pixout = bufferout + wrapout * y + x;

		pixin = bufferin + wrapin * (y + dy) + (x + dx);

		while (x <= x2) {

			fx1 = std::max(radius0, -dx - x);

			fx2 = std::min(radius1, -dx - x + rin->getLx() - 1);

			if (x > -dx && x < -dx + rin->getLx() - 1)

				n = std::min(-dx + rin->getLx() - 1 - x, x2 - x + 1);

			else

				n = 1;

			if (n < 1)

				break;

			pixn = 0;

			for (fy = fy1; fy <= fy2; fy++)

				for (fx = fx1; fx <= fx2; fx++) {

					pixarr[pixn] = pixin + fy * wrapin + fx;

					w[pixn] = (long)(conv[(fy - radius0) * radius + fx - radius0] * (1 << 16));

					pixn++;

				}

			doConvolve_row_i<pixout, pixin="">(pixout, n, pixarr.get(), w.get(), pixn);</pixout,>

			x += n;

			pixin += n;

			pixout += n;

		}

	}

	rin->unlock();

	rout->unlock();

}

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

template <class pixout=""></class>

void doConvolve_cm32_3_i(TRasterPT<pixout> rout,</pixout>

						 TRasterCM32P rin,

						 const TPaletteP &palette,

						 int dx, int dy,

						 double conv[])

{

	TPixelCM32 *pixin;

	PIXOUT *pixout;

	TPixelCM32 *pixarr[9];

	long w[9];

	int pixn;

	int wrapin, wrapout;

	int x, y, n;

	int x1, y1, x2, y2;

	int fx1, fy1, fx2, fy2, fx, fy;

	rout->clear();

	wrapin = rin->getWrap();

	wrapout = rout->getWrap();

	/* calcolo l'area di output interessata */

	x1 = std::max(0, -dx - 1);

	y1 = std::max(0, -dy - 1);

	x2 = std::min(rout->getLx() - 1, -dx + rin->getLx());

	y2 = std::min(rout->getLy() - 1, -dy + rin->getLy());

	int colorCount = palette->getStyleCount();

	colorCount = std::max({colorCount, TPixelCM32::getMaxInk(), TPixelCM32::getMaxPaint()});

	std::vector<tpixel32> paints(colorCount);</tpixel32>

	std::vector<tpixel32> inks(colorCount);</tpixel32>

	rin->lock();

	rout->lock();

	TPixelCM32 *bufferin = rin->pixels();

	PIXOUT *bufferout = rout->pixels();

	for (int i = 0; i < palette->getStyleCount(); i++)

		paints[i] = inks[i] = palette->getStyle(i)->getAverageColor();

	for (y = y1; y <= y2; y++) {

		fy1 = std::max(-1, -dy - y);

		fy2 = std::min(1, -dy + rin->getLy() - 1 - y);

		if (fy1 > fy2)

			continue;

		x = x1;

		pixout = bufferout + wrapout * y + x;

		pixin = bufferin + wrapin * (y + dy) + (x + dx);

		while (x <= x2) {

			fx1 = std::max(-1, -dx - x);

			fx2 = std::min(1, -dx + rin->getLx() - 1 - x);

			if (x > -dx && x < -dx + rin->getLx() - 1)

				n = std::min(-dx + rin->getLx() - 1 - x, x2 - x + 1);

			else

				n = 1;

			if (n < 1)

				break;

			pixn = 0;

			for (fy = fy1; fy <= fy2; fy++)

				for (fx = fx1; fx <= fx2; fx++) {

					pixarr[pixn] = pixin + fy * wrapin + fx;

					w[pixn] = (long)(conv[(fy + 1) * 3 + fx + 1] * (1 << 16));

					pixn++;

				}

			if (pixn == 9)

				doConvolve_cm32_row_9_i<pixout>(pixout, n, pixarr, w, paints, inks);</pixout>

			else

				doConvolve_cm32_row_i<pixout>(pixout, n, pixarr, w, pixn, paints, inks);</pixout>

			x += n;

			pixin += n;

			pixout += n;

		}

	}

	rin->unlock();

	rout->unlock();

}

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

template <class pixout=""></class>

void doConvolve_cm32_i(TRasterPT<pixout> rout,</pixout>

					   TRasterCM32P rin,

					   const TPaletteP &palette,

					   int dx, int dy,

					   double conv[], int radius)

{

	TPixelCM32 *pixin;

	PIXOUT *pixout;

	int radiusSquare = sq(radius);

	std::unique_ptr<tpixelcm32*[]> pixarr(new TPixelCM32*[radiusSquare]);</tpixelcm32*[]>

	std::unique_ptr<long[]> w(new long[radiusSquare]);</long[]>

	int pixn;

	int wrapin, wrapout;

	int x, y, n;

	int x1, y1, x2, y2;

	int fx1, fy1, fx2, fy2, fx, fy;

	int radius1 = radius / 2;

	int radius0 = radius1 - radius + 1;

	rout->clear();

	wrapin = rin->getWrap();

	wrapout = rout->getWrap();

	/* calcolo l'area di output interessata */

	x1 = std::max(0, -dx - 1);

	y1 = std::max(0, -dy - 1);

	x2 = std::min(rout->getLx() - 1, -dx + rin->getLx());

	y2 = std::min(rout->getLy() - 1, -dy + rin->getLy());

	int colorCount = palette->getStyleCount();

	colorCount = std::max({colorCount, TPixelCM32::getMaxInk(), TPixelCM32::getMaxPaint()});

	std::vector<tpixel32> paints(colorCount);</tpixel32>

	std::vector<tpixel32> inks(colorCount);</tpixel32>

	rin->lock();

	rout->lock();

	TPixelCM32 *bufferin = rin->pixels();

	PIXOUT *bufferout = rout->pixels();

	for (int i = 0; i < palette->getStyleCount(); i++)

		paints[i] = inks[i] = palette->getStyle(i)->getAverageColor();

	for (y = y1; y <= y2; y++) {

		fy1 = std::max(radius0, -dy - y);

		fy2 = std::min(radius1, -dy + rin->getLy() - 1 - y);

		if (fy1 > fy2)

			continue;

		x = x1;

		pixout = bufferout + wrapout * y + x;

		pixin = bufferin + wrapin * (y + dy) + (x + dx);

		while (x <= x2) {

			fx1 = std::max(radius0, -dx - x);

			fx2 = std::min(radius1, -dx + rin->getLx() - 1 - x);

			if (x > -dx && x < -dx + rin->getLx() - 1)

				n = std::min(-dx + rin->getLx() - 1 - x, x2 - x + 1);

			else

				n = 1;

			if (n < 1)

				break;

			pixn = 0;

			for (fy = fy1; fy <= fy2; fy++)

				for (fx = fx1; fx <= fx2; fx++) {

					pixarr[pixn] = pixin + fy * wrapin + fx;

					w[pixn] = (long)(conv[(fy - radius0) * radius + fx - radius0] * (1 << 16));

					pixn++;

				}

			doConvolve_cm32_row_i<pixout>(pixout, n, pixarr.get(), w.get(), pixn, paints, inks);</pixout>

			x += n;

			pixin += n;

			pixout += n;

		}

	}

	rin->unlock();

	rout->unlock();

}

} // anonymous namespace

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

void TRop::convolve_3_i(TRasterP rout, TRasterP rin, int dx, int dy, double conv[])

{

	TRaster32P rin32 = rin;

	if (rin32) {

		TRaster32P rout32 = rout;

		if (rout32) {

			doConvolve_3_i<tpixel32, tpixel32="">(rout32, rin32, dx, dy, conv);</tpixel32,>

			return;

		}

		TRaster64P rout64 = rout;

		if (rout64) {

			doConvolve_3_i<tpixel64, tpixel32="">(rout64, rin32, dx, dy, conv);</tpixel64,>

			return;

		}

	} else {

		TRaster64P rin64 = rin;

		if (rin64) {

			TRaster32P rout32 = rout;

			if (rout32) {

				doConvolve_3_i<tpixel32, tpixel64="">(rout32, rin64, dx, dy, conv);</tpixel32,>

				return;

			}

			TRaster64P rout64 = rout;

			if (rout64) {

				doConvolve_3_i<tpixel64, tpixel64="">(rout64, rin64, dx, dy, conv);</tpixel64,>

				return;

			}

		}

	}

	throw TRopException("TRop::convolve_3_i: unsupported pixel type");

}

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

void TRop::convolve_3_i(TRasterP rout, TRasterCM32P rin, const TPaletteP &palette,

						int dx, int dy, double conv[])

{

	TRaster32P rout32 = rout;

	if (rout32) {

		doConvolve_cm32_3_i<tpixel32>(rout32, rin, palette, dx, dy, conv);</tpixel32>

		return;

	}

	TRaster64P rout64 = rout;

	if (rout64) {

		doConvolve_cm32_3_i<tpixel64>(rout64, rin, palette, dx, dy, conv);</tpixel64>

		return;

	}

	throw TRopException("TRop::convolve_3_i: unsupported pixel type");

}

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

void TRop::convolve_i(TRasterP rout, TRasterP rin, int dx, int dy, double conv[], int radius)

{

	TRaster32P rin32 = rin;

	if (rin32) {

		TRaster32P rout32 = rout;

		if (rout32) {

			doConvolve_i<tpixel32, tpixel32="">(rout32, rin32, dx, dy, conv, radius);</tpixel32,>

			return;

		}

		TRaster64P rout64 = rout;

		if (rout64) {

			doConvolve_i<tpixel64, tpixel32="">(rout64, rin32, dx, dy, conv, radius);</tpixel64,>

			return;

		}

	} else {

		TRaster64P rin64 = rin;

		if (rin64) {

			TRaster32P rout32 = rout;

			if (rout32) {

				doConvolve_i<tpixel32, tpixel64="">(rout32, rin64, dx, dy, conv, radius);</tpixel32,>

				return;

			}

			TRaster64P rout64 = rout;

			if (rout64) {

				doConvolve_i<tpixel64, tpixel64="">(rout64, rin64, dx, dy, conv, radius);</tpixel64,>

				return;

			}

		}

	}

	throw TRopException("TRop::convolve_i: unsupported pixel type");

}

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

void TRop::convolve_i(TRasterP rout, TRasterCM32P rin, const TPaletteP &palette,

					  int dx, int dy, double conv[], int radius)

{

	TRaster32P rout32 = rout;

	if (rout32) {

		doConvolve_cm32_i<tpixel32>(rout32, rin, palette, dx, dy, conv, radius);</tpixel32>

		return;

	}

	TRaster64P rout64 = rout;

	if (rout64) {

		doConvolve_cm32_i<tpixel64>(rout64, rin, palette, dx, dy, conv, radius);</tpixel64>

		return;

	}

	throw TRopException("TRop::convolve_i: unsupported pixel type");

}