From 7bae599232ab5d00a0b7a9e310741df6fbea204f Mon Sep 17 00:00:00 2001 From: Jacques Lucke Date: Fri, 10 Jul 2020 14:18:51 +0200 Subject: Nodes: add redundant name check in debug builds to prevent errors --- source/blender/blenkernel/BKE_derived_node_tree.hh | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'source/blender') 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_; -- cgit v1.2.3