#include "stdfx.h"

#ifdef _WIN32
#pragma warning(disable : 4996)
#include "windows.h"
#endif
#include <vector>

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <sys/types.h>
#include <sys/timeb.h>
#include <string.h>
#include <math.h>

#include "tfxparam.h"
#include "tdoubleparam.h"
#include "ttzpimagefx.h"

//===================================================================

class CalligraphicFx : public TStandardRasterFx
{
	FX_PLUGIN_DECLARATION(CalligraphicFx)
	TRasterFxPort m_input;
	TStringParamP m_colorIndex;
	TDoubleParamP m_thickness;
	TDoubleParamP m_horizontal;
	TDoubleParamP m_upWDiagonal;
	TDoubleParamP m_vertical;
	TDoubleParamP m_doWDiagonal;
	TDoubleParamP m_accuracy;
	TDoubleParamP m_noise;

public:
	CalligraphicFx()
		: m_colorIndex(L"1,2,3"), m_thickness(5.0), m_horizontal(100.0), m_upWDiagonal(50.0), m_vertical(0.0), m_doWDiagonal(50.0), m_accuracy(50.0), m_noise(0.0)
	{
		m_thickness->setMeasureName("fxLength");
		addInputPort("Source", m_input);
		bindParam(this, "Color_Index", m_colorIndex);
		bindParam(this, "Thickness", m_thickness);
		bindParam(this, "Accuracy", m_accuracy);
		bindParam(this, "Noise", m_noise);
		bindParam(this, "Horizontal", m_horizontal);
		bindParam(this, "upWDiagonal", m_upWDiagonal);
		bindParam(this, "Vertical", m_vertical);
		bindParam(this, "doWDiagonal", m_doWDiagonal);
		m_thickness->setValueRange(0.0, 60.0);
		m_horizontal->setValueRange(0.0, 100.0);
		m_upWDiagonal->setValueRange(0.0, 100.0);
		m_vertical->setValueRange(0.0, 100.0);
		m_doWDiagonal->setValueRange(0.0, 100.0);
		m_accuracy->setValueRange(0.0, 100.0);
		m_noise->setValueRange(0.0, 100.0);
	}

	~CalligraphicFx(){};

	bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)
	{
		if (m_input.isConnected()) {
			TRenderSettings info2(info);

			int shrink = tround((info.m_shrinkX + info.m_shrinkY) / 2.0);
			int argc = 8;
			const char *argv[8];
			argv[0] = strsave(toString(m_colorIndex->getValue()).c_str());
			getValues(argv, argc, frame);
			SandorFxRenderData *calligraphicData = new SandorFxRenderData(Calligraphic, argc, argv, 0, shrink);
			CalligraphicParams &params = calligraphicData->m_callParams;
			params.m_accuracy = m_accuracy->getValue(frame);
			params.m_horizontal = m_horizontal->getValue(frame);
			params.m_colorIndex = m_colorIndex->getValue();
			params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
			params.m_noise = m_noise->getValue(frame);
			params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
			params.m_thickness = m_thickness->getValue(frame) * 0.5;
			params.m_vertical = m_vertical->getValue(frame);
			info2.m_data.push_back(calligraphicData);

			return m_input->doGetBBox(frame, bBox, info2);
		} else {
			bBox = TRectD();
			return false;
		}
	}

	bool canHandle(const TRenderSettings &info, double frame) { return true; }

	bool allowUserCacheOnPort(int port) { return false; }

