GMMK Pro RGB Support (#13147)
* Enable SPI1 for GMMK pro * Setup initial boilerplate for new LED driver * RGB matrix minimally functional * Map full LED matrix * Return keymap to default * Fix printscreen LED mapping * Reduce max brightness * Default values for AW20216 * Add documentation for AW20216 * Disable console and warnings * Run cformat * Update drivers/awinic/aw20216.h Co-authored-by: Drashna Jaelre <drashna@live.com> * make aw struct match issi struct Co-authored-by: Drashna Jaelre <drashna@live.com> * add led location defines Co-authored-by: Drashna Jaelre <drashna@live.com> * Use led pin definitions in keyboard.c * Add driver indices to led map * Fix elif typo * Run cformat * Update docs * Fix typo in docs * Document global brightness limits Co-authored-by: Drashna Jaelre <drashna@live.com>python_optimizations
Gigahawk
4 years ago
committed by
GitHub
parent
e21a039901
commit
32b2ac0a80
@ -0,0 +1,166 @@ |
||||
/* Copyright 2021 Jasper Chan
|
||||
* |
||||
* 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 "aw20216.h" |
||||
#include "spi_master.h" |
||||
|
||||
/* The AW20216 appears to be somewhat similar to the IS31FL743, although quite
|
||||
* a few things are different, such as the command byte format and page ordering. |
||||
* The LED addresses start from 0x00 instead of 0x01. |
||||
*/ |
||||
#define AWINIC_ID 0b1010 << 4 |
||||
|
||||
#define AW_PAGE_FUNCTION 0x00 << 1 // PG0, Function registers
|
||||
#define AW_PAGE_PWM 0x01 << 1 // PG1, LED PWM control
|
||||
#define AW_PAGE_SCALING 0x02 << 1 // PG2, LED current scaling control
|
||||
#define AW_PAGE_PATCHOICE 0x03 << 1 // PG3, Pattern choice?
|
||||
#define AW_PAGE_PWMSCALING 0x04 << 1 // PG4, LED PWM + Scaling control?
|
||||
|
||||
#define AW_WRITE 0 |
||||
#define AW_READ 1 |
||||
|
||||
#define AW_REG_CONFIGURATION 0x00 // PG0
|
||||
#define AW_REG_GLOBALCURRENT 0x01 // PG0
|
||||
|
||||
// Default value of AW_REG_CONFIGURATION
|
||||
// D7:D4 = 1011, SWSEL (SW1~SW12 active)
|
||||
// D3 = 0?, reserved (apparently this should be 1 but it doesn't seem to matter)
|
||||
// D2:D1 = 00, OSDE (open/short detection enable)
|
||||
// D0 = 0, CHIPEN (write 1 to enable LEDs when hardware enable pulled high)
|
||||
#define AW_CONFIG_DEFAULT 0b10110000 |
||||
#define AW_CHIPEN 1 |
||||
|
||||
#ifndef AW_SCALING_MAX |
||||
# define AW_SCALING_MAX 150 |
||||
#endif |
||||
|
||||
#ifndef AW_GLOBAL_CURRENT_MAX |
||||
# define AW_GLOBAL_CURRENT_MAX 150 |
||||
#endif |
||||
|
||||
#ifndef DRIVER_1_CS |
||||
# define DRIVER_1_CS B13 |
||||
#endif |
||||
|
||||
#ifndef DRIVER_1_EN |
||||
# define DRIVER_1_EN C13 |
||||
#endif |
||||
|
||||
uint8_t g_spi_transfer_buffer[20] = {0}; |
||||
aw_led g_pwm_buffer[DRIVER_LED_TOTAL]; |
||||
bool g_pwm_buffer_update_required[DRIVER_LED_TOTAL]; |
||||
|
||||
bool AW20216_write_register(pin_t slave_pin, uint8_t page, uint8_t reg, uint8_t data) { |
||||
// Do we need to call spi_stop() if this fails?
|
||||
if (!spi_start(slave_pin, false, 0, 16)) { |
||||
return false; |
||||
} |
||||
|
||||
g_spi_transfer_buffer[0] = (AWINIC_ID | page | AW_WRITE); |
||||
g_spi_transfer_buffer[1] = reg; |
||||
g_spi_transfer_buffer[2] = data; |
||||
|
||||
if (spi_transmit(g_spi_transfer_buffer, 3) != SPI_STATUS_SUCCESS) { |
||||
spi_stop(); |
||||
return false; |
||||
} |
||||
spi_stop(); |
||||
return true; |
||||
} |
||||
|
||||
bool AW20216_init_scaling(void) { |
||||
// Set constant current to the max, control brightness with PWM
|
||||
aw_led led; |
||||
for (uint8_t i = 0; i < DRIVER_LED_TOTAL; i++) { |
||||
led = g_aw_leds[i]; |
||||
if (led.driver == 0) { |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_SCALING, led.r, AW_SCALING_MAX); |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_SCALING, led.g, AW_SCALING_MAX); |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_SCALING, led.b, AW_SCALING_MAX); |
||||
} |
||||
#ifdef DRIVER_2_CS |
||||
else if (led.driver == 1) { |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_SCALING, led.r, AW_SCALING_MAX); |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_SCALING, led.g, AW_SCALING_MAX); |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_SCALING, led.b, AW_SCALING_MAX); |
||||
} |
||||
#endif |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
bool AW20216_soft_enable(void) { |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_FUNCTION, AW_REG_CONFIGURATION, AW_CONFIG_DEFAULT | AW_CHIPEN); |
||||
#ifdef DRIVER_2_CS |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_FUNCTION, AW_REG_CONFIGURATION, AW_CONFIG_DEFAULT | AW_CHIPEN); |
||||
#endif |
||||
return true; |
||||
} |
||||
|
||||
void AW20216_update_pwm(int index, uint8_t red, uint8_t green, uint8_t blue) { |
||||
aw_led led = g_aw_leds[index]; |
||||
if (led.driver == 0) { |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_PWM, led.r, red); |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_PWM, led.g, green); |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_PWM, led.b, blue); |
||||
} |
||||
#ifdef DRIVER_2_CS |
||||
else if (led.driver == 1) { |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_PWM, led.r, red); |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_PWM, led.g, green); |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_PWM, led.b, blue); |
||||
} |
||||
#endif |
||||
return; |
||||
} |
||||
|
||||
void AW20216_init(void) { |
||||
// All LEDs should start with all scaling and PWM registers as off
|
||||
setPinOutput(DRIVER_1_EN); |
||||
writePinHigh(DRIVER_1_EN); |
||||
AW20216_write_register(DRIVER_1_CS, AW_PAGE_FUNCTION, AW_REG_GLOBALCURRENT, AW_GLOBAL_CURRENT_MAX); |
||||
#ifdef DRIVER_2_EN |
||||
setPinOutput(DRIVER_2_EN); |
||||
writePinHigh(DRIVER_2_EN); |
||||
AW20216_write_register(DRIVER_2_CS, AW_PAGE_FUNCTION, AW_REG_GLOBALCURRENT, AW_GLOBAL_CURRENT_MAX); |
||||
#endif |
||||
AW20216_init_scaling(); |
||||
AW20216_soft_enable(); |
||||
return; |
||||
} |
||||
|
||||
void AW20216_set_color(int index, uint8_t red, uint8_t green, uint8_t blue) { |
||||
g_pwm_buffer[index].r = red; |
||||
g_pwm_buffer[index].g = green; |
||||
g_pwm_buffer[index].b = blue; |
||||
g_pwm_buffer_update_required[index] = true; |
||||
return; |
||||
} |
||||
void AW20216_set_color_all(uint8_t red, uint8_t green, uint8_t blue) { |
||||
for (uint8_t i = 0; i < DRIVER_LED_TOTAL; i++) { |
||||
AW20216_set_color(i, red, green, blue); |
||||
} |
||||
return; |
||||
} |
||||
void AW20216_update_pwm_buffers(void) { |
||||
for (uint8_t i = 0; i < DRIVER_LED_TOTAL; i++) { |
||||
if (g_pwm_buffer_update_required[i]) { |
||||
AW20216_update_pwm(i, g_pwm_buffer[i].r, g_pwm_buffer[i].g, g_pwm_buffer[i].b); |
||||
g_pwm_buffer_update_required[i] = false; |
||||
} |
||||
} |
||||
return; |
||||
} |
||||
@ -0,0 +1,251 @@ |
||||
/* Copyright 2021 Jasper Chan (Gigahawk)
|
||||
* |
||||
* 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/>.
|
||||
*/ |
||||
|
||||
#pragma once |
||||
|
||||
#include <stdint.h> |
||||
#include <stdbool.h> |
||||
|
||||
typedef struct aw_led { |
||||
uint8_t driver : 2; |
||||
uint8_t r; |
||||
uint8_t g; |
||||
uint8_t b; |
||||
} aw_led; |
||||
|
||||
extern const aw_led g_aw_leds[DRIVER_LED_TOTAL]; |
||||
|
||||
void AW20216_init(void); |
||||
void AW20216_set_color(int index, uint8_t red, uint8_t green, uint8_t blue); |
||||
void AW20216_set_color_all(uint8_t red, uint8_t green, uint8_t blue); |
||||
void AW20216_update_pwm_buffers(void); |
||||
|
||||
#define CS1_SW1 0x00 |
||||
#define CS2_SW1 0x01 |
||||
#define CS3_SW1 0x02 |
||||
#define CS4_SW1 0x03 |
||||
#define CS5_SW1 0x04 |
||||
#define CS6_SW1 0x05 |
||||
#define CS7_SW1 0x06 |
||||
#define CS8_SW1 0x07 |
||||
#define CS9_SW1 0x08 |
||||
#define CS10_SW1 0x09 |
||||
#define CS11_SW1 0x0A |
||||
#define CS12_SW1 0x0B |
||||
#define CS13_SW1 0x0C |
||||
#define CS14_SW1 0x0D |
||||
#define CS15_SW1 0x0E |
||||
#define CS16_SW1 0x0F |
||||
#define CS17_SW1 0x10 |
||||
#define CS18_SW1 0x11 |
||||
#define CS1_SW2 0x12 |
||||
#define CS2_SW2 0x13 |
||||
#define CS3_SW2 0x14 |
||||
#define CS4_SW2 0x15 |
||||
#define CS5_SW2 0x16 |
||||
#define CS6_SW2 0x17 |
||||
#define CS7_SW2 0x18 |
||||
#define CS8_SW2 0x19 |
||||
#define CS9_SW2 0x1A |
||||
#define CS10_SW2 0x1B |
||||
#define CS11_SW2 0x1C |
||||
#define CS12_SW2 0x1D |
||||
#define CS13_SW2 0x1E |
||||
#define CS14_SW2 0x1F |
||||
#define CS15_SW2 0x20 |
||||
#define CS16_SW2 0x21 |
||||
#define CS17_SW2 0x22 |
||||
#define CS18_SW2 0x23 |
||||
#define CS1_SW3 0x24 |
||||
#define CS2_SW3 0x25 |
||||
#define CS3_SW3 0x26 |
||||
#define CS4_SW3 0x27 |
||||
#define CS5_SW3 0x28 |
||||
#define CS6_SW3 0x29 |
||||
#define CS7_SW3 0x2A |
||||
#define CS8_SW3 0x2B |
||||
#define CS9_SW3 0x2C |
||||
#define CS10_SW3 0x2D |
||||
#define CS11_SW3 0x2E |
||||
#define CS12_SW3 0x2F |
||||
#define CS13_SW3 0x30 |
||||
#define CS14_SW3 0x31 |
||||
#define CS15_SW3 0x32 |
||||
#define CS16_SW3 0x33 |
||||
#define CS17_SW3 0x34 |
||||
#define CS18_SW3 0x35 |
||||
#define CS1_SW4 0x36 |
||||
#define CS2_SW4 0x37 |
||||
#define CS3_SW4 0x38 |
||||
#define CS4_SW4 0x39 |
||||
#define CS5_SW4 0x3A |
||||
#define CS6_SW4 0x3B |
||||
#define CS7_SW4 0x3C |
||||
#define CS8_SW4 0x3D |
||||
#define CS9_SW4 0x3E |
||||
#define CS10_SW4 0x3F |
||||
#define CS11_SW4 0x40 |
||||
#define CS12_SW4 0x41 |
||||
#define CS13_SW4 0x42 |
||||
#define CS14_SW4 0x43 |
||||
#define CS15_SW4 0x44 |
||||
#define CS16_SW4 0x45 |
||||
#define CS17_SW4 0x46 |
||||
#define CS18_SW4 0x47 |
||||
#define CS1_SW5 0x48 |
||||
#define CS2_SW5 0x49 |
||||
#define CS3_SW5 0x4A |
||||
#define CS4_SW5 0x4B |
||||
#define CS5_SW5 0x4C |
||||
#define CS6_SW5 0x4D |
||||
#define CS7_SW5 0x4E |
||||
#define CS8_SW5 0x4F |
||||
#define CS9_SW5 0x50 |
||||
#define CS10_SW5 0x51 |
||||
#define CS11_SW5 0x52 |
||||
#define CS12_SW5 0x53 |
||||
#define CS13_SW5 0x54 |
||||
#define CS14_SW5 0x55 |
||||
#define CS15_SW5 0x56 |
||||
#define CS16_SW5 0x57 |
||||
#define CS17_SW5 0x58 |
||||
#define CS18_SW5 0x59 |
||||
#define CS1_SW6 0x5A |
||||
#define CS2_SW6 0x5B |
||||
#define CS3_SW6 0x5C |
||||
#define CS4_SW6 0x5D |
||||
#define CS5_SW6 0x5E |
||||
#define CS6_SW6 0x5F |
||||
#define CS7_SW6 0x60 |
||||
#define CS8_SW6 0x61 |
||||
#define CS9_SW6 0x62 |
||||
#define CS10_SW6 0x63 |
||||
#define CS11_SW6 0x64 |
||||
#define CS12_SW6 0x65 |
||||
#define CS13_SW6 0x66 |
||||
#define CS14_SW6 0x67 |
||||
#define CS15_SW6 0x68 |
||||
#define CS16_SW6 0x69 |
||||
#define CS17_SW6 0x6A |
||||
#define CS18_SW6 0x6B |
||||
#define CS1_SW7 0x6C |
||||
#define CS2_SW7 0x6D |
||||
#define CS3_SW7 0x6E |
||||
#define CS4_SW7 0x6F |
||||
#define CS5_SW7 0x70 |
||||
#define CS6_SW7 0x71 |
||||
#define CS7_SW7 0x72 |
||||
#define CS8_SW7 0x73 |
||||
#define CS9_SW7 0x74 |
||||
#define CS10_SW7 0x75 |
||||
#define CS11_SW7 0x76 |
||||
#define CS12_SW7 0x77 |
||||
#define CS13_SW7 0x78 |
||||
#define CS14_SW7 0x79 |
||||
#define CS15_SW7 0x7A |
||||
#define CS16_SW7 0x7B |
||||
#define CS17_SW7 0x7C |
||||
#define CS18_SW7 0x7D |
||||
#define CS1_SW8 0x7E |
||||
#define CS2_SW8 0x7F |
||||
#define CS3_SW8 0x80 |
||||
#define CS4_SW8 0x81 |
||||
#define CS5_SW8 0x82 |
||||
#define CS6_SW8 0x83 |
||||
#define CS7_SW8 0x84 |
||||
#define CS8_SW8 0x85 |
||||
#define CS9_SW8 0x86 |
||||
#define CS10_SW8 0x87 |
||||
#define CS11_SW8 0x88 |
||||
#define CS12_SW8 0x89 |
||||
#define CS13_SW8 0x8A |
||||
#define CS14_SW8 0x8B |
||||
#define CS15_SW8 0x8C |
||||
#define CS16_SW8 0x8D |
||||
#define CS17_SW8 0x8E |
||||
#define CS18_SW8 0x8F |
||||
#define CS1_SW9 0x90 |
||||
#define CS2_SW9 0x91 |
||||
#define CS3_SW9 0x92 |
||||
#define CS4_SW9 0x93 |
||||
#define CS5_SW9 0x94 |
||||
#define CS6_SW9 0x95 |
||||
#define CS7_SW9 0x96 |
||||
#define CS8_SW9 0x97 |
||||
#define CS9_SW9 0x98 |
||||
#define CS10_SW9 0x99 |
||||
#define CS11_SW9 0x9A |
||||
#define CS12_SW9 0x9B |
||||
#define CS13_SW9 0x9C |
||||
#define CS14_SW9 0x9D |
||||
#define CS15_SW9 0x9E |
||||
#define CS16_SW9 0x9F |
||||
#define CS17_SW9 0xA0 |
||||
#define CS18_SW9 0xA1 |
||||
#define CS1_SW10 0xA2 |
||||
#define CS2_SW10 0xA3 |
||||
#define CS3_SW10 0xA4 |
||||
#define CS4_SW10 0xA5 |
||||
#define CS5_SW10 0xA6 |
||||
#define CS6_SW10 0xA7 |
||||
#define CS7_SW10 0xA8 |
||||
#define CS8_SW10 0xA9 |
||||
#define CS9_SW10 0xAA |
||||
#define CS10_SW10 0xAB |
||||
#define CS11_SW10 0xAC |
||||
#define CS12_SW10 0xAD |
||||
#define CS13_SW10 0xAE |
||||
#define CS14_SW10 0xAF |
||||
#define CS15_SW10 0xB0 |
||||
#define CS16_SW10 0xB1 |
||||
#define CS17_SW10 0xB2 |
||||
#define CS18_SW10 0xB3 |
||||
#define CS1_SW11 0xB4 |
||||
#define CS2_SW11 0xB5 |
||||
#define CS3_SW11 0xB6 |
||||
#define CS4_SW11 0xB7 |
||||
#define CS5_SW11 0xB8 |
||||
#define CS6_SW11 0xB9 |
||||
#define CS7_SW11 0xBA |
||||
#define CS8_SW11 0xBB |
||||
#define CS9_SW11 0xBC |
||||
#define CS10_SW11 0xBD |
||||
#define CS11_SW11 0xBE |
||||
#define CS12_SW11 0xBF |
||||
#define CS13_SW11 0xC0 |
||||
#define CS14_SW11 0xC1 |
||||
#define CS15_SW11 0xC2 |
||||
#define CS16_SW11 0xC3 |
||||
#define CS17_SW11 0xC4 |
||||
#define CS18_SW11 0xC5 |
||||
#define CS1_SW12 0xC6 |
||||
#define CS2_SW12 0xC7 |
||||
#define CS3_SW12 0xC8 |
||||
#define CS4_SW12 0xC9 |
||||
#define CS5_SW12 0xCA |
||||
#define CS6_SW12 0xCB |
||||
#define CS7_SW12 0xCC |
||||
#define CS8_SW12 0xCD |
||||
#define CS9_SW12 0xCE |
||||
#define CS10_SW12 0xCF |
||||
#define CS11_SW12 0xD0 |
||||
#define CS12_SW12 0xD1 |
||||
#define CS13_SW12 0xD2 |
||||
#define CS14_SW12 0xD3 |
||||
#define CS15_SW12 0xD4 |
||||
#define CS16_SW12 0xD5 |
||||
#define CS17_SW12 0xD6 |
||||
#define CS18_SW12 0xD7 |
||||
@ -0,0 +1,7 @@ |
||||
#pragma once |
||||
|
||||
#define HAL_USE_SPI TRUE |
||||
#define SPI_USE_WAIT TRUE |
||||
#define SPI_SELECT_MODE SPI_SELECT_MODE_PAD |
||||
|
||||
#include_next <halconf.h> |
||||
@ -0,0 +1,6 @@ |
||||
#pragma once |
||||
|
||||
#include_next <mcuconf.h> |
||||
|
||||
#undef STM32_SPI_USE_SPI1 |
||||
#define STM32_SPI_USE_SPI1 TRUE |
||||
Loading…
Reference in new issue