tahoma2d/toonz/sources/stdfx/diamondgradientfx.cpp
2016-06-20 11:34:15 +09:00

89 lines
3 KiB
C++

#include "texception.h"
#include "tfxparam.h"
#include "stdfx.h"
#include "tspectrumparam.h"
#include "tparamuiconcept.h"
class DiamondGradientFx : public TStandardZeraryFx {
FX_PLUGIN_DECLARATION(DiamondGradientFx)
TSpectrumParamP m_colors;
TDoubleParamP m_size;
public:
DiamondGradientFx() : m_size(100.0) {
TSpectrum::ColorKey colors[] = {TSpectrum::ColorKey(0, TPixel32::White),
TSpectrum::ColorKey(0.2, TPixel32::Yellow),
TSpectrum::ColorKey(0.4, TPixel32::Blue),
TSpectrum::ColorKey(0.6, TPixel32::Green),
TSpectrum::ColorKey(0.8, TPixel32::Magenta),
TSpectrum::ColorKey(1, TPixel32::Red)};
m_colors = TSpectrumParamP(tArrayCount(colors), colors);
m_size->setMeasureName("fxLength");
bindParam(this, "colors", m_colors);
bindParam(this, "size", m_size);
}
~DiamondGradientFx(){};
bool doGetBBox(double, TRectD &bBox, const TRenderSettings &info) override {
bBox = TConsts::infiniteRectD;
return true;
};
void doCompute(TTile &tile, double frame, const TRenderSettings &ri) override;
bool canHandle(const TRenderSettings &info, double frame) override { return false; }
void getParamUIs(TParamUIConcept *&concepts, int &length) override {
concepts = new TParamUIConcept[length = 1];
concepts[0].m_type = TParamUIConcept::RADIUS;
concepts[0].m_label = "Size";
concepts[0].m_params.push_back(m_size);
}
};
template <typename PIXEL>
void doDiamondGradient(const TRasterPT<PIXEL> &ras,
const TSpectrumT<PIXEL> &spectrum, TPointD &tilepos,
double size) {
PIXEL outpixel = spectrum.getPremultipliedValue(1.);
int j;
ras->lock();
for (j = 0; j < ras->getLy(); j++) {
TPointD pos = tilepos;
pos.y += j;
double fy = fabs(pos.y) / size;
PIXEL *pix = ras->pixels(j);
PIXEL *endPix = pix + ras->getLx();
while (pix < endPix) {
double s;
double fx = fabs(pos.x) / size;
if ((fx * fy) < 1) {
s = fx * fy;
*pix++ = spectrum.getPremultipliedValue(s);
} else
*pix++ = outpixel;
pos.x += 1.0;
}
}
ras->unlock();
}
//------------------------------------------------------------------
void DiamondGradientFx::doCompute(TTile &tile, double frame,
const TRenderSettings &ri) {
double size = m_size->getValue(frame) * ri.m_affine.a11 / ri.m_shrinkX;
TPointD pos = tile.m_pos;
if (TRaster32P raster32 = tile.getRaster())
doDiamondGradient<TPixel32>(raster32, m_colors->getValue(frame), pos, size);
else if (TRaster64P raster64 = tile.getRaster())
doDiamondGradient<TPixel64>(raster64, m_colors->getValue64(frame), pos,
size);
else
throw TException("DiamondGradientFx: unsupported Pixel Type");
}
//==============================================================================
FX_PLUGIN_IDENTIFIER(DiamondGradientFx, "diamondGradientFx");