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

use crate::line::{ChoiceData, LineData, ParsedLine};

use std::cell::Cell;

use super::parse::parse_full_node;

#[derive(Debug)]
/// Node in a graph representation of a dialogue tree.
pub struct DialogueNode {
    /// Children of current node.
    pub items: Vec<NodeItem>,
    pub num_visited: Cell<u32>,
}

impl DialogueNode {
    /// Parse a set of `ParsedLine` items and create a full graph representation of it.
    pub fn from_lines(lines: &[ParsedLine]) -> Self {
        parse_full_node(lines)
    }

    pub fn with_items(items: Vec<NodeItem>) -> Self {
        DialogueNode {
            items,
            num_visited: Cell::new(0),
        }
    }
}

#[derive(Debug)]
pub enum NodeItem {
    /// Regular line of marked up text.
    Line(LineData),
    /// Nested node, either a `ChoiceSet` which has `Choice`s as children, or a
    /// `Choice` which has more `Line`s and possibly further `ChoiceSet`s.
    Node {
        kind: NodeType,
        node: Box<DialogueNode>,
    },
}

#[derive(Debug)]
pub enum NodeType {
    /// Root of a set of choices. All node items will be of type `Choice`.
    ChoiceSet,
    /// Choice in a set of choices. All node items will be lines or further `ChoiceSet` nodes.
    Choice(ChoiceData),
}