class HorizontalHistogramItem(HistogramLUTItem):
"""
This is a graphicsWidget which provides controls for adjusting the display of an image.
Includes:
- Image histogram
- Movable region over histogram to select black/white levels
- Gradient editor to define color lookup table for single-channel images
"""
# CONSTANTS
# the minThresholdLabel and maxThresholdLabel heigth of the histogram widget
HISTOGRAM_MAX_HEIGHT = 152
HISTOGRAM_MIN_HEIGHT = 45
DECIMAL_PLACES = 2
def __init__(self, image=None, fillHistogram=True):
"""
If *image* (ImageItem) is provided, then the control will be automatically linked
to the image and changes to the control will be immediately reflected in the image's appearance.
By default, the histogram is rendered with a fill. For performance, set *fillHistogram* = False.
"""
HistogramLUTItem.__init__(self, image, fillHistogram)
self.layout = QtGui.QGraphicsGridLayout()
self.layout.setContentsMargins(1, 1, 1, 1)
self.layout.setSpacing(0)
# the viewbox in which would be displayed the histogram
# get a new and descriptive name for the view box
self.view_boxHistogram = self.vb
self.view_boxHistogram = ViewBox()
self.view_boxHistogram.setMaximumHeight(self.HISTOGRAM_MAX_HEIGHT)
self.view_boxHistogram.setMinimumHeight(self.HISTOGRAM_MIN_HEIGHT)
self.view_boxHistogram.setMouseEnabled(x=False, y=False)
# set defaults gradient position and color bar
self.gradient.setOrientation('bottom')
self.gradient.loadPreset('thermal')
# todo remove this parche the region of threshold selection
self.region.setVisible(False)
self._region = LinearRegionItem(values=[-40, 0], orientation=LinearRegionItem.Vertical)
self.view_boxHistogram.addItem(self._region)
self.axis = AxisItem('top', linkView=self.view_boxHistogram, maxTickLength=-10, showValues=False)
self.layout.addItem(self.axis, 0, 0)
self.layout.addItem(self.view_boxHistogram, 1, 0)
self.layout.addItem(self.gradient, 2, 0)
self.gradient.setFlag(self.gradient.ItemStacksBehindParent)
self.view_boxHistogram.setFlag(self.gradient.ItemStacksBehindParent)
self.plot = PlotDataItem()
self.fillHistogram(fillHistogram)
self.view_boxHistogram.addItem(self.plot)
self.autoHistogramRange()
# set the fixed range of visualization
self.view_boxHistogram.setXRange(-120, 5)
self._region.setBounds((-120, 5))
y_range = self.view_boxHistogram.viewRange()[1]
# the minThresholdLabel and maxThresholdLabel labels
self.minThresholdLabel = pg.TextItem(self.tr(u'Min'), color=(255, 255, 255), anchor=(1, 0.5))
self.view_boxHistogram.addItem(self.minThresholdLabel)
self.maxThresholdLabel = pg.TextItem(self.tr(u'Max'), color=(255, 255, 255), anchor=(0, 0.5))
self.view_boxHistogram.addItem(self.maxThresholdLabel)
self.gradient.sigGradientChanged.connect(self.gradientChanged)
self._region.sigRegionChanged.connect(self.regionChanging)
self._region.sigRegionChangeFinished.connect(self.regionChanged)
labels_ypos = self.view_boxHistogram.viewRange()[1][1] * 0.7
self.maxThresholdLabel.setPos(self._region.getRegion()[1], labels_ypos)
self.minThresholdLabel.setPos(self._region.getRegion()[0], labels_ypos)
# update the tooltips and position of the limit labels
self._region.sigRegionChangeFinished.connect(
lambda: self.maxThresholdLabel.setToolTip(self.tr(u'Max Threshold') + u": " +
unicode(round(self._region.getRegion()[1],
self.DECIMAL_PLACES))))
self._region.sigRegionChangeFinished.connect(
lambda: self.minThresholdLabel.setToolTip(self.tr(u'Min Threshold') + u": " +
unicode(round(self._region.getRegion()[0],
self.DECIMAL_PLACES))))
# set the y position of the labels to the 70% of the visible y range of the viewbox
self._region.sigRegionChanged.connect(
lambda: self.maxThresholdLabel.setPos(self._region.getRegion()[1],
self.view_boxHistogram.viewRange()[1][1] * 0.7))
self._region.sigRegionChanged.connect(
lambda: self.minThresholdLabel.setPos(self._region.getRegion()[0],
self.view_boxHistogram.viewRange()[1][1] * 0.7))
if image is not None:
self.setImageItem(image)
self.setLayout(self.layout)
# self.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Expanding)
def paint(self, p, *args):
"""
:param p:
:param args:
:return:
"""
pen = self._region.lines[0].pen
rgn = self._region.getRegion()
class HistogramLUTItem(GraphicsWidget):
"""
This is a graphicsWidget which provides controls for adjusting the display of an image.
Includes:
- Image histogram
- Movable region over histogram to select black/white levels
- Gradient editor to define color lookup table for single-channel images
"""
sigLookupTableChanged = QtCore.Signal(object)
sigLevelsChanged = QtCore.Signal(object)
sigLevelChangeFinished = QtCore.Signal(object)
def __init__(self,
image=None,
fillHistogram=True,
orientation='right',
gradients=None):
"""
If *image* (ImageItem) is provided, then the control will be automatically linked to the image and changes to the control will be immediately reflected in the image's appearance.
By default, the histogram is rendered with a fill. For performance, set *fillHistogram* = False.
"""
GraphicsWidget.__init__(self)
self.lut = None
self.imageItem = lambda: None # fake a dead weakref
self.layout = QtGui.QGraphicsGridLayout()
self.setLayout(self.layout)
self.layout.setContentsMargins(1, 1, 1, 1)
self.layout.setSpacing(0)
self.vb = ViewBox(parent=self)
if orientation in ['right', 'left']:
self.vb.setMaximumWidth(150)
self.vb.setMinimumWidth(15)
self.vb.setMouseEnabled(x=False, y=True)
elif orientation in ['bottom', 'top']:
self.vb.setMaximumHeight(150)
self.vb.setMinimumHeight(15)
self.vb.setMouseEnabled(x=True, y=False)
self.orientation = orientation
orientationGEI = orientation
if orientation == 'right':
orientationLRI = 'horizontal'
orientationAI = 'left'
elif orientation == 'left':
orientationLRI = 'horizontal'
orientationAI = 'right'
elif orientation == 'bottom':
orientationLRI = 'vertical'
orientationAI = 'top'
elif orientation == 'top':
orientationLRI = 'vertical'
orientationAI = 'bottom'
self.gradient = GradientEditorItem(gradients=gradients)
self.gradient.setOrientation(orientationGEI)
self.gradient.loadPreset('grey')
self.region = LinearRegionItem([0, 1], orientation=orientationLRI)
self.region.setZValue(1000)
self.vb.addItem(self.region)
self.axis = AxisItem(orientationAI,
linkView=self.vb,
maxTickLength=-10,
parent=self)
if orientation == 'right':
self.layout.addItem(self.axis, 0, 0)
self.layout.addItem(self.vb, 0, 1)
self.layout.addItem(self.gradient, 0, 2)
elif orientation == 'left':
self.layout.addItem(self.gradient, 0, 0)
self.layout.addItem(self.vb, 0, 1)
self.layout.addItem(self.axis, 0, 2)
elif orientation == 'bottom':
self.layout.addItem(self.axis, 0, 0)
self.layout.addItem(self.vb, 1, 0)
self.layout.addItem(self.gradient, 2, 0)
elif orientation == 'top':
self.layout.addItem(self.gradient, 0, 0)
self.layout.addItem(self.vb, 1, 0)
self.layout.addItem(self.axis, 2, 0)
self.range = None
self.gradient.setFlag(self.gradient.ItemStacksBehindParent)
self.vb.setFlag(self.gradient.ItemStacksBehindParent)
#self.grid = GridItem()
#self.vb.addItem(self.grid)
self.gradient.sigGradientChanged.connect(self.gradientChanged)
self.region.sigRegionChanged.connect(self.regionChanging)
self.region.sigRegionChangeFinished.connect(self.regionChanged)
self.vb.sigRangeChanged.connect(self.viewRangeChanged)
self.plot = PlotDataItem()
self.plot.rotate(90)
self.fillHistogram(fillHistogram)
self.vb.addItem(self.plot)
self.autoHistogramRange()
if image is not None:
self.setImageItem(image)
#self.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Expanding)
def fillHistogram(self, fill=True, level=0.0, color=(100, 100, 200)):
if fill:
self.plot.setFillLevel(level)
self.plot.setFillBrush(color)
else:
self.plot.setFillLevel(None)
#def sizeHint(self, *args):
#return QtCore.QSizeF(115, 200)
def paint(self, p, *args):
pen = self.region.lines[0].pen
rgn = self.getLevels()
if self.orientation == 'right':
p1 = self.vb.mapFromViewToItem(
self, Point(self.vb.viewRect().center().x(), rgn[0]))
p2 = self.vb.mapFromViewToItem(
self, Point(self.vb.viewRect().center().x(), rgn[1]))
gradRect = self.gradient.mapRectToParent(
self.gradient.gradRect.rect())
for pen in [fn.mkPen('k', width=3), pen]:
p.setPen(pen)
p.drawLine(p1, gradRect.bottomLeft())
p.drawLine(p2, gradRect.topLeft())
p.drawLine(gradRect.topLeft(), gradRect.topRight())
p.drawLine(gradRect.bottomLeft(), gradRect.bottomRight())
elif self.orientation == 'left':
p1 = self.vb.mapFromViewToItem(
self, Point(self.vb.viewRect().center().x(), rgn[0]))
p2 = self.vb.mapFromViewToItem(
self, Point(self.vb.viewRect().center().x(), rgn[1]))
gradRect = self.gradient.mapRectToParent(
self.gradient.gradRect.rect())
for pen in [fn.mkPen('k', width=3), pen]:
p.setPen(pen)
p.drawLine(p1, gradRect.bottomRight())
p.drawLine(p2, gradRect.topRight())
p.drawLine(gradRect.topLeft(), gradRect.topRight())
p.drawLine(gradRect.bottomLeft(), gradRect.bottomRight())
elif self.orientation == 'bottom':
p1 = self.vb.mapFromViewToItem(
self, Point(rgn[0],
self.vb.viewRect().center().y()))
p2 = self.vb.mapFromViewToItem(
self, Point(rgn[1],
self.vb.viewRect().center().y()))
gradRect = self.gradient.mapRectToParent(
self.gradient.gradRect.rect())
for pen in [fn.mkPen('k', width=3), pen]:
p.setPen(pen)
p.drawLine(p1, gradRect.topLeft())
p.drawLine(p2, gradRect.topRight())
p.drawLine(gradRect.topLeft(), gradRect.bottomLeft())
p.drawLine(gradRect.topRight(), gradRect.bottomRight())
elif self.orientation == 'top':
p1 = self.vb.mapFromViewToItem(
self, Point(rgn[0],
self.vb.viewRect().center().y()))
p2 = self.vb.mapFromViewToItem(
self, Point(rgn[1],
self.vb.viewRect().center().y()))
gradRect = self.gradient.mapRectToParent(
self.gradient.gradRect.rect())
for pen in [fn.mkPen('k', width=3), pen]:
p.setPen(pen)
p.drawLine(p1, gradRect.bottomLeft())
p.drawLine(p2, gradRect.bottomRight())
p.drawLine(gradRect.topLeft(), gradRect.bottomLeft())
p.drawLine(gradRect.topRight(), gradRect.bottomRight())
#p.drawRect(self.boundingRect())
def setHistogramRange(self, mn, mx, padding=0.1):
"""Set the Y range on the histogram plot. This disables auto-scaling."""
self.vb.enableAutoRange(self.vb.YAxis, False)
self.vb.setYRange(mn, mx, padding)
#d = mx-mn
#mn -= d*padding
#mx += d*padding
#self.range = [mn,mx]
#self.updateRange()
#self.vb.setMouseEnabled(False, True)
#self.region.setBounds([mn,mx])
def autoHistogramRange(self):
"""Enable auto-scaling on the histogram plot."""
self.vb.enableAutoRange(self.vb.XYAxes)
#self.range = None
#self.updateRange()
#self.vb.setMouseEnabled(False, False)
#def updateRange(self):
#self.vb.autoRange()
#if self.range is not None:
#self.vb.setYRange(*self.range)
#vr = self.vb.viewRect()
#self.region.setBounds([vr.top(), vr.bottom()])
def setImageItem(self, img):
"""Set an ImageItem to have its levels and LUT automatically controlled
by this HistogramLUTItem.
"""
self.imageItem = weakref.ref(img)
img.sigImageChanged.connect(self.imageChanged)
img.setLookupTable(
self.getLookupTable) ## send function pointer, not the result
#self.gradientChanged()
self.regionChanged()
self.imageChanged(autoLevel=True)
#self.vb.autoRange()
def viewRangeChanged(self):
self.update()
def gradientChanged(self):
if self.imageItem() is not None:
if self.gradient.isLookupTrivial():
self.imageItem().setLookupTable(
None) #lambda x: x.astype(np.uint8))
else:
self.imageItem().setLookupTable(
self.getLookupTable
) ## send function pointer, not the result
self.lut = None
#if self.imageItem is not None:
#self.imageItem.setLookupTable(self.gradient.getLookupTable(512))
self.sigLookupTableChanged.emit(self)
def getLookupTable(self, img=None, n=None, alpha=None):
"""Return a lookup table from the color gradient defined by this
HistogramLUTItem.
"""
if n is None:
if img.dtype == np.uint8:
n = 256
else:
n = 512
if self.lut is None:
self.lut = self.gradient.getLookupTable(n, alpha=alpha)
return self.lut
def regionChanged(self):
if self.imageItem() is not None:
self.imageItem().setLevels(self.region.getRegion())
self.sigLevelChangeFinished.emit(self)
#self.update()
def regionChanging(self):
if self.imageItem() is not None:
self.imageItem().setLevels(self.region.getRegion())
self.sigLevelsChanged.emit(self)
self.update()
def imageChanged(self, autoLevel=False, autoRange=False):
profiler = debug.Profiler()
h = self.imageItem().getHistogram()
profiler('get histogram')
if h[0] is None:
return
self.plot.setData(*h)
profiler('set plot')
if autoLevel:
mn = h[0][0]
mx = h[0][-1]
self.region.setRegion([mn, mx])
profiler('set region')
def getLevels(self):
"""Return the min and max levels.
"""
return self.region.getRegion()
def setLevels(self, mn, mx):
"""Set the min and max levels.
"""
self.region.setRegion([mn, mx])
self.update()
