view raster function

Author: oliche

viewspikes/gui.py

@@ -6,12 +6,17 @@
 from PyQt5 import QtWidgets, QtCore, QtGui, uic
 import pyqtgraph as pg
 from brainbox.processing import bincount2D
-import qt
-from iblutil.numerical import ismember
+
 from easyqc.gui import viewseis
+import one.alf.io as alfio
+from one.alf.files import get_session_path
+from iblutil.numerical import ismember
 from ibllib.io import spikeglx
 from ibllib.dsp import voltage
 
+import qt
+from brainbox.io.one import SpikeSortingLoader
+
 T_BIN = .007  # time bin size in secs
 D_BIN = 10  # depth bin size in um
 
@@ -28,6 +33,17 @@
 
 YMAX = 4000
 
+
+def view_raster(bin_file):
+    bin_file = Path(bin_file)
+    pname = bin_file.parent.name
+    session_path = get_session_path(bin_file)
+    ssl = SpikeSortingLoader(session_path=session_path, pname=pname)
+    spikes, clusters, channels = ssl.load_spike_sorting(dataset_types=['spikes.samples'])
+    trials = alfio.load_object(ssl.session_path.joinpath('alf'), 'trials')
+    return RasterView(bin_file, spikes, clusters, trials=trials)
+
+
 class RasterView(QtWidgets.QMainWindow):
     def __init__(self, bin_file, spikes, clusters, channels=None, trials=None, *args, **kwargs):
         self.sr = spikeglx.Reader(bin_file)