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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
use super::*;
use std::cmp;
pub fn count(list: &[Group]) -> usize {
let mut n: usize = 0;
for g in list.iter() {
n += g.nodes.len();
}
n
}
pub fn get_parameters(group: &Group) -> (Coordinate, Coordinate) {
let mut min_x: i16 = 0;
let mut min_y: i16 = 0;
let mut max_x: i16 = 0;
let mut max_y: i16 = 0;
for node in &group.nodes {
let (min, max) = node.get_parameters();
max_x = std::cmp::max(max_x, max.x);
min_x = std::cmp::min(min_x, min.x);
max_y = std::cmp::max(max_y, max.y);
min_y = std::cmp::min(min_y, min.y);
}
(Coordinate::new(min_x, min_y), Coordinate::new(max_x, max_y))
}
pub fn add_node(group: &mut Group, name: Option<&str>, min: Option<u32>, max: Option<u32>) {
let name = name.unwrap_or("");
let min = min.unwrap_or(0);
let max = max.unwrap_or(group.get_dynamic_radius());
let mi = cmp::min(min, max);
let ma = cmp::max(min, max);
let geo = coordinate::gen_radius(group.settings.geo, mi, ma);
let mut node = Node::new(name, geo);
node.color = group.gen_color(geo);
node.radius = group.settings.radius;
group.push(node);
}
#[cfg(test)]
mod tests {
use super::{super::Node, *};
#[test]
fn test_count_none() {
let groups = Group::from_list(&[(0, 0), (100, 100)]);
assert_eq!(count(&groups), 0);
}
#[test]
fn test_count_some() {
let mut groups = Group::from_list(&[(0, 0), (100, 100)]);
groups[0].nodes = Node::from_list(&[(0, 0), (0, 0)]);
assert_eq!(count(&groups), 2);
}
#[test]
fn test_add_node() {
let mut group = Group::new_simple(0, 0);
add_node(&mut group, None, None, None);
add_node(&mut group, Some("name"), Some(50), Some(20));
assert_eq!(group.nodes.len(), 2);
}
}