/* Copyright 2021 dogspace * * 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 . */ #include QMK_KEYBOARD_H enum layer_names { _LAY0, _LAY1, _LAY2, _LAY3 }; const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = { [_LAY0] = LAYOUT( KC_PSLS, KC_PAST, KC_PMNS, KC_P7, KC_P8, KC_P9, KC_PPLS, KC_P4, KC_P5, KC_P6, KC_PPLS, KC_P1, KC_P2, KC_P3, KC_PENT, KC_P0, KC_P0, KC_PDOT, KC_PENT ), [_LAY1] = LAYOUT( _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______ ), [_LAY2] = LAYOUT( _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______ ), [_LAY3] = LAYOUT( _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______, _______ ) }; #ifdef ENCODER_MAP_ENABLE const uint16_t PROGMEM encoder_map[][NUM_ENCODERS][NUM_DIRECTIONS] = { [0] = { ENCODER_CCW_CW(KC_VOLD, KC_VOLU), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), }, [1] = { ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), }, [2] = { ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), }, [3] = { ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), ENCODER_CCW_CW(_______, _______), }, }; #endif #ifdef OLED_ENABLE /*=========================================== OLED CONFIGURATION ===========================================*/ #define OLED_ROTATE true // OLED rotation (flip 180* from default orientation) #define GRAPH_DIRECTION true // Graph movement (true = right to left, false = left to right) #define GRAPH_TOP_WPM 100.0 // Minimum WPM required to reach the top of the graph #define GRAPH_REFRESH 1000 // In milliseconds, determines the graph-line frequency #define ICON_MED_WPM 10 // WPM required to display the medium snail #define ICON_FAST_WPM 25 // WPM required to display the fast snail // Layer names: Should be exactly 5 characters in length if vertical display, or 6 characters if horizontal #define MA_LAYER_NAME "LAY 0" // Layer _MA name #define L1_LAYER_NAME "LAY 1" // Layer _L1 name #define L2_LAYER_NAME "LAY 2" // Layer _L2 name #define L3_LAYER_NAME "LAY 3" // Layer _L3 name #define CAPLCK_STR "CAPLK" // Caps Lock string #define NUMLCK_STR "NUMLK" // Num Lock string #define SCRLK_STR "SCRLK" // Scroll Lock string #define EMPTY_STR " " // Empty string /*================================================================================================================*/ typedef struct oled_params { bool first_loop : 1; uint8_t wpm_icon : 7; uint16_t timer; uint8_t wpm_limit; uint8_t max_wpm; uint8_t graph_lines[32]; } oled_params; oled_params oled_data; void oled_init_data(void) { // Initialize oled params oled_data.first_loop = true; oled_data.wpm_icon = 5; oled_data.timer = 0; oled_data.wpm_limit = 20; oled_data.max_wpm = 0; for (int i=0; i<32; i++) { oled_data.graph_lines[i] = 0; } } // Set OLED rotation oled_rotation_t oled_init_user(oled_rotation_t rotation) { oled_init_data(); return OLED_ROTATE ? OLED_ROTATION_270 : OLED_ROTATION_90; } // Draw static background image to OLED (keyboard with no bottom row) static void render_background(void) { static const char PROGMEM nullbits_n_oled[] = { 0x00, 0xe0, 0xf0, 0xf0, 0xf8, 0xf8, 0xf0, 0xf0, 0xe0, 0x80, 0x20, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xf0, 0xe0, 0xe0, 0xe0, 0xc0, 0x80, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x1f, 0x1f, 0x1f, 0x1f, 0x1f, 0x1f, 0x1f, 0x1f, 0x1f, 0x3f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xfe, 0xfc, 0xf0, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xfe, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x00, 0x07, 0x0f, 0x0f, 0x0f, 0x0f, 0x0f, 0x0f, 0x07, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07, 0x0f, 0x0f, 0x0f, 0x0f, 0x0f, 0x0f, 0x07, 0x03, 0x00, }; oled_write_raw_P(nullbits_n_oled, sizeof(nullbits_n_oled)); } // Toggles pixel on/off, converts horizontal coordinates to vertical equivalent if necessary static void write_pixel(uint8_t x, uint8_t y, bool onoff) { oled_write_pixel(y, 127 - x, onoff); } // Write active layer name static void render_layer_state(void) { oled_set_cursor(0, 15); switch (get_highest_layer(layer_state)) { case _LAY0: oled_write_P(PSTR(MA_LAYER_NAME), false); break; case _LAY1: oled_write_P(PSTR(L1_LAYER_NAME), false); break; case _LAY2: oled_write_P(PSTR(L2_LAYER_NAME), false); break; case _LAY3: oled_write_P(PSTR(L3_LAYER_NAME), false); break; default: oled_write("ERROR", false); break; } } // Update WPM counters static void render_wpm_counters(uint8_t current_wpm) { uint8_t cursorposition_cur = 13; uint8_t cursorposition_max = 14; oled_set_cursor(0, cursorposition_cur); oled_write(get_u8_str(current_wpm, '0'), false); if (current_wpm > oled_data.max_wpm) { oled_data.max_wpm = current_wpm; oled_data.wpm_limit = oled_data.max_wpm + 20; oled_set_cursor(0, cursorposition_max); oled_write(get_u8_str(current_wpm, '0'), false); } } static void render_led_status(void) { // Host Keyboard LED Status led_t led_state = host_keyboard_led_state(); oled_set_cursor(0, 8); oled_write_P(led_state.caps_lock ? PSTR(CAPLCK_STR) : PSTR(EMPTY_STR), false); oled_set_cursor(0, 9); oled_write_P(led_state.num_lock ? PSTR(NUMLCK_STR) : PSTR(EMPTY_STR), false); oled_set_cursor(0, 10); oled_write_P(led_state.scroll_lock ? PSTR(SCRLK_STR) : PSTR(EMPTY_STR), false); } // Update WPM snail icon static void render_wpm_icon(uint8_t current_wpm) { // wpm_icon is used to prevent unnecessary redraw if ((current_wpm < ICON_MED_WPM) && (oled_data.wpm_icon != 0)) { oled_data.wpm_icon = 0; } else if ((current_wpm >= ICON_MED_WPM) && (current_wpm < ICON_FAST_WPM) && (oled_data.wpm_icon != 1)) { oled_data.wpm_icon = 1; } else if ((current_wpm >= ICON_FAST_WPM) && (oled_data.wpm_icon != 2)) { oled_data.wpm_icon = 2; } else { return; } static const char PROGMEM snails[][2][24] = { {{0x00, 0x00, 0x00, 0x00, 0x00, 0x80, 0x40, 0x20, 0xA0, 0x20, 0x40, 0x40, 0x80, 0x00, 0x00, 0x00, 0x80, 0x40, 0x20, 0x50, 0x88, 0x04, 0x00, 0x00}, {0x40, 0x60, 0x50, 0x4E, 0x51, 0x64, 0x4A, 0x51, 0x54, 0x49, 0x41, 0x62, 0x54, 0x49, 0x46, 0x41, 0x40, 0x30, 0x09, 0x04, 0x02, 0x01, 0x00, 0x00}}, {{0x00, 0x00, 0x00, 0x00, 0x00, 0x80, 0x40, 0x40, 0x40, 0x40, 0x40, 0x80, 0x80, 0x00, 0x00, 0x00, 0x04, 0x98, 0x60, 0x80, 0x00, 0x00, 0x00, 0x00}, {0x60, 0x50, 0x54, 0x4A, 0x51, 0x64, 0x4A, 0x51, 0x55, 0x49, 0x41, 0x62, 0x54, 0x49, 0x46, 0x41, 0x21, 0x10, 0x0A, 0x08, 0x05, 0x02, 0x00, 0x00}}, {{0x00, 0x00, 0x00, 0x00, 0x00, 0x80, 0x40, 0x40, 0x40, 0x40, 0x80, 0x80, 0x10, 0x10, 0x10, 0x20, 0x40, 0x40, 0xC0, 0x80, 0x80, 0x00, 0x00, 0x00}, {0x60, 0x58, 0x54, 0x62, 0x49, 0x54, 0x52, 0x51, 0x55, 0x49, 0x62, 0x52, 0x4D, 0x45, 0x46, 0x22, 0x21, 0x11, 0x10, 0x0A, 0x08, 0x05, 0x02, 0x00}} }; oled_set_cursor(0, 11); oled_write_raw_P(snails[oled_data.wpm_icon][0], sizeof(snails[oled_data.wpm_icon][0])); oled_set_cursor(0, 12); oled_write_raw_P(snails[oled_data.wpm_icon][1], sizeof(snails[oled_data.wpm_icon][1])); } // Update WPM graph static void render_wpm_graph(uint8_t current_wpm) { uint8_t line_height = ((current_wpm / GRAPH_TOP_WPM) * 7); if (line_height > 7) { line_height = 7; } // Count graph line pixels, return if nothing to draw uint8_t pixel_count = line_height; for (int i = 0; i < 31; i++) { pixel_count += oled_data.graph_lines[i]; } if (pixel_count == 0) { return; } // Shift array elements left or right depending on GRAPH_DIRECTION pend new graph line if (GRAPH_DIRECTION) { for (int i = 0; i < 31; i++) { oled_data.graph_lines[i] = oled_data.graph_lines[i + 1]; } oled_data.graph_lines[31] = line_height; } else { for (int i = 31; i > 0; i--) { oled_data.graph_lines[i] = oled_data.graph_lines[i - 1]; } oled_data.graph_lines[0] = line_height; } // Draw all graph lines (left to right, bottom to top) uint16_t draw_count, arrpos; for (int x = 1; x <= 63; x += 2) { arrpos = x / 2; draw_count = oled_data.graph_lines[arrpos]; for (int y = 31; y >= 25; y--) { if (draw_count > 0) { write_pixel(x, y, true); draw_count--; } else { write_pixel(x, y, false); } } } } // Call OLED functions bool oled_task_user(void) { // Draw OLED keyboard, prevent redraw if (oled_data.first_loop) { render_background(); oled_data.first_loop = false; } // Get current WPM, subtract 25% for accuracy and prevent large jumps caused by simultaneous keypresses uint8_t current_wpm = get_current_wpm(); // Write active layer name to display render_layer_state(); // Update WPM counters render_wpm_counters(current_wpm); // Update WPM snail icon render_wpm_icon(current_wpm); // Update LED status render_led_status(); // Update WPM graph every graph_refresh milliseconds if (timer_elapsed(oled_data.timer) > GRAPH_REFRESH) { render_wpm_graph(current_wpm); oled_data.timer = timer_read(); } return false; } #endif bool wpm_keycode_user(uint16_t keycode) { // Count all keycodes on the macropad return true; }