413 lines
11 KiB
Rust
413 lines
11 KiB
Rust
use crate::Span;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub enum EventKind<C, A> {
|
|
Enter(C),
|
|
Inline,
|
|
Exit(C),
|
|
Atom(A),
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
pub struct Event<C, A> {
|
|
pub kind: EventKind<C, A>,
|
|
pub span: Span,
|
|
}
|
|
|
|
pub struct Tree<C, A>(Box<[Node<C, A>]>);
|
|
|
|
#[derive(Clone)]
|
|
pub struct Inlines<'t, C, A> {
|
|
iter: std::slice::Iter<'t, Node<C, A>>,
|
|
}
|
|
|
|
impl<'t, C, A> Iterator for Inlines<'t, C, A> {
|
|
type Item = Span;
|
|
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
self.iter.next().map(|n| n.span)
|
|
}
|
|
}
|
|
|
|
impl<C, A> Tree<C, A> {
|
|
pub fn root(&self) -> Branch<C, A> {
|
|
let head = self.0[NodeIndex::root().index()].next;
|
|
// SAFETY: tree must outlive the branch
|
|
let nodes = unsafe { std::mem::transmute::<&[Node<C, A>], &'static [Node<C, A>]>(&self.0) };
|
|
Branch {
|
|
nodes,
|
|
branch: Vec::new(),
|
|
head,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct Branch<C: 'static, A: 'static> {
|
|
nodes: &'static [Node<C, A>],
|
|
branch: Vec<NodeIndex>,
|
|
head: Option<NodeIndex>,
|
|
}
|
|
|
|
impl<C, A> Branch<C, A> {
|
|
pub fn empty() -> Self {
|
|
Self {
|
|
nodes: &[],
|
|
branch: Vec::new(),
|
|
head: None,
|
|
}
|
|
}
|
|
|
|
/// Count number of direct children nodes.
|
|
pub fn count_children(&self) -> usize {
|
|
let mut head = self.head;
|
|
let mut count = 0;
|
|
while let Some(h) = head {
|
|
let n = &self.nodes[h.index()];
|
|
head = n.next;
|
|
count += 1;
|
|
}
|
|
count
|
|
}
|
|
|
|
pub fn take_branch(&mut self) -> Self {
|
|
let head = self.head.take();
|
|
self.head = self.branch.pop();
|
|
if let Some(h) = self.head {
|
|
let n = &self.nodes[h.index()];
|
|
self.head = n.next;
|
|
}
|
|
Self {
|
|
nodes: self.nodes,
|
|
branch: Vec::new(),
|
|
head,
|
|
}
|
|
}
|
|
|
|
/// Retrieve all inlines until the end of the current container. Panics if any upcoming node is
|
|
/// not an inline node.
|
|
pub fn take_inlines(&mut self) -> impl Iterator<Item = Span> + '_ {
|
|
let mut head = self.head.take();
|
|
std::iter::from_fn(move || {
|
|
head.take().map(|h| {
|
|
let n = &self.nodes[h.index()];
|
|
assert!(matches!(n.kind, NodeKind::Inline));
|
|
head = n.next;
|
|
n.span
|
|
})
|
|
})
|
|
}
|
|
}
|
|
|
|
impl<C: Clone, A: Clone> Iterator for Branch<C, A> {
|
|
type Item = Event<C, A>;
|
|
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
if let Some(head) = self.head {
|
|
let n = &self.nodes[head.index()];
|
|
let kind = match &n.kind {
|
|
NodeKind::Root => unreachable!(),
|
|
NodeKind::Container(c, child) => {
|
|
self.branch.push(head);
|
|
self.head = *child;
|
|
EventKind::Enter(c.clone())
|
|
}
|
|
NodeKind::Atom(a) => {
|
|
self.head = n.next;
|
|
EventKind::Atom(a.clone())
|
|
}
|
|
NodeKind::Inline => {
|
|
self.head = n.next;
|
|
EventKind::Inline
|
|
}
|
|
};
|
|
Some(Event { kind, span: n.span })
|
|
} else if let Some(block_ni) = self.branch.pop() {
|
|
let Node { next, kind, span } = &self.nodes[block_ni.index()];
|
|
let kind = match kind {
|
|
NodeKind::Container(c, _) => EventKind::Exit(c.clone()),
|
|
_ => panic!(),
|
|
};
|
|
self.head = *next;
|
|
Some(Event { kind, span: *span })
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
struct NodeIndex(std::num::NonZeroUsize);
|
|
|
|
impl NodeIndex {
|
|
fn new(i: usize) -> Self {
|
|
assert_ne!(i, usize::MAX);
|
|
Self((i + 1).try_into().unwrap())
|
|
}
|
|
|
|
fn root() -> Self {
|
|
Self::new(0)
|
|
}
|
|
|
|
fn index(self) -> usize {
|
|
usize::from(self.0) - 1
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
enum NodeKind<C, A> {
|
|
Root,
|
|
Container(C, Option<NodeIndex>),
|
|
Atom(A),
|
|
Inline,
|
|
}
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct Node<C, A> {
|
|
span: Span,
|
|
kind: NodeKind<C, A>,
|
|
next: Option<NodeIndex>,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct Builder<C, A> {
|
|
nodes: Vec<Node<C, A>>,
|
|
branch: Vec<NodeIndex>,
|
|
head: Option<NodeIndex>,
|
|
}
|
|
|
|
impl<C: Clone, A: Clone> Builder<C, A> {
|
|
pub(super) fn new() -> Self {
|
|
Builder {
|
|
nodes: vec![Node {
|
|
span: Span::default(),
|
|
kind: NodeKind::Root,
|
|
next: None,
|
|
}],
|
|
branch: vec![],
|
|
head: Some(NodeIndex::root()),
|
|
}
|
|
}
|
|
|
|
pub(super) fn atom(&mut self, a: A, span: Span) {
|
|
self.add_node(Node {
|
|
span,
|
|
kind: NodeKind::Atom(a),
|
|
next: None,
|
|
});
|
|
}
|
|
|
|
pub(super) fn inline(&mut self, span: Span) {
|
|
self.add_node(Node {
|
|
span,
|
|
kind: NodeKind::Inline,
|
|
next: None,
|
|
});
|
|
}
|
|
|
|
pub(super) fn enter(&mut self, c: C, span: Span) {
|
|
self.add_node(Node {
|
|
span,
|
|
kind: NodeKind::Container(c, None),
|
|
next: None,
|
|
});
|
|
}
|
|
|
|
pub(super) fn exit(&mut self) {
|
|
if self.head.is_some() {
|
|
self.head = None;
|
|
} else {
|
|
let last = self.branch.pop();
|
|
assert_ne!(last, None);
|
|
}
|
|
}
|
|
|
|
pub(super) fn finish(self) -> Tree<C, A> {
|
|
Tree(self.nodes.into_boxed_slice())
|
|
}
|
|
|
|
fn add_node(&mut self, node: Node<C, A>) {
|
|
let ni = NodeIndex::new(self.nodes.len());
|
|
self.nodes.push(node);
|
|
if let Some(head_ni) = &mut self.head {
|
|
let mut head = &mut self.nodes[head_ni.index()];
|
|
match &mut head.kind {
|
|
NodeKind::Root | NodeKind::Inline | NodeKind::Atom(_) => {
|
|
// set next pointer of previous node
|
|
assert_eq!(head.next, None);
|
|
head.next = Some(ni);
|
|
}
|
|
NodeKind::Container(_, child) => {
|
|
self.branch.push(*head_ni);
|
|
// set child pointer of current container
|
|
assert_eq!(*child, None);
|
|
*child = Some(ni);
|
|
}
|
|
}
|
|
} else if let Some(block) = self.branch.pop() {
|
|
let mut block = &mut self.nodes[block.index()];
|
|
assert!(matches!(block.kind, NodeKind::Container(..)));
|
|
block.next = Some(ni);
|
|
} else {
|
|
panic!()
|
|
}
|
|
self.head = Some(ni);
|
|
}
|
|
}
|
|
|
|
impl<C: std::fmt::Debug + Clone + 'static, A: std::fmt::Debug + Clone + 'static> std::fmt::Debug
|
|
for Builder<C, A>
|
|
{
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
self.clone().finish().root().fmt(f)
|
|
}
|
|
}
|
|
|
|
impl<C: std::fmt::Debug + Clone, A: std::fmt::Debug + Clone> std::fmt::Debug for Branch<C, A> {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
const INDENT: &str = " ";
|
|
let mut level = 0;
|
|
write!(f, "\n")?;
|
|
for e in self.clone() {
|
|
let indent = INDENT.repeat(level);
|
|
match e.kind {
|
|
EventKind::Enter(c) => {
|
|
write!(f, "{}{:?}", indent, c)?;
|
|
level += 1;
|
|
}
|
|
EventKind::Inline => write!(f, "{}Inline", indent)?,
|
|
EventKind::Exit(..) => {
|
|
level -= 1;
|
|
continue;
|
|
}
|
|
EventKind::Atom(a) => write!(f, "{}{:?}", indent, a)?,
|
|
}
|
|
writeln!(f, " ({}:{})", e.span.start(), e.span.end())?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use crate::Span;
|
|
|
|
use super::Event;
|
|
use super::EventKind;
|
|
|
|
#[test]
|
|
fn fmt() {
|
|
let mut tree = super::Builder::new();
|
|
tree.enter(1, Span::new(0, 1));
|
|
tree.atom(11, Span::new(0, 1));
|
|
tree.atom(12, Span::new(0, 1));
|
|
tree.exit();
|
|
tree.enter(2, Span::new(1, 5));
|
|
tree.enter(21, Span::new(2, 5));
|
|
tree.enter(211, Span::new(3, 4));
|
|
tree.atom(2111, Span::new(3, 4));
|
|
tree.exit();
|
|
tree.exit();
|
|
tree.enter(22, Span::new(4, 5));
|
|
tree.atom(221, Span::new(4, 5));
|
|
tree.exit();
|
|
tree.exit();
|
|
tree.enter(3, Span::new(5, 6));
|
|
tree.atom(31, Span::new(5, 6));
|
|
tree.exit();
|
|
assert_eq!(
|
|
format!("{:?}", tree),
|
|
concat!(
|
|
"\n",
|
|
"1 (0:1)\n",
|
|
" 11 (0:1)\n",
|
|
" 12 (0:1)\n",
|
|
"2 (1:5)\n",
|
|
" 21 (2:5)\n",
|
|
" 211 (3:4)\n",
|
|
" 2111 (3:4)\n",
|
|
" 22 (4:5)\n",
|
|
" 221 (4:5)\n",
|
|
"3 (5:6)\n",
|
|
" 31 (5:6)\n",
|
|
)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn branch_take_branch() {
|
|
let mut b = super::Builder::new();
|
|
let sp = Span::new(0, 0);
|
|
b.enter(1, sp);
|
|
b.atom(11, sp);
|
|
b.exit();
|
|
b.enter(2, sp);
|
|
b.enter(21, sp);
|
|
b.atom(211, sp);
|
|
b.exit();
|
|
b.exit();
|
|
b.enter(3, sp);
|
|
b.atom(31, sp);
|
|
let tree = b.finish();
|
|
|
|
let mut root_branch = tree.root();
|
|
assert_eq!(
|
|
(&mut root_branch).take(3).collect::<Vec<_>>(),
|
|
&[
|
|
Event {
|
|
kind: EventKind::Enter(1),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Atom(11),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Exit(1),
|
|
span: sp
|
|
},
|
|
]
|
|
);
|
|
assert_eq!(
|
|
root_branch.next(),
|
|
Some(Event {
|
|
kind: EventKind::Enter(2),
|
|
span: sp
|
|
})
|
|
);
|
|
assert_eq!(
|
|
root_branch.take_branch().collect::<Vec<_>>(),
|
|
&[
|
|
Event {
|
|
kind: EventKind::Enter(21),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Atom(211),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Exit(21),
|
|
span: sp
|
|
},
|
|
]
|
|
);
|
|
assert_eq!(
|
|
root_branch.collect::<Vec<_>>(),
|
|
&[
|
|
Event {
|
|
kind: EventKind::Enter(3),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Atom(31),
|
|
span: sp
|
|
},
|
|
Event {
|
|
kind: EventKind::Exit(3),
|
|
span: sp
|
|
},
|
|
]
|
|
);
|
|
}
|
|
}
|