unveil-python 0.2.0__tar.gz → 0.4.0__tar.gz

{unveil_python-0.2.0/src/unveil_python.egg-info → unveil_python-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unveil-python
-Version: 0.2.0
+Version: 0.4.0
 Summary: Implementation of UNVEIL
 Author-email: Nicolas Delinte <nicolas.delinte@uclouvain.be>
 License-Expression: GPL-3.0-only
@@ -25,10 +25,16 @@ Dynamic: license-file
 
 A simple tractography viewer implemented in python.
 
-![image](https://github.com/user-attachments/assets/4fbefa7a-3fc2-446a-8835-2844266b14fa)
+<img width="1297" height="757" alt="image" src="https://github.com/user-attachments/assets/28807e18-17fb-485b-a8ce-2179cdef796b" />
 
-The GUI is launched with the following python command:
+
+The GUI is launched with the following python command in a python console:
 
 ```
 import unveil.__main__
 ```
+or with
+```
+python -c "import unveil.__main__"
+```
+from a terminal with a python environment activated.

{unveil_python-0.2.0 → unveil_python-0.4.0}/README.md

@@ -6,10 +6,16 @@
 
 A simple tractography viewer implemented in python.
 
-![image](https://github.com/user-attachments/assets/4fbefa7a-3fc2-446a-8835-2844266b14fa)
+<img width="1297" height="757" alt="image" src="https://github.com/user-attachments/assets/28807e18-17fb-485b-a8ce-2179cdef796b" />
 
-The GUI is launched with the following python command:
+
+The GUI is launched with the following python command in a python console:
 
 ```
 import unveil.__main__
 ```
+or with
+```
+python -c "import unveil.__main__"
+```
+from a terminal with a python environment activated.

{unveil_python-0.2.0 → unveil_python-0.4.0}/src/unveil/__init__.py

@@ -1,5 +1,5 @@
 """
 This package contains an UNVEIL, a lightweight tractography viewer implemented in python."""
 
-__version__ = "0.2.0"
+__version__ = "0.4.0"
 __author__ = 'Nicolas Delinte'

{unveil_python-0.2.0 → unveil_python-0.4.0}/src/unveil/core.py

@@ -13,12 +13,12 @@ from PyQt6.QtCore import Qt
 from PyQt6.QtGui import QAction
 from PyQt6.QtWidgets import (QApplication, QWidget, QHBoxLayout, QVBoxLayout,
                              QPushButton, QFileDialog, QCheckBox, QSlider,
-                             QLabel, QComboBox, QMainWindow, QLineEdit)
+                             QLabel, QComboBox, QMainWindow, QLineEdit,
+                             QDockWidget, QTreeWidget, QTreeWidgetItem)
 from dipy.io.streamline import load_tractogram
 from matplotlib import pyplot as plt
 from matplotlib.colors import LinearSegmentedColormap
 # from unravel.utils import get_streamline_density
-# from unravel.viz import plot_trk   # We'll use our local version below
 
 
 def gifti_to_pyvista(gii_path):
@@ -27,17 +27,6 @@ def gifti_to_pyvista(gii_path):
     using the inverse of the given NIfTI affine.
     """
 
-    # Affine from MNI152 1mm template (voxel→world)
-    affine = np.array([
-        [-1.0,  0.0,  0.0,  90.0],
-        [0.0,  1.0,  0.0, -126.0],
-        [0.0,  0.0,  1.0,  -72.0],
-        [0.0,  0.0,  0.0,   1.0]
-    ])
-
-    # Compute inverse to go from world (mm) → voxel coordinates
-    inv_affine = np.linalg.inv(affine)
-
     # Load GIFTI file
     img = nib.load(gii_path)
     coords_list = img.get_arrays_from_intent('NIFTI_INTENT_POINTSET')
@@ -51,22 +40,38 @@ def gifti_to_pyvista(gii_path):
     coords_world = coords_list[0].data.astype(np.float32)
     faces = face_list[0].data.astype(np.int32)
 
-    # Transform to voxel space
-    coords_voxel = nib.affines.apply_affine(inv_affine, coords_world)
-
     # Build PyVista face array (VTK expects [3, i1, i2, i3, 3, ...])
     faces_pv = np.hstack(
         (np.full((faces.shape[0], 1), 3, dtype=np.int32), faces)).astype(np.int32)
 
     # Create PolyData mesh in voxel space
-    mesh = pv.PolyData(coords_voxel, faces_pv)
+    mesh = pv.PolyData(coords_world, faces_pv)
 
     return mesh
 
 
+def convert_rgb_array_to_new_basis(rgb_array):
+    """
+    Converts an array of RGB colors shape [n, 3] into a new basis.
+
+    rgb_array: numpy array of shape (n, 3) where each row is [R, G, B]
+    Returns: numpy array of shape (n, 3) where each row is [newR, newG, newB]
+    """
+
+    # Based on the Color-Blind Safe IBM Design Library Color Palette
+    C1 = np.array([220, 38, 127])
+    C2 = np.array([255, 176, 0])
+    C3 = np.array([120, 94, 240])
+
+    B = np.column_stack([C1, C2, C3])
+    new_rgb = rgb_array.dot(B.T)
+
+    return new_rgb
+
+
 def plot_trk(trk_file, scalar=None, color_map='plasma', opacity: float = 1,
              show_points: bool = False, background: str = 'black', plotter=None,
-             name=None, reset_camera=False):
+             name=None, reset_camera=False, color_blind: bool = False):
     '''
     3D render for .trk files.
 
@@ -128,10 +133,12 @@ def plot_trk(trk_file, scalar=None, color_map='plasma', opacity: float = 1,
         point = streamlines._data
         next_point = np.roll(point, -1, axis=0)
         vs = next_point-point
+        ends = (streamlines._offsets+streamlines._lengths-1)
+        if color_blind:
+            vs = convert_rgb_array_to_new_basis(np.abs(vs))
         norm = np.linalg.norm(vs, axis=1)
         norm = np.stack((norm,)*3, axis=1, dtype=np.float32)
         norm = np.divide(vs, norm, dtype=np.float64)
-        ends = (streamlines._offsets+streamlines._lengths-1)
         norm[ends, :] = norm[ends-1, :]
         scalars = np.abs(norm)
         rgb = True
@@ -157,23 +164,22 @@ def plot_trk(trk_file, scalar=None, color_map='plasma', opacity: float = 1,
                    interpolate_before_map=False, render_lines_as_tubes=True,
                    line_width=2, point_size=point_size, rgb=rgb,
                    cmap=color_map, clim=color_lim, scalars=scalars, name=name,
-                   reset_camera=reset_camera)
+                   reset_camera=reset_camera, user_matrix=trk.affine)
 
     elif color_map == 'flesh':
         p.add_mesh(mesh, opacity=opacity, diffuse=0.4, ambient=ambient,
                    interpolate_before_map=False, render_lines_as_tubes=True,
                    line_width=2, point_size=point_size, rgb=rgb,
                    color=[250, 225, 210], name=name,
-                   reset_camera=reset_camera)
+                   reset_camera=reset_camera, user_matrix=trk.affine)
     else:
         p.add_mesh(mesh, opacity=opacity, diffuse=diffuse, ambient=ambient,
                    interpolate_before_map=False, render_lines_as_tubes=True,
                    line_width=2, point_size=point_size, rgb=rgb,
                    cmap=color_map, scalars=scalars, name=name,
-                   reset_camera=reset_camera)
+                   reset_camera=reset_camera, user_matrix=trk.affine)
 
     p.background_color = background
-    # Do not call p.show() here when using an embedded interactor
 
 
 class TrkViewer(QWidget):
@@ -184,12 +190,13 @@ class TrkViewer(QWidget):
         self.nii_data = None
         self.grid = None
         self.names = []
-        self.selected_name = None
         self.x = 0
         self.y = 0
         self.z = 0
         self.initUI()
         self.background = 'white'
+        self.color_blind = False
+        self.flicker = False
 
     def initUI(self):
         # Create the main layout
@@ -257,9 +264,8 @@ class TrkViewer(QWidget):
             'Insert color map name. Name must be in the matplotlib library. Scalar nii.gz must be loaded.')
         control_layout.addWidget(self.colorMapEdit)
 
-        self.showVolumeCheckbox = QCheckBox('Show Volume')
-        self.showVolumeCheckbox.stateChanged.connect(self.update_nii_viewer)
-        control_layout.addWidget(self.showVolumeCheckbox)
+        self.volume_label = QLabel('Volume')
+        control_layout.addWidget(self.volume_label)
 
         self.nii_opacitySlider = QSlider(Qt.Orientation.Horizontal, self)
         self.nii_opacitySlider.setMinimum(0)
@@ -272,27 +278,24 @@ class TrkViewer(QWidget):
         self.showSlicesCheckbox.stateChanged.connect(self.update_nii_viewer)
         control_layout.addWidget(self.showSlicesCheckbox)
 
-        self.showXCheckbox = QCheckBox('X')
-        self.showXCheckbox.stateChanged.connect(self._update_nii_x)
-        control_layout.addWidget(self.showXCheckbox)
+        self.x_label = QLabel('X')
+        control_layout.addWidget(self.x_label)
         self.XSlider = QSlider(Qt.Orientation.Horizontal, self)
         self.XSlider.setMinimum(0)
         self.XSlider.setMaximum(100)
         self.XSlider.setValue(100)
         self.XSlider.sliderReleased.connect(self._update_nii_x)
         control_layout.addWidget(self.XSlider)
-        self.showYCheckbox = QCheckBox('Y')
-        self.showYCheckbox.stateChanged.connect(self._update_nii_y)
-        control_layout.addWidget(self.showYCheckbox)
+        self.y_label = QLabel('Y')
+        control_layout.addWidget(self.y_label)
         self.YSlider = QSlider(Qt.Orientation.Horizontal, self)
         self.YSlider.setMinimum(0)
         self.YSlider.setMaximum(100)
         self.YSlider.setValue(100)
         self.YSlider.sliderReleased.connect(self._update_nii_y)
         control_layout.addWidget(self.YSlider)
-        self.showZCheckbox = QCheckBox('Z')
-        self.showZCheckbox.stateChanged.connect(self._update_nii_z)
-        control_layout.addWidget(self.showZCheckbox)
+        self.z_label = QLabel('Z')
+        control_layout.addWidget(self.z_label)
         self.ZSlider = QSlider(Qt.Orientation.Horizontal, self)
         self.ZSlider.setMinimum(0)
         self.ZSlider.setMaximum(100)
@@ -301,9 +304,8 @@ class TrkViewer(QWidget):
         control_layout.addWidget(self.ZSlider)
 
         # Gifti
-        self.showSurfaceCheckbox = QCheckBox('Show Surface')
-        self.showSurfaceCheckbox.stateChanged.connect(self.update_gii_viewer)
-        control_layout.addWidget(self.showSurfaceCheckbox)
+        self.surface_label = QLabel('Surface')
+        control_layout.addWidget(self.surface_label)
 
         self.gii_opacitySlider = QSlider(Qt.Orientation.Horizontal, self)
         self.gii_opacitySlider.setMinimum(0)
@@ -324,15 +326,19 @@ class TrkViewer(QWidget):
 
     def loadTrkFile(self):
         options = QFileDialog.Options()
-        filePath, _ = QFileDialog.getOpenFileName(
-            self, "Open TRK File", "", "Tractography Files (*.trk)", options=options)
-        if filePath:
-            self.trk_file = filePath
+        filePaths, _ = QFileDialog.getOpenFileNames(self, "Open TRK Files", "",
+                                                    "Tractography Files (*.trk)",
+                                                    options=options)
+        if not filePaths:
+            return
+
+        for trk_path in filePaths:
+            self.trk_file = trk_path
             print(f"Loaded TRK file: {self.trk_file}")
             self.names.append(self.trk_file)
-            self.selected_name = self.trk_file
 
         self.update_trk_viewer(reset_camera=True)
+        self.parent().refreshActorList()
 
     def loadNiftiFile(self):
         options = QFileDialog.Options()
@@ -342,12 +348,10 @@ class TrkViewer(QWidget):
             self.nii_file = filePath
             print(f"Loaded NIfTI file: {self.nii_file}")
             img = nib.load(filePath)
+            self.nii_affine = img.affine
             self.nii_data = img.get_fdata()
             grid = pv.ImageData()
             grid.dimensions = np.array(self.nii_data.shape) + 1
-            # World coordinates
-            # grid.origin = img.affine[:3, 3]
-            # grid.spacing = np.diag(img.affine)[:3]
             grid.cell_data['values'] = self.nii_data.flatten(order='F')
             self.grid = grid
 
@@ -355,7 +359,38 @@ class TrkViewer(QWidget):
         self.YSlider.setMaximum(self.nii_data.shape[1])
         self.ZSlider.setMaximum(self.nii_data.shape[2])
 
-        self.update_nii_viewer(reset_camera=True)
+        self.update_nii_viewer(reset_camera=False)
+        self.parent().refreshActorList()
+
+    def loadROIFile(self):
+        """Load one or multiple ROI nifti volumes and render as surfaces."""
+        options = QFileDialog.Options()
+        filePaths, _ = QFileDialog.getOpenFileNames(self, "Open ROI Files", "",
+                                                    "Nifti Files (*.nii *.nii.gz)",
+                                                    options=options)
+
+        if not filePaths:
+            return
+
+        for roi_path in filePaths:
+            img = nib.load(roi_path)
+            roi = img.get_fdata()
+
+            datapv = pv.wrap(roi)
+            datapv.cell_data['labels'] = roi[:-1, :-1, :-1].flatten(order='F')
+            vol = datapv.threshold(value=1, scalars='labels')
+            mesh = vol.extract_surface()
+
+            smooth = mesh.smooth_taubin(n_iter=12)
+
+            actor_name = f"roi_{roi_path.split('/')[-1]}"
+
+            self.plotter.add_mesh(smooth, name=actor_name, opacity=1,
+                                  color="white", user_matrix=img.affine,
+                                  reset_camera=False, point_size=0,
+                                  render_lines_as_tubes=True)
+
+            self.parent().refreshActorList()
 
     def loadGiftiFile(self):
         options = QFileDialog.Options()
@@ -366,7 +401,8 @@ class TrkViewer(QWidget):
             print(f"Loaded GIfTI file: {self.gii_file}")
             self.gii_mesh = gifti_to_pyvista(filePath)
 
-        self.update_gii_viewer(reset_camera=True)
+        self.update_gii_viewer(reset_camera=False)
+        self.parent().refreshActorList()
 
     def set_background_color(self):
 
@@ -377,25 +413,30 @@ class TrkViewer(QWidget):
 
         self.plotter.background_color = self.background
 
+    def toggleColorBlind(self):
+
+        self.color_blind = not self.color_blind
+        self.update_trk_viewer()
+
     def view_isometric(self):
-        # equivalent to BackgroundPlotter 'iso' camera position
         self.plotter.view_isometric()
-        self.plotter.render()
+        self.plotter.camera.azimuth += self.flicker*180
+        self.flicker = not self.flicker
 
     def view_xy(self):
-        # view from +Z axis, XY plane (axial)
-        self.plotter.view_xy()   # pyvista method
-        self.plotter.render()
+        self.plotter.view_xy()
+        self.plotter.camera.azimuth += self.flicker*180
+        self.flicker = not self.flicker
 
     def view_xz(self):
-        # view from +Y axis, XZ plane (sagittal)
         self.plotter.view_xz()
-        self.plotter.render()
+        self.plotter.camera.azimuth += self.flicker*180
+        self.flicker = not self.flicker
 
     def view_yz(self):
-        # view from +X axis, YZ plane (coronal)
         self.plotter.view_yz()
-        self.plotter.render()
+        self.plotter.camera.azimuth += self.flicker*180
+        self.flicker = not self.flicker
 
     def take_screenshot(self):
         """Take a screenshot of the current 3D view."""
@@ -426,7 +467,7 @@ class TrkViewer(QWidget):
 
         # Copy all meshes and volumes from the main plotter to the offscreen one
         for actor_name, actor in self.plotter.actors.items():
-            offscreen.add_actor(actor.copy())
+            offscreen.add_actor(actor)
 
         # Match background color and camera
         offscreen.background_color = self.plotter.background_color
@@ -446,7 +487,7 @@ class TrkViewer(QWidget):
             self.plotter.render()
         offscreen.close()
 
-    def update_trk_viewer(self, reset_camera=False):
+    def update_trk_viewer(self, reset_camera: bool = False):
 
         opacity = self.opacitySlider.value() / 100.0
         show_points = self.showPointsCheckbox.isChecked()
@@ -467,7 +508,7 @@ class TrkViewer(QWidget):
             plot_trk(file, opacity=opacity, plotter=self.plotter, scalar=scalar,
                      show_points=show_points, color_map=color_map,
                      name=file, background=self.background,
-                     reset_camera=reset_camera)
+                     reset_camera=reset_camera, color_blind=self.color_blind)
 
     def _update_nii_x(self):
 
@@ -476,14 +517,14 @@ class TrkViewer(QWidget):
             center = (self.XSlider.value(),
                       self.YSlider.value(), self.ZSlider.value())
 
-            if self.showXCheckbox.isChecked():
-                slice_x = self.grid.slice('x', center)
-                self.plotter.add_mesh(slice_x, cmap='grey', name='nii_x',
-                                      show_scalar_bar=False, point_size=0,
-                                      render_lines_as_tubes=True,
-                                      reset_camera=False)
-            else:
-                self.plotter.remove_actor('nii_x')
+            slice_x = self.grid.slice('x', center)
+            self.plotter.add_mesh(slice_x, cmap='grey', name='nii_x',
+                                  show_scalar_bar=False, point_size=0,
+                                  render_lines_as_tubes=True,
+                                  reset_camera=False,
+                                  user_matrix=self.nii_affine)
+        else:
+            self.plotter.remove_actor('nii_x')
 
     def _update_nii_y(self):
 
@@ -492,14 +533,14 @@ class TrkViewer(QWidget):
             center = (self.XSlider.value(),
                       self.YSlider.value(), self.ZSlider.value())
 
-            if self.showYCheckbox.isChecked():
-                slice_y = self.grid.slice('y', center)
-                self.plotter.add_mesh(slice_y, cmap='grey', name='nii_y',
-                                      show_scalar_bar=False, point_size=0,
-                                      render_lines_as_tubes=True,
-                                      reset_camera=False)
-            else:
-                self.plotter.remove_actor('nii_y')
+            slice_y = self.grid.slice('y', center)
+            self.plotter.add_mesh(slice_y, cmap='grey', name='nii_y',
+                                  show_scalar_bar=False, point_size=0,
+                                  render_lines_as_tubes=True,
+                                  reset_camera=False,
+                                  user_matrix=self.nii_affine)
+        else:
+            self.plotter.remove_actor('nii_y')
 
     def _update_nii_z(self):
 
@@ -508,48 +549,35 @@ class TrkViewer(QWidget):
             center = (self.XSlider.value(),
                       self.YSlider.value(), self.ZSlider.value())
 
-            if self.showZCheckbox.isChecked():
-                slice_z = self.grid.slice('z', center)
-                self.plotter.add_mesh(slice_z, cmap='grey', name='nii_z',
-                                      show_scalar_bar=False, point_size=0,
-                                      render_lines_as_tubes=True,
-                                      reset_camera=False)
-            else:
-                self.plotter.remove_actor('nii_z')
-
-    def update_nii_viewer(self, reset_camera=False):
-
-        if self.showSlicesCheckbox.isChecked():
-
-            self._update_nii_x()
-            self._update_nii_y()
-            self._update_nii_z()
-
+            slice_z = self.grid.slice('z', center)
+            self.plotter.add_mesh(slice_z, cmap='grey', name='nii_z',
+                                  show_scalar_bar=False, point_size=0,
+                                  render_lines_as_tubes=True,
+                                  reset_camera=False,
+                                  user_matrix=self.nii_affine)
         else:
-            self.plotter.remove_actor('nii_x')
-            self.plotter.remove_actor('nii_y')
             self.plotter.remove_actor('nii_z')
 
-        if self.showVolumeCheckbox.isChecked():
-            opacity = self.nii_opacitySlider.value()/1000
-            self.plotter.add_volume(self.grid, cmap='gray', opacity=[0, opacity],
-                                    show_scalar_bar=False, name='nii_volume',
-                                    reset_camera=reset_camera)
-        else:
-            self.plotter.remove_actor('nii_volume', reset_camera=reset_camera)
+    def update_nii_viewer(self, reset_camera: bool = False):
 
-    def update_gii_viewer(self, reset_camera=False):
+        self._update_nii_x()
+        self._update_nii_y()
+        self._update_nii_z()
 
-        if self.showSurfaceCheckbox.isChecked():
+        opacity = self.nii_opacitySlider.value()/1000
+        self.plotter.add_volume(self.grid, cmap='gray', opacity=[0, opacity],
+                                show_scalar_bar=False, name='nii_volume',
+                                reset_camera=reset_camera,
+                                user_matrix=self.nii_affine)
 
-            opacity = self.gii_opacitySlider.value()/100
-            self.plotter.add_mesh(self.gii_mesh, color="ghostwhite",
-                                  culling='front', smooth_shading=True,
-                                  opacity=opacity, name='gii_surface',
-                                  reset_camera=reset_camera)
-        else:
-            self.plotter.remove_actor('gii_surface', reset_camera=reset_camera)
-            # self.plotter.actors['gii_surface'].visibility = False
+    def update_gii_viewer(self, reset_camera: bool = False):
+
+        opacity = self.gii_opacitySlider.value()/100
+        self.plotter.add_mesh(self.gii_mesh, color="ghostwhite",
+                              culling='back', smooth_shading=True,
+                              opacity=opacity, name='gii_surface',
+                              reset_camera=reset_camera, point_size=0,
+                              render_lines_as_tubes=True,)
 
 
 class MainWindow(QMainWindow):
@@ -557,6 +585,7 @@ class MainWindow(QMainWindow):
         super().__init__()
         self.viewer = TrkViewer()
         self.setCentralWidget(self.viewer)
+        self.initActorDock()
         self.initMenuBar()
         self.setWindowTitle("UNVEIL - Tractography Viewer")
 
@@ -581,16 +610,108 @@ class MainWindow(QMainWindow):
         loadGiftiAction.triggered.connect(self.viewer.loadGiftiFile)
         fileMenu.addAction(loadGiftiAction)
 
+        # Load ROIs
+        loadROIAction = QAction('Load ROIs (.nii/.nii.gz)', self)
+        loadROIAction.triggered.connect(self.viewer.loadROIFile)
+        fileMenu.addAction(loadROIAction)
+
         # Optional: Exit action
         exitAction = QAction('Exit', self)
         exitAction.triggered.connect(self.close)
         fileMenu.addAction(exitAction)
 
         # View menu
-        fileMenu = menubar.addMenu('View')
+        viewMenu = menubar.addMenu('View')
         setBackColorAction = QAction('Background Color', self)
         setBackColorAction.triggered.connect(self.viewer.set_background_color)
-        fileMenu.addAction(setBackColorAction)
+        viewMenu.addAction(setBackColorAction)
+        toggleActorListAction = QAction("Toggle Actor List", self,
+                                        checkable=True)
+        toggleActorListAction.triggered.connect(self.toggleActorDock)
+        viewMenu.addAction(toggleActorListAction)
+        toggleColorBlindAction = QAction("Use color-blind colors", self,
+                                         checkable=True)
+        toggleColorBlindAction.triggered.connect(self.viewer.toggleColorBlind)
+        viewMenu.addAction(toggleColorBlindAction)
+
+    def initActorDock(self):
+        self.actorDock = QDockWidget("Scene Actors", self)
+        self.actorDock.setAllowedAreas(Qt.DockWidgetArea.RightDockWidgetArea)
+
+        self.actorTree = QTreeWidget()
+        self.actorTree.setHeaderLabels(["Actor", "Visible"])
+        self.actorTree.setColumnCount(2)
+        self.actorTree.setColumnWidth(0, 200)
+
+        self.actorDock.setWidget(self.actorTree)
+        self.addDockWidget(
+            Qt.DockWidgetArea.RightDockWidgetArea, self.actorDock)
+
+        self.actorDock.hide()  # start hidden
+
+        # Refresh the list whenever actors change
+        self.viewer.plotter.mesh_added_callback = self.refreshActorList
+        self.viewer.plotter.mesh_removed_callback = self.refreshActorList
+
+    def toggleActorDock(self, checked):
+        if checked:
+
+            self.actorDock.show()
+        else:
+            self.actorDock.hide()
+
+    def refreshActorList(self):
+        """Rebuild the hierarchical actor list grouped by class."""
+        self.actorTree.clear()
+
+        # Groups
+        group_trk = QTreeWidgetItem(["TRK", ""])
+        group_nii = QTreeWidgetItem(["NIfTI Volume", ""])
+        group_roi = QTreeWidgetItem(["ROI Surfaces", ""])
+        group_gii = QTreeWidgetItem(["GIFTI", ""])
+
+        self.actorTree.addTopLevelItem(group_trk)
+        self.actorTree.addTopLevelItem(group_nii)
+        self.actorTree.addTopLevelItem(group_gii)
+        self.actorTree.addTopLevelItem(group_roi)
+        # Loop through PyVista actors
+        for name, actor in self.viewer.plotter.actors.items():
+            item = QTreeWidgetItem([name, ""])
+            item.setCheckState(
+                1, Qt.CheckState.Checked if actor.GetVisibility() else Qt.CheckState.Unchecked)
+
+            # Store name for callback
+            item.actor_name = name
+
+            # Insert into the correct group
+            if name.endswith(".trk"):
+                group_trk.addChild(item)
+            elif name.startswith("nii_"):
+                group_nii.addChild(item)
+            elif name.startswith("gii_"):
+                group_gii.addChild(item)
+            elif name.startswith("roi_"):
+                group_roi.addChild(item)
+            else:
+                group_trk.addChild(item)  # default bucket
+
+        # Connect visibility toggles
+        self.actorTree.itemChanged.connect(self.onActorVisibilityChanged)
+
+    def onActorVisibilityChanged(self, item, column):
+        """Toggle visibility when user clicks checkbox."""
+        if column != 1:
+            return
+
+        actor_name = item.actor_name
+        visible = item.checkState(1) == Qt.CheckState.Checked
+
+        try:
+            self.viewer.plotter.actors[actor_name].SetVisibility(visible)
+        except KeyError:
+            pass
+
+        self.viewer.plotter.render()
 
 
 if __name__ == '__main__':

{unveil_python-0.2.0 → unveil_python-0.4.0/src/unveil_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unveil-python
-Version: 0.2.0
+Version: 0.4.0
 Summary: Implementation of UNVEIL
 Author-email: Nicolas Delinte <nicolas.delinte@uclouvain.be>
 License-Expression: GPL-3.0-only
@@ -25,10 +25,16 @@ Dynamic: license-file
 
 A simple tractography viewer implemented in python.
 
-![image](https://github.com/user-attachments/assets/4fbefa7a-3fc2-446a-8835-2844266b14fa)
+<img width="1297" height="757" alt="image" src="https://github.com/user-attachments/assets/28807e18-17fb-485b-a8ce-2179cdef796b" />
 
-The GUI is launched with the following python command:
+
+The GUI is launched with the following python command in a python console:
 
 ```
 import unveil.__main__
 ```
+or with
+```
+python -c "import unveil.__main__"
+```
+from a terminal with a python environment activated.