245 lines
8.4 KiB
C++
245 lines
8.4 KiB
C++
#ifdef _WIN32
|
|
#pragma warning(disable : 4996)
|
|
#endif
|
|
|
|
#include "ttzpimagefx.h"
|
|
#include "texception.h"
|
|
#include "tfxparam.h"
|
|
#include "stdfx.h"
|
|
#include "trasterfx.h"
|
|
#include "tspectrumparam.h"
|
|
|
|
class ArtContourFx : public TStandardRasterFx {
|
|
FX_PLUGIN_DECLARATION(ArtContourFx)
|
|
|
|
TRasterFxPort m_input;
|
|
TRasterFxPort m_controller;
|
|
TStringParamP m_colorIndex;
|
|
TBoolParamP m_keepColor;
|
|
TBoolParamP m_keepLine;
|
|
TBoolParamP m_includeAlpha;
|
|
TDoubleParamP m_density;
|
|
TRangeParamP m_distance;
|
|
TBoolParamP m_randomness;
|
|
TRangeParamP m_orientation;
|
|
TRangeParamP m_size;
|
|
|
|
public:
|
|
ArtContourFx()
|
|
: m_colorIndex(L"1,2,3")
|
|
, m_keepColor(false)
|
|
, m_keepLine(false)
|
|
, m_includeAlpha(true)
|
|
, m_density(0.0)
|
|
, m_distance(DoublePair(30.0, 30.0))
|
|
, m_randomness(true)
|
|
, m_orientation(DoublePair(0.0, 180.0))
|
|
, m_size(DoublePair(30.0, 30.0)) {
|
|
bindParam(this, "Color_Index", m_colorIndex);
|
|
bindParam(this, "Keep_color", m_keepColor);
|
|
bindParam(this, "Keep_Line", m_keepLine);
|
|
bindParam(this, "Include_Alpha", m_includeAlpha);
|
|
bindParam(this, "Density", m_density);
|
|
bindParam(this, "Distance", m_distance);
|
|
bindParam(this, "Randomness", m_randomness);
|
|
bindParam(this, "Orientation", m_orientation);
|
|
bindParam(this, "Size", m_size);
|
|
addInputPort("Source", m_input);
|
|
addInputPort("Controller", m_controller);
|
|
m_density->setValueRange(0.0, 100.0);
|
|
m_distance->getMin()->setValueRange(0.0, 1000.0);
|
|
m_distance->getMax()->setValueRange(0.0, 1000.0);
|
|
m_orientation->getMin()->setValueRange(-180.0, 180.0);
|
|
m_orientation->getMax()->setValueRange(-180.0, 180.0);
|
|
m_orientation->getMin()->setMeasureName("angle");
|
|
m_orientation->getMax()->setMeasureName("angle");
|
|
m_size->getMin()->setValueRange(0.0, 1000.0);
|
|
m_size->getMax()->setValueRange(0.0, 1000.0);
|
|
}
|
|
|
|
~ArtContourFx() {}
|
|
|
|
//----------------------------------------------------------------------------
|
|
|
|
SandorFxRenderData *buildRenderData(double frame, int shrink,
|
|
const TRectD &controlBox,
|
|
const std::string &controllerAlias) {
|
|
int argc = 12;
|
|
const char *argv[12];
|
|
argv[0] = strsave(::to_string(m_colorIndex->getValue()).c_str());
|
|
getValues(argv, argc, frame);
|
|
|
|
SandorFxRenderData *artContourData =
|
|
new SandorFxRenderData(ArtAtContour, argc, argv, 0, shrink, controlBox);
|
|
ArtAtContourParams ¶ms = artContourData->m_contourParams;
|
|
params.m_density = m_density->getValue(frame) / 100;
|
|
params.m_colorIndex = m_colorIndex->getValue();
|
|
params.m_keepLine = m_keepLine->getValue();
|
|
params.m_includeAlpha = m_includeAlpha->getValue();
|
|
params.m_maxOrientation = m_orientation->getValue(frame).second;
|
|
params.m_maxDistance = m_distance->getValue(frame).second / 10;
|
|
params.m_maxSize = m_size->getValue(frame).second / 100;
|
|
params.m_minOrientation = m_orientation->getValue(frame).first;
|
|
params.m_minDistance = m_distance->getValue(frame).first / 10;
|
|
params.m_minSize = m_size->getValue(frame).first / 100;
|
|
params.m_randomness = m_randomness->getValue();
|
|
params.m_keepColor = m_keepColor->getValue();
|
|
artContourData->m_controllerAlias = controllerAlias;
|
|
|
|
return artContourData;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
|
|
bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &ri) {
|
|
if (m_input.isConnected() && m_controller.isConnected()) {
|
|
TRectD controlBox, inputBox;
|
|
|
|
TRenderSettings ri2(ri);
|
|
ri2.m_affine = TAffine();
|
|
|
|
m_controller->getBBox(frame, controlBox, ri2);
|
|
|
|
TRenderSettings ri3(ri);
|
|
|
|
int shrink = tround((ri.m_shrinkX + ri.m_shrinkY) / 2.0);
|
|
// Should be there no need for the alias...
|
|
SandorFxRenderData *artContourData =
|
|
buildRenderData(frame, shrink, controlBox, "");
|
|
ri3.m_data.push_back(artContourData);
|
|
|
|
return m_input->doGetBBox(frame, bBox, ri3);
|
|
} else if (m_input.isConnected()) {
|
|
m_input->doGetBBox(frame, bBox, ri);
|
|
return false;
|
|
}
|
|
bBox = TRectD();
|
|
return false;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool canHandle(const TRenderSettings &info, double frame) { return true; }
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool allowUserCacheOnPort(int port) { return port != 0; }
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void doDryCompute(TRectD &rect, double frame, const TRenderSettings &ri);
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void doCompute(TTile &tile, double frame, const TRenderSettings &ri);
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
private:
|
|
void getValues(const char *argv[], int argc, double frame) {
|
|
double values[12];
|
|
values[1] = m_size->getValue(frame).second / 100;
|
|
values[2] = m_size->getValue(frame).first / 100;
|
|
values[3] = m_orientation->getValue(frame).second;
|
|
values[4] = m_orientation->getValue(frame).first;
|
|
values[5] = m_randomness->getValue() ? 1.0 : 0.0;
|
|
values[6] = m_distance->getValue(frame).second / 10;
|
|
values[7] = m_distance->getValue(frame).first / 10;
|
|
values[8] = m_density->getValue(frame) / 100;
|
|
values[9] = m_keepLine->getValue() ? 1.0 : 0.0;
|
|
values[10] = m_keepColor->getValue() ? 1.0 : 0.0;
|
|
values[11] = m_includeAlpha->getValue() ? 1.0 : 0.0;
|
|
convertParam(values, argv, argc);
|
|
}
|
|
//----------------------------------------------------------------------
|
|
|
|
char *strsave(const char *t) {
|
|
char *s;
|
|
s = (char *)malloc(strlen(t) + 1);
|
|
strcpy(s, t);
|
|
return s;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------
|
|
|
|
void convertParam(double param[], const char *cParam[], int cParamLen) {
|
|
std::string app;
|
|
for (int i = 1; i <= 11; i++) {
|
|
app = std::to_string(param[i]);
|
|
cParam[i] = strsave(app.c_str());
|
|
}
|
|
}
|
|
};
|
|
|
|
FX_PLUGIN_IDENTIFIER(ArtContourFx, "artContourFx");
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void ArtContourFx::doDryCompute(TRectD &rect, double frame,
|
|
const TRenderSettings &ri) {
|
|
if (!m_input.isConnected()) return;
|
|
if (!m_controller.isConnected()) return;
|
|
|
|
TRenderSettings ri2(ri);
|
|
ri2.m_affine = TAffine();
|
|
|
|
TRectD controlBox;
|
|
m_controller->getBBox(frame, controlBox, ri2);
|
|
|
|
TDimension dim = convert(controlBox).getSize();
|
|
TRectD controlRect(controlBox.getP00(), TDimensionD(dim.lx, dim.ly));
|
|
|
|
m_controller->dryCompute(controlRect, frame, ri2);
|
|
|
|
TRenderSettings ri3(ri);
|
|
|
|
int shrink = tround((ri.m_shrinkX + ri.m_shrinkY) / 2.0);
|
|
std::string controlAlias = m_controller->getAlias(frame, ri2);
|
|
SandorFxRenderData *artContourData =
|
|
buildRenderData(frame, shrink, controlBox, controlAlias);
|
|
ri3.m_data.push_back(artContourData);
|
|
ri3.m_userCachable = false;
|
|
|
|
m_input->dryCompute(rect, frame, ri3);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void ArtContourFx::doCompute(TTile &tile, double frame,
|
|
const TRenderSettings &ri) {
|
|
if (!m_input.isConnected()) return;
|
|
|
|
if (!m_controller.isConnected()) {
|
|
m_input->compute(tile, frame, ri);
|
|
return;
|
|
}
|
|
|
|
TRenderSettings ri2(ri);
|
|
ri2.m_affine = TAffine();
|
|
|
|
TRectD controlBox;
|
|
m_controller->getBBox(frame, controlBox, ri2);
|
|
TTile ctrTile;
|
|
ctrTile.m_pos = controlBox.getP00();
|
|
TDimension dim = convert(controlBox).getSize();
|
|
|
|
m_controller->allocateAndCompute(ctrTile, ctrTile.m_pos, dim,
|
|
tile.getRaster(), frame, ri2);
|
|
|
|
TRenderSettings ri3(ri);
|
|
|
|
// Build the render data
|
|
int shrink = tround((ri.m_shrinkX + ri.m_shrinkY) / 2.0);
|
|
std::string controlAlias = m_controller->getAlias(frame, ri2);
|
|
SandorFxRenderData *artContourData =
|
|
buildRenderData(frame, shrink, controlBox, controlAlias);
|
|
|
|
// Add the controller raster
|
|
artContourData->m_controller = ctrTile.getRaster();
|
|
|
|
// Push the data among the others
|
|
ri3.m_data.push_back(artContourData);
|
|
ri3.m_userCachable = false;
|
|
|
|
m_input->compute(tile, frame, ri3);
|
|
}
|