Nodes: add redundant name check in debug builds to prevent errors

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_;