summaryrefslogtreecommitdiff
path: root/keyboards/neson_design/n6/n6.c
diff options
context:
space:
mode:
Diffstat (limited to 'keyboards/neson_design/n6/n6.c')
-rw-r--r--keyboards/neson_design/n6/n6.c319
1 files changed, 319 insertions, 0 deletions
diff --git a/keyboards/neson_design/n6/n6.c b/keyboards/neson_design/n6/n6.c
new file mode 100644
index 0000000000..800aa0ae9a
--- /dev/null
+++ b/keyboards/neson_design/n6/n6.c
@@ -0,0 +1,319 @@
+/**
+ * @file n6.c
+ *
+ Copyright 2021 astro
+
+ 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 "n6.h"
+#include "i2c_master.h"
+#include "issi/is31fl3731.h"
+
+enum {
+ SELF_TESTING,
+ CAPS_ALERT,
+ NORMAL,
+};
+
+enum {
+ ST_STAGE_1,
+ ST_STAGE_2,
+ ST_STAGE_3,
+};
+
+// alert state update interval
+#define ALERT_INTERVAL 600
+// self testing state update interval
+#define ST_INTERVAL 100
+// self testing start index
+#define ST_DEFAULT_INDEX 15
+// self testing stage delay
+#define ST_STAGE_DELAY 10
+// self testing stage cycle count
+#define ST_STAGE_COUNT 4
+// self testing stage end duration
+#define ST_END_DURATION 10
+
+#ifdef RGBLIGHT_ENABLE
+extern rgblight_config_t rgblight_config;
+
+typedef struct {
+ uint8_t state;
+ uint8_t testing;
+ bool alert;
+ uint8_t index;
+ uint8_t delay;
+ uint8_t count;
+ bool dir;
+ uint8_t duration;
+ uint16_t ticks;
+} rgb_state_t;
+
+static rgb_state_t rgb_state = {
+ .state = //NORMAL,
+ SELF_TESTING,
+ .testing = ST_STAGE_1,
+ .ticks = 0,
+ .alert = false,
+ .index = ST_DEFAULT_INDEX,
+ .delay = ST_STAGE_DELAY,
+ .count = ST_STAGE_COUNT,
+ .dir = true,
+ .duration = ST_END_DURATION,
+};
+
+static void update_ticks(void)
+{
+ rgb_state.ticks = timer_read();
+}
+
+static void self_testing(void)
+{
+ if (timer_elapsed(rgb_state.ticks) < ST_INTERVAL) return;
+ HSV hsv;
+ hsv.h = rgblight_config.hue;
+ hsv.s = rgblight_config.sat;
+ hsv.v = rgblight_config.val;
+
+ RGB led = hsv_to_rgb(hsv);
+ switch(rgb_state.testing) {
+ case ST_STAGE_1:
+ if (rgb_state.index !=0 ) {
+ IS31FL3731_set_color_all(0, 0, 0);
+ }
+
+ if (rgb_state.index >= 1) {
+ for (int i = rgb_state.index-1; i < 32-rgb_state.index+1; i++) {
+ IS31FL3731_set_color(i, led.r, led.g, led.b);
+ }
+ if (rgb_state.index==1) {
+ rgb_state.index=0;
+ } else {
+ rgb_state.index -= 2;
+ }
+ } else{
+ if (rgb_state.delay > 0) {
+ rgb_state.delay--;
+ } else {
+ // move to stage 2
+ rgb_state.index = 2;
+ rgb_state.testing = ST_STAGE_2;
+ }
+ }
+ break;
+ case ST_STAGE_2: {
+ // clear all
+ IS31FL3731_set_color_all(0, 0, 0);
+ // light left and right
+ IS31FL3731_set_color(0, led.r, led.g, led.b);
+ IS31FL3731_set_color(1, led.r, led.g, led.b);
+ IS31FL3731_set_color(30, led.r, led.g, led.b);
+ IS31FL3731_set_color(31, led.r, led.g, led.b);
+ if (rgb_state.dir) {
+ // left to right
+ for (int i = rgb_state.index; i < rgb_state.index+4; i++) {
+ IS31FL3731_set_color(i, led.r, led.g, led.b);
+ }
+ rgb_state.index += 4;
+ if (rgb_state.index == 30) {
+ rgb_state.dir = !rgb_state.dir;
+ rgb_state.count--;
+ }
+ } else {
+ // right to left
+ for (int i = rgb_state.index-4; i < rgb_state.index; i++) {
+ IS31FL3731_set_color(i, led.r, led.g, led.b);
+ }
+ rgb_state.index -= 4;
+ if (rgb_state.index == 2) {
+ rgb_state.dir = !rgb_state.dir;
+ rgb_state.count--;
+ }
+ }
+
+ if (rgb_state.count == 0) {
+ // move to stage 3
+ rgb_state.testing = ST_STAGE_3;
+ rgb_state.index = 0;
+ rgb_state.delay = ST_STAGE_DELAY;
+ rgb_state.duration = ST_END_DURATION;
+ }
+ }
+ break;
+ case ST_STAGE_3:
+ if (rgb_state.index != 16) {
+ IS31FL3731_set_color_all(0, 0, 0);
+ }
+
+ // light left and right
+
+ if (rgb_state.index == 16) {
+ if (rgb_state.duration) {
+ rgb_state.duration--;
+ } else {
+ if (IS_HOST_LED_ON(USB_LED_CAPS_LOCK)) {
+ rgb_state.state = CAPS_ALERT;
+ } else {
+ rgb_state.state = NORMAL;
+ rgblight_set();
+ }
+ }
+ } else {
+ // left
+ for (int i = 0; i < rgb_state.index+1; i++) {
+ IS31FL3731_set_color(i, led.r, led.g, led.b);
+ }
+ // right
+ for (int i = 31; i > 31-rgb_state.index-1; i--) {
+ IS31FL3731_set_color(i, led.r, led.g, led.b);
+ }
+ rgb_state.index ++;
+ }
+ break;
+ }
+
+ update_ticks();
+}
+
+const is31_led g_is31_leds[DRIVER_LED_TOTAL] = {
+/* Refer to IS31 manual for these locations
+ * driver
+ * | R location
+ * | | G location
+ * | | | B location
+ * | | | | */
+// left CA
+ {0, C1_1, C3_2, C4_2},
+ {0, C1_2, C2_2, C4_3},
+ {0, C1_3, C2_3, C3_3},
+ {0, C1_4, C2_4, C3_4},
+ {0, C1_5, C2_5, C3_5},
+ {0, C1_6, C2_6, C3_6},
+ {0, C1_7, C2_7, C3_7},
+ {0, C1_8, C2_8, C3_8},
+
+ {0, C9_1, C8_1, C7_1},
+ {0, C9_2, C8_2, C7_2},
+ {0, C9_3, C8_3, C7_3},
+ {0, C9_4, C8_4, C7_4},
+ {0, C9_5, C8_5, C7_5},
+ {0, C9_6, C8_6, C7_6},
+ {0, C9_7, C8_7, C6_6},
+ {0, C9_8, C7_7, C6_7},
+// left CB
+ {0, C1_9, C3_10, C4_10},
+ {0, C1_10, C2_10, C4_11},
+ {0, C1_11, C2_11, C3_11},
+ {0, C1_12, C2_12, C3_12},
+ {0, C1_13, C2_13, C3_13},
+ {0, C1_14, C2_14, C3_14},
+ {0, C1_15, C2_15, C3_15},
+ {0, C1_16, C2_16, C3_16},
+
+ {0, C9_9, C8_9, C7_9},
+ {0, C9_10, C8_10, C7_10},
+ {0, C9_11, C8_11, C7_11},
+ {0, C9_12, C8_12, C7_12},
+ {0, C9_13, C8_13, C7_13},
+ {0, C9_14, C8_14, C7_14},
+ {0, C9_15, C8_15, C6_14},
+ {0, C9_16, C7_15, C6_15},
+};
+#endif
+__attribute__((weak))
+void matrix_init_user(void) {}
+
+void matrix_init_kb(void)
+{
+ // clear caps led
+ setPinOutput(CAPS_PIN);
+ writePinLow(CAPS_PIN);
+#ifdef RGBLIGHT_ENABLE
+ i2c_init();
+ IS31FL3731_init(DRIVER_ADDR_1);
+ for (int index = 0; index < DRIVER_LED_TOTAL; index++) {
+ IS31FL3731_set_led_control_register(index, true, true, true);
+ }
+ IS31FL3731_update_led_control_registers(DRIVER_ADDR_1, 0);
+ update_ticks();
+#endif
+ matrix_init_user();
+}
+
+#ifdef RGBLIGHT_ENABLE
+void housekeeping_task_kb(void)
+{
+ if (rgb_state.state == SELF_TESTING) {
+ self_testing();
+ } else if (rgb_state.state == CAPS_ALERT) {
+ //gold 0xFF, 0xD9, 0x00
+ LED_TYPE led = {
+ .r = 0xFF,
+ .g = 0xD9,
+ .b = 0x00,
+ };
+ if (rgb_state.alert) {
+ IS31FL3731_set_color_all(led.r, led.g, led.b);
+ ws2812_setleds(&led, 1);
+ } else {
+ led.r = 0;
+ led.g = 0;
+ led.b = 0;
+ IS31FL3731_set_color_all(0, 0, 0);
+ ws2812_setleds(&led, 1);
+ }
+
+ if (timer_elapsed(rgb_state.ticks) > ALERT_INTERVAL) {
+ rgb_state.alert = !rgb_state.alert;
+ update_ticks();
+ }
+ }
+
+ IS31FL3731_update_pwm_buffers(DRIVER_ADDR_1,0);
+
+ housekeeping_task_user();
+}
+
+void rgblight_call_driver(LED_TYPE *start_led, uint8_t num_leds)
+{
+ if (rgb_state.state != NORMAL) return;
+
+ for (uint8_t i = 0; i < DRIVER_LED_TOTAL; i++) {
+ IS31FL3731_set_color(i, start_led[i].r, start_led[i].g, start_led[i].b);
+ }
+ ws2812_setleds(start_led+DRIVER_LED_TOTAL, 1);
+}
+#endif
+
+bool led_update_kb(led_t led_state)
+{
+ bool res = led_update_user(led_state);
+ if (res) {
+ writePin(CAPS_PIN, led_state.caps_lock);
+
+#ifdef RGBLIGHT_ENABLE
+ if (rgb_state.state != SELF_TESTING) {
+ if (led_state.caps_lock) {
+ rgb_state.state = CAPS_ALERT;
+ update_ticks();
+ } else {
+ rgb_state.state = NORMAL;
+ rgblight_set();
+ }
+ }
+#endif
+ }
+ return res;
+} \ No newline at end of file
generated by cgit v1.2.1 (git 2.18.0) at 2024-11-10 11:33:43 +0000