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
113
114
115
116
117
118
|
"""
This example demonstrates linearized ColorMap objects using colormap.makeMonochrome()
or using the `ColorMap`'s `linearize()` method.
"""
import numpy as np
import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui, QtWidgets
name_list = (
'warm','neutral','cool',
'green','amber','blue','red','pink','lavender',
(0.5, 0.2, 0.1, 0.8)
)
ramp_list = [
pg.colormap.makeMonochrome(name)
for name in name_list
]
cm_list = []
# Create a gray ramp for demonstrating the idea:
cm = pg.ColorMap( None, [
QtGui.QColor( 0, 0, 0),
QtGui.QColor( 10, 10, 10),
QtGui.QColor(127, 127, 127),
QtGui.QColor(240, 240, 240),
QtGui.QColor(255, 255, 255)
])
cm_list.append(('Distorted gray ramp',cm))
# Create a rainbow scale in HSL color space:
length = 41
col_list = []
for idx in range(length):
frac = idx/(length-1)
qcol = QtGui.QColor()
qcol.setHslF( (2*frac-0.15)%1.0, 0.8, 0.5-0.5*np.cos(np.pi*frac) )
col_list.append(qcol)
cm = pg.ColorMap( None, col_list )
cm_list.append( ('Distorted HSL spiral', cm) )
# Create some random examples:
for example_idx in range(3):
previous = None
col_list = []
for idx in range(8):
values = np.random.random((3))
if previous is not None:
intermediate = (values + previous) / 2
qcol = QtGui.QColor()
qcol.setRgbF( *intermediate )
col_list.append( qcol)
qcol1 = QtGui.QColor()
qcol1.setRgbF( *values )
col_list.append( qcol1)
previous = values
cm = pg.ColorMap( None, col_list )
cm_list.append( (f'random {example_idx+1}', cm) )
app = pg.mkQApp()
win = QtWidgets.QMainWindow()
win.resize(1000,800)
lw = pg.GraphicsLayoutWidget()
lw.setFixedWidth(1000)
lw.setSizePolicy(QtWidgets.QSizePolicy.Policy.Expanding, QtWidgets.QSizePolicy.Policy.Expanding)
scr = QtWidgets.QScrollArea()
scr.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
scr.setWidget(lw)
win.setCentralWidget(scr)
win.setWindowTitle('pyqtgraph example: Linearized color maps')
win.show()
bar_width = 32
bar_data = pg.colormap.modulatedBarData(width=bar_width)
num_bars = 0
def add_heading(lw, name):
global num_bars
lw.addLabel('=== '+name+' ===')
num_bars += 1
lw.nextRow()
def add_bar(lw, name, cm):
global num_bars
lw.addLabel(name)
imi = pg.ImageItem( bar_data )
imi.setLookupTable( cm.getLookupTable(alpha=True) )
vb = lw.addViewBox(lockAspect=True, enableMouse=False)
vb.addItem( imi )
num_bars += 1
lw.nextRow()
add_heading(lw, 'ramp generator')
for cm in ramp_list:
add_bar(lw, cm.name, cm)
add_heading(lw, 'linearizer demonstration')
for (name, cm) in cm_list:
add_bar(lw, name, cm)
cm.linearize()
add_bar(lw, '> linearized', cm)
add_heading(lw, 'consistency with included maps')
for name in ('CET-C3', 'CET-L17', 'CET-L2'):
# lw.addLabel(str(name))
cm = pg.colormap.get(name)
add_bar(lw, name, cm)
cm.linearize()
add_bar(lw, '> linearized', cm)
lw.setFixedHeight(num_bars * (bar_width+5) )
if __name__ == '__main__':
pg.exec()
|
