add button node_prop && fix autoCook bug

Author: ArnoChenFx

NodeGraphQt/constants.py

@@ -81,6 +81,8 @@
 NODE_PROP_FLOAT = 14
 #: Property type represented with int widget in the properties bin.
 NODE_PROP_INT = 15
+#: Property type represented with button widget in the properties bin.
+NODE_PROP_BUTTON = 16
 
 # === NODE VIEWER ===
 

NodeGraphQt/widgets/properties.py

@@ -15,7 +15,8 @@
                                    NODE_PROP_VECTOR3,
                                    NODE_PROP_VECTOR4,
                                    NODE_PROP_FLOAT,
-                                   NODE_PROP_INT)
+                                   NODE_PROP_INT,
+                                   NODE_PROP_BUTTON)
 from NodeGraphQt.widgets.file_dialog import file_dialog
 
 
@@ -425,7 +426,6 @@ def mouseMoveEvent(self, event):
                 value = self.pre_val + int(delta*self._speed) * self._step
                 self.setValue(value)
 
-
         super(_valueEdit,self).mouseMoveEvent(event)
 
     def mousePressEvent(self, event):
@@ -662,6 +662,21 @@ def __init__(self, parent=None):
         self.set_data_type(int)
 
 
+class PropButton(QtWidgets.QPushButton):
+    value_changed = QtCore.Signal(str, object)
+
+    def __init__(self, parent=None):
+        super(PropButton, self).__init__(parent)
+
+    def set_value(self, value):
+        # value: list of functions
+        for func in value:
+            self.clicked.connect(func)
+
+    def get_value(self):
+        return None
+
+
 WIDGET_MAP = {
     NODE_PROP_QLABEL: PropLabel,
     NODE_PROP_QLINEEDIT: PropLineEdit,
@@ -677,6 +692,7 @@ def __init__(self, parent=None):
     NODE_PROP_VECTOR4: PropVector4,
     NODE_PROP_FLOAT: PropFloat,
     NODE_PROP_INT: PropInt,
+    NODE_PROP_BUTTON: PropButton
 }
 
 
@@ -716,11 +732,14 @@ def add_widget(self, name, widget, value, label=None):
         if row > 0:
             row += 1
 
+        label = QtWidgets.QLabel(label)
         label_flags = QtCore.Qt.AlignCenter | QtCore.Qt.AlignRight
         if widget.__class__.__name__ == 'PropTextEdit':
             label_flags = label_flags | QtCore.Qt.AlignTop
-
-        self.__layout.addWidget(QtWidgets.QLabel(label), row, 0, label_flags)
+        elif widget.__class__.__name__ == 'PropButton':
+            label.setVisible(False)
+            widget.setText(name)
+        self.__layout.addWidget(label, row, 0, label_flags)
         self.__layout.addWidget(widget, row, 1)
 
     def get_widget(self, name):

example_auto_nodes/node_base/auto_node.py

@@ -4,7 +4,7 @@
 from NodeGraphQt import QtCore
 import random
 import copy
-
+import time
 
 def rand_color(seed_type):
     seed = id(seed_type)
@@ -35,6 +35,9 @@ def __init__(self, defaultInputType=None, defaultOutputType=None):
         self.defaultInputType = defaultInputType
         self.defaultOutputType = defaultOutputType
 
+        self._cookTime = 0.0
+        self._toolTip = self._setup_tool_tip()
+
     @property
     def autoCook(self):
         return self._autoCook
@@ -51,6 +54,15 @@ def autoCook(self, mode):
             self.defaultColor = self.get_property("color")
             self.set_property('color', self.stopCookColor)
 
+    @property
+    def cookTime(self):
+        return self._cookTime
+
+    @autoCook.setter
+    def cookTime(self, time):
+        self._cookTime = time
+        self._update_tool_tip()
+
     def cookNextNode(self):
         for nodeList in self.connected_output_nodes().values():
             for n in nodeList:
@@ -80,6 +92,11 @@ def getInputData(self, port):
             data = from_port.node().get_property(from_port.name())
             return copy.deepcopy(data)
 
+    def when_disabled(self):
+        num = len(self.input_ports())
+        for index, out_port in enumerate(self.output_ports()):
+            self.set_property(out_port.name(), self.getInputData(index % num))
+
     def cook(self, forceCook=False):
         if not self._autoCook and forceCook is not True:
             return
@@ -88,18 +105,20 @@ def cook(self, forceCook=False):
         self._autoCook = False
 
         if self.disabled():
-            num = len(self.input_ports())
-            for index, out_port in enumerate(self.output_ports()):
-                self.set_property(out_port.name(), self.getInputData(index % num))
+            self._autoCook = _tmp
+            self.when_disabled()
             self.cookNextNode()
             return
 
         if not self.needCook:
+            self._autoCook = _tmp
             return
 
         if self.error():
             self._close_error()
 
+        _start_time = time.time()
+
         try:
             self.run()
         except Exception as error:
@@ -109,6 +128,8 @@ def cook(self, forceCook=False):
         if self.error():
             return
 
+        self.cookTime = time.time() - _start_time
+
         self.cooked.emit()
         self.cookNextNode()
 
@@ -206,18 +227,30 @@ def add_output(self, name='output', data_type=None, multi_output=True, display_n
     def _close_error(self):
         self._error = False
         self.set_property('color', self.defaultColor)
-        self._view._tooltip_disable(False)
+        self._update_tool_tip()
 
     def _show_error(self, message):
         if not self._error:
             self.defaultColor = self.get_property("color")
 
         self._error = True
         self.set_property('color', self.errorColor)
-        tooltip = '<b>{}</b>'.format(self.name())
-        tooltip += ' <font color="red"><br>({})</br></font>'.format(message)
-        tooltip += '<br/>{}<br/>'.format(self._view.type_)
-        self._view.setToolTip(tooltip)
+        tooltip = '<font color="red"><br>({})</br></font>'.format(message)
+        self._update_tool_tip(tooltip)
+
+    def _update_tool_tip(self, message = None):
+        if message is None:
+            tooltip = self._toolTip.format(self._cookTime)
+        else:
+            tooltip = '<b>{}</b>'.format(self.name())
+            tooltip += message
+            tooltip += '<br/>{}<br/>'.format(self._view.type_)
+        self.view.setToolTip(tooltip)
+        return tooltip
+
+    def _setup_tool_tip(self):
+        tooltip = '<br> last cook used: {}s</br>'
+        return self._update_tool_tip(tooltip)
 
     def error(self, message=None):
         if message is None: