wolf-graph-dot 0.1.0

Adds support for generating Graphviz DOT files from wolf-graph graphs.
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

use std::fmt;

use crate::{ArrowDirection, ArrowType, Color, EdgeStyle};

#[derive(Debug, Clone, PartialEq)]
pub struct EdgeAttributes {
    pub label: Option<String>,
    pub color: Option<Color>,
    pub style: Option<EdgeStyle>,
    pub shape: Option<String>,
    pub arrow_direction: Option<ArrowDirection>,
    pub arrow_head: Option<ArrowType>,
    pub arrow_tail: Option<ArrowType>,
}

impl EdgeAttributes {
    pub fn new(
        label: Option<String>,
        color: Option<Color>,
        style: Option<EdgeStyle>,
        shape: Option<String>,
        arrow_direction: Option<ArrowDirection>,
        arrow_head: Option<ArrowType>,
        arrow_tail: Option<ArrowType>,
    ) -> Self {
        Self {
            label,
            color,
            style,
            shape,
            arrow_direction,
            arrow_head,
            arrow_tail,
        }
    }

    pub fn attributes(&self) -> Option<String> {
        crate::details::encode::encode_attributes(
            &[
                ("label", self.label.as_ref().map(|s| s as &dyn ToString)),
                ("color", self.color.as_ref().map(|c| c as &dyn ToString)),
                ("style", self.style.as_ref().map(|s| s as &dyn ToString)),
                ("shape", self.shape.as_ref().map(|s| s as &dyn ToString)),
                ("dir", self.arrow_direction.as_ref().map(|s| s as &dyn ToString)),
                ("arrowhead", self.arrow_head.as_ref().map(|s| s as &dyn ToString)),
                ("arrowtail", self.arrow_tail.as_ref().map(|s| s as &dyn ToString)),
            ],
            true,
        )
    }
}

impl Default for EdgeAttributes {
    fn default() -> Self {
        Self::new(None, None, None, None, None, None, None)
    }
}

impl fmt::Display for EdgeAttributes {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        if let Some(encoded) = self.attributes() {
            write!(f, "{}", encoded)
        } else {
            Ok(())
        }
    }
}