/*
 * Copyright 2011, Blender Foundation.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * Contributor: 
 *		Jeroen Bakker 
 *		Monique Dewanchand
 */

#include "COM_Converter.h"
#include "BKE_node.h"
#include "COM_CompositorNode.h"
#include "COM_RenderLayersNode.h"
#include "COM_ColorToBWNode.h"
#include "string.h"
#include "COM_SocketConnection.h"
#include "COM_ConvertColourToValueProg.h"
#include "COM_ConvertValueToColourProg.h"
#include "COM_ConvertColorToVectorOperation.h"
#include "COM_ConvertValueToVectorOperation.h"
#include "COM_ConvertVectorToColorOperation.h"
#include "COM_ConvertVectorToValueOperation.h"
#include "COM_ExecutionSystem.h"
#include "COM_MixNode.h"
#include "COM_MuteNode.h"
#include "COM_TranslateNode.h"
#include "COM_RotateNode.h"
#include "COM_ScaleNode.h"
#include "COM_FlipNode.h"
#include "COM_IDMaskNode.h"
#include "COM_FilterNode.h"
#include "COM_BrightnessNode.h"
#include "COM_SeparateRGBANode.h"
#include "COM_CombineRGBANode.h"
#include "COM_SeparateHSVANode.h"
#include "COM_CombineHSVANode.h"
#include "COM_SeparateYUVANode.h"
#include "COM_CombineYUVANode.h"
#include "COM_SeparateYCCANode.h"
#include "COM_CombineYCCANode.h"
#include "COM_AlphaOverNode.h"
#include "COM_ColorBalanceNode.h"
#include "COM_ViewerNode.h"
#include "COM_SplitViewerNode.h"
#include "COM_InvertNode.h"
#include "COM_GroupNode.h"
#include "COM_NormalNode.h"
#include "COM_NormalizeNode.h"
#include "COM_ImageNode.h"
#include "COM_BokehImageNode.h"
#include "COM_ColorCurveNode.h"
#include "COM_VectorCurveNode.h"
#include "COM_SetAlphaNode.h"
#include "COM_ConvertAlphaNode.h"
#include "COM_MapUVNode.h"
#include "COM_DisplaceNode.h"
#include "COM_MathNode.h"
#include "COM_HueSaturationValueNode.h"
#include "COM_HueSaturationValueCorrectNode.h"
#include "COM_ColorCorrectionNode.h"
#include "COM_BoxMaskNode.h"
#include "COM_EllipseMaskNode.h"
#include "COM_GammaNode.h"
#include "COM_ColorRampNode.h"
#include "COM_DifferenceMatteNode.h"
#include "COM_LuminanceMatteNode.h"
#include "COM_DistanceMatteNode.h"
#include "COM_ChromaMatteNode.h"
#include "COM_ColorMatteNode.h"
#include "COM_ChannelMatteNode.h"
#include "COM_BlurNode.h"
#include "COM_BokehBlurNode.h"
#include "COM_DilateErodeNode.h"
#include "COM_TranslateOperation.h"
#include "COM_LensDistortionNode.h"
#include "COM_TextureNode.h"
#include "COM_ColorNode.h"
#include "COM_ValueNode.h"
#include "COM_TimeNode.h"
#include "COM_DirectionalBlurNode.h"
#include "COM_ZCombineNode.h"
#include "COM_SetValueOperation.h"
#include "COM_ScaleOperation.h"
#include "COM_ExecutionSystemHelper.h"
#include "COM_TonemapNode.h"
#include "COM_SwitchNode.h"
#include "COM_GlareNode.h"
#include "COM_MovieClipNode.h"
#include "COM_ColorSpillNode.h"
#include "COM_OutputFileNode.h"
#include "COM_MapValueNode.h"
#include "COM_TransformNode.h"
#include "COM_Stabilize2dNode.h"
#include "COM_BilateralBlurNode.h"
#include "COM_VectorBlurNode.h"
#include "COM_MovieDistortionNode.h"
#include "COM_ViewLevelsNode.h"
#include "COM_DefocusNode.h"
#include "COM_DoubleEdgeMaskNode.h"
#include "COM_CropNode.h"

