1 files changed, 128 insertions, 0 deletions

diff --git a/source/blender/compositor/intern/COM_NodeBase.cpp b/source/blender/compositor/intern/COM_NodeBase.cpp
new file mode 100644
index 00000000000..6e6be889572
--- /dev/null
+++ b/source/blender/compositor/intern/COM_NodeBase.cpp
@@ -0,0 +1,128 @@
+/*
+ * 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_NodeBase.h"
+#include "string.h"
+#include "COM_NodeOperation.h"
+#include "BKE_node.h"
+#include "COM_SetValueOperation.h"
+#include "COM_SetColorOperation.h"
+#include "COM_SocketConnection.h"
+#include "COM_ExecutionSystem.h"
+
+NodeBase::NodeBase() {
+}
+
+
+NodeBase::~NodeBase(){
+	while (!this->outputsockets.empty()) {
+		delete (this->outputsockets.back());
+		this->outputsockets.pop_back();
+	}
+	while (!this->inputsockets.empty()) {
+		delete (this->inputsockets.back());
+		this->inputsockets.pop_back();
+	}
+}
+
+void NodeBase::addInputSocket(DataType datatype) {
+	this->addInputSocket(datatype, COM_SC_CENTER, NULL);
+}
+
+void NodeBase::addInputSocket(DataType datatype, InputSocketResizeMode resizeMode) {
+	this->addInputSocket(datatype, resizeMode, NULL);
+}
+void NodeBase::addInputSocket(DataType datatype, InputSocketResizeMode resizeMode, bNodeSocket* bSocket) {
+	InputSocket *socket = new InputSocket(datatype, resizeMode);
+	socket->setEditorSocket(bSocket);
+	socket->setNode(this);
+	this->inputsockets.push_back(socket);
+}
+
+void NodeBase::addOutputSocket(DataType datatype) {
+	this->addOutputSocket(datatype, NULL);
+	
+}
+void NodeBase::addOutputSocket(DataType datatype, bNodeSocket* bSocket) {
+	OutputSocket *socket = new OutputSocket(datatype);
+	socket->setEditorSocket(bSocket);
+	socket->setNode(this);
+	this->outputsockets.push_back(socket);
+}
+const bool NodeBase::isInputNode() const {
+	return this->inputsockets.size() == 0;
+}
+
+OutputSocket* NodeBase::getOutputSocket(int index) {
+	return this->outputsockets[index];
+}
+
+InputSocket* NodeBase::getInputSocket(int index) {
+	return this->inputsockets[index];
+}
+
+
+void NodeBase::determineActualSocketDataTypes() {
+	unsigned int index;
+	for (index = 0 ; index < this->outputsockets.size() ; index ++) {
+		OutputSocket* socket = this->outputsockets[index];
+		if (socket->getActualDataType() ==COM_DT_UNKNOWN && socket->isConnected()) {
+			socket->determineActualDataType();
+		}
+	}
+	for (index = 0 ; index < this->inputsockets.size() ; index ++) {
+		InputSocket* socket = this->inputsockets[index];
+		if (socket->getActualDataType() ==COM_DT_UNKNOWN) {
+			socket->determineActualDataType();
+		}
+	}
+}
+
+DataType NodeBase::determineActualDataType(OutputSocket *outputsocket) {
+	const int inputIndex = outputsocket->getInputSocketDataTypeDeterminatorIndex();
+	if (inputIndex != -1) {
+		return this->getInputSocket(inputIndex)->getActualDataType();
+	} else {
+		return outputsocket->getDataType();
+	}
+}
+
+void NodeBase::notifyActualDataTypeSet(InputSocket *socket, DataType actualType) {
+	unsigned int index;
+	int socketIndex = -1;
+	for (index = 0 ; index < this->inputsockets.size() ; index ++) {
+		if (this->inputsockets[index] == socket) {
+			socketIndex = (int)index;
+			break;
+		}
+	}
+	if (socketIndex == -1) return;
+	
+	for (index = 0 ; index < this->outputsockets.size() ; index ++) {
+		OutputSocket* socket = this->outputsockets[index];
+		if (socket->isActualDataTypeDeterminedByInputSocket() &&
+				socket->getInputSocketDataTypeDeterminatorIndex() == socketIndex) {
+			socket->setActualDataType(actualType);
+			socket->fireActualDataType();
+		}
+	}
+}