New branch for keymap-in-eeprom feature

common.mk

@@ -64,6 +64,11 @@ ifdef KEYMAP_SECTION_ENABLE
     EXTRALDFLAGS = -Wl,-L$(TOP_DIR),-Tldscript_keymap_avr5.x
 endif
 
+ifdef KEYMAP_IN_EEPROM_ENABLE
+    SRC += $(COMMON_DIR)/keymap_in_eeprom.c
+    OPT_DEFS += -DKEYMAP_IN_EEPROM_ENABLE
+endif
+
 # Version string
 OPT_DEFS += -DVERSION=$(shell (git describe --always --dirty || echo 'unknown') 2> /dev/null)
 

common/bootmagic.c

@@ -30,6 +30,7 @@ void bootmagic(void)
 
     /* eeconfig clear */
     if (bootmagic_scan_keycode(BOOTMAGIC_KEY_EEPROM_CLEAR)) {
+        eeconfig_disable();
         eeconfig_init();
     }
 

common/eeconfig.c

@@ -2,6 +2,7 @@
 #include <stdbool.h>
 #include <avr/eeprom.h>
 #include "eeconfig.h"
+#include "keymap_in_eeprom.h"
 
 void eeconfig_init(void)
 {
@@ -13,6 +14,9 @@ void eeconfig_init(void)
 #ifdef BACKLIGHT_ENABLE
     eeprom_write_byte(EECONFIG_BACKLIGHT,      0);
 #endif
+#ifdef KEYMAP_IN_EEPROM_ENABLE
+    keymap_in_eeprom_init();
+#endif
 }
 
 void eeconfig_enable(void)
@@ -23,6 +27,9 @@ void eeconfig_enable(void)
 void eeconfig_disable(void)
 {
     eeprom_write_word(EECONFIG_MAGIC, 0xFFFF);
+#ifdef KEYMAP_IN_EEPROM_ENABLE
+    keymap_in_eeprom_disable();
+#endif
 }
 
 bool eeconfig_is_enabled(void)

common/keyboard.c

@@ -31,6 +31,7 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #include "bootmagic.h"
 #include "eeconfig.h"
 #include "backlight.h"
+#include "keymap_in_eeprom.h"
 #ifdef MOUSEKEY_ENABLE
 #   include "mousekey.h"
 #endif
@@ -72,6 +73,10 @@ void keyboard_init(void)
 #ifdef BACKLIGHT_ENABLE
     backlight_init();
 #endif
