VesselVio.py

"""
The main entry point to build the GUI of the application.
"""

__author__ = "Jacob Bumgarner <jrbumgarner@mix.wvu.edu>"
__license__ = "GPLv3 - GNU General Pulic License v3 (see LICENSE)"
__copyright__ = "Copyright 2022 by Jacob Bumgarner"
__webpage__ = "https://jacobbumgarner.github.io/VesselVio/"
__download__ = "https://jacobbumgarner.github.io/VesselVio/Downloads"

import os
import sys

os.environ["QT_MAC_WANTS_LAYER"] = "1"

from multiprocessing import freeze_support

import pyvista as pv

from library import image_processing as ImProc, qt_threading as QtTh
from library.gui import (
    analysis_page,
    annotation_page,
    left_menu,
    qt_objects as QtO,
    update_alert,
    visualization_page,
)
from library.helpers import load_icon

from PyQt5.QtCore import QTimer
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import QApplication, QMainWindow, QStackedWidget
from pyvistaqt import QtInteractor

######
__version__ = "V 1.1.2"
######


# Main window for the application.
class mainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        # Setup the main window.
        self.plotter = QtInteractor(self)
        light = pv.Light(light_type="headlight", intensity=0.1)
        self.plotter.add_light(light)

        # Name & Color.
        self.setWindowTitle("VesselVio")
        self.centralWidget = QtO.new_widget()
        self.setCentralWidget(self.centralWidget)
        self.centralLayout = QtO.new_layout(self.centralWidget, no_spacing=True)

        # Setup the individual pages of the application.
        self.leftMenu = left_menu.LeftMenu(__version__)

        self.qStack = QStackedWidget()  # Stacked widget for the pages

        self.page1 = analysis_page.AnalysisPage()
        self.page2 = visualization_page.VisualizationPage(self.plotter, self)
        self.page3 = annotation_page.AnnotationPage()

        self.pageStack = QtO.new_stacked([self.page1, self.page2, self.page3])

        # Connect sour left menu to our pages.
        self.leftMenu.pageTabs.currentRowChanged.connect(self.pageStack.setCurrentIndex)
        self.leftMenu.pageTabs.setCurrentRow(0)

        # Add pages to the app
        QtO.add_widgets(self.centralLayout, [self.leftMenu, self.pageStack])

        # Make sure we delete the labeled_cache_volume if it exists
        ImProc.clear_labeled_cache()

        # Run our tiny file to prep our numba jit compilers.
        QtTh.prepare_compilers()

        # Check for updates
        update_alert.local_version = __version__[2:]
        QTimer.singleShot(500, update_alert.version_check)

        # Present the application
        self.show()
        return

    # Close all associated widgets if the main thread is ended.s
    def closeEvent(self, event):
        try:
            self.page1.a_thread.stop()
            self.page2.v_thread.close()
        except AttributeError:
            pass
        event.accept()


if __name__ == "__main__":
    freeze_support()
    app = QApplication(sys.argv)
    app.setWindowIcon(QIcon(load_icon()))
    main_app = mainWindow()
    sys.exit(app.exec_())