diff options
| -rw-r--r-- | source/blender/blenkernel/BKE_derived_node_tree.hh | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/source/blender/blenkernel/BKE_derived_node_tree.hh b/source/blender/blenkernel/BKE_derived_node_tree.hh index 3790d06a534..083057835a5 100644 --- a/source/blender/blenkernel/BKE_derived_node_tree.hh +++ b/source/blender/blenkernel/BKE_derived_node_tree.hh @@ -140,6 +140,9 @@ class DNode : NonCopyable, NonMovable { const DInputSocket &input(uint index) const; const DOutputSocket &output(uint index) const; + const DInputSocket &input(uint index, StringRef expected_name) const; + const DOutputSocket &output(uint index, StringRef expected_name) const; + uint id() const; PointerRNA *rna() const; @@ -408,6 +411,22 @@ inline const DOutputSocket &DNode::output(uint index) const return *outputs_[index]; } +inline const DInputSocket &DNode::input(uint index, StringRef expected_name) const +{ + const DInputSocket &socket = *inputs_[index]; + BLI_assert(socket.name() == expected_name); + UNUSED_VARS_NDEBUG(expected_name); + return socket; +} + +inline const DOutputSocket &DNode::output(uint index, StringRef expected_name) const +{ + const DOutputSocket &socket = *outputs_[index]; + BLI_assert(socket.name() == expected_name); + UNUSED_VARS_NDEBUG(expected_name); + return socket; +} + inline uint DNode::id() const { return id_; |