class FilterPreviews(GraphicsLayoutWidget):
image_before: ImageItem
image_after: ImageItem
image_diff: ImageItem
histogram_before: Optional[PlotItem]
histogram_after: Optional[PlotItem]
histogram: Optional[PlotItem]
def __init__(self, parent=None, **kwargs):
super(FilterPreviews, self).__init__(parent, **kwargs)
widget_location = self.mapToGlobal(QPoint(self.width() / 2, 0))
# allow the widget to take up to 80% of the desktop's height
self.ALLOWED_HEIGHT: QRect = QGuiApplication.screenAt(widget_location).availableGeometry().height() * 0.8
self.before_histogram_data = None
self.after_histogram_data = None
self.histogram = None
self.before_histogram = None
self.after_histogram = None
self.combined_histograms = True
self.histogram_legend_visible = True
self.addLabel("Image before")
self.addLabel("Image after")
self.addLabel("Image difference")
self.nextRow()
self.image_before, self.image_before_vb, self.image_before_hist = self.image_in_vb(name="before")
self.image_after, self.image_after_vb, self.image_after_hist = self.image_in_vb(name="after")
self.image_difference, self.image_difference_vb, self.image_difference_hist = self.image_in_vb(
name="difference")
self.image_after_overlay = ImageItem()
self.image_after_overlay.setZValue(10)
self.image_after_vb.addItem(self.image_after_overlay)
# Ensure images resize equally
self.image_layout: GraphicsLayout = self.addLayout(colspan=6)
self.image_layout.addItem(self.image_before_vb, 0, 0)
self.image_layout.addItem(self.image_before_hist, 0, 1)
self.image_layout.addItem(self.image_after_vb, 0, 2)
self.image_layout.addItem(self.image_after_hist, 0, 3)
self.image_layout.addItem(self.image_difference_vb, 0, 4)
self.image_layout.addItem(self.image_difference_hist, 0, 5)
self.nextRow()
before_details = self.addLabel("")
after_details = self.addLabel("")
difference_details = self.addLabel("")
self.display_formatted_detail = {
self.image_before: lambda val: before_details.setText(f"Before: {val:.6f}"),
self.image_after: lambda val: after_details.setText(f"After: {val:.6f}"),
self.image_difference: lambda val: difference_details.setText(f"Difference: {val:.6f}"),
}
for img in self.image_before, self.image_after, self.image_difference:
img.hoverEvent = lambda ev: self.mouse_over(ev)
self.init_histogram()
def resizeEvent(self, ev: QResizeEvent):
if ev is not None:
size = ev.size()
self.image_layout.setFixedHeight(min(size.height() * 0.7, self.ALLOWED_HEIGHT))
super().resizeEvent(ev)
def image_in_vb(self, name=None):
im = ImageItem()
vb = ViewBox(invertY=True, lockAspect=True, name=name)
vb.addItem(im)
hist = HistogramLUTItem(im)
return im, vb, hist
def clear_items(self):
self.image_before.clear()
self.image_after.clear()
self.image_difference.clear()
self.image_after_overlay.clear()
def init_histogram(self):
self.histogram = self.addPlot(row=histogram_coords["combined"].row,
col=histogram_coords["combined"].col,
labels=histogram_axes_labels,
lockAspect=True,
colspan=3)
self.addLabel("Pixel values", row=label_coords["combined"].row, col=label_coords["combined"].col)
self.legend = self.histogram.addLegend()
def update_histogram_data(self):
# Plot any histogram that has data, and add a legend if both exist
before_data = self.image_before.getHistogram()
after_data = self.image_after.getHistogram()
if _data_valid_for_histogram(before_data):
if self.combined_histograms:
before_plot = self.histogram.plot(*before_data, pen=before_pen, clear=True)
self.legend.addItem(before_plot, "Before")
else:
self.before_histogram.plot(*before_data, pen=before_pen, clear=True)
if _data_valid_for_histogram(after_data):
if self.combined_histograms:
after_plot = self.histogram.plot(*after_data, pen=after_pen)
self.legend.addItem(after_plot, "After")
else:
self.after_histogram.plot(*after_data, pen=after_pen, clear=True)
def init_separate_histograms(self):
hc = histogram_coords
self.before_histogram = self.addPlot(row=hc["before"].row,
col=hc["before"].col,
labels=histogram_axes_labels,
lockAspect=True)
self.after_histogram = self.addPlot(row=hc["after"].row,
col=hc["after"].col,
labels=histogram_axes_labels,
lockAspect=True)
lc = label_coords
self.addLabel("Pixel values before", row=lc["before"].row, col=lc["before"].col)
self.addLabel("Pixel values after", row=lc["after"].row, col=lc["after"].col)
if _data_valid_for_histogram(self.before_histogram_data):
self.before_histogram.plot(*self.before_histogram_data, pen=before_pen)
if _data_valid_for_histogram(self.after_histogram_data):
self.after_histogram.plot(*self.after_histogram_data, pen=after_pen)
def delete_histograms(self):
coords = set(c for c in histogram_coords.values())
histograms = (self.getItem(*coord) for coord in coords)
for histogram in filter(lambda h: h is not None, histograms):
self.removeItem(histogram)
self.histogram = None
self.before_histogram = None
self.after_histogram = None
def delete_histogram_labels(self):
coords = set(c for c in label_coords.values())
labels = (self.getItem(*coord) for coord in coords)
for label in filter(lambda h: h is not None, labels):
self.removeItem(label)
@property
def histogram_legend(self) -> Optional[LegendItem]:
if self.histogram and self.histogram.legend:
return self.histogram.legend
return None
def mouse_over(self, ev):
# Ignore events triggered by leaving window or right clicking
if ev.exit:
return
pos = CloseEnoughPoint(ev.pos())
for img in self.image_before, self.image_after, self.image_difference:
if img.image is not None and pos.x < img.image.shape[0] and pos.y < img.image.shape[1]:
pixel_value = img.image[pos.y, pos.x]
self.display_formatted_detail[img](pixel_value)
def link_all_views(self):
for view1, view2 in [[self.image_before_vb, self.image_after_vb],
[self.image_after_vb, self.image_difference_vb],
[self.image_after_hist.vb, self.image_before_hist.vb]]:
view1.linkView(ViewBox.XAxis, view2)
view1.linkView(ViewBox.YAxis, view2)
def unlink_all_views(self):
for view in self.image_before_vb, self.image_after_vb, self.image_after_hist.vb:
view.linkView(ViewBox.XAxis, None)
view.linkView(ViewBox.YAxis, None)
def add_difference_overlay(self, diff):
diff = -diff
diff[diff > 0.0] = 1.0
pos = np.array([0, 1])
color = np.array([[0, 0, 0, 0], [255, 0, 0, 255]], dtype=np.ubyte)
map = ColorMap(pos, color)
self.image_after_overlay.setOpacity(1)
self.image_after_overlay.setImage(diff)
lut = map.getLookupTable(0, 1, 2)
self.image_after_overlay.setLookupTable(lut)
def hide_difference_overlay(self):
self.image_after_overlay.setOpacity(0)
def auto_range(self):
# This will cause the previews to all show by just causing autorange on self.image_before_vb
self.image_before_vb.autoRange()
class FilterPreviews(GraphicsLayoutWidget):
image_before: ImageItem
image_after: ImageItem
image_diff: ImageItem
histogram_before: Optional[PlotItem]
histogram_after: Optional[PlotItem]
histogram: Optional[PlotItem]
def __init__(self, parent=None, **kwargs):
super(FilterPreviews, self).__init__(parent, **kwargs)
self.before_histogram_data = None
self.after_histogram_data = None
self.histogram = None
self.before_histogram = None
self.after_histogram = None
self.combined_histograms = True
self.histogram_legend_visible = True
self.addLabel("Image before")
self.addLabel("Image after")
self.addLabel("Image difference")
self.nextRow()
self.image_before, self.image_before_vb = self.image_in_vb(
name="before")
self.image_after, self.image_after_vb = self.image_in_vb(name="after")
self.image_difference, self.image_difference_vb = self.image_in_vb(
name="difference")
self.image_after_overlay = ImageItem()
self.image_after_overlay.setZValue(10)
self.image_after_vb.addItem(self.image_after_overlay)
# Ensure images resize equally
image_layout = self.addLayout(colspan=3)
image_layout.addItem(self.image_before_vb, 0, 0)
image_layout.addItem(self.image_after_vb, 0, 1)
image_layout.addItem(self.image_difference_vb, 0, 2)
self.nextRow()
before_details = self.addLabel("")
after_details = self.addLabel("")
difference_details = self.addLabel("")
self.display_formatted_detail = {
self.image_before:
lambda val: before_details.setText(f"Before: {val:.6f}"),
self.image_after:
lambda val: after_details.setText(f"After: {val:.6f}"),
self.image_difference:
lambda val: difference_details.setText(f"Difference: {val:.6f}"),
}
for img in self.image_before, self.image_after, self.image_difference:
img.hoverEvent = lambda ev: self.mouse_over(ev)
def image_in_vb(self, name=None):
im = ImageItem()
im.setAutoDownsample(False)
vb = ViewBox(invertY=True, lockAspect=True, name=name)
vb.addItem(im)
return im, vb
def clear_items(self):
self.image_before.clear()
self.image_after.clear()
self.image_difference.clear()
self.image_after_overlay.clear()
self.delete_histograms()
# There seems to be a bug with pyqtgraph.PlotDataItem.setData not forcing a redraw.
# We work around this by redrawing everything completely every time, which is unreasonably fast anyway.
def redraw_histograms(self):
self.delete_histograms()
self.delete_histogram_labels()
if self.combined_histograms:
self.draw_combined_histogram()
else:
self.draw_separate_histograms()
def delete_histograms(self):
coords = set(c for c in histogram_coords.values())
histograms = (self.getItem(*coord) for coord in coords)
for histogram in filter(lambda h: h is not None, histograms):
self.removeItem(histogram)
self.histogram = None
self.before_histogram = None
self.after_histogram = None
self.diff_histogram = None
def delete_histogram_labels(self):
coords = set(c for c in label_coords.values())
labels = (self.getItem(*coord) for coord in coords)
for label in filter(lambda h: h is not None, labels):
self.removeItem(label)
def draw_combined_histogram(self):
self.histogram = self.addPlot(row=histogram_coords["combined"].row,
col=histogram_coords["combined"].col,
labels=histogram_axes_labels,
lockAspect=True,
colspan=3)
self.addLabel("Pixel values",
row=label_coords["combined"].row,
col=label_coords["combined"].col)
legend = self.histogram.addLegend()
# Plot any histogram that has data, and add a legend if both exist
if _data_valid_for_histogram(self.before_histogram_data):
before_plot = self.histogram.plot(*self.before_histogram_data,
pen=before_pen)
legend.addItem(before_plot, "Before")
if _data_valid_for_histogram(self.after_histogram_data):
after_plot = self.histogram.plot(*self.after_histogram_data,
pen=after_pen)
legend.addItem(after_plot, "After")
def draw_separate_histograms(self):
hc = histogram_coords
self.before_histogram = self.addPlot(row=hc["before"].row,
col=hc["before"].col,
labels=histogram_axes_labels,
lockAspect=True)
self.after_histogram = self.addPlot(row=hc["after"].row,
col=hc["after"].col,
labels=histogram_axes_labels,
lockAspect=True)
lc = label_coords
self.addLabel("Pixel values before",
row=lc["before"].row,
col=lc["before"].col)
self.addLabel("Pixel values after",
row=lc["after"].row,
col=lc["after"].col)
if _data_valid_for_histogram(self.before_histogram_data):
self.before_histogram.plot(*self.before_histogram_data,
pen=before_pen)
if _data_valid_for_histogram(self.after_histogram_data):
self.after_histogram.plot(*self.after_histogram_data,
pen=after_pen)
def set_before_histogram(self, data: Tuple[ndarray]):
self.before_histogram_data = data
self.redraw_histograms()
def set_after_histogram(self, data: Tuple[ndarray]):
self.after_histogram_data = data
self.redraw_histograms()
@property
def histogram_legend(self) -> Optional[LegendItem]:
if self.histogram and self.histogram.legend:
return self.histogram.legend
return None
def mouse_over(self, ev):
# Ignore events triggered by leaving window or right clicking
if ev.exit:
return
pos = CloseEnoughPoint(ev.pos())
for img in self.image_before, self.image_after, self.image_difference:
if img.image is not None and pos.x < img.image.shape[
0] and pos.y < img.image.shape[1]:
pixel_value = img.image[pos.y, pos.x]
self.display_formatted_detail[img](pixel_value)
def link_all_views(self):
for view1, view2 in zip(
[self.image_before_vb, self.image_after_vb],
[self.image_after_vb, self.image_difference_vb]):
view1.linkView(ViewBox.XAxis, view2)
view1.linkView(ViewBox.YAxis, view2)
def unlink_all_views(self):
for view in self.image_before_vb, self.image_after_vb, self.image_difference_vb:
view.linkView(ViewBox.XAxis, None)
view.linkView(ViewBox.YAxis, None)
def add_difference_overlay(self, diff):
diff = -diff
diff[diff > 0.0] = 1.0
pos = np.array([0, 1])
color = np.array([[0, 0, 0, 0], [255, 0, 0, 255]], dtype=np.ubyte)
map = ColorMap(pos, color)
self.image_after_overlay.setOpacity(1)
self.image_after_overlay.setImage(diff)
lut = map.getLookupTable(0, 1, 2)
self.image_after_overlay.setLookupTable(lut)
def hide_difference_overlay(self):
self.image_after_overlay.setOpacity(0)