Add experimental overlay

Author: pshriwise

openmc_plotter/main_window.py

@@ -9,7 +9,7 @@
 from PySide6.QtWidgets import (QApplication, QLabel, QSizePolicy, QMainWindow,
                                QScrollArea, QMessageBox, QFileDialog,
                                QColorDialog, QInputDialog, QWidget,
-                               QGestureEvent, QProgressBar)
+                               QGestureEvent)
 
 import openmc
 import openmc.lib
@@ -115,12 +115,6 @@ def loadGui(self, use_settings_pkl=True):
         self.coord_label = QLabel()
         self.statusBar().addPermanentWidget(self.coord_label)
         self.coord_label.hide()
-        self.busyIndicator = QProgressBar()
-        self.busyIndicator.setRange(0, 0)
-        self.busyIndicator.setMaximumWidth(self.font_metric.averageCharWidth() * 12)
-        self.busyIndicator.setMaximumHeight(self.font_metric.height())
-        self.busyIndicator.hide()
-        self.statusBar().addPermanentWidget(self.busyIndicator)
 
         self.plot_manager = self.model.plot_manager
         self.plot_manager.plot_started.connect(self._on_plot_started)
@@ -1252,11 +1246,12 @@ def waitForPlotIdle(self, timeout_ms=None):
         return True
 
     def _on_plot_started(self):
-        self.busyIndicator.show()
-        self.statusBar().showMessage('Generating Plot...')
+        if hasattr(self, "plotIm") and self.plotIm is not None:
+            self.plotIm.showUpdatingOverlay("Generating Plot...")
 
     def _on_plot_queued(self):
-        self.statusBar().showMessage('Generating Plot... (update queued)')
+        if hasattr(self, "plotIm") and self.plotIm is not None:
+            self.plotIm.showUpdatingOverlay("Generating Plot... (update queued)")
 
     def _on_plot_finished(self, view_snapshot, view_params, ids_map, properties):
         if view_params != self.plot_manager.latest_view_params:
@@ -1273,8 +1268,8 @@ def _on_plot_error(self, error_msg):
         msg_box.exec()
 
     def _on_plot_idle(self):
-        self.busyIndicator.hide()
-        self.statusBar().showMessage('')
+        if hasattr(self, "plotIm") and self.plotIm is not None:
+            self.plotIm.hideUpdatingOverlay()
 
     def saveSettings(self):
         if self.model.statepoint:

openmc_plotter/plotgui.py

@@ -2,7 +2,7 @@
 
 from PySide6 import QtCore, QtGui
 from PySide6.QtWidgets import (QWidget, QPushButton, QHBoxLayout, QVBoxLayout,
-                               QFormLayout, QComboBox, QSpinBox,
+                               QFormLayout, QComboBox, QSpinBox, QLabel,
                                QDoubleSpinBox, QSizePolicy, QMessageBox,
                                QCheckBox, QRubberBand, QMenu, QDialog,
                                QTabWidget, QTableView, QHeaderView)
@@ -21,6 +21,34 @@
 from .custom_widgets import HorizontalLine
 
 
+class PlotUpdateOverlay(QWidget):
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setAttribute(QtCore.Qt.WA_TransparentForMouseEvents, True)
+        self.setAttribute(QtCore.Qt.WA_StyledBackground, True)
+        self.setFocusPolicy(QtCore.Qt.NoFocus)
+        self.setStyleSheet("background-color: rgba(20, 20, 20, 140);")
+
+        layout = QVBoxLayout(self)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setAlignment(QtCore.Qt.AlignCenter)
+
+        self.label = QLabel("Generating Plot...", self)
+        self.label.setAlignment(QtCore.Qt.AlignCenter)
+        font = self.label.font()
+        font.setPointSize(max(12, font.pointSize() + 6))
+        font.setWeight(QtGui.QFont.DemiBold)
+        self.label.setFont(font)
+        self.label.setStyleSheet("color: white;")
+        layout.addWidget(self.label)
+
+        self.hide()
+
+    def set_message(self, message: str):
+        self.label.setText(message)
+
+
 
 class PlotImage(FigureCanvas):
 
@@ -59,6 +87,7 @@ def __init__(self, model: PlotModel, parent, main_window):
         self._last_data_indicator_value = None
 
         self.menu = QMenu(self)
+        self.update_overlay = PlotUpdateOverlay(self)
 
     def enterEvent(self, event):
         self.setCursor(QtCore.Qt.CrossCursor)
@@ -122,6 +151,24 @@ def _resize(self):
         self.resize(self.parent.width() * z,
                     self.parent.height() * z)
 
+    def resizeEvent(self, event):
+        super().resizeEvent(event)
+        if self.update_overlay is not None:
+            self.update_overlay.setGeometry(self.rect())
+
+    def showUpdatingOverlay(self, message: str = "Generating Plot..."):
+        if self.update_overlay is None:
+            return
+        self.update_overlay.set_message(message)
+        self.update_overlay.setGeometry(self.rect())
+        self.update_overlay.raise_()
+        self.update_overlay.show()
+
+    def hideUpdatingOverlay(self):
+        if self.update_overlay is None:
+            return
+        self.update_overlay.hide()
+
     def saveImage(self, filename):
         """Save an image of the current view