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()