From 1a3064afb186328c17c679094b0f92cbd46844b5 Mon Sep 17 00:00:00 2001
From: Gregorio <38576492+ohchiko@users.noreply.github.com>
Date: Mon, 31 May 2021 11:04:26 +0700
Subject: [Keyboard] Add Rubi Numpad (#12283)

Co-authored-by: Drashna Jaelre <drashna@live.com>
Co-authored-by: Ryan <fauxpark@gmail.com>
---
 keyboards/rubi/lib/oled.c | 268 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 268 insertions(+)
 create mode 100644 keyboards/rubi/lib/oled.c

(limited to 'keyboards/rubi/lib/oled.c')

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;
+}
-- 
cgit v1.2.3