mosamatic2 2.0.24__py3-none-any.whl

mosamatic2/ui/widgets/panels/tasks/selectslicefromscanstaskpanel.py

@@ -0,0 +1,193 @@
+import os
+
+from PySide6.QtWidgets import (
+    QLineEdit,
+    QCheckBox,
+    QComboBox,
+    QHBoxLayout,
+    QVBoxLayout,
+    QFormLayout,
+    QPushButton,
+    QFileDialog,
+    QMessageBox,
+)
+from PySide6.QtCore import (
+    QThread, 
+    Slot,
+)
+
+from mosamatic2.core.managers.logmanager import LogManager
+from mosamatic2.ui.widgets.panels.tasks.taskpanel import TaskPanel
+from mosamatic2.ui.settings import Settings
+from mosamatic2.ui.utils import is_macos
+from mosamatic2.ui.worker import Worker
+from mosamatic2.core.tasks import SelectSliceFromScansTask
+
+LOG = LogManager()
+
+PANEL_TITLE = 'SelectSliceFromScansTaskPanel'
+PANEL_NAME = 'selectslicefromscanstaskpanel'
+
+
+class SelectSliceFromScansTaskPanel(TaskPanel):
+    def __init__(self):
+        super(SelectSliceFromScansTaskPanel, self).__init__()
+        self.set_title(PANEL_TITLE)
+        self._scans_dir_line_edit = None
+        self._scans_dir_select_button = None
+        self._vertebra_combobox = None
+        self._output_dir_line_edit = None
+        self._output_dir_select_button = None
+        self._overwrite_checkbox = None
+        self._form_layout = None
+        self._run_task_button = None
+        self._settings = None
+        self._task = None
+        self._worker = None
+        self._thread = None
+        self.init_layout()
+
+    def scans_dir_line_edit(self):
+        if not self._scans_dir_line_edit:
+            self._scans_dir_line_edit = QLineEdit(self.settings().get(f'{PANEL_NAME}/scans_dir'))
+        return self._scans_dir_line_edit
+    
+    def scans_dir_select_button(self):
+        if not self._scans_dir_select_button:
+            self._scans_dir_select_button = QPushButton('Select')
+            self._scans_dir_select_button.clicked.connect(self.handle_scans_dir_select_button)
+        return self._scans_dir_select_button
+    
+    def vertebra_combobox(self):
+        if not self._vertebra_combobox:
+            self._vertebra_combobox = QComboBox()
+            self._vertebra_combobox.addItems(['L3'])
+            self._vertebra_combobox.setCurrentText(self.settings().get(f'{PANEL_NAME}/vertebra'))
+        return self._vertebra_combobox
+    
+    def output_dir_line_edit(self):
+        if not self._output_dir_line_edit:
+            self._output_dir_line_edit = QLineEdit(self.settings().get(f'{PANEL_NAME}/output_dir'))
+        return self._output_dir_line_edit
+    
+    def output_dir_select_button(self):
+        if not self._output_dir_select_button:
+            self._output_dir_select_button = QPushButton('Select')
+            self._output_dir_select_button.clicked.connect(self.handle_output_dir_select_button)
+        return self._output_dir_select_button
+    
+    def overwrite_checkbox(self):
+        if not self._overwrite_checkbox:
+            self._overwrite_checkbox = QCheckBox('')
+            self._overwrite_checkbox.setChecked(self.settings().get_bool(f'{PANEL_NAME}/overwrite', True))
+        return self._overwrite_checkbox
+    
+    def form_layout(self):
+        if not self._form_layout:
+            self._form_layout = QFormLayout()
+            if is_macos():
+                self._form_layout.setFieldGrowthPolicy(QFormLayout.ExpandingFieldsGrow)
+        return self._form_layout
+    
+    def run_task_button(self):
+        if not self._run_task_button:
+            self._run_task_button = QPushButton('Run task')
+            self._run_task_button.clicked.connect(self.handle_run_task_button)
+        return self._run_task_button
+    
+    def settings(self):
+        if not self._settings:
+            self._settings = Settings()
+        return self._settings
+    
+    def init_layout(self):
+        scans_dir_layout = QHBoxLayout()
+        scans_dir_layout.addWidget(self.scans_dir_line_edit())
+        scans_dir_layout.addWidget(self.scans_dir_select_button())
+        output_dir_layout = QHBoxLayout()
+        output_dir_layout.addWidget(self.output_dir_line_edit())
+        output_dir_layout.addWidget(self.output_dir_select_button())
+        self.form_layout().addRow('Scans directory', scans_dir_layout)
+        # self.form_layout().addRow('Vertebra', self.vertebra_combobox())
+        self.form_layout().addRow('Output directory', output_dir_layout)
+        self.form_layout().addRow('Overwrite', self.overwrite_checkbox())
+        layout = QVBoxLayout()
+        layout.addLayout(self.form_layout())
+        layout.addWidget(self.run_task_button())
+        self.setLayout(layout)
+        self.setObjectName(PANEL_NAME)
+
+    def handle_scans_dir_select_button(self):
+        last_directory = self.settings().get('last_directory')
+        directory = QFileDialog.getExistingDirectory(dir=last_directory)
+        if directory:
+            self.scans_dir_line_edit().setText(directory)
+            self.settings().set('last_directory', directory)
+
+    def handle_output_dir_select_button(self):
+        last_directory = self.settings().get('last_directory')
+        directory = QFileDialog.getExistingDirectory(dir=last_directory)
+        if directory:
+            self.output_dir_line_edit().setText(directory)
+            self.settings().set('last_directory', directory)
+
+    def handle_run_task_button(self):
+        errors = self.check_inputs_and_parameters()
+        if len(errors) > 0:
+            error_message = 'Following errors were encountered:\n'
+            for error in errors:
+                error_message += f' - {error}\n'
+            QMessageBox.information(self, 'Error', error_message)
+        else:
+            LOG.info('Running task...')
+            self.run_task_button().setEnabled(False)
+            self.save_inputs_and_parameters()
+            self._task = SelectSliceFromScansTask(
+                inputs={'scans': self.scans_dir_line_edit().text()},
+                params={'vertebra': 'L3'},
+                output=self.output_dir_line_edit().text(),
+                overwrite=self.overwrite_checkbox().isChecked(),
+            )
+            self._worker = Worker(self._task)
+            self._thread = QThread()
+            self._worker.moveToThread(self._thread)
+            self._thread.started.connect(self._worker.run)
+            self._worker.progress.connect(self.handle_progress)
+            self._worker.status.connect(self.handle_status)
+            self._worker.finished.connect(self.handle_finished)
+            self._worker.finished.connect(self._thread.quit)
+            self._worker.finished.connect(self._worker.deleteLater)
+            self._thread.finished.connect(self._thread.deleteLater)
+            self._thread.start()
+
+    @Slot(int)
+    def handle_progress(self, progress):
+        LOG.info(f'Progress: {progress} / 100%')
+
+    @Slot(str)
+    def handle_status(self, status):
+        LOG.info(f'Status: {status}')
+
+    @Slot()
+    def handle_finished(self):
+        self.run_task_button().setEnabled(True)
+
+    # HELPERS
+
+    def check_inputs_and_parameters(self):
+        errors = []
+        if self.scans_dir_line_edit().text() == '':
+            errors.append('Empty scans directory path')
+        if not os.path.isdir(self.scans_dir_line_edit().text()):
+            errors.append('Scans directory does not exist')
+        if self.output_dir_line_edit().text() == '':
+            errors.append('Empty output directory path')
+        if os.path.isdir(self.output_dir_line_edit().text()) and not self.overwrite_checkbox().isChecked():
+            errors.append('Output directory exists but overwrite=False. Please remove output directory first')
+        return errors
+    
+    def save_inputs_and_parameters(self):
+        self.settings().set(f'{PANEL_NAME}/scans_dir', self.scans_dir_line_edit().text())
+        self.settings().set(f'{PANEL_NAME}/vertebra', self.vertebra_combobox().currentText())
+        self.settings().set(f'{PANEL_NAME}/output_dir', self.output_dir_line_edit().text())
+        self.settings().set(f'{PANEL_NAME}/overwrite', self.overwrite_checkbox().isChecked())

