X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=keyboards%2Fv60_type_r%2Fv60_type_r.c;h=016348db9f7f582a1f2c4014ea2c16a07a3e8eaa;hb=4e92dceed815d00a5492c5606ce7d863606d6507;hp=4f88919bc364d660b92e1f20295102b23bbf106d;hpb=6ec7ccec63a5f434af3b6d5726fac858adfdbb61;p=qmk_firmware.git

diff --git a/keyboards/v60_type_r/v60_type_r.c b/keyboards/v60_type_r/v60_type_r.c
index 4f88919bc..016348db9 100644
--- a/keyboards/v60_type_r/v60_type_r.c
+++ b/keyboards/v60_type_r/v60_type_r.c
@@ -15,6 +15,200 @@
  */
 #include "v60_type_r.h"
 
+#include "rgblight.h"
+
+#include <avr/pgmspace.h>
+
+#include "action_layer.h"
+#include "quantum.h"
+
+// if we've got an RGB underglow!
+#ifdef V60_POLESTAR
+#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_user(void) {
+	rgb_init();
+}
+
+bool process_record_user(uint16_t keycode, keyrecord_t *record) {
+	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 true;
+}
+
+
+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;
+  	}
+}
+#else
+
+void matrix_init_user(void) {
+}
+
+bool process_record_user(uint16_t keycode, keyrecord_t *record) {
+	return true;
+}
+
+#endif // V60_POLESTAR
+
+// we need these functions for both versions
+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 matrix_scan_user(void) {
+}
+
 void matrix_init_kb(void) {
 	// put your keyboard start-up code here
 	// runs once when the firmware starts up
@@ -35,9 +229,3 @@ bool process_record_kb(uint16_t keycode, keyrecord_t *record) {
 
 	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);
-}