Library tree filled with items

src/bedit_gui/controllers/library_controller.py

@@ -0,0 +1,48 @@
+from PySide6.QtCore import QObject, QSize
+from PySide6.QtWidgets import QAbstractItemView, QHeaderView
+
+from bedit_core.models import Component, ComponentID, GraphImplementation
+from bedit_gui.models import Icon, IconDatabase
+from bedit_gui.services.application_settings import ApplicationSettings
+from bedit_gui.services.libraries import load_library_documents
+from bedit_gui.utils.icon import render_fitted_icon
+from bedit_gui.views.main_window import MainWindow
+from bedit_gui.views.models.document_tree_model import DocumentTreeModel
+
+ICON_SIZE = QSize(32, 32)
+
+
+class LibraryController(QObject):
+    def __init__(self, window: MainWindow, settings: ApplicationSettings) -> None:
+        super().__init__(window)
+        self.window = window
+        self.settings = settings
+        self.model = DocumentTreeModel(editable=False)
+
+        tree = window.ui.libraryTree
+        tree.setModel(self.model)
+        tree.setHeaderHidden(True)
+        tree.setSelectionMode(QAbstractItemView.SelectionMode.ExtendedSelection)
+        tree.setSelectionBehavior(QAbstractItemView.SelectionBehavior.SelectRows)
+        tree.setIconSize(QSize(48, 48))
+        tree.header().setStretchLastSection(False)
+        tree.header().setSectionResizeMode(0, QHeaderView.ResizeMode.Stretch)
+        tree.header().setSectionResizeMode(1, QHeaderView.ResizeMode.Fixed)
+        tree.setColumnWidth(1, 56)
+
+        self.reload()
+
+    def reload(self) -> None:
+        libraries = load_library_documents(self.settings.library_paths)
+        self.model.set_documents([library.document for library in libraries])
+        for library in libraries:
+            database = library.document.metadata.get("icon_database") if library.document.metadata is not None else None
+            icons = database.icons if isinstance(database, IconDatabase) else {}
+            self._set_component_icons(library.document.root, icons)
+        self.window.ui.libraryTree.expandAll()
+
+    def _set_component_icons(self, components: dict[ComponentID, Component], icons: dict[ComponentID, Icon]) -> None:
+        for component_id, component in components.items():
+            self.model.set_component_icon(component_id, render_fitted_icon(icons.get(component_id, Icon()), component.interface.ports, ICON_SIZE))
+            if isinstance(component.implementation, GraphImplementation):
+                self._set_component_icons(component.implementation.graph.components, icons)

src/bedit_gui/controllers/settings_controller.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from collections.abc import Callable
 from typing import Protocol
 
-from PySide6.QtCore import QObject
+from PySide6.QtCore import QObject, Signal
 from PySide6.QtWidgets import QDialog
 
 from bedit_gui.services.application_logging import get_logger, set_log_level
@@ -33,6 +33,8 @@ SettingsDialogFactory = Callable[[int, int, list[str], MainWindow], SettingsDial
 class SettingsController(QObject):
     """Opens the settings dialog and applies accepted preferences."""
 
+    library_paths_changed = Signal()
+
     def __init__(
         self,
         window: MainWindow,
@@ -51,9 +53,12 @@ class SettingsController(QObject):
         if dialog.exec() != QDialog.DialogCode.Accepted:
             return
 
+        old_library_paths = self.settings.library_paths
         self.settings.log_level = dialog.log_level
         self.settings.snap_to_grid_size = dialog.snap_to_grid_size
         self.settings.library_paths = dialog.library_paths
         self.window.graph_editor.set_snap_to_grid_size(dialog.snap_to_grid_size)
         set_log_level(dialog.log_level)
+        if self.settings.library_paths != old_library_paths:
+            self.library_paths_changed.emit()
         logger.info("Application settings updated")