+
+#ifdef KEYMAP_IN_EEPROM_ENABLE
+    keymap_in_eeprom_init();
+#endif
 }
 
 /*

common/keymap_in_eeprom.c

@@ -0,0 +1,108 @@
+/*
+Copyright 2013,2014 Kai Ryu <kai1103@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 2 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <avr/pgmspace.h>
+#include <avr/eeprom.h>
+#include "eeconfig.h"
+#include "keymap_in_eeprom.h"
+#include "debug.h"
+
+#ifdef KEYMAP_IN_EEPROM_ENABLE
+
+void keymap_in_eeprom_init(void) {
+    if (!check_keymap_in_eeprom()) {
+        write_keymap_to_eeprom();
+    }
+}
+
+void keymap_in_eeprom_disable(void) {
+    eeprom_write_word(EECONFIG_KEYMAP_CHECKSUM, eeprom_read_word(EECONFIG_KEYMAP_CHECKSUM) + 1);
+}
+
+bool check_keymap_in_eeprom(void) {
+    uint16_t checksum_in_eeprom = eeprom_read_word(EECONFIG_KEYMAP_CHECKSUM);
+    uint16_t checksum = EECONFIG_MAGIC_NUMBER;
+    for (uint16_t i = 0; i < KEYMAP_WORD_SIZE; i++) {
+        checksum += eeprom_read_word(EECONFIG_KEYMAP_FN_ACTIONS + i);
+    }
+#ifdef DEBUG
+    eeprom_write_word(EECONFIG_KEYMAP_DEBUG, checksum);
+#endif
+    return (checksum_in_eeprom == checksum);
+}
+
+void write_keymap_to_eeprom(void) {
+    uint16_t checksum = EECONFIG_MAGIC_NUMBER;
+    const uint16_t *fn_actions = fn_actions_pointer();
+    const uint8_t *keymaps = keymaps_pointer();
+    // write fn_actions
+    if (fn_actions != NULL) {
+        uint16_t fn_actions_count_in_flash = fn_actions_count();
+        for (uint16_t i = 0; i < FN_ACTIONS_COUNT; i++) {
+            uint16_t fn_action = 0;
+            if (i < fn_actions_count_in_flash) {
+                fn_action = pgm_read_word(fn_actions + i);
+            }
+            eeconfig_write_keymap_fn_action(i, fn_action);
+            checksum += fn_action;
+        }
+    }
+    // write keymaps
+    if (keymaps != NULL) {
+        uint16_t keys_count_in_flash = keys_count();
+        for (uint16_t i = 0; i < KEYS_COUNT; i++) {
+            uint8_t keymap = 0;
+            if (i < keys_count_in_flash) {
+                keymap = pgm_read_byte(keymaps + i);
+            }
+            eeconfig_write_keymap_key_by_index(i, keymap);
+            uint16_t keymap_word = keymap;
+            if (i & 1) {
+                keymap_word = keymap << 8;
+            }
+            checksum += keymap_word;
+        }
+    }
+    // write checksum
+    eeprom_write_word(EECONFIG_KEYMAP_CHECKSUM, checksum);
+}
+
+uint16_t eeconfig_read_keymap_fn_action(uint8_t index) {
+    return eeprom_read_word(EECONFIG_KEYMAP_FN_ACTIONS + index);
+}
+
+void eeconfig_write_keymap_fn_action(uint8_t index, uint16_t fn_action) {
+    return eeprom_write_word(EECONFIG_KEYMAP_FN_ACTIONS + index, fn_action);
+}
+
+uint8_t eeconfig_read_keymap_key(uint8_t layer, uint8_t row, uint8_t col) {
+    return eeprom_read_byte(EECONFIG_KEYMAP_KEYMAPS + KEY_OFFSET(layer, row, col));
+}
+
+void eeconfig_write_keymap_key(uint8_t layer, uint8_t row, uint8_t col, uint8_t key) {
+    return eeprom_write_byte(EECONFIG_KEYMAP_KEYMAPS + KEY_OFFSET(layer, row, col), key);
+}
+
+uint8_t eeconfig_read_keymap_key_by_index(uint16_t index) {
+    return eeprom_read_byte(EECONFIG_KEYMAP_KEYMAPS + index);
+}
+
+void eeconfig_write_keymap_key_by_index(uint16_t index, uint8_t key) {
+    return eeprom_write_byte(EECONFIG_KEYMAP_KEYMAPS + index, key);
+}
+
+#endif

common/keymap_in_eeprom.h

@@ -0,0 +1,68 @@
+/*
+Copyright 2013,2014 Kai Ryu <kai1103@gmail.com>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 2 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef KEYMAP_IN_EEPROM_H
+#define KEYMAP_IN_EEPROM_H
+
+#ifdef KEYMAP_IN_EEPROM_ENABLE
+
+#include <stdint.h>
+#include <stdbool.h>
+
+#define EECONFIG_KEYMAP_IN_EEPROM 0x10
+#ifndef FN_ACTIONS_COUNT
+#define FN_ACTIONS_COUNT 32
+#endif
+#ifndef KEYMAPS_COUNT
+#define KEYMAPS_COUNT 1
+#endif
+#define KEYS_COUNT (KEYMAPS_COUNT * MATRIX_ROWS * MATRIX_COLS)
+
+typedef struct {
+    uint16_t checksum;
+    uint16_t fn_actions[FN_ACTIONS_COUNT];
+    uint8_t keymaps[KEYMAPS_COUNT][MATRIX_ROWS][MATRIX_COLS];
+} keymap_in_eeprom_t;
+
+#define EECONFIG_KEYMAP_CHECKSUM   (uint16_t *)(EECONFIG_KEYMAP_IN_EEPROM)
+#define EECONFIG_KEYMAP_FN_ACTIONS (uint16_t *)(EECONFIG_KEYMAP_CHECKSUM + 1)
+#define EECONFIG_KEYMAP_KEYMAPS    (uint8_t *)(EECONFIG_KEYMAP_FN_ACTIONS + FN_ACTIONS_COUNT)
+#define EECONFIG_KEYMAP_DEBUG      (uint16_t *)(EECONFIG_KEYMAP_KEYMAPS + KEYS_COUNT)
+
+#define KEYMAP_SIZE (sizeof(uint16_t) * FN_ACTIONS_COUNT + sizeof(uint8_t) * KEYS_COUNT)
+#define KEYMAP_WORD_SIZE ((KEYMAP_SIZE + 1) / 2)
+#define KEY_OFFSET(layer, row, col) (layer * MATRIX_ROWS * MATRIX_COLS + row * MATRIX_COLS + col)
+
+void keymap_in_eeprom_init(void);
+void keymap_in_eeprom_disable(void);
+bool check_keymap_in_eeprom(void);
+void write_keymap_to_eeprom(void);
+uint8_t eeconfig_read_keymap_key(uint8_t layer, uint8_t row, uint8_t col);
+void eeconfig_write_keymap_key(uint8_t layer, uint8_t row, uint8_t col, uint8_t key);
+uint8_t eeconfig_read_keymap_key_by_index(uint16_t index);
+void eeconfig_write_keymap_key_by_index(uint16_t index, uint8_t key);
+uint16_t eeconfig_read_keymap_fn_action(uint8_t index);
+void eeconfig_write_keymap_fn_action(uint8_t index, uint16_t fn_action);
+
+const uint8_t* keymaps_pointer(void);
+const uint16_t* fn_actions_pointer(void);
+uint16_t keys_count(void);
+uint16_t fn_actions_count(void);
+
+#endif
+
+#endif