use crate::{
fib::{Fiber, FiberState, RootFiber},
thr::prelude::*,
};
use core::{
ops::{Generator, GeneratorState},
pin::Pin,
};
pub struct FiberGen<G>(G)
where
G: Generator;
impl<G> Fiber for FiberGen<G>
where
G: Generator,
{
type Input = ();
type Return = G::Return;
type Yield = G::Yield;
#[inline]
fn resume(self: Pin<&mut Self>, (): ()) -> FiberState<G::Yield, G::Return> {
let gen = unsafe { self.map_unchecked_mut(|x| &mut x.0) };
gen.resume(()).into()
}
}
impl<G> RootFiber for FiberGen<G>
where
G: Generator<Yield = (), Return = ()>,
G: 'static,
{
#[inline]
fn advance(self: Pin<&mut Self>) -> bool {
match self.resume(()) {
FiberState::Yielded(()) => false,
FiberState::Complete(()) => true,
}
}
}
impl<Y, R> From<GeneratorState<Y, R>> for FiberState<Y, R> {
#[inline]
fn from(state: GeneratorState<Y, R>) -> Self {
match state {
GeneratorState::Yielded(val) => FiberState::Yielded(val),
GeneratorState::Complete(val) => FiberState::Complete(val),
}
}
}
#[inline]
pub fn new<G>(gen: G) -> FiberGen<G>
where
G: Generator,
{
FiberGen(gen)
}
pub trait ThrFiberGen: ThrToken {
#[inline]
fn add<G>(self, gen: G)
where
G: Generator<Yield = (), Return = ()>,
G: Send + 'static,
{
self.add_fib(new(gen))
}
#[inline]
fn add_factory<C, G>(self, factory: C)
where
C: FnOnce() -> G + Send + 'static,
G: Generator<Yield = (), Return = ()>,
G: 'static,
{
self.add_fib_factory(|| new(factory()))
}
}
impl<T: ThrToken> ThrFiberGen for T {}