Add signal reset()
This commit is contained in:
parent
def225b982
commit
a2735a716c
1 changed files with 28 additions and 25 deletions
|
@ -26,8 +26,7 @@ impl<T: Send> Signal<T> {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn signal(&self, val: T) {
|
pub fn signal(&self, val: T) {
|
||||||
unsafe {
|
cortex_m::interrupt::free(|_| unsafe {
|
||||||
cortex_m::interrupt::free(|_| {
|
|
||||||
let state = &mut *self.state.get();
|
let state = &mut *self.state.get();
|
||||||
match mem::replace(state, State::Signaled(val)) {
|
match mem::replace(state, State::Signaled(val)) {
|
||||||
State::Waiting(waker) => waker.wake(),
|
State::Waiting(waker) => waker.wake(),
|
||||||
|
@ -35,6 +34,12 @@ impl<T: Send> Signal<T> {
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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 {
|
pub fn wait<'a>(&'a self) -> impl Future<Output = T> + 'a {
|
||||||
|
@ -50,8 +55,7 @@ impl<'a, T: Send> Future for WaitFuture<'a, T> {
|
||||||
type Output = T;
|
type Output = T;
|
||||||
|
|
||||||
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<T> {
|
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<T> {
|
||||||
unsafe {
|
cortex_m::interrupt::free(|_| unsafe {
|
||||||
cortex_m::interrupt::free(|_| {
|
|
||||||
let state = &mut *self.signal.state.get();
|
let state = &mut *self.signal.state.get();
|
||||||
match state {
|
match state {
|
||||||
State::None => {
|
State::None => {
|
||||||
|
@ -68,4 +72,3 @@ impl<'a, T: Send> Future for WaitFuture<'a, T> {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in a new issue