ratex-parser 0.1.1

LaTeX parser for RaTeX
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
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

use std::collections::HashMap;

use crate::error::{ParseError, ParseResult};
use crate::functions::{define_function_full, FunctionContext, FunctionSpec};
use crate::parse_node::ParseNode;

pub fn register(map: &mut HashMap<&'static str, FunctionSpec>) {
    define_function_full(
        map,
        &["\\left"],
        "leftright",
        1, 0, None,
        false, false, true, false,
        true, // primitive
        handle_left,
    );

    define_function_full(
        map,
        &["\\right"],
        "leftright-right",
        1, 0, None,
        false, false, true, false,
        true, // primitive
        handle_right,
    );

    define_function_full(
        map,
        &["\\middle"],
        "middle",
        1, 0, None,
        false, false, true, false,
        true,
        handle_middle,
    );
}

fn handle_left(
    ctx: &mut FunctionContext,
    args: Vec<ParseNode>,
    _opt_args: Vec<Option<ParseNode>>,
) -> ParseResult<ParseNode> {
    let delim = get_delim_text(&args[0])?;

    ctx.parser.leftright_depth += 1;
    let body = ctx.parser.parse_expression(false, None)?;
    ctx.parser.leftright_depth -= 1;

    // Expect \right, but don't consume it yet
    ctx.parser.expect("\\right", false)?;
    // Parse \right as a function, which returns a leftright-right node
    let right_node = ctx.parser.parse_function(None, None)?;

    let (right, right_color) = match right_node {
        Some(ParseNode::LeftRightRight { delim, color, .. }) => (delim, color),
        _ => {
            return Err(ParseError::msg("Expected \\right after \\left"));
        }
    };

    Ok(ParseNode::LeftRight {
        mode: ctx.parser.mode,
        body,
        left: delim,
        right,
        right_color,
        loc: None,
    })
}

fn handle_right(
    ctx: &mut FunctionContext,
    args: Vec<ParseNode>,
    _opt_args: Vec<Option<ParseNode>>,
) -> ParseResult<ParseNode> {
    let delim = get_delim_text(&args[0])?;

    Ok(ParseNode::LeftRightRight {
        mode: ctx.parser.mode,
        delim,
        color: None,
        loc: None,
    })
}

fn handle_middle(
    ctx: &mut FunctionContext,
    args: Vec<ParseNode>,
    _opt_args: Vec<Option<ParseNode>>,
) -> ParseResult<ParseNode> {
    let delim = get_delim_text(&args[0])?;

    if ctx.parser.leftright_depth <= 0 {
        return Err(ParseError::msg(
            "\\middle must be within \\left and \\right",
        ));
    }

    Ok(ParseNode::Middle {
        mode: ctx.parser.mode,
        delim,
        loc: None,
    })
}

fn get_delim_text(node: &ParseNode) -> ParseResult<String> {
    if let Some(text) = node.symbol_text() {
        return Ok(text.to_string());
    }
    Err(ParseError::msg(format!(
        "Invalid delimiter type '{}'",
        node.type_name(),
    )))
}