mosamatic2/ui/widgets/panels/tasks/taskpanel.py

@@ -0,0 +1,6 @@
+from mosamatic2.ui.widgets.panels.defaultpanel import DefaultPanel
+
+
+class TaskPanel(DefaultPanel):
+    def __init__(self):
+        super(TaskPanel, self).__init__()

mosamatic2/ui/widgets/panels/tasks/totalsegmentatortaskpanel.py

@@ -0,0 +1,195 @@
+import os
+
+from PySide6.QtWidgets import (
+    QLineEdit,
+    QCheckBox,
+    QComboBox,
+    QHBoxLayout,
+    QVBoxLayout,
+    QFormLayout,
+    QPushButton,
+    QFileDialog,
+    QMessageBox,
+)
+from PySide6.QtCore import (
+    QThread, 
+    Slot,
+)
+
+from mosamatic2.core.managers.logmanager import LogManager
+from mosamatic2.ui.widgets.panels.tasks.taskpanel import TaskPanel
+from mosamatic2.ui.settings import Settings
+from mosamatic2.ui.utils import is_macos
+from mosamatic2.ui.worker import Worker
+from mosamatic2.core.tasks import TotalSegmentatorTask
+
+LOG = LogManager()
+
+PANEL_TITLE = 'TotalSegmentatorTaskPanel'
+PANEL_NAME = 'totalsegmentatortaskpanel'
+
+
+class TotalSegmentatorTaskPanel(TaskPanel):
+    def __init__(self):
+        super(TotalSegmentatorTaskPanel, self).__init__()
+        self.set_title(PANEL_TITLE)
+        self._scans_dir_line_edit = None
+        self._scans_dir_select_button = None
+        self._tasks_line_edit = None
+        self._output_dir_line_edit = None
+        self._output_dir_select_button = None
+        self._overwrite_checkbox = None
+        self._form_layout = None
+        self._run_task_button = None
+        self._settings = None
+        self._task = None
+        self._worker = None
+        self._thread = None
+        self.init_layout()
+
+    def scans_dir_line_edit(self):
+        if not self._scans_dir_line_edit:
+            self._scans_dir_line_edit = QLineEdit(self.settings().get(f'{PANEL_NAME}/scans_dir'))
+        return self._scans_dir_line_edit
+    
+    def scans_dir_select_button(self):
+        if not self._scans_dir_select_button:
+            self._scans_dir_select_button = QPushButton('Select')
+            self._scans_dir_select_button.clicked.connect(self.handle_scans_dir_select_button)
+        return self._scans_dir_select_button
+    
+    def tasks_line_edit(self):
+        if not self._tasks_line_edit:
+            self._tasks_line_edit = QLineEdit(self.settings().get(f'{PANEL_NAME}/tasks'))
+            if self._tasks_line_edit.text() == '':
+                self._tasks_line_edit.setText('total')
+        return self._tasks_line_edit
+    
+    def output_dir_line_edit(self):
+        if not self._output_dir_line_edit:
+            self._output_dir_line_edit = QLineEdit(self.settings().get(f'{PANEL_NAME}/output_dir'))
+        return self._output_dir_line_edit
+    
+    def output_dir_select_button(self):
+        if not self._output_dir_select_button:
+            self._output_dir_select_button = QPushButton('Select')
+            self._output_dir_select_button.clicked.connect(self.handle_output_dir_select_button)
+        return self._output_dir_select_button
+    
+    def overwrite_checkbox(self):
+        if not self._overwrite_checkbox:
+            self._overwrite_checkbox = QCheckBox('')
+            self._overwrite_checkbox.setChecked(self.settings().get_bool(f'{PANEL_NAME}/overwrite', True))
+        return self._overwrite_checkbox
+    
+    def form_layout(self):
+        if not self._form_layout:
+            self._form_layout = QFormLayout()
+            if is_macos():
+                self._form_layout.setFieldGrowthPolicy(QFormLayout.ExpandingFieldsGrow)
+        return self._form_layout
+    
+    def run_task_button(self):
+        if not self._run_task_button:
+            self._run_task_button = QPushButton('Run task')
+            self._run_task_button.clicked.connect(self.handle_run_task_button)
+        return self._run_task_button
+    
+    def settings(self):
+        if not self._settings:
+            self._settings = Settings()
+        return self._settings
+    
+    def init_layout(self):
+        scans_dir_layout = QHBoxLayout()
+        scans_dir_layout.addWidget(self.scans_dir_line_edit())
+        scans_dir_layout.addWidget(self.scans_dir_select_button())
+        output_dir_layout = QHBoxLayout()
+        output_dir_layout.addWidget(self.output_dir_line_edit())
+        output_dir_layout.addWidget(self.output_dir_select_button())
+        self.form_layout().addRow('Scans directory', scans_dir_layout)
+        self.form_layout().addRow('Tasks', self.tasks_line_edit())
+        self.form_layout().addRow('Output directory', output_dir_layout)
+        self.form_layout().addRow('Overwrite', self.overwrite_checkbox())
+        layout = QVBoxLayout()
+        layout.addLayout(self.form_layout())
+        layout.addWidget(self.run_task_button())
+        self.setLayout(layout)
+        self.setObjectName(PANEL_NAME)
+
+    def handle_scans_dir_select_button(self):
+        last_directory = self.settings().get('last_directory')
+        directory = QFileDialog.getExistingDirectory(dir=last_directory)
+        if directory:
+            self.scans_dir_line_edit().setText(directory)
+            self.settings().set('last_directory', directory)
+
+    def handle_output_dir_select_button(self):
+        last_directory = self.settings().get('last_directory')
+        directory = QFileDialog.getExistingDirectory(dir=last_directory)
+        if directory:
+            self.output_dir_line_edit().setText(directory)
+            self.settings().set('last_directory', directory)
+
+    def handle_run_task_button(self):
+        errors = self.check_inputs_and_parameters()
+        if len(errors) > 0:
+            error_message = 'Following errors were encountered:\n'
+            for error in errors:
+                error_message += f' - {error}\n'
+            QMessageBox.information(self, 'Error', error_message)
+        else:
+            LOG.info('Running task...')
+            self.run_task_button().setEnabled(False)
+            self.save_inputs_and_parameters()
+            self._task = TotalSegmentatorTask(
+                inputs={'scans': self.scans_dir_line_edit().text()},
+                params={'tasks': self.tasks_line_edit().text()},
+                output=self.output_dir_line_edit().text(),
+                overwrite=self.overwrite_checkbox().isChecked(),
+            )
+            self._worker = Worker(self._task)
+            self._thread = QThread()
+            self._worker.moveToThread(self._thread)
+            self._thread.started.connect(self._worker.run)
+            self._worker.progress.connect(self.handle_progress)
+            self._worker.status.connect(self.handle_status)
+            self._worker.finished.connect(self.handle_finished)
+            self._worker.finished.connect(self._thread.quit)
+            self._worker.finished.connect(self._worker.deleteLater)
+            self._thread.finished.connect(self._thread.deleteLater)
+            self._thread.start()
+
+    @Slot(int)
+    def handle_progress(self, progress):
+        LOG.info(f'Progress: {progress} / 100%')
+
+    @Slot(str)
+    def handle_status(self, status):
+        LOG.info(f'Status: {status}')
+
+    @Slot()
+    def handle_finished(self):
+        self.run_task_button().setEnabled(True)
+
+    # HELPERS
+
+    def check_inputs_and_parameters(self):
+        errors = []
+        if self.scans_dir_line_edit().text() == '':
+            errors.append('Empty scans directory path')
+        if not os.path.isdir(self.scans_dir_line_edit().text()):
+            errors.append('Scans directory does not exist')
+        if self.output_dir_line_edit().text() == '':
+            errors.append('Empty output directory path')
+        if os.path.isdir(self.output_dir_line_edit().text()) and not self.overwrite_checkbox().isChecked():
+            errors.append('Output directory exists but overwrite=False. Please remove output directory first')
+        if self.tasks_line_edit().text() == '':
+            errors.append('Tasks cannot be empty and must be comma-separated list of values')
+        return errors
+    
+    def save_inputs_and_parameters(self):
+        self.settings().set(f'{PANEL_NAME}/scans_dir', self.scans_dir_line_edit().text())
+        self.settings().set(f'{PANEL_NAME}/tasks', self.tasks_line_edit().text())
+        self.settings().set(f'{PANEL_NAME}/output_dir', self.output_dir_line_edit().text())
+        self.settings().set(f'{PANEL_NAME}/overwrite', self.overwrite_checkbox().isChecked())

