aboutsummaryrefslogtreecommitdiff
path: root/src/util.rs
blob: 4b47afc5dba8d2b8498696cbc10f6c74bc674a64 (plain) (blame)
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
#![allow(non_snake_case, non_upper_case_globals)]

use crate::ast::*;

// intentionally small: i want to run into errors
/// assumption: the count is instantiated to zero
pub fn unique_ident(count: &mut u8) -> String {
    *count += 1;
    if *count == 0 {
        panic!("we've overflowed!");
    } else {
        return String::from(format!("{:X}", count));
    }
}

pub fn Ann(expr: Expression, kind: Type) -> Expression {
    return Expression::Annotation {
        expr: Box::new(expr),
        kind: kind
    };
}

pub fn Const(term: Term) -> Expression {
    return Expression::Constant { term };
}

pub fn Var(id: &str) -> Expression {
    return Expression::Variable {
        id: String::from(id)
    };
}

pub fn Abs(param: &str, func: Expression) -> Expression {
    return Expression::Abstraction {
        param: String::from(param),
        func: Box::new(func),
    };
}

pub fn App(func: Expression, arg: Expression) -> Expression {
    return Expression::Application {
        func: Box::new(func),
        arg: Box::new(arg)
    };
}

pub fn Cond(if_cond: Expression, if_then: Expression, if_else: Expression) -> Expression {
    return Expression::Conditional {
        if_cond: Box::new(if_cond),
        if_then: Box::new(if_then),
        if_else: Box::new(if_else)
    };
}

pub fn Func(from: Type, to: Type) -> Type {
    return Type::Function {
        from: Box::new(from),
        to: Box::new(to)
    }
}

pub const Empty: Type = Type::Empty;
pub const Error: Type = Type::Empty;
pub const Unit: Type = Type::Unit;
pub const Bool: Type = Type::Boolean;
pub const Nat: Type = Type::Natural;
pub const Int: Type = Type::Integer;

pub fn Float(term: f32) -> Term {
    return Term::Float(term)
}

pub fn Str(len: usize, cap: usize, data: Vec<usize>) -> Term {
    return Term::String { len, cap, data }
}

pub fn Enum(val: usize, data: Vec<Type>) -> Term {
    return Term::Enum { val, data }
}