diff --git a/source/blender/blenkernel/BKE_derived_node_tree.hh b/source/blender/blenkernel/BKE_derived_node_tree.hh index b5ced6ab772..84b36f675e0 100644 --- a/source/blender/blenkernel/BKE_derived_node_tree.hh +++ b/source/blender/blenkernel/BKE_derived_node_tree.hh @@ -69,6 +69,7 @@ class DSocket : NonCopyable, NonMovable { StringRefNull name() const; const SocketRef &socket_ref() const; + bNodeSocket *bsocket() const; bool is_available() const; }; @@ -111,6 +112,7 @@ class DGroupInput : NonCopyable, NonMovable { public: const InputSocketRef &socket_ref() const; + bNodeSocket *bsocket() const; const DParentNode *parent() const; Span linked_sockets() const; uint id() const; @@ -291,6 +293,11 @@ inline const SocketRef &DSocket::socket_ref() const return *m_socket_ref; } +inline bNodeSocket *DSocket::bsocket() const +{ + return m_socket_ref->bsocket(); +} + inline bool DSocket::is_available() const { return (m_socket_ref->bsocket()->flag & SOCK_UNAVAIL) == 0; @@ -343,6 +350,11 @@ inline const InputSocketRef &DGroupInput::socket_ref() const return *m_socket_ref; } +inline bNodeSocket *DGroupInput::bsocket() const +{ + return m_socket_ref->bsocket(); +} + inline const DParentNode *DGroupInput::parent() const { return m_parent;