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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
#[cfg(test)]
mod tests;
use std::collections::VecDeque;
use std::fmt::{self, Display};
use std::rc::Rc;
#[derive(Debug, Clone)]
pub struct Tree<D: Display> {
pub root: D,
pub leaves: Vec<Tree<D>>,
multiline: bool,
glyphs: GlyphPalette,
}
impl<D: Display> Tree<D> {
pub fn new(root: D) -> Self {
Tree {
root,
leaves: Vec::new(),
multiline: false,
glyphs: GlyphPalette::new(),
}
}
pub fn with_leaves(mut self, leaves: impl IntoIterator<Item = impl Into<Tree<D>>>) -> Self {
self.leaves = leaves.into_iter().map(Into::into).collect();
self
}
pub fn with_multiline(mut self, yes: bool) -> Self {
self.multiline = yes;
self
}
pub fn with_glyphs(mut self, glyphs: GlyphPalette) -> Self {
self.glyphs = glyphs;
self
}
}
impl<D: Display> Tree<D> {
pub fn set_multiline(&mut self, yes: bool) -> &mut Self {
self.multiline = yes;
self
}
pub fn set_glyphs(&mut self, glyphs: GlyphPalette) -> &mut Self {
self.glyphs = glyphs;
self
}
}
impl<D: Display> Tree<D> {
pub fn push(&mut self, leaf: impl Into<Tree<D>>) -> &mut Self {
self.leaves.push(leaf.into());
self
}
}
impl<D: Display> From<D> for Tree<D> {
fn from(inner: D) -> Self {
Self::new(inner)
}
}
impl<D: Display> Extend<D> for Tree<D> {
fn extend<T: IntoIterator<Item = D>>(&mut self, iter: T) {
self.leaves.extend(iter.into_iter().map(Into::into))
}
}
impl<D: Display> Extend<Tree<D>> for Tree<D> {
fn extend<T: IntoIterator<Item = Tree<D>>>(&mut self, iter: T) {
self.leaves.extend(iter)
}
}
impl<D: Display> Display for Tree<D> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
writeln!(f, "{}", self.root)?;
let mut queue = DisplauQueue::new();
let no_space = Rc::new(Vec::new());
enqueue_leaves(&mut queue, self, no_space);
while let Some((last, leaf, spaces)) = queue.pop_front() {
let mut prefix = (
if last {
leaf.glyphs.last_item
} else {
leaf.glyphs.middle_item
},
leaf.glyphs.item_indent,
);
if leaf.multiline {
let rest_prefix = (
if last {
leaf.glyphs.last_skip
} else {
leaf.glyphs.middle_skip
},
leaf.glyphs.skip_indent,
);
debug_assert_eq!(prefix.0.chars().count(), rest_prefix.0.chars().count());
debug_assert_eq!(prefix.1.chars().count(), rest_prefix.1.chars().count());
let root = leaf.root.to_string();
for line in root.lines() {
for s in spaces.as_slice() {
if *s {
write!(f, "{}{}", self.glyphs.last_skip, self.glyphs.skip_indent)?;
} else {
write!(f, "{}{}", self.glyphs.middle_skip, self.glyphs.skip_indent)?;
}
}
writeln!(f, "{}{}{}", prefix.0, prefix.1, line)?;
prefix = rest_prefix;
}
} else {
for s in spaces.as_slice() {
if *s {
write!(f, "{}{}", self.glyphs.last_skip, self.glyphs.skip_indent)?;
} else {
write!(f, "{}{}", self.glyphs.middle_skip, self.glyphs.skip_indent)?;
}
}
writeln!(f, "{}{}{}", prefix.0, prefix.1, leaf.root)?;
}
if !leaf.leaves.is_empty() {
let s: &Vec<bool> = &spaces;
let mut child_spaces = s.clone();
child_spaces.push(last);
let child_spaces = Rc::new(child_spaces);
enqueue_leaves(&mut queue, leaf, child_spaces);
}
}
Ok(())
}
}
type DisplauQueue<'t, D> = VecDeque<(bool, &'t Tree<D>, Rc<Vec<bool>>)>;
fn enqueue_leaves<'t, D: Display>(
queue: &mut DisplauQueue<'t, D>,
parent: &'t Tree<D>,
spaces: Rc<Vec<bool>>,
) {
for (i, leaf) in parent.leaves.iter().rev().enumerate() {
let last = i == 0;
queue.push_front((last, leaf, spaces.clone()));
}
}
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
pub struct GlyphPalette {
pub middle_item: &'static str,
pub last_item: &'static str,
pub item_indent: &'static str,
pub middle_skip: &'static str,
pub last_skip: &'static str,
pub skip_indent: &'static str,
}
impl GlyphPalette {
pub const fn new() -> Self {
Self {
middle_item: "├",
last_item: "└",
item_indent: "── ",
middle_skip: "│",
last_skip: " ",
skip_indent: " ",
}
}
}
impl Default for GlyphPalette {
fn default() -> Self {
Self::new()
}
}