winit-async 0.4.0

An async runtime based on the winit event loop.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::tls::{runtime, try_runtime};
use async_local_channel::spsc;
use std::pin::{Pin, pin};
use std::task::{Context, Poll};
use winit::event::WindowEvent;
use winit::window::{Window, WindowId};

pub struct WindowEvents {
    window_id: WindowId,
    rx: spsc::Receiver<WindowEvent>,
}

impl WindowEvents {
    #[inline]
    pub const fn next_event(&mut self) -> WindowEventFuture<'_> {
        WindowEventFuture { inner: self }
    }
}

pub struct WindowEventFuture<'a> {
    inner: &'a WindowEvents,
}

impl Future for WindowEventFuture<'_> {
    type Output = Option<WindowEvent>;

    #[inline]
    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        pin!(self.inner.rx.recv()).poll(cx).map(Result::ok)
    }
}

impl Drop for WindowEvents {
    #[inline]
    fn drop(&mut self) {
        if let Some(mut runtime) = try_runtime() {
            runtime.window_event_receiver_dropped(self.window_id);
        }
    }
}

pub trait WindowExtAsync {
    fn events(&self) -> Option<WindowEvents>;
}

impl WindowExtAsync for Window {
    #[inline]
    fn events(&self) -> Option<WindowEvents> {
        let window_id = self.id();
        let rx = runtime().window_event_receiver(window_id)?;
        Some(WindowEvents { window_id, rx })
    }
}