ASCEE/pyqtgraph
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

more readable names for color map references in ColorBarItem API (#1908)

Browse Source 
* more readable names for color map references

* changed example code

* activated deprecation warnings

* added accessor methods for cmap/colorMap
This commit is contained in:
Nils Nemitz 2021-07-24 06:45:57 +09:00 committed by GitHub
parent 81823768c0
commit 7009084e4c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 50 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/ColorBarItem.py
View File

@ -41,7 +41,7 @@ class MainWindow(QtWidgets.QMainWindow):
cmap = pg.colormap.get('CET-L9') cmap = pg.colormap.get('CET-L9')
bar = pg.ColorBarItem( bar = pg.ColorBarItem(
interactive=False, values= (0, 30_000), cmap=cmap, interactive=False, values= (0, 30_000), colorMap=cmap,
label='vertical fixed color bar' label='vertical fixed color bar'
) )
bar.setImageItem( i1, insert_in=p1 ) bar.setImageItem( i1, insert_in=p1 )
@ -59,11 +59,11 @@ class MainWindow(QtWidgets.QMainWindow):
cmap = pg.colormap.get('CET-L4') cmap = pg.colormap.get('CET-L4')
bar = pg.ColorBarItem( bar = pg.ColorBarItem(
values = (0, 30_000), values = (0, 30_000),
cmap=cmap, colorMap=cmap,
label='horizontal color bar', label='horizontal color bar',
limits = (0, None), limits = (0, None),
rounding=1000, rounding=1000,
orientation = 'horizontal', orientation = 'h',
pen='#8888FF', hoverPen='#EEEEFF', hoverBrush='#EEEEFF80' pen='#8888FF', hoverPen='#EEEEFF', hoverBrush='#EEEEFF80'
) )
bar.setImageItem( i2, insert_in=p2 ) bar.setImageItem( i2, insert_in=p2 )
@ -83,7 +83,7 @@ class MainWindow(QtWidgets.QMainWindow):
limits = (-30_000, 30_000), # start with full range... limits = (-30_000, 30_000), # start with full range...
rounding=1000, rounding=1000,
width = 10, width = 10,
cmap=cmap ) colorMap=cmap )
bar.setImageItem( [i3, i4] ) bar.setImageItem( [i3, i4] )
bar.setLevels( low=-5_000, high=15_000) # ... then adjust to retro sunset. bar.setLevels( low=-5_000, high=15_000) # ... then adjust to retro sunset.

61
pyqtgraph/graphicsItems/ColorBarItem.py
View File

