rhai/src/packages/iter_basic.rs

106 lines
3.0 KiB
Rust
Raw Normal View History

2020-10-28 15:30:35 +01:00
use crate::dynamic::Variant;
use crate::stdlib::{
boxed::Box,
ops::{Add, Range},
};
2020-11-16 16:10:14 +01:00
use crate::{def_package, EvalAltResult, INT};
fn get_range<T: Variant + Clone>(from: T, to: T) -> Result<Range<T>, Box<EvalAltResult>> {
2020-05-13 13:21:42 +02:00
Ok(from..to)
}
// Register range function with step
#[derive(Debug, Clone, Copy, Hash, Eq, PartialEq)]
struct StepRange<T>(T, T, T)
where
for<'a> &'a T: Add<&'a T, Output = T>,
T: Variant + Clone + PartialOrd;
impl<T> Iterator for StepRange<T>
where
for<'a> &'a T: Add<&'a T, Output = T>,
T: Variant + Clone + PartialOrd,
{
type Item = T;
fn next(&mut self) -> Option<T> {
if self.0 < self.1 {
let v = self.0.clone();
2020-11-25 02:36:06 +01:00
self.0 = self.0.add(&self.2);
Some(v)
} else {
None
}
}
}
fn get_step_range<T>(from: T, to: T, step: T) -> Result<StepRange<T>, Box<EvalAltResult>>
2020-05-13 13:21:42 +02:00
where
for<'a> &'a T: Add<&'a T, Output = T>,
T: Variant + Clone + PartialOrd,
{
Ok(StepRange::<T>(from, to, step))
}
2020-10-25 15:08:02 +01:00
#[cfg(not(feature = "only_i32"))]
#[cfg(not(feature = "only_i64"))]
2020-10-19 17:49:01 +02:00
macro_rules! reg_range {
($lib:expr, $x:expr, $( $y:ty ),*) => (
$(
$lib.set_iterator::<Range<$y>>();
2020-11-22 10:21:34 +01:00
let hash = $lib.set_fn_2($x, get_range::<$y>);
2020-11-22 15:15:17 +01:00
$lib.update_fn_metadata(hash, [
concat!("from: ", stringify!($y)),
concat!("to: ", stringify!($y)),
concat!("Iterator<Item=", stringify!($y), ">")
]);
2020-10-19 17:49:01 +02:00
)*
)
}
2020-10-25 15:08:02 +01:00
#[cfg(not(feature = "only_i32"))]
#[cfg(not(feature = "only_i64"))]
2020-10-19 17:49:01 +02:00
macro_rules! reg_step {
($lib:expr, $x:expr, $( $y:ty ),*) => (
$(
$lib.set_iterator::<StepRange<$y>>();
2020-11-22 10:21:34 +01:00
let hash = $lib.set_fn_3($x, get_step_range::<$y>);
2020-11-22 15:15:17 +01:00
$lib.update_fn_metadata(hash, [
concat!("from: ", stringify!($y)),
concat!("to: ", stringify!($y)),
concat!("step: ", stringify!($y)), concat!("Iterator<Item=", stringify!($y), ">")
]);
2020-10-19 17:49:01 +02:00
)*
)
}
2020-05-13 13:21:42 +02:00
def_package!(crate:BasicIteratorPackage:"Basic range iterators.", lib, {
2020-10-15 16:11:18 +02:00
lib.set_iterator::<Range<INT>>();
2020-11-22 10:21:34 +01:00
let hash = lib.set_fn_2("range", get_range::<INT>);
2020-11-22 15:15:17 +01:00
lib.update_fn_metadata(hash, ["from: INT", "to: INT", "Iterator<Item=INT>"]);
2020-04-21 17:01:10 +02:00
2020-10-19 17:49:01 +02:00
#[cfg(not(feature = "only_i32"))]
#[cfg(not(feature = "only_i64"))]
{
2020-06-17 10:50:46 +02:00
reg_range!(lib, "range", i8, u8, i16, u16, i32, i64, u32, u64);
2020-07-31 16:30:23 +02:00
if cfg!(not(target_arch = "wasm32")) {
reg_range!(lib, "range", i128, u128);
}
2020-04-21 17:01:10 +02:00
}
2020-10-15 16:11:18 +02:00
lib.set_iterator::<StepRange<INT>>();
2020-11-22 10:21:34 +01:00
let hash = lib.set_fn_3("range", get_step_range::<INT>);
2020-11-22 15:15:17 +01:00
lib.update_fn_metadata(hash, ["from: INT", "to: INT", "step: INT", "Iterator<Item=INT>"]);
2020-04-21 17:01:10 +02:00
2020-10-19 17:49:01 +02:00
#[cfg(not(feature = "only_i32"))]
#[cfg(not(feature = "only_i64"))]
{
2020-06-17 10:50:46 +02:00
reg_step!(lib, "range", i8, u8, i16, u16, i32, i64, u32, u64);
2020-07-31 16:30:23 +02:00
if cfg!(not(target_arch = "wasm32")) {
reg_step!(lib, "range", i128, u128);
}
}
2020-04-21 17:01:10 +02:00
});