example_autocorrelograms.py

# 8/31/22
# https://figurl.org/f?v=gs://figurl/spikesortingview-8&d=sha1://5cb75552ad2f7316bf1a06af8cd6564c0fd9d497&label=Autocorrelograms%20example

from typing import List
import sortingview.views as vv
import spikeinterface as si
import spikeinterface.extractors as se
import kachery as ka
from helpers.compute_correlogram_data import compute_correlogram_data


def main():
    ka.use_sandbox()
    _, sorting = se.toy_example(num_units=18, duration=300, seed=0, num_segments=1)

    view = example_autocorrelograms(sorting=sorting)
    url = view.url(label="Autocorrelograms example")
    print(url)


def example_autocorrelograms(*, sorting: si.BaseSorting, height=400):
    autocorrelogram_items: List[vv.AutocorrelogramItem] = []
    for unit_id in sorting.get_unit_ids():
        a = compute_correlogram_data(sorting=sorting, unit_id1=unit_id, unit_id2=None, window_size_msec=50, bin_size_msec=1)
        bin_edges_sec = a["bin_edges_sec"]
        bin_counts = a["bin_counts"]
        autocorrelogram_items.append(vv.AutocorrelogramItem(unit_id=unit_id, bin_edges_sec=bin_edges_sec, bin_counts=bin_counts))
    view = vv.Autocorrelograms(autocorrelograms=autocorrelogram_items, height=height)
    return view


if __name__ == "__main__":
    main()