opacity_in_viewer.py

import sys

import vtk
from ccpi.viewer import viewer3D
from ccpi.viewer.iviewer import SingleViewerCenterWidget
from ccpi.viewer.QCILViewerWidget import QCILViewerWidget
from eqt.ui.UIFormWidget import FormDockWidget
from PySide2 import QtWidgets
from PySide2.QtCore import Qt
from PySide2.QtWidgets import QComboBox
from ccpi.viewer.utils import example_data

try:
    import qdarkstyle
    from qdarkstyle.dark.palette import DarkPalette
    set_style = True
except:
    set_style = False


class OpacityViewerWidget(SingleViewerCenterWidget):

    def __init__(self, parent=None, viewer=viewer3D):
        SingleViewerCenterWidget.__init__(self, parent)

        self.frame = QCILViewerWidget(parent, viewer=viewer, shape=(600, 600))

        self.setCentralWidget(self.frame)

        self.create_settings_dockwidget()

        if set_style:
            self.set_app_style()

        self.show()

    def set_input(self, data):
        self.frame.viewer.setInputData(data)

    def create_settings_dockwidget(self):
        form_dock_widget = FormDockWidget(title='')
        drop_down = QComboBox()
        drop_down.addItems(['gradient', 'scalar'])
        drop_down.currentTextChanged.connect(
            lambda: self.frame.viewer.setVolumeRenderOpacityMethod(drop_down.currentText()))
        form_dock_widget.addWidget(drop_down, "Select Opacity Method:", 'select_opacity')
        self.addDockWidget(Qt.TopDockWidgetArea, form_dock_widget)

    def set_app_style(self):
        '''Sets app stylesheet '''
        style = qdarkstyle.load_stylesheet(palette=DarkPalette)
        self.setStyleSheet(style)


class viewer_window(object):
    '''
    a Qt interactive viewer with one single dataset
    Parameters
    ----------
    data: vtkImageData
        image to be displayed       
    '''

    def __init__(self, data):
        '''Creator'''
        app = QtWidgets.QApplication(sys.argv)
        self.app = app

        self.setUp(data)
        self.show()

    def setUp(self, data):
        window = OpacityViewerWidget()
        window.set_input(data)
        self.window = window
        self.has_run = None

    def show(self):
        if self.has_run is None:
            self.has_run = self.app.exec_()
        else:
            print('No instance can be run interactively again. Delete and re-instantiate.')


if __name__ == "__main__":

    err = vtk.vtkFileOutputWindow()
    err.SetFileName("viewer.log")
    vtk.vtkOutputWindow.SetInstance(err)

    # To use your own dataset:
    # reader = vtk.vtkMetaImageReader()
    # reader.SetFileName(r'head.mha')
    # reader.Update()
    # viewer_window(reader.GetOutput())

    data = example_data.HEAD.get()
    viewer_window(data)