// TnzCore includes
#include "trop.h"
#include "tpixelutils.h"

// TnzBase includes
#include "tparamset.h"
#include "tdoubleparam.h"
#include "tzeraryfx.h"
#include "tfxparam.h"
#include "tparamuiconcept.h"

#include "tbasefx.h"
#include "tzeraryfx.h"

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

class ColorCardFx : public TBaseZeraryFx
{
	FX_DECLARATION(ColorCardFx)

	TPixelParamP m_color;

public:
	ColorCardFx() : m_color(TPixel32::Green)
	{
		bindParam(this, "color", m_color);
		m_color->setDefaultValue(TPixel32::Green);
		setName(L"ColorCardFx");
	}

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

	bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)
	{
		bBox = TConsts::infiniteRectD;
		return true;
	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &)
	{
		TRaster32P raster32 = tile.getRaster();
		if (raster32)
			raster32->fill(m_color->getPremultipliedValue(frame));
		else {
			TRaster64P ras64 = tile.getRaster();
			if (ras64)
				ras64->fill(toPixel64(m_color->getPremultipliedValue(frame)));
			else
				throw TException("ColorCardFx unsupported pixel type");
		}
	};
};

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

class CheckBoardFx : public TBaseZeraryFx
{
	FX_DECLARATION(CheckBoardFx)

	TPixelParamP m_color1, m_color2;
	TDoubleParamP m_size;

public:
	CheckBoardFx() : m_color1(TPixel32::Black), m_color2(TPixel32::White), m_size(50)
	{
		m_size->setMeasureName("fxLength");
		bindParam(this, "color1", m_color1);
		bindParam(this, "color2", m_color2);
		bindParam(this, "size", m_size);
		m_color1->setDefaultValue(TPixel32::Black);
		m_color2->setDefaultValue(TPixel32::White);
		m_size->setValueRange(1, 1000);
		m_size->setDefaultValue(50);
		setName(L"CheckBoardFx");
	}

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

	bool doGetBBox(double, TRectD &bBox, const TRenderSettings &info)
	{
		bBox = TConsts::infiniteRectD;
		return true;
	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &info)
	{
		const TPixel32 &c1 = m_color1->getValue(frame);
		const TPixel32 &c2 = m_color2->getValue(frame);

		double size = m_size->getValue(frame);

		assert(info.m_shrinkX == info.m_shrinkY);
		size *= info.m_affine.a11 / info.m_shrinkX;

		TDimensionD dim(size, size);
		TRop::checkBoard(tile.getRaster(), c1, c2, dim, tile.m_pos);
	}

	void getParamUIs(TParamUIConcept *&concepts, int &length)
	{
		concepts = new TParamUIConcept[length = 1];

		concepts[0].m_type = TParamUIConcept::SIZE;
		concepts[0].m_label = "Size";
		concepts[0].m_params.push_back(m_size);
	}
};

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

FX_IDENTIFIER(ColorCardFx, "colorCardFx")
FX_IDENTIFIER(CheckBoardFx, "checkBoardFx")