Functions: Support getting MFSocket based on its id

1 files changed, 13 insertions, 0 deletions

diff --git a/source/blender/functions/FN_multi_function_network.hh b/source/blender/functions/FN_multi_function_network.hh
index 195711a7c62..493b26e6510 100644
--- a/source/blender/functions/FN_multi_function_network.hh
+++ b/source/blender/functions/FN_multi_function_network.hh
@@ -226,6 +226,9 @@ class MFNetwork : NonCopyable, NonMovable {
   MFNode *node_or_null_by_id(uint id);
   const MFNode *node_or_null_by_id(uint id) const;
 
+  MFSocket *socket_or_null_by_id(uint id);
+  const MFSocket *socket_or_null_by_id(uint id) const;
+
   std::string to_dot(Span<const MFNode *> marked_nodes = {}) const;
 };
 
@@ -505,6 +508,16 @@ inline const MFNode *MFNetwork::node_or_null_by_id(uint id) const
   return node_or_null_by_id_[id];
 }
 
+inline MFSocket *MFNetwork::socket_or_null_by_id(uint id)
+{
+  return socket_or_null_by_id_[id];
+}
+
+inline const MFSocket *MFNetwork::socket_or_null_by_id(uint id) const
+{
+  return socket_or_null_by_id_[id];
+}
+
 inline uint MFNetwork::socket_id_amount() const
 {
   return socket_or_null_by_id_.size();