#include "texception.h"

#include "tfxparam.h"

#include "trop.h"

#include "stdfx.h"

#include "trasterfx.h"

#include "tparamset.h"

#include "tparamuiconcept.h"

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

class BaseRaylitFx : public TStandardRasterFx

{

protected:

	TRasterFxPort m_input;

	TPointParamP m_p;

	TDoubleParamP m_z;

	TDoubleParamP m_intensity;

	TDoubleParamP m_decay;

	TDoubleParamP m_smoothness;

	TBoolParamP m_includeInput;

public:

	BaseRaylitFx()

		: m_p(TPointD(0, 0)), m_z(300.0), m_intensity(60), m_decay(1.0), m_smoothness(100), m_includeInput(false)

	{

		m_p->getX()->setMeasureName("fxLength");

		m_p->getY()->setMeasureName("fxLength");

		bindParam(this, "p", m_p);

		bindParam(this, "z", m_z);

		bindParam(this, "intensity", m_intensity);

		bindParam(this, "decay", m_decay);

		bindParam(this, "smoothness", m_smoothness);

		bindParam(this, "includeInput", m_includeInput);

		addInputPort("Source", m_input);

	}

	~BaseRaylitFx() {}

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

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

	int getMemoryRequirement(const TRectD &rect, double frame, const TRenderSettings &info);

	void doDryCompute(TRectD &rect, double frame, const TRenderSettings &info);

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

	{

		concepts = new TParamUIConcept[length = 1];

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

		concepts[0].m_label = "Center";

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

	}

};

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

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

{

	if (m_input.isConnected()) {

		bool ret = m_input->doGetBBox(frame, bBox, info);

		if (ret)

			bBox = TConsts::infiniteRectD;

		return ret;

	} else {

		bBox = TRectD();

		return false;

	}

};

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

void BaseRaylitFx::doDryCompute(TRectD &rect,

								double frame,

								const TRenderSettings &ri)

{

	if (!m_input.isConnected())

		return;

	TRectD bboxIn;

	m_input->getBBox(frame, bboxIn, ri);

	if (bboxIn == TConsts::infiniteRectD)

		bboxIn = rect;

	TDimension sizeIn(std::max(tceil(bboxIn.getLx()), 1), std::max(tceil(bboxIn.getLy()), 1));

	bboxIn = TRectD(bboxIn.getP00(), TDimensionD(sizeIn.lx, sizeIn.ly));

	m_input->dryCompute(bboxIn, frame, ri);

}

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

int BaseRaylitFx::getMemoryRequirement(const TRectD &rect, double frame, const TRenderSettings &info)

{

	TRectD bboxIn;

	m_input->getBBox(frame, bboxIn, info);

	if (bboxIn == TConsts::infiniteRectD)

		return -1;

	if (bboxIn.isEmpty())

		return 0;

	return TRasterFx::memorySize(bboxIn, info.m_bpp);

}

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

class RaylitFx : public BaseRaylitFx

{

	FX_PLUGIN_DECLARATION(RaylitFx)

protected:

	TPixelParamP m_color;

	TBoolParamP m_invert;

public:

	RaylitFx()

		: m_color(TPixel(255, 80, 0)), m_invert(false)

	{

		bindParam(this, "color", m_color);

		bindParam(this, "invert", m_invert);

	}

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

};

FX_PLUGIN_IDENTIFIER(RaylitFx, "raylitFx")

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

void RaylitFx::doCompute(TTile &tileOut, double frame, const TRenderSettings &ri)

