200 lines
6.6 KiB
C++
200 lines
6.6 KiB
C++
|
|
|
|
#include "tfxparam.h"
|
|
#include "stdfx.h"
|
|
#include "tspectrumparam.h"
|
|
#include "tpixelutils.h"
|
|
#include "tparamuiconcept.h"
|
|
|
|
namespace {
|
|
inline void pixelConvert(TPixel32 &dst, const TPixel32 &src) { dst = src; }
|
|
inline void pixelConvert(TPixel64 &dst, const TPixel32 &src) {
|
|
dst = toPixel64(src);
|
|
}
|
|
} // namespace
|
|
|
|
class FourPointsGradientFx : public TStandardZeraryFx {
|
|
FX_PLUGIN_DECLARATION(FourPointsGradientFx)
|
|
TSpectrumParamP m_colors;
|
|
TPointParamP m_point1;
|
|
TPointParamP m_point2;
|
|
TPointParamP m_point3;
|
|
TPointParamP m_point4;
|
|
TPixelParamP m_color1;
|
|
TPixelParamP m_color2;
|
|
TPixelParamP m_color3;
|
|
TPixelParamP m_color4;
|
|
|
|
public:
|
|
FourPointsGradientFx()
|
|
: m_point1(TPointD(200.0, 200.0))
|
|
, m_point2(TPointD(-200.0, 200.0))
|
|
, m_point3(TPointD(-200.0, -200.0))
|
|
, m_point4(TPointD(200.0, -200.0))
|
|
, m_color1(TPixel32::Red)
|
|
, m_color2(TPixel32::Green)
|
|
, m_color3(TPixel32::Blue)
|
|
, m_color4(TPixel32::Yellow) {
|
|
m_point1->getX()->setMeasureName("fxLength");
|
|
m_point1->getY()->setMeasureName("fxLength");
|
|
m_point2->getX()->setMeasureName("fxLength");
|
|
m_point2->getY()->setMeasureName("fxLength");
|
|
m_point3->getX()->setMeasureName("fxLength");
|
|
m_point3->getY()->setMeasureName("fxLength");
|
|
m_point4->getX()->setMeasureName("fxLength");
|
|
m_point4->getY()->setMeasureName("fxLength");
|
|
/*
|
|
bindParam(this,"point1", m_point1);
|
|
bindParam(this,"point2", m_point2);
|
|
bindParam(this,"point3", m_point3);
|
|
bindParam(this,"point4", m_point4);
|
|
bindParam(this,"color_1", m_color1);
|
|
bindParam(this,"color_2", m_color2);
|
|
bindParam(this,"color_3", m_color3);
|
|
bindParam(this,"color_4", m_color4);
|
|
*/
|
|
bindParam(this, "Point_1", m_point1);
|
|
bindParam(this, "Color_1", m_color1);
|
|
|
|
bindParam(this, "Point_2", m_point2);
|
|
bindParam(this, "Color_2", m_color2);
|
|
|
|
bindParam(this, "Point_3", m_point3);
|
|
bindParam(this, "Color_3", m_color3);
|
|
|
|
bindParam(this, "Point_4", m_point4);
|
|
bindParam(this, "Color_4", m_color4);
|
|
}
|
|
~FourPointsGradientFx(){};
|
|
|
|
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 true; }
|
|
|
|
void getParamUIs(TParamUIConcept *&concepts, int &length) {
|
|
concepts = new TParamUIConcept[length = 4];
|
|
|
|
concepts[0].m_type = TParamUIConcept::POINT;
|
|
concepts[0].m_label = "Point 1";
|
|
concepts[0].m_params.push_back(m_point1);
|
|
|
|
concepts[1].m_type = TParamUIConcept::POINT;
|
|
concepts[1].m_label = "Point 2";
|
|
concepts[1].m_params.push_back(m_point2);
|
|
|
|
concepts[2].m_type = TParamUIConcept::POINT;
|
|
concepts[2].m_label = "Point 3";
|
|
concepts[2].m_params.push_back(m_point3);
|
|
|
|
concepts[3].m_type = TParamUIConcept::POINT;
|
|
concepts[3].m_label = "Point 4";
|
|
concepts[3].m_params.push_back(m_point4);
|
|
}
|
|
};
|
|
|
|
template <typename PIXEL, typename CHANNEL_TYPE>
|
|
void doFourPointsGradient(const TRasterPT<PIXEL> &ras, TPointD tilepos,
|
|
TPointD pos1, TPointD pos2, TPointD pos3,
|
|
TPointD pos4, TPixel32 ccol1, TPixel32 ccol2,
|
|
TPixel32 ccol3, TPixel32 ccol4) {
|
|
PIXEL col1, col2, col3, col4;
|
|
pixelConvert(col1, ccol1);
|
|
pixelConvert(col2, ccol2);
|
|
pixelConvert(col3, ccol3);
|
|
pixelConvert(col4, ccol4);
|
|
|
|
int j;
|
|
ras->lock();
|
|
for (j = 0; j < ras->getLy(); j++) {
|
|
TPointD pos = tilepos;
|
|
pos.y += j;
|
|
PIXEL *pix = ras->pixels(j);
|
|
PIXEL *endPix = pix + ras->getLx();
|
|
while (pix < endPix) {
|
|
double d1 = sqrt((pos1.x - pos.x) * (pos1.x - pos.x) +
|
|
(pos1.y - pos.y) * (pos1.y - pos.y));
|
|
if (!d1) {
|
|
*pix = col1;
|
|
pos.x += 1;
|
|
++pix;
|
|
continue;
|
|
}
|
|
double d2 = sqrt((pos2.x - pos.x) * (pos2.x - pos.x) +
|
|
(pos2.y - pos.y) * (pos2.y - pos.y));
|
|
if (!d2) {
|
|
*pix = col2;
|
|
pos.x += 1;
|
|
++pix;
|
|
continue;
|
|
}
|
|
double d3 = sqrt((pos3.x - pos.x) * (pos3.x - pos.x) +
|
|
(pos3.y - pos.y) * (pos3.y - pos.y));
|
|
if (!d3) {
|
|
*pix = col3;
|
|
pos.x += 1;
|
|
++pix;
|
|
continue;
|
|
}
|
|
double d4 = sqrt((pos4.x - pos.x) * (pos4.x - pos.x) +
|
|
(pos4.y - pos.y) * (pos4.y - pos.y));
|
|
if (!d4) {
|
|
*pix = col4;
|
|
pos.x += 1;
|
|
++pix;
|
|
continue;
|
|
}
|
|
double dtotal = 1 / d1 + 1 / d2 + 1 / d3 + 1 / d4;
|
|
pix->r = (CHANNEL_TYPE)(
|
|
((1 / d1) / dtotal) * col1.r + ((1 / d2) / dtotal) * col2.r +
|
|
((1 / d3) / dtotal) * col3.r + ((1 / d4) / dtotal) * col4.r);
|
|
pix->g = (CHANNEL_TYPE)(
|
|
((1 / d1) / dtotal) * col1.g + ((1 / d2) / dtotal) * col2.g +
|
|
((1 / d3) / dtotal) * col3.g + ((1 / d4) / dtotal) * col4.g);
|
|
pix->b = (CHANNEL_TYPE)(
|
|
((1 / d1) / dtotal) * col1.b + ((1 / d2) / dtotal) * col2.b +
|
|
((1 / d3) / dtotal) * col3.b + ((1 / d4) / dtotal) * col4.b);
|
|
pix->m = (CHANNEL_TYPE)(
|
|
((1 / d1) / dtotal) * col1.m + ((1 / d2) / dtotal) * col2.m +
|
|
((1 / d3) / dtotal) * col3.m + ((1 / d4) / dtotal) * col4.m);
|
|
pos.x += 1.0;
|
|
++pix;
|
|
}
|
|
}
|
|
ras->unlock();
|
|
}
|
|
|
|
void FourPointsGradientFx::doCompute(TTile &tile, double frame,
|
|
const TRenderSettings &ri) {
|
|
TPointD pos1 = m_point1->getValue(frame) * (1.0 / ri.m_shrinkX);
|
|
TPointD pos2 = m_point2->getValue(frame) * (1.0 / ri.m_shrinkX);
|
|
TPointD pos3 = m_point3->getValue(frame) * (1.0 / ri.m_shrinkX);
|
|
TPointD pos4 = m_point4->getValue(frame) * (1.0 / ri.m_shrinkX);
|
|
pos1 = ri.m_affine * pos1;
|
|
pos2 = ri.m_affine * pos2;
|
|
pos3 = ri.m_affine * pos3;
|
|
pos4 = ri.m_affine * pos4;
|
|
TPixel32 col1 = m_color1->getPremultipliedValue(frame);
|
|
TPixel32 col2 = m_color2->getPremultipliedValue(frame);
|
|
TPixel32 col3 = m_color3->getPremultipliedValue(frame);
|
|
TPixel32 col4 = m_color4->getPremultipliedValue(frame);
|
|
|
|
TPointD pos = tile.m_pos;
|
|
TRaster32P raster32 = tile.getRaster();
|
|
|
|
if (raster32)
|
|
doFourPointsGradient<TPixel32, UCHAR>(raster32, pos, pos1, pos2, pos3, pos4,
|
|
col1, col2, col3, col4);
|
|
else {
|
|
TRaster64P raster64 = tile.getRaster();
|
|
if (raster64)
|
|
doFourPointsGradient<TPixel64, USHORT>(raster64, pos, pos1, pos2, pos3,
|
|
pos4, col1, col2, col3, col4);
|
|
else
|
|
throw TException("Brightness&Contrast: unsupported Pixel Type");
|
|
}
|
|
}
|
|
|
|
FX_PLUGIN_IDENTIFIER(FourPointsGradientFx, "fourPointsGradientFx");
|