]> git.donarmstrong.com Git - qmk_firmware.git/blob - quantum/split_common/transport.c
Fix i2c splits with >8 columns (#5454)
[qmk_firmware.git] / quantum / split_common / transport.c
1 #include <string.h>
2
3 #include "config.h"
4 #include "matrix.h"
5 #include "quantum.h"
6
7 #define ROWS_PER_HAND (MATRIX_ROWS / 2)
8
9 #ifdef RGBLIGHT_ENABLE
10 #  include "rgblight.h"
11 #endif
12
13 #ifdef BACKLIGHT_ENABLE
14 #  include "backlight.h"
15 extern backlight_config_t backlight_config;
16 #endif
17
18 #if defined(USE_I2C) || defined(EH)
19
20 #  include "i2c_master.h"
21 #  include "i2c_slave.h"
22
23 #  define I2C_BACKLIT_START 0x00
24 // Need 4 bytes for RGB (32 bit)
25 #  define I2C_RGB_START 0x01
26 #  define I2C_KEYMAP_START 0x05
27
28 #  define TIMEOUT 100
29
30 #  ifndef SLAVE_I2C_ADDRESS
31 #    define SLAVE_I2C_ADDRESS 0x32
32 #  endif
33
34 // Get rows from other half over i2c
35 bool transport_master(matrix_row_t matrix[]) {
36   i2c_readReg(SLAVE_I2C_ADDRESS, I2C_KEYMAP_START, (void *)matrix, ROWS_PER_HAND * sizeof(matrix_row_t), TIMEOUT);
37
38   // write backlight info
39 #  ifdef BACKLIGHT_ENABLE
40   static uint8_t prev_level = ~0;
41   uint8_t        level      = get_backlight_level();
42   if (level != prev_level) {
43     if (i2c_writeReg(SLAVE_I2C_ADDRESS, I2C_BACKLIT_START, (void *)&level, sizeof(level), TIMEOUT) >= 0) {
44       prev_level = level;
45     }
46   }
47 #  endif
48
49 #  ifdef RGBLIGHT_ENABLE
50   static uint32_t prev_rgb = ~0;
51   uint32_t        rgb      = rgblight_read_dword();
52   if (rgb != prev_rgb) {
53     if (i2c_writeReg(SLAVE_I2C_ADDRESS, I2C_RGB_START, (void *)&rgb, sizeof(rgb), TIMEOUT) >= 0) {
54       prev_rgb = rgb;
55     }
56   }
57 #  endif
58
59   return true;
60 }
61
62 void transport_slave(matrix_row_t matrix[]) {
63   // Copy matrix to I2C buffer
64   memcpy((void*)(i2c_slave_reg + I2C_KEYMAP_START), (void *)matrix, ROWS_PER_HAND * sizeof(matrix_row_t) );
65
66 // Read Backlight Info
67 #  ifdef BACKLIGHT_ENABLE
68   backlight_set(i2c_slave_reg[I2C_BACKLIT_START]);
69 #  endif
70
71 #  ifdef RGBLIGHT_ENABLE
72   uint32_t rgb = *(uint32_t *)(i2c_slave_reg + I2C_RGB_START);
73   // Update the RGB with the new data
74   rgblight_update_dword(rgb);
75 #  endif
76 }
77
78 void transport_master_init(void) { i2c_init(); }
79
80 void transport_slave_init(void) { i2c_slave_init(SLAVE_I2C_ADDRESS); }
81
82 #else  // USE_SERIAL
83
84 #  include "serial.h"
85
86 typedef struct _Serial_s2m_buffer_t {
87   // TODO: if MATRIX_COLS > 8 change to uint8_t packed_matrix[] for pack/unpack
88   matrix_row_t smatrix[ROWS_PER_HAND];
89 } Serial_s2m_buffer_t;
90
91 typedef struct _Serial_m2s_buffer_t {
92 #  ifdef BACKLIGHT_ENABLE
93   uint8_t           backlight_level;
94 #  endif
95 #  if defined(RGBLIGHT_ENABLE) && defined(RGBLIGHT_SPLIT)
96   rgblight_config_t rgblight_config;  // not yet use
97   //
98   // When MCUs on both sides drive their respective RGB LED chains,
99   // it is necessary to synchronize, so it is necessary to communicate RGB
100   // information. In that case, define the RGBLIGHT_SPLIT macro.
101   //
102   // Otherwise, if the master side MCU drives both sides RGB LED chains,
103   // there is no need to communicate.
104 #  endif
105 } Serial_m2s_buffer_t;
106
107 volatile Serial_s2m_buffer_t serial_s2m_buffer = {};
108 volatile Serial_m2s_buffer_t serial_m2s_buffer = {};
109 uint8_t volatile status0                       = 0;
110
111 SSTD_t transactions[] = {
112     {
113         (uint8_t *)&status0,
114         sizeof(serial_m2s_buffer),
115         (uint8_t *)&serial_m2s_buffer,
116         sizeof(serial_s2m_buffer),
117         (uint8_t *)&serial_s2m_buffer,
118     },
119 };
120
121 void transport_master_init(void) { soft_serial_initiator_init(transactions, TID_LIMIT(transactions)); }
122
123 void transport_slave_init(void) { soft_serial_target_init(transactions, TID_LIMIT(transactions)); }
124
125 bool transport_master(matrix_row_t matrix[]) {
126   if (soft_serial_transaction()) {
127     return false;
128   }
129
130   // TODO:  if MATRIX_COLS > 8 change to unpack()
131   for (int i = 0; i < ROWS_PER_HAND; ++i) {
132     matrix[i] = serial_s2m_buffer.smatrix[i];
133   }
134
135 #  if defined(RGBLIGHT_ENABLE) && defined(RGBLIGHT_SPLIT)
136   // Code to send RGB over serial goes here (not implemented yet)
137 #  endif
138
139 #  ifdef BACKLIGHT_ENABLE
140   // Write backlight level for slave to read
141   serial_m2s_buffer.backlight_level = backlight_config.enable ? backlight_config.level : 0;
142 #  endif
143
144   return true;
145 }
146
147 void transport_slave(matrix_row_t matrix[]) {
148   // TODO: if MATRIX_COLS > 8 change to pack()
149   for (int i = 0; i < ROWS_PER_HAND; ++i) {
150     serial_s2m_buffer.smatrix[i] = matrix[i];
151   }
152 #  ifdef BACKLIGHT_ENABLE
153   backlight_set(serial_m2s_buffer.backlight_level);
154 #  endif
155 #  if defined(RGBLIGHT_ENABLE) && defined(RGBLIGHT_SPLIT)
156 // Add serial implementation for RGB here
157 #  endif
158 }
159
160 #endif