keychron_qmk_firmware/keyboards/crkbd/keymaps/mcrown/oled.c

356 lines
15 KiB
C

/** @file oled.h
* @brief mcrown oled service implementation.
*
* 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/>.
*
* @author Mario Corona (mariocc@comunidad.unam.mx) 2021
*
*/
#include QMK_KEYBOARD_H
#include <stdio.h>
#include <string.h>
#include "mcrown.h"
#include "oled.h"
#define ASCII_TABLE_LENGTH (0x80)
#define KEYLOG_STRING_STARTUP (KEYLOG_EOL_LEN+1)
#define ALT_CODE (0x7E)
#define SPECIAL_KEYS_SHIFT(kc) (0x18+(kc))
static void render_keylogger_status(void);
static char keylog_str[KEYLOG_EOL_LEN] = {' '};
static uint16_t log_timer = 0;
static uint8_t current_cursor_pos=0;
static uint32_t cursor_oled_timer = 0;
static uint32_t standby_oled_timer = 0;
static char last_c=' ';
/* Provides the ASCII value or the address of the character selected of the OLED font specified in glcfont.c */
static const char ascii_t[ASCII_TABLE_LENGTH] = {
/* 0 1 2 3 4 5 6 7 8 9 A B C D E F */
/* | | | | | | | | | | | | | | | | */
0x0F, 0x1A, 0x1B, 0x19, 0x18, 0x0E, ' ', ' ', 0x11, 0x1C, 0x97, ' ', ' ', ' ', ' ', ' ', /* 0 */
/* | | | | | | | | | | | | | | | | */
' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ', 0x1D, ' ', ' ', ' ', ' ', /* 1 */
/* | | | | | | | | | | | | | | | | */
0x16, '!', '"', '#', '$', '%', '&', '\'', '(', ')', '*', '+', ',', '-', '.', '/', /* 2 */
/* | | | | | | | | | | | | | | | | */
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', ':', ';', '<', '=', '>', '?', /* 3 */
/* | | | | | | | | | | | | | | | | */
'@', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', /* 4 */
/* | | | | | | | | | | | | | | | | */
'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '[', '\\', ']', '^', '_', /* 5 */
/* | | | | | | | | | | | | | | | | */
'`', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', /* 6 */
/* | | | | | | | | | | | | | | | | */
'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '{', '|', '}', '~', 0x7F, /* 7 */
};
/* This table is to remap and get the corresponding ASCII value based on the KEYCODE (taken as the index of the array) of quatum_keycodes.h module */
static const unsigned char code_to_ascii[ASCII_TABLE_LENGTH] = {
/* 0 1 2 3 4 5 6 7 8 9 A B C D E F */
/* | | | | | | | | | | | | | | | | */
0x00, 0x00, 0x00, 0x00, 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', /* 0 */
/* | | | | | | | | | | | | | | | | */
'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '1', '2', /* 1 */
/* | | | | | | | | | | | | | | | | */
'3', '4', '5', '6', '7', '8', '9', '0', 0x0A, 0x1B, 0x08, 0x09, ' ', '-', '=', '[', /* 2 */
/* | | | | | | | | | | | | | | | | */
']', '\\', 0x00, ';', '\'', '`', ',', '.', '/', 0x00, 0x00, 0x00, 0x00, 0x00, '!', '@', /* 3 */
/* | | | | | | | | | | | | | | | | */
'#', '$', '%', '^', '&', '*', '(', ')', 0x00, 0x00, 0x00, 0x00, 0x00, '_', '+', '{', /* 4 */
/* | | | | | | | | | | | | | | | | */
'}', '|', 0x00, 0x00, 0x00, '~', 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* 5 */
/* | | | | | | | | | | | | | | | | */
0x00, 0x00, 0x00, 0x00, 0x7F, 0x00, 0x00, 0x01, 0x02, 0x03, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, /* 6 */
/* | | | | | | | | | | | | | | | | */
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x05, 0x00, /* 7 */
};
/** @brief maps the keycode to get the ascii value.
*
* If any argument is invalid, the function has no effect.
*
* @param keycode value of the pressed key.
* @return ascii value of the pressed key or a special value for non-ascii keys.
*/
inline static char get_ascii(int16_t keycode){
uint8_t ascii_idx=0x00;
if(keycode<KC_F1){
ascii_idx=code_to_ascii[(uint8_t)keycode];
}else if(keycode<KC_KP_ENTER){
ascii_idx=code_to_ascii[SPECIAL_KEYS_SHIFT(keycode)];
}else if(KC_LNG1==keycode){
ascii_idx=code_to_ascii[ALT_CODE];
}else if( QK_LSFT==(QK_LSFT&keycode) ){
ascii_idx=code_to_ascii[RM_LSFT(keycode)];
}
return ascii_t[ascii_idx];
}
/** @brief detect retuns the rotation of the display based on the keyboard side.
*
* If any argument is invalid, the function has no effect.
*
* @param oled_rotation_t rotation
* @return rotation of the display.
*/
oled_rotation_t oled_init_user(oled_rotation_t rotation){
oled_rotation_t oled_rot=OLED_ROTATION_180;
if(true==is_keyboard_master()){
#ifdef OLED_VERTICAL
oled_rot=OLED_ROTATION_270;
#else
oled_rot=OLED_ROTATION_0;
#endif
}
return oled_rot;
}
/** @brief renders the keylog string and display it. This function also toggles the cursor.
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
static void render_keylogger_status(void){
static bool cursor_f=true;
if(timer_elapsed32(cursor_oled_timer) > 300){
cursor_oled_timer = timer_read32();
cursor_f=!cursor_f;
}
oled_write_P(PSTR("\n>:"), false);
if(current_cursor_pos>(KEYLOG_LEN-1)){
current_cursor_pos=0;
memset(keylog_str, ' ', sizeof(char)*KEYLOG_EOL_LEN);
/* Here the EOL is to clear with white spaces all the keylog area */
keylog_str[KEYLOG_EOL_LEN-1] = '\0';
oled_write(keylog_str, false);
/* Reset EOL to the begining of the keylog string */
keylog_str[0] = '\0';
}
oled_write(keylog_str, false);
oled_write_char(last_c, cursor_f);
}
/** @brief displays the current active layout.
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
void render_layout_state(void){
CUSTOM_LAYERS_T current_layer;
current_layer=(CUSTOM_LAYERS_T)get_highest_layer(layer_state);
#ifdef OLED_VERTICAL
oled_write_P(PSTR("Lyt:\n"), false);
#else
oled_write_P(PSTR("Layout: "), false);
#endif
switch (current_layer){
#ifdef OLED_VERTICAL
case _COLEMAK:
oled_write_P(PSTR("Clmak\n"), false);
break;
case _DVORAK:
oled_write_P(PSTR("Dvak\n"), false);
break;
case _QWERTY:
oled_write_P(PSTR("Qwty\n"), false);
break;
default:
oled_write_P(PSTR("Undf\n"), false);
break;
#else
case _COLEMAK:
oled_write_P(PSTR("Colemak\n"), false);
break;
case _DVORAK:
oled_write_P(PSTR("Dvorak\n"), false);
break;
case _QWERTY:
oled_write_P(PSTR("Qwerty\n"), false);
break;
default:
oled_write_P(PSTR("Undefined\n"), false);
break;
#endif
}
}
/** @brief displays the current active layer.
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
void render_layer_state(void){
CUSTOM_LAYERS_T current_layer;
current_layer=(CUSTOM_LAYERS_T)get_highest_layer(layer_state);
#ifdef OLED_VERTICAL
oled_write_P(PSTR("Lyr:\n"), false);
#else
oled_write_P(PSTR("Layer:"), false);
#endif
switch(current_layer){
#ifdef OLED_VERTICAL
case _LOWER:
oled_write_P(PSTR("Lwr\n"), true);
break;
case _RAISE:
oled_write_P(PSTR("Ris\n"), true);
break;
case _ADJUST:
oled_write_P(PSTR("Adj\n"), true);
break;
case _NUMPAD:
oled_write_P(PSTR("Num\n"), true);
break;
default:
oled_write_P(PSTR("Def\n"), false);
break;
#else
case _LOWER:
oled_write_P(PSTR(" Lower "), true);
break;
case _RAISE:
oled_write_P(PSTR(" Raise "), true);
break;
case _ADJUST:
oled_write_P(PSTR(" Adjust "), true);
break;
case _NUMPAD:
oled_write_P(PSTR(" Numpad "), true);
break;
default:
oled_write_P(PSTR(" Default "), false);
break;
#endif
}
}
/** @brief displays the current status of the main display/
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
void render_status(void){
render_layout_state();
oled_write_P(PSTR("\n"), false);
render_layer_state();
render_keylogger_status();
}
/** @brief renders the logo to be displayed.
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
static void render_logo(void){
static const char PROGMEM qmk_logo[] = {
0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89, 0x8A, 0x8B, 0x8C, 0x8D, 0x8E, 0x8F, 0x90, 0x91, 0x92, 0x93, 0x94,
0xA0, 0xA1, 0xA2, 0xA3, 0xA4, 0xA5, 0xA6, 0xA7, 0xA8, 0xA9, 0xAA, 0xAB, 0xAC, 0xAD, 0xAE, 0xAF, 0xB0, 0xB1, 0xB2, 0xB3, 0xB4,
0xC0, 0xC1, 0xC2, 0xC3, 0xC4, 0xC5, 0xC6, 0xC7, 0xC8, 0xC9, 0xCA, 0xCB, 0xCC, 0xCD, 0xCE, 0xCF, 0xD0, 0xD1, 0xD2, 0xD3, 0xD4,
0x00};
oled_write_P(qmk_logo, false);
}
/** @brief executes the actions for both displays.
*
* If any argument is invalid, the function has no effect.
*
* @param void.
* @return void.
*/
bool oled_task_user(void){
if (timer_elapsed32(standby_oled_timer) > 15000){
oled_off();
}else{
oled_on();
if(true==is_keyboard_master()){
render_status();
}else{
render_logo();
oled_write_P(PSTR("\n"), false);
oled_scroll_left();
}
}
return false;
}
/** @brief process the current key and add it to the keylog string.
*
* If any argument is invalid, the function has no effect.
*
* @param keycode pressed key.
* @return void.
*/
extern void add_keylog(uint16_t keycode){
if ((keycode >= QK_MOD_TAP && keycode <= QK_MOD_TAP_MAX) || (keycode >= QK_LAYER_TAP && keycode <= QK_LAYER_TAP_MAX)){
keycode&=0x00FF;
}
if(current_cursor_pos>(KEYLOG_LEN-1)||(current_cursor_pos>KEYLOG_STRING_STARTUP)){
current_cursor_pos=0;
last_c=get_ascii(keycode);
current_cursor_pos++;
}else{
if(keycode <= KC_TILD){
keylog_str[current_cursor_pos]=last_c;
last_c=get_ascii(keycode);
current_cursor_pos++;
}
keylog_str[current_cursor_pos] = '\0';
}
log_timer = timer_read();
standby_oled_timer = timer_read32();
}