tahoma2d/toonz/sources/stdfx/despecklefx.cpp
2021-10-07 12:17:57 -04:00

84 lines
2.2 KiB
C++

// TnzCore includes
#include "trop.h"
// TnzBase includes
#include "tfxparam.h"
// TnzStdfx includes
#include "stdfx.h"
#include "globalcontrollablefx.h"
//--------------------------------------------------------------------------
class DespeckleFx final : public GlobalControllableFx {
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) override;
void doCompute(TTile &tile, double frame, const TRenderSettings &) override;
void doDryCompute(TRectD &rect, double frame,
const TRenderSettings &info) override;
bool canHandle(const TRenderSettings &info, double frame) override {
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")