forceatlas2 0.8.0

fast force-directed generic n-dimension graph layout
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

use crate::{layout::*, node::*, util::*};

use rayon::prelude::*;

pub(crate) fn apply_gravity<
	T: Coord + Send + Sync,
	const N: usize,
	E,
	O: Nodes<T, N, Id>,
	Id: Sync,
>(
	layout: &mut Layout<T, N, E, O, Id>,
) {
	layout.nodes.par_iter_nodes_mut().for_each(|node| {
		let mut d = T::zero();
		for x in node.pos {
			d += x * x;
		}
		if d.is_zero() {
			return;
		}
		node.speed -= node.pos * (node.mass + T::one()) * layout.settings.kg / d.sqrt();
	})
}

pub(crate) fn apply_gravity_sg<
	T: Coord + Send + Sync,
	const N: usize,
	E,
	O: Nodes<T, N, Id>,
	Id: Sync,
>(
	layout: &mut Layout<T, N, E, O, Id>,
) {
	layout.nodes.par_iter_nodes_mut().for_each(|node| {
		node.speed -= node.pos * (node.mass + T::one()) * layout.settings.kg;
	})
}