Node* Converter::convert(bNode *bNode) {
	Node * node;

	if (bNode->flag & NODE_MUTED) {
		node = new MuteNode(bNode);
		return node;
	}

	switch (bNode->type) {
	case CMP_NODE_COMPOSITE:
		node = new CompositorNode(bNode);
		break;
	case CMP_NODE_R_LAYERS:
		node = new RenderLayersNode(bNode);
		break;
	case CMP_NODE_TEXTURE:
		node = new TextureNode(bNode);
		break;
	case CMP_NODE_RGBTOBW:
		node = new ColourToBWNode(bNode);
		break;
	case CMP_NODE_MIX_RGB:
		node = new MixNode(bNode);
		break;
	case CMP_NODE_TRANSLATE:
		node = new TranslateNode(bNode);
		break;
	case CMP_NODE_SCALE:
		node = new ScaleNode(bNode);
		break;
	case CMP_NODE_ROTATE:
		node = new RotateNode(bNode);
		break;
	case CMP_NODE_FLIP:
		node = new FlipNode(bNode);
		break;
	case CMP_NODE_FILTER:
		node = new FilterNode(bNode);
		break;
	case CMP_NODE_ID_MASK:
		node = new IDMaskNode(bNode);
		break;
	case CMP_NODE_BRIGHTCONTRAST:
		node = new BrightnessNode(bNode);
		break;
	case CMP_NODE_SEPRGBA:
		node = new SeparateRGBANode(bNode);
		break;
	case CMP_NODE_COMBRGBA:
		node = new CombineRGBANode(bNode);
		break;
	case CMP_NODE_SEPHSVA:
		node = new SeparateHSVANode(bNode);
		break;
	case CMP_NODE_COMBHSVA:
		node = new CombineHSVANode(bNode);
		break;
	case CMP_NODE_SEPYUVA:
		node = new SeparateYUVANode(bNode);
		break;
	case CMP_NODE_COMBYUVA:
		node = new CombineYUVANode(bNode);
		break;
	case CMP_NODE_SEPYCCA:
		node = new SeparateYCCANode(bNode);
		break;
	case CMP_NODE_COMBYCCA:
		node = new CombineYCCANode(bNode);
		break;
	case CMP_NODE_ALPHAOVER:
		node = new AlphaOverNode(bNode);
		break;
	case CMP_NODE_COLORBALANCE:
		node = new ColorBalanceNode(bNode);
		break;
	case CMP_NODE_VIEWER:
		node = new ViewerNode(bNode);
		break;
	case CMP_NODE_SPLITVIEWER:
		node = new SplitViewerNode(bNode);
		break;
	case CMP_NODE_INVERT:
		node = new InvertNode(bNode);
		break;
	case NODE_GROUP:
		node = new GroupNode(bNode);
		break;
	case CMP_NODE_NORMAL:
		node = new NormalNode(bNode);
		break;
	case CMP_NODE_NORMALIZE:
		node = new NormalizeNode(bNode);
		break;
	case CMP_NODE_IMAGE:
		node = new ImageNode(bNode);
		break;
	case CMP_NODE_SETALPHA:
		node = new SetAlphaNode(bNode);
		break;
	case CMP_NODE_PREMULKEY:
		node = new ConvertAlphaNode(bNode);
		break;
	case CMP_NODE_MATH:
		node = new MathNode(bNode);
		break;
	case CMP_NODE_HUE_SAT:
		node = new HueSaturationValueNode(bNode);
		break;
	case CMP_NODE_COLORCORRECTION:
		node = new ColorCorrectionNode(bNode);
		break;
	case CMP_NODE_MASK_BOX:
		node = new BoxMaskNode(bNode);
		break;
	case CMP_NODE_MASK_ELLIPSE:
		node = new EllipseMaskNode(bNode);
		break;
	case CMP_NODE_GAMMA:
		node = new GammaNode(bNode);
		break;
	case CMP_NODE_CURVE_RGB:
		node = new ColorCurveNode(bNode);
		break;
	case CMP_NODE_CURVE_VEC:
		node = new VectorCurveNode(bNode);
		break;
	case CMP_NODE_HUECORRECT:
		node = new HueSaturationValueCorrectNode(bNode);
		break;
	case CMP_NODE_MAP_UV:
		node = new MapUVNode(bNode);
		break;
	case CMP_NODE_DISPLACE:
		node = new DisplaceNode(bNode);
		break;
	case CMP_NODE_VALTORGB:
		node = new ColorRampNode(bNode);
		break;
	case CMP_NODE_DIFF_MATTE:
		node = new DifferenceMatteNode(bNode);
		break;
	case CMP_NODE_LUMA_MATTE:
		node = new LuminanceMatteNode(bNode);
		break;
	case CMP_NODE_DIST_MATTE:
		node = new DistanceMatteNode(bNode);
		break;
	case CMP_NODE_CHROMA_MATTE:
		node = new ChromaMatteNode(bNode);
		break;
	case CMP_NODE_COLOR_MATTE:
		node = new ColorMatteNode(bNode);
		break;
	case CMP_NODE_CHANNEL_MATTE:
		node = new ChannelMatteNode(bNode);
		break;
	case CMP_NODE_BLUR:
		node = new BlurNode(bNode);
		break;
	case CMP_NODE_BOKEHIMAGE:
		node = new BokehImageNode(bNode);
		break;
	case CMP_NODE_BOKEHBLUR:
		node = new BokehBlurNode(bNode);
		break;
	case CMP_NODE_DILATEERODE:
		node = new DilateErodeNode(bNode);
		break;
	case CMP_NODE_LENSDIST:
		node = new LensDistortionNode(bNode);
		break;
	case CMP_NODE_RGB:
		node = new ColorNode(bNode);
		break;
	case CMP_NODE_VALUE:
		node = new ValueNode(bNode);
		break;
	case CMP_NODE_TIME:
		node = new TimeNode(bNode);
		break;
	case CMP_NODE_DBLUR:
		node = new DirectionalBlurNode(bNode);
		break;
	case CMP_NODE_ZCOMBINE:
		node = new ZCombineNode(bNode);
		break;
	case CMP_NODE_TONEMAP:
		node = new TonemapNode(bNode);
		break;
	case CMP_NODE_SWITCH:
		node = new SwitchNode(bNode);
		break;
	case CMP_NODE_GLARE:
		node = new GlareNode(bNode);
		break;
	case CMP_NODE_MOVIECLIP:
		node = new MovieClipNode(bNode);
		break;
	case CMP_NODE_COLOR_SPILL:
		node = new ColorSpillNode(bNode);
		break;
case CMP_NODE_OUTPUT_FILE:
		node = new OutputFileNode(bNode);
		break;
	case CMP_NODE_MAP_VALUE:
		node = new MapValueNode(bNode);
		break;
	case CMP_NODE_TRANSFORM:
		node = new TransformNode(bNode);
		break;
	case CMP_NODE_STABILIZE2D:
		node = new Stabilize2dNode(bNode);
		break;
	case CMP_NODE_BILATERALBLUR:
		node = new BilateralBlurNode(bNode);
		break;
	case CMP_NODE_VECBLUR:
		node = new VectorBlurNode(bNode);
		break;
	case CMP_NODE_MOVIEDISTORTION:
		node = new MovieDistortionNode(bNode);
		break;
	case CMP_NODE_VIEW_LEVELS:
		node = new ViewLevelsNode(bNode);
		break;
	case CMP_NODE_DEFOCUS:
		node = new DefocusNode(bNode);
		break;
	case CMP_NODE_DOUBLEEDGEMASK:
		node = new DoubleEdgeMaskNode(bNode);
		break;
	case CMP_NODE_CROP:
		node = new CropNode(bNode);
		break;
	/* not inplemented yet */
	default:
		node = new MuteNode(bNode);
		break;
	}
	return node;
}
void Converter::convertDataType(SocketConnection* connection, ExecutionSystem *system) {
	OutputSocket* outputSocket = connection->getFromSocket();
	InputSocket* inputSocket = connection->getToSocket();
	DataType fromDatatype = outputSocket->getActualDataType();
	DataType toDatatype = inputSocket->getActualDataType();
	NodeOperation * converter = NULL;
	if (fromDatatype == COM_DT_VALUE && toDatatype == COM_DT_COLOR) {
		converter = new ConvertValueToColourProg();
	}
	else if (fromDatatype == COM_DT_VALUE && toDatatype == COM_DT_VECTOR) {
		converter = new ConvertValueToVectorOperation();
	}
	else if (fromDatatype == COM_DT_COLOR && toDatatype == COM_DT_VALUE) {
		converter = new ConvertColourToValueProg();
	}
	else if (fromDatatype == COM_DT_COLOR && toDatatype == COM_DT_VECTOR) {
		converter = new ConvertColorToVectorOperation();
	}
	else if (fromDatatype == COM_DT_VECTOR && toDatatype == COM_DT_VALUE) {
		converter = new ConvertVectorToValueOperation();
	}
	else if (fromDatatype == COM_DT_VECTOR && toDatatype == COM_DT_COLOR) {
		converter = new ConvertVectorToColorOperation();
	}
	if (converter != NULL) {
		inputSocket->relinkConnections(converter->getInputSocket(0));
		ExecutionSystemHelper::addLink(system->getConnections(), converter->getOutputSocket(), inputSocket);
		system->addOperation(converter);
	}
}

