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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
use crate::diff::{match_graphs, DiffGraph, Match};
use crate::{MultiGraph, Edge, Graph, NodeStyle};
use std::collections::HashSet;
/// Returns a MultiGraph containing the diff of the two graphs.
/// To be visualized with dot.
pub fn visualize_diff(d1: &DiffGraph, d2: &DiffGraph) -> MultiGraph {
let matches = match_graphs(d1, d2);
let mut matched1 = HashSet::new();
let mut matched2 = HashSet::new();
let mut partial1 = HashSet::new();
let mut partial2 = HashSet::new();
for mch in matches {
match mch {
Match::Full(m) => {
matched1.insert(m.from);
matched2.insert(m.to);
}
Match::Partial(m) => {
partial1.insert(m.from);
partial2.insert(m.to);
}
}
}
let added_style = NodeStyle {
title_bg: Some("green".into()),
..Default::default()
};
let removed_style = NodeStyle {
title_bg: Some("red".into()),
..Default::default()
};
let changed_style = NodeStyle {
title_bg: Some("yellow".into()),
..Default::default()
};
let default_style = NodeStyle {
..Default::default()
};
let edges1: Vec<Edge> = d1
.graph
.edges
.iter()
.map(|e| {
Edge::new(
format!("{}_diff1", e.from),
format!("{}_diff1", e.to),
e.label.clone(),
)
})
.collect();
let edges2: Vec<Edge> = d2
.graph
.edges
.iter()
.map(|e| {
Edge::new(
format!("{}_diff2", e.from),
format!("{}_diff2", e.to),
e.label.clone(),
)
})
.collect();
let mut nodes1 = Vec::new();
for node in &d1.graph.nodes {
let label = node.label.as_str();
let mut node_cloned = node.clone();
node_cloned.label = format!("{}_diff1", node.label);
if matched1.contains(label) {
node_cloned.style = default_style.clone();
nodes1.push(node_cloned);
} else if partial1.contains(label) {
node_cloned.style = changed_style.clone();
nodes1.push(node_cloned);
} else {
node_cloned.style = removed_style.clone();
nodes1.push(node_cloned);
}
}
let mut nodes2 = Vec::new();
for node in &d2.graph.nodes {
let label = node.label.as_str();
let mut node_cloned = node.clone();
node_cloned.label = format!("{}_diff2", node.label);
if matched2.contains(label) {
node_cloned.style = default_style.clone();
nodes2.push(node_cloned);
} else if partial2.contains(label) {
node_cloned.style = changed_style.clone();
nodes2.push(node_cloned);
} else {
node_cloned.style = added_style.clone();
nodes2.push(node_cloned);
}
}
let newg1 = Graph::new("diff1".to_owned(), nodes1, edges1);
let newg2 = Graph::new("diff2".to_owned(), nodes2, edges2);
MultiGraph::new("diff".to_owned(), vec![newg1, newg2])
}
|
