1 files changed, 115 insertions, 0 deletions
diff --git a/source/blender/blenkernel/intern/virtual_node_tree.cc b/source/blender/blenkernel/intern/virtual_node_tree.cc
new file mode 100644
index 00000000000..81698d0e524
--- /dev/null
+++ b/source/blender/blenkernel/intern/virtual_node_tree.cc
@@ -0,0 +1,115 @@
+#include "BKE_virtual_node_tree.h"
+
+#include "BLI_listbase_wrapper.h"
+#include "BLI_map.h"
+
+namespace BKE {
+
+using BLI::Map;
+using BSocketList = BLI::IntrusiveListBaseWrapper<bNodeSocket>;
+using BNodeList = BLI::IntrusiveListBaseWrapper<bNode>;
+using BLinkList = BLI::IntrusiveListBaseWrapper<bNodeLink>;
+
+static bool is_reroute_node(const VNode &vnode)
+{
+  return vnode.idname() == "NodeReroute";
+}
+
+VirtualNodeTree::VirtualNodeTree(bNodeTree *btree) : m_btree(btree)
+{
+  BLI_assert(btree != nullptr);
+
+  VirtualNodeTree &vtree = *this;
+
+  Map<bNode *, VNode *> node_mapping;
+
+  for (bNode *bnode : BNodeList(btree->nodes)) {
+    VNode &vnode = *vtree.m_allocator.construct<VNode>();
+
+    vnode.m_vtree = &vtree;
+    vnode.m_bnode = bnode;
+    vnode.m_id = vtree.m_nodes_by_id.append_and_get_index(&vnode);
+    RNA_pointer_create(&btree->id, &RNA_Node, bnode, &vnode.m_rna);
+
+    for (bNodeSocket *bsocket : BSocketList(bnode->inputs)) {
+      VInputSocket &vsocket = *vtree.m_allocator.construct<VInputSocket>();
+
+      vsocket.m_node = &vnode;
+      vsocket.m_index = vnode.m_inputs.append_and_get_index(&vsocket, m_allocator);
+      vsocket.m_is_input = true;
+      vsocket.m_bsocket = bsocket;
+      vsocket.m_id = vtree.m_sockets_by_id.append_and_get_index(&vsocket);
+      RNA_pointer_create(&btree->id, &RNA_NodeSocket, bsocket, &vsocket.m_rna);
+
+      vtree.m_input_sockets.append(&vsocket);
+    }
+
+    for (bNodeSocket *bsocket : BSocketList(bnode->outputs)) {
+      VOutputSocket &vsocket = *vtree.m_allocator.construct<VOutputSocket>();
+
+      vsocket.m_node = &vnode;
+      vsocket.m_index = vnode.m_outputs.append_and_get_index(&vsocket, m_allocator);
+      vsocket.m_is_input = false;
+      vsocket.m_bsocket = bsocket;
+      vsocket.m_id = vtree.m_sockets_by_id.append_and_get_index(&vsocket);
+      RNA_pointer_create(&btree->id, &RNA_NodeSocket, bsocket, &vsocket.m_rna);
+
+      vtree.m_output_sockets.append(&vsocket);
+    }
+
+    node_mapping.add_new(bnode, &vnode);
+  }
+
+  for (bNodeLink *blink : BLinkList(btree->links)) {
+    VOutputSocket &from_vsocket =
+        *node_mapping.lookup(blink->fromnode)
+             ->m_outputs[BSocketList(blink->fromnode->outputs).index_of(blink->fromsock)];
+    VInputSocket &to_vsocket =
+        *node_mapping.lookup(blink->tonode)
+             ->m_inputs[BSocketList(blink->tonode->inputs).index_of(blink->tosock)];
+
+    from_vsocket.m_directly_linked_sockets.append(&to_vsocket, m_allocator);
+    to_vsocket.m_directly_linked_sockets.append(&from_vsocket, m_allocator);
+  }
+
+  for (VOutputSocket *socket : vtree.m_output_sockets) {
+    if (!is_reroute_node(socket->node())) {
+      vtree.find_targets_skipping_reroutes(*socket, socket->m_linked_sockets);
+      for (VSocket *target : socket->m_linked_sockets) {
+        target->m_linked_sockets.append(socket, m_allocator);
+      }
+    }
+  }
+
+  for (VNode *vnode : vtree.m_nodes_by_id) {
+    vtree.m_nodes_by_idname.add(vnode->idname(), vnode);
+  }
+}
+
+void VirtualNodeTree::find_targets_skipping_reroutes(VOutputSocket &vsocket,
+                                                     LinearAllocatedVector<VSocket *> &r_targets)
+{
+  for (VSocket *direct_target : vsocket.m_directly_linked_sockets) {
+    if (is_reroute_node(*direct_target->m_node)) {
+      this->find_targets_skipping_reroutes(*direct_target->m_node->m_outputs[0], r_targets);
+    }
+    else if (!r_targets.contains(direct_target)) {
+      r_targets.append(direct_target, m_allocator);
+    }
+  }
+}
+
+VirtualNodeTree::~VirtualNodeTree()
+{
+  for (VNode *node : m_nodes_by_id) {
+    node->~VNode();
+  }
+  for (VInputSocket *socket : m_input_sockets) {
+    socket->~VInputSocket();
+  }
+  for (VOutputSocket *socket : m_output_sockets) {
+    socket->~VOutputSocket();
+  }
+}
+
+};  // namespace BKE