graph-force/src/lib.rs

mod graph;
mod runner;
mod spring_model;
mod utils;

use pyo3::prelude::*;

/// Formats the sum of two numbers as string.
#[pyfunction(number_of_nodes, edges, "*", iter = 500, threads = 0)]
fn layout_from_edge_list(
    number_of_nodes: usize,
    edges: Vec<(u32, u32)>,
    iter: usize,
    threads: usize,
) -> PyResult<Vec<(f32, f32)>> {
    let r = runner::Runner::new(iter, threads);
    Ok(r.layout(number_of_nodes, edges))
}

/// A Python module implemented in Rust.
#[pymodule]
fn graph_force(_py: Python, m: &PyModule) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(layout_from_edge_list, m)?)?;
    Ok(())
}