Merge pull request #356 from wpfff/fix/pyqtgraphcolors

fix: pyqtgraph colormaps
toolsforexperiments · Dec 6, 2022 · 3635c6c · 3635c6c
2 parents b2cd882 + de18c45
commit 3635c6c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/plottr/config/plottrcfg_main.py b/plottr/config/plottrcfg_main.py
@@ -39,5 +39,6 @@
         'line_symbols': ['o', ],
         'line_symbol_size': 7,
         'minimum_plot_size': (400, 400),
+        'default_colormap': 'magma',
     }
 }
diff --git a/plottr/plot/pyqtgraph/plots.py b/plottr/plot/pyqtgraph/plots.py
@@ -6,7 +6,7 @@
 import numpy as np
 import pyqtgraph as pg
 
-from plottr import QtCore, QtWidgets
+from plottr import QtCore, QtWidgets, config_entry
 
 __all__ = ['PlotBase', 'Plot']
 
@@ -73,9 +73,14 @@ class PlotWithColorbar(PlotBase):
     def __init__(self, parent: Optional[QtWidgets.QWidget] = None) -> None:
         super().__init__(parent)
 
-        cmap = pg.colormap.get('viridis', source='matplotlib')
+        cmap_name = config_entry('main', 'pyqtgraph', 'default_colormap')
+        try:
+            cmap = pg.colormap.get(cmap_name)
+        except:
+            cmap = pg.colormap.get('CET-L1')
+
         self.colorbar: pg.ColorBarItem = pg.ColorBarItem(interactive=True, values=(0, 1),
-                                                         cmap=cmap, width=15)
+                                                         colorMap=cmap, width=15)
         self.graphicsLayout.addItem(self.colorbar)
 
         self.img: Optional[pg.ImageItem] = None