diff options
Diffstat (limited to 'quantum/template/ps2avrgb/template.c')
-rw-r--r-- | quantum/template/ps2avrgb/template.c | 41 |
1 files changed, 37 insertions, 4 deletions
diff --git a/quantum/template/ps2avrgb/template.c b/quantum/template/ps2avrgb/template.c index 16fcc8469f..acc8698f56 100644 --- a/quantum/template/ps2avrgb/template.c +++ b/quantum/template/ps2avrgb/template.c @@ -15,10 +15,43 @@ */ #include "%KEYBOARD%.h" -#include <avr/pgmspace.h> -#include "action_layer.h" -#include "i2c.h" -#include "quantum.h" +#ifdef RGBLIGHT_ENABLE + +# include <string.h> +# include "i2c_master.h" +# include "rgblight.h" + +extern rgblight_config_t rgblight_config; + +void matrix_init_kb(void) { + i2c_init(); + // call user level keymaps, if any + matrix_init_user(); +} + +// custom RGB driver +void rgblight_set(void) { + if (!rgblight_config.enable) { + memset(led, 0, 3 * RGBLED_NUM); + } + + i2c_transmit(0xb0, (uint8_t*)led, 3 * RGBLED_NUM, 100); +} + +bool rgb_init = false; + +void matrix_scan_kb(void) { + // if LEDs were previously on before poweroff, turn them back on + if (rgb_init == false && rgblight_config.enable) { + i2c_transmit(0xb0, (uint8_t*)led, 3 * RGBLED_NUM, 100); + rgb_init = true; + } + + rgblight_task(); + matrix_scan_user(); +} + +#endif __attribute__ ((weak)) void matrix_scan_user(void) { |