Fix incorrect connections for muted nodes in tile compositor
[blender.git] / source / blender / compositor / nodes / COM_MuteNode.h
index aab37e5f888225439fdadd0a81221d3085609d5d..44fbe6e80c6145ba3dda7c1b937f0cc2aba68ea7 100644 (file)
 #ifndef _COM_MuteNode_h_
 #define _COM_MuteNode_h_
 
+#include <map>
+
 #include "COM_Node.h"
 
+extern "C" {
+       #include "BKE_node.h"
+}
+
 /**
  * @brief MuteNode
  * @ingroup Node
@@ -34,7 +40,11 @@ public:
        MuteNode(bNode *editorNode);
        void convertToOperations(ExecutionSystem *graph, CompositorContext *context);
 private:
+       typedef std::map<bNodeSocket *, Socket *> SocketMap;
+
        void reconnect(ExecutionSystem *graph, OutputSocket *output);
+
+       template<class SocketType> void fillSocketMap(vector<SocketType *> &sockets, SocketMap &socketMap);
 };
 
 #endif