Differential D2493 Diff 12620 source/blender/compositor/nodes/COM_DistanceTransformNode.cpp

Changeset View

Standalone View

source/blender/compositor/nodes/COM_DistanceTransformNode.cpp

This file was added.

				/*
				* Copyright 2017, 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: IRIE Shinsuke
				*/

				#include "COM_DistanceTransformNode.h"
				#include "COM_DistanceTransformOperation.h"
				#include "COM_ExecutionSystem.h"
				#include "COM_ConvertOperation.h"

				DistanceTransformNode::DistanceTransformNode(bNode *editorNode) : Node(editorNode)
				{
				/* pass */
				}

				void DistanceTransformNode::convertToOperations(NodeConverter &converter, const CompositorContext &/context/) const
				{
				DistanceTransformOperation *operation = new DistanceTransformOperation();
				bNode *editorNode = this->getbNode();
				operation->setInvert(editorNode->custom1);
				operation->setRelative(editorNode->custom2);
				operation->setThreshold(editorNode->custom3);
				converter.addOperation(operation);

				converter.mapInputSocket(getInputSocket(0), operation->getInputSocket(0));

				/* Distance */
				SeparateChannelOperation *operationD = new SeparateChannelOperation();
				operationD->setChannel(0);
				converter.addOperation(operationD);

				converter.addLink(operation->getOutputSocket(), operationD->getInputSocket(0));
				converter.mapOutputSocket(getOutputSocket(0), operationD->getOutputSocket());

				/* Vector X */
				SeparateChannelOperation *operationX = new SeparateChannelOperation();
				operationX->setChannel(1);
				converter.addOperation(operationX);

				converter.addLink(operation->getOutputSocket(), operationX->getInputSocket(0));
				converter.mapOutputSocket(getOutputSocket(1), operationX->getOutputSocket());

				/* Vector Y */
				SeparateChannelOperation *operationY = new SeparateChannelOperation();
				operationY->setChannel(2);
				converter.addOperation(operationY);

				converter.addLink(operation->getOutputSocket(), operationY->getInputSocket(0));
				converter.mapOutputSocket(getOutputSocket(2), operationY->getOutputSocket());
				}