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
54
55
56
57
58
59

//! Window-based inputs captured during event loop processing.

mod key_buf;
mod key_map;
mod mouse_buf;

use winit::event::Event;

pub use self::{
    key_buf::KeyBuf,
    key_map::KeyMap,
    mouse_buf::{MouseBuf, MouseButton},
};

/// Handles keyboard and mouse `Event`s while updating the provided buffers.
pub fn update_input<'a>(
    keyboard: &'a mut KeyBuf,
    mouse: &'a mut MouseBuf,
    events: impl IntoIterator<Item = &'a Event<()>>,
) {
    update_input_opt(Some(keyboard), Some(mouse), events);
}

/// Handles keyboard `Event`s while updating the provided buffer.
pub fn update_keyboard<'a>(
    keyboard: &'a mut KeyBuf,
    events: impl IntoIterator<Item = &'a Event<()>>,
) {
    update_input_opt(Some(keyboard), None, events);
}

/// Handles mouse `Event`s while updating the provided buffer.
pub fn update_mouse<'a>(mouse: &'a mut MouseBuf, events: impl IntoIterator<Item = &'a Event<()>>) {
    update_input_opt(None, Some(mouse), events);
}

fn update_input_opt<'a>(
    mut keyboard: Option<&'a mut KeyBuf>,
    mut mouse: Option<&'a mut MouseBuf>,
    events: impl IntoIterator<Item = &'a Event<()>>,
) {
    if let Some(keyboard) = keyboard.as_mut() {
        keyboard.update();
    }

    if let Some(mouse) = mouse.as_mut() {
        mouse.update();
    }

    for event in events.into_iter() {
        if let Some(keyboard) = keyboard.as_mut() {
            keyboard.handle_event(event);
        }

        if let Some(mouse) = mouse.as_mut() {
            mouse.handle_event(event);
        }
    }
}