#include "quickputP.h"

#include "tpixelutils.h"

#include "trastercm.h"

#include "tsystem.h"

#include "tropcm.h"

#include "tpalette.h"

#ifdef _WIN32

#include <emmintrin.h> // per SSE2</emmintrin.h>

#endif

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

namespace

{

inline bool transp(const TPixel32 &p)

{

	return p.m == 0;

}

inline bool transp(const TPixel64 &p)

{

	return p.m == 0;

}

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

inline bool opaque(const TPixel32 &p)

{

	return p.m == 0xff;

}

inline bool opaque(const TPixel64 &p)

{

	return p.m == 0xffff;

}

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

#define MODO2

#define VELOCE

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

void do_overT3(TRasterPT<t> rout, const TRasterPT<t> &rdn, const TRasterPT<t> &rup)</t></t></t>

{

	for (int y = 0; y < rout->getLy(); y++) {

#ifdef MODO1

		const T *dn_pix = rdn->pixels(y);

		const T *up_pix = rup->pixels(y);

		T *out_pix = rout->pixels(y);

#else

#ifdef MODO2

		const T *dn_pix = ((T *)rdn->getRawData()) + y * rdn->getWrap();

		const T *up_pix = ((T *)rup->getRawData()) + y * rup->getWrap();

		T *out_pix = ((T *)rout->getRawData()) + y * rout->getWrap();

#endif

#endif

		const T *dn_limit = dn_pix + rdn->getLx();

		for (; dn_pix < dn_limit; dn_pix++, up_pix++, out_pix++) {

#ifdef VELOCE

			if (transp(*up_pix))

				*out_pix = *dn_pix;

			else if (opaque(*up_pix))

				*out_pix = *up_pix;

			else {

				*out_pix = overPix(*dn_pix, *up_pix);

			}

#else

			T topval = *up_pix;

			if (transp(topval))

				*out_pix = *dn_pix;

			else if (opaque(topval))

				*out_pix = topval;

			else {

				*out_pix = overPix(*dn_pix, topval);

			}

#endif

		}

	}

}

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

template <typename pixtypedn,="" pixtypeout,="" pixtypeup="" typename=""></typename>

void do_over(TRasterPT<pixtypeout> rout, const TRasterPT<pixtypedn> &rdn,</pixtypedn></pixtypeout>

			 const TRasterPT<pixtypeup> &rup, const TRasterGR8P rmask)</pixtypeup>

{

	for (int y = 0; y < rout->getLy(); y++) {

		const PixTypeDn *dn_pix = ((PixTypeDn *)rdn->getRawData()) + y * rdn->getWrap();

		const PixTypeUp *up_pix = ((PixTypeUp *)rup->getRawData()) + y * rup->getWrap();

		PixTypeOut *out_pix = ((PixTypeOut *)rout->getRawData()) + y * rout->getWrap();

		TPixelGR8 *mask_pix = ((TPixelGR8 *)rmask->getRawData()) + y * rmask->getWrap();

		const PixTypeDn *dn_limit = dn_pix + rout->getLx();

		for (; dn_pix < dn_limit; dn_pix++, up_pix++, out_pix++, mask_pix++) {

			if (mask_pix->value == 0x00)

				*out_pix = *dn_pix;

			else if (mask_pix->value == 0xff)

				*out_pix = *up_pix;

			else {

				PixTypeUp p(*up_pix);

				p.m = mask_pix->value;

				*out_pix = overPix(*dn_pix, p); //hei!

			}

		}

	}

}

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

void do_over(TRasterCM32P rout, const TRasterCM32P &rup)

{

	assert(rout->getSize() == rup->getSize());

	for (int y = 0; y < rout->getLy(); y++) {

		TPixelCM32 *out_pix = rout->pixels(y);

		TPixelCM32 *const out_end = out_pix + rout->getLx();

		const TPixelCM32 *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			if (!up_pix->isPureInk() &&

				up_pix->getPaint() != 0) // BackgroundStyle)

				*out_pix = *up_pix;

			else if (!up_pix->isPurePaint()) {

				TUINT32 *outl = (TUINT32 *)out_pix, *upl = (TUINT32 *)up_pix;

				*outl = ((*upl) & (TPixelCM32::getInkMask())) |

						((*outl) & (TPixelCM32::getPaintMask())) |

						std::min(up_pix->getTone(), out_pix->getTone());

			}

		}

	}

}

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

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

