QtOrthoViewer.py

from OrthoViewer import *

from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from Worker import *

from QtViewer import *

class QtOrthoViewer(QtViewer):

    # Constructor
    def __init__(self, vtkBaseClass, orientation, label:str="Orthogonal Viewer"):
        super(QtOrthoViewer, self).__init__()

        # Properties
        self.orientation = orientation
        self.status = False
        self.label = label
        
        ## Render Viewer
        self.viewer = OrthoViewer(vtkBaseClass, self.orientation, self.label)

        # Initialize the UI        
        self._init_UI()

        # Thread
        self.thread = QThread()
        self.worker = Worker(self.slider)
        
        # Connect signals and slots
        self.connect()

    # Initialize the UI
    def _init_UI(self):
        super()._init_UI()
        
        # PyQt Stuff
        ## Slider
        self.slider = QSlider(Qt.Vertical)
        self.slider.setSingleStep(1)
        self.slider.setValue(0)
        self.slider.setEnabled(False)
        self.viewer.commandSliceSelect.sliders[self.orientation] = self.slider
        
        ## Buttons
        self.buttonsLayout = QHBoxLayout()
        
        self.prevBtn = QPushButton()
        self.prevBtn.setIcon(QIcon("./assets/decrease.svg"))
        self.prevBtn.setStyleSheet("font-size:15px; border-radius: 6px;border: 1px solid rgba(27, 31, 35, 0.15);padding: 5px 15px; background: black")
        self.prevBtn.setDisabled(True)
        
        self.playBtn = QPushButton()
        self.playBtn.setIcon(QIcon("./assets/play.ico"))
        self.playBtn.setStyleSheet("font-size:15px; border-radius: 6px;border: 1px solid rgba(27, 31, 35, 0.15);padding: 5px 15px;")
        self.playBtn.setDisabled(True)
        
        self.nextBtn = QPushButton()
        self.nextBtn.setIcon(QIcon("./assets/increase.svg"))
        self.nextBtn.setStyleSheet("font-size:15px; border-radius: 6px;border: 1px solid rgba(27, 31, 35, 0.15);padding: 5px 15px; background: black")
        self.nextBtn.setDisabled(True)
        
        self.buttonsLayout.addSpacerItem(QSpacerItem(80, 10))
        self.buttonsLayout.addWidget(self.prevBtn,4)
        self.buttonsLayout.addWidget(self.playBtn,5)
        self.buttonsLayout.addWidget(self.nextBtn,4)
        self.buttonsLayout.addSpacerItem(QSpacerItem(80, 10))
        
        # Set up the layouts
        self.topLayout.addWidget(self.slider)
        self.mainLayout.addLayout(self.buttonsLayout)
    
    # Connect signals and slots        
    def connect(self):
        # Connect slider signals to slice update slots
        self.slider.valueChanged.connect(self.update_slice)
        
        # Connect buttons to slots
        self.prevBtn.clicked.connect(lambda: self.next_prev_btn(self.slider.value()-10))
        self.playBtn.clicked.connect(self.play_pause_btn)
        self.nextBtn.clicked.connect(lambda: self.next_prev_btn(self.slider.value()+10))

    # Update slice
    def update_slice(self, slice_index):
        self.viewer.set_slice(slice_index)

    # Connect on data
    def connect_on_data(self, path):
        super().connect_on_data(path)
        
        # Settings of the button
        self.prevBtn.setEnabled(True)
        self.playBtn.setEnabled(True)
        self.nextBtn.setEnabled(True)
    
        # Settings of the slider
        self.slider.setEnabled(True)
        self.slider.setMinimum(self.viewer.min_slice)
        self.slider.setMaximum(self.viewer.max_slice)
        self.slider.setValue((self.slider.maximum() + self.slider.minimum())//2)
    
    # Next/Previous button function
    def next_prev_btn(self, slice_index):
        if slice_index < self.slider.minimum():
            slice_index = self.slider.minimum()
        elif slice_index > self.slider.maximum():
            slice_index = self.slider.maximum()
            
        self.slider.setValue(slice_index)
        self.viewer.set_slice(slice_index)
        
    # Play slices            
    def play_slices(self):
        self.thread = QThread()
        self.worker = Worker(self.slider)
        self.status = True
        
        # Play Button icon
        self.playBtn.setIcon(QIcon("./assets/pause.ico"))
                
        # Final resets
        # Move worker to the thread
        self.worker.moveToThread(self.thread)
        
        # Connect signals and slots
        self.thread.started.connect(self.worker.play)
        self.worker.finished.connect(self.thread.quit)
        self.worker.finished.connect(self.worker.deleteLater)
        self.thread.finished.connect(self.thread.deleteLater)
        self.worker.progress.connect(self.update_slice)
        
        # Start the thread
        self.thread.start()
        self.slider.setHidden(True)
        self.thread.finished.connect(
            lambda: self.slider.setHidden(False)
        )
        self.thread.finished.connect(
            self.pause_slices
        )

    # Pause slices
    def pause_slices(self):
        self.playBtn.setIcon(QIcon("./assets/play.ico"))
        self.worker.pause()
        self.status = False

    # Play/Pause button function    
    def play_pause_btn(self):
        if self.status == False:
            self.play_slices()
        else:
            self.pause_slices()