dspk-synapse 0.0.5

Electrophysiology for dspk: recording and sorting formats, probes, spike detection, waveforms, features, clustering, localization, quality metrics, and the shared parts of the sorters
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! Neural spike detection, spatial deduplication, waveform alignment, feature extraction,
//! localization, drift estimation, kriging, clustering, template matching, correlograms,
//! Allen/IBL quality metrics, sorter comparison, and neural frequency bands.
//!
//! Recording and sorting file formats (NWB, SpikeGLX, mtscomp, Phy, `.sorting.zarr`) and probe
//! geometry (layouts, presets, nearest-site queries) are in [`io`].
//!
//! # Where things are
//!
//! | Step | Module | Start with |
//! |---|---|---|
//! | Find spikes | [`detection`] | [`detect_spikes_multichannel`], [`deduplicate_spikes_spatial`] |
//! | Whole recordings, bounded memory | [`streaming`] | [`streaming::StreamingDetector`] |
//! | Cut waveforms | [`extraction`] | [`extract_snippets_multichannel`] |
//! | Features | [`features`] | [`PcaFeatureEmbedder`] |
//! | Positions and drift | [`spatial`] | [`PeakLocalizer`] implementations |
//! | Cluster into units | [`sorting`] | k-means, GMM, HDBSCAN, bipartite modularity |
//! | Quality and comparison | [`metrics`] | ISI violations, presence, amplitude cutoff; sorter agreement |
//! | Results on disk | [`storage`] | [`SortingOutput`] (Phy, Zarr, NWB) |
//! | Recordings, probes, sorting files | [`io`] | [`io::open`], [`io::probe`] |
//!
//! Complete sorters are separate crates, one per upstream project, each under its upstream's
//! licence: `dspk-sorter-kilosort` (Kilosort4, EMUsort), `dspk-sorter-ms5` (MountainSort 5),
//! `dspk-sorter-si` (SpyKING CIRCUS 2, Tridesclous 2). Pretrained models: `dspk-synapse-ml`.
//!
//! # Example
//!
//! Detect threshold crossings, then keep one spike per event: a spike seen on neighbouring
//! channels is one spike, assigned to the channel where it is largest.
//!
//! ```
//! use dspk_synapse::io::probe::{Position3D, SensorLayout, SensorSite};
//! use dspk_synapse::{deduplicate_spikes_spatial, detect_spikes_multichannel, SpikePolarity, SpikeSpacing};
//!
//! // 4 channels on a line, 20 µm apart
//! let sites = (0..4).map(|c| SensorSite::new(c, Position3D::new(0.0, 20.0 * c as f32, 0.0), 0)).collect();
//! let layout = SensorLayout::new("linear", sites);
//!
//! // Small noise, and one spike at sample 500: −80 µV on channel 1, −40 µV on its neighbour
//! let (channels, samples) = (4, 1_000);
//! let mut x: Vec<f32> = (0..channels * samples).map(|i| ((i * 7_919) % 200) as f32 / 100.0 - 1.0).collect();
//! x[samples + 500] = -80.0;
//! x[2 * samples + 500] = -40.0;
//!
//! // Troughs below −5 σ (σ per channel, from the median absolute deviation), ≥ 1 ms apart at 30 kHz
//! let crossings = detect_spikes_multichannel(&x, channels, samples, 5.0, SpikePolarity::Negative, SpikeSpacing::new(30));
//! assert_eq!(crossings.len(), 2); // the same spike, on two channels
//!
//! // Crossings within 50 µm and 10 samples of a larger one belong to it
//! let spikes = deduplicate_spikes_spatial(&crossings, &layout, 50.0, 10);
//! assert_eq!(spikes.len(), 1);
//! assert_eq!(spikes[0].primary_channel, 1);
//! assert_eq!(spikes[0].participating_channels, vec![1, 2]);
//! ```

pub mod core;
pub mod detection;
pub mod extraction;
pub mod features;
pub mod io;
pub mod metrics;
pub mod provenance;
pub mod sorter;
pub mod sorting;
pub mod spatial;
pub mod storage;
pub mod streaming;