void do_overT2(TRasterPT<t> rout, const TRasterPT<t> &rup)</t></t>

{

	UINT max = T::maxChannelValue;

	double maxD = max;

	assert(rout->getSize() == rup->getSize());

	for (int y = 0; y < rout->getLy(); y++) {

		T *out_pix = rout->pixels(y);

		T *const out_end = out_pix + rout->getLx();

		const T *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			if (up_pix->m == max)

				*out_pix = *up_pix;

			else if (up_pix->m > 0) {

				TUINT32 r, g, b;

				r = up_pix->r + (out_pix->r * (max - up_pix->m)) / maxD;

				g = up_pix->g + (out_pix->g * (max - up_pix->m)) / maxD;

				b = up_pix->b + (out_pix->b * (max - up_pix->m)) / maxD;

				out_pix->r = (r < max) ? (Q)r : (Q)max;

				out_pix->g = (g < max) ? (Q)g : (Q)max;

				out_pix->b = (b < max) ? (Q)b : (Q)max;

				out_pix->m = up_pix->m + (out_pix->m * (max - up_pix->m)) / maxD;

			}

		}

	}

}

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

#ifdef _WIN32

void do_over_SSE2(TRaster32P rout, const TRaster32P &rup)

{

	__m128i zeros = _mm_setzero_si128();

	__m128i out_pix_packed_i, up_pix_packed_i;

	__m128 out_pix_packed, up_pix_packed;

	float maxChannelValue = 255.0;

	float maxChannelValueInv = 1.0f / maxChannelValue;

	__m128 maxChanneValue_packed = _mm_load1_ps(&maxChannelValue);

	assert(rout->getSize() == rup->getSize());

	for (int y = 0; y < rout->getLy(); y++) {

		TPixel32 *out_pix = rout->pixels(y);

		TPixel32 *const out_end = out_pix + rout->getLx();

		const TPixel32 *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			if (up_pix->m == 0xff)

				*out_pix = *up_pix;

			else if (up_pix->m > 0) {

				float factor = (255.0f - up_pix->m) / 255.0f;

				__m128 factor_packed = _mm_load1_ps(&factor);

				// carica up_pix e out_pix in due registri a 128 bit

				up_pix_packed_i = _mm_unpacklo_epi8(_mm_cvtsi32_si128(*(DWORD *)up_pix), zeros);

				up_pix_packed = _mm_cvtepi32_ps(_mm_unpacklo_epi16(up_pix_packed_i, zeros));

				out_pix_packed_i = _mm_unpacklo_epi8(_mm_cvtsi32_si128(*(DWORD *)out_pix), zeros);

				out_pix_packed = _mm_cvtepi32_ps(_mm_unpacklo_epi16(out_pix_packed_i, zeros));

				out_pix_packed = _mm_add_ps(up_pix_packed, _mm_mul_ps(out_pix_packed, factor_packed));

				out_pix_packed = _mm_min_ps(maxChanneValue_packed, out_pix_packed);

				out_pix_packed_i = _mm_cvtps_epi32(out_pix_packed);

				out_pix_packed_i = _mm_packs_epi32(out_pix_packed_i, zeros);

				out_pix_packed_i = _mm_packus_epi16(out_pix_packed_i, zeros);

				*(DWORD *)(out_pix) = _mm_cvtsi128_si32(out_pix_packed_i);

			}

		}

	}

}

#endif

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

void do_over(TRaster32P rout, const TRasterGR8P &rup)

{

	assert(rout->getSize() == rup->getSize());

	for (int y = rout->getLy(); --y >= 0;) {

		TPixel32 *out_pix = rout->pixels(y);

		TPixel32 *const out_end = out_pix + rout->getLx();

		const TPixelGR8 *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			int v = up_pix->value;

			out_pix->r = out_pix->r * v / 255;

			out_pix->g = out_pix->r;

			out_pix->b = out_pix->r;

		}

	}

}

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

void do_over(TRasterGR8P rout, const TRaster32P &rup)

{

	assert(rout->getSize() == rup->getSize());

	for (int y = rout->getLy(); --y >= 0;) {

		TPixelGR8 *out_pix = rout->pixels(y);

		TPixelGR8 *const out_end = out_pix + rout->getLx();

		const TPixel32 *up_pix = rup->pixels(y);

		TPixel32 *temp_pix = new TPixel32();

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			temp_pix->r = out_pix->value;

			temp_pix->g = out_pix->value;

			temp_pix->b = out_pix->value;

			temp_pix->m = 0xff;

			TPixel32 out32_pix = overPix(*temp_pix, *up_pix);

			*out_pix = out_pix->from(out32_pix);

		}

		delete temp_pix;

	}

}

} // namespace

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