mosamatic2/ui/widgets/panels/visualizations/liversegmentvisualization/liversegmentpicker.py

@@ -0,0 +1,96 @@
+import vtk
+
+
+class LiverSegmentPicker:
+    def __init__(self, renderer, interactor, actors, volumes, selected_volume_actor):
+        self.ren = renderer
+        self.iren = interactor
+        self.actors = actors
+        self.actor_keys = {v: k for k, v in actors.items()}
+        self.volumes = volumes
+        self.selected_volume_actor = selected_volume_actor
+        self.selected = set()
+        self._orig = {}
+        self._opacity_selected = 1.0
+        self._opacity_not_selected = 0.25
+
+        self.picker = vtk.vtkCellPicker()
+        self.picker.SetTolerance(0.001)
+
+        # Use trackball so it rotates only when you move
+        self.style = vtk.vtkInteractorStyleTrackballCamera()
+        self.iren.SetInteractorStyle(self.style)
+
+        # Hook events
+        self.iren.AddObserver("LeftButtonPressEvent", self.on_left_down)
+        self.iren.AddObserver("LeftButtonReleaseEvent", self.on_left_up)
+
+        self._allow_camera_drag = True
+
+    def _save(self, a):
+        if a in self._orig:
+            return
+
+    def _highlight(self, a, on):
+        self._save(a)
+        p = a.GetProperty()
+        if on:
+            p.SetOpacity(self._opacity_selected)
+        else:
+            p.SetOpacity(self._opacity_not_selected)
+
+    def clear(self):
+        for a in list(self.selected):
+            self._highlight(a, False)
+        self.selected.clear()
+
+    def toggle(self, a):
+        if a in self.selected:
+            self._highlight(a, False)
+            self.selected.remove(a)
+        else:
+            self._highlight(a, True)
+            self.selected.add(a)
+        total_selected_volume = 0
+        for a in self.selected:
+            a_key = self.actor_keys[a]
+            volume = int(self.volumes[a_key])
+            total_selected_volume += volume
+        self.selected_volume_actor.SetInput(f'Selected volume: {total_selected_volume} mL')
+            
+
+    def on_left_down(self, iren, evt):
+        x, y = iren.GetEventPosition()
+
+        # Only do picking when Shift or Ctrl is held.
+        do_pick = (iren.GetShiftKey() or iren.GetControlKey())
+
+        if do_pick:
+            self.picker.Pick(x, y, 0, self.ren)
+            a = self.picker.GetActor()
+            a_key = self.actor_keys[a]
+
+            if a in self.actors.values():
+                # Additive (Shift/Ctrl): toggle selection
+                self.toggle(a)
+            else:
+                # Shift/Ctrl + empty click: clear selection (optional)
+                self.clear()
+
+            iren.GetRenderWindow().Render()
+            self._allow_camera_drag = False  # don't also start a camera drag
+            return
+
+        # No modifier => normal rotate/pan/zoom interaction
+        self._allow_camera_drag = True
+        self.style.OnLeftButtonDown()
+
+    def on_left_up(self, iren, evt):
+        if self._allow_camera_drag:
+            self.style.OnLeftButtonUp()
+
+    def set_opacity_selected(self, opacity_selected):
+        self._opacity_selected = opacity_selected
+
+    def set_opacity_not_selected(self, opacity_not_selected):
+        self._opacity_not_selected = opacity_not_selected

