tomwer 1.3.26__py3-none-any.whl → 1.4.0__py3-none-any.whl

tomwer/gui/reconstruction/nabu/nabuconfig/reconstruction.py

@@ -1,40 +1,15 @@
 # coding: utf-8
-# /*##########################################################################
-#
-# Copyright (c) 2016-2020 European Synchrotron Radiation Facility
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-#
-# ###########################################################################*/
-
-__authors__ = ["H. Payno"]
-__license__ = "MIT"
-__date__ = "12/02/2020"
+from __future__ import annotations
+
 
 import functools
 import logging
-import typing
 
 from silx.gui import qt
 from silx.gui.dialog.DataFileDialog import DataFileDialog
 from silx.utils.enum import Enum as _Enum
 
+from nxtomomill.io.utils import convert_str_to_tuple
 from tomwer.core.process.reconstruction.nabu.plane import NabuPlane
 from tomwer.core.process.reconstruction.nabu.nabuslices import NabuSliceMode
 from tomwer.core.process.reconstruction.nabu.utils import (
@@ -50,6 +25,8 @@ from tomwer.gui.utils.scrollarea import (
     QDoubleSpinBoxIgnoreWheel,
     QSpinBoxIgnoreWheel,
 )
+from tomwer.gui.utils.RangeWidget import RangeWidget
+from tomwer.gui import icons
 from tomwer.utils import docstring
 
 _logger = logging.getLogger(__name__)
@@ -61,20 +38,31 @@ class SliceGroupBox(qt.QGroupBox):
     sigSlicesChanged = qt.Signal()
     """Signal emitted when the selected slices change"""
 
-    def __init__(self, parent, scrollArea):
+    def __init__(self, parent):
         qt.QGroupBox.__init__(self, "slices", parent)
         self.setCheckable(True)
-
         self.setLayout(qt.QHBoxLayout())
-        self._modeCB = QComboBoxIgnoreWheel(parent=self, scrollArea=scrollArea)
+        # mode
+        self._modeCB = QComboBoxIgnoreWheel(parent=self)
         for mode in NabuSliceMode:
             self._modeCB.addItem(mode.value)
         self.layout().addWidget(self._modeCB)
         self._modeCB.setFocusPolicy(qt.Qt.FocusPolicy.NoFocus)
 
+        # slice line edit
         self._sliceQLE = SelectionLineEdit(parent=self, allow_negative_indices=True)
         self.layout().addWidget(self._sliceQLE)
 
+        # warning icon
+        warningIcon = icons.getQIcon("warning")
+        self._noSliceWarning = qt.QLabel("")
+        self._noSliceWarning.setPixmap(warningIcon.pixmap(20, state=qt.QIcon.On))
+        self._noSliceWarning.setToolTip(
+            "no slice defined, No slice reconstruction will be done"
+        )
+        self.layout().addWidget(self._noSliceWarning)
+
+        # spacer
         spacer = qt.QWidget(parent=self)
         spacer.setSizePolicy(qt.QSizePolicy.Minimum, qt.QSizePolicy.Expanding)
         self.layout().addWidget(spacer)
@@ -83,22 +71,36 @@ class SliceGroupBox(qt.QGroupBox):
         self.setChecked(True)
         self.setMode(NabuSliceMode.MIDDLE)
         self._sliceQLE.setVisible(False)
+        self._noSliceWarning.setVisible(False)
 
         # connect signal / slot
-        self._modeCB.currentIndexChanged.connect(self._updateSliceQLEVisibilty)
+        self._modeCB.currentIndexChanged.connect(self._updateSliceQLEVisibility)
         self._modeCB.currentIndexChanged.connect(self.sigSlicesChanged)
         self._sliceQLE.editingFinished.connect(self.sigSlicesChanged)
+        self._sliceQLE.textChanged.connect(self._updateNoSliceWarningVisibility)
         self.toggled.connect(self.sigSlicesChanged)
 
-    def _updateSliceQLEVisibilty(self, *args, **kwargs):
+    def _updateSliceQLEVisibility(self, *args, **kwargs):
         self._sliceQLE.setVisible(self.getMode() == NabuSliceMode.OTHER)
+        self._updateNoSliceWarningVisibility()
+
+    def _updateNoSliceWarningVisibility(self):
+        def no_other_slice_defined() -> bool:
+            slices_defined = self._sliceQLE.text().replace(" ", "")
+            slices_defined = self._sliceQLE.text().replace(",", "")
+            slices_defined = self._sliceQLE.text().replace(";", "")
+            return slices_defined == ""
+
+        self._noSliceWarning.setVisible(
+            self.getMode() is NabuSliceMode.OTHER and no_other_slice_defined()
+        )
 
     def setMode(self, mode: NabuSliceMode):
         mode = NabuSliceMode.from_value(mode)
         item_index = self._modeCB.findText(mode.value)
         self._modeCB.setCurrentIndex(item_index)
         old = self.blockSignals(True)
-        self._updateSliceQLEVisibilty()
+        self._updateSliceQLEVisibility()
         self.blockSignals(old)
 
     def getMode(self) -> NabuSliceMode:
@@ -279,7 +281,8 @@ class AnglesFileWidget(qt.QWidget):
 
 class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
     """
-    Widget to define the configuration of nabu reconstruction processing
+    Widget to define the configuration of nabu reconstruction processing and dataset configuration.
+    At the gui side the "dataset" section as no real reason to be on its own. So we melt the two there.
     """
 
     sigConfChanged = qt.Signal(str)
@@ -287,7 +290,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
     modified
     """
 
-    def __init__(self, parent, scrollArea):
+    def __init__(self, parent):
         qt.QWidget.__init__(self, parent, stage=_NabuStages.PRE)
         _NabuStageConfigBase.__init__(self, stage=_NabuStages.PRE)
         self.setLayout(qt.QGridLayout())
@@ -295,7 +298,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         self.__optimizationIteAlgo = "chambolle-pock"
 
         # slices to be reconstructed online
-        self._slicesWidget = SliceGroupBox(parent=self, scrollArea=scrollArea)
+        self._slicesWidget = SliceGroupBox(parent=self)
         self.layout().addWidget(self._slicesWidget, 0, 0, 1, 2)
         # ignore listener of the mouse for the slice widget, can
         # come with an issue as we are on a scroll area
@@ -305,7 +308,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         self._axisLabel = qt.QLabel("reconstruction plane", self)
         self._axisLabel.setToolTip("Over which axis the slice must be picked")
         self.layout().addWidget(self._axisLabel)
-        self._axisQCB = qt.QComboBox(self)
+        self._axisQCB = QComboBoxIgnoreWheel(self)
         axis_tooltips = {
             NabuPlane.YZ.value: "along axis x (aka axis 2 - expected to be slow)",
             NabuPlane.XZ.value: "along axis y (aka axis 1 - expected to be slow)",
@@ -323,7 +326,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         # method
         self._methodLabel = qt.QLabel("method", self)
         self.layout().addWidget(self._methodLabel, 2, 0, 1, 1)
-        self._methodQCB = QComboBoxIgnoreWheel(parent=self, scrollArea=scrollArea)
+        self._methodQCB = QComboBoxIgnoreWheel(parent=self)
         for method in _NabuReconstructionMethods:
             self._methodQCB.addItem(method.value)
         self.layout().addWidget(self._methodQCB, 2, 1, 1, 1)
@@ -333,7 +336,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         # angle_offset
         self._labelOffsetLabel = qt.QLabel("angle offset (in degree)", self)
         self.layout().addWidget(self._labelOffsetLabel, 3, 0, 1, 1)
-        self._angleOffsetQDSB = QDoubleSpinBoxIgnoreWheel(self, scrollArea)
+        self._angleOffsetQDSB = QDoubleSpinBoxIgnoreWheel(self)
         self._angleOffsetQDSB.setMaximum(-180)
         self._angleOffsetQDSB.setMaximum(180)
         self.layout().addWidget(self._angleOffsetQDSB, 3, 1, 1, 1)
@@ -343,7 +346,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         # fbp filter type
         self._fbpFilterCB = qt.QCheckBox("fbp filter", self)
         self.layout().addWidget(self._fbpFilterCB, 4, 0, 1, 1)
-        self._fbpFilterType = QComboBoxIgnoreWheel(self, scrollArea)
+        self._fbpFilterType = QComboBoxIgnoreWheel(self)
         for filter_type in _NabuFBPFilterType:
             self._fbpFilterType.addItem(filter_type.value)
         self.layout().addWidget(self._fbpFilterType, 4, 1, 1, 1)
@@ -353,7 +356,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         # padding type
         self._paddingTypeLabel = qt.QLabel("padding type", self)
         self.layout().addWidget(self._paddingTypeLabel, 5, 0, 1, 1)
-        self._paddingType = QComboBoxIgnoreWheel(self, scrollArea)
+        self._paddingType = QComboBoxIgnoreWheel(self)
         for fbp_padding_type in _NabuPaddingType:
             self._paddingType.addItem(fbp_padding_type.value)
         self.layout().addWidget(self._paddingType, 5, 1, 1, 1)
@@ -377,7 +380,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
 
         # binning - subsampling
         self._binSubSamplingGB = _BinSubSampling(
-            "binning and subsampling", parent=self, scrollArea=scrollArea
+            "binning and sub-sampling", parent=self
         )
         self.layout().addWidget(self._binSubSamplingGB, 9, 0, 1, 2)
 
@@ -445,6 +448,19 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         self.registerWidget(self._angleFileLabel, "advanced")
         self.registerWidget(self._anglesFileWidget, "advanced")
 
+        # exclude projections: [A, B] (if user want's to skip projection from degree A to degree B (included))
+        self._excludeProjectionsQCB = qt.QCheckBox("exclude projections angles", self)
+        self.layout().addWidget(self._excludeProjectionsQCB, 24, 0, 1, 1)
+        self._excludeProjectionsWidget = RangeWidget(self)
+        self._excludeProjectionsWidget.setRange(0.0, 360.0)
+        self._excludeProjectionsWidget.setSuffix("º")
+        self._excludeProjectionsWidget.setToolTip(
+            "Projections with a rotation in [a, b] range will be ignored."
+        )
+        self.layout().addWidget(self._excludeProjectionsWidget, 24, 1, 1, 1)
+        self.registerWidget(self._excludeProjectionsQCB, "advanced")
+        self.registerWidget(self._excludeProjectionsWidget, "advanced")
+
         # spacer for style
         spacer = qt.QWidget(self)
         spacer.setSizePolicy(qt.QSizePolicy.Minimum, qt.QSizePolicy.Expanding)
@@ -466,6 +482,7 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         self._positivityConstraintCB.setChecked(True)
         self._clipOuterCircleCB.setChecked(False)
         self._centeredAxisCB.setChecked(False)
+        self._excludeProjectionsWidget.setEnabled(False)
 
         # connect signal / slot
         self._axisQCB.currentIndexChanged.connect(self._slicesChanged)
@@ -485,6 +502,13 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         self._anglesFileWidget._qle.textChanged.connect(self._anglesFilechanged)
         self._clipOuterCircleCB.toggled.connect(self._clipOuterCircleChanged)
         self._centeredAxisCB.toggled.connect(self._centeredAxisChanged)
+        self._excludeProjectionsQCB.toggled.connect(self._excludeProjectionsChanged)
+        self._excludeProjectionsQCB.toggled.connect(
+            self._excludeProjectionsWidget.setEnabled
+        )
+        self._excludeProjectionsWidget.sigChanged.connect(
+            self._excludeProjectionsChanged
+        )
 
     def getSlices(self):
         return self._slicesWidget.getSlices()
@@ -542,6 +566,9 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
     def _anglesFilechanged(self, *args, **kwargs):
         self._signalConfChanged("angles_file")
 
+    def _excludeProjectionsChanged(self):
+        self._signalConfChanged("exclude_projections")
+
     def setScan(self, scan):
         raise NotImplementedError()
 
@@ -556,10 +583,10 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
     def getAngleOffset(self) -> float:
         return self._angleOffsetQDSB.value()
 
-    def setAngleOffset(self, value: typing.Union[str, float]):
+    def setAngleOffset(self, value: str | float):
         self._angleOffsetQDSB.setValue(float(value))
 
-    def getFBPFilterType(self) -> typing.Union[_NabuFBPFilterType, None]:
+    def getFBPFilterType(self) -> _NabuFBPFilterType | None:
         if self._fbpFilterCB.isChecked():
             return _NabuFBPFilterType.from_value(self._fbpFilterType.currentText())
         else:
@@ -599,13 +626,13 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
     def isPreconditioningFilterEnable(self):
         return self._preconditioningFilter.isChecked()
 
-    def setPreconditioningFilterEnable(self, enable: typing.Union[bool, int]):
+    def setPreconditioningFilterEnable(self, enable: bool | int):
         self._preconditioningFilter.setChecked(bool(enable))
 
     def isPositivityConstraintEnable(self):
         return self._positivityConstraintCB.isChecked()
 
-    def setPositivityConstraintEnable(self, enable: typing.Union[bool, int]):
+    def setPositivityConstraintEnable(self, enable: bool | int):
         self._positivityConstraintCB.setChecked(bool(enable))
 
     def getTranslationMvtFile(self):
@@ -654,10 +681,27 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         """return over which axis we expect to do the reconstruction"""
         return NabuPlane.from_value(self._axisQCB.currentText())
 
-    def setNabuPlane(self, axis: typing.Union[str, NabuPlane]):
+    def setNabuPlane(self, axis: str | NabuPlane):
         axis = NabuPlane.from_value(axis)
         self._axisQCB.setCurrentText(axis.value)
 
+    def getExcludeProjections(self) -> tuple:
+        if self._excludeProjectionsQCB.isChecked():
+            return self._excludeProjectionsWidget.getRange()
+        else:
+            return tuple()
+
+    def setExcludeProjections(self, projections: tuple):
+        if not isinstance(projections, tuple):
+            raise TypeError(
+                f"projections should be an instance of tuple. Got {type(projections)}"
+            )
+        if len(projections) == 0:
+            self._excludeProjectionsQCB.setChecked(False)
+        else:
+            self._excludeProjectionsQCB.setChecked(True)
+            self._excludeProjectionsWidget.setRange(*projections)
+
     @docstring(_NabuStageConfigBase)
     def getConfiguration(self) -> dict:
         fbp_filter_type = self.getFBPFilterType()
@@ -687,9 +731,22 @@ class _NabuReconstructionConfig(qt.QWidget, _NabuStageConfigBase):
         return config
 
     def getDatasetConfiguration(self) -> dict:
-        return self._binSubSamplingGB.getConfiguration()
+        exclude_projections = self.getExcludeProjections()
+        if len(exclude_projections) == 0:
+            exclude_projections = ""
+        else:
+            exclude_projections = f"angular_range={str(list(exclude_projections))}"
+        config = self._binSubSamplingGB.getConfiguration()
+        config["exclude_projections"] = exclude_projections
+        return config
 
     def setDatasetConfiguration(self, config):
+        exclude_proj = config.pop("exclude_projections", None)
+        if exclude_proj is not None:
+            if isinstance(exclude_proj, str):
+                exclude_proj = exclude_proj.replace("angular_range=", "")
+            exclude_proj = convert_str_to_tuple(exclude_proj)
+            self.setExcludeProjections(exclude_proj)
         return self._binSubSamplingGB.setConfiguration(config)
 
     @docstring(_NabuStageConfigBase)
@@ -791,7 +848,7 @@ class _SubRegionEditor(qt.QObject):
     def layout(self):
         return self._layout
 
-    def getSubRegionMin(self) -> typing.Union[None, int]:
+    def getSubRegionMin(self) -> int | None:
         """
 
         :return: None if region is unbounded else the value of the bound
@@ -801,7 +858,7 @@ class _SubRegionEditor(qt.QObject):
         else:
             return None
 
-    def getSubRegionMax(self) -> typing.Union[None, int]:
+    def getSubRegionMax(self) -> int | None:
         """
 
         :return: None if region is unbounded else the value of the bound
@@ -811,7 +868,7 @@ class _SubRegionEditor(qt.QObject):
         else:
             return None
 
-    def setSubRegionMin(self, sub_region_min: typing.Union[None, int]) -> None:
+    def setSubRegionMin(self, sub_region_min: int | None) -> None:
         """
 
         :param min: if min is None or -1 wr will expand it to - infinity
@@ -843,20 +900,14 @@ class _SubRegionEditor(qt.QObject):
         """
 
         :return: min, max
-        :rtype: tuple
         """
         return self.getSubRegionMin(), self.getSubRegionMax()
 
     def setSubRegion(
         self,
-        sub_region_min: typing.Union[None, int],
-        sub_region_max: typing.Union[None, int],
+        sub_region_min: int | None,
+        sub_region_max: int | None,
     ):
-        """
-
-        :param tuple min:
-        :param tuple max:
-        """
         self.setSubRegionMin(sub_region_min)
         self.setSubRegionMax(sub_region_max)
 
@@ -957,11 +1008,11 @@ class _BinSubSampling(qt.QGroupBox):
     binningChanged = qt.Signal()
     """signal emitted when binning change"""
 
-    def __init__(self, text, parent, scrollArea):
+    def __init__(self, text, parent):
         qt.QGroupBox.__init__(self, text)
         self.setLayout(qt.QFormLayout())
         # horizontal binning
-        self._hBinningSB = QSpinBoxIgnoreWheel(self, scrollArea=scrollArea)
+        self._hBinningSB = QSpinBoxIgnoreWheel(self)
         self._hBinningSB.setMinimum(1)
         self._hBinningSB.setMaximum(3)
         self.layout().addRow("horizontal binning", self._hBinningSB)
@@ -971,11 +1022,11 @@ class _BinSubSampling(qt.QGroupBox):
         self._vBinningSB.setMaximum(100)
         # self.layout().addRow('vertical binning', self._vBinningSB)
         self._vBinningSB.setVisible(False)
-        # projection subsampling
-        self._projSubsamplingSB = QSpinBoxIgnoreWheel(self, scrollArea=scrollArea)
+        # projection sub-sampling
+        self._projSubsamplingSB = QSpinBoxIgnoreWheel(self)
         self._projSubsamplingSB.setMinimum(1)
         self._projSubsamplingSB.setMaximum(100)
-        self.layout().addRow("projection subsampling", self._projSubsamplingSB)
+        self.layout().addRow("projection sub-sampling", self._projSubsamplingSB)
 
         # connect signal / slot
         self._vBinningSB.valueChanged.connect(self._valueUpdated)

tomwer/gui/reconstruction/nabu/nabuflow.py

@@ -1,34 +1,9 @@
 # coding: utf-8
-# /*##########################################################################
-#
-# Copyright (c) 2016-2017 European Synchrotron Radiation Facility
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-#
-# ###########################################################################*/
-
-__authors__ = ["H. Payno"]
-__license__ = "MIT"
-__date__ = "04/03/2019"
+from __future__ import annotations
+
 
 import functools
-import typing
+from typing import Iterable
 
 from silx.gui import qt
 
@@ -168,7 +143,7 @@ class NabuFlowControl(qt.QWidget):
         """
         change visibility of the widgets relative to initialization process
 
-        :param bool visible:
+        :param visible:
         """
         self._iniProcessWidgets.setVisible(visible)
 
@@ -176,40 +151,34 @@ class NabuFlowControl(qt.QWidget):
         """
         change visibility of the widgets relative to pre processing process
 
-        :param bool visible:
+        :param visible:
         """
         self._preProcessWidgets.setVisible(visible)
 
-    def setPhaseVisible(self, visible):
+    def setPhaseVisible(self, visible: bool):
         """
         change visibility of the widgets relative to pre processing process
-
-        :param bool visible:
         """
         self._phaseProcessWidgets.setVisible(visible)
 
-    def setProcVisible(self, visible):
+    def setProcVisible(self, visible: bool):
         """
         change visibility of the widgets relative to processing process
-
-        :param bool visible:
         """
         self._processWidgets.setVisible(visible)
 
-    def setPostVisible(self, visible):
+    def setPostVisible(self, visible: bool):
         """
         change visibility of the widgets relative to post processing process
-
-        :param bool visible:
         """
         self._postProcessWidgets.setVisible(visible)
 
-    def setIniProcessing(self, processes: typing.Iterable, icons: typing.Iterable):
+    def setIniProcessing(self, processes: Iterable, icons: Iterable):
         """
         Defines the processing to execute for initialization
 
-        :param Iterable processes: processes to display
-        :param Iterable icons: icons associated to the processes if any
+        :param processes: processes to display
+        :param icons: icons associated to the processes if any
         """
         draggability = [False] * len(processes)
         self._iniProcessWidgets.setProcesses(
@@ -217,11 +186,11 @@ class NabuFlowControl(qt.QWidget):
         )
         self.setIniprocVisible(len(processes) != 0)
 
-    def setPreProcessing(self, processes: typing.Iterable, icons: typing.Iterable):
+    def setPreProcessing(self, processes: Iterable, icons: Iterable):
         """
 
-        :param Iterable processes: processes to display
-        :param Iterable icons: icons associated to the processes if any
+        :param processes: processes to display
+        :param icons: icons associated to the processes if any
         """
         draggability = [False] * len(processes)
         self._preProcessWidgets.setProcesses(
@@ -229,11 +198,11 @@ class NabuFlowControl(qt.QWidget):
         )
         self.setPreProcVisible(len(processes) != 0)
 
-    def setPhaseProcessing(self, processes: typing.Iterable, icons: typing.Iterable):
+    def setPhaseProcessing(self, processes: Iterable, icons: Iterable):
         """
 
-        :param Iterable processes: processes to display
-        :param Iterable icons: icons associated to the processes if any
+        :param processes: processes to display
+        :param icons: icons associated to the processes if any
         """
         draggability = [False] * len(processes)
         self._phaseProcessWidgets.setProcesses(
@@ -241,11 +210,11 @@ class NabuFlowControl(qt.QWidget):
         )
         self.setPhaseVisible(len(processes) != 0)
 
-    def setProcessing(self, processes: typing.Iterable, icons: typing.Iterable):
+    def setProcessing(self, processes: Iterable, icons: Iterable):
         """
 
-        :param Iterable processes: processes to display
-        :param Iterable icons: icons associated to the processes if any
+        :param processes: processes to display
+        :param icons: icons associated to the processes if any
         """
         draggability = [False] * len(processes)
         self._processWidgets.setProcesses(
@@ -253,11 +222,11 @@ class NabuFlowControl(qt.QWidget):
         )
         self.setProcVisible(len(processes) != 0)
 
-    def setPostProcessing(self, processes: typing.Iterable, icons: typing.Iterable):
+    def setPostProcessing(self, processes: Iterable, icons: Iterable):
         """
 
-        :param Iterable processes: processes to display
-        :param Iterable icons: icons associated to the processes if any
+        :param processes: processes to display
+        :param icons: icons associated to the processes if any
         """
         draggability = [False] * len(processes)
         self._postProcessWidgets.setProcesses(
@@ -293,13 +262,15 @@ class NabuFlowControl(qt.QWidget):
         """
         return self._postProcessWidgets.getFlow()
 
-    def getFlow(self):
+    def getFlow(self) -> dict:
         """
+        Returns the concatenation of pre-processing, processing, and post-processing flows.
+
+        :return: Dictionary where the keys are Nabu stages, and the values are lists of flows for each stage.
+        :rtype: dict
 
-        :return: Return the concatenation of pre processing, processing and
-                 post processing
-        :rtype: dict. keys are nabu stage, values are list of flow for each
-                stage.
+        This method fetches the processing flows for all stages and returns them
+        as a dictionary, allowing easy access to the current configuration.
         """
         return {
             _NabuStages.INI: self._iniProcessWidget.getFlow(),
@@ -309,11 +280,10 @@ class NabuFlowControl(qt.QWidget):
             _NabuStages.POST: self._processWidgets.getFlow(),
         }
 
-    def getProcessFocused(self):
+    def getProcessFocused(self) -> tuple:
         """
 
         :return: the process under focus
-        :rtype: tuple (stage or None, None or process_id (str or class))
         """
         stages = (
             _NabuStages.INI,