diff options
Diffstat (limited to 'source/blender/nodes/NOD_node_tree_ref.hh')
-rw-r--r-- | source/blender/nodes/NOD_node_tree_ref.hh | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/source/blender/nodes/NOD_node_tree_ref.hh b/source/blender/nodes/NOD_node_tree_ref.hh index 8b430ea1c48..5795617deb0 100644 --- a/source/blender/nodes/NOD_node_tree_ref.hh +++ b/source/blender/nodes/NOD_node_tree_ref.hh @@ -132,23 +132,31 @@ class SocketRef : NonCopyable, NonMovable { class InputSocketRef final : public SocketRef { public: + friend NodeTreeRef; + Span<const OutputSocketRef *> logically_linked_sockets() const; Span<const OutputSocketRef *> directly_linked_sockets() const; bool is_multi_input_socket() const; + private: void foreach_logical_origin(FunctionRef<void(const OutputSocketRef &)> origin_fn, FunctionRef<void(const SocketRef &)> skipped_fn, - bool only_follow_first_input_link = false) const; + bool only_follow_first_input_link, + Vector<const InputSocketRef *> &handled_sockets) const; }; class OutputSocketRef final : public SocketRef { public: + friend NodeTreeRef; + Span<const InputSocketRef *> logically_linked_sockets() const; Span<const InputSocketRef *> directly_linked_sockets() const; + private: void foreach_logical_target(FunctionRef<void(const InputSocketRef &)> target_fn, - FunctionRef<void(const SocketRef &)> skipped_fn) const; + FunctionRef<void(const SocketRef &)> skipped_fn, + Vector<const OutputSocketRef *> &handled_sockets) const; }; class NodeRef : NonCopyable, NonMovable { |