[Keyboard] V60 Type R - Turn on leds for Configurator + Refactor (#5546)

[qmk_firmware.git] / keyboards / v60_type_r / v60_type_r.c

diff --git a/keyboards/v60_type_r/v60_type_r.c b/keyboards/v60_type_r/v60_type_r.c
index 4f88919bc364d660b92e1f20295102b23bbf106d..41eb75807d080e79f5dc16398c120f957f5a0b0f 100644 (file)
--- a/keyboards/v60_type_r/v60_type_r.c
+++ b/keyboards/v60_type_r/v60_type_r.c
@@ -1,4 +1,4 @@
-/* Copyright 2017 REPLACE_WITH_YOUR_NAME
+/* Copyright 2017 benlyall, MechMerlin
  *
  * 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
@@ -15,29 +15,180 @@
  */
 #include "v60_type_r.h"
 
-void matrix_init_kb(void) {
-       // put your keyboard start-up code here
-       // runs once when the firmware starts up
+#include "quantum.h"
 
-       matrix_init_user();
-}
+// if we've got an RGB underglow!
+#ifdef V60_POLESTAR
+
+#include "rgblight.h"
+
+#include <avr/pgmspace.h>
 
-void matrix_scan_kb(void) {
-       // put your looping keyboard code here
-       // runs every cycle (a lot)
+#include "action_layer.h"
 
-       matrix_scan_user();
+#define SOFTPWM_LED_TIMER_TOP F_CPU/(256*64)
+
+extern rgblight_config_t rgblight_config;
+static uint8_t softpwm_buff[3] = {0};
+
+void matrix_init_kb(void) {
+       rgb_init();
+       matrix_init_user();
 }
 
 bool process_record_kb(uint16_t keycode, keyrecord_t *record) {
-       // put your per-action keyboard code here
-       // runs for every action, just before processing by the firmware
+       uint8_t r = led[0].r, g = led[0].g, b = led[0].b;
+       switch(keycode) {
+               case RGB_RI:
+                       if (record->event.pressed) {
+                               r += RGB_STEP;
+                               if (r < led[0].r) {
+                                       r = 255;
+                               }
+                               rgblight_setrgb(r, g, b);
+                       }
+
+                       return false;
+               case RGB_RD:
+                 if (record->event.pressed) {
+                       r -= RGB_STEP;
+                               if (r > led[0].r) {
+                                       r = 0;
+                               }
+                               rgblight_setrgb(r, g, b);
+                 }
+
+                       return false;
+               case RGB_BI:
+                       if (record->event.pressed) {
+                               b += RGB_STEP;
+                               if (b < led[0].b) {
+                                       b = 255;
+                               }
+                               rgblight_setrgb(r, g, b);
+                       }
+
+                       return false;
+               case RGB_BD:
+                 if (record->event.pressed) {
+                       b -= RGB_STEP;
+                               if (b > led[0].b) {
+                                       b = 0;
+                               }
+                               rgblight_setrgb(r, g, b);
+                 }
+
+                       return false;
+               case RGB_GI:
+                       if (record->event.pressed) {
+                               g += RGB_STEP;
+                               if (g < led[0].g) {
+                                       g = 255;
+                               }
+                               rgblight_setrgb(r, g, b);
+                       }
+
+                       return false;
+               case RGB_GD:
+                 if (record->event.pressed) {
+                       g -= RGB_STEP;
+                               if (g > led[0].g) {
+                                       g = 0;
+                               }
+                               rgblight_setrgb(r, g, b);
+                       }
+
+                       return false;
+       }
 
        return process_record_user(keycode, record);
 }
 
-void led_set_kb(uint8_t usb_led) {
-       // put your keyboard LED indicator (ex: Caps Lock LED) toggling code here
 
-       led_set_user(usb_led);
+void rgb_timer_init(void) {
+    /* Timer1 setup */
+    /* CTC mode */
+    TCCR1B |= (1<<WGM12);
+    /* Clock selelct: clk/8 */
+    TCCR1B |= (1<<CS10);
+    /* Set TOP value */
+    uint8_t sreg = SREG;
+    cli();
+    OCR1AH = (SOFTPWM_LED_TIMER_TOP >> 8) & 0xff;
+    OCR1AL = SOFTPWM_LED_TIMER_TOP & 0xff;
+    SREG = sreg;
+
+    // Enable the compare match interrupt on timer 1
+    TIMSK1 |= (1<<OCIE1A);
+}
+
+void rgb_init(void) {
+    DDRF  |=  (1<<PF6 | 1<<PF5 | 1<<PF4);
+    PORTF |=  (1<<PF6 | 1<<PF5 | 1<<PF4);
+
+    rgb_timer_init();
+}
+
+void set_rgb_pin_on(uint8_t pin) {
+       PORTF &= ~(1<<pin);
+}
+
+void set_rgb_pin_off(uint8_t pin) {
+       PORTF |= (1<<pin);
+}
+
+void rgblight_set(void) {
+         // xprintf("Setting RGB underglow\n");
+    if (!rgblight_config.enable) {
+          led[0].r = 0;
+          led[0].g = 0;
+          led[0].b = 0;
+          set_rgb_pin_off(RGB_RED_PIN);
+          set_rgb_pin_off(RGB_GREEN_PIN);
+          set_rgb_pin_off(RGB_BLUE_PIN);
+    }
+
+   //  //xprintf("Red: %u, Green: %u, Blue: %u\n", led[0].r, led[0].g, led[0].b);
+}
+
+ISR(TIMER1_COMPA_vect)
+{
+    static uint8_t pwm = 0;
+    pwm++;
+
+    // turn the LEDS on
+    if (pwm == 0) {
+       if (softpwm_buff[0]) {
+               set_rgb_pin_on(RGB_RED_PIN);
+               softpwm_buff[0] = led[0].r;
+       }
+
+       if (softpwm_buff[1]) {
+               set_rgb_pin_on(RGB_GREEN_PIN);
+               softpwm_buff[1] = led[0].g;
+       }
+
+       if (softpwm_buff[2]) {
+               set_rgb_pin_on(RGB_BLUE_PIN);
+               softpwm_buff[2] = led[0].b;
+       }
+    }
+
+    // turn em off
+       if (pwm == softpwm_buff[0]) {
+               set_rgb_pin_off(RGB_RED_PIN);
+               softpwm_buff[0] = led[0].r;
+
+       }
+
+       if (pwm == softpwm_buff[1]) {
+               set_rgb_pin_off(RGB_GREEN_PIN);
+       softpwm_buff[1] = led[0].g;
+       }
+
+       if (pwm == softpwm_buff[2]) {
+               set_rgb_pin_off(RGB_BLUE_PIN);
+       softpwm_buff[2] = led[0].b;
+       }
 }
+#endif // V60_POLESTAR