{

	if (!m_input.isConnected())

		return;

	double scale = sqrt(fabs(ri.m_affine.det()));

	int shrink = (ri.m_shrinkX + ri.m_shrinkY) / 2;

	TPointD p(ri.m_affine * m_p->getValue(frame));

	TRectD rectIn(tileOut.m_pos, TDimensionD(tileOut.getRaster()->getLx(), tileOut.getRaster()->getLy()));

	TRectD bboxIn;

	m_input->getBBox(frame, bboxIn, ri);

	if (bboxIn == TConsts::infiniteRectD)

		bboxIn = rectIn;

	if (!bboxIn.isEmpty()) {

		TPoint posIn, posOut;

		TTile tileIn;

		TDimension sizeIn(std::max(tceil(bboxIn.getLx()), 1), std::max(tceil(bboxIn.getLy()), 1));

		m_input->allocateAndCompute(tileIn, bboxIn.getP00(), sizeIn,

									tileOut.getRaster(), frame, ri);

		TRop::RaylitParams params;

		params.m_scale = scale;

		params.m_lightOriginSrc.x = params.m_lightOriginDst.x = p.x;

		params.m_lightOriginSrc.y = params.m_lightOriginDst.y = p.y;

		params.m_lightOriginSrc.z = params.m_lightOriginDst.z = (int)m_z->getValue(frame);

		params.m_color = m_color->getValue(frame);

		params.m_intensity = m_intensity->getValue(frame);

		params.m_decay = m_decay->getValue(frame);

		params.m_smoothness = m_smoothness->getValue(frame);

		params.m_invert = m_invert->getValue();

		params.m_includeInput = m_includeInput->getValue();

		params.m_lightOriginSrc.x -= (int)tileIn.m_pos.x;

		params.m_lightOriginSrc.y -= (int)tileIn.m_pos.y;

		params.m_lightOriginDst.x -= (int)tileOut.m_pos.x;

		params.m_lightOriginDst.y -= (int)tileOut.m_pos.y;

		TRop::raylit(tileOut.getRaster(), tileIn.getRaster(), params);

	}

}

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

class ColorRaylitFx : public BaseRaylitFx

{

	FX_PLUGIN_DECLARATION(ColorRaylitFx)

public:

	ColorRaylitFx() : BaseRaylitFx() {}

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

};

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

void ColorRaylitFx::doCompute(TTile &tileOut, double frame, const TRenderSettings &ri)

{

	if (!m_input.isConnected())

		return;

	double scale = sqrt(fabs(ri.m_affine.det()));

	int shrink = (ri.m_shrinkX + ri.m_shrinkY) / 2;

	TPointD p(ri.m_affine * m_p->getValue(frame));

	TRectD rectIn(tileOut.m_pos, TDimensionD(tileOut.getRaster()->getLx(), tileOut.getRaster()->getLy()));

	TRectD bboxIn;

	m_input->getBBox(frame, bboxIn, ri);

	if (bboxIn == TConsts::infiniteRectD)

		bboxIn = rectIn;

	if (!bboxIn.isEmpty()) {

		TPoint posIn, posOut;

		TTile tileIn;

		TDimension sizeIn(std::max(tceil(bboxIn.getLx()), 1), std::max(tceil(bboxIn.getLy()), 1));

		m_input->allocateAndCompute(tileIn, bboxIn.getP00(), sizeIn,

									tileOut.getRaster(), frame, ri);

		TRop::RaylitParams params;

		params.m_scale = scale;

		params.m_lightOriginSrc.x = params.m_lightOriginDst.x = p.x;

		params.m_lightOriginSrc.y = params.m_lightOriginDst.y = p.y;

		params.m_lightOriginSrc.z = params.m_lightOriginDst.z = (int)m_z->getValue(frame);

		params.m_intensity = m_intensity->getValue(frame);

		params.m_decay = m_decay->getValue(frame);

		params.m_smoothness = m_smoothness->getValue(frame);

		params.m_includeInput = m_includeInput->getValue();

		params.m_lightOriginSrc.x -= (int)tileIn.m_pos.x;

		params.m_lightOriginSrc.y -= (int)tileIn.m_pos.y;

		params.m_lightOriginDst.x -= (int)tileOut.m_pos.x;

		params.m_lightOriginDst.y -= (int)tileOut.m_pos.y;

		TRop::glassRaylit(tileOut.getRaster(), tileIn.getRaster(), params);

	}

}

FX_PLUGIN_IDENTIFIER(ColorRaylitFx, "colorRaylitFx")