forked from Orchid/orchid
Lawrence Bethlenfalvy
cce4b8f11c
- removed many panics from the pipeline - extracted project and const tree to representations - extended STL list support - improved loops
67 lines
2.1 KiB
Rust
67 lines
2.1 KiB
Rust
use chumsky::Parser;
|
|
use ordered_float::NotNan;
|
|
|
|
use super::litconv::with_lit;
|
|
use super::{ArithmeticError, AssertionError};
|
|
use crate::foreign::ExternError;
|
|
use crate::interner::Interner;
|
|
use crate::parse::{float_parser, int_parser};
|
|
use crate::{define_fn, ConstTree, Literal};
|
|
|
|
define_fn! {
|
|
/// parse a number. Accepts the same syntax Orchid does.
|
|
ToFloat = |x| with_lit(x, |l| match l {
|
|
Literal::Str(s) => float_parser()
|
|
.parse(s.as_str())
|
|
.map_err(|_| AssertionError::ext(
|
|
x.clone(),
|
|
"cannot be parsed into a float"
|
|
)),
|
|
Literal::Num(n) => Ok(*n),
|
|
Literal::Uint(i) => NotNan::new(*i as f64)
|
|
.map_err(|_| ArithmeticError::NaN.into_extern()),
|
|
Literal::Char(char) => char
|
|
.to_digit(10)
|
|
.map(|i| NotNan::new(i as f64).expect("u32 to f64 yielded NaN"))
|
|
.ok_or_else(|| AssertionError::ext(x.clone(), "is not a decimal digit")),
|
|
}).map(|nn| Literal::Num(nn).into())
|
|
}
|
|
|
|
define_fn! {
|
|
/// Parse an unsigned integer. Accepts the same formats Orchid does. If the
|
|
/// input is a number, floors it.
|
|
ToUint = |x| with_lit(x, |l| match l {
|
|
Literal::Str(s) => int_parser()
|
|
.parse(s.as_str())
|
|
.map_err(|_| AssertionError::ext(
|
|
x.clone(),
|
|
"cannot be parsed into an unsigned int",
|
|
)),
|
|
Literal::Num(n) => Ok(n.floor() as u64),
|
|
Literal::Uint(i) => Ok(*i),
|
|
Literal::Char(char) => char
|
|
.to_digit(10)
|
|
.map(u64::from)
|
|
.ok_or(AssertionError::ext(x.clone(), "is not a decimal digit")),
|
|
}).map(|u| Literal::Uint(u).into())
|
|
}
|
|
|
|
define_fn! {
|
|
/// Convert a literal to a string using Rust's conversions for floats, chars and
|
|
/// uints respectively
|
|
ToString = |x| with_lit(x, |l| Ok(match l {
|
|
Literal::Char(c) => c.to_string(),
|
|
Literal::Uint(i) => i.to_string(),
|
|
Literal::Num(n) => n.to_string(),
|
|
Literal::Str(s) => s.clone(),
|
|
})).map(|s| Literal::Str(s).into())
|
|
}
|
|
|
|
pub fn conv(i: &Interner) -> ConstTree {
|
|
ConstTree::tree([
|
|
(i.i("to_float"), ConstTree::xfn(ToFloat)),
|
|
(i.i("to_uint"), ConstTree::xfn(ToUint)),
|
|
(i.i("to_string"), ConstTree::xfn(ToString)),
|
|
])
|
|
}
|