diff options
Diffstat (limited to 'source/blender/compositor/intern/COM_Converter.cpp')
| -rw-r--r-- | source/blender/compositor/intern/COM_Converter.cpp | 492 |
1 files changed, 492 insertions, 0 deletions
diff --git a/source/blender/compositor/intern/COM_Converter.cpp b/source/blender/compositor/intern/COM_Converter.cpp new file mode 100644 index 00000000000..56b1da97df8 --- /dev/null +++ b/source/blender/compositor/intern/COM_Converter.cpp @@ -0,0 +1,492 @@ +/* + * 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); +} |