Nodes: provide access to type specific data through node tree ref

1 files changed, 26 insertions, 0 deletions

diff --git a/source/blender/nodes/NOD_node_tree_ref.hh b/source/blender/nodes/NOD_node_tree_ref.hh
index 6b8eb21bf9a..b9c5d41658a 100644
--- a/source/blender/nodes/NOD_node_tree_ref.hh
+++ b/source/blender/nodes/NOD_node_tree_ref.hh
@@ -116,6 +116,9 @@ class SocketRef : NonCopyable, NonMovable {
   bNodeTree *btree() const;
 
   bool is_available() const;
+
+  void *default_value() const;
+  template<typename T> T *default_value() const;
 };
 
 class InputSocketRef final : public SocketRef {
@@ -169,6 +172,9 @@ class NodeRef : NonCopyable, NonMovable {
   bool is_group_input_node() const;
   bool is_group_output_node() const;
   bool is_muted() const;
+
+  void *storage() const;
+  template<typename T> T *storage() const;
 };
 
 class LinkRef : NonCopyable, NonMovable {
@@ -379,6 +385,16 @@ inline bool SocketRef::is_available() const
   return (bsocket_->flag & SOCK_UNAVAIL) == 0;
 }
 
+inline void *SocketRef::default_value() const
+{
+  return bsocket_->default_value;
+}
+
+template<typename T> inline T *SocketRef::default_value() const
+{
+  return (T *)bsocket_->default_value;
+}
+
 /* --------------------------------------------------------------------
  * InputSocketRef inline methods.
  */
@@ -507,6 +523,16 @@ inline bool NodeRef::is_muted() const
   return (bnode_->flag & NODE_MUTED) != 0;
 }
 
+inline void *NodeRef::storage() const
+{
+  return bnode_->storage;
+}
+
+template<typename T> inline T *NodeRef::storage() const
+{
+  return (T *)bnode_->storage;
+}
+
 /* --------------------------------------------------------------------
  * LinkRef inline methods.
  */