Add checkbox for reversing colormap (#137)

openmc-dev · Apr 25, 2024 · e93a120 · e93a120
1 parent 3c3e04c
commit e93a120
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 2 deletions.
diff --git a/openmc_plotter/docks.py b/openmc_plotter/docks.py
@@ -776,6 +776,11 @@ def __init__(self, model, main_window, field, colormaps=None):
         cmap_connector = partial(main_window.editTallyDataColormap)
         self.colormapBox.currentTextChanged[str].connect(cmap_connector)
 
+        # Color map reverse
+        self.reverseCmapBox = QCheckBox()
+        reverse_connector = partial(main_window.toggleReverseCmap)
+        self.reverseCmapBox.stateChanged.connect(reverse_connector)
+
         # Data indicator line check box
         self.dataIndicatorCheckBox = QCheckBox()
         data_indicator_connector = partial(
@@ -831,6 +836,7 @@ def __init__(self, model, main_window, field, colormaps=None):
         self.layout.addRow("Visible:", self.visibilityBox)
         self.layout.addRow("Alpha: ", self.alphaBox)
         self.layout.addRow("Colormap: ", self.colormapBox)
+        self.layout.addRow("Reverse colormap: ", self.reverseCmapBox)
         self.layout.addRow("Data Indicator: ", self.dataIndicatorCheckBox)
         self.layout.addRow("Custom Min/Max: ", self.userMinMaxBox)
         self.layout.addRow("Min: ", self.minBox)

diff --git a/openmc_plotter/main_window.py b/openmc_plotter/main_window.py
@@ -1004,6 +1004,10 @@ def editTallyDataColormap(self, cmap, apply=False):
         if apply:
             self.applyChanges()
 
+    def toggleReverseCmap(self, state):
+        av = self.model.activeView
+        av.tallyDataReverseCmap = bool(state)
+
     def updateTallyMinMax(self):
         self.tallyDock.updateMinMax()
 

diff --git a/openmc_plotter/plotgui.py b/openmc_plotter/plotgui.py
@@ -587,21 +587,25 @@ def updatePixmap(self):
 
             norm = SymLogNorm(1E-30) if cv.tallyDataLogScale else None
 
+            cmap = cv.tallyDataColormap
+            if cv.tallyDataReverseCmap:
+                cmap += '_r'
+
             if cv.tallyContours:
                 # parse the levels line
                 levels = self.parseContoursLine(cv.tallyContourLevels)
                 self.tally_image = self.ax.contour(image_data,
                                                    origin='image',
                                                    levels=levels,
                                                    alpha=cv.tallyDataAlpha,
-                                                   cmap=cv.tallyDataColormap,
+                                                   cmap=cmap,
                                                    norm=norm,
                                                    extent=extents)
 
             else:
                 self.tally_image = self.ax.imshow(image_data,
                                                   alpha=cv.tallyDataAlpha,
-                                                  cmap=cv.tallyDataColormap,
+                                                  cmap=cmap,
                                                   norm=norm,
                                                   extent=extents)
             # add colorbar

diff --git a/openmc_plotter/plotmodel.py b/openmc_plotter/plotmodel.py
@@ -921,6 +921,7 @@ def __init__(self):
 
         # Tally Viz Settings
         self.tallyDataColormap = 'Spectral'
+        self.tallyDataReverseCmap = False
         self.tallyDataVisible = True
         self.tallyDataAlpha = 1.0
         self.tallyDataIndicator = False