pyqtgraph/examples/colorMapsLinearized.py
Ogi Moore d455da9aec Use Qt6 Enum Namespace
This namespace appears to be valid in PySide2/PyQt5 5.12+ so we may as
well migrate to the newer namespace ourselves.
2021-06-08 21:23:49 -07:00

121 lines
3.1 KiB
Python

# -*- coding: utf-8 -*-
"""
This example demonstrates linearized ColorMap objects using colormap.makeMonochrome()
or using the `ColorMap`'s `linearize()` method.
"""
# Add path to library (just for examples; you do not need this)
import initExample
import numpy as np
from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph as pg
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 = QtGui.QMainWindow()
win.resize(1000,800)
lw = pg.GraphicsLayoutWidget()
lw.setFixedWidth(1000)
lw.setSizePolicy(QtGui.QSizePolicy.Policy.Expanding, QtGui.QSizePolicy.Policy.Expanding)
scr = QtGui.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()