NaxdyOrg/embassy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add signal reset()

Browse source
This commit is contained in:
Dario Nieuwenhuis 2020-11-08 18:41:22 +01:00
parent def225b982
commit a2735a716c
1 changed files with 28 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
embassy/src/util/signal.rs
View file

@ -26,15 +26,20 @@ impl<T: Send> Signal<T> {
}
pub fn signal(&self, val: T) {
unsafe {
cortex_m::interrupt::free(|_| {
let state = &mut *self.state.get();
match mem::replace(state, State::Signaled(val)) {
State::Waiting(waker) => waker.wake(),
_ => {}
}
})
}
cortex_m::interrupt::free(|_| unsafe {
let state = &mut *self.state.get();
match mem::replace(state, State::Signaled(val)) {
State::Waiting(waker) => waker.wake(),
_ => {}
}
})
}
pub fn reset(&self) {
cortex_m::interrupt::free(|_| unsafe {
let state = &mut *self.state.get();
*state = State::None
})
}
pub fn wait<'a>(&'a self) -> impl Future<Output = T> + 'a {
@ -50,22 +55,20 @@ impl<'a, T: Send> Future for WaitFuture<'a, T> {
type Output = T;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<T> {
unsafe {
cortex_m::interrupt::free(|_| {
let state = &mut *self.signal.state.get();
match state {
State::None => {
*state = State::Waiting(cx.waker().clone());
Poll::Pending
}
State::Waiting(w) if w.will_wake(cx.waker()) => Poll::Pending,
State::Waiting(_) => panic!("waker overflow"),
State::Signaled(_) => match mem::replace(state, State::None) {
State::Signaled(res) => Poll::Ready(res),
_ => unreachable!(),
},
cortex_m::interrupt::free(|_| unsafe {
let state = &mut *self.signal.state.get();
match state {
State::None => {
*state = State::Waiting(cx.waker().clone());
Poll::Pending
}
})
}
State::Waiting(w) if w.will_wake(cx.waker()) => Poll::Pending,
State::Waiting(_) => panic!("waker overflow"),
State::Signaled(_) => match mem::replace(state, State::None) {
State::Signaled(res) => Poll::Ready(res),
_ => unreachable!(),
},
}
})
}
}