mosamatic2/ui/widgets/panels/visualizations/liversegmentvisualization/liversegmentviewer.py

@@ -0,0 +1,130 @@
+import os
+import vtk
+import pandas as pd
+from PySide6.QtWidgets import (
+    QWidget, 
+    QVBoxLayout,
+)
+from vtkmodules.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
+from mosamatic2.ui.widgets.panels.visualizations.liversegmentvisualization.liversegmentpicker import LiverSegmentPicker
+from mosamatic2.core.managers.logmanager import LogManager
+
+LOG = LogManager()
+SEGMENT_COLORS = [
+    (0.80, 0.25, 0.25),  # muted red
+    (0.25, 0.55, 0.80),  # steel blue
+    (0.30, 0.70, 0.45),  # green
+    (0.85, 0.65, 0.25),  # amber
+    (0.55, 0.40, 0.75),  # purple
+    (0.90, 0.45, 0.15),  # orange
+    (0.20, 0.75, 0.75),  # teal
+    (0.75, 0.75, 0.30),  # olive
+]
+
+vtk.vtkObject.GlobalWarningDisplayOff()
+
+
+class LiverSegmentViewer(QWidget):
+    def __init__(self):
+        super(LiverSegmentViewer, self).__init__()
+        self._total_volume = 0.0
+        self._selected_volume = 0.0
+        self._liver_segment_actors = {}
+        self._liver_volumes = {}
+        self._vtk_widget = QVTKRenderWindowInteractor(self)
+        self._render_window = self._vtk_widget.GetRenderWindow()
+        self._interactor = self._render_window.GetInteractor()
+        self._interactor.SetInteractorStyle(vtk.vtkInteractorStyleTrackballCamera())
+        layout = QVBoxLayout()
+        layout.addWidget(self._vtk_widget)
+        self.setLayout(layout)
+        self._selected_volume_actor = vtk.vtkTextActor()
+        self._selected_volume_actor.SetInput(f'Selected volume: {self._selected_volume} mL')
+        self._selected_volume_actor.GetTextProperty().SetFontSize(18)
+        self._selected_volume_actor.GetTextProperty().SetColor(1, 1, 1)  # RGB 0–1
+        self._selected_volume_actor.SetDisplayPosition(20, 50)
+        self._total_volume_actor = vtk.vtkTextActor()
+        self._total_volume_actor.SetInput(f'Total volume: {self._total_volume} mL')
+        self._total_volume_actor.GetTextProperty().SetFontSize(18)
+        self._total_volume_actor.GetTextProperty().SetColor(1, 1, 1)  # RGB 0–1
+        self._total_volume_actor.SetDisplayPosition(20, 20)
+        self._renderer = vtk.vtkRenderer()
+        self._renderer.AddViewProp(self._total_volume_actor)
+        self._renderer.AddViewProp(self._selected_volume_actor)
+        self._render_window.AddRenderer(self._renderer)
+        self._render_window.Render()
+
+    def image_to_surface(self, image: vtk.vtkImageData, label=1) -> vtk.vtkPolyData:
+        dmc = vtk.vtkDiscreteMarchingCubes()
+        dmc.SetInputData(image)
+        dmc.SetValue(0, label)
+        dmc.Update()
+        cleaner = vtk.vtkCleanPolyData()
+        cleaner.SetInputConnection(dmc.GetOutputPort())
+        cleaner.Update()
+        smoother = vtk.vtkWindowedSincPolyDataFilter()
+        smoother.SetInputConnection(cleaner.GetOutputPort())
+        smoother.SetNumberOfIterations(20)   # 10–30 typical
+        smoother.SetPassBand(0.1)              # smaller = smoother (e.g. 0.05–0.2)
+        smoother.SetFeatureAngle(120.0)      # preserve sharper edges
+        smoother.BoundarySmoothingOff()              # usually best for closed organs
+        smoother.FeatureEdgeSmoothingOff()           # keep features; turn On if you want more smoothing
+        smoother.NonManifoldSmoothingOn()
+        smoother.NormalizeCoordinatesOn()            # helps numerical stability
+        smoother.Update()
+        normals = vtk.vtkPolyDataNormals()
+        normals.SetInputConnection(smoother.GetOutputPort())
+        normals.SetComputePointNormals(True)
+        normals.SetComputeCellNormals(False)
+        normals.SetAutoOrientNormals(True)
+        normals.SetConsistency(True)
+        normals.Update()
+        return normals.GetOutput()
+
+    def surface_to_actor(self, polydata: vtk.vtkPolyData) -> vtk.vtkActor:
+        mapper = vtk.vtkPolyDataMapper()
+        mapper.SetInputData(polydata)
+        mapper.ScalarVisibilityOff()
+        actor = vtk.vtkActor()
+        actor.SetMapper(mapper)
+        return actor
+    
+    def load_segments_and_volumes(self, liver_segments_dir, liver_volumes_file):
+        # Load volumes
+        df = pd.read_csv(liver_volumes_file, sep=';')
+        for _, row in df.iterrows():
+            self._liver_volumes[row['file']] = row['volume_mL']
+        self._total_volume = sum(self._liver_volumes.values())
+        self._total_volume_actor.SetInput(f'Total volume: {self._total_volume} mL')
+        # Load liver segment actors
+        i = 0
+        for f in os.listdir(liver_segments_dir):
+            segment_name = f
+            f_path = os.path.join(liver_segments_dir, f)
+            reader = vtk.vtkNIFTIImageReader()
+            reader.SetFileName(f_path)
+            reader.Update()
+            image = reader.GetOutput()
+            surface = self.image_to_surface(image)
+            actor = self.surface_to_actor(surface)
+            prop = actor.GetProperty()
+            prop.SetSpecular(0.3)
+            prop.SetSpecularPower(20)
+            prop.SetDiffuse(0.7)
+            prop.SetAmbient(0.1)
+            prop.SetColor(*SEGMENT_COLORS[i])
+            prop.SetOpacity(0.5)
+            self._renderer.AddActor(actor)
+            self._liver_segment_actors[segment_name] = actor
+            i += 1
+        picker = LiverSegmentPicker(
+            self._renderer, 
+            self._interactor, 
+            self._liver_segment_actors, 
+            self._liver_volumes, 
+            self._selected_volume_actor,
+        )
+        self._renderer.SetBackground(0.1, 0.1, 0.12)
+        self._renderer.ResetCamera()
+        self._render_window.Render()
+        self._interactor.Start()