use super::{Error, Event, Instance};
use crate::{mcu::peripherals::I2C, pac::interrupt};
use core::{future::Future, marker::PhantomData, task::Poll};
use embassy_sync::waitqueue::AtomicWaker;
use enumset::EnumSet;
static WAKER: [AtomicWaker; 1] = [AtomicWaker::new(); 1];
pub struct EventFuture<T: Instance> {
_t: PhantomData<T>,
events: EnumSet<Event>,
}
impl<T: Instance> EventFuture<T> {
pub fn new(events: EnumSet<Event>) -> Self {
events.iter().for_each(|event| T::event_config(event, true));
Self {
_t: PhantomData,
events,
}
}
#[inline]
unsafe fn on_interrupt() {
EnumSet::all().iter().for_each(|event| {
if T::is_event_match(event) {
T::event_clear(event);
T::event_config(event, false);
}
});
WAKER[T::id() as usize].wake()
}
}
impl<T: Instance> Future for EventFuture<T> {
type Output = Result<(), Error>;
fn poll(
self: core::pin::Pin<&mut Self>,
cx: &mut core::task::Context<'_>,
) -> core::task::Poll<Self::Output> {
WAKER[T::id() as usize].register(cx.waker());
for event in self.events {
if T::event_flag(event) {
T::event_clear(event);
return Poll::Ready(Ok(()));
}
}
Poll::Pending
}
}
#[interrupt]
fn I2C1() {
critical_section::with(|_cs| unsafe { EventFuture::<I2C>::on_interrupt() })
}