Add blitting support to the RadioButtons widget

QuLogic · QuLogic · commit 75b05deda370 · 2022-07-18T22:22:17.000-05:00
diff --git a/lib/matplotlib/widgets.py b/lib/matplotlib/widgets.py
@@ -1406,7 +1406,8 @@ class RadioButtons(AxesWidget):
         The label text of the currently selected button.
     """
 
-    def __init__(self, ax, labels, active=0, activecolor='blue'):
+    def __init__(self, ax, labels, active=0, activecolor='blue',
+                 useblit=False):
         """
         Add radio buttons to an `~.axes.Axes`.
 
@@ -1420,6 +1421,9 @@ def __init__(self, ax, labels, active=0, activecolor='blue'):
             The index of the initially selected button.
         activecolor : color
             The color of the selected button.
+        useblit : bool, default: False
+            Use blitting for faster drawing if supported by the backend.
+            See the tutorial :doc:`/tutorials/advanced/blitting` for details.
         """
         super().__init__(ax)
         self.activecolor = activecolor
@@ -1438,6 +1442,9 @@ def __init__(self, ax, labels, active=0, activecolor='blue'):
         # default to hard-coded value if the radius becomes too large
         circle_radius = min(circle_radius, 0.05)
 
+        self._useblit = useblit and self.canvas.supports_blit
+        animated_style = {'animated': True} if self._useblit else {}
+
         self.labels = []
         self.circles = []
         for y, label in zip(ys, labels):
@@ -1452,17 +1459,27 @@ def __init__(self, ax, labels, active=0, activecolor='blue'):
                 facecolor = axcolor
 
             p = Circle(xy=(0.15, y), radius=circle_radius, edgecolor='black',
-                       facecolor=facecolor, transform=ax.transAxes)
+                       facecolor=facecolor, transform=ax.transAxes,
+                       **animated_style)
 
             self.labels.append(t)
             self.circles.append(p)
             ax.add_patch(p)
             cnt += 1
 
         self.connect_event('button_press_event', self._clicked)
+        if self._useblit:
+            self.connect_event('draw_event', self._clear)
 
         self._observers = cbook.CallbackRegistry(signals=["clicked"])
 
+    def _clear(self, event):
+        """Internal event handler to clear the buttons."""
+        if self.ignore(event):
+            return
+        for circle in self.circles:
+            self.ax.draw_artist(circle)
+
     def _clicked(self, event):
         if self.ignore(event) or event.button != 1 or event.inaxes != self.ax:
             return
@@ -1493,9 +1510,14 @@ def set_active(self, index):
             else:
                 color = self.ax.get_facecolor()
             p.set_facecolor(color)
+            if self.drawon and self._useblit:
+                self.ax.draw_artist(p)
 
         if self.drawon:
-            self.ax.figure.canvas.draw()
+            if self._useblit:
+                self.canvas.blit(self.ax.bbox)
+            else:
+                self.canvas.draw()
 
         if self.eventson:
             self._observers.process('clicked', self.labels[index].get_text())
