2020-04-21 17:01:10 +02:00
|
|
|
use super::{reg_binary, reg_trinary, reg_unary_mut, PackageStore};
|
2020-04-20 18:11:25 +02:00
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
use crate::def_package;
|
2020-04-20 18:11:25 +02:00
|
|
|
use crate::any::{Dynamic, Union, Variant};
|
|
|
|
use crate::engine::{Array, Map};
|
|
|
|
use crate::fn_register::map_dynamic as map;
|
|
|
|
use crate::parser::INT;
|
|
|
|
|
|
|
|
use crate::stdlib::{
|
|
|
|
any::TypeId,
|
2020-04-21 17:01:10 +02:00
|
|
|
ops::{Add, Range},
|
2020-04-20 18:11:25 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
// Register range function
|
2020-04-21 17:01:10 +02:00
|
|
|
fn reg_range<T: Variant + Clone>(lib: &mut PackageStore)
|
2020-04-20 18:11:25 +02:00
|
|
|
where
|
|
|
|
Range<T>: Iterator<Item = T>,
|
|
|
|
{
|
2020-04-21 17:01:10 +02:00
|
|
|
lib.type_iterators.insert(
|
2020-04-20 18:11:25 +02:00
|
|
|
TypeId::of::<Range<T>>(),
|
|
|
|
Box::new(|source: &Dynamic| {
|
|
|
|
Box::new(
|
|
|
|
source
|
|
|
|
.downcast_ref::<Range<T>>()
|
|
|
|
.cloned()
|
|
|
|
.unwrap()
|
|
|
|
.map(|x| x.into_dynamic()),
|
|
|
|
) as Box<dyn Iterator<Item = Dynamic>>
|
|
|
|
}),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
// 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();
|
|
|
|
self.0 = &v + &self.2;
|
|
|
|
Some(v)
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
fn reg_step<T>(lib: &mut PackageStore)
|
2020-04-20 18:11:25 +02:00
|
|
|
where
|
|
|
|
for<'a> &'a T: Add<&'a T, Output = T>,
|
|
|
|
T: Variant + Clone + PartialOrd,
|
|
|
|
StepRange<T>: Iterator<Item = T>,
|
|
|
|
{
|
2020-04-21 17:01:10 +02:00
|
|
|
lib.type_iterators.insert(
|
2020-04-20 18:11:25 +02:00
|
|
|
TypeId::of::<StepRange<T>>(),
|
|
|
|
Box::new(|source: &Dynamic| {
|
|
|
|
Box::new(
|
|
|
|
source
|
|
|
|
.downcast_ref::<StepRange<T>>()
|
|
|
|
.cloned()
|
|
|
|
.unwrap()
|
|
|
|
.map(|x| x.into_dynamic()),
|
|
|
|
) as Box<dyn Iterator<Item = Dynamic>>
|
|
|
|
}),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
def_package!(BasicIteratorPackage:"Basic range iterators.", lib, {
|
|
|
|
fn get_range<T>(from: T, to: T) -> Range<T> {
|
|
|
|
from..to
|
2020-04-20 18:11:25 +02:00
|
|
|
}
|
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
reg_range::<INT>(lib);
|
|
|
|
reg_binary(lib, "range", get_range::<INT>, map);
|
|
|
|
|
|
|
|
#[cfg(not(feature = "only_i32"))]
|
|
|
|
#[cfg(not(feature = "only_i64"))]
|
|
|
|
{
|
|
|
|
macro_rules! reg_range {
|
|
|
|
($self:expr, $x:expr, $( $y:ty ),*) => (
|
|
|
|
$(
|
|
|
|
reg_range::<$y>($self);
|
|
|
|
reg_binary($self, $x, get_range::<$y>, map);
|
|
|
|
)*
|
|
|
|
)
|
2020-04-20 18:11:25 +02:00
|
|
|
}
|
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
reg_range!(lib, "range", i8, u8, i16, u16, i32, i64, u32, u64, i128, u128);
|
|
|
|
}
|
2020-04-20 18:11:25 +02:00
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
reg_step::<INT>(lib);
|
|
|
|
reg_trinary(lib, "range", StepRange::<INT>, map);
|
|
|
|
|
|
|
|
#[cfg(not(feature = "only_i32"))]
|
|
|
|
#[cfg(not(feature = "only_i64"))]
|
|
|
|
{
|
|
|
|
macro_rules! reg_step {
|
|
|
|
($self:expr, $x:expr, $( $y:ty ),*) => (
|
|
|
|
$(
|
|
|
|
reg_step::<$y>($self);
|
|
|
|
reg_trinary($self, $x, StepRange::<$y>, map);
|
|
|
|
)*
|
|
|
|
)
|
2020-04-20 18:11:25 +02:00
|
|
|
}
|
|
|
|
|
2020-04-21 17:01:10 +02:00
|
|
|
reg_step!(lib, "range", i8, u8, i16, u16, i32, i64, u32, u64, i128, u128);
|
2020-04-20 18:11:25 +02:00
|
|
|
}
|
2020-04-21 17:01:10 +02:00
|
|
|
});
|