Change docks to tabs

Author: paulromano

openmc_plotter/docks.py

@@ -7,7 +7,8 @@
                                QGroupBox, QFormLayout, QLabel, QLineEdit,
                                QComboBox, QSpinBox, QDoubleSpinBox, QSizePolicy,
                                QCheckBox, QDockWidget, QScrollArea, QListWidget,
-                               QListWidgetItem, QTreeWidget, QTreeWidgetItem)
+                               QListWidgetItem, QTreeWidget, QTreeWidgetItem,
+                               QTabWidget)
 import matplotlib.pyplot as plt
 import numpy as np
 import openmc
@@ -18,26 +19,26 @@
                         _REACTION_UNITS, _SPATIAL_FILTERS)
 
 
-class PlotterDock(QDockWidget):
+class PlotterPanel(QWidget):
     """
-    Dock widget with common settings for the plotting application
+    Base panel widget with common settings for the plotting application
     """
 
-    def __init__(self, model, font_metric, parent=None):
+    def __init__(self, model, font_metric, main_window, parent=None):
         super().__init__(parent)
 
         self.model = model
         self.font_metric = font_metric
-        self.main_window = parent
+        self.main_window = main_window
 
         self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)
 
 
-class MeshAnnotationDock(PlotterDock):
-    """Dock for mesh annotation options"""
+class MeshAnnotationPanel(PlotterPanel):
+    """Panel for mesh annotation options"""
 
-    def __init__(self, model, font_metric, parent=None):
-        super().__init__(model, font_metric, parent)
+    def __init__(self, model, font_metric, main_window, parent=None):
+        super().__init__(model, font_metric, main_window, parent)
 
         self.treeLayout = QVBoxLayout()
         self.meshTree = QTreeWidget()
@@ -56,55 +57,88 @@ def __init__(self, model, font_metric, parent=None):
 
         self.meshTree.setHeaderHidden(True)
 
-        # Create submit button
-        self.applyButton = QPushButton("Apply Changes")
-        # Mac bug fix
-        self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
-        self.applyButton.clicked.connect(self.main_window.applyChanges)
-
-        label = QLabel("Mesh Annotations")
-        self.treeLayout.addWidget(label)
         self.treeLayout.addWidget(self.meshTree)
-        self.treeLayout.addWidget(HorizontalLine())
-        self.treeLayout.addWidget(self.applyButton)
 
-        self.optionsWidget = QWidget()
-        self.optionsWidget.setLayout(self.treeLayout)
-        self.setWidget(self.optionsWidget)
+        self.setLayout(self.treeLayout)
 
     def get_checked_meshes(self):
         return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]
 
     def update(self):
         pass
 
+
+class TabbedDock(QDockWidget):
+    """
+    Dock widget containing tabbed panels for Geometry, Tallies, and Mesh Annotations
+    """
+
+    def __init__(self, model, font_metric, parent=None):
+        super().__init__(parent)
+
+        self.model = model
+        self.font_metric = font_metric
+        self.main_window = parent
+
+        self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)
+        self.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
+
+        # Create the tab widget
+        self.tabWidget = QTabWidget()
+
+        # Create the three panels
+        self.geometryPanel = GeometryPanel(model, font_metric, parent, self)
+        self.tallyPanel = TallyPanel(model, font_metric, parent, self)
+        self.meshAnnotationPanel = MeshAnnotationPanel(model, font_metric, parent, self)
+
+        # Add panels as tabs
+        self.tabWidget.addTab(self.geometryPanel, "Geometry")
+        self.tabWidget.addTab(self.tallyPanel, "Tallies")
+        self.tabWidget.addTab(self.meshAnnotationPanel, "Mesh Annotations")
+
+        # Create Apply Changes button
+        self.applyButton = QPushButton("Apply Changes")
+        self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
+        self.applyButton.clicked.connect(self.main_window.applyChanges)
+
+        # Main layout with tabs and apply button
+        self.mainLayout = QVBoxLayout()
+        self.mainLayout.addWidget(self.tabWidget)
+        self.mainLayout.addWidget(HorizontalLine())
+        self.mainLayout.addWidget(self.applyButton)
+
+        # Create container widget
+        self.containerWidget = QWidget()
+        self.containerWidget.setLayout(self.mainLayout)
+        self.setWidget(self.containerWidget)
+
+    def updateDock(self):
+        """Update geometry panel"""
+        self.geometryPanel.updateDock()
+
+    def update(self):
+        """Update tally panel"""
+        self.tallyPanel.update()
+
     def resizeEvent(self, event):
         self.main_window.resizeEvent(event)
 
     hideEvent = showEvent = moveEvent = resizeEvent
 
 
-class DomainDock(PlotterDock):
+class GeometryPanel(PlotterPanel):
     """
-    Domain options dock
+    Geometry options panel
     """
 