void Converter::convertResolution(SocketConnection *connection, ExecutionSystem *system) {
	InputSocketResizeMode mode = connection->getToSocket()->getResizeMode();

	NodeOperation * toOperation = (NodeOperation*)connection->getToNode();
	const float toWidth = toOperation->getWidth();
	const float toHeight = toOperation->getHeight();
	NodeOperation * fromOperation = (NodeOperation*)connection->getFromNode();
	const float fromWidth = fromOperation->getWidth();
	const float fromHeight = fromOperation->getHeight();
	bool doCenter = false;
	bool doScale = false;
	float addX=	(toWidth-fromWidth)/2.0f;
	float addY = (toHeight-fromHeight)/2.0f;
	float scaleX=0;
	float scaleY=0;

	switch (mode) {
	case COM_SC_NO_RESIZE:
		break;
	case COM_SC_CENTER:
		doCenter = true;
		break;
	case COM_SC_FIT_WIDTH:
		doCenter = true;
		doScale = true;
		scaleX = scaleY = toWidth/fromWidth;
		break;
	case COM_SC_FIT_HEIGHT:
		doCenter = true;
		doScale = true;
		scaleX = scaleY = toHeight/fromHeight;
		break;
	case COM_SC_FIT:
		doCenter = true;
		doScale = true;
		scaleX = toWidth/fromWidth;
		scaleY = toHeight/fromHeight;
		if (scaleX < scaleY) {
			scaleX = scaleY;
		}
		else {
			scaleY = scaleX;
		}
		break;
	case COM_SC_STRETCH:
		doCenter = true;
		doScale = true;
		scaleX = toWidth/fromWidth;
		scaleY = toHeight/fromHeight;
		break;

	}

	if (doCenter) {
		NodeOperation *first = NULL;
		SocketConnection *c;
		ScaleOperation * scaleOperation = NULL;
		if (doScale) {
			scaleOperation = new ScaleOperation();
			first = scaleOperation;
			SetValueOperation * sxop = new SetValueOperation();
			sxop->setValue(scaleX);
			c = ExecutionSystemHelper::addLink(system->getConnections(), sxop->getOutputSocket(), scaleOperation->getInputSocket(1));
			c->setIgnoreResizeCheck(true);
			SetValueOperation * syop = new SetValueOperation();
			syop->setValue(scaleY);
			c = ExecutionSystemHelper::addLink(system->getConnections(), syop->getOutputSocket(), scaleOperation->getInputSocket(2));
			c->setIgnoreResizeCheck(true);
			system->addOperation(sxop);
			system->addOperation(syop);

			unsigned int resolution[2] = {fromWidth, fromHeight};
			scaleOperation->setResolution(resolution);
			sxop->setResolution(resolution);
			syop->setResolution(resolution);
			system->addOperation(scaleOperation);

			c->setIgnoreResizeCheck(true);
		}

		TranslateOperation * translateOperation = new TranslateOperation();
		if (!first) first = translateOperation;
		SetValueOperation * xop = new SetValueOperation();
		xop->setValue(addX);
		c = ExecutionSystemHelper::addLink(system->getConnections(), xop->getOutputSocket(), translateOperation->getInputSocket(1));
		c->setIgnoreResizeCheck(true);
		SetValueOperation * yop = new SetValueOperation();
		yop->setValue(addY);
		c = ExecutionSystemHelper::addLink(system->getConnections(), yop->getOutputSocket(), translateOperation->getInputSocket(2));
		c->setIgnoreResizeCheck(true);
		system->addOperation(xop);
		system->addOperation(yop);

		unsigned int resolution[2] = {toWidth, toHeight};
		translateOperation->setResolution(resolution);
		xop->setResolution(resolution);
		yop->setResolution(resolution);
		system->addOperation(translateOperation);

		if (doScale) {
			c = ExecutionSystemHelper::addLink(system->getConnections(), scaleOperation->getOutputSocket(), translateOperation->getInputSocket(0));
			c->setIgnoreResizeCheck(true);
		}

		InputSocket * inputSocket = connection->getToSocket();
		inputSocket->relinkConnections(first->getInputSocket(0));
		c = ExecutionSystemHelper::addLink(system->getConnections(), translateOperation->getOutputSocket(), inputSocket);
		c->setIgnoreResizeCheck(true);
	}

	connection->setIgnoreResizeCheck(true);
}