Merge pull request #50 from pshriwise/universe_levels

Plotting by universe level

Author: paulromano

openmc_plotter/docks.py

@@ -145,6 +145,14 @@ def _createOptionsBox(self):
         self.colorbyBox.currentTextChanged[str].connect(
             self.main_window.editColorBy)
 
+        # Universe level (applies to cell coloring only)
+        self.universeLevelBox = QComboBox(self)
+        self.universeLevelBox.addItem('all')
+        for i in range(self.model.max_universe_levels):
+            self.universeLevelBox.addItem(str(i))
+        self.universeLevelBox.currentTextChanged[str].connect(
+            self.main_window.editUniverseLevel)
+
         # Alpha
         self.domainAlphaBox = QDoubleSpinBox(self)
         self.domainAlphaBox.setValue(self.model.activeView.domainAlpha)
@@ -180,6 +188,7 @@ def _createOptionsBox(self):
         self.opLayout.addRow('Height:', self.heightBox)
         self.opLayout.addRow('Basis:', self.basisBox)
         self.opLayout.addRow('Color By:', self.colorbyBox)
+        self.opLayout.addRow('Universe Level:', self.universeLevelBox)
         self.opLayout.addRow('Plot alpha:', self.domainAlphaBox)
         self.opLayout.addRow('Visible:', self.visibilityBox)
         self.opLayout.addRow('Outlines:', self.outlinesBox)
@@ -229,6 +238,7 @@ def updateDock(self):
         self.updateWidth()
         self.updateHeight()
         self.updateColorBy()
+        self.updateUniverseLevel()
         self.updatePlotAlpha()
         self.updatePlotVisibility()
         self.updateOutlines()
@@ -250,6 +260,13 @@ def updateHeight(self):
 
     def updateColorBy(self):
         self.colorbyBox.setCurrentText(self.model.activeView.colorby)
+        if self.model.activeView.colorby != 'cell':
+            self.universeLevelBox.setEnabled(False)
+        else:
+            self.universeLevelBox.setEnabled(True)
+
+    def updateUniverseLevel(self):
+        self.universeLevelBox.setCurrentIndex(self.model.activeView.level + 1)
 
     def updatePlotAlpha(self):
         self.domainAlphaBox.setValue(self.model.activeView.domainAlpha)

openmc_plotter/main_window.py

@@ -26,6 +26,7 @@
 from .overlays import ShortcutsOverlay
 from .tools import ExportDataDialog
 
+_COORD_LEVELS = 0
 
 def _openmcReload():
     # reset OpenMC memory, instances
@@ -35,7 +36,6 @@ def _openmcReload():
     openmc.lib.settings.output_summary = False
     openmc.lib.init(["-c"])
 
-
 class MainWindow(QMainWindow):
     def __init__(self, font=QtGui.QFontMetrics(QtGui.QFont()), screen_size=QtCore.QSize()):
         super().__init__()
@@ -627,6 +627,16 @@ def editColorBy(self, domain_kind, apply=False):
         if apply:
             self.applyChanges()
 
+    def editUniverseLevel(self, level, apply=False):
+        if level == 'all':
+            self.model.activeView.level = -1
+        else:
+            self.model.activeView.level = int(level)
+        self.dock.updateUniverseLevel()
+        self.colorDialog.updateUniverseLevel()
+        if apply:
+            self.applyChanges()
+
     def toggleOverlaps(self, state, apply=False):
         self.model.activeView.color_overlaps = bool(state)
         self.colorDialog.updateOverlap()

openmc_plotter/plotgui.py

@@ -776,6 +776,13 @@ def createGeneralTab(self):
         self.colorbyBox.addItem("cell")
         self.colorbyBox.addItem("temperature")
         self.colorbyBox.addItem("density")
+        self.colorbyBox.currentTextChanged[str].connect(main_window.editColorBy)
+
+        self.universeLevelBox = QComboBox(self)
+        self.universeLevelBox.addItem('all')
+        for i in range(self.model.max_universe_levels):
+            self.universeLevelBox.addItem(str(i))
+        self.universeLevelBox.currentTextChanged[str].connect(main_window.editUniverseLevel)
 
         # Overlap plotting
         self.overlapCheck = QCheckBox('', self)
@@ -788,8 +795,6 @@ def createGeneralTab(self):
         self.overlapColorButton.setFixedHeight(self.font_metric.height() * 1.5)
         self.overlapColorButton.clicked.connect(main_window.editOverlapColor)
 
-        self.colorbyBox.currentTextChanged[str].connect(main_window.editColorBy)
-
         self.colorResetButton = QPushButton("&Reset Colors")
         self.colorResetButton.setCursor(QtCore.Qt.PointingHandCursor)
         self.colorResetButton.clicked.connect(main_window.resetColors)
@@ -813,6 +818,7 @@ def createGeneralTab(self):
         formLayout.addRow('OVerlap Color:', self.overlapColorButton)
         formLayout.addRow(HorizontalLine())
         formLayout.addRow('Color Plot By:', self.colorbyBox)
+        formLayout.addRow('Universe Level:', self.universeLevelBox)
         formLayout.addRow(self.colorResetButton, None)
 
         generalLayout = QHBoxLayout()
@@ -973,6 +979,7 @@ def updateDialogValues(self):
         self.updateSeed()
         self.updateBackgroundColor()
         self.updateColorBy()
+        self.updateUniverseLevel()
         self.updateDomainTabs()
         self.updateOverlap()
         self.updateOverlapColor()
@@ -1049,6 +1056,14 @@ def updateColorBy(self):
         colorby = self.model.activeView.colorby
         self.colorbyBox.setCurrentText(colorby)
         self.overlapCheck.setEnabled(colorby in ("cell", "material"))
+        self.universeLevelBox.setEnabled(colorby == 'cell')
+
+    def updateUniverseLevel(self):
+        level = self.model.activeView.level
+        if level == -1:
+            self.universeLevelBox.setCurrentText('all')
+        else:
+            self.universeLevelBox.setCurrentText(str(level))
 
     def updateDomainTabs(self):
         self.cellTable.setModel(self.main_window.cellsModel)

openmc_plotter/plotmodel.py

@@ -96,6 +96,7 @@ def __init__(self):
         # Retrieve OpenMC Cells/Materials
         self.modelCells = openmc.lib.cells
         self.modelMaterials = openmc.lib.materials
+        self.max_universe_levels = openmc.lib._coord_levels()
 
         # Cell/Material ID by coordinates
         self.ids = None
@@ -791,7 +792,7 @@ def urc(self):
         else:
             x = self.origin[0] + self.width / 2.0
             y = self.origin[1]
-            z = self.origin[2] + height / 2.0
+            z = self.origin[2] + self.height / 2.0
         return x, y, z
 
     def adopt_plotbase(self, view):