-    def __init__(self, model, font_metric, parent=None):
-        super().__init__(model, font_metric, parent)
-
-        self.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea)
+    def __init__(self, model, font_metric, main_window, parent=None):
+        super().__init__(model, font_metric, main_window, parent)
 
         # Create Controls
         self._createOriginBox()
         self._createOptionsBox()
         self._createResolutionBox()
 
-        # Create submit button
-        self.applyButton = QPushButton("Apply Changes")
-        # Mac bug fix
-        self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
-        self.applyButton.clicked.connect(self.main_window.applyChanges)
-
         # Create Zoom box
         self.zoomBox = QSpinBox()
         self.zoomBox.setSuffix(' %')
@@ -120,22 +154,15 @@ def __init__(self, model, font_metric, parent=None):
         self.zoomWidget.setLayout(self.zoomLayout)
 
         # Create Layout
-        self.dockLayout = QVBoxLayout()
-        self.dockLayout.addWidget(QLabel("Geometry/Properties"))
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addWidget(self.originGroupBox)
-        self.dockLayout.addWidget(self.optionsGroupBox)
-        self.dockLayout.addWidget(self.resGroupBox)
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addWidget(self.zoomWidget)
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addStretch()
-        self.dockLayout.addWidget(self.applyButton)
-        self.dockLayout.addWidget(HorizontalLine())
-
-        self.optionsWidget = QWidget()
-        self.optionsWidget.setLayout(self.dockLayout)
-        self.setWidget(self.optionsWidget)
+        self.panelLayout = QVBoxLayout()
+        self.panelLayout.addWidget(self.originGroupBox)
+        self.panelLayout.addWidget(self.optionsGroupBox)
+        self.panelLayout.addWidget(self.resGroupBox)
+        self.panelLayout.addWidget(HorizontalLine())
+        self.panelLayout.addWidget(self.zoomWidget)
+        self.panelLayout.addStretch()
+
+        self.setLayout(self.panelLayout)
 
     def _createOriginBox(self):
 
@@ -362,20 +389,12 @@ def revertToCurrent(self):
         self.widthBox.setValue(cv.width)
         self.heightBox.setValue(cv.height)
 
-    def resizeEvent(self, event):
-        self.main_window.resizeEvent(event)
-
-    hideEvent = showEvent = moveEvent = resizeEvent
-
-
-class TallyDock(PlotterDock):
-
-    def __init__(self, model, font_metric, parent=None):
-        super().__init__(model, font_metric, parent)
+class TallyPanel(PlotterPanel):
 
-        self.setAllowedAreas(QtCore.Qt.RightDockWidgetArea)
+    def __init__(self, model, font_metric, main_window, parent=None):
+        super().__init__(model, font_metric, main_window, parent)
 
-        # Dock maps for tally information
+        # Panel maps for tally information
         self.tally_map = {}
         self.filter_map = {}
         self.score_map = {}
@@ -395,11 +414,6 @@ def __init__(self, model, font_metric, parent=None):
         self.tallyGroupBox = QGroupBox('Selected Tally')
         self.tallyGroupBox.setLayout(self.tallySelectorLayout)
 
-        # Create submit button
-        self.applyButton = QPushButton("Apply Changes")
-        self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
-        self.applyButton.clicked.connect(self.main_window.applyChanges)
-
         # Color options section
         self.tallyColorForm = ColorForm(self.model, self.main_window, 'tally')
         self.scoresGroupBox = Expander(title="Scores:")
@@ -408,23 +422,24 @@ def __init__(self, model, font_metric, parent=None):
         self.nuclidesListWidget = QListWidget()
 
         # Main layout
-        self.dockLayout = QVBoxLayout()
-        self.dockLayout.addWidget(QLabel("Tallies"))
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addWidget(self.tallyGroupBox)
-        self.dockLayout.addStretch()
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addWidget(self.tallyColorForm)
-        self.dockLayout.addWidget(HorizontalLine())
-        self.dockLayout.addWidget(self.applyButton)
-
-        # Create widget for dock and apply main layout
+        self.panelLayout = QVBoxLayout()
+        self.panelLayout.addWidget(self.tallyGroupBox)
+        self.panelLayout.addStretch()
+        self.panelLayout.addWidget(HorizontalLine())
+        self.panelLayout.addWidget(self.tallyColorForm)
+
+        # Create widget for scroll area and apply main layout
         self.scroll = QScrollArea()
         self.scroll.setWidgetResizable(True)
         self.widget = QWidget()
-        self.widget.setLayout(self.dockLayout)
+        self.widget.setLayout(self.panelLayout)
         self.scroll.setWidget(self.widget)
-        self.setWidget(self.scroll)
+
+        # Set scroll area as main layout
+        mainLayout = QVBoxLayout()
+        mainLayout.setContentsMargins(0, 0, 0, 0)
+        mainLayout.addWidget(self.scroll)
+        self.setLayout(mainLayout)
 
     def _createFilterTree(self, spatial_filters):
         av = self.model.activeView