// Copyright 2023 NaturalZh (@NaturalZh) // SPDX-License-Identifier: GPL-2.0-or-later #include "quantum.h" #define HC595_ST_PIN B1 //74HC595 storage register clock input #define HC595_SH_PIN B0 //74HC595 shift register clock input #define HC595_DS B3 // 74HC595 serial data input #include "led_hc595.c" uint8_t IND = 0; //buffer of LED Display int FN_ON = 0; _Bool WIN_LOCK = 0; #if defined(RGB_MATRIX_ENABLE) /*&& defined(CAPS_LOCK_INDEX)*/ const aw20216s_led_t PROGMEM g_aw20216s_leds[AW20216S_LED_COUNT] = { /* Each AW20216 channel is controlled by a register at some offset between 0x00 * and 0xD7 inclusive. * See drivers/awinic/aw20216.h for the mapping between register offsets and * driver pin locations. * driver * | R location * | | G location * | | | B location * | | | | */ {0, SW1_CS1, SW1_CS2, SW1_CS3}, //ESC {0, SW3_CS1, SW3_CS2, SW3_CS3}, //F1 {0, SW4_CS1, SW4_CS2, SW4_CS3}, {0, SW5_CS1, SW5_CS2, SW5_CS3}, //F3 {0, SW6_CS1, SW6_CS2, SW6_CS3}, {0, SW7_CS1, SW7_CS2, SW7_CS3}, //F5 {0, SW8_CS1, SW8_CS2, SW8_CS3}, {0, SW9_CS1, SW9_CS2, SW9_CS3}, //F7 {0, SW10_CS1, SW10_CS2, SW10_CS3}, {0, SW11_CS1, SW11_CS2, SW11_CS3}, //F9 {0, SW12_CS1, SW12_CS2, SW12_CS3}, //F10 {0, SW2_CS10, SW2_CS11, SW2_CS12}, {0, SW3_CS10, SW3_CS11, SW3_CS12}, //F12 {0, SW4_CS10, SW4_CS11, SW4_CS12}, //PrtSc {0, SW5_CS10, SW5_CS11, SW5_CS12}, //Scroll {0, SW6_CS10, SW6_CS11, SW6_CS12}, //Pause //encoder no led //encoder no led {0, SW1_CS4, SW1_CS5, SW1_CS6}, //`~ {0, SW2_CS4, SW2_CS5, SW2_CS6}, //1 {0, SW3_CS4, SW3_CS5, SW3_CS6}, //2 {0, SW4_CS4, SW4_CS5, SW4_CS6}, {0, SW5_CS4, SW5_CS5, SW5_CS6}, //4 {0, SW6_CS4, SW6_CS5, SW6_CS6}, {0, SW7_CS4, SW7_CS5, SW7_CS6}, //6 {0, SW8_CS4, SW8_CS5, SW8_CS6}, {0, SW9_CS4, SW9_CS5, SW9_CS6}, //8 {0, SW10_CS4, SW10_CS5, SW10_CS6}, {0, SW11_CS4, SW11_CS5, SW11_CS6}, //0 {0, SW12_CS4, SW12_CS5, SW12_CS6}, //-_ {0, SW2_CS13, CS14_SW2, CS15_SW2}, //=+ {0, SW3_CS13, SW3_CS14, SW3_CS15}, //BS {0, SW4_CS13, SW4_CS14, SW4_CS15}, //INS {0, SW5_CS13, SW5_CS14, SW5_CS15}, //Home {0, SW6_CS13, SW6_CS14, SW6_CS15}, //PgUp {0, SW7_CS13, SW7_CS14, SW7_CS15}, //NumLK {0, SW8_CS13, SW8_CS14, SW8_CS15}, //P/ {0, SW9_CS13, SW9_CS14, SW9_CS15}, //P* {0, SW10_CS13, SW10_CS14, SW10_CS15}, //P- {0, SW1_CS7, SW1_CS8, SW1_CS9}, //Tab {0, SW2_CS7, SW2_CS8, SW2_CS9}, //Q {0, SW3_CS7, SW3_CS8, SW3_CS9}, //W {0, SW4_CS7, SW4_CS8, SW4_CS9}, //E {0, SW5_CS7, SW5_CS8, SW5_CS9}, //R {0, SW6_CS7, SW6_CS8, SW6_CS9}, //T {0, SW7_CS7, SW7_CS8, SW7_CS9}, //Y {0, SW8_CS7, SW8_CS8, SW8_CS9}, //U {0, SW9_CS7, SW9_CS8, SW9_CS9}, //I {0, SW10_CS7, SW10_CS8, SW10_CS9}, //O {0, SW11_CS7, SW11_CS8, SW11_CS9}, //P {0, SW12_CS7, SW12_CS8, SW12_CS9}, //[ {0, SW2_CS16, SW2_CS17, SW2_CS18}, //] {0, SW3_CS16, SW3_CS17, SW3_CS18}, //| {0, SW4_CS16, SW4_CS17, SW4_CS18}, //DEL {0, SW5_CS16, SW5_CS17, SW5_CS18}, //END {0, SW6_CS16, SW6_CS17, SW6_CS18}, //PgDn {0, SW7_CS16, SW7_CS17, SW7_CS18}, //P7 {0, SW8_CS16, SW8_CS17, SW8_CS18}, //P8 {0, SW9_CS16, SW9_CS17, SW9_CS18}, //P9 {0, SW10_CS16, SW10_CS17, SW10_CS18}, //P+ {1, SW1_CS1, SW1_CS2, SW1_CS3}, //Cap {1, SW2_CS1, SW2_CS2, SW2_CS3}, //A {1, SW3_CS1, SW3_CS2, SW3_CS3}, //S {1, SW4_CS1, SW4_CS2, SW4_CS3}, //D {1, SW5_CS1, SW5_CS2, SW5_CS3}, //F {1, SW6_CS1, SW6_CS2, SW6_CS3}, //G {1, SW7_CS1, SW7_CS2, SW7_CS3}, //H {1, SW8_CS1, SW8_CS2, SW8_CS3}, //J {1, SW9_CS1, SW9_CS2, SW9_CS3}, //K {1, SW10_CS1, SW10_CS2, SW10_CS3}, //L {1, SW11_CS1, SW11_CS2, SW11_CS3}, //; {1, SW12_CS1, SW12_CS2, SW12_CS3}, //. {1, SW2_CS10, SW2_CS11, SW2_CS12}, //Ent {1, SW5_CS10, SW5_CS11, SW5_CS12}, //P4 {1, SW6_CS10, SW6_CS11, SW6_CS12}, //P5 {1, SW7_CS10, SW7_CS11, SW7_CS12}, //P6 {1, SW1_CS4, SW1_CS5, SW1_CS6}, //Shift {1, SW2_CS4, SW2_CS5, SW2_CS6}, //Z {1, SW3_CS4, SW3_CS5, SW3_CS6}, //X {1, SW4_CS4, SW4_CS5, SW4_CS6}, //C {1, SW5_CS4, SW5_CS5, SW5_CS6}, //V {1, SW6_CS4, SW6_CS5, SW6_CS6}, //B {1, SW7_CS4, SW7_CS5, SW7_CS6}, //N {1, SW8_CS4, SW8_CS5, SW8_CS6}, //M {1, SW9_CS4, SW9_CS5, SW9_CS6}, //, {1, SW10_CS4, SW10_CS5, SW10_CS6}, //. {1, SW11_CS4, SW11_CS5, SW11_CS6}, //? {1, SW2_CS13, SW2_CS14, SW2_CS15}, //Shift {1, SW3_CS10, SW3_CS11, SW3_CS12}, //Up {1, SW5_CS13, SW5_CS14, SW5_CS15}, //P1 {1, SW6_CS13, SW6_CS14, SW6_CS15}, //P2 {1, SW7_CS13, SW7_CS14, SW7_CS15}, //P3 {1, SW7_CS16, SW7_CS17, SW7_CS18}, //PENT {1, SW1_CS7, SW1_CS8, SW1_CS9}, //Ctrl {1, SW2_CS7, SW2_CS8, SW2_CS9}, //Com {1, SW3_CS7, SW3_CS8, SW3_CS9}, //Alt {1, SW4_CS7, SW4_CS8, SW4_CS9}, //SPAC1 {1, SW6_CS7, SW6_CS8, SW6_CS9}, //SPAC {1, SW8_CS7, SW8_CS8, SW8_CS9}, //SPAC2 {1, SW10_CS7, SW10_CS8, SW10_CS9}, //Alt {1, SW11_CS7, SW11_CS8, SW11_CS9}, //Fn {1, SW12_CS7, SW12_CS8, SW12_CS9}, //Menu {1, SW2_CS16, SW2_CS17, SW2_CS18}, //Ctrl {1, SW3_CS16, SW3_CS17, SW3_CS18}, //Left {1, SW3_CS13, SW3_CS14, SW3_CS15}, //Down {1, SW4_CS16, SW4_CS17, SW4_CS18}, //Right {1, SW5_CS16, SW5_CS17, SW5_CS18}, //P0 {1, SW6_CS16, SW6_CS17, SW6_CS18} //PDot }; #endif bool process_record_kb(uint16_t keycode, keyrecord_t *record) { if (!process_record_user(keycode, record)) { return false; } switch (keycode) { # ifdef RGB_MATRIX_ENABLE case RGB_TOG: if (record->event.pressed) { switch (rgb_matrix_get_flags()) { case LED_FLAG_ALL: { rgb_matrix_set_flags(LED_FLAG_NONE); rgb_matrix_set_color_all(0, 0, 0); } break; default: { rgb_matrix_set_flags(LED_FLAG_ALL); } break; } } return false; case RGB_VAI: rgb_matrix_set_flags(LED_FLAG_ALL); return true; # endif case MO(1): if (record->event.pressed) { FN_ON = 1; } else { FN_ON = 0; } return true; case MO(2): if (record->event.pressed) { FN_ON = 1; } else { FN_ON = 0; } return true; case MO(3): if (record->event.pressed) { FN_ON = 1; } else { FN_ON = 0; } return true; case KC_LGUI: if (FN_ON){ if ( record->event.pressed){ WIN_LOCK = !WIN_LOCK ; //change win lock state } if (!WIN_LOCK) { IND = IND & (~WINLK_ON); //Close win lock display s_serial_to_parallel(IND); return false; //windows key locked do nothing } s_serial_to_parallel(IND); } if (WIN_LOCK) { IND = IND | WINLK_ON; //Open win lock display s_serial_to_parallel(IND); return false; //windows key locked do nothing } return true; // continue all further processing of this key case TO(0): set_single_persistent_default_layer(0); return true; case TO(1): set_single_persistent_default_layer(1); return true; default: return true; } s_serial_to_parallel(IND); } void suspend_power_down_kb() { # ifdef RGB_MATRIX_ENABLE writePinLow(SDB); # endif writePinLow(MAC_PIN); s_serial_to_parallel(0); suspend_power_down_user(); } void suspend_wakeup_init_kb() { # ifdef RGB_MATRIX_ENABLE writePinHigh(SDB); # endif s_serial_to_parallel(IND); suspend_wakeup_init_user(); } bool shutdown_kb(bool jump_to_bootloader) { writePinLow(SDB); s_serial_to_parallel(0); return true; } layer_state_t default_layer_state_set_kb(layer_state_t state) { switch (get_highest_layer(state)) { case 0: writePinLow(MAC_PIN); //switch to win layer display IND = IND & (~MAC_ON); IND = IND | WIN_ON; break; case 1: writePinHigh(MAC_PIN); //switch to mac layer display IND = IND & (~WIN_ON); IND = IND | MAC_ON; break; } s_serial_to_parallel(IND); return state; } bool rgb_matrix_indicators_advanced_kb(uint8_t led_min, uint8_t led_max) { if (!rgb_matrix_indicators_advanced_user(led_min, led_max)) { return false; } //caps lock display if (host_keyboard_led_state().caps_lock) { RGB_MATRIX_INDICATOR_SET_COLOR(CAPS_LOCK_INDEX, 255, 255, 255); IND = IND | CAPS_ON; } else { if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(CAPS_LOCK_INDEX, 0, 0, 0); } IND = IND & (~CAPS_ON); } //number lock display if (host_keyboard_led_state().num_lock) { RGB_MATRIX_INDICATOR_SET_COLOR(NUM_LOCK_INDEX, 255, 255, 255); IND = IND | NUM_ON; } else { if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(NUM_LOCK_INDEX, 0, 0, 0); } IND = IND & (~NUM_ON); } //scroll lock display if (host_keyboard_led_state().scroll_lock) { RGB_MATRIX_INDICATOR_SET_COLOR(SCR_LOCK_INDEX, 255, 255, 255); IND = IND | SCR_ON; } else { if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(SCR_LOCK_INDEX, 0, 0, 0); } IND = IND & (~SCR_ON); } switch (get_highest_layer(layer_state)) { case 0:{ if (FN_ON) { RGB_MATRIX_INDICATOR_SET_COLOR(WIN_MOD_INDEX, 255, 255, 255); if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(MAC_MOD_INDEX, 0, 0, 0); RGB_MATRIX_INDICATOR_SET_COLOR(WIN_MOD_INDEX, 0, 0, 0); } } } break; case 1:{ if (FN_ON) { RGB_MATRIX_INDICATOR_SET_COLOR(MAC_MOD_INDEX, 255, 255, 255); if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(WIN_MOD_INDEX, 0, 0, 0); RGB_MATRIX_INDICATOR_SET_COLOR(MAC_MOD_INDEX, 0, 0, 0); } } } break; default:{ if (!rgb_matrix_get_flags()) { RGB_MATRIX_INDICATOR_SET_COLOR(WIN_MOD_INDEX, 0, 0, 0); RGB_MATRIX_INDICATOR_SET_COLOR(MAC_MOD_INDEX, 0, 0, 0); } } return true; } s_serial_to_parallel(IND); return true; } void board_init(void) { // JTAG-DP Disabled and SW-DP Disabled AFIO->MAPR = (AFIO->MAPR & ~AFIO_MAPR_SWJ_CFG_Msk) | AFIO_MAPR_SWJ_CFG_DISABLE; # ifdef RGB_MATRIX_ENABLE setPinOutput(SDB); writePinHigh(SDB); # endif setPinOutput(MAC_PIN); writePinHigh(MAC_PIN); s_serial_to_parallel(0xFF); IND = SKYLOONG; }