ASCEE/pyqtgraph
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
7aa5d0cbf4
pyqtgraph/examples/colorMaps.py

76 lines
2.1 KiB
Python
Raw Normal View History

Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
# -*- coding: utf-8 -*-
"""
Fix Matplotlib color map import, add CC0 maps locally (#1865) * accept Matplotlib colormap data as numpy array, add CC0 Matplotlib maps locally * add cividis, prioritize lower case in example/atlas
2021-06-26 15:41:57 +00:00
This example displays all color maps currently available, either as local data
or imported from Matplotlib of ColorCET.
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
"""
## Add path to library (just for examples; you do not need this)
import initExample
from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph as pg
app = pg.mkQApp()
win = QtGui.QMainWindow()
win.resize(1000,800)
lw = pg.GraphicsLayoutWidget()
lw.setFixedWidth(1000)
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-06 01:17:43 +00:00
lw.setSizePolicy(QtGui.QSizePolicy.Policy.Expanding, QtGui.QSizePolicy.Policy.Expanding)
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
scr = QtGui.QScrollArea()
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-06 01:17:43 +00:00
scr.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
scr.setWidget(lw)
win.setCentralWidget(scr)
win.setWindowTitle('pyqtgraph example: Color maps')
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
win.show()
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
bar_width = 32
bar_data = pg.colormap.modulatedBarData(width=bar_width)
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
num_bars = 0
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
def add_heading(lw, name):
global num_bars
lw.addLabel('=== '+name+' ===')
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
num_bars += 1
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
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) )
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
vb = lw.addViewBox(lockAspect=True, enableMouse=False)
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
vb.addItem( imi )
num_bars += 1
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
lw.nextRow()
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
add_heading(lw, 'local color maps')
list_of_maps = pg.colormap.listMaps()
Fix Matplotlib color map import, add CC0 maps locally (#1865) * accept Matplotlib colormap data as numpy array, add CC0 Matplotlib maps locally * add cividis, prioritize lower case in example/atlas
2021-06-26 15:41:57 +00:00
list_of_maps = sorted( list_of_maps, key=lambda x: x.swapcase() )
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
for map_name in list_of_maps:
cm = pg.colormap.get(map_name)
add_bar(lw, map_name, cm)
add_heading(lw, 'Matplotlib import')
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
list_of_maps = pg.colormap.listMaps('matplotlib')
Fix Matplotlib color map import, add CC0 maps locally (#1865) * accept Matplotlib colormap data as numpy array, add CC0 Matplotlib maps locally * add cividis, prioritize lower case in example/atlas
2021-06-26 15:41:57 +00:00
list_of_maps = sorted( list_of_maps, key=lambda x: x.lower() )
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
for map_name in list_of_maps:
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
cm = pg.colormap.get(map_name, source='matplotlib', skipCache=True)
if cm is not None:
add_bar(lw, map_name, cm)
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
add_heading(lw, 'ColorCET import')
Fix Matplotlib color map import, add CC0 maps locally (#1865) * accept Matplotlib colormap data as numpy array, add CC0 Matplotlib maps locally * add cividis, prioritize lower case in example/atlas
2021-06-26 15:41:57 +00:00
list_of_maps = pg.colormap.listMaps('colorcet')
list_of_maps = sorted( list_of_maps, key=lambda x: x.lower() )
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
for map_name in list_of_maps:
Color map linearization (#1797) * extended color map functions * assertion string, documentation pass, comment purge * fix some documentation links * simplify assert statement * removed comments and redundancy, renamed modulated bar * include modulatedBarData in documentation * test running matplotlib with dummy backend * skip color maps example on OSX/PySide6.1.1 * removed skipping of tests * reverted some accidental whitespace, removed unneeded numpy import * removed unneded alpha parameter
2021-06-09 03:41:46 +00:00
cm = pg.colormap.get(map_name, source='colorcet', skipCache=True)
if cm is not None:
add_bar(lw, map_name, cm)
lw.setFixedHeight(num_bars * (bar_width+5) )
Colormap extension (#1424) * extend ColorMap functionality for palette management * manual merge with changes in master branch * extend ColorMap functionality for palette management * manual merge with changes in master branch * light mode / dark mode swapping demo * color map updates * added ColorCET and Matplotlib import for color maps * minor cleanup * restored lost indent * Code cleanup * Add colorcet as optional dependency Co-authored-by: Ogi Moore <ognyan.moore@gmail.com>
2021-01-05 06:01:30 +00:00
if __name__ == '__main__':
change all examples to use pg.exec()
2021-05-13 21:28:22 +00:00
pg.exec()
Reference in New Issue Copy Permalink