SwissFEL/grum
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

allow to set color map

Browse Source
This commit is contained in:
augustin_s
2023-05-12 14:59:18 +02:00
parent cf24d54b62
commit dcc554eab8
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
grum/descs/imgdesc.py
View File

@ -1,16 +1,18 @@
import numpy as np import numpy as np
import pyqtgraph as pg
from .desc import Description from .desc import Description
class ImageDescription(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.name = name
self.title = title self.title = title
self.xlabel = xlabel self.xlabel = xlabel
self.ylabel = ylabel self.ylabel = ylabel
self.image = image self.image = image
self.levels = levels #TODO: might be better to use vmin and vmax self.levels = levels #TODO: might be better to use vmin and vmax
self.cmap = cmap
@property @property
def data(self): def data(self):
@ -40,6 +42,10 @@ class ImageDescription(Description):
if self.ylabel: if self.ylabel:
plotwidget.getView().setLabel("left", self.ylabel) plotwidget.getView().setLabel("left", self.ylabel)
if self.cmap:
cm = pg.colormap.get(self.cmap)
plotwidget.setColorMap(cm)
return res return res