aboutsummaryrefslogtreecommitdiff
path: root/drivers/avr
diff options
context:
space:
mode:
authoryiancar <yiangosyiangou@cytanet.com.cy>2018-08-15 08:19:38 +0300
committerJack Humbert <jack.humb@gmail.com>2018-08-15 01:19:38 -0400
commitad2bb529c795be066b279f52bebec03257992fc2 (patch)
treeaddf2d3b665316c985f0b0d1c524a60977d3ed72 /drivers/avr
parentfeec8ad4694814ed8953c6f72798f51f39724af5 (diff)
downloadqmk_firmware-ad2bb529c795be066b279f52bebec03257992fc2.tar.gz
qmk_firmware-ad2bb529c795be066b279f52bebec03257992fc2.zip
Rgb matrix arm (#3648)
* Addition of I2C master driver for STM32, Generalization of ISSI3731 driver - Addition of an i2c_master driver for STM32 to replicate expectations of AVR driver. - Moved ISSI3731 driver one level up to make it accesible by both architectures. - Renamed ISSI3731 functions to a more general name for preparation of other ISSI drivers. - Added compiler directives where necessary to differenciate each architecture. * converted tabs to spaces
Diffstat (limited to 'drivers/avr')
-rw-r--r--drivers/avr/is31fl3731.c262
-rw-r--r--drivers/avr/is31fl3731.h214
2 files changed, 0 insertions, 476 deletions
diff --git a/drivers/avr/is31fl3731.c b/drivers/avr/is31fl3731.c
deleted file mode 100644
index 70813464b..000000000
--- a/drivers/avr/is31fl3731.c
+++ /dev/null
@@ -1,262 +0,0 @@
1/* Copyright 2017 Jason Williams
2 * Copyright 2018 Jack Humbert
3 *
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17
18#include "is31fl3731.h"
19#include <avr/interrupt.h>
20#include <avr/io.h>
21#include <util/delay.h>
22#include <string.h>
23#include "i2c_master.h"
24#include "progmem.h"
25
26// This is a 7-bit address, that gets left-shifted and bit 0
27// set to 0 for write, 1 for read (as per I2C protocol)
28// The address will vary depending on your wiring:
29// 0b1110100 AD <-> GND
30// 0b1110111 AD <-> VCC
31// 0b1110101 AD <-> SCL
32// 0b1110110 AD <-> SDA
33#define ISSI_ADDR_DEFAULT 0x74
34
35#define ISSI_REG_CONFIG 0x00
36#define ISSI_REG_CONFIG_PICTUREMODE 0x00
37#define ISSI_REG_CONFIG_AUTOPLAYMODE 0x08
38#define ISSI_REG_CONFIG_AUDIOPLAYMODE 0x18
39
40#define ISSI_CONF_PICTUREMODE 0x00
41#define ISSI_CONF_AUTOFRAMEMODE 0x04
42#define ISSI_CONF_AUDIOMODE 0x08
43
44#define ISSI_REG_PICTUREFRAME 0x01
45
46#define ISSI_REG_SHUTDOWN 0x0A
47#define ISSI_REG_AUDIOSYNC 0x06
48
49#define ISSI_COMMANDREGISTER 0xFD
50#define ISSI_BANK_FUNCTIONREG 0x0B // helpfully called 'page nine'
51
52#ifndef ISSI_TIMEOUT
53 #define ISSI_TIMEOUT 100
54#endif
55
56#ifndef ISSI_PERSISTENCE
57 #define ISSI_PERSISTENCE 0
58#endif
59
60// Transfer buffer for TWITransmitData()
61uint8_t g_twi_transfer_buffer[20];
62
63// These buffers match the IS31FL3731 PWM registers 0x24-0xB3.
64// Storing them like this is optimal for I2C transfers to the registers.
65// We could optimize this and take out the unused registers from these
66// buffers and the transfers in IS31FL3731_write_pwm_buffer() but it's
67// probably not worth the extra complexity.
68uint8_t g_pwm_buffer[DRIVER_COUNT][144];
69bool g_pwm_buffer_update_required = false;
70
71uint8_t g_led_control_registers[DRIVER_COUNT][18] = { { 0 }, { 0 } };
72bool g_led_control_registers_update_required = false;
73
74// This is the bit pattern in the LED control registers
75// (for matrix A, add one to register for matrix B)
76//
77// reg - b7 b6 b5 b4 b3 b2 b1 b0
78// 0x00 - R08,R07,R06,R05,R04,R03,R02,R01
79// 0x02 - G08,G07,G06,G05,G04,G03,G02,R00
80// 0x04 - B08,B07,B06,B05,B04,B03,G01,G00
81// 0x06 - - , - , - , - , - ,B02,B01,B00
82// 0x08 - - , - , - , - , - , - , - , -
83// 0x0A - B17,B16,B15, - , - , - , - , -
84// 0x0C - G17,G16,B14,B13,B12,B11,B10,B09
85// 0x0E - R17,G15,G14,G13,G12,G11,G10,G09
86// 0x10 - R16,R15,R14,R13,R12,R11,R10,R09
87
88
89void IS31FL3731_write_register( uint8_t addr, uint8_t reg, uint8_t data )
90{
91 g_twi_transfer_buffer[0] = reg;
92 g_twi_transfer_buffer[1] = data;
93
94 #if ISSI_PERSISTENCE > 0
95 for (uint8_t i = 0; i < ISSI_PERSISTENCE; i++) {
96 if (i2c_transmit(addr << 1, g_twi_transfer_buffer, 2, ISSI_TIMEOUT) == 0)
97 break;
98 }
99 #else
100 i2c_transmit(addr << 1, g_twi_transfer_buffer, 2, ISSI_TIMEOUT);
101 #endif
102}
103
104void IS31FL3731_write_pwm_buffer( uint8_t addr, uint8_t *pwm_buffer )
105{
106 // assumes bank is already selected
107
108 // transmit PWM registers in 9 transfers of 16 bytes
109 // g_twi_transfer_buffer[] is 20 bytes
110
111 // iterate over the pwm_buffer contents at 16 byte intervals
112 for ( int i = 0; i < 144; i += 16 ) {
113 // set the first register, e.g. 0x24, 0x34, 0x44, etc.
114 g_twi_transfer_buffer[0] = 0x24 + i;
115 // copy the data from i to i+15
116 // device will auto-increment register for data after the first byte
117 // thus this sets registers 0x24-0x33, 0x34-0x43, etc. in one transfer
118 for ( int j = 0; j < 16; j++ ) {
119 g_twi_transfer_buffer[1 + j] = pwm_buffer[i + j];
120 }
121
122 #if ISSI_PERSISTENCE > 0
123 for (uint8_t i = 0; i < ISSI_PERSISTENCE; i++) {
124 if (i2c_transmit(addr << 1, g_twi_transfer_buffer, 17, ISSI_TIMEOUT) == 0)
125 break;
126 }
127 #else
128 i2c_transmit(addr << 1, g_twi_transfer_buffer, 17, ISSI_TIMEOUT);
129 #endif
130 }
131}
132
133void IS31FL3731_init( uint8_t addr )
134{
135 // In order to avoid the LEDs being driven with garbage data
136 // in the LED driver's PWM registers, first enable software shutdown,
137 // then set up the mode and other settings, clear the PWM registers,
138 // then disable software shutdown.
139
140 // select "function register" bank
141 IS31FL3731_write_register( addr, ISSI_COMMANDREGISTER, ISSI_BANK_FUNCTIONREG );
142
143 // enable software shutdown
144 IS31FL3731_write_register( addr, ISSI_REG_SHUTDOWN, 0x00 );
145 // this delay was copied from other drivers, might not be needed
146 _delay_ms( 10 );
147
148 // picture mode
149 IS31FL3731_write_register( addr, ISSI_REG_CONFIG, ISSI_REG_CONFIG_PICTUREMODE );
150 // display frame 0
151 IS31FL3731_write_register( addr, ISSI_REG_PICTUREFRAME, 0x00 );
152 // audio sync off
153 IS31FL3731_write_register( addr, ISSI_REG_AUDIOSYNC, 0x00 );
154
155 // select bank 0
156 IS31FL3731_write_register( addr, ISSI_COMMANDREGISTER, 0 );
157
158 // turn off all LEDs in the LED control register
159 for ( int i = 0x00; i <= 0x11; i++ )
160 {
161 IS31FL3731_write_register( addr, i, 0x00 );
162 }
163
164 // turn off all LEDs in the blink control register (not really needed)
165 for ( int i = 0x12; i <= 0x23; i++ )
166 {
167 IS31FL3731_write_register( addr, i, 0x00 );
168 }
169
170 // set PWM on all LEDs to 0
171 for ( int i = 0x24; i <= 0xB3; i++ )
172 {
173 IS31FL3731_write_register( addr, i, 0x00 );
174 }
175
176 // select "function register" bank
177 IS31FL3731_write_register( addr, ISSI_COMMANDREGISTER, ISSI_BANK_FUNCTIONREG );
178
179 // disable software shutdown
180 IS31FL3731_write_register( addr, ISSI_REG_SHUTDOWN, 0x01 );
181
182 // select bank 0 and leave it selected.
183 // most usage after initialization is just writing PWM buffers in bank 0
184 // as there's not much point in double-buffering
185 IS31FL3731_write_register( addr, ISSI_COMMANDREGISTER, 0 );
186
187}
188
189void IS31FL3731_set_color( int index, uint8_t red, uint8_t green, uint8_t blue )
190{
191 if ( index >= 0 && index < DRIVER_LED_TOTAL ) {
192 is31_led led = g_is31_leds[index];
193
194 // Subtract 0x24 to get the second index of g_pwm_buffer
195 g_pwm_buffer[led.driver][led.r - 0x24] = red;
196 g_pwm_buffer[led.driver][led.g - 0x24] = green;
197 g_pwm_buffer[led.driver][led.b - 0x24] = blue;
198 g_pwm_buffer_update_required = true;
199 }
200}
201
202void IS31FL3731_set_color_all( uint8_t red, uint8_t green, uint8_t blue )
203{
204 for ( int i = 0; i < DRIVER_LED_TOTAL; i++ )
205 {
206 IS31FL3731_set_color( i, red, green, blue );
207 }
208}
209
210void IS31FL3731_set_led_control_register( uint8_t index, bool red, bool green, bool blue )
211{
212 is31_led led = g_is31_leds[index];
213
214 uint8_t control_register_r = (led.r - 0x24) / 8;
215 uint8_t control_register_g = (led.g - 0x24) / 8;
216 uint8_t control_register_b = (led.b - 0x24) / 8;
217 uint8_t bit_r = (led.r - 0x24) % 8;
218 uint8_t bit_g = (led.g - 0x24) % 8;
219 uint8_t bit_b = (led.b - 0x24) % 8;
220
221 if ( red ) {
222 g_led_control_registers[led.driver][control_register_r] |= (1 << bit_r);
223 } else {
224 g_led_control_registers[led.driver][control_register_r] &= ~(1 << bit_r);
225 }
226 if ( green ) {
227 g_led_control_registers[led.driver][control_register_g] |= (1 << bit_g);
228 } else {
229 g_led_control_registers[led.driver][control_register_g] &= ~(1 << bit_g);
230 }
231 if ( blue ) {
232 g_led_control_registers[led.driver][control_register_b] |= (1 << bit_b);
233 } else {
234 g_led_control_registers[led.driver][control_register_b] &= ~(1 << bit_b);
235 }
236
237 g_led_control_registers_update_required = true;
238
239}
240
241void IS31FL3731_update_pwm_buffers( uint8_t addr1, uint8_t addr2 )
242{
243 if ( g_pwm_buffer_update_required )
244 {
245 IS31FL3731_write_pwm_buffer( addr1, g_pwm_buffer[0] );
246 IS31FL3731_write_pwm_buffer( addr2, g_pwm_buffer[1] );
247 }
248 g_pwm_buffer_update_required = false;
249}
250
251void IS31FL3731_update_led_control_registers( uint8_t addr1, uint8_t addr2 )
252{
253 if ( g_led_control_registers_update_required )
254 {
255 for ( int i=0; i<18; i++ )
256 {
257 IS31FL3731_write_register(addr1, i, g_led_control_registers[0][i] );
258 IS31FL3731_write_register(addr2, i, g_led_control_registers[1][i] );
259 }
260 }
261}
262
diff --git a/drivers/avr/is31fl3731.h b/drivers/avr/is31fl3731.h
deleted file mode 100644
index 3d30fc67b..000000000
--- a/drivers/avr/is31fl3731.h
+++ /dev/null
@@ -1,214 +0,0 @@
1/* Copyright 2017 Jason Williams
2 * Copyright 2018 Jack Humbert
3 *
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17
18
19#ifndef IS31FL3731_DRIVER_H
20#define IS31FL3731_DRIVER_H
21
22#include <stdint.h>
23#include <stdbool.h>
24
25typedef struct is31_led {
26 uint8_t driver:2;
27 uint8_t r;
28 uint8_t g;
29 uint8_t b;
30} __attribute__((packed)) is31_led;
31
32extern const is31_led g_is31_leds[DRIVER_LED_TOTAL];
33
34void IS31FL3731_init( uint8_t addr );
35void IS31FL3731_write_register( uint8_t addr, uint8_t reg, uint8_t data );
36void IS31FL3731_write_pwm_buffer( uint8_t addr, uint8_t *pwm_buffer );
37
38void IS31FL3731_set_color( int index, uint8_t red, uint8_t green, uint8_t blue );
39void IS31FL3731_set_color_all( uint8_t red, uint8_t green, uint8_t blue );
40
41void IS31FL3731_set_led_control_register( uint8_t index, bool red, bool green, bool blue );
42
43// This should not be called from an interrupt
44// (eg. from a timer interrupt).
45// Call this while idle (in between matrix scans).
46// If the buffer is dirty, it will update the driver with the buffer.
47void IS31FL3731_update_pwm_buffers( uint8_t addr1, uint8_t addr2 );
48void IS31FL3731_update_led_control_registers( uint8_t addr1, uint8_t addr2 );
49
50#define C1_1 0x24
51#define C1_2 0x25
52#define C1_3 0x26
53#define C1_4 0x27
54#define C1_5 0x28
55#define C1_6 0x29
56#define C1_7 0x2A
57#define C1_8 0x2B
58
59#define C1_9 0x2C
60#define C1_10 0x2D
61#define C1_11 0x2E
62#define C1_12 0x2F
63#define C1_13 0x30
64#define C1_14 0x31
65#define C1_15 0x32
66#define C1_16 0x33
67
68#define C2_1 0x34
69#define C2_2 0x35
70#define C2_3 0x36
71#define C2_4 0x37
72#define C2_5 0x38
73#define C2_6 0x39
74#define C2_7 0x3A
75#define C2_8 0x3B
76
77#define C2_9 0x3C
78#define C2_10 0x3D
79#define C2_11 0x3E
80#define C2_12 0x3F
81#define C2_13 0x40
82#define C2_14 0x41
83#define C2_15 0x42
84#define C2_16 0x43
85
86#define C3_1 0x44
87#define C3_2 0x45
88#define C3_3 0x46
89#define C3_4 0x47
90#define C3_5 0x48
91#define C3_6 0x49
92#define C3_7 0x4A
93#define C3_8 0x4B
94
95#define C3_9 0x4C
96#define C3_10 0x4D
97#define C3_11 0x4E
98#define C3_12 0x4F
99#define C3_13 0x50
100#define C3_14 0x51
101#define C3_15 0x52
102#define C3_16 0x53
103
104#define C4_1 0x54
105#define C4_2 0x55
106#define C4_3 0x56
107#define C4_4 0x57
108#define C4_5 0x58
109#define C4_6 0x59
110#define C4_7 0x5A
111#define C4_8 0x5B
112
113#define C4_9 0x5C
114#define C4_10 0x5D
115#define C4_11 0x5E
116#define C4_12 0x5F
117#define C4_13 0x60
118#define C4_14 0x61
119#define C4_15 0x62
120#define C4_16 0x63
121
122#define C5_1 0x64
123#define C5_2 0x65
124#define C5_3 0x66
125#define C5_4 0x67
126#define C5_5 0x68
127#define C5_6 0x69
128#define C5_7 0x6A
129#define C5_8 0x6B
130
131#define C5_9 0x6C
132#define C5_10 0x6D
133#define C5_11 0x6E
134#define C5_12 0x6F
135#define C5_13 0x70
136#define C5_14 0x71
137#define C5_15 0x72
138#define C5_16 0x73
139
140#define C6_1 0x74
141#define C6_2 0x75
142#define C6_3 0x76
143#define C6_4 0x77
144#define C6_5 0x78
145#define C6_6 0x79
146#define C6_7 0x7A
147#define C6_8 0x7B
148
149#define C6_9 0x7C
150#define C6_10 0x7D
151#define C6_11 0x7E
152#define C6_12 0x7F
153#define C6_13 0x80
154#define C6_14 0x81
155#define C6_15 0x82
156#define C6_16 0x83
157
158#define C7_1 0x84
159#define C7_2 0x85
160#define C7_3 0x86
161#define C7_4 0x87
162#define C7_5 0x88
163#define C7_6 0x89
164#define C7_7 0x8A
165#define C7_8 0x8B
166
167#define C7_9 0x8C
168#define C7_10 0x8D
169#define C7_11 0x8E
170#define C7_12 0x8F
171#define C7_13 0x90
172#define C7_14 0x91
173#define C7_15 0x92
174#define C7_16 0x93
175
176#define C8_1 0x94
177#define C8_2 0x95
178#define C8_3 0x96
179#define C8_4 0x97
180#define C8_5 0x98
181#define C8_6 0x99
182#define C8_7 0x9A
183#define C8_8 0x9B
184
185#define C8_9 0x9C
186#define C8_10 0x9D
187#define C8_11 0x9E
188#define C8_12 0x9F
189#define C8_13 0xA0
190#define C8_14 0xA1
191#define C8_15 0xA2
192#define C8_16 0xA3
193
194#define C9_1 0xA4
195#define C9_2 0xA5
196#define C9_3 0xA6
197#define C9_4 0xA7
198#define C9_5 0xA8
199#define C9_6 0xA9
200#define C9_7 0xAA
201#define C9_8 0xAB
202
203#define C9_9 0xAC
204#define C9_10 0xAD
205#define C9_11 0xAE
206#define C9_12 0xAF
207#define C9_13 0xB0
208#define C9_14 0xB1
209#define C9_15 0xB2
210#define C9_16 0xB3
211
212
213
214#endif // IS31FL3731_DRIVER_H