|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
//#include "trop.h"
|
|
Toshihiro Shimizu |
890ddd |
#include "tfxparam.h"
|
|
Toshihiro Shimizu |
890ddd |
#include <math.h></math.h>
|
|
Toshihiro Shimizu |
890ddd |
#include "stdfx.h"
|
|
Toshihiro Shimizu |
890ddd |
#include "hsvutil.h"
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
class HSVKeyFx : public TStandardRasterFx
|
|
Toshihiro Shimizu |
890ddd |
{
|
|
Toshihiro Shimizu |
890ddd |
FX_PLUGIN_DECLARATION(HSVKeyFx)
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
TRasterFxPort m_input;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_h;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_s;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_v;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_hrange;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_srange;
|
|
Toshihiro Shimizu |
890ddd |
TDoubleParamP m_vrange;
|
|
Toshihiro Shimizu |
890ddd |
TBoolParamP m_gender;
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
public:
|
|
Toshihiro Shimizu |
890ddd |
HSVKeyFx()
|
|
Toshihiro Shimizu |
890ddd |
: m_h(0.0), m_s(0.0), m_v(0.0), m_hrange(0.0), m_srange(0.0), m_vrange(0.0), m_gender(false)
|
|
Toshihiro Shimizu |
890ddd |
{
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "h", m_h);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "s", m_s);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "v", m_v);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "h_range", m_hrange);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "s_range", m_srange);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "v_range", m_vrange);
|
|
Toshihiro Shimizu |
890ddd |
bindParam(this, "invert", m_gender);
|
|
Toshihiro Shimizu |
890ddd |
m_h->setValueRange(0.0, 360.0);
|
|
Toshihiro Shimizu |
890ddd |
m_s->setValueRange(0.0, 1.0);
|
|
Toshihiro Shimizu |
890ddd |
m_v->setValueRange(0.0, 1.0);
|
|
Toshihiro Shimizu |
890ddd |
m_hrange->setValueRange(0.0, 360.0);
|
|
Toshihiro Shimizu |
890ddd |
m_srange->setValueRange(0.0, 1.0);
|
|
Toshihiro Shimizu |
890ddd |
m_vrange->setValueRange(0.0, 1.0);
|
|
Toshihiro Shimizu |
890ddd |
addInputPort("Source", m_input);
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
~HSVKeyFx(){};
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)
|
|
Toshihiro Shimizu |
890ddd |
{
|
|
Toshihiro Shimizu |
890ddd |
if (m_input.isConnected()) {
|
|
Toshihiro Shimizu |
890ddd |
m_input->doGetBBox(frame, bBox, info);
|
|
Toshihiro Shimizu |
890ddd |
return true;
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
return false;
|
|
Toshihiro Shimizu |
890ddd |
};
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
void doCompute(TTile &tile, double frame, const TRenderSettings &);
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
bool canHandle(const TRenderSettings &info, double frame) { return true; }
|
|
Toshihiro Shimizu |
890ddd |
};
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
//------------------------------------------------------------------------------
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
template <typename pixel=""></typename>
|
|
Toshihiro Shimizu |
890ddd |
void doHSVKey(const TRasterPT<pixel> &ras, double lowH, double highH,</pixel>
|
|
Toshihiro Shimizu |
890ddd |
double lowS, double highS, double lowV, double highV, bool gender)
|
|
Toshihiro Shimizu |
890ddd |
{
|
|
Toshihiro Shimizu |
890ddd |
double aux = (double)PIXEL::maxChannelValue;
|
|
Toshihiro Shimizu |
890ddd |
int j;
|
|
Toshihiro Shimizu |
890ddd |
ras->lock();
|
|
Toshihiro Shimizu |
890ddd |
for (j = 0; j < ras->getLy(); j++) {
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
PIXEL *pix = ras->pixels(j);
|
|
Toshihiro Shimizu |
890ddd |
PIXEL *endPix = pix + ras->getLx();
|
|
Toshihiro Shimizu |
890ddd |
while (pix < endPix) {
|
|
Toshihiro Shimizu |
890ddd |
double h, s, v;
|
|
Toshihiro Shimizu |
890ddd |
OLDRGB2HSV(pix->r / aux, pix->g / aux, pix->b / aux, &h, &s, &v);
|
|
Toshihiro Shimizu |
890ddd |
bool condition = h >= lowH && h <= highH && s >= lowS && s <= highS && v >= lowV && v <= highV;
|
|
Toshihiro Shimizu |
890ddd |
if (condition != gender)
|
|
Toshihiro Shimizu |
890ddd |
*pix = PIXEL::Transparent;
|
|
Toshihiro Shimizu |
890ddd |
pix++;
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
ras->unlock();
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
//------------------------------------------------------------------------------
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
void HSVKeyFx::doCompute(TTile &tile, double frame, const TRenderSettings &ri)
|
|
Toshihiro Shimizu |
890ddd |
{
|
|
Toshihiro Shimizu |
890ddd |
if (!m_input.isConnected())
|
|
Toshihiro Shimizu |
890ddd |
return;
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
m_input->compute(tile, frame, ri);
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
double h_ref = m_h->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
double s_ref = m_s->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
double v_ref = m_v->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
double h_range = m_hrange->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
double s_range = m_srange->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
double v_range = m_vrange->getValue(frame);
|
|
Toshihiro Shimizu |
890ddd |
bool gender = (int)m_gender->getValue();
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
double lowH = tmax(0.0, h_ref - h_range);
|
|
Toshihiro Shimizu |
890ddd |
double highH = tmin(360.0, h_ref + h_range);
|
|
Toshihiro Shimizu |
890ddd |
double lowS = tmax(0.0, s_ref - s_range);
|
|
Toshihiro Shimizu |
890ddd |
double highS = tmin(1.0, s_ref + s_range);
|
|
Toshihiro Shimizu |
890ddd |
double lowV = tmax(0.0, v_ref - v_range);
|
|
Toshihiro Shimizu |
890ddd |
double highV = tmin(1.0, v_ref + v_range);
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
TRaster32P raster32 = tile.getRaster();
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
if (raster32)
|
|
Toshihiro Shimizu |
890ddd |
doHSVKey<tpixel32>(raster32, lowH, highH, lowS, highS, lowV, highV, gender);</tpixel32>
|
|
Toshihiro Shimizu |
890ddd |
else {
|
|
Toshihiro Shimizu |
890ddd |
TRaster64P raster64 = tile.getRaster();
|
|
Toshihiro Shimizu |
890ddd |
if (raster64)
|
|
Toshihiro Shimizu |
890ddd |
doHSVKey<tpixel64>(raster64, lowH, highH, lowS, highS, lowV, highV, gender);</tpixel64>
|
|
Toshihiro Shimizu |
890ddd |
else
|
|
Toshihiro Shimizu |
890ddd |
throw TException("HSVKey: unsupported Pixel Type");
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
}
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
//------------------------------------------------------------------
|
|
Toshihiro Shimizu |
890ddd |
|
|
Toshihiro Shimizu |
890ddd |
FX_PLUGIN_IDENTIFIER(HSVKeyFx, "hsvKeyFx")
|