// 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) override { return true; }
bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info) override {
bBox = TConsts::infiniteRectD;
return true;
}
void doCompute(TTile &tile, double frame, const TRenderSettings &) override {
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) override { return false; }
bool doGetBBox(double, TRectD &bBox, const TRenderSettings &info) override {
bBox = TConsts::infiniteRectD;
return true;
}
void doCompute(TTile &tile, double frame, const TRenderSettings &info) override {
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) override {
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")