rmk 0.1.3

Keyboard firmware written in Rust
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
54
55
56
57
58
59
60
61
62
63
64

use crate::{
    action::KeyAction,
    via::keycode_convert::{from_via_keycode, to_via_keycode},
};
use byteorder::{BigEndian, ByteOrder};
use embedded_storage::nor_flash::NorFlash;

use super::{eeconfig::DYNAMIC_KEYMAP_ADDR, Eeprom};

impl<F: NorFlash, const EEPROM_SIZE: usize> Eeprom<F, EEPROM_SIZE> {
    pub(crate) fn set_keymap<const ROW: usize, const COL: usize, const NUM_LAYER: usize>(
        &mut self,
        keymap: &[[[KeyAction; COL]; ROW]; NUM_LAYER],
    ) {
        keymap
            .iter()
            .flatten()
            .flatten()
            .enumerate()
            .for_each(|(i, action)| {
                // 2-byte value, relative addr should be i*2
                let addr = DYNAMIC_KEYMAP_ADDR + (i * 2) as u16;
                let mut buf: [u8; 2] = [0xFF; 2];
                BigEndian::write_u16(&mut buf, to_via_keycode(action.clone()));
                self.write_byte(addr, &buf);
            });
    }

    pub(crate) fn set_keymap_action(
        &mut self,
        row: usize,
        col: usize,
        layer: usize,
        action: KeyAction,
    ) {
        let addr = self.get_keymap_addr(row, col, layer);
        let mut buf: [u8; 2] = [0xFF; 2];
        BigEndian::write_u16(&mut buf, to_via_keycode(action));
        self.write_byte(addr, &buf);
    }

    pub(crate) fn read_keymap<const ROW: usize, const COL: usize, const NUM_LAYER: usize>(
        &self,
        keymap: &mut [[[KeyAction; COL]; ROW]; NUM_LAYER],
    ) {
        for (layer, layer_data) in keymap.iter_mut().enumerate() {
            for (row, row_data) in layer_data.iter_mut().enumerate() {
                for (col, value) in row_data.iter_mut().enumerate() {
                    let addr = self.get_keymap_addr(row, col, layer);
                    let data = self.read_byte(addr, 2);
                    *value = from_via_keycode(BigEndian::read_u16(data));
                }
            }
        }
    }

    fn get_keymap_addr(&self, row: usize, col: usize, layer: usize) -> u16 {
        DYNAMIC_KEYMAP_ADDR
            + ((self.keymap_config.col * self.keymap_config.row * layer
                + self.keymap_config.col * row
                + col)
                * 2) as u16
    }
}