void do_over(TRaster32P rout, const TRasterGR8P &rup, const TPixel32 &color)

{

	assert(rout->getSize() == rup->getSize());

	for (int y = rout->getLy(); --y >= 0;) {

		TPixel32 *out_pix = rout->pixels(y);

		TPixel32 *const out_end = out_pix + rout->getLx();

		const TPixelGR8 *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			double v = up_pix->value / 255.0;

			TPixel32 up(troundp(v * color.r), troundp(v * color.g), troundp(v * color.b), troundp(v * color.m));

			*out_pix = overPix(*out_pix, up);

		}

	}

}

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

void TRop::over(TRaster32P rout, const TRasterGR8P &rup, const TPixel32 &color)

{

	rout->lock();

	do_over(rout, rup, color);

	rout->unlock();

}

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

void TRop::over(const TRasterP &rout, const TRasterP &rdn, const TRasterP &rup)

{

	TRect rect = rout->getBounds() * rdn->getBounds() * rup->getBounds();

	if (rect.isEmpty())

		return;

	TRasterP cRout = rout->extract(rect);

	TRasterP cRdn = rdn->extract(rect);

	TRasterP cRup = rup->extract(rect);

	rout->lock();

	rdn->lock();

	rup->lock();

	TRaster32P rout32 = cRout, rdn32 = cRdn, rup32 = cRup;

	TRaster64P rout64 = cRout, rdn64 = cRdn, rup64 = cRup;

	if (rout32 && rdn32 && rup32)

		do_overT3<tpixel32>(rout32, rdn32, rup32);</tpixel32>

	else if (rout64 && rdn64 && rup64)

		do_overT3<tpixel64>(rout64, rdn64, rup64);</tpixel64>

	else {

		rout->unlock();

		rdn->unlock();

		rup->unlock();

		throw TRopException("unsupported pixel type");

	}

	rout->unlock();

	rdn->unlock();

	rup->unlock();

}

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

void TRop::over(const TRasterP &rout, const TRasterP &rup, const TPoint &pos)

{

	TRect outRect(rout->getBounds());

	TRect upRect(rup->getBounds() + pos);

	TRect intersection = outRect * upRect;

	if (intersection.isEmpty())

		return;

	TRasterP cRout = rout->extract(intersection);

	TRect r = intersection - pos;

	TRasterP cRup = rup->extract(r);

	TRaster32P rout32 = cRout, rup32 = cRup;

	TRaster64P rout64 = cRout, rup64 = cRup;

	TRasterGR8P rout8 = cRout, rup8 = cRup;

	TRasterCM32P routCM32 = cRout, rupCM32 = cRup;

	rout->lock();

	rup->lock();

	// TRaster64P rout64 = rout, rin64 = rin;

	if (rout32 && rup32) {

#ifdef _WIN32

		if (TSystem::getCPUExtensions() & TSystem::CpuSupportsSse2)

			do_over_SSE2(rout32, rup32);

		else

#endif

			do_overT2<tpixel32, uchar="">(rout32, rup32);</tpixel32,>

	} else if (rout64) {

		if (!rup64) {

			TRaster64P raux(cRup->getSize());

			TRop::convert(raux, cRup);

			rup64 = raux;

		}

		do_overT2<tpixel64, ushort="">(rout64, rup64);</tpixel64,>

	} else if (rout32 && rup8)

		do_over(rout32, rup8);

	else if (rout8 && rup32)

		do_over(rout8, rup32);

	else if (rout8 && rup8)

		TRop::copy(rout8, rup8);

	else if (routCM32 && rupCM32)

		do_over(routCM32, rupCM32);

	else {

		rout->unlock();

		rup->unlock();

		throw TRopException("unsupported pixel type");

	}

	rout->unlock();

	rup->unlock();

}

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

static void addBackground32(TRaster32P ras, const TPixel32 &col)

{

	ras->lock();

	int nrows = ras->getLy();

	while (nrows-- > 0) {

		TPixel32 *pix = ras->pixels(nrows);

		TPixel32 *endPix = pix + ras->getLx();

		while (pix < endPix) {

			*pix = overPix(col, *pix);

			pix++;

		}

	}

	ras->unlock();

}

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

