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