Add clear_points (#7386)

mne/viz/_brain/_timeviewer.py

@@ -494,6 +494,7 @@ def __init__(self, brain, show_traces=False):
             'i': self.toggle_interface,
             's': self.apply_auto_scaling,
             'r': self.restore_user_scaling,
+            'c': self.clear_points,
             ' ': self.toggle_playback,
         }
         menu = self.plotter.main_menu.addMenu('Help')
@@ -722,6 +723,14 @@ def remove_point(self, mesh):
         self.picked_points[mesh._hemi].remove(mesh._vertex_id)
         self.plotter.remove_actor(mesh._actors)
 
+    def clear_points(self):
+        for sphere in self._spheres:
+            vertex_id = sphere._vertex_id
+            hemi = sphere._hemi
+            if vertex_id in self.picked_points[hemi]:
+                self.remove_point(sphere)
+        self._spheres.clear()
+
     def plot_time_course(self, hemi, vertex_id, color):
         time = self.brain._data['time']
         hemi_str = 'L' if hemi == 'lh' else 'R'
@@ -750,6 +759,7 @@ def help(self):
             ('i', 'Toggle interface'),
             ('s', 'Apply auto-scaling'),
             ('r', 'Restore original clim'),
+            ('c', 'Clear all traces'),
             ('Space', 'Start/Pause playback'),
         ]
         text1, text2 = zip(*pairs)

mne/viz/_brain/tests/test_brain.py

@@ -199,11 +199,9 @@ def test_brain_timeviewer_traces(renderer_interactive, hemi):
     assert len(spheres) == len(hemi_str)
 
     # test removing points
-    for sphere in spheres:
-        time_viewer.remove_point(sphere)
+    time_viewer.clear_points()
     assert len(picked_points['lh']) == 0
     assert len(picked_points['rh']) == 0
-    spheres.clear()  # necessary for the rest of the test
 
     # test picking a cell at random
     for idx, current_hemi in enumerate(hemi_str):