1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
import numpy as np
from .. import colormap
from .. import functions as fn
from ..Qt import QtCore, QtGui, QtWidgets
from .ColorMapMenu import ColorMapMenu
__all__ = ['ColorMapButton']
class ColorMapDisplayMixin:
# Note that this Mixin attempts to be safe for inheritance
# on either the lhs or rhs. To this end:
# 1) the __init__ is safe to be called more than once
# 2) it does not override any main class virtual methods
def __init__(self, *, orientation):
self.horizontal = orientation == 'horizontal'
self._menu = None
self._setColorMap(None)
def setMaximumThickness(self, val):
# calls main class methods
Thickness = 'Height' if self.horizontal else 'Width'
getattr(self, f'setMaximum{Thickness}')(val)
def _setColorMap(self, cmap):
if isinstance(cmap, str):
try:
cmap = colormap.get(cmap)
except FileNotFoundError:
cmap = None
if cmap is None:
cmap = colormap.ColorMap(None, [0.0, 1.0])
self._cmap = cmap
self._image = None
def setColorMap(self, cmap):
# calls main class methods
self._setColorMap(cmap)
self.colorMapChanged()
def colorMap(self):
return self._cmap
def getImage(self):
if self._image is None:
lut = self._cmap.getLookupTable(nPts=256, alpha=True)
lut = np.expand_dims(lut, axis=0 if self.horizontal else 1)
qimg = fn.ndarray_to_qimage(lut, QtGui.QImage.Format.Format_RGBA8888)
self._image = qimg if self.horizontal else qimg.mirrored()
return self._image
def getMenu(self):
if self._menu is None:
self._menu = ColorMapMenu(showColorMapSubMenus=True)
self._menu.sigColorMapTriggered.connect(self.setColorMap)
return self._menu
def paintColorMap(self, painter, rect):
painter.save()
image = self.getImage()
painter.drawImage(rect, image)
if not self.horizontal:
painter.translate(rect.center())
painter.rotate(-90)
painter.translate(-rect.center())
text = self.colorMap().name
wpen = QtGui.QPen(QtCore.Qt.GlobalColor.white)
bpen = QtGui.QPen(QtCore.Qt.GlobalColor.black)
# get an estimate of the lightness of the colormap
# from its center element
lightness = image.pixelColor(image.rect().center()).lightnessF()
pen = bpen if lightness >= 0.55 else wpen
AF = QtCore.Qt.AlignmentFlag
trect = painter.boundingRect(rect, AF.AlignCenter, text)
# draw the foreground text
painter.setPen(pen)
painter.drawText(trect, text)
painter.restore()
class ColorMapButton(ColorMapDisplayMixin, QtWidgets.QWidget):
sigColorMapChanged = QtCore.Signal(object)
def __init__(self):
QtWidgets.QWidget.__init__(self)
ColorMapDisplayMixin.__init__(self, orientation='horizontal')
def colorMapChanged(self):
cmap = self.colorMap()
self.sigColorMapChanged.emit(cmap)
self.update()
def paintEvent(self, evt):
painter = QtGui.QPainter(self)
self.paintColorMap(painter, self.contentsRect())
painter.end()
def mouseReleaseEvent(self, evt):
if evt.button() != QtCore.Qt.MouseButton.LeftButton:
return
# position the menu below the widget
pos = self.mapToGlobal(self.pos())
pos.setY(pos.y() + self.height())
self.getMenu().popup(pos)
|
