porcel8/src/device/keyboard.rs

use std::sync::mpsc::TryRecvError;
use crate::util::{EmulatorError, EmulatorResult};


/// An emulated keyboard that receives keyboard data as input
pub struct Keyboard {
    /// Current keyboard state
    bitflags: u16,
    /// Receives keyboard events from main thread
    keyboard_event_receiver: std::sync::mpsc::Receiver<KeyboardEvent>,
}

#[derive(Clone, Debug, Eq, PartialEq)]
pub enum KeyboardEvent {
    KeyUp(u8),
    KeyDown(u8),
}

impl Keyboard {
    pub fn new(keyboard_event_receiver: std::sync::mpsc::Receiver<KeyboardEvent>) -> Keyboard {
        Keyboard {
            bitflags: 0,
            keyboard_event_receiver,
        }
    }

    /// Update keyboard based on pending keyboard events.
    /// If no events are presents, it will return without any action.
    pub fn update_keyboard(&mut self) -> EmulatorResult<()> {
        loop {
            let keyboard_event_recv_res = self.keyboard_event_receiver.try_recv();
            match keyboard_event_recv_res {
                Ok(event) => {
                    log::debug!("Processing {:?}",event);
                    self.update_keyboard_state(event);
                }
                Err(TryRecvError::Empty) => {
                    break Ok(());
                }
                Err(TryRecvError::Disconnected) => {
                    break Err(EmulatorError::IOError("Keyboard updater disconnected".into()));
                }
            }
        }
    }

    /// Query if key is down
    pub fn query_key_down(&self, key_num: u8) -> bool {
        (self.bitflags | 1 << key_num) == (1 << key_num)
    }

    fn update_keyboard_state(&mut self, keyboard_event: KeyboardEvent) {
        match keyboard_event {
            KeyboardEvent::KeyUp(key) => {
                self.bitflags &= !((1 << key) as u16);
            }
            KeyboardEvent::KeyDown(key) => {
                self.bitflags |= 1 << key;
            }
        }
    }
}