heatmap: Adapt to palette change

Author: ales-erjavec

Orange/widgets/visualize/utils/heatmap.py

@@ -19,8 +19,6 @@
     QGraphicsLayoutItem
 )
 
-import pyqtgraph as pg
-
 from Orange.clustering import hierarchical
 from Orange.clustering.hierarchical import Tree
 from Orange.widgets.utils import apply_all
@@ -32,6 +30,7 @@
 
 from Orange.widgets.utils.graphicstextlist import TextListWidget
 from Orange.widgets.utils.dendrogram import DendrogramWidget
+from Orange.widgets.visualize.utils.plotutils import AxisItem
 
 
 def leaf_indices(tree: Tree) -> Sequence[int]:
@@ -936,6 +935,11 @@ def setShowAverages(self, visible):
             item.setVisible(visible)
             item.setPreferredWidth(0 if not visible else 10)
 
+    def changeEvent(self, event):
+        if event.type() == QEvent.PaletteChange:
+            self.__update_palette()
+        super().changeEvent(event)
+
     def event(self, event):
         # type: (QEvent) -> bool
         rval = super().event(event)
@@ -992,6 +996,12 @@ def __select_by_cluster(self, item, dendrogramindex):
                 node.value.first, node.value.last - 1, hm,
                 clear=clear, remove=remove, append=append)
 
+    def __update_palette(self):
+        for item in layout_items_recursive(self.layout()):
+            if isinstance(item, SimpleLayoutItem) \
+                    and isinstance(item.item, QGraphicsSimpleTextItem):
+                item.item.setBrush(self.palette().text())
+
     def heatmapAtPos(self, pos: QPointF) -> Optional['GraphicsHeatmapWidget']:
         for hw in chain.from_iterable(self.heatmap_widget_grid):
             if hw.contains(hw.mapFromItem(self, pos)):
@@ -1234,7 +1244,7 @@ def remove_item(item: QGraphicsItem) -> None:
         item.setParentItem(None)
 
 
-class _GradientLegendAxisItem(pg.AxisItem):
+class _GradientLegendAxisItem(AxisItem):
     def boundingRect(self):
         br = super().boundingRect()
         if self.orientation in ["top", "bottom"]:
@@ -1335,12 +1345,6 @@ def __update(self):
 
         self.updateGeometry()
 
-    def changeEvent(self, event: QEvent) -> None:
-        if event.type() == QEvent.PaletteChange:
-            pen = QPen(self.palette().color(QPalette.Text))
-            self.__axis.setPen(pen)
-        super().changeEvent(event)
-
 
 class CategoricalColorLegend(QGraphicsWidget):
     def __init__(
@@ -1431,20 +1435,31 @@ def legend_item_pair(color: QColor, size: float, text: str):
     def changeEvent(self, event: QEvent) -> None:
         if event.type() == QEvent.FontChange:
             self._updateFont(self.font())
+        elif event.type() == QEvent.PaletteChange:
+            self._updatePalette()
         super().changeEvent(event)
 
     def _updateFont(self, font):
         w = QFontMetrics(font).horizontalAdvance("X")
-        for item in filter(
-                lambda item: isinstance(item, SimpleLayoutItem),
-                layout_items_recursive(self.__layout)
-        ):
+        for item in self.__layoutItems():
             if isinstance(item.item, QGraphicsSimpleTextItem):
                 item.item.setFont(font)
             elif isinstance(item.item, QGraphicsRectItem):
                 item.item.setRect(QRectF(0, 0, w, w))
             item.updateGeometry()
 
+    def _updatePalette(self):
+        palette = self.palette()
+        for item in self.__layoutItems():
+            if isinstance(item.item, QGraphicsSimpleTextItem):
+                item.item.setBrush(palette.brush(QPalette.Text))
+
+    def __layoutItems(self):
+        return filter(
+            lambda item: isinstance(item, SimpleLayoutItem),
+            layout_items_recursive(self.__layout)
+        )
+
 
 def layout_items(layout: QGraphicsLayout) -> Iterable[QGraphicsLayoutItem]:
     for item in map(layout.itemAt, range(layout.count())):