End painter when drawing colormap (#262)

* End painter when drawing colormap * Only end painter if we created it
pyapp-kit · Dec 13, 2024 · 2f3113f · 2f3113f
1 parent c9528ff
commit 2f3113f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/superqt/cmap/_cmap_utils.py b/src/superqt/cmap/_cmap_utils.py
@@ -121,6 +121,10 @@ def paintEvent(self, event) -> None:
         painter.setBrush(gradient)
         painter.drawRect(rect)
 
+    # If we created a new Painter, free its resources
+    if isinstance(painter_or_device, QPaintDevice):
+        painter.end()
+
 
 def _draw_checkerboard(
     painter: QPainter, rect: QRect | QRectF, checker_size: int