example_splitter_layout.py

# 8/31/22
# https://figurl.org/f?v=gs://figurl/spikesortingview-8&d=sha1://428301bc920d206a927143a924165a0a91dc8a63&label=Splitter%20layout%20example

import sortingview.views as vv
import spikeinterface.extractors as se
import spikeinterface as si
import kachery as ka
from example_autocorrelograms import example_autocorrelograms
from example_cross_correlograms import example_cross_correlograms
from example_raster_plot import example_raster_plot
from example_average_waveforms import example_average_waveforms
from example_units_table import example_units_table
from example_unit_similarity_matrix import example_unit_unit_similarity_matrix


def main():
    ka.use_sandbox()
    R, S = se.toy_example(num_units=12, duration=300, seed=0, num_segments=1)
    assert isinstance(R, si.BaseRecording)

    v_units_table = example_units_table(recording=R, sorting=S)
    v_raster_plot = example_raster_plot(recording=R, sorting=S)
    v_autocorrelograms = example_autocorrelograms(sorting=S)
    v_average_waveforms = example_average_waveforms(recording=R, sorting=S)
    v_cross_correlograms = example_cross_correlograms(sorting=S, hide_unit_selector=True)
    v_unit_similarity_matrix = example_unit_unit_similarity_matrix(recording=R, sorting=S)

    view = vv.Splitter(
        direction="vertical",
        item1=(
            vv.LayoutItem(
                vv.Box(
                    direction="horizontal",
                    items=[vv.LayoutItem(v_units_table, stretch=1), vv.LayoutItem(v_raster_plot, stretch=2), vv.LayoutItem(v_unit_similarity_matrix, min_size=100, max_size=200)],
                )
            )
        ),
        item2=(
            vv.LayoutItem(
                vv.Splitter(
                    direction="horizontal",
                    item1=(
                        vv.LayoutItem(
                            vv.Box(
                                direction="horizontal",
                                items=[
                                    vv.LayoutItem(v_autocorrelograms, min_size=300, max_size=350),
                                    vv.LayoutItem(v_average_waveforms),
                                ],
                            )
                        )
                    ),
                    item2=(vv.LayoutItem(v_cross_correlograms)),
                )
            )
        ),
    )

    url = view.url(label="Splitter layout example")
    print(url)


if __name__ == "__main__":
    main()