use super::Signal;
use core::mem;
#[derive(Clone)]
pub struct Const<T> {
pub(super) value: T,
}
impl<T: Clone> Signal for Const<T> {
type Type = T;
#[inline]
fn next(&mut self) -> Self::Type {
self.value.clone()
}
}
#[derive(Clone)]
pub struct Gen<F> {
pub(super) gen: F,
}
impl<F, T> Signal for Gen<F>
where
F: FnMut() -> T,
{
type Type = T;
#[inline]
fn next(&mut self) -> Self::Type {
(self.gen)()
}
}
#[derive(Clone)]
pub struct Successors<T, F> {
pub(super) state: T,
pub(super) successor: F,
}
impl<T, F> Signal for Successors<T, F>
where
F: FnMut(&T) -> T,
{
type Type = T;
#[inline]
fn next(&mut self) -> Self::Type {
let next_state = (self.successor)(&self.state);
mem::replace(&mut self.state, next_state)
}
}
pub struct Cycle<I> {
pub(super) orig: I,
pub(super) iter: I,
}
impl<I> Signal for Cycle<I>
where
I: Iterator + Clone,
{
type Type = I::Item;
fn next(&mut self) -> Self::Type {
match self.iter.next() {
Some(item) => item,
None => {
self.iter = self.orig.clone();
self.iter
.next()
.expect("The iterator always returns `None`")
}
}
}
}