find_spots/imageCompareDialog.py at main · StathopoulosLab/find_spots · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# imageCompareDialog.py

from qtpy.QtCore import Qt, Slot
from qtpy.QtGui import QPixmap
from qtpy.QtWidgets import QDialog, QDialogButtonBox

from numpy import ndarray
from imageCompareDialog_ui import Ui_ImageCompareDialog

from algorithms.denoise import ProcessStepDenoiseConcurrent
from algorithms.threshold_mask import ProcessStepThresholdMask
from processing import ProcessStep, ProcessStatus
from typing import Callable, Dict

class ImageCompareDialog(QDialog):
    """
    """
    DiscardResults = 2

    def __init__(self, parent=None):
        super().__init__(parent)
        self.ui = Ui_ImageCompareDialog()
        self.ui.setupUi(self)
        # synchronize slice across both sides
        self.ui.leftGraphicsView.sliceChanged.connect(self.ui.rightGraphicsView.updateSlice)
        self.ui.rightGraphicsView.sliceChanged.connect(self.ui.leftGraphicsView.updateSlice)
        self.ui.leftGraphicsView.horizontalScrollBar().valueChanged.connect(self.ui.rightGraphicsView.horizontalScrollBar().setValue)
        self.ui.leftGraphicsView.verticalScrollBar().valueChanged.connect(self.ui.rightGraphicsView.verticalScrollBar().setValue)
        self.ui.rightGraphicsView.horizontalScrollBar().valueChanged.connect(self.ui.leftGraphicsView.horizontalScrollBar().setValue)
        self.ui.rightGraphicsView.verticalScrollBar().valueChanged.connect(self.ui.leftGraphicsView.verticalScrollBar().setValue)
        # make the discard button return a unique value
        self.ui.buttonBox.button(QDialogButtonBox.Discard).clicked.connect(self.discardResults)

    def setLeftImageVolume(self, volume: ndarray):
        self.ui.leftGraphicsView.setImageVolume(volume)

    def setRightImageVolume(self, volume: ndarray):
        self.ui.rightGraphicsView.setImageVolume(volume)

    @Slot()
    def discardResults(self):
        self.done(self.DiscardResults)

class ProcessStepVisualizeDenoise(ProcessStep):
    """
    ProcessStep to visualize the denoising process.

    Inputs:
        List[beforeDenoising, afterDenoising]

    Step Outputs:
        afterDenoising image

    Status:
        Depends on user response, as follows:
        Clicking "Okay" yields ProcessStatus.COMPLETED
        Clicking "Reject/Don't Save" yields ProcessStatus.REJECTED
        Clicking "Cancel" yields ProcessStatus.CANCELLED
        While user is still running the tool, status is ProcessStatus.RUNNING
    """

    def __init__(self, params: Dict = {}):
        super().__init__(params)
        self._stepName = "Visualize"

    def run(self, progressCallback: Callable[[int, str], None] = None):
        self._status = ProcessStatus.RUNNING
        step = ProcessStepDenoiseConcurrent(self._params)
        step.setApp(self._app)
        step.setInputs(self._inputs)
        step.run(progressCallback)
        stepOutputs = step.stepOutputs()
        endOutputs = step.endOutputs()
        viewer = ImageCompareDialog()
        viewer.setLeftImageVolume(self._inputs[0])
        viewer.setRightImageVolume(stepOutputs[0])
        result = viewer.exec()
        if result == QDialog.Accepted:
            # The user accepted the current parameters, so we press forward
            self._status = ProcessStatus.COMPLETED
        elif result == QDialog.Rejected:
            # confusingly, this is the Cancel button, so we abort the processing
            self._status = ProcessStatus.ABORTED
        elif result == ImageCompareDialog.DiscardResults:
            # The user didn't like the results and wants to retry with different parameters
            self._status = ProcessStatus.REJECTED
        else:
            raise ValueError(f"Unexpected return value: {result}")
        self._stepOutputs = stepOutputs
        self._endOutputs = endOutputs