diff options
Diffstat (limited to 'source/blender/nodes/NOD_XXX_node_tree.hh')
-rw-r--r-- | source/blender/nodes/NOD_XXX_node_tree.hh | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/source/blender/nodes/NOD_XXX_node_tree.hh b/source/blender/nodes/NOD_XXX_node_tree.hh index d1446857442..aecc968aae8 100644 --- a/source/blender/nodes/NOD_XXX_node_tree.hh +++ b/source/blender/nodes/NOD_XXX_node_tree.hh @@ -17,6 +17,7 @@ #pragma once #include "BLI_function_ref.hh" +#include "BLI_vector_set.hh" #include "NOD_node_tree_ref.hh" @@ -125,12 +126,16 @@ class XXXNodeTree { private: LinearAllocator<> allocator_; XXXNodeTreeContext *root_context_; + VectorSet<const NodeTreeRef *> used_node_tree_refs_; public: XXXNodeTree(bNodeTree &btree, NodeTreeRefMap &node_tree_refs); ~XXXNodeTree(); const XXXNodeTreeContext &root_context() const; + Span<const NodeTreeRef *> used_node_tree_refs() const; + + bool has_link_cycles() const; private: XXXNodeTreeContext &construct_context_recursively(XXXNodeTreeContext *parent_context, @@ -354,4 +359,9 @@ inline const XXXNodeTreeContext &XXXNodeTree::root_context() const return *root_context_; } +inline Span<const NodeTreeRef *> XXXNodeTree::used_node_tree_refs() const +{ + return used_node_tree_refs_; +} + } // namespace blender::nodes |