#include "texception.h"

//#include "tfxparam.h"

#include "stdfx.h"

namespace

{

TPixel32 unmultiply(const TPixel32 &pix)

{

	if (!pix.m)

		return pix;

	double depremult = 255.0 / pix.m;

	return TPixel32((UCHAR)(pix.r * depremult),

					(UCHAR)(pix.g * depremult),

					(UCHAR)(pix.b * depremult),

					pix.m);

}

}

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

class UnmultiplyFx : public TStandardRasterFx

{

	FX_PLUGIN_DECLARATION(UnmultiplyFx)

	TRasterFxPort m_input;

public:

	UnmultiplyFx() { addInputPort("Source", m_input); }

	~UnmultiplyFx(){};

	bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)

	{

		if (m_input.isConnected())

			return m_input->doGetBBox(frame, bBox, info);

		else {

			bBox = TRectD();

			return false;

		}

	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &ri);

	bool canHandle(const TRenderSettings &info, double frame) { return true; }

};

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

void UnmultiplyFx::doCompute(TTile &tile, double frame,

							 const TRenderSettings &ri)

{

	if (!m_input.isConnected())

		return;

	m_input->compute(tile, frame, ri);

	TRaster32P raster32 = tile.getRaster();

	assert(raster32); // per ora gestisco solo i Raster32

	int j;

	raster32->lock();

	for (j = 0; j < raster32->getLy(); j++) {

		TPixel32 *pix = raster32->pixels(j);

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

		while (pix < endPix) {

			if (pix->m) {

				double depremult = 255.0 / pix->m;

				pix->r = (UCHAR)(pix->r * depremult);

				pix->g = (UCHAR)(pix->g * depremult);

				pix->b = (UCHAR)(pix->b * depremult);

			}

			//*pix=unmultiply(*pix);

			pix++;

		}

	}

	raster32->unlock();

}

FX_PLUGIN_IDENTIFIER(UnmultiplyFx, "unmultiplyFx");