- added new nodes_registered signal.

- fixed bugs in ``NodesPaletteWidget`` widgets.

Author: jchanvfx

NodeGraphQt/base/graph.py

@@ -44,6 +44,14 @@ class NodeGraph(QtCore.QObject):
         :width: 60%
     """
 
+    nodes_registered = QtCore.Signal(list)
+    """
+    Signal triggered when a node is registered into the node graph.
+
+    :parameters: list[:class:`NodeGraphQt.NodeObject`]
+    :emits: registered nodes
+    """
+
     node_created = QtCore.Signal(NodeObject)
     """
     Signal triggered when a node is created in the node graph.
@@ -918,6 +926,7 @@ def register_node(self, node, alias=None):
         """
         self._node_factory.register_node(node, alias)
         self._viewer.rebuild_tab_search()
+        self.nodes_registered.emit([node])
 
     def register_nodes(self, nodes):
         """
@@ -928,6 +937,7 @@ def register_nodes(self, nodes):
         """
         [self._node_factory.register_node(n) for n in nodes]
         self._viewer.rebuild_tab_search()
+        self.nodes_registered.emit(nodes)
 
     def create_node(self, node_type, name=None, selected=True, color=None,
                     text_color=None, pos=None, push_undo=True):

NodeGraphQt/custom_widgets/nodes_palette.py

@@ -146,7 +146,7 @@ def __init__(self, parent=None):
         self.setItemDelegate(NodesGridDelagate(self))
 
     def clear(self):
-        self.model().sourceMode().clear()
+        self.model().sourceModel().clear()
 
     def add_item(self, label, tooltip=''):
         item = QtGui.QStandardItem(label)
@@ -200,21 +200,58 @@ def __init__(self, parent=None, node_graph=None):
 
         self._build_ui()
 
+        # update the ui if new nodes are registered post init.
+        node_graph.nodes_registered.connect(self._on_nodes_registered)
+
     def __repr__(self):
         return '<{} object at {}>'.format(
             self.__class__.__name__, hex(id(self))
         )
 
+    def _on_nodes_registered(self, nodes):
+        """
+        Slot function when a new node has been registered into the node graph.
+
+        Args:
+            nodes (list[NodeObject]): node objects.
+        """
+        node_types = defaultdict(list)
+        for node in nodes:
+            name = node.NODE_NAME
+            node_type = node.type_
+            category = '.'.join(node_type.split('.')[:-1])
+            node_types[category].append((node_type, name))
+
+        update_tabs = False
+        for category, nodes_list in node_types.items():
+            if not update_tabs and category not in self._category_tabs:
+                update_tabs = True
+            grid_view = self._add_category_tab(category)
+            for node_id, node_name in nodes_list:
+                grid_view.add_item(node_name, node_id)
+
+        if update_tabs:
+            self._update_tab_labels()
+
+    def _update_tab_labels(self):
+        """
+        Update the tab labels.
+        """
+        tabs_idx = {self._tab_widget.tabText(x): x
+                    for x in range(self._tab_widget.count())}
+        for category, label in self._custom_labels.items():
+            if category in tabs_idx:
+                idx = tabs_idx[category]
+                self._tab_widget.setTabText(idx, label)
+
     def _build_ui(self):
         """
         populate the ui
         """
-        categories = set()
         node_types = defaultdict(list)
         for name, node_ids in self._factory.names.items():
             for nid in node_ids:
                 category = '.'.join(nid.split('.')[:-1])
-                categories.add(category)
                 node_types[category].append((nid, name))
 
         for category, nodes_list in node_types.items():
@@ -267,13 +304,34 @@ def set_category_label(self, category, label):
                 break
         self._custom_labels[category] = label
 
+    def tab_widget(self):
+        """
+        Get the tab widget.
+
+        Returns:
+            QtWidgets.QTabWidget: tab widget.
+        """
+        return self._tab_widget
+
     def update(self):
         """
         Update and refresh the node palette widget.
         """
-        self._build_tree()
-
+        for category, grid_view in self._category_tabs.items():
+            grid_view.clear()
 
+        node_types = defaultdict(list)
+        for name, node_ids in self._factory.names.items():
+            for nid in node_ids:
+                category = '.'.join(nid.split('.')[:-1])
+                node_types[category].append((nid, name))
 
+        for category, nodes_list in node_types.items():
+            grid_view = self._category_tabs.get(category)
+            if not grid_view:
+                grid_view = self._add_category_tab(category)
 
+            for node_id, node_name in nodes_list:
+                grid_view.add_item(node_name, node_id)
 
+        self._update_tab_labels()