@ -7,6 +7,7 @@ from .LinearRegionItem import LinearRegionItem
import weakref import weakref
import math import math
import warnings
import numpy as np import numpy as np
__all__ = ['ColorBarItem'] __all__ = ['ColorBarItem']
@ -35,33 +36,40 @@ class ColorBarItem(PlotItem):
sigLevelsChanged = QtCore.Signal(object) sigLevelsChanged = QtCore.Signal(object)
sigLevelsChangeFinished = QtCore.Signal(object) sigLevelsChangeFinished = QtCore.Signal(object)
def __init__(self, values=(0,1), width=25, cmap=None, label=None, def __init__(self, values=(0,1), width=25, colorMap=None, label=None,
interactive=True, limits=None, rounding=1, interactive=True, limits=None, rounding=1,
orientation='vertical', pen='w', hoverPen='r', hoverBrush='#FF000080' ): orientation='vertical', pen='w', hoverPen='r', hoverBrush='#FF000080', cmap=None ):
""" """
Create a new ColorBarItem. Creates a new ColorBarItem.
============== =========================================================================== ============== ===========================================================================
**Arguments:** **Arguments:**
values The range of values as tuple (min, max) values The range of values as tuple (min, max)
width (default=25) The width of the displayed color bar width (default=25) The width of the displayed color bar
cmap ColorMap object, look-up table is also applied to assigned ImageItem(s) colorMap ColorMap object, look-up table is also applied to assigned ImageItem(s)
label (default=None) Label applied to color bar axis label (default=None) Label applied to color bar axis
interactive (default=True) Handles are displayed to interactively adjust level range interactive (default=True) Handles are displayed to interactively adjust level range
limits Limits to adjustment range as (low, high) tuple, None disables limit limits Limits to adjustment range as (low, high) tuple, None disables limit
rounding (default=1) Adjusted range values are rounded to multiples of this values rounding (default=1) Adjusted range values are rounded to multiples of this values
orientation (default='vertical') 'horizontal' gives a horizontal color bar orientation (default='vertical') 'horizontal' or 'h' gives a horizontal color bar.
pen color of adjustement handles in interactive mode pen color of adjustement handles in interactive mode
hoverPen color of adjustement handles when hovered over hoverPen color of adjustement handles when hovered over
hoverBrush color of movable center region when hovered over hoverBrush color of movable center region when hovered over
============== =========================================================================== ============== ===========================================================================
""" """
super().__init__() super().__init__()
if cmap is not None:
warnings.warn(
"The parameter 'cmap' has been renamed to 'colorMap' for clarity. "
"The old name will no longer be available in any version of PyQtGraph released after July 2022.",
DeprecationWarning, stacklevel=2
)
colorMap = cmap
self.img_list = [] # list of controlled ImageItems self.img_list = [] # list of controlled ImageItems
self.values = values self.values = values
self.cmap = cmap self._colorMap = colorMap
self.rounding = rounding self.rounding = rounding
self.horizontal = bool(orientation == 'horizontal') self.horizontal = bool( orientation in ('h', 'horizontal') )
self.lo_prv, self.hi_prv = self.values # remember previous values while adjusting range self.lo_prv, self.hi_prv = self.values # remember previous values while adjusting range
if limits is None: if limits is None:
@ -107,7 +115,7 @@ class ColorBarItem(PlotItem):
self.bar.setImage( np.linspace(0, 1, 256).reshape( (1,-1) ) ) self.bar.setImage( np.linspace(0, 1, 256).reshape( (1,-1) ) )
if label is not None: self.getAxis('left').setLabel(label) if label is not None: self.getAxis('left').setLabel(label)
self.addItem(self.bar) self.addItem(self.bar)
if cmap is not None: self.setCmap(cmap) if cmap is not None: self.setColorMap(cmap)
if interactive: if interactive:
if self.horizontal: if self.horizontal:
@ -156,21 +164,44 @@ class ColorBarItem(PlotItem):
insert_in.layout.setColumnFixedWidth(4, 5) # enforce some space to axis on the left insert_in.layout.setColumnFixedWidth(4, 5) # enforce some space to axis on the left
self._update_items( update_cmap = True ) self._update_items( update_cmap = True )
# Maintain compatibility for old name of color bar setting method.
def setCmap(self, cmap): def setCmap(self, cmap):
warnings.warn(
"The method 'setCmap' has been renamed to 'setColorMap' for clarity. "
"The old name will no longer be available in any version of PyQtGraph released after July 2022.",
DeprecationWarning, stacklevel=2
)
self.setColorMap(cmap)
def setColorMap(self, colorMap):
""" """
sets a ColorMap object to determine the ColorBarItem's look-up table. The same Sets a ColorMap object to determine the ColorBarItem's look-up table. The same
look-up table is applied to any assigned ImageItem. look-up table is applied to any assigned ImageItem.
""" """
self.cmap = cmap self._colorMap = colorMap
self._update_items( update_cmap = True ) self._update_items( update_cmap = True )
def colorMap(self):
"""
Returns the assigned ColorMap object.
"""
return self._colorMap
@property
def cmap(self):
warnings.warn(
"Direct access to ColorMap.cmap is deprecated and will no longer be available in any "
"version of PyQtGraph released after July 2022. Please use 'ColorMap.colorMap()' instead.",
DeprecationWarning, stacklevel=2)
return self._colorMap
def setLevels(self, values=None, low=None, high=None ): def setLevels(self, values=None, low=None, high=None ):
""" """
Sets the displayed range of levels as specified. Sets the displayed range of levels as specified.
============== =========================================================================== ============== ===========================================================================
**Arguments:** **Arguments:**
values Specify levels by tuple (low, high). Either value can be None to leave values specifies levels as tuple (low, high). Either value can be None to leave
to previous value unchanged. Takes precedence over low and high parameters. to previous value unchanged. Takes precedence over low and high parameters.
low new low level to be applied to color bar and assigned images low new low level to be applied to color bar and assigned images
high new high level to be applied to color bar and assigned images high new high level to be applied to color bar and assigned images
@ -199,15 +230,15 @@ class ColorBarItem(PlotItem):
""" internal: update color maps for bar and assigned ImageItems """ """ internal: update color maps for bar and assigned ImageItems """
# update color bar: # update color bar:
self.axis.setRange( self.values[0], self.values[1] ) self.axis.setRange( self.values[0], self.values[1] )
if update_cmap and self.cmap is not None: if update_cmap and self._colorMap is not None:
self.bar.setLookupTable( self.cmap.getLookupTable(nPts=256) ) self.bar.setLookupTable( self._colorMap.getLookupTable(nPts=256) )
# update assigned ImageItems, too: # update assigned ImageItems, too:
for img_weakref in self.img_list: for img_weakref in self.img_list:
img = img_weakref() img = img_weakref()
if img is None: continue # dereference weakref if img is None: continue # dereference weakref
img.setLevels( self.values ) # (min,max) tuple img.setLevels( self.values ) # (min,max) tuple
if update_cmap and self.cmap is not None: if update_cmap and self._colorMap is not None:
img.setLookupTable( self.cmap.getLookupTable(nPts=256) ) img.setLookupTable( self._colorMap.getLookupTable(nPts=256) )
def _regionChanged(self): def _regionChanged(self):
""" internal: snap adjusters back to default positions on release """ """ internal: snap adjusters back to default positions on release """