sc_neurocore_engine/bindings/hardware/
akida.rs1use pyo3::prelude::*;
10use pyo3::types::PyDict;
11
12use crate::neurons;
13
14#[pyclass(
15 name = "AkidaNeuron",
16 module = "sc_neurocore_engine.sc_neurocore_engine"
17)]
18#[derive(Clone)]
19pub struct PyAkidaNeuron {
20 inner: neurons::AkidaNeuron,
21}
22
23#[pymethods]
24impl PyAkidaNeuron {
25 #[new]
26 #[pyo3(signature = (threshold=100))]
27 fn new(threshold: i32) -> Self {
28 Self {
29 inner: neurons::AkidaNeuron::new(threshold),
30 }
31 }
32
33 fn step(&mut self, weight: i32) -> i32 {
34 self.inner.step(weight as f64)
35 }
36
37 fn reset(&mut self) {
38 self.inner.reset();
39 }
40
41 fn get_state(&self, py: Python<'_>) -> PyResult<Py<PyAny>> {
42 let d = PyDict::new(py);
43 d.set_item("v", self.inner.v)?;
44 d.set_item("rank", self.inner.rank)?;
45 Ok(d.into_any().unbind())
46 }
47}
48
49pub(super) fn register(module: &Bound<'_, PyModule>) -> PyResult<()> {
50 module.add_class::<PyAkidaNeuron>()?;
51 Ok(())
52}