pub use core::{
    DeduplicatedSpike, FeatureEmbedder, MatchedSpike, NeuralBand, PeakLocalizer, RecordingMeta,
    SnippetBatch, SortedUnit, SortingOutput, SpikeDetector, SpikeEvent, SpikeMatcher,
    UnitQualityLabel, WaveformDenoiser, WaveformSnippet, WaveformTemplate, compute_mean_template,
    dense_waveform, pack_templates, unpack_template,
};
pub use detection::{
    AdaptiveThresholdDetector, DistanceRule, MatchedFilterSpikeDetector, NeoSpikeDetector,
    SpikePolarity, SpikeSpacing, ThresholdSpikeDetector, canonical_biphasic_prototype,
    DedupNeighbours, compute_neo_energy_1d, deduplicate_spikes_spatial, deduplicate_spikes_spatial_gpu,
    detect_spikes_matched_filter, detect_spikes_multichannel, detect_spikes_neo,
    detect_spikes_with_sigma, detection_heights, estimate_noise_std, execute_detect_spikes_in_vram,
};
pub use extraction::{
    execute_extract_sinc_in_vram, extract_snippet_batch_multichannel, extract_snippets_kernel,
    extract_snippets_multichannel, extract_snippets_single_channel, read_snippets,
    trough_shift_kernel,
};
pub use features::{
    ConductionVelocityEstimate, PcaFeatureEmbedder, PpcaFeatureEmbedder, SpikeMorphology,
    WaveletFeatureEmbedder, compute_morphology, estimate_hdemg_conduction_velocity,
    extract_waveform_pca, haar_dwt_multilevel_1d,
};
pub use metrics::{
    BurstEpoch, Correlogram, FiringRateCurve, IsiMetrics, MepMetrics, PairwiseTrainMatch,
    PsthResult, QualityCriteria, SortingComparison, StimulusTriggeredAverage, UnitMatchSummary,
    compare_sortings, compare_spike_trains, compute_amplitude_cutoff,
    compute_amplitude_cutoff_with, compute_autocorrelogram, compute_crosscorrelogram,
    compute_d_prime, compute_instantaneous_firing_rate, compute_isi_violations,
    compute_isolation_distance, compute_llobet_contamination, compute_presence_ratio,
    compute_psth, compute_silhouette_score, compute_snr, compute_stimulus_triggered_average,
    count_refractory_violations, detect_burst_epochs, isi_histogram, quantify_mep,
};
pub use sorting::{
    ConvolutiveBssDecomposer, DensityPeaksResult, GmmClusterer, GmmCovarianceKind, GmmResult,
    KdeMergeResult, MotorUnitPulseTrain, MatchingPursuitMatcher, cluster_density_peaks,
    cluster_density_peaks_capped, cluster_gmm_bic, cluster_kde_merge,
    compute_template_similarity_matrix, match_spikes_matching_pursuit,
    suggest_template_merges, template_max_cosine_similarity,
};
pub use spatial::{
    CenterOfMassLocalizer, DipoleEstimate, DipoleLocalizer, DriftEstimate, GridConvolutionLocalizer,
    KRIGING_REGULARIZATION, MonopolarTriangulator, NonRigidDriftEstimate, TraceKriging,
    compute_kriging_weight_matrix,
    correct_snippet_batch_drift_kriging, correct_traces_drift_kriging, estimate_nonrigid_drift,
    estimate_rigid_drift, localize_spike_center_of_mass, localize_spike_dipole,
    localize_spike_grid_convolution, localize_spike_monopolar,
};
pub use streaming::{
    BatchTemplateStats, SINC_RESAMPLE_MARGIN, StreamingDetectionConfig, StreamingDetectionResult,
    StreamingDetector, TemplateAccumulator, execute_reduce_templates_in_vram,
    reduce_channel_templates_kernel,
};
pub use provenance::{ArtifactSource, Attributed, Paper, Provenance, ProvenanceKind, UpstreamCode};
pub use storage::{
    fill_similarity, load_nwb_units, load_phy_folder, load_sorting,
    load_sorting_zarr, load_spikes, save_nwb_units, save_phy_folder, save_sorting,
    save_sorting_zarr,
};