Add checkbox for mesh volume normalization

Author: paulromano

openmc_plotter/docks.py

@@ -826,6 +826,11 @@ def __init__(self, model, main_window, field, colormaps=None):
         zero_connector = partial(main_window.toggleTallyMaskZero)
         self.maskZeroBox.stateChanged.connect(zero_connector)
 
+        # Volume normalization check box
+        self.volumeNormBox = QCheckBox()
+        volume_connector = partial(main_window.toggleTallyVolumeNorm)
+        self.volumeNormBox.stateChanged.connect(volume_connector)
+
         # Clip data to min/max check box
         self.clipDataBox = QCheckBox()
         clip_connector = partial(main_window.toggleTallyDataClip)
@@ -849,6 +854,7 @@ def __init__(self, model, main_window, field, colormaps=None):
         self.layout.addRow("Log Scale: ", self.scaleBox)
         self.layout.addRow("Clip Data: ", self.clipDataBox)
         self.layout.addRow("Mask Zeros: ", self.maskZeroBox)
+        self.layout.addRow("Volume normalize: ", self.volumeNormBox)
         self.layout.addRow("Contours: ", self.contoursBox)
         self.layout.addRow("Contour Levels:", self.contourLevelsLine)
         self.setLayout(self.layout)

openmc_plotter/main_window.py

@@ -951,6 +951,10 @@ def toggleTallyMaskZero(self, state):
         av = self.model.activeView
         av.tallyMaskZeroValues = bool(state)
 
+    def toggleTallyVolumeNorm(self, state):
+        av = self.model.activeView
+        av.tallyVolumeNorm = bool(state)
+
     def editTallyAlpha(self, value, apply=False):
         av = self.model.activeView
         av.tallyDataAlpha = value

openmc_plotter/plotmodel.py

@@ -696,10 +696,6 @@ def _do_op(array, tally_value, ax=0):
                 selected_scores.append(idx)
         data = _do_op(data[np.array(selected_scores)], tally_value)
 
-        # get dataset's min/max
-        data_min = np.min(data)
-        data_max = np.max(data)
-
         # Account for mesh filter translation
         if mesh_filter.translation is not None:
             t = mesh_filter.translation
@@ -716,11 +712,19 @@ def _do_op(array, tally_value, ax=0):
             pixels=(view.h_res, view.v_res),
         )
 
+        # Apply volume normalization
+        if view.tallyVolumeNorm:
+            data /= mesh_cpp.volumes
+
         # set image data
         image_data = np.full_like(self.ids, np.nan, dtype=float)
         mask = (mesh_bins >= 0)
         image_data[mask] = data[mesh_bins[mask]]
 
+        # get dataset's min/max
+        data_min = np.min(data)
+        data_max = np.max(data)
+
         return image_data, None, data_min, data_max
 
     @property
@@ -911,6 +915,7 @@ def __init__(self):
         self.tallyDataMax = np.inf
         self.tallyDataLogScale = False
         self.tallyMaskZeroValues = False
+        self.tallyVolumeNorm = False
         self.clipTallyData = False
         self.tallyValue = "Mean"
         self.tallyContours = False