101 lines
2.7 KiB
C++
101 lines
2.7 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)
|
||
|
{
|
||
|
bBox = TConsts::infiniteRectD;
|
||
|
return true;
|
||
|
};
|
||
|
void doCompute(TTile &tile, double frame, const TRenderSettings &ri);
|
||
|
bool canHandle(const TRenderSettings &info, double frame) { return false; }
|
||
|
|
||
|
void getParamUIs(TParamUIConcept *&concepts, int &length)
|
||
|
{
|
||
|
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");
|