tahoma2d/toonz/sources/stdfx/artcontourfx.cpp

#ifdef WIN32
#pragma warning(disable : 4996)
#define NOMINMAX
#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 string &controllerAlias)
	{
		int argc = 12;
		const char *argv[12];
		argv[0] = strsave(toString(m_colorIndex->getValue()).c_str());
		getValues(argv, argc, frame);

		SandorFxRenderData *artContourData = new SandorFxRenderData(ArtAtContour, argc, argv,
																	0, shrink, controlBox);
		ArtAtContourParams &params = 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)
	{
		string app;
		for (int i = 1; i <= 11; i++) {
			app = toString(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);
	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);
	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);
}