GUI plots now save to results dir in png format

kilosort/gui/message_log_box.py

@@ -58,7 +58,7 @@ def print_settings(self):
         # values and the values that are actually being used.
         settings_text = "settings = "
         settings = self.gui.settings_box.settings.copy()
-        settings['probe'] = '... (use dump probe)'
+        settings['probe'] = '... (use print probe)'
         s = pprint.pformat(settings, indent=4, sort_dicts=False)
         settings_text += s[0] + '\n ' + s[1:-1] + '\n' + s[-1]
 

kilosort/gui/run_box.py

@@ -210,6 +210,5 @@ def add_plot_data(self, plot_type):
             clu = self.current_worker.clu
             tF = self.current_worker.tF
             is_refractory = self.current_worker.is_refractory
-            device = self.parent.device
             plot_spike_positions(plot_window, ops, st, clu, tF, is_refractory,
-                                 device)
+                                 settings)

kilosort/gui/sanity_plots.py

@@ -1,5 +1,7 @@
+from pathlib import Path
+
 import pyqtgraph as pg
-import scipy
+import pyqtgraph.exporters
 import numpy as np
 import matplotlib
 import torch
@@ -46,6 +48,8 @@ def plot_drift_amount(plot_window, dshift, settings):
         p1.plot(t, dshift[:,i], pen=color)
 
     plot_window.show()
+    save_path = str(Path(settings['results_dir']) / 'drift_amount.png')
+    pg.exporters.ImageExporter(plot_window.plot_widget.scene()).export(save_path)
 
 
 def plot_drift_scatter(plot_window, st0, settings):
@@ -93,6 +97,8 @@ def plot_drift_scatter(plot_window, st0, settings):
     left_ax.setTickPen('k')
 
     plot_window.show()
+    save_path = str(Path(settings['results_dir']) / 'drift_scatter.png')
+    pg.exporters.ImageExporter(plot_window.plot_widget.scene()).export(save_path)
 
 
 def plot_diagnostics(plot_window, wPCA, Wall0, clu0, settings):
@@ -142,10 +148,11 @@ def plot_diagnostics(plot_window, wPCA, Wall0, clu0, settings):
 
     # Finished, draw plot
     plot_window.show()
+    save_path = str(Path(settings['results_dir']) / 'diagnostics.png')
+    pg.exporters.ImageExporter(plot_window.plot_widget.scene()).export(save_path)
 
 
-def plot_spike_positions(plot_window, ops, st, clu, tF, is_refractory,
-                         device=None):
+def plot_spike_positions(plot_window, ops, st, clu, tF, is_refractory, settings):
 
     p1 = plot_window.plot_widget.addPlot(
         row=0, col=0, labels={'bottom': 'Depth (um)', 'left': 'Lateral (um)'}
@@ -177,3 +184,5 @@ def plot_spike_positions(plot_window, ops, st, clu, tF, is_refractory,
                                  brush=brushes)
     p1.addItem(scatter)
     plot_window.show()
+    save_path = str(Path(settings['results_dir']) / 'spike_positions.png')
+    pg.exporters.ImageExporter(plot_window.plot_widget.scene()).export(save_path)