dagre_rust 0.0.5

Dagre implementation in Rust
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
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

use graphlib_rust::Graph;
use crate::{GraphConfig, GraphEdge, GraphNode};

pub fn adjust(g: &mut Graph<GraphConfig, GraphNode, GraphEdge>) {
  let rank_dir = g.graph().rankdir.clone().unwrap();
  if &rank_dir == "lr" || &rank_dir == "rl" {
    swap_width_height(g);
  }
}

pub fn undo(g: &mut Graph<GraphConfig, GraphNode, GraphEdge>) {
  let rank_dir = g.graph().rankdir.clone().unwrap();
  if &rank_dir == "bt" || &rank_dir == "rl" {
    reverse_y(g);
  }

  if &rank_dir == "lr" || &rank_dir == "rl" {
    swap_x_y(g);
    swap_width_height(g);
  }
}

fn swap_width_height(g: &mut Graph<GraphConfig, GraphNode, GraphEdge>) {
  let nodes = g.nodes();
  // =swapWidthHeightOne
  nodes.iter().for_each(|v| {
    let node = g.node_mut(v).unwrap();
    let w = node.width.clone();
    node.width = node.height;
    node.height = w;
  });
  let edges = g.edges();
  // =swapWidthHeightOne
  edges.iter().for_each(|e| {
    let edge_label = g.edge_mut_with_obj(&e).unwrap();
    let w = edge_label.width.clone();
    edge_label.width = edge_label.height;
    edge_label.height = w;
  });
}

fn reverse_y(g: &mut Graph<GraphConfig, GraphNode, GraphEdge>) {
  let nodes = g.nodes();
  nodes.iter().for_each(|v| {
    // =reverseYOne
    let node = g.node_mut(v).unwrap();
    node.y = -node.y;
  });

  let edges = g.edges();
  edges.iter().for_each(|e| {
    // =reverseYOne
    let edge_label = g.edge_mut_with_obj(&e).unwrap();
    let mut points = edge_label.points.clone().unwrap_or(vec![]);
    points.iter_mut().for_each(|point| {
      point.y = -point.y;
    });
    edge_label.points = Some(points);
    edge_label.y = -edge_label.y;
  });
}

fn swap_x_y(g: &mut Graph<GraphConfig, GraphNode, GraphEdge>) {
  let nodes = g.nodes();
  nodes.iter().for_each(|v| {
    // =swapXYOne
    let node = g.node_mut(v).unwrap();
    let x = node.x.clone();
    node.x = node.y;
    node.y = x;
  });

  let edges = g.edges();
  edges.iter().for_each(|e| {
    // =swapXYOne
    let edge_label = g.edge_mut_with_obj(&e).unwrap();
    let mut points = edge_label.points.clone().unwrap_or(vec![]);
    points.iter_mut().for_each(|point| {
      let x = point.x.clone();
      point.x = point.y;
      point.y = x;
    });
    edge_label.points = Some(points);

    let x = edge_label.x.clone();
    edge_label.x = edge_label.y;
    edge_label.y = x;
  });
}