#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 PIXEL, typename PIXELGRAY, typename CHANNEL_TYPE>
void doMultiTone(const TRasterPT<PIXEL> &ras, const TSpectrumT<PIXEL> &spectrum)
{
	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>(
			raster32, m_colors->getValue(frame));
	else if (TRaster64P raster64 = tile.getRaster())
		doMultiTone<TPixel64, TPixelGR16, USHORT>(
			raster64, m_colors->getValue64(frame));
	else
		throw TException("MultiToneFx: unsupported Pixel Type");
}

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

FX_PLUGIN_IDENTIFIER(MultiToneFx, "multiToneFx")