diff options
Diffstat (limited to 'crates')
| -rw-r--r-- | crates/teleia/src/script.rs | 34 |
1 files changed, 33 insertions, 1 deletions
diff --git a/crates/teleia/src/script.rs b/crates/teleia/src/script.rs index 88fe759..2e7ea3c 100644 --- a/crates/teleia/src/script.rs +++ b/crates/teleia/src/script.rs @@ -36,7 +36,7 @@ extern "C" fn unwrap_nativefunc(rt: *mut PitRuntime, args: PitValue, data: *mut #[repr(C)] #[derive(Debug, Clone, Copy)] -enum PitSort { +pub enum Sort { Double, Integer, Symbol, @@ -81,6 +81,17 @@ struct PitParser { #[allow(dead_code)] unsafe extern "C" { + static PIT__SORT_DOUBLE: i64; + static PIT__SORT_INTEGER: i64; + static PIT__SORT_SYMBOL: i64; + static PIT__SORT_CELL: i64; + static PIT__SORT_CONS: i64; + static PIT__SORT_ARRAY: i64; + static PIT__SORT_BYTES: i64; + static PIT__SORT_FUNC: i64; + static PIT__SORT_NATIVEFUNC: i64; + static PIT__SORT_NATIVEDATA: i64; + pub fn pit_runtime_test(out: *mut u8, out_len: i64, buf: *mut u8, len: i64) -> c_int; fn pit_runtime_new(buf: *mut MaybeUninit<u8>, len: i64) -> *mut PitRuntime; fn pit_runtime_set_fuel(rt: *mut PitRuntime, fuel: i64); @@ -104,6 +115,8 @@ unsafe extern "C" { fn pit_symtab_get(rt: *mut PitRuntime, sym: PitValue) -> PitValue; fn pit_symtab_fget(rt: *mut PitRuntime, sym: PitValue) -> PitValue; + fn pit_value_sort(x: PitValue) -> i64; + fn pit_value_sort_or_heavy_sort(x: PitValue) -> i64; fn pit_value_eq(x: PitValue, y: PitValue) -> bool; fn pit_value_as_integer(rt: *mut PitRuntime, v: PitValue) -> i64; @@ -168,6 +181,25 @@ impl Runtime { pit_runtime_set_fuel(self.rt, fuel); } } + pub fn sort_of(&mut self, v: Value) -> utils::Erm<Sort> { + unsafe { + let vt = pit_value_sort_or_heavy_sort(v.val); + self.error()?; + if vt == PIT__SORT_DOUBLE { Ok(Sort::Double) } + else if vt == PIT__SORT_INTEGER { Ok(Sort::Integer) } + else if vt == PIT__SORT_SYMBOL { Ok(Sort::Symbol) } + else if vt == PIT__SORT_CELL { Ok(Sort::Cell) } + else if vt == PIT__SORT_CONS { Ok(Sort::Cons) } + else if vt == PIT__SORT_ARRAY { Ok(Sort::Array) } + else if vt == PIT__SORT_BYTES { Ok(Sort::Bytes) } + else if vt == PIT__SORT_FUNC { Ok(Sort::Func) } + else if vt == PIT__SORT_NATIVEFUNC { Ok(Sort::NativeFunc) } + else if vt == PIT__SORT_NATIVEDATA { Ok(Sort::NativeData) } + else { + utils::erm(Error { msg: format!("unknown pit sort index: {}", vt) }) + } + } + } pub fn parse(&mut self, s: &str) -> utils::Erm<Value> { let lexer = Lexer::from_bytes(s.as_bytes()); let mut parser = Parser::from_lexer(lexer); |