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

use std::fmt::{Display, Formatter, LowerHex, UpperHex, Write};

mod as_tex;
mod as_wolfram;
mod predefined;
mod utils;

pub use self::{predefined::*, utils::List};

pub enum ASTNode {
    Sequence(Sequence),
    List(List),
    Variable(String),
    Function(String, Vec<ASTNode>),
    Evaluated(Box<dyn WoTeXFunction>),
}

pub trait WoTeXFunction: LowerHex + UpperHex {
    fn as_tex(&self) -> String {
        format!("{:x}", self)
    }
    fn as_wolfram(&self) -> String {
        format!("{:X}", self)
    }
}

pub struct Sequence {
    inner: Vec<ASTNode>,
}

impl Default for Sequence {
    fn default() -> Self {
        Self { inner: vec![] }
    }
}

pub struct Integrate {
    pub kind: String,
    pub regional: bool,
    /// A sequence of ASTNodes
    pub expr: Sequence,
    pub vars: Vec<List>,
}

/// Functions that can be added with or without parentheses
///
/// eg: sin cos tan arctan
pub struct MaybeParentheses {
    head: String,
    operator_name: bool,
    rest: Vec<Sequence>,
}

#[test]
fn test() {
    let seq = Sequence { inner: vec![ASTNode::Variable("a".to_string()), ASTNode::Variable("b".to_string())] };
    let sin = TrigonometricFunction::sin(seq);
    let ast = ASTNode::Evaluated(sin);
    println!("{:x}", ast);

    let expr = Sequence { inner: vec![ASTNode::Variable("x".to_string()), ASTNode::Variable("y".to_string())] };
    let vars = List::one(ASTNode::Variable("x".to_string()));
    let int = Integrate::int(expr, vec![vars], false);
    let ast = ASTNode::Evaluated(int);
    println!("{:x}", ast);
}