void TRop::addBackground(TRasterP ras, const TPixel32 &col)

{

	TRaster32P ras32 = ras;

	if (ras32)

		addBackground32(ras32, col);

	else

		throw TRopException("unsupported pixel type");

}

//===================================================================

//Usata tinylinetest

static void my_do_over(TRaster32P rout, const TRasterGR8P &rup)

{

	assert(rout->getSize() == rup->getSize());

	for (int y = rout->getLy(); --y >= 0;) {

		TPixel32 *out_pix = rout->pixels(y);

		TPixel32 *const out_end = out_pix + rout->getLx();

		const TPixelGR8 *up_pix = rup->pixels(y);

		for (; out_pix < out_end; ++out_pix, ++up_pix) {

			int v = up_pix->value;

			out_pix->r = out_pix->r * v / 255;

			out_pix->g = out_pix->r;

			out_pix->b = out_pix->r;

		}

	}

}

//===================================================================

void TRop::over(const TRasterP &out,

				const TRasterP &up,

				const TAffine &aff,

				ResampleFilterType filterType)

{

	out->lock();

	up->lock();

	if (filterType == ClosestPixel || filterType == Bilinear)

		::quickPut(out, up, aff, filterType);

	else {

		TRect rasterBounds = up->getBounds();

		TRectD dbounds(rasterBounds.x0, rasterBounds.y0, rasterBounds.x1 + 1, rasterBounds.y1 + 1);

		dbounds = aff * dbounds;

		TRect bounds(tfloor(dbounds.x0), tfloor(dbounds.y0), tceil(dbounds.x1) - 1, tceil(dbounds.y1) - 1);

		TRasterP tmp = up->create(bounds.getLx(), bounds.getLy());

		resample(tmp, up, TTranslation(-bounds.x0, -bounds.y0) * aff, filterType);

		over(out, tmp, bounds.getP00());

	}

	out->unlock();

	up->unlock();

}

void TRop::over(const TRasterP &out, const TRasterP &up, const TPoint &pos, const TAffine &aff,

				ResampleFilterType filterType)

{

	if (aff.isIdentity())

		//simple over with offset

		TRop::over(out, up, pos);

	else {

		TRect rasterBounds = up->getBounds();

		TRectD dbounds(rasterBounds.x0, rasterBounds.y0, rasterBounds.x1, rasterBounds.y1);

		dbounds = aff * dbounds;

		TRect bounds(tfloor(dbounds.x0), tfloor(dbounds.y0), tceil(dbounds.x1), tceil(dbounds.y1));

		TRasterP tmp = up->create(bounds.getLx(), bounds.getLy());

		resample(tmp, up, TTranslation(-dbounds.getP00()) * aff, filterType);

		TRop::over(out, tmp, pos);

	}

}

void TRop::over(TRasterP rout, const TRasterCM32P &rup, TPalette *palette, const TPoint &point, const TAffine &aff)

{

	TRaster32P app(rup->getSize());

	TRop::convert(app, rup, palette);

	TRop::over(rout, app, point, aff);

}

//===================================================================

void TRop::quickPut(const TRasterP &out,

					const TRasterP &up,

					const TAffine &aff,

					const TPixel32 &colorScale,

					bool doPremultiply, bool whiteTransp, bool firstColumn,

					bool doRasterDarkenBlendedView)

{

	::quickPut(out, up, aff, ClosestPixel, colorScale, doPremultiply, whiteTransp, firstColumn, doRasterDarkenBlendedView);

}

//===================================================================

void TRop::over(const TRasterP &out, const TRasterP &dn, const TRasterP &up, const TRasterGR8P &mask)

{

	out->lock();

	up->lock();

	dn->lock();

	TRaster32P out32 = out;

	TRaster32P dn32 = dn;

	TRaster32P up32 = up;

	if (out32 && dn32 && up32)

		do_over<tpixel32, tpixel32="" tpixel32,="">(out32, dn32, up32, mask);</tpixel32,>

	else {

		TRaster64P out64 = out;

		TRaster64P dn64 = dn;

		TRaster64P up64 = up;

		if (out64 && dn64 && up64)

			do_over<tpixel64, tpixel64="" tpixel64,="">(out64, dn64, up64, mask);</tpixel64,>

		else

			throw TRopException("unsupported pixel type");

	}

	out->unlock();

	up->unlock();

	dn->unlock();

}