#include "stdfx.h"

#include "tfxparam.h"

#include "tspectrumparam.h"

class MultiToneFx : public TStandardRasterFx

{

	FX_PLUGIN_DECLARATION(MultiToneFx)

	TRasterFxPort m_input;

	TSpectrumParamP m_colors;

public:

	MultiToneFx()

	{

		TSpectrum::ColorKey colors[] = {

			TSpectrum::ColorKey(0, TPixel32::White),

			TSpectrum::ColorKey(0.5, TPixel32::Yellow),

			TSpectrum::ColorKey(1, TPixel32::Red)};

		m_colors = TSpectrumParamP(tArrayCount(colors), colors);

		bool ret = m_colors->isKeyframe(0);

		bindParam(this, "colors", m_colors);

		ret = getParams()->getParam(0)->isKeyframe(0);

		addInputPort("Source", m_input);

	}

	~MultiToneFx(){};

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

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

};

template <typename channel_type="" pixel,="" pixelgray,="" typename=""></typename>

void doMultiTone(const TRasterPT<pixel> &ras, const TSpectrumT<pixel> &spectrum)</pixel></pixel>

{

	double aux = (double)PIXEL::maxChannelValue;

	int j;

	ras->lock();

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

		PIXEL *pix = ras->pixels(j);

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

		while (pix < endPix) {

			if (pix->m) {

				CHANNEL_TYPE reference;

				double s;

				reference = PIXELGRAY::from(*pix).value;

				s = reference / aux;

				if (pix->m == aux) {

					*pix = spectrum.getPremultipliedValue(s);

				} else {

					PIXEL tmp = spectrum.getPremultipliedValue(s);

					s = pix->m / aux;

					tmp.r = (CHANNEL_TYPE)(tmp.r * s);

					tmp.g = (CHANNEL_TYPE)(tmp.g * s);

					tmp.b = (CHANNEL_TYPE)(tmp.b * s);

					tmp.m = (CHANNEL_TYPE)(tmp.m * s);

					*pix = tmp;

				}

			}

			pix++;

		}

	}

	ras->unlock();

}

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

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

{

	if (!m_input.isConnected())

		return;

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

	if (TRaster32P raster32 = tile.getRaster())

		doMultiTone<tpixel32, tpixelgr8,="" uchar="">(</tpixel32,>

			raster32, m_colors->getValue(frame));

	else if (TRaster64P raster64 = tile.getRaster())

		doMultiTone<tpixel64, tpixelgr16,="" ushort="">(</tpixel64,>

			raster64, m_colors->getValue64(frame));

	else

		throw TException("MultiToneFx: unsupported Pixel Type");

}

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

FX_PLUGIN_IDENTIFIER(MultiToneFx, "multiToneFx")