[Bf-blender-cvs] [1b90194608a] functions: allow declarations to draw in node

[Jacques Lucke]

Commit: 1b90194608a15bb284aeb0c33c4fd5250f9be5f7
Author: Jacques Lucke
Date:   Wed Mar 20 18:16:55 2019 +0100
Branches: functions
https://developer.blender.org/rB1b90194608a15bb284aeb0c33c4fd5250f9be5f7

allow declarations to draw in node

===================================================================

M	release/scripts/startup/function_nodes/base.py
M	release/scripts/startup/function_nodes/socket_decl.py

===================================================================

diff --git a/release/scripts/startup/function_nodes/base.py b/release/scripts/startup/function_nodes/base.py
index 8fc36353c53..397446c36ce 100644
--- a/release/scripts/startup/function_nodes/base.py
+++ b/release/scripts/startup/function_nodes/base.py
@@ -105,6 +105,8 @@ class BaseNode:
 
     def draw_buttons(self, context, layout):
         self.draw(layout)
+        for decl in self.storage.sockets_per_decl.keys():
+            decl.draw_node(layout, self)
 
     def draw(self, layout):
         pass
diff --git a/release/scripts/startup/function_nodes/socket_decl.py b/release/scripts/startup/function_nodes/socket_decl.py
index c04adfc0ed8..1c4229f12eb 100644
--- a/release/scripts/startup/function_nodes/socket_decl.py
+++ b/release/scripts/startup/function_nodes/socket_decl.py
@@ -12,6 +12,9 @@ class SocketDeclBase:
     def amount(self, node):
         raise NotImplementedError()
 
+    def draw_node(self, layout, node):
+        pass
+
     def draw_socket(self, layout, node, socket, index):
         socket.draw_self(layout, self)