1 files changed, 105 insertions, 0 deletions
diff --git a/source/blender/compositor/intern/COM_OutputSocket.h b/source/blender/compositor/intern/COM_OutputSocket.h
new file mode 100644
index 00000000000..8815f0bb3a8
--- /dev/null
+++ b/source/blender/compositor/intern/COM_OutputSocket.h
@@ -0,0 +1,105 @@
+/*
+ * 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
+ */
+
+#ifndef _COM_OutputSocket_h
+#define _COM_OutputSocket_h
+
+#include <vector>
+#include "COM_Socket.h"
+#include "COM_ChannelInfo.h"
+
+using namespace std;
+class SocketConnection;
+class Node;
+class InputSocket;
+class WriteBufferOperation;
+
+//#define COM_ST_INPUT 0
+//#define COM_ST_OUTPUT 1
+
+/**
+  * @brief OutputSocket are sockets that can send data/input
+  * @ingroup Model
+  */
+class OutputSocket : public Socket {
+private:
+	vector<SocketConnection*> connections;
+	
+	/**
+	  * @brief index of the inputsocket that determines the datatype of this outputsocket
+	  * -1 will not use any inputsocket to determine the datatype, but use the outputsocket
+	  * default datatype.
+	  */
+	int inputSocketDataTypeDeterminatorIndex;
+	
+	ChannelInfo channelinfo[4];
+	void removeFirstConnection();
+public:
+	OutputSocket(DataType datatype);
+	OutputSocket(DataType datatype, int inputSocketDataTypeDeterminatorIndex);
+	OutputSocket(OutputSocket * from);
+	void addConnection(SocketConnection *connection);
+	SocketConnection* getConnection(unsigned int index) {return this->connections[index];}
+	const int isConnected() const;
+	int isOutputSocket() const;
+	
+	/**
+	  * @brief determine the resolution of this socket
+	  * @param resolution the result of this operation
+	  * @param preferredResolution the preferrable resolution as no resolution could be determined
+	  */
+	void determineResolution(unsigned int resolution[], unsigned int preferredResolution[]);
+	
+	/**
+	  * @brief determine the actual data type and channel info.
+	  */
+	void determineActualDataType();
+	void relinkConnections(OutputSocket *relinkToSocket) {this->relinkConnections(relinkToSocket, false);};
+	void relinkConnections(OutputSocket *relinkToSocket, bool single);
+	bool isActualDataTypeDeterminedByInputSocket() {
+		return this->inputSocketDataTypeDeterminatorIndex>-1;
+	}
+	const int getNumberOfConnections() {return connections.size();}
+	
+	/**
+	  * @brief get the index of the inputsocket that determines the datatype of this outputsocket
+	  */
+	int getInputSocketDataTypeDeterminatorIndex() {return this->inputSocketDataTypeDeterminatorIndex;}
+	void clearConnections();
+	
+	/**
+	  * @brief find a connected write buffer operation to this OutputSocket
+	  * @return WriteBufferOperation or NULL
+	  */
+	WriteBufferOperation* findAttachedWriteBufferOperation() const;
+	ChannelInfo* getChannelInfo(const int channelnumber);
+	
+	/**
+	  * @brief trigger determine actual data type to all connected sockets
+	  * @note will only be triggered just after the actual data type is set.
+	  */
+	void fireActualDataType();
+
+private:
+
+};
+#endif