private:
	void getValues(const char *argv[], int argc, double frame)
	{
		double values[8];
		values[1] = m_noise->getValue(frame);
		values[2] = m_accuracy->getValue(frame);
		values[3] = m_doWDiagonal->getValue(frame);
		values[4] = m_vertical->getValue(frame);
		values[5] = m_upWDiagonal->getValue(frame);
		values[6] = m_horizontal->getValue(frame);
		values[7] = m_thickness->getValue(frame) * 0.5;
		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 < cParamLen; i++) {
			app = toString(param[i]);
			cParam[i] = strsave(app.c_str());
		}
	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &ri)
	{
		if (!m_input.isConnected())
			return;

		int shrink = tround((ri.m_shrinkX + ri.m_shrinkY) / 2.0);
		int argc = 8;
		const char *argv[8];
		argv[0] = strsave(toString(m_colorIndex->getValue()).c_str());
		getValues(argv, argc, frame);
		TRenderSettings ri2(ri);
		SandorFxRenderData *calligraphicData = new SandorFxRenderData(Calligraphic, argc, argv, 0, shrink);
		CalligraphicParams &params = calligraphicData->m_callParams;
		params.m_accuracy = m_accuracy->getValue(frame);
		params.m_horizontal = m_horizontal->getValue(frame);
		params.m_colorIndex = m_colorIndex->getValue();
		params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
		params.m_noise = m_noise->getValue(frame);
		params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
		params.m_thickness = m_thickness->getValue(frame) * 0.5;
		params.m_vertical = m_vertical->getValue(frame);
		ri2.m_data.push_back(calligraphicData);
		ri2.m_userCachable = false;
		m_input->compute(tile, frame, ri2);
	}

	void transform(double frame,
				   int port,
				   const TRectD &rectOnOutput,
				   const TRenderSettings &infoOnOutput,
				   TRectD &rectOnInput,
				   TRenderSettings &infoOnInput)
	{
		rectOnInput = rectOnOutput;
		infoOnInput = infoOnOutput;

		int shrink = tround((infoOnOutput.m_shrinkX + infoOnOutput.m_shrinkY) / 2.0);
		int argc = 8;
		const char *argv[8];
		argv[0] = strsave(toString(m_colorIndex->getValue()).c_str());
		getValues(argv, argc, frame);
		SandorFxRenderData *calligraphicData = new SandorFxRenderData(Calligraphic, argc, argv, 0, shrink);
		CalligraphicParams &params = calligraphicData->m_callParams;
		params.m_accuracy = m_accuracy->getValue(frame);
		params.m_horizontal = m_horizontal->getValue(frame);
		params.m_colorIndex = m_colorIndex->getValue();
		params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
		params.m_noise = m_noise->getValue(frame);
		params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
		params.m_thickness = m_thickness->getValue(frame);
		params.m_vertical = m_vertical->getValue(frame);
		infoOnInput.m_data.push_back(calligraphicData);
		infoOnInput.m_userCachable = false;
	}
};

//------------------------------------------------------------------

//===================================================================
class OutBorderFx : public TStandardRasterFx
{
	FX_PLUGIN_DECLARATION(OutBorderFx)
	TRasterFxPort m_input;
	TDoubleParamP m_thickness;
	TDoubleParamP m_horizontal;
	TDoubleParamP m_upWDiagonal;
	TDoubleParamP m_vertical;
	TDoubleParamP m_doWDiagonal;
	TDoubleParamP m_accuracy;
	TDoubleParamP m_noise;

public:
	OutBorderFx()
		: m_thickness(5.0), m_horizontal(100.0), m_upWDiagonal(100.0), m_vertical(100.0), m_doWDiagonal(100.0), m_accuracy(50.0), m_noise(0.0)
	{
		m_thickness->setMeasureName("fxLength");
		addInputPort("Source", m_input);
		bindParam(this, "Thickness", m_thickness);
		bindParam(this, "Accuracy", m_accuracy);
		bindParam(this, "Noise", m_noise);
		bindParam(this, "Horizontal", m_horizontal);
		bindParam(this, "upWDiagonal", m_upWDiagonal);
		bindParam(this, "Vertical", m_vertical);
		bindParam(this, "doWDiagonal", m_doWDiagonal);
		m_thickness->setValueRange(0.0, 30.0);
		m_horizontal->setValueRange(0.0, 100.0);
		m_upWDiagonal->setValueRange(0.0, 100.0);
		m_vertical->setValueRange(0.0, 100.0);
		m_doWDiagonal->setValueRange(0.0, 100.0);
		m_accuracy->setValueRange(0.0, 100.0);
		m_noise->setValueRange(0.0, 100.0);
	}

	~OutBorderFx(){};

	bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info)
	{
		if (m_input.isConnected()) {
			TRenderSettings info2(info);

			int shrink = tround((info.m_shrinkX + info.m_shrinkY) / 2.0);
			int argc = 8;
			const char *argv[8];
			argv[0] = strsave(toString(L"-1").c_str());
			getValues(argv, argc, frame);
			SandorFxRenderData *outBorderData = new SandorFxRenderData(OutBorder, argc, argv, 0, shrink);
			CalligraphicParams &params = outBorderData->m_callParams;
			params.m_accuracy = m_accuracy->getValue(frame);
			params.m_horizontal = m_horizontal->getValue(frame);
			params.m_colorIndex = L"-1";
			params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
			params.m_noise = m_noise->getValue(frame);
			params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
			params.m_thickness = m_thickness->getValue(frame);
			params.m_vertical = m_vertical->getValue(frame);
			info2.m_data.push_back(outBorderData);

			return m_input->doGetBBox(frame, bBox, info2);
		} else {
			bBox = TRectD();
			return false;
		}
	}

	bool canHandle(const TRenderSettings &info, double frame) { return true; }

private:
	void getValues(const char *argv[], int argc, double frame)
	{
		double values[8];
		values[1] = m_noise->getValue(frame);
		values[2] = m_accuracy->getValue(frame);
		values[3] = m_doWDiagonal->getValue(frame);
		values[4] = m_vertical->getValue(frame);
		values[5] = m_upWDiagonal->getValue(frame);
		values[6] = m_horizontal->getValue(frame);
		values[7] = m_thickness->getValue(frame);
		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 < cParamLen; i++) {
			app = toString(param[i]);
			cParam[i] = strsave(app.c_str());
		}
	}

	void doCompute(TTile &tile, double frame, const TRenderSettings &ri)
	{
		if (!m_input.isConnected())
			return;

		int shrink = tround((ri.m_shrinkX + ri.m_shrinkY) / 2.0);
		int argc = 8;
		const char *argv[8];
		argv[0] = strsave(toString(L"-1").c_str());
		getValues(argv, argc, frame);
		TRenderSettings ri2(ri);
		SandorFxRenderData *outBorderData = new SandorFxRenderData(OutBorder, argc, argv, 0, shrink);
		CalligraphicParams &params = outBorderData->m_callParams;
		params.m_accuracy = m_accuracy->getValue(frame);
		params.m_horizontal = m_horizontal->getValue(frame);
		params.m_colorIndex = L"-1";
		params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
		params.m_noise = m_noise->getValue(frame);
		params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
		params.m_thickness = m_thickness->getValue(frame);
		params.m_vertical = m_vertical->getValue(frame);
		ri2.m_data.push_back(outBorderData);
		m_input->compute(tile, frame, ri2);
	}

	void transform(double frame,
				   int port,
				   const TRectD &rectOnOutput,
				   const TRenderSettings &infoOnOutput,
				   TRectD &rectOnInput,
				   TRenderSettings &infoOnInput)
	{
		rectOnInput = rectOnOutput;
		infoOnInput = infoOnOutput;

		int shrink = tround((infoOnOutput.m_shrinkX + infoOnOutput.m_shrinkY) / 2.0);
		int argc = 8;
		const char *argv[8];
		argv[0] = strsave(toString(L"-1").c_str());
		getValues(argv, argc, frame);
		SandorFxRenderData *outBorderData = new SandorFxRenderData(OutBorder, argc, argv, 0, shrink);
		CalligraphicParams &params = outBorderData->m_callParams;
		params.m_accuracy = m_accuracy->getValue(frame);
		params.m_horizontal = m_horizontal->getValue(frame);
		params.m_colorIndex = L"-1";
		params.m_upWDiagonal = m_upWDiagonal->getValue(frame);
		params.m_noise = m_noise->getValue(frame);
		params.m_doWDiagonal = m_doWDiagonal->getValue(frame);
		params.m_thickness = m_thickness->getValue(frame);
		params.m_vertical = m_vertical->getValue(frame);
		infoOnInput.m_data.push_back(outBorderData);
	}
};

//------------------------------------------------------------------

FX_PLUGIN_IDENTIFIER(CalligraphicFx, "calligraphicFx");

FX_PLUGIN_IDENTIFIER(OutBorderFx, "outBorderFx");