diff options
Diffstat (limited to 'keyboards/rubi/lib/oled.c')
| -rw-r--r-- | keyboards/rubi/lib/oled.c | 268 |
1 files changed, 268 insertions, 0 deletions
diff --git a/keyboards/rubi/lib/oled.c b/keyboards/rubi/lib/oled.c new file mode 100644 index 0000000000..d48d27323c --- /dev/null +++ b/keyboards/rubi/lib/oled.c @@ -0,0 +1,268 @@ +/* +Copyright 2021 gregorio + +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 QMK_KEYBOARD_H +#include "./lib/oled.h" + +bool process_record_user_oled(uint16_t keycode, keyrecord_t *record) { + return process_record_user(keycode, record); +} + +void change_oled_mode(void) { + oled_mode = (oled_mode + 1) % _NUM_OLED_MODES; +} + +void render_layer_section(void) { + // Layer indicators + static const char PROGMEM layer_0[] = {0xc8, 0xc9, 0}; + static const char PROGMEM layer_1[] = {0xca, 0xcb, 0}; + static const char PROGMEM layer_2[] = {0xcc, 0xcd, 0}; + static const char PROGMEM layer_3[] = {0xce, 0xcf, 0}; + + oled_set_cursor(oled_max_chars()-15, 0); + oled_write_P(PSTR("LAYER"), false); + switch (get_highest_layer(layer_state)) { + case 0: + oled_write_P(layer_0, false); + break; + case 1: + oled_write_P(layer_1, false); + break; + case 2: + oled_write_P(layer_2, false); + break; + case 3: + oled_write_P(layer_3, false); + break; + default: + oled_write_P(PSTR("? "), false); + break; + } +} + +void render_encoder_section(void) { + static const char PROGMEM enc_vol[] = {0x88, 0x89, 0x8a, 0x8b, 0}; + static const char PROGMEM enc_med[] = {0xa8, 0xa9, 0xaa, 0xab, 0}; + static const char PROGMEM enc_brt[] = {0x8c, 0x8d, 0x8e, 0x8f, 0}; + + oled_set_cursor(oled_max_chars()-7, 0); + oled_write_P(PSTR("ENC"), false); + switch (encoder_mode) { + default: + case ENC_MODE_VOLUME: + oled_write_P(enc_vol, false); + break; + case ENC_MODE_MEDIA: + oled_write_P(enc_med, false); + break; + case ENC_MODE_BRIGHTNESS: + oled_write_P(enc_brt, false); + break; + } +} + +void render_numlock_section(void) { + static const char PROGMEM num_on[] = {0x80, 0x81, 0x82, 0x83, 0}; + static const char PROGMEM num_off[] = {0xa0, 0xa1, 0xa2, 0xa3, 0}; + static const char PROGMEM cap_on[] = {0x84, 0x85, 0x86, 0x87, 0}; + static const char PROGMEM cap_off[] = {0xa4, 0xa5, 0xa6, 0xa7, 0}; + static const char PROGMEM scr_on[] = {0xba, 0xbb, 0xbc, 0xbd, 0}; + static const char PROGMEM scr_off[] = {0xda, 0xdb, 0xdc, 0xdd, 0}; + + led_t led_state = host_keyboard_led_state(); + + oled_set_cursor(oled_max_chars()-12, 3); + // num lock + oled_write_P(led_state.num_lock ? num_on : num_off, false); + oled_write_P(led_state.caps_lock ? cap_on : cap_off, false); + oled_write_P(led_state.scroll_lock ? scr_on : scr_off, false); +} + +void render_mode_section(void) { + static const char PROGMEM pad_on[] = {0xb0, 0xb1, 0xb2, 0xb3, 0}; + static const char PROGMEM pad_off[] = {0xc4, 0xc5, 0xc6, 0xc7, 0}; + static const char PROGMEM cal_on[] = {0x90, 0x91, 0x92, 0x93, 0}; + static const char PROGMEM cal_off[] = {0xc0, 0xc1, 0xc2, 0xc3, 0}; + + if (oled_mode == OLED_MODE_CALC) { + oled_set_cursor(0, 0); + oled_write_P(pad_off, false); + oled_set_cursor(0, 1); + oled_write_P(cal_on, false); + } else { + oled_set_cursor(0, 0); + oled_write_P(pad_on, false); + oled_set_cursor(0, 1); + oled_write_P(cal_off, false); + } +} + +void render_calc_section(void) { + static const char PROGMEM add_on[] = {0xd4, 0xd5, 0}; + static const char PROGMEM add_off[] = {0x94, 0x95, 0}; + static const char PROGMEM sub_on[] = {0xd6, 0xd7, 0}; + static const char PROGMEM sub_off[] = {0x96, 0x97, 0}; + static const char PROGMEM mul_on[] = {0xd8, 0xd9, 0}; + static const char PROGMEM mul_off[] = {0x9c, 0x9d, 0}; + static const char PROGMEM div_on[] = {0xb8, 0xb9, 0}; + static const char PROGMEM div_off[] = {0x9e, 0x9f, 0}; + static const char PROGMEM sqr_on[] = {0x98, 0x99, 0}; + static const char PROGMEM sqr_off[] = {0xbe, 0xbf, 0}; + static const char PROGMEM rec_on[] = {0x9a, 0x9b, 0}; + static const char PROGMEM rec_off[] = {0xde, 0xdf, 0}; + static const char PROGMEM neg_on[] = {0xb4, 0xb5, 0}; + static const char PROGMEM neg_off[] = {0xb6, 0xb7, 0}; + + + if (oled_mode == OLED_MODE_CALC) { + if (get_highest_layer(layer_state) == 1) { + oled_set_cursor(oled_max_chars()-8, 0); + + switch (calc_operator_display) { + case '+': + oled_write_P(div_off, false); + oled_write_P(mul_off, false); + oled_write_P(sub_off, false); + oled_write_P(add_on, false); + break; + case '-': + oled_write_P(div_off, false); + oled_write_P(mul_off, false); + oled_write_P(sub_on, false); + oled_write_P(add_off, false); + break; + case '*': + oled_write_P(div_off, false); + oled_write_P(mul_on, false); + oled_write_P(sub_off, false); + oled_write_P(add_off, false); + break; + case '/': + oled_write_P(div_on, false); + oled_write_P(mul_off, false); + oled_write_P(sub_off, false); + oled_write_P(add_off, false); + break; + case 's': + case 'r': + case 'n': + layer_on(2); + break; + default: + oled_write_P(div_off, false); + oled_write_P(mul_off, false); + oled_write_P(sub_off, false); + oled_write_P(add_off, false); + break; + } + } else if (get_highest_layer(layer_state) == 2) { + oled_set_cursor(oled_max_chars()-6, 0); + + switch (calc_operator_display) { + case '+': + case '-': + case '*': + case '/': + layer_off(2); + break; + case 's': + oled_write_P(neg_off, false); + oled_write_P(sqr_on, false); + oled_write_P(rec_off, false); + break; + case 'r': + oled_write_P(neg_off, false); + oled_write_P(sqr_off, false); + oled_write_P(rec_on, false); + break; + case 'n': + oled_write_P(neg_on, false); + oled_write_P(sqr_off, false); + oled_write_P(rec_off, false); + break; + default: + oled_write_P(neg_off, false); + oled_write_P(sqr_off, false); + oled_write_P(rec_off, false); + break; + } + } + + if (calc_display_lines == 1) { + oled_set_cursor(oled_max_chars()-strlen(calc_result_display)-2, 3); + oled_write_char(calc_operator_display, false); + } else if (calc_display_lines == 2) { + oled_set_cursor(oled_max_chars()-strlen(calc_status_display), 2); + oled_write(calc_status_display, false); + } + oled_set_cursor(oled_max_chars()-strlen(calc_result_display), 3); + oled_write(calc_result_display, false); + } +} + +static void render_logo(void) { + oled_write_raw_P(raw_logo, sizeof(raw_logo)); +} + +void render_frame(void) { + if (oled_logo_expired) { + if (oled_mode == OLED_MODE_DEFAULT) { + render_mode_section(); + render_layer_section(); + render_encoder_section(); + render_numlock_section(); + } else if (oled_mode == OLED_MODE_CALC) { + render_mode_section(); + render_calc_section(); + } else if (oled_mode == OLED_MODE_OFF) { + if (is_oled_on()) { + oled_off(); + } + } + } else { + render_logo(); + oled_logo_expired = timer_elapsed(oled_logo_timer) > OLED_LOGO_TIMEOUT; + } +} + +__attribute__((weak)) void oled_task_user(void) { + if (timer_elapsed(oled_frame_timer) > OLED_FRAME_TIMEOUT) { + oled_clear(); + oled_frame_timer = timer_read(); + render_frame(); + } + + if (get_highest_layer(layer_state) == 1) { + oled_mode = OLED_MODE_CALC; + } else if (get_highest_layer(layer_state) == 2) { + if (IS_LAYER_ON(1)) { + oled_mode = OLED_MODE_CALC; + } else { + oled_mode = OLED_MODE_DEFAULT; + } + } else if (get_highest_layer(layer_state) == 3) { + oled_mode = OLED_MODE_OFF; + } else { + oled_mode = OLED_MODE_DEFAULT; + } +} + +oled_rotation_t oled_init_user(oled_rotation_t rotation) { + oled_logo_timer = timer_read(); + oled_frame_timer = timer_read(); + return rotation; +} |