// TnzCore includes

#include "trop.h"

// TnzBase includes

#include "tfxparam.h"

// TnzStdfx includes

#include "stdfx.h"

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

class DespeckleFx : public TStandardRasterFx

{

	FX_PLUGIN_DECLARATION(DespeckleFx)

	TRasterFxPort m_input;

	TIntParamP m_size;

	TIntEnumParamP m_transparenceType;

public:

	DespeckleFx()

		: m_size(1), m_transparenceType(new TIntEnumParam(0, "Transparent Bg"))

	{

		bindParam(this, "size", m_size);

		bindParam(this, "detect_speckles_on", m_transparenceType);

		m_transparenceType->addItem(1, "White Bg");

		addInputPort("Source", m_input);

		m_size->setValueRange(1, 1000);

	}

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

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

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

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

};

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

bool DespeckleFx::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 DespeckleFx::doCompute(TTile &tile, double frame, const TRenderSettings &ri)

{

	if (!m_input.isConnected())

		return;

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

	int size = m_size->getValue();

	TRop::despeckle(tile.getRaster(), size, false, m_transparenceType->getValue() == 1);

}

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

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

{

	if (!m_input.isConnected())

		return;

	m_input->dryCompute(rect, frame, info);

}

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

FX_PLUGIN_IDENTIFIER(DespeckleFx, "despeckleFx")