add register custom prop widget

Author: ArnoChenFx

NodeGraphQt/widgets/properties.py

@@ -324,7 +324,9 @@ def _on_select_file(self):
         if file:
             self.set_value(file)
 
-    def _on_value_change(self, value):
+    def _on_value_change(self, value=None):
+        if value is None:
+            value = self._ledit.text()
         self.value_changed.emit(self.toolTip(), value)
 
     def get_value(self):
@@ -708,6 +710,12 @@ def get_value(self):
 }
 
 
+def registerPropType(name, prop_class, override=False):
+    global WIDGET_MAP
+    if name in WIDGET_MAP.keys() and not override:
+        raise Exception("Prop type {} has already exists, u can use override=True to override)".format(name))
+    WIDGET_MAP[name] = prop_class
+
 # main property widgets.
 
 

NodeGraphQt/widgets/properties_bin.py

@@ -42,12 +42,15 @@ def __init__(self, parent=None):
         self.setItemDelegate(PropertiesDelegate())
         self.setColumnCount(1)
         self.setShowGrid(False)
-        QtCompat.QHeaderView.setSectionResizeMode(
-            self.verticalHeader(), QtWidgets.QHeaderView.ResizeToContents)
         self.verticalHeader().hide()
-        QtCompat.QHeaderView.setSectionResizeMode(
-            self.horizontalHeader(), 0, QtWidgets.QHeaderView.Stretch)
         self.horizontalHeader().hide()
+        self.horizontalHeader().setResizeMode(QtWidgets.QHeaderView.Stretch)
+        self.verticalHeader().setResizeMode(QtWidgets.QHeaderView.Stretch)
+        self.setVerticalScrollMode(QtWidgets.QAbstractItemView.ScrollPerPixel)
+
+    def wheelEvent(self, event):
+        delta = event.delta() * 0.2
+        self.verticalScrollBar().setValue(self.verticalScrollBar().value() - delta)
 
 
 class PropertiesBinWidget(QtWidgets.QWidget):