From dcc554eab84b025cfeac3211d128bd177849b090 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Fri, 12 May 2023 14:59:18 +0200
Subject: [PATCH] allow to set color map

---
 grum/descs/imgdesc.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/grum/descs/imgdesc.py b/grum/descs/imgdesc.py
index 4706638..0d29e75 100644
--- a/grum/descs/imgdesc.py
+++ b/grum/descs/imgdesc.py
@@ -1,16 +1,18 @@
 import numpy as np
+import pyqtgraph as pg
 from .desc import Description
 
 
 class ImageDescription(Description):
 
-    def __init__(self, name, title=None, xlabel=None, ylabel=None, image=None, levels=None):
+    def __init__(self, name, title=None, xlabel=None, ylabel=None, image=None, levels=None, cmap="viridis"):
         self.name = name
         self.title = title
         self.xlabel = xlabel
         self.ylabel = ylabel
         self.image = image
         self.levels = levels #TODO: might be better to use vmin and vmax
+        self.cmap = cmap
 
     @property
     def data(self):
@@ -40,6 +42,10 @@ class ImageDescription(Description):
         if self.ylabel:
             plotwidget.getView().setLabel("left", self.ylabel)
 
+        if self.cmap:
+            cm = pg.colormap.get(self.cmap)
+            plotwidget.setColorMap(cm)
+
         return res