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
|
use std::{fmt::Debug, process, time::Instant};
use clap::Parser;
use rebel_lang::{
func::{Func, FuncDef, FuncType},
typing::{ArrayLen, Type, TypeFamily},
value::{Context, EvalError, Result, Value},
};
use rebel_parse::{recipe, tokenize};
#[derive(Clone, Debug, Parser)]
struct Opts {
input: String,
}
fn intrinsic_array_len(params: &[Value]) -> Result<Value> {
assert!(params.len() == 1);
let Value::Array(array) = ¶ms[0] else {
panic!();
};
Ok(Value::Integer(array.len().try_into().or(Err(EvalError))?))
}
fn intrinsic_string_len(params: &[Value]) -> Result<Value> {
assert!(params.len() == 1);
let Value::Str(string) = ¶ms[0] else {
panic!();
};
Ok(Value::Integer(
string.chars().count().try_into().or(Err(EvalError))?,
))
}
fn main() {
let opts: Opts = Opts::parse();
let input = opts.input.trim();
let start = Instant::now();
let result = tokenize::token_stream(input);
let dur = Instant::now().duration_since(start);
println!("Tokenization took {} µs", dur.as_micros());
let tokens = match result {
Ok(value) => value,
Err(err) => {
println!("{err}");
process::exit(1);
}
};
let start = Instant::now();
let result = recipe::expr(&tokens);
let dur = Instant::now().duration_since(start);
println!("Parsing took {} µs", dur.as_micros());
let expr = match result {
Ok(value) => value,
Err(err) => {
println!("{err}");
process::exit(1);
}
};
let mut ctx = Context::default();
ctx.methods.entry(TypeFamily::Array).or_default().insert(
"len",
Func {
typ: FuncType {
params: vec![Type::Array(Box::new(Type::Free), ArrayLen::Dynamic)],
ret: Type::Int,
},
def: FuncDef::Intrinsic(intrinsic_array_len),
},
);
ctx.methods.entry(TypeFamily::Str).or_default().insert(
"len",
Func {
typ: FuncType {
params: vec![Type::Str],
ret: Type::Int,
},
def: FuncDef::Intrinsic(intrinsic_string_len),
},
);
let start = Instant::now();
let result = ctx.eval(&expr);
let dur = Instant::now().duration_since(start);
println!("Eval took {} µs", dur.as_micros());
let value = match result {
Ok(value) => value,
Err(err) => {
println!("{err:?}");
process::exit(1);
}
};
println!("{value}");
}
|
