#include "texception.h"

#include "tfxparam.h"

#include "stdfx.h"

#include "tspectrumparam.h"

#include "tparamuiconcept.h"

class DiamondGradientFx : public TStandardZeraryFx

{

	FX_PLUGIN_DECLARATION(DiamondGradientFx)

	TSpectrumParamP m_colors;

	TDoubleParamP m_size;

public:

	DiamondGradientFx()

		: m_size(100.0)

	{

		TSpectrum::ColorKey colors[] = {

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

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

			TSpectrum::ColorKey(0.4, TPixel32::Blue),

			TSpectrum::ColorKey(0.6, TPixel32::Green),

			TSpectrum::ColorKey(0.8, TPixel32::Magenta),

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

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

		m_size->setMeasureName("fxLength");

		bindParam(this, "colors", m_colors);

		bindParam(this, "size", m_size);

	}

	~DiamondGradientFx(){};

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

	{

		bBox = TConsts::infiniteRectD;

		return true;

	};

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

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

	void getParamUIs(TParamUIConcept *&concepts, int &length)

	{

		concepts = new TParamUIConcept[length = 1];

		concepts[0].m_type = TParamUIConcept::RADIUS;

		concepts[0].m_label = "Size";

		concepts[0].m_params.push_back(m_size);

	}

};

template <typename pixel=""></typename>

void doDiamondGradient(const TRasterPT<pixel> &ras, const TSpectrumT<pixel> &spectrum, TPointD &tilepos, double size)</pixel></pixel>

{

	PIXEL outpixel = spectrum.getPremultipliedValue(1.);

	int j;

	ras->lock();

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

		TPointD pos = tilepos;

		pos.y += j;

		double fy = fabs(pos.y) / size;

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

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

		while (pix < endPix) {

			double s;

			double fx = fabs(pos.x) / size;

			if ((fx * fy) < 1) {

				s = fx * fy;

				*pix++ = spectrum.getPremultipliedValue(s);

			} else

				*pix++ = outpixel;

			pos.x += 1.0;

		}

	}

	ras->unlock();

}

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

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

{

	double size = m_size->getValue(frame) * ri.m_affine.a11 / ri.m_shrinkX;

	TPointD pos = tile.m_pos;

	if (TRaster32P raster32 = tile.getRaster())

		doDiamondGradient<tpixel32>(</tpixel32>

			raster32,

			m_colors->getValue(frame),

			pos, size);

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

		doDiamondGradient<tpixel64>(</tpixel64>

			raster64,

			m_colors->getValue64(frame),

			pos, size);

	else

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

}

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

FX_PLUGIN_IDENTIFIER(DiamondGradientFx, "diamondGradientFx");