sc_neurocore_engine/bindings/rate/
parallel_spiking.rs1use pyo3::prelude::*;
10
11use crate::neurons;
12
13#[pyclass(
14 name = "ParallelSpikingNeuron",
15 module = "sc_neurocore_engine.sc_neurocore_engine"
16)]
17#[derive(Clone)]
18pub struct PyParallelSpikingNeuron {
19 inner: neurons::ParallelSpikingNeuron,
20}
21
22#[pymethods]
23impl PyParallelSpikingNeuron {
24 #[new]
25 #[pyo3(signature = (kernel_size=8, v_threshold=1.0))]
26 fn new(kernel_size: usize, v_threshold: f64) -> Self {
27 Self {
28 inner: neurons::ParallelSpikingNeuron::new(kernel_size, v_threshold),
29 }
30 }
31
32 fn step(&mut self, current: f64) -> i32 {
33 self.inner.step(current)
34 }
35
36 fn reset(&mut self) {
37 self.inner.reset();
38 }
39}
40
41pub(super) fn register(module: &Bound<'_, PyModule>) -> PyResult<()> {
42 module.add_class::<PyParallelSpikingNeuron>()?;
43 Ok(())
44}