#include "trop.h"

#include "tpixelgr.h"

namespace

{

template <typename pixtype=""></typename>

inline void do_invert(TRasterPT<pixtype> ras)</pixtype>

{

	int wrap = ras->getWrap();

	int lx = ras->getLx();

	PixType *rowIn = ras->pixels();

	PixType *lastPix = rowIn + wrap * ras->getLy();

	PixType *pixIn = 0;

	PixType *endPix = 0;

	while (pixIn < lastPix) {

		pixIn = rowIn;

		endPix = pixIn + lx;

		while (pixIn < endPix) {

			pixIn->r = pixIn->m - pixIn->r;

			pixIn->g = pixIn->m - pixIn->g;

			pixIn->b = pixIn->m - pixIn->b; /*pixIn->m = pixIn->m;*/

			++pixIn;

		}

		rowIn += wrap;

	}

}

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

template <typename pixtype=""></typename>

inline void do_invert(TRasterPT<pixtype> ras, bool invRed, bool invGreen, bool invBlue, bool invMatte)</pixtype>

{

	int wrap = ras->getWrap();

	int lx = ras->getLx();

	PixType *rowIn = ras->pixels();

	PixType *lastPix = rowIn + wrap * ras->getLy();

	PixType *pixIn = 0;

	PixType *endPix = 0;

	while (pixIn < lastPix) {

		pixIn = rowIn;

		endPix = pixIn + lx;

		while (pixIn < endPix) {

			if (invRed)

				pixIn->r = pixIn->m - pixIn->r;

			if (invGreen)

				pixIn->g = pixIn->m - pixIn->g;

			if (invBlue)

				pixIn->b = pixIn->m - pixIn->b;

			if (invMatte)

				pixIn->m = ~pixIn->m;

			++pixIn;

		}

		rowIn += wrap;

	}

}

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

template <>

inline void do_invert<tpixelgr8>(TRasterPT<tpixelgr8> ras)</tpixelgr8></tpixelgr8>

{

	int wrap = ras->getWrap();

	int lx = ras->getLx();

	TPixelGR8 *rowIn = ras->pixels();

	TPixelGR8 *lastPix = rowIn + wrap * ras->getLy();

	TPixelGR8 *pixIn = 0;

	TPixelGR8 *endPix = 0;

	while (pixIn < lastPix) {

		pixIn = rowIn;

		endPix = pixIn + lx;

		while (pixIn < endPix) {

			pixIn->value = 255 - pixIn->value;

			++pixIn;

		}

		rowIn += wrap;

	}

}

}

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

void TRop::invert(TRasterP ras, bool invRed, bool invGreen, bool invBlue, bool invMatte)

{

	if (!invRed && !invGreen && !invBlue)

		return;

	bool flag = invRed && invGreen && invBlue && !invMatte;

	TRaster32P ras32 = ras;

	ras->lock();

	if (ras32)

		if (flag)

			do_invert<tpixel32>(ras32);</tpixel32>

		else

			do_invert<tpixel32>(ras, invRed, invGreen, invBlue, invMatte);</tpixel32>

	else {

		TRaster64P ras64 = ras;

		if (ras64)

			if (flag)

				do_invert<tpixel64>(ras64);</tpixel64>

			else

				do_invert<tpixel64>(ras64, invRed, invGreen, invBlue, invMatte);</tpixel64>

		else {

			TRasterGR8P ras8 = ras;

			if (ras8)

				do_invert<tpixelgr8>(ras8);</tpixelgr8>

			else {

				ras->unlock();

				throw TRopException("unsupported pixel type");

			}

		}

	}

	ras->unlock();

}

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