tahoma2d/toonz/sources/stdfx/despecklefx.cpp

82 lines
1.9 KiB
C++
Raw Normal View History

2016-03-19 06:57:51 +13:00
// 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")