dspk-synapse 0.0.8

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

//! Unified polymorphic trait contracts for classical DSP (`dspk-synapse`) and
//! neural network inference (`dspk-synapse-ml`).
//!
//! Every stage returns a [`DspResult`]: learned models can fail at run time (an ONNX graph with an
//! unexpected input shape, an unsupported operator) and callers, including Python, get the error.

use dspk_core::DspResult;
use crate::io::probe::SensorLayout;
use super::events::{MatchedSpike, SpikeEvent};
use super::snippets::SnippetBatch;
use super::template::WaveformTemplate;

/// Polymorphic contract for Stage 1: Spike Detection (Classical Threshold/NEO vs. 1D-CNN/YASS).
pub trait SpikeDetector: Send + Sync {
    fn detect(
        &self,
        data: &[f32],
        channels: usize,
        samples: usize,
        sample_rate_hz: f64,
    ) -> DspResult<Vec<SpikeEvent>>;
}

/// Polymorphic contract for Stage 2: Waveform Denoising (Identity/Filter vs. 1D-UNet/Autoencoder).
pub trait WaveformDenoiser: Send + Sync {
    fn denoise(&self, batch: &SnippetBatch) -> DspResult<SnippetBatch>;
}

/// Polymorphic contract for Stage 3: Dimensionality Reduction & Latent Embedding (PCA vs. Conv-AE/VAE/SimCLR).
pub trait FeatureEmbedder: Send + Sync {
    /// Projects `[N, K, T]` snippets into a flat `[N, D]` embedding matrix and returns `(flat_embeddings, embedding_dim)`.
    fn embed(&self, batch: &SnippetBatch) -> DspResult<(Vec<f32>, usize)>;
}

/// Polymorphic contract for Stage 4: 3D Physical Source Localization (CoM/Monopolar vs. Dipole MLP).
pub trait PeakLocalizer: Send + Sync {
    /// Estimates 3D physical coordinates `[x_um, y_um, z_um]` for each spike in `batch`.
    fn localize(&self, batch: &SnippetBatch, layout: &SensorLayout) -> DspResult<Vec<[f32; 3]>>;
}

/// Polymorphic contract for Stage 5: Template Matching & Collision Deconvolution (matching pursuit or learned separators).
/// Template matching on a compute device (the caller chooses it; see `dspk_core::compute`).
pub trait SpikeMatcher: Send + Sync {
    fn match_spikes(
        &self,
        client: &cubecl::prelude::Client,
        data: &[f32],
        channels: usize,
        samples: usize,
        templates: &[WaveformTemplate],
    ) -> DspResult<Vec<MatchedSpike>>;
}