diff options
Diffstat (limited to 'tmk_core/protocol/lufa')
| -rw-r--r-- | tmk_core/protocol/lufa/adafruit_ble.cpp | 805 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/adafruit_ble.h | 60 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/descriptor.c | 98 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/descriptor.h | 35 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/lufa.c | 305 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/lufa.h | 17 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/outputselect.c | 56 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/outputselect.h | 40 | ||||
| -rw-r--r-- | tmk_core/protocol/lufa/ringbuffer.hpp | 66 |
9 files changed, 1407 insertions, 75 deletions
diff --git a/tmk_core/protocol/lufa/adafruit_ble.cpp b/tmk_core/protocol/lufa/adafruit_ble.cpp new file mode 100644 index 000000000..fd6edd42c --- /dev/null +++ b/tmk_core/protocol/lufa/adafruit_ble.cpp | |||
| @@ -0,0 +1,805 @@ | |||
| 1 | #include "adafruit_ble.h" | ||
| 2 | #include <stdio.h> | ||
| 3 | #include <stdlib.h> | ||
| 4 | #include <alloca.h> | ||
| 5 | #include <util/delay.h> | ||
| 6 | #include <util/atomic.h> | ||
| 7 | #include "debug.h" | ||
| 8 | #include "pincontrol.h" | ||
| 9 | #include "timer.h" | ||
| 10 | #include "action_util.h" | ||
| 11 | #include "ringbuffer.hpp" | ||
| 12 | #include <string.h> | ||
| 13 | |||
| 14 | // These are the pin assignments for the 32u4 boards. | ||
| 15 | // You may define them to something else in your config.h | ||
| 16 | // if yours is wired up differently. | ||
| 17 | #ifndef AdafruitBleResetPin | ||
| 18 | #define AdafruitBleResetPin D4 | ||
| 19 | #endif | ||
| 20 | |||
| 21 | #ifndef AdafruitBleCSPin | ||
| 22 | #define AdafruitBleCSPin B4 | ||
| 23 | #endif | ||
| 24 | |||
| 25 | #ifndef AdafruitBleIRQPin | ||
| 26 | #define AdafruitBleIRQPin E6 | ||
| 27 | #endif | ||
| 28 | |||
| 29 | |||
| 30 | #define SAMPLE_BATTERY | ||
| 31 | #define ConnectionUpdateInterval 1000 /* milliseconds */ | ||
| 32 | |||
| 33 | static struct { | ||
| 34 | bool is_connected; | ||
| 35 | bool initialized; | ||
| 36 | bool configured; | ||
| 37 | |||
| 38 | #define ProbedEvents 1 | ||
| 39 | #define UsingEvents 2 | ||
| 40 | bool event_flags; | ||
| 41 | |||
| 42 | #ifdef SAMPLE_BATTERY | ||
| 43 | uint16_t last_battery_update; | ||
| 44 | uint32_t vbat; | ||
| 45 | #endif | ||
| 46 | uint16_t last_connection_update; | ||
| 47 | } state; | ||
| 48 | |||
| 49 | // Commands are encoded using SDEP and sent via SPI | ||
| 50 | // https://github.com/adafruit/Adafruit_BluefruitLE_nRF51/blob/master/SDEP.md | ||
| 51 | |||
| 52 | #define SdepMaxPayload 16 | ||
| 53 | struct sdep_msg { | ||
| 54 | uint8_t type; | ||
| 55 | uint8_t cmd_low; | ||
| 56 | uint8_t cmd_high; | ||
| 57 | struct __attribute__((packed)) { | ||
| 58 | uint8_t len:7; | ||
| 59 | uint8_t more:1; | ||
| 60 | }; | ||
| 61 | uint8_t payload[SdepMaxPayload]; | ||
| 62 | } __attribute__((packed)); | ||
| 63 | |||
| 64 | // The recv latency is relatively high, so when we're hammering keys quickly, | ||
| 65 | // we want to avoid waiting for the responses in the matrix loop. We maintain | ||
| 66 | // a short queue for that. Since there is quite a lot of space overhead for | ||
| 67 | // the AT command representation wrapped up in SDEP, we queue the minimal | ||
| 68 | // information here. | ||
| 69 | |||
| 70 | enum queue_type { | ||
| 71 | QTKeyReport, // 1-byte modifier + 6-byte key report | ||
| 72 | QTConsumer, // 16-bit key code | ||
| 73 | #ifdef MOUSE_ENABLE | ||
| 74 | QTMouseMove, // 4-byte mouse report | ||
| 75 | #endif | ||
| 76 | }; | ||
| 77 | |||
| 78 | struct queue_item { | ||
| 79 | enum queue_type queue_type; | ||
| 80 | uint16_t added; | ||
| 81 | union __attribute__((packed)) { | ||
| 82 | struct __attribute__((packed)) { | ||
| 83 | uint8_t modifier; | ||
| 84 | uint8_t keys[6]; | ||
| 85 | } key; | ||
| 86 | |||
| 87 | uint16_t consumer; | ||
| 88 | struct __attribute__((packed)) { | ||
| 89 | int8_t x, y, scroll, pan; | ||
| 90 | } mousemove; | ||
| 91 | }; | ||
| 92 | }; | ||
| 93 | |||
| 94 | // Items that we wish to send | ||
| 95 | static RingBuffer<queue_item, 40> send_buf; | ||
| 96 | // Pending response; while pending, we can't send any more requests. | ||
| 97 | // This records the time at which we sent the command for which we | ||
| 98 | // are expecting a response. | ||
| 99 | static RingBuffer<uint16_t, 2> resp_buf; | ||
| 100 | |||
| 101 | static bool process_queue_item(struct queue_item *item, uint16_t timeout); | ||
| 102 | |||
| 103 | enum sdep_type { | ||
| 104 | SdepCommand = 0x10, | ||
| 105 | SdepResponse = 0x20, | ||
| 106 | SdepAlert = 0x40, | ||
| 107 | SdepError = 0x80, | ||
| 108 | SdepSlaveNotReady = 0xfe, // Try again later | ||
| 109 | SdepSlaveOverflow = 0xff, // You read more data than is available | ||
| 110 | }; | ||
| 111 | |||
| 112 | enum ble_cmd { | ||
| 113 | BleInitialize = 0xbeef, | ||
| 114 | BleAtWrapper = 0x0a00, | ||
| 115 | BleUartTx = 0x0a01, | ||
| 116 | BleUartRx = 0x0a02, | ||
| 117 | }; | ||
| 118 | |||
| 119 | enum ble_system_event_bits { | ||
| 120 | BleSystemConnected = 0, | ||
| 121 | BleSystemDisconnected = 1, | ||
| 122 | BleSystemUartRx = 8, | ||
| 123 | BleSystemMidiRx = 10, | ||
| 124 | }; | ||
| 125 | |||
| 126 | // The SDEP.md file says 2MHz but the web page and the sample driver | ||
| 127 | // both use 4MHz | ||
| 128 | #define SpiBusSpeed 4000000 | ||
| 129 | |||
| 130 | #define SdepTimeout 150 /* milliseconds */ | ||
| 131 | #define SdepShortTimeout 10 /* milliseconds */ | ||
| 132 | #define SdepBackOff 25 /* microseconds */ | ||
| 133 | #define BatteryUpdateInterval 10000 /* milliseconds */ | ||
| 134 | |||
| 135 | static bool at_command(const char *cmd, char *resp, uint16_t resplen, | ||
| 136 | bool verbose, uint16_t timeout = SdepTimeout); | ||
| 137 | static bool at_command_P(const char *cmd, char *resp, uint16_t resplen, | ||
| 138 | bool verbose = false); | ||
| 139 | |||
| 140 | struct SPI_Settings { | ||
| 141 | uint8_t spcr, spsr; | ||
| 142 | }; | ||
| 143 | |||
| 144 | static struct SPI_Settings spi; | ||
| 145 | |||
| 146 | // Initialize 4Mhz MSBFIRST MODE0 | ||
| 147 | void SPI_init(struct SPI_Settings *spi) { | ||
| 148 | spi->spcr = _BV(SPE) | _BV(MSTR); | ||
| 149 | spi->spsr = _BV(SPI2X); | ||
| 150 | |||
| 151 | static_assert(SpiBusSpeed == F_CPU / 2, "hard coded at 4Mhz"); | ||
| 152 | |||
| 153 | ATOMIC_BLOCK(ATOMIC_RESTORESTATE) { | ||
| 154 | // Ensure that SS is OUTPUT High | ||
| 155 | digitalWrite(B0, PinLevelHigh); | ||
| 156 | pinMode(B0, PinDirectionOutput); | ||
| 157 | |||
| 158 | SPCR |= _BV(MSTR); | ||
| 159 | SPCR |= _BV(SPE); | ||
| 160 | pinMode(B1 /* SCK */, PinDirectionOutput); | ||
| 161 | pinMode(B2 /* MOSI */, PinDirectionOutput); | ||
| 162 | } | ||
| 163 | } | ||
| 164 | |||
| 165 | static inline void SPI_begin(struct SPI_Settings*spi) { | ||
| 166 | SPCR = spi->spcr; | ||
| 167 | SPSR = spi->spsr; | ||
| 168 | } | ||
| 169 | |||
| 170 | static inline uint8_t SPI_TransferByte(uint8_t data) { | ||
| 171 | SPDR = data; | ||
| 172 | asm volatile("nop"); | ||
| 173 | while (!(SPSR & _BV(SPIF))) { | ||
| 174 | ; // wait | ||
| 175 | } | ||
| 176 | return SPDR; | ||
| 177 | } | ||
| 178 | |||
| 179 | static inline void spi_send_bytes(const uint8_t *buf, uint8_t len) { | ||
| 180 | if (len == 0) return; | ||
| 181 | const uint8_t *end = buf + len; | ||
| 182 | while (buf < end) { | ||
| 183 | SPDR = *buf; | ||
| 184 | while (!(SPSR & _BV(SPIF))) { | ||
| 185 | ; // wait | ||
| 186 | } | ||
| 187 | ++buf; | ||
| 188 | } | ||
| 189 | } | ||
| 190 | |||
| 191 | static inline uint16_t spi_read_byte(void) { | ||
| 192 | return SPI_TransferByte(0x00 /* dummy */); | ||
| 193 | } | ||
| 194 | |||
| 195 | static inline void spi_recv_bytes(uint8_t *buf, uint8_t len) { | ||
| 196 | const uint8_t *end = buf + len; | ||
| 197 | if (len == 0) return; | ||
| 198 | while (buf < end) { | ||
| 199 | SPDR = 0; // write a dummy to initiate read | ||
| 200 | while (!(SPSR & _BV(SPIF))) { | ||
| 201 | ; // wait | ||
| 202 | } | ||
| 203 | *buf = SPDR; | ||
| 204 | ++buf; | ||
| 205 | } | ||
| 206 | } | ||
| 207 | |||
| 208 | #if 0 | ||
| 209 | static void dump_pkt(const struct sdep_msg *msg) { | ||
| 210 | print("pkt: type="); | ||
| 211 | print_hex8(msg->type); | ||
| 212 | print(" cmd="); | ||
| 213 | print_hex8(msg->cmd_high); | ||
| 214 | print_hex8(msg->cmd_low); | ||
| 215 | print(" len="); | ||
| 216 | print_hex8(msg->len); | ||
| 217 | print(" more="); | ||
| 218 | print_hex8(msg->more); | ||
| 219 | print("\n"); | ||
| 220 | } | ||
| 221 | #endif | ||
| 222 | |||
| 223 | // Send a single SDEP packet | ||
| 224 | static bool sdep_send_pkt(const struct sdep_msg *msg, uint16_t timeout) { | ||
| 225 | SPI_begin(&spi); | ||
| 226 | |||
| 227 | digitalWrite(AdafruitBleCSPin, PinLevelLow); | ||
| 228 | uint16_t timerStart = timer_read(); | ||
| 229 | bool success = false; | ||
| 230 | bool ready = false; | ||
| 231 | |||
| 232 | do { | ||
| 233 | ready = SPI_TransferByte(msg->type) != SdepSlaveNotReady; | ||
| 234 | if (ready) { | ||
| 235 | break; | ||
| 236 | } | ||
| 237 | |||
| 238 | // Release it and let it initialize | ||
| 239 | digitalWrite(AdafruitBleCSPin, PinLevelHigh); | ||
| 240 | _delay_us(SdepBackOff); | ||
| 241 | digitalWrite(AdafruitBleCSPin, PinLevelLow); | ||
| 242 | } while (timer_elapsed(timerStart) < timeout); | ||
| 243 | |||
| 244 | if (ready) { | ||
| 245 | // Slave is ready; send the rest of the packet | ||
| 246 | spi_send_bytes(&msg->cmd_low, | ||
| 247 | sizeof(*msg) - (1 + sizeof(msg->payload)) + msg->len); | ||
| 248 | success = true; | ||
| 249 | } | ||
| 250 | |||
| 251 | digitalWrite(AdafruitBleCSPin, PinLevelHigh); | ||
| 252 | |||
| 253 | return success; | ||
| 254 | } | ||
| 255 | |||
| 256 | static inline void sdep_build_pkt(struct sdep_msg *msg, uint16_t command, | ||
| 257 | const uint8_t *payload, uint8_t len, | ||
| 258 | bool moredata) { | ||
| 259 | msg->type = SdepCommand; | ||
| 260 | msg->cmd_low = command & 0xff; | ||
| 261 | msg->cmd_high = command >> 8; | ||
| 262 | msg->len = len; | ||
| 263 | msg->more = (moredata && len == SdepMaxPayload) ? 1 : 0; | ||
| 264 | |||
| 265 | static_assert(sizeof(*msg) == 20, "msg is correctly packed"); | ||
| 266 | |||
| 267 | memcpy(msg->payload, payload, len); | ||
| 268 | } | ||
| 269 | |||
| 270 | // Read a single SDEP packet | ||
| 271 | static bool sdep_recv_pkt(struct sdep_msg *msg, uint16_t timeout) { | ||
| 272 | bool success = false; | ||
| 273 | uint16_t timerStart = timer_read(); | ||
| 274 | bool ready = false; | ||
| 275 | |||
| 276 | do { | ||
| 277 | ready = digitalRead(AdafruitBleIRQPin); | ||
| 278 | if (ready) { | ||
| 279 | break; | ||
| 280 | } | ||
| 281 | _delay_us(1); | ||
| 282 | } while (timer_elapsed(timerStart) < timeout); | ||
| 283 | |||
| 284 | if (ready) { | ||
| 285 | SPI_begin(&spi); | ||
| 286 | |||
| 287 | digitalWrite(AdafruitBleCSPin, PinLevelLow); | ||
| 288 | |||
| 289 | do { | ||
| 290 | // Read the command type, waiting for the data to be ready | ||
| 291 | msg->type = spi_read_byte(); | ||
| 292 | if (msg->type == SdepSlaveNotReady || msg->type == SdepSlaveOverflow) { | ||
| 293 | // Release it and let it initialize | ||
| 294 | digitalWrite(AdafruitBleCSPin, PinLevelHigh); | ||
| 295 | _delay_us(SdepBackOff); | ||
| 296 | digitalWrite(AdafruitBleCSPin, PinLevelLow); | ||
| 297 | continue; | ||
| 298 | } | ||
| 299 | |||
| 300 | // Read the rest of the header | ||
| 301 | spi_recv_bytes(&msg->cmd_low, sizeof(*msg) - (1 + sizeof(msg->payload))); | ||
| 302 | |||
| 303 | // and get the payload if there is any | ||
| 304 | if (msg->len <= SdepMaxPayload) { | ||
| 305 | spi_recv_bytes(msg->payload, msg->len); | ||
| 306 | } | ||
| 307 | success = true; | ||
| 308 | break; | ||
| 309 | } while (timer_elapsed(timerStart) < timeout); | ||
| 310 | |||
| 311 | digitalWrite(AdafruitBleCSPin, PinLevelHigh); | ||
| 312 | } | ||
| 313 | return success; | ||
| 314 | } | ||
| 315 | |||
| 316 | static void resp_buf_read_one(bool greedy) { | ||
| 317 | uint16_t last_send; | ||
| 318 | if (!resp_buf.peek(last_send)) { | ||
| 319 | return; | ||
| 320 | } | ||
| 321 | |||
| 322 | if (digitalRead(AdafruitBleIRQPin)) { | ||
| 323 | struct sdep_msg msg; | ||
| 324 | |||
| 325 | again: | ||
| 326 | if (sdep_recv_pkt(&msg, SdepTimeout)) { | ||
| 327 | if (!msg.more) { | ||
| 328 | // We got it; consume this entry | ||
| 329 | resp_buf.get(last_send); | ||
| 330 | dprintf("recv latency %dms\n", TIMER_DIFF_16(timer_read(), last_send)); | ||
| 331 | } | ||
| 332 | |||
| 333 | if (greedy && resp_buf.peek(last_send) && digitalRead(AdafruitBleIRQPin)) { | ||
| 334 | goto again; | ||
| 335 | } | ||
| 336 | } | ||
| 337 | |||
| 338 | } else if (timer_elapsed(last_send) > SdepTimeout * 2) { | ||
| 339 | dprintf("waiting_for_result: timeout, resp_buf size %d\n", | ||
| 340 | (int)resp_buf.size()); | ||
| 341 | |||
| 342 | // Timed out: consume this entry | ||
| 343 | resp_buf.get(last_send); | ||
| 344 | } | ||
| 345 | } | ||
| 346 | |||
| 347 | static void send_buf_send_one(uint16_t timeout = SdepTimeout) { | ||
| 348 | struct queue_item item; | ||
| 349 | |||
| 350 | // Don't send anything more until we get an ACK | ||
| 351 | if (!resp_buf.empty()) { | ||
| 352 | return; | ||
| 353 | } | ||
| 354 | |||
| 355 | if (!send_buf.peek(item)) { | ||
| 356 | return; | ||
| 357 | } | ||
| 358 | if (process_queue_item(&item, timeout)) { | ||
| 359 | // commit that peek | ||
| 360 | send_buf.get(item); | ||
| 361 | dprintf("send_buf_send_one: have %d remaining\n", (int)send_buf.size()); | ||
| 362 | } else { | ||
| 363 | dprint("failed to send, will retry\n"); | ||
| 364 | _delay_ms(SdepTimeout); | ||
| 365 | resp_buf_read_one(true); | ||
| 366 | } | ||
| 367 | } | ||
| 368 | |||
| 369 | static void resp_buf_wait(const char *cmd) { | ||
| 370 | bool didPrint = false; | ||
| 371 | while (!resp_buf.empty()) { | ||
| 372 | if (!didPrint) { | ||
| 373 | dprintf("wait on buf for %s\n", cmd); | ||
| 374 | didPrint = true; | ||
| 375 | } | ||
| 376 | resp_buf_read_one(true); | ||
| 377 | } | ||
| 378 | } | ||
| 379 | |||
| 380 | static bool ble_init(void) { | ||
| 381 | state.initialized = false; | ||
| 382 | state.configured = false; | ||
| 383 | state.is_connected = false; | ||
| 384 | |||
| 385 | pinMode(AdafruitBleIRQPin, PinDirectionInput); | ||
| 386 | pinMode(AdafruitBleCSPin, PinDirectionOutput); | ||
| 387 | digitalWrite(AdafruitBleCSPin, PinLevelHigh); | ||
| 388 | |||
| 389 | SPI_init(&spi); | ||
| 390 | |||
| 391 | // Perform a hardware reset | ||
| 392 | pinMode(AdafruitBleResetPin, PinDirectionOutput); | ||
| 393 | digitalWrite(AdafruitBleResetPin, PinLevelHigh); | ||
| 394 | digitalWrite(AdafruitBleResetPin, PinLevelLow); | ||
| 395 | _delay_ms(10); | ||
| 396 | digitalWrite(AdafruitBleResetPin, PinLevelHigh); | ||
| 397 | |||
| 398 | _delay_ms(1000); // Give it a second to initialize | ||
| 399 | |||
| 400 | state.initialized = true; | ||
| 401 | return state.initialized; | ||
| 402 | } | ||
| 403 | |||
| 404 | static inline uint8_t min(uint8_t a, uint8_t b) { | ||
| 405 | return a < b ? a : b; | ||
| 406 | } | ||
| 407 | |||
| 408 | static bool read_response(char *resp, uint16_t resplen, bool verbose) { | ||
| 409 | char *dest = resp; | ||
| 410 | char *end = dest + resplen; | ||
| 411 | |||
| 412 | while (true) { | ||
| 413 | struct sdep_msg msg; | ||
| 414 | |||
| 415 | if (!sdep_recv_pkt(&msg, 2 * SdepTimeout)) { | ||
| 416 | dprint("sdep_recv_pkt failed\n"); | ||
| 417 | return false; | ||
| 418 | } | ||
| 419 | |||
| 420 | if (msg.type != SdepResponse) { | ||
| 421 | *resp = 0; | ||
| 422 | return false; | ||
| 423 | } | ||
| 424 | |||
| 425 | uint8_t len = min(msg.len, end - dest); | ||
| 426 | if (len > 0) { | ||
| 427 | memcpy(dest, msg.payload, len); | ||
| 428 | dest += len; | ||
| 429 | } | ||
| 430 | |||
| 431 | if (!msg.more) { | ||
| 432 | // No more data is expected! | ||
| 433 | break; | ||
| 434 | } | ||
| 435 | } | ||
| 436 | |||
| 437 | // Ensure the response is NUL terminated | ||
| 438 | *dest = 0; | ||
| 439 | |||
| 440 | // "Parse" the result text; we want to snip off the trailing OK or ERROR line | ||
| 441 | // Rewind past the possible trailing CRLF so that we can strip it | ||
| 442 | --dest; | ||
| 443 | while (dest > resp && (dest[0] == '\n' || dest[0] == '\r')) { | ||
| 444 | *dest = 0; | ||
| 445 | --dest; | ||
| 446 | } | ||
| 447 | |||
| 448 | // Look back for start of preceeding line | ||
| 449 | char *last_line = strrchr(resp, '\n'); | ||
| 450 | if (last_line) { | ||
| 451 | ++last_line; | ||
| 452 | } else { | ||
| 453 | last_line = resp; | ||
| 454 | } | ||
| 455 | |||
| 456 | bool success = false; | ||
| 457 | static const char kOK[] PROGMEM = "OK"; | ||
| 458 | |||
| 459 | success = !strcmp_P(last_line, kOK ); | ||
| 460 | |||
| 461 | if (verbose || !success) { | ||
| 462 | dprintf("result: %s\n", resp); | ||
| 463 | } | ||
| 464 | return success; | ||
| 465 | } | ||
| 466 | |||
| 467 | static bool at_command(const char *cmd, char *resp, uint16_t resplen, | ||
| 468 | bool verbose, uint16_t timeout) { | ||
| 469 | const char *end = cmd + strlen(cmd); | ||
| 470 | struct sdep_msg msg; | ||
| 471 | |||
| 472 | if (verbose) { | ||
| 473 | dprintf("ble send: %s\n", cmd); | ||
| 474 | } | ||
| 475 | |||
| 476 | if (resp) { | ||
| 477 | // They want to decode the response, so we need to flush and wait | ||
| 478 | // for all pending I/O to finish before we start this one, so | ||
| 479 | // that we don't confuse the results | ||
| 480 | resp_buf_wait(cmd); | ||
| 481 | *resp = 0; | ||
| 482 | } | ||
| 483 | |||
| 484 | // Fragment the command into a series of SDEP packets | ||
| 485 | while (end - cmd > SdepMaxPayload) { | ||
| 486 | sdep_build_pkt(&msg, BleAtWrapper, (uint8_t *)cmd, SdepMaxPayload, true); | ||
| 487 | if (!sdep_send_pkt(&msg, timeout)) { | ||
| 488 | return false; | ||
| 489 | } | ||
| 490 | cmd += SdepMaxPayload; | ||
| 491 | } | ||
| 492 | |||
| 493 | sdep_build_pkt(&msg, BleAtWrapper, (uint8_t *)cmd, end - cmd, false); | ||
| 494 | if (!sdep_send_pkt(&msg, timeout)) { | ||
| 495 | return false; | ||
| 496 | } | ||
| 497 | |||
| 498 | if (resp == NULL) { | ||
| 499 | auto now = timer_read(); | ||
| 500 | while (!resp_buf.enqueue(now)) { | ||
| 501 | resp_buf_read_one(false); | ||
| 502 | } | ||
| 503 | auto later = timer_read(); | ||
| 504 | if (TIMER_DIFF_16(later, now) > 0) { | ||
| 505 | dprintf("waited %dms for resp_buf\n", TIMER_DIFF_16(later, now)); | ||
| 506 | } | ||
| 507 | return true; | ||
| 508 | } | ||
| 509 | |||
| 510 | return read_response(resp, resplen, verbose); | ||
| 511 | } | ||
| 512 | |||
| 513 | bool at_command_P(const char *cmd, char *resp, uint16_t resplen, bool verbose) { | ||
| 514 | auto cmdbuf = (char *)alloca(strlen_P(cmd) + 1); | ||
| 515 | strcpy_P(cmdbuf, cmd); | ||
| 516 | return at_command(cmdbuf, resp, resplen, verbose); | ||
| 517 | } | ||
| 518 | |||
| 519 | bool adafruit_ble_is_connected(void) { | ||
| 520 | return state.is_connected; | ||
| 521 | } | ||
| 522 | |||
| 523 | bool adafruit_ble_enable_keyboard(void) { | ||
| 524 | char resbuf[128]; | ||
| 525 | |||
| 526 | if (!state.initialized && !ble_init()) { | ||
| 527 | return false; | ||
| 528 | } | ||
| 529 | |||
| 530 | state.configured = false; | ||
| 531 | |||
| 532 | // Disable command echo | ||
| 533 | static const char kEcho[] PROGMEM = "ATE=0"; | ||
| 534 | // Make the advertised name match the keyboard | ||
| 535 | static const char kGapDevName[] PROGMEM = | ||
| 536 | "AT+GAPDEVNAME=" STR(PRODUCT) " " STR(DESCRIPTION); | ||
| 537 | // Turn on keyboard support | ||
| 538 | static const char kHidEnOn[] PROGMEM = "AT+BLEHIDEN=1"; | ||
| 539 | |||
| 540 | // Adjust intervals to improve latency. This causes the "central" | ||
| 541 | // system (computer/tablet) to poll us every 10-30 ms. We can't | ||
| 542 | // set a smaller value than 10ms, and 30ms seems to be the natural | ||
| 543 | // processing time on my macbook. Keeping it constrained to that | ||
| 544 | // feels reasonable to type to. | ||
| 545 | static const char kGapIntervals[] PROGMEM = "AT+GAPINTERVALS=10,30,,"; | ||
| 546 | |||
| 547 | // Reset the device so that it picks up the above changes | ||
| 548 | static const char kATZ[] PROGMEM = "ATZ"; | ||
| 549 | |||
| 550 | // Turn down the power level a bit | ||
| 551 | static const char kPower[] PROGMEM = "AT+BLEPOWERLEVEL=-12"; | ||
| 552 | static PGM_P const configure_commands[] PROGMEM = { | ||
| 553 | kEcho, | ||
| 554 | kGapIntervals, | ||
| 555 | kGapDevName, | ||
| 556 | kHidEnOn, | ||
| 557 | kPower, | ||
| 558 | kATZ, | ||
| 559 | }; | ||
| 560 | |||
| 561 | uint8_t i; | ||
| 562 | for (i = 0; i < sizeof(configure_commands) / sizeof(configure_commands[0]); | ||
| 563 | ++i) { | ||
| 564 | PGM_P cmd; | ||
| 565 | memcpy_P(&cmd, configure_commands + i, sizeof(cmd)); | ||
| 566 | |||
| 567 | if (!at_command_P(cmd, resbuf, sizeof(resbuf))) { | ||
| 568 | dprintf("failed BLE command: %S: %s\n", cmd, resbuf); | ||
| 569 | goto fail; | ||
| 570 | } | ||
| 571 | } | ||
| 572 | |||
| 573 | state.configured = true; | ||
| 574 | |||
| 575 | // Check connection status in a little while; allow the ATZ time | ||
| 576 | // to kick in. | ||
| 577 | state.last_connection_update = timer_read(); | ||
| 578 | fail: | ||
| 579 | return state.configured; | ||
| 580 | } | ||
| 581 | |||
| 582 | static void set_connected(bool connected) { | ||
| 583 | if (connected != state.is_connected) { | ||
| 584 | if (connected) { | ||
| 585 | print("****** BLE CONNECT!!!!\n"); | ||
| 586 | } else { | ||
| 587 | print("****** BLE DISCONNECT!!!!\n"); | ||
| 588 | } | ||
| 589 | state.is_connected = connected; | ||
| 590 | |||
| 591 | // TODO: if modifiers are down on the USB interface and | ||
| 592 | // we cut over to BLE or vice versa, they will remain stuck. | ||
| 593 | // This feels like a good point to do something like clearing | ||
| 594 | // the keyboard and/or generating a fake all keys up message. | ||
| 595 | // However, I've noticed that it takes a couple of seconds | ||
| 596 | // for macOS to to start recognizing key presses after BLE | ||
| 597 | // is in the connected state, so I worry that doing that | ||
| 598 | // here may not be good enough. | ||
| 599 | } | ||
| 600 | } | ||
| 601 | |||
| 602 | void adafruit_ble_task(void) { | ||
| 603 | char resbuf[48]; | ||
| 604 | |||
| 605 | if (!state.configured && !adafruit_ble_enable_keyboard()) { | ||
| 606 | return; | ||
| 607 | } | ||
| 608 | resp_buf_read_one(true); | ||
| 609 | send_buf_send_one(SdepShortTimeout); | ||
| 610 | |||
| 611 | if (resp_buf.empty() && (state.event_flags & UsingEvents) && | ||
| 612 | digitalRead(AdafruitBleIRQPin)) { | ||
| 613 | // Must be an event update | ||
| 614 | if (at_command_P(PSTR("AT+EVENTSTATUS"), resbuf, sizeof(resbuf))) { | ||
| 615 | uint32_t mask = strtoul(resbuf, NULL, 16); | ||
| 616 | |||
| 617 | if (mask & BleSystemConnected) { | ||
| 618 | set_connected(true); | ||
| 619 | } else if (mask & BleSystemDisconnected) { | ||
| 620 | set_connected(false); | ||
| 621 | } | ||
| 622 | } | ||
| 623 | } | ||
| 624 | |||
| 625 | if (timer_elapsed(state.last_connection_update) > ConnectionUpdateInterval) { | ||
| 626 | bool shouldPoll = true; | ||
| 627 | if (!(state.event_flags & ProbedEvents)) { | ||
| 628 | // Request notifications about connection status changes. | ||
| 629 | // This only works in SPIFRIEND firmware > 0.6.7, which is why | ||
| 630 | // we check for this conditionally here. | ||
| 631 | // Note that at the time of writing, HID reports only work correctly | ||
| 632 | // with Apple products on firmware version 0.6.7! | ||
| 633 | // https://forums.adafruit.com/viewtopic.php?f=8&t=104052 | ||
| 634 | if (at_command_P(PSTR("AT+EVENTENABLE=0x1"), resbuf, sizeof(resbuf))) { | ||
| 635 | at_command_P(PSTR("AT+EVENTENABLE=0x2"), resbuf, sizeof(resbuf)); | ||
| 636 | state.event_flags |= UsingEvents; | ||
| 637 | } | ||
| 638 | state.event_flags |= ProbedEvents; | ||
| 639 | |||
| 640 | // leave shouldPoll == true so that we check at least once | ||
| 641 | // before relying solely on events | ||
| 642 | } else { | ||
| 643 | shouldPoll = false; | ||
| 644 | } | ||
| 645 | |||
| 646 | static const char kGetConn[] PROGMEM = "AT+GAPGETCONN"; | ||
| 647 | state.last_connection_update = timer_read(); | ||
| 648 | |||
| 649 | if (at_command_P(kGetConn, resbuf, sizeof(resbuf))) { | ||
| 650 | set_connected(atoi(resbuf)); | ||
| 651 | } | ||
| 652 | } | ||
| 653 | |||
| 654 | #ifdef SAMPLE_BATTERY | ||
| 655 | // I don't know if this really does anything useful yet; the reported | ||
| 656 | // voltage level always seems to be around 3200mV. We may want to just rip | ||
| 657 | // this code out. | ||
| 658 | if (timer_elapsed(state.last_battery_update) > BatteryUpdateInterval && | ||
| 659 | resp_buf.empty()) { | ||
| 660 | state.last_battery_update = timer_read(); | ||
| 661 | |||
| 662 | if (at_command_P(PSTR("AT+HWVBAT"), resbuf, sizeof(resbuf))) { | ||
| 663 | state.vbat = atoi(resbuf); | ||
| 664 | } | ||
| 665 | } | ||
| 666 | #endif | ||
| 667 | } | ||
| 668 | |||
| 669 | static bool process_queue_item(struct queue_item *item, uint16_t timeout) { | ||
| 670 | char cmdbuf[48]; | ||
| 671 | char fmtbuf[64]; | ||
| 672 | |||
| 673 | // Arrange to re-check connection after keys have settled | ||
| 674 | state.last_connection_update = timer_read(); | ||
| 675 | |||
| 676 | #if 1 | ||
| 677 | if (TIMER_DIFF_16(state.last_connection_update, item->added) > 0) { | ||
| 678 | dprintf("send latency %dms\n", | ||
| 679 | TIMER_DIFF_16(state.last_connection_update, item->added)); | ||
| 680 | } | ||
| 681 | #endif | ||
| 682 | |||
| 683 | switch (item->queue_type) { | ||
| 684 | case QTKeyReport: | ||
| 685 | strcpy_P(fmtbuf, | ||
| 686 | PSTR("AT+BLEKEYBOARDCODE=%02x-00-%02x-%02x-%02x-%02x-%02x-%02x")); | ||
| 687 | snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->key.modifier, | ||
| 688 | item->key.keys[0], item->key.keys[1], item->key.keys[2], | ||
| 689 | item->key.keys[3], item->key.keys[4], item->key.keys[5]); | ||
| 690 | return at_command(cmdbuf, NULL, 0, true, timeout); | ||
| 691 | |||
| 692 | case QTConsumer: | ||
| 693 | strcpy_P(fmtbuf, PSTR("AT+BLEHIDCONTROLKEY=0x%04x")); | ||
| 694 | snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->consumer); | ||
| 695 | return at_command(cmdbuf, NULL, 0, true, timeout); | ||
| 696 | |||
| 697 | #ifdef MOUSE_ENABLE | ||
| 698 | case QTMouseMove: | ||
| 699 | strcpy_P(fmtbuf, PSTR("AT+BLEHIDMOUSEMOVE=%d,%d,%d,%d")); | ||
| 700 | snprintf(cmdbuf, sizeof(cmdbuf), fmtbuf, item->mousemove.x, | ||
| 701 | item->mousemove.y, item->mousemove.scroll, item->mousemove.pan); | ||
| 702 | return at_command(cmdbuf, NULL, 0, true, timeout); | ||
| 703 | #endif | ||
| 704 | default: | ||
| 705 | return true; | ||
| 706 | } | ||
| 707 | } | ||
| 708 | |||
| 709 | bool adafruit_ble_send_keys(uint8_t hid_modifier_mask, uint8_t *keys, | ||
| 710 | uint8_t nkeys) { | ||
| 711 | struct queue_item item; | ||
| 712 | bool didWait = false; | ||
| 713 | |||
| 714 | item.queue_type = QTKeyReport; | ||
| 715 | item.key.modifier = hid_modifier_mask; | ||
| 716 | item.added = timer_read(); | ||
| 717 | |||
| 718 | while (nkeys >= 0) { | ||
| 719 | item.key.keys[0] = keys[0]; | ||
| 720 | item.key.keys[1] = nkeys >= 1 ? keys[1] : 0; | ||
| 721 | item.key.keys[2] = nkeys >= 2 ? keys[2] : 0; | ||
| 722 | item.key.keys[3] = nkeys >= 3 ? keys[3] : 0; | ||
| 723 | item.key.keys[4] = nkeys >= 4 ? keys[4] : 0; | ||
| 724 | item.key.keys[5] = nkeys >= 5 ? keys[5] : 0; | ||
| 725 | |||
| 726 | if (!send_buf.enqueue(item)) { | ||
| 727 | if (!didWait) { | ||
| 728 | dprint("wait for buf space\n"); | ||
| 729 | didWait = true; | ||
| 730 | } | ||
| 731 | send_buf_send_one(); | ||
| 732 | continue; | ||
| 733 | } | ||
| 734 | |||
| 735 | if (nkeys <= 6) { | ||
| 736 | return true; | ||
| 737 | } | ||
| 738 | |||
| 739 | nkeys -= 6; | ||
| 740 | keys += 6; | ||
| 741 | } | ||
| 742 | |||
| 743 | return true; | ||
| 744 | } | ||
| 745 | |||
| 746 | bool adafruit_ble_send_consumer_key(uint16_t keycode, int hold_duration) { | ||
| 747 | struct queue_item item; | ||
| 748 | |||
| 749 | item.queue_type = QTConsumer; | ||
| 750 | item.consumer = keycode; | ||
| 751 | |||
| 752 | while (!send_buf.enqueue(item)) { | ||
| 753 | send_buf_send_one(); | ||
| 754 | } | ||
| 755 | return true; | ||
| 756 | } | ||
| 757 | |||
| 758 | #ifdef MOUSE_ENABLE | ||
| 759 | bool adafruit_ble_send_mouse_move(int8_t x, int8_t y, int8_t scroll, | ||
| 760 | int8_t pan) { | ||
| 761 | struct queue_item item; | ||
| 762 | |||
| 763 | item.queue_type = QTMouseMove; | ||
| 764 | item.mousemove.x = x; | ||
| 765 | item.mousemove.y = y; | ||
| 766 | item.mousemove.scroll = scroll; | ||
| 767 | item.mousemove.pan = pan; | ||
| 768 | |||
| 769 | while (!send_buf.enqueue(item)) { | ||
| 770 | send_buf_send_one(); | ||
| 771 | } | ||
| 772 | return true; | ||
| 773 | } | ||
| 774 | #endif | ||
| 775 | |||
| 776 | uint32_t adafruit_ble_read_battery_voltage(void) { | ||
| 777 | return state.vbat; | ||
| 778 | } | ||
| 779 | |||
| 780 | bool adafruit_ble_set_mode_leds(bool on) { | ||
| 781 | if (!state.configured) { | ||
| 782 | return false; | ||
| 783 | } | ||
| 784 | |||
| 785 | // The "mode" led is the red blinky one | ||
| 786 | at_command_P(on ? PSTR("AT+HWMODELED=1") : PSTR("AT+HWMODELED=0"), NULL, 0); | ||
| 787 | |||
| 788 | // Pin 19 is the blue "connected" LED; turn that off too. | ||
| 789 | // When turning LEDs back on, don't turn that LED on if we're | ||
| 790 | // not connected, as that would be confusing. | ||
| 791 | at_command_P(on && state.is_connected ? PSTR("AT+HWGPIO=19,1") | ||
| 792 | : PSTR("AT+HWGPIO=19,0"), | ||
| 793 | NULL, 0); | ||
| 794 | return true; | ||
| 795 | } | ||
| 796 | |||
| 797 | // https://learn.adafruit.com/adafruit-feather-32u4-bluefruit-le/ble-generic#at-plus-blepowerlevel | ||
| 798 | bool adafruit_ble_set_power_level(int8_t level) { | ||
| 799 | char cmd[46]; | ||
| 800 | if (!state.configured) { | ||
| 801 | return false; | ||
| 802 | } | ||
| 803 | snprintf(cmd, sizeof(cmd), "AT+BLEPOWERLEVEL=%d", level); | ||
| 804 | return at_command(cmd, NULL, 0, false); | ||
| 805 | } | ||
diff --git a/tmk_core/protocol/lufa/adafruit_ble.h b/tmk_core/protocol/lufa/adafruit_ble.h new file mode 100644 index 000000000..b3bab3ca0 --- /dev/null +++ b/tmk_core/protocol/lufa/adafruit_ble.h | |||
| @@ -0,0 +1,60 @@ | |||
| 1 | /* Bluetooth Low Energy Protocol for QMK. | ||
| 2 | * Author: Wez Furlong, 2016 | ||
| 3 | * Supports the Adafruit BLE board built around the nRF51822 chip. | ||
| 4 | */ | ||
| 5 | #pragma once | ||
| 6 | #ifdef MODULE_ADAFRUIT_BLE | ||
| 7 | #include <stdbool.h> | ||
| 8 | #include <stdint.h> | ||
| 9 | #include <string.h> | ||
| 10 | |||
| 11 | #ifdef __cplusplus | ||
| 12 | extern "C" { | ||
| 13 | #endif | ||
| 14 | |||
| 15 | /* Instruct the module to enable HID keyboard support and reset */ | ||
| 16 | extern bool adafruit_ble_enable_keyboard(void); | ||
| 17 | |||
| 18 | /* Query to see if the BLE module is connected */ | ||
| 19 | extern bool adafruit_ble_query_is_connected(void); | ||
| 20 | |||
| 21 | /* Returns true if we believe that the BLE module is connected. | ||
| 22 | * This uses our cached understanding that is maintained by | ||
| 23 | * calling ble_task() periodically. */ | ||
| 24 | extern bool adafruit_ble_is_connected(void); | ||
| 25 | |||
| 26 | /* Call this periodically to process BLE-originated things */ | ||
| 27 | extern void adafruit_ble_task(void); | ||
| 28 | |||
| 29 | /* Generates keypress events for a set of keys. | ||
| 30 | * The hid modifier mask specifies the state of the modifier keys for | ||
| 31 | * this set of keys. | ||
| 32 | * Also sends a key release indicator, so that the keys do not remain | ||
| 33 | * held down. */ | ||
| 34 | extern bool adafruit_ble_send_keys(uint8_t hid_modifier_mask, uint8_t *keys, | ||
| 35 | uint8_t nkeys); | ||
| 36 | |||
| 37 | /* Send a consumer keycode, holding it down for the specified duration | ||
| 38 | * (milliseconds) */ | ||
| 39 | extern bool adafruit_ble_send_consumer_key(uint16_t keycode, int hold_duration); | ||
| 40 | |||
| 41 | #ifdef MOUSE_ENABLE | ||
| 42 | /* Send a mouse/wheel movement report. | ||
| 43 | * The parameters are signed and indicate positive of negative direction | ||
| 44 | * change. */ | ||
| 45 | extern bool adafruit_ble_send_mouse_move(int8_t x, int8_t y, int8_t scroll, | ||
| 46 | int8_t pan); | ||
| 47 | #endif | ||
| 48 | |||
| 49 | /* Compute battery voltage by reading an analog pin. | ||
| 50 | * Returns the integer number of millivolts */ | ||
| 51 | extern uint32_t adafruit_ble_read_battery_voltage(void); | ||
| 52 | |||
| 53 | extern bool adafruit_ble_set_mode_leds(bool on); | ||
| 54 | extern bool adafruit_ble_set_power_level(int8_t level); | ||
| 55 | |||
| 56 | #ifdef __cplusplus | ||
| 57 | } | ||
| 58 | #endif | ||
| 59 | |||
| 60 | #endif // MODULE_ADAFRUIT_BLE | ||
diff --git a/tmk_core/protocol/lufa/descriptor.c b/tmk_core/protocol/lufa/descriptor.c index 6f2407f58..feeea76df 100644 --- a/tmk_core/protocol/lufa/descriptor.c +++ b/tmk_core/protocol/lufa/descriptor.c | |||
| @@ -40,6 +40,9 @@ | |||
| 40 | #include "report.h" | 40 | #include "report.h" |
| 41 | #include "descriptor.h" | 41 | #include "descriptor.h" |
| 42 | 42 | ||
| 43 | #ifndef USB_MAX_POWER_CONSUMPTION | ||
| 44 | #define USB_MAX_POWER_CONSUMPTION 500 | ||
| 45 | #endif | ||
| 43 | 46 | ||
| 44 | /******************************************************************************* | 47 | /******************************************************************************* |
| 45 | * HID Report Descriptors | 48 | * HID Report Descriptors |
| @@ -140,10 +143,10 @@ const USB_Descriptor_HIDReport_Datatype_t PROGMEM ExtrakeyReport[] = | |||
| 140 | HID_RI_USAGE(8, 0x80), /* System Control */ | 143 | HID_RI_USAGE(8, 0x80), /* System Control */ |
| 141 | HID_RI_COLLECTION(8, 0x01), /* Application */ | 144 | HID_RI_COLLECTION(8, 0x01), /* Application */ |
| 142 | HID_RI_REPORT_ID(8, REPORT_ID_SYSTEM), | 145 | HID_RI_REPORT_ID(8, REPORT_ID_SYSTEM), |
| 143 | HID_RI_LOGICAL_MINIMUM(16, 0x0081), | 146 | HID_RI_LOGICAL_MINIMUM(16, 0x0001), |
| 144 | HID_RI_LOGICAL_MAXIMUM(16, 0x00B7), | 147 | HID_RI_LOGICAL_MAXIMUM(16, 0x0003), |
| 145 | HID_RI_USAGE_MINIMUM(16, 0x0081), /* System Power Down */ | 148 | HID_RI_USAGE_MINIMUM(16, 0x0081), /* System Power Down */ |
| 146 | HID_RI_USAGE_MAXIMUM(16, 0x00B7), /* System Display LCD Autoscale */ | 149 | HID_RI_USAGE_MAXIMUM(16, 0x0083), /* System Wake Up */ |
| 147 | HID_RI_REPORT_SIZE(8, 16), | 150 | HID_RI_REPORT_SIZE(8, 16), |
| 148 | HID_RI_REPORT_COUNT(8, 1), | 151 | HID_RI_REPORT_COUNT(8, 1), |
| 149 | HID_RI_INPUT(8, HID_IOF_DATA | HID_IOF_ARRAY | HID_IOF_ABSOLUTE), | 152 | HID_RI_INPUT(8, HID_IOF_DATA | HID_IOF_ARRAY | HID_IOF_ABSOLUTE), |
| @@ -164,6 +167,28 @@ const USB_Descriptor_HIDReport_Datatype_t PROGMEM ExtrakeyReport[] = | |||
| 164 | }; | 167 | }; |
| 165 | #endif | 168 | #endif |
| 166 | 169 | ||
| 170 | #ifdef RAW_ENABLE | ||
| 171 | const USB_Descriptor_HIDReport_Datatype_t PROGMEM RawReport[] = | ||
| 172 | { | ||
| 173 | HID_RI_USAGE_PAGE(16, 0xFF60), /* Vendor Page 0xFF60 */ | ||
| 174 | HID_RI_USAGE(8, 0x61), /* Vendor Usage 0x61 */ | ||
| 175 | HID_RI_COLLECTION(8, 0x01), /* Application */ | ||
| 176 | HID_RI_USAGE(8, 0x62), /* Vendor Usage 0x62 */ | ||
| 177 | HID_RI_LOGICAL_MINIMUM(8, 0x00), | ||
| 178 | HID_RI_LOGICAL_MAXIMUM(16, 0x00FF), | ||
| 179 | HID_RI_REPORT_COUNT(8, RAW_EPSIZE), | ||
| 180 | HID_RI_REPORT_SIZE(8, 0x08), | ||
| 181 | HID_RI_INPUT(8, HID_IOF_DATA | HID_IOF_VARIABLE | HID_IOF_ABSOLUTE), | ||
| 182 | HID_RI_USAGE(8, 0x63), /* Vendor Usage 0x63 */ | ||
| 183 | HID_RI_LOGICAL_MINIMUM(8, 0x00), | ||
| 184 | HID_RI_LOGICAL_MAXIMUM(16, 0x00FF), | ||
| 185 | HID_RI_REPORT_COUNT(8, RAW_EPSIZE), | ||
| 186 | HID_RI_REPORT_SIZE(8, 0x08), | ||
| 187 | HID_RI_OUTPUT(8, HID_IOF_DATA | HID_IOF_VARIABLE | HID_IOF_ABSOLUTE | HID_IOF_NON_VOLATILE), | ||
| 188 | HID_RI_END_COLLECTION(0), | ||
| 189 | }; | ||
| 190 | #endif | ||
| 191 | |||
| 167 | #ifdef CONSOLE_ENABLE | 192 | #ifdef CONSOLE_ENABLE |
| 168 | const USB_Descriptor_HIDReport_Datatype_t PROGMEM ConsoleReport[] = | 193 | const USB_Descriptor_HIDReport_Datatype_t PROGMEM ConsoleReport[] = |
| 169 | { | 194 | { |
| @@ -272,7 +297,7 @@ const USB_Descriptor_Configuration_t PROGMEM ConfigurationDescriptor = | |||
| 272 | 297 | ||
| 273 | .ConfigAttributes = (USB_CONFIG_ATTR_RESERVED | USB_CONFIG_ATTR_REMOTEWAKEUP), | 298 | .ConfigAttributes = (USB_CONFIG_ATTR_RESERVED | USB_CONFIG_ATTR_REMOTEWAKEUP), |
| 274 | 299 | ||
| 275 | .MaxPowerConsumption = USB_CONFIG_POWER_MA(500) | 300 | .MaxPowerConsumption = USB_CONFIG_POWER_MA(USB_MAX_POWER_CONSUMPTION) |
| 276 | }, | 301 | }, |
| 277 | 302 | ||
| 278 | /* | 303 | /* |
| @@ -399,6 +424,58 @@ const USB_Descriptor_Configuration_t PROGMEM ConfigurationDescriptor = | |||
| 399 | }, | 424 | }, |
| 400 | #endif | 425 | #endif |
| 401 | 426 | ||
| 427 | /* | ||
| 428 | * Raw | ||
| 429 | */ | ||
| 430 | #ifdef RAW_ENABLE | ||
| 431 | .Raw_Interface = | ||
| 432 | { | ||
| 433 | .Header = {.Size = sizeof(USB_Descriptor_Interface_t), .Type = DTYPE_Interface}, | ||
| 434 | |||
| 435 | .InterfaceNumber = RAW_INTERFACE, | ||
| 436 | .AlternateSetting = 0x00, | ||
| 437 | |||
| 438 | .TotalEndpoints = 2, | ||
| 439 | |||
| 440 | .Class = HID_CSCP_HIDClass, | ||
| 441 | .SubClass = HID_CSCP_NonBootSubclass, | ||
| 442 | .Protocol = HID_CSCP_NonBootProtocol, | ||
| 443 | |||
| 444 | .InterfaceStrIndex = NO_DESCRIPTOR | ||
| 445 | }, | ||
| 446 | |||
| 447 | .Raw_HID = | ||
| 448 | { | ||
| 449 | .Header = {.Size = sizeof(USB_HID_Descriptor_HID_t), .Type = HID_DTYPE_HID}, | ||
| 450 | |||
| 451 | .HIDSpec = VERSION_BCD(1,1,1), | ||
| 452 | .CountryCode = 0x00, | ||
| 453 | .TotalReportDescriptors = 1, | ||
| 454 | .HIDReportType = HID_DTYPE_Report, | ||
| 455 | .HIDReportLength = sizeof(RawReport) | ||
| 456 | }, | ||
| 457 | |||
| 458 | .Raw_INEndpoint = | ||
| 459 | { | ||
| 460 | .Header = {.Size = sizeof(USB_Descriptor_Endpoint_t), .Type = DTYPE_Endpoint}, | ||
| 461 | |||
| 462 | .EndpointAddress = (ENDPOINT_DIR_IN | RAW_IN_EPNUM), | ||
| 463 | .Attributes = (EP_TYPE_INTERRUPT | ENDPOINT_ATTR_NO_SYNC | ENDPOINT_USAGE_DATA), | ||
| 464 | .EndpointSize = RAW_EPSIZE, | ||
| 465 | .PollingIntervalMS = 0x01 | ||
| 466 | }, | ||
| 467 | |||
| 468 | .Raw_OUTEndpoint = | ||
| 469 | { | ||
| 470 | .Header = {.Size = sizeof(USB_Descriptor_Endpoint_t), .Type = DTYPE_Endpoint}, | ||
| 471 | |||
| 472 | .EndpointAddress = (ENDPOINT_DIR_OUT | RAW_OUT_EPNUM), | ||
| 473 | .Attributes = (EP_TYPE_INTERRUPT | ENDPOINT_ATTR_NO_SYNC | ENDPOINT_USAGE_DATA), | ||
| 474 | .EndpointSize = RAW_EPSIZE, | ||
| 475 | .PollingIntervalMS = 0x01 | ||
| 476 | }, | ||
| 477 | #endif | ||
| 478 | |||
| 402 | /* | 479 | /* |
| 403 | * Console | 480 | * Console |
| 404 | */ | 481 | */ |
| @@ -754,7 +831,6 @@ const USB_Descriptor_Configuration_t PROGMEM ConfigurationDescriptor = | |||
| 754 | .PollingIntervalMS = 0x05 | 831 | .PollingIntervalMS = 0x05 |
| 755 | }, | 832 | }, |
| 756 | #endif | 833 | #endif |
| 757 | |||
| 758 | }; | 834 | }; |
| 759 | 835 | ||
| 760 | 836 | ||
| @@ -846,6 +922,12 @@ uint16_t CALLBACK_USB_GetDescriptor(const uint16_t wValue, | |||
| 846 | Size = sizeof(USB_HID_Descriptor_HID_t); | 922 | Size = sizeof(USB_HID_Descriptor_HID_t); |
| 847 | break; | 923 | break; |
| 848 | #endif | 924 | #endif |
| 925 | #ifdef RAW_ENABLE | ||
| 926 | case RAW_INTERFACE: | ||
| 927 | Address = &ConfigurationDescriptor.Raw_HID; | ||
| 928 | Size = sizeof(USB_HID_Descriptor_HID_t); | ||
| 929 | break; | ||
| 930 | #endif | ||
| 849 | #ifdef CONSOLE_ENABLE | 931 | #ifdef CONSOLE_ENABLE |
| 850 | case CONSOLE_INTERFACE: | 932 | case CONSOLE_INTERFACE: |
| 851 | Address = &ConfigurationDescriptor.Console_HID; | 933 | Address = &ConfigurationDescriptor.Console_HID; |
| @@ -878,6 +960,12 @@ uint16_t CALLBACK_USB_GetDescriptor(const uint16_t wValue, | |||
| 878 | Size = sizeof(ExtrakeyReport); | 960 | Size = sizeof(ExtrakeyReport); |
| 879 | break; | 961 | break; |
| 880 | #endif | 962 | #endif |
| 963 | #ifdef RAW_ENABLE | ||
| 964 | case RAW_INTERFACE: | ||
| 965 | Address = &RawReport; | ||
| 966 | Size = sizeof(RawReport); | ||
| 967 | break; | ||
| 968 | #endif | ||
| 881 | #ifdef CONSOLE_ENABLE | 969 | #ifdef CONSOLE_ENABLE |
| 882 | case CONSOLE_INTERFACE: | 970 | case CONSOLE_INTERFACE: |
| 883 | Address = &ConsoleReport; | 971 | Address = &ConsoleReport; |
diff --git a/tmk_core/protocol/lufa/descriptor.h b/tmk_core/protocol/lufa/descriptor.h index c6c94e361..24ce420e6 100644 --- a/tmk_core/protocol/lufa/descriptor.h +++ b/tmk_core/protocol/lufa/descriptor.h | |||
| @@ -71,6 +71,14 @@ typedef struct | |||
| 71 | USB_Descriptor_Endpoint_t Extrakey_INEndpoint; | 71 | USB_Descriptor_Endpoint_t Extrakey_INEndpoint; |
| 72 | #endif | 72 | #endif |
| 73 | 73 | ||
| 74 | #ifdef RAW_ENABLE | ||
| 75 | // Raw HID Interface | ||
| 76 | USB_Descriptor_Interface_t Raw_Interface; | ||
| 77 | USB_HID_Descriptor_HID_t Raw_HID; | ||
| 78 | USB_Descriptor_Endpoint_t Raw_INEndpoint; | ||
| 79 | USB_Descriptor_Endpoint_t Raw_OUTEndpoint; | ||
| 80 | #endif | ||
| 81 | |||
| 74 | #ifdef CONSOLE_ENABLE | 82 | #ifdef CONSOLE_ENABLE |
| 75 | // Console HID Interface | 83 | // Console HID Interface |
| 76 | USB_Descriptor_Interface_t Console_Interface; | 84 | USB_Descriptor_Interface_t Console_Interface; |
| @@ -137,10 +145,16 @@ typedef struct | |||
| 137 | # define EXTRAKEY_INTERFACE MOUSE_INTERFACE | 145 | # define EXTRAKEY_INTERFACE MOUSE_INTERFACE |
| 138 | #endif | 146 | #endif |
| 139 | 147 | ||
| 148 | #ifdef RAW_ENABLE | ||
| 149 | # define RAW_INTERFACE (EXTRAKEY_INTERFACE + 1) | ||
| 150 | #else | ||
| 151 | # define RAW_INTERFACE EXTRAKEY_INTERFACE | ||
| 152 | #endif | ||
| 153 | |||
| 140 | #ifdef CONSOLE_ENABLE | 154 | #ifdef CONSOLE_ENABLE |
| 141 | # define CONSOLE_INTERFACE (EXTRAKEY_INTERFACE + 1) | 155 | # define CONSOLE_INTERFACE (RAW_INTERFACE + 1) |
| 142 | #else | 156 | #else |
| 143 | # define CONSOLE_INTERFACE EXTRAKEY_INTERFACE | 157 | # define CONSOLE_INTERFACE RAW_INTERFACE |
| 144 | #endif | 158 | #endif |
| 145 | 159 | ||
| 146 | #ifdef NKRO_ENABLE | 160 | #ifdef NKRO_ENABLE |
| @@ -182,12 +196,19 @@ typedef struct | |||
| 182 | # define EXTRAKEY_IN_EPNUM MOUSE_IN_EPNUM | 196 | # define EXTRAKEY_IN_EPNUM MOUSE_IN_EPNUM |
| 183 | #endif | 197 | #endif |
| 184 | 198 | ||
| 199 | #ifdef RAW_ENABLE | ||
| 200 | # define RAW_IN_EPNUM (EXTRAKEY_IN_EPNUM + 1) | ||
| 201 | # define RAW_OUT_EPNUM (EXTRAKEY_IN_EPNUM + 2) | ||
| 202 | #else | ||
| 203 | # define RAW_OUT_EPNUM EXTRAKEY_IN_EPNUM | ||
| 204 | #endif | ||
| 205 | |||
| 185 | #ifdef CONSOLE_ENABLE | 206 | #ifdef CONSOLE_ENABLE |
| 186 | # define CONSOLE_IN_EPNUM (EXTRAKEY_IN_EPNUM + 1) | 207 | # define CONSOLE_IN_EPNUM (RAW_OUT_EPNUM + 1) |
| 187 | # define CONSOLE_OUT_EPNUM (EXTRAKEY_IN_EPNUM + 1) | 208 | //# define CONSOLE_OUT_EPNUM (RAW_OUT_EPNUM + 2) |
| 188 | //# define CONSOLE_OUT_EPNUM (EXTRAKEY_IN_EPNUM + 2) | 209 | # define CONSOLE_OUT_EPNUM (RAW_OUT_EPNUM + 1) |
| 189 | #else | 210 | #else |
| 190 | # define CONSOLE_OUT_EPNUM EXTRAKEY_IN_EPNUM | 211 | # define CONSOLE_OUT_EPNUM RAW_OUT_EPNUM |
| 191 | #endif | 212 | #endif |
| 192 | 213 | ||
| 193 | #ifdef NKRO_ENABLE | 214 | #ifdef NKRO_ENABLE |
| @@ -217,7 +238,6 @@ typedef struct | |||
| 217 | # define CDC_OUT_EPNUM MIDI_STREAM_OUT_EPNUM | 238 | # define CDC_OUT_EPNUM MIDI_STREAM_OUT_EPNUM |
| 218 | #endif | 239 | #endif |
| 219 | 240 | ||
| 220 | |||
| 221 | #if defined(__AVR_ATmega32U2__) && CDC_OUT_EPNUM > 4 | 241 | #if defined(__AVR_ATmega32U2__) && CDC_OUT_EPNUM > 4 |
| 222 | # error "Endpoints are not available enough to support all functions. Remove some in Makefile.(MOUSEKEY, EXTRAKEY, CONSOLE, NKRO, MIDI, SERIAL)" | 242 | # error "Endpoints are not available enough to support all functions. Remove some in Makefile.(MOUSEKEY, EXTRAKEY, CONSOLE, NKRO, MIDI, SERIAL)" |
| 223 | #endif | 243 | #endif |
| @@ -225,6 +245,7 @@ typedef struct | |||
| 225 | #define KEYBOARD_EPSIZE 8 | 245 | #define KEYBOARD_EPSIZE 8 |
| 226 | #define MOUSE_EPSIZE 8 | 246 | #define MOUSE_EPSIZE 8 |
| 227 | #define EXTRAKEY_EPSIZE 8 | 247 | #define EXTRAKEY_EPSIZE 8 |
| 248 | #define RAW_EPSIZE 32 | ||
| 228 | #define CONSOLE_EPSIZE 32 | 249 | #define CONSOLE_EPSIZE 32 |
| 229 | #define NKRO_EPSIZE 32 | 250 | #define NKRO_EPSIZE 32 |
| 230 | #define MIDI_STREAM_EPSIZE 64 | 251 | #define MIDI_STREAM_EPSIZE 64 |
diff --git a/tmk_core/protocol/lufa/lufa.c b/tmk_core/protocol/lufa/lufa.c index 01c0e45b0..60cba8d2a 100644 --- a/tmk_core/protocol/lufa/lufa.c +++ b/tmk_core/protocol/lufa/lufa.c | |||
| @@ -51,6 +51,9 @@ | |||
| 51 | 51 | ||
| 52 | #include "descriptor.h" | 52 | #include "descriptor.h" |
| 53 | #include "lufa.h" | 53 | #include "lufa.h" |
| 54 | #include "quantum.h" | ||
| 55 | #include <util/atomic.h> | ||
| 56 | #include "outputselect.h" | ||
| 54 | 57 | ||
| 55 | #ifdef NKRO_ENABLE | 58 | #ifdef NKRO_ENABLE |
| 56 | #include "keycode_config.h" | 59 | #include "keycode_config.h" |
| @@ -64,13 +67,29 @@ | |||
| 64 | #endif | 67 | #endif |
| 65 | 68 | ||
| 66 | #ifdef BLUETOOTH_ENABLE | 69 | #ifdef BLUETOOTH_ENABLE |
| 70 | #ifdef MODULE_ADAFRUIT_BLE | ||
| 71 | #include "adafruit_ble.h" | ||
| 72 | #else | ||
| 67 | #include "bluetooth.h" | 73 | #include "bluetooth.h" |
| 74 | #endif | ||
| 68 | #endif | 75 | #endif |
| 69 | 76 | ||
| 70 | #ifdef VIRTSER_ENABLE | 77 | #ifdef VIRTSER_ENABLE |
| 71 | #include "virtser.h" | 78 | #include "virtser.h" |
| 72 | #endif | 79 | #endif |
| 73 | 80 | ||
| 81 | #if (defined(RGB_MIDI) | defined(RGBLIGHT_ANIMATIONS)) & defined(RGBLIGHT_ENABLE) | ||
| 82 | #include "rgblight.h" | ||
| 83 | #endif | ||
| 84 | |||
| 85 | #ifdef MIDI_ENABLE | ||
| 86 | #include "sysex_tools.h" | ||
| 87 | #endif | ||
| 88 | |||
| 89 | #ifdef RAW_ENABLE | ||
| 90 | #include "raw_hid.h" | ||
| 91 | #endif | ||
| 92 | |||
| 74 | uint8_t keyboard_idle = 0; | 93 | uint8_t keyboard_idle = 0; |
| 75 | /* 0: Boot Protocol, 1: Report Protocol(default) */ | 94 | /* 0: Boot Protocol, 1: Report Protocol(default) */ |
| 76 | uint8_t keyboard_protocol = 1; | 95 | uint8_t keyboard_protocol = 1; |
| @@ -79,9 +98,9 @@ static uint8_t keyboard_led_stats = 0; | |||
| 79 | static report_keyboard_t keyboard_report_sent; | 98 | static report_keyboard_t keyboard_report_sent; |
| 80 | 99 | ||
| 81 | #ifdef MIDI_ENABLE | 100 | #ifdef MIDI_ENABLE |
| 82 | void usb_send_func(MidiDevice * device, uint16_t cnt, uint8_t byte0, uint8_t byte1, uint8_t byte2); | 101 | static void usb_send_func(MidiDevice * device, uint16_t cnt, uint8_t byte0, uint8_t byte1, uint8_t byte2); |
| 83 | void usb_get_midi(MidiDevice * device); | 102 | static void usb_get_midi(MidiDevice * device); |
| 84 | void midi_usb_init(MidiDevice * device); | 103 | static void midi_usb_init(MidiDevice * device); |
| 85 | #endif | 104 | #endif |
| 86 | 105 | ||
| 87 | /* Host driver */ | 106 | /* Host driver */ |
| @@ -166,6 +185,80 @@ USB_ClassInfo_CDC_Device_t cdc_device = | |||
| 166 | }; | 185 | }; |
| 167 | #endif | 186 | #endif |
| 168 | 187 | ||
| 188 | #ifdef RAW_ENABLE | ||
| 189 | |||
| 190 | void raw_hid_send( uint8_t *data, uint8_t length ) | ||
| 191 | { | ||
| 192 | // TODO: implement variable size packet | ||
| 193 | if ( length != RAW_EPSIZE ) | ||
| 194 | { | ||
| 195 | return; | ||
| 196 | } | ||
| 197 | |||
| 198 | if (USB_DeviceState != DEVICE_STATE_Configured) | ||
| 199 | { | ||
| 200 | return; | ||
| 201 | } | ||
| 202 | |||
| 203 | // TODO: decide if we allow calls to raw_hid_send() in the middle | ||
| 204 | // of other endpoint usage. | ||
| 205 | uint8_t ep = Endpoint_GetCurrentEndpoint(); | ||
| 206 | |||
| 207 | Endpoint_SelectEndpoint(RAW_IN_EPNUM); | ||
| 208 | |||
| 209 | // Check to see if the host is ready to accept another packet | ||
| 210 | if (Endpoint_IsINReady()) | ||
| 211 | { | ||
| 212 | // Write data | ||
| 213 | Endpoint_Write_Stream_LE(data, RAW_EPSIZE, NULL); | ||
| 214 | // Finalize the stream transfer to send the last packet | ||
| 215 | Endpoint_ClearIN(); | ||
| 216 | } | ||
| 217 | |||
| 218 | Endpoint_SelectEndpoint(ep); | ||
| 219 | } | ||
| 220 | |||
| 221 | __attribute__ ((weak)) | ||
| 222 | void raw_hid_receive( uint8_t *data, uint8_t length ) | ||
| 223 | { | ||
| 224 | // Users should #include "raw_hid.h" in their own code | ||
| 225 | // and implement this function there. Leave this as weak linkage | ||
| 226 | // so users can opt to not handle data coming in. | ||
| 227 | } | ||
| 228 | |||
| 229 | static void raw_hid_task(void) | ||
| 230 | { | ||
| 231 | // Create a temporary buffer to hold the read in data from the host | ||
| 232 | uint8_t data[RAW_EPSIZE]; | ||
| 233 | bool data_read = false; | ||
| 234 | |||
| 235 | // Device must be connected and configured for the task to run | ||
| 236 | if (USB_DeviceState != DEVICE_STATE_Configured) | ||
| 237 | return; | ||
| 238 | |||
| 239 | Endpoint_SelectEndpoint(RAW_OUT_EPNUM); | ||
| 240 | |||
| 241 | // Check to see if a packet has been sent from the host | ||
| 242 | if (Endpoint_IsOUTReceived()) | ||
| 243 | { | ||
| 244 | // Check to see if the packet contains data | ||
| 245 | if (Endpoint_IsReadWriteAllowed()) | ||
| 246 | { | ||
| 247 | /* Read data */ | ||
| 248 | Endpoint_Read_Stream_LE(data, sizeof(data), NULL); | ||
| 249 | data_read = true; | ||
| 250 | } | ||
| 251 | |||
| 252 | // Finalize the stream transfer to receive the last packet | ||
| 253 | Endpoint_ClearOUT(); | ||
| 254 | |||
| 255 | if ( data_read ) | ||
| 256 | { | ||
| 257 | raw_hid_receive( data, sizeof(data) ); | ||
| 258 | } | ||
| 259 | } | ||
| 260 | } | ||
| 261 | #endif | ||
| 169 | 262 | ||
| 170 | /******************************************************************************* | 263 | /******************************************************************************* |
| 171 | * Console | 264 | * Console |
| @@ -285,10 +378,14 @@ void EVENT_USB_Device_WakeUp() | |||
| 285 | #endif | 378 | #endif |
| 286 | } | 379 | } |
| 287 | 380 | ||
| 381 | |||
| 382 | |||
| 288 | #ifdef CONSOLE_ENABLE | 383 | #ifdef CONSOLE_ENABLE |
| 289 | static bool console_flush = false; | 384 | static bool console_flush = false; |
| 290 | #define CONSOLE_FLUSH_SET(b) do { \ | 385 | #define CONSOLE_FLUSH_SET(b) do { \ |
| 291 | uint8_t sreg = SREG; cli(); console_flush = b; SREG = sreg; \ | 386 | ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {\ |
| 387 | console_flush = b; \ | ||
| 388 | } \ | ||
| 292 | } while (0) | 389 | } while (0) |
| 293 | 390 | ||
| 294 | // called every 1ms | 391 | // called every 1ms |
| @@ -302,6 +399,7 @@ void EVENT_USB_Device_StartOfFrame(void) | |||
| 302 | Console_Task(); | 399 | Console_Task(); |
| 303 | console_flush = false; | 400 | console_flush = false; |
| 304 | } | 401 | } |
| 402 | |||
| 305 | #endif | 403 | #endif |
| 306 | 404 | ||
| 307 | /** Event handler for the USB_ConfigurationChanged event. | 405 | /** Event handler for the USB_ConfigurationChanged event. |
| @@ -330,6 +428,14 @@ void EVENT_USB_Device_ConfigurationChanged(void) | |||
| 330 | EXTRAKEY_EPSIZE, ENDPOINT_BANK_SINGLE); | 428 | EXTRAKEY_EPSIZE, ENDPOINT_BANK_SINGLE); |
| 331 | #endif | 429 | #endif |
| 332 | 430 | ||
| 431 | #ifdef RAW_ENABLE | ||
| 432 | /* Setup Raw HID Report Endpoints */ | ||
| 433 | ConfigSuccess &= ENDPOINT_CONFIG(RAW_IN_EPNUM, EP_TYPE_INTERRUPT, ENDPOINT_DIR_IN, | ||
| 434 | RAW_EPSIZE, ENDPOINT_BANK_SINGLE); | ||
| 435 | ConfigSuccess &= ENDPOINT_CONFIG(RAW_OUT_EPNUM, EP_TYPE_INTERRUPT, ENDPOINT_DIR_OUT, | ||
| 436 | RAW_EPSIZE, ENDPOINT_BANK_SINGLE); | ||
| 437 | #endif | ||
| 438 | |||
| 333 | #ifdef CONSOLE_ENABLE | 439 | #ifdef CONSOLE_ENABLE |
| 334 | /* Setup Console HID Report Endpoints */ | 440 | /* Setup Console HID Report Endpoints */ |
| 335 | ConfigSuccess &= ENDPOINT_CONFIG(CONSOLE_IN_EPNUM, EP_TYPE_INTERRUPT, ENDPOINT_DIR_IN, | 441 | ConfigSuccess &= ENDPOINT_CONFIG(CONSOLE_IN_EPNUM, EP_TYPE_INTERRUPT, ENDPOINT_DIR_IN, |
| @@ -485,7 +591,6 @@ void EVENT_USB_Device_ControlRequest(void) | |||
| 485 | 591 | ||
| 486 | /******************************************************************************* | 592 | /******************************************************************************* |
| 487 | * Host driver | 593 | * Host driver |
| 488 | p | ||
| 489 | ******************************************************************************/ | 594 | ******************************************************************************/ |
| 490 | static uint8_t keyboard_leds(void) | 595 | static uint8_t keyboard_leds(void) |
| 491 | { | 596 | { |
| @@ -494,18 +599,25 @@ static uint8_t keyboard_leds(void) | |||
| 494 | 599 | ||
| 495 | static void send_keyboard(report_keyboard_t *report) | 600 | static void send_keyboard(report_keyboard_t *report) |
| 496 | { | 601 | { |
| 602 | uint8_t timeout = 255; | ||
| 603 | uint8_t where = where_to_send(); | ||
| 497 | 604 | ||
| 498 | #ifdef BLUETOOTH_ENABLE | 605 | #ifdef BLUETOOTH_ENABLE |
| 499 | bluefruit_serial_send(0xFD); | 606 | if (where == OUTPUT_BLUETOOTH || where == OUTPUT_USB_AND_BT) { |
| 500 | for (uint8_t i = 0; i < KEYBOARD_EPSIZE; i++) { | 607 | #ifdef MODULE_ADAFRUIT_BLE |
| 608 | adafruit_ble_send_keys(report->mods, report->keys, sizeof(report->keys)); | ||
| 609 | #else | ||
| 610 | bluefruit_serial_send(0xFD); | ||
| 611 | for (uint8_t i = 0; i < KEYBOARD_EPSIZE; i++) { | ||
| 501 | bluefruit_serial_send(report->raw[i]); | 612 | bluefruit_serial_send(report->raw[i]); |
| 502 | } | 613 | } |
| 614 | #endif | ||
| 615 | } | ||
| 503 | #endif | 616 | #endif |
| 504 | 617 | ||
| 505 | uint8_t timeout = 255; | 618 | if (where != OUTPUT_USB && where != OUTPUT_USB_AND_BT) { |
| 506 | 619 | return; | |
| 507 | if (USB_DeviceState != DEVICE_STATE_Configured) | 620 | } |
| 508 | return; | ||
| 509 | 621 | ||
| 510 | /* Select the Keyboard Report Endpoint */ | 622 | /* Select the Keyboard Report Endpoint */ |
| 511 | #ifdef NKRO_ENABLE | 623 | #ifdef NKRO_ENABLE |
| @@ -543,23 +655,31 @@ static void send_keyboard(report_keyboard_t *report) | |||
| 543 | static void send_mouse(report_mouse_t *report) | 655 | static void send_mouse(report_mouse_t *report) |
| 544 | { | 656 | { |
| 545 | #ifdef MOUSE_ENABLE | 657 | #ifdef MOUSE_ENABLE |
| 658 | uint8_t timeout = 255; | ||
| 659 | uint8_t where = where_to_send(); | ||
| 546 | 660 | ||
| 547 | #ifdef BLUETOOTH_ENABLE | 661 | #ifdef BLUETOOTH_ENABLE |
| 548 | bluefruit_serial_send(0xFD); | 662 | if (where == OUTPUT_BLUETOOTH || where == OUTPUT_USB_AND_BT) { |
| 549 | bluefruit_serial_send(0x00); | 663 | #ifdef MODULE_ADAFRUIT_BLE |
| 550 | bluefruit_serial_send(0x03); | 664 | // FIXME: mouse buttons |
| 551 | bluefruit_serial_send(report->buttons); | 665 | adafruit_ble_send_mouse_move(report->x, report->y, report->v, report->h); |
| 552 | bluefruit_serial_send(report->x); | 666 | #else |
| 553 | bluefruit_serial_send(report->y); | 667 | bluefruit_serial_send(0xFD); |
| 554 | bluefruit_serial_send(report->v); // should try sending the wheel v here | 668 | bluefruit_serial_send(0x00); |
| 555 | bluefruit_serial_send(report->h); // should try sending the wheel h here | 669 | bluefruit_serial_send(0x03); |
| 556 | bluefruit_serial_send(0x00); | 670 | bluefruit_serial_send(report->buttons); |
| 671 | bluefruit_serial_send(report->x); | ||
| 672 | bluefruit_serial_send(report->y); | ||
| 673 | bluefruit_serial_send(report->v); // should try sending the wheel v here | ||
| 674 | bluefruit_serial_send(report->h); // should try sending the wheel h here | ||
| 675 | bluefruit_serial_send(0x00); | ||
| 676 | #endif | ||
| 677 | } | ||
| 557 | #endif | 678 | #endif |
| 558 | 679 | ||
| 559 | uint8_t timeout = 255; | 680 | if (where != OUTPUT_USB && where != OUTPUT_USB_AND_BT) { |
| 560 | 681 | return; | |
| 561 | if (USB_DeviceState != DEVICE_STATE_Configured) | 682 | } |
| 562 | return; | ||
| 563 | 683 | ||
| 564 | /* Select the Mouse Report Endpoint */ | 684 | /* Select the Mouse Report Endpoint */ |
| 565 | Endpoint_SelectEndpoint(MOUSE_IN_EPNUM); | 685 | Endpoint_SelectEndpoint(MOUSE_IN_EPNUM); |
| @@ -585,7 +705,7 @@ static void send_system(uint16_t data) | |||
| 585 | 705 | ||
| 586 | report_extra_t r = { | 706 | report_extra_t r = { |
| 587 | .report_id = REPORT_ID_SYSTEM, | 707 | .report_id = REPORT_ID_SYSTEM, |
| 588 | .usage = data | 708 | .usage = data - SYSTEM_POWER_DOWN + 1 |
| 589 | }; | 709 | }; |
| 590 | Endpoint_SelectEndpoint(EXTRAKEY_IN_EPNUM); | 710 | Endpoint_SelectEndpoint(EXTRAKEY_IN_EPNUM); |
| 591 | 711 | ||
| @@ -599,27 +719,34 @@ static void send_system(uint16_t data) | |||
| 599 | 719 | ||
| 600 | static void send_consumer(uint16_t data) | 720 | static void send_consumer(uint16_t data) |
| 601 | { | 721 | { |
| 722 | uint8_t timeout = 255; | ||
| 723 | uint8_t where = where_to_send(); | ||
| 602 | 724 | ||
| 603 | #ifdef BLUETOOTH_ENABLE | 725 | #ifdef BLUETOOTH_ENABLE |
| 604 | static uint16_t last_data = 0; | 726 | if (where == OUTPUT_BLUETOOTH || where == OUTPUT_USB_AND_BT) { |
| 605 | if (data == last_data) return; | 727 | #ifdef MODULE_ADAFRUIT_BLE |
| 606 | last_data = data; | 728 | adafruit_ble_send_consumer_key(data, 0); |
| 607 | uint16_t bitmap = CONSUMER2BLUEFRUIT(data); | 729 | #else |
| 608 | bluefruit_serial_send(0xFD); | 730 | static uint16_t last_data = 0; |
| 609 | bluefruit_serial_send(0x00); | 731 | if (data == last_data) return; |
| 610 | bluefruit_serial_send(0x02); | 732 | last_data = data; |
| 611 | bluefruit_serial_send((bitmap>>8)&0xFF); | 733 | uint16_t bitmap = CONSUMER2BLUEFRUIT(data); |
| 612 | bluefruit_serial_send(bitmap&0xFF); | 734 | bluefruit_serial_send(0xFD); |
| 613 | bluefruit_serial_send(0x00); | 735 | bluefruit_serial_send(0x00); |
| 614 | bluefruit_serial_send(0x00); | 736 | bluefruit_serial_send(0x02); |
| 615 | bluefruit_serial_send(0x00); | 737 | bluefruit_serial_send((bitmap>>8)&0xFF); |
| 616 | bluefruit_serial_send(0x00); | 738 | bluefruit_serial_send(bitmap&0xFF); |
| 739 | bluefruit_serial_send(0x00); | ||
| 740 | bluefruit_serial_send(0x00); | ||
| 741 | bluefruit_serial_send(0x00); | ||
| 742 | bluefruit_serial_send(0x00); | ||
| 743 | #endif | ||
| 744 | } | ||
| 617 | #endif | 745 | #endif |
| 618 | 746 | ||
| 619 | uint8_t timeout = 255; | 747 | if (where != OUTPUT_USB && where != OUTPUT_USB_AND_BT) { |
| 620 | 748 | return; | |
| 621 | if (USB_DeviceState != DEVICE_STATE_Configured) | 749 | } |
| 622 | return; | ||
| 623 | 750 | ||
| 624 | report_extra_t r = { | 751 | report_extra_t r = { |
| 625 | .report_id = REPORT_ID_CONSUMER, | 752 | .report_id = REPORT_ID_CONSUMER, |
| @@ -709,7 +836,7 @@ int8_t sendchar(uint8_t c) | |||
| 709 | ******************************************************************************/ | 836 | ******************************************************************************/ |
| 710 | 837 | ||
| 711 | #ifdef MIDI_ENABLE | 838 | #ifdef MIDI_ENABLE |
| 712 | void usb_send_func(MidiDevice * device, uint16_t cnt, uint8_t byte0, uint8_t byte1, uint8_t byte2) { | 839 | static void usb_send_func(MidiDevice * device, uint16_t cnt, uint8_t byte0, uint8_t byte1, uint8_t byte2) { |
| 713 | MIDI_EventPacket_t event; | 840 | MIDI_EventPacket_t event; |
| 714 | event.Data1 = byte0; | 841 | event.Data1 = byte0; |
| 715 | event.Data2 = byte1; | 842 | event.Data2 = byte1; |
| @@ -769,7 +896,7 @@ void usb_send_func(MidiDevice * device, uint16_t cnt, uint8_t byte0, uint8_t byt | |||
| 769 | USB_USBTask(); | 896 | USB_USBTask(); |
| 770 | } | 897 | } |
| 771 | 898 | ||
| 772 | void usb_get_midi(MidiDevice * device) { | 899 | static void usb_get_midi(MidiDevice * device) { |
| 773 | MIDI_EventPacket_t event; | 900 | MIDI_EventPacket_t event; |
| 774 | while (MIDI_Device_ReceiveEventPacket(&USB_MIDI_Interface, &event)) { | 901 | while (MIDI_Device_ReceiveEventPacket(&USB_MIDI_Interface, &event)) { |
| 775 | 902 | ||
| @@ -799,12 +926,12 @@ void usb_get_midi(MidiDevice * device) { | |||
| 799 | USB_USBTask(); | 926 | USB_USBTask(); |
| 800 | } | 927 | } |
| 801 | 928 | ||
| 802 | void midi_usb_init(MidiDevice * device){ | 929 | static void midi_usb_init(MidiDevice * device){ |
| 803 | midi_device_init(device); | 930 | midi_device_init(device); |
| 804 | midi_device_set_send_func(device, usb_send_func); | 931 | midi_device_set_send_func(device, usb_send_func); |
| 805 | midi_device_set_pre_input_process_func(device, usb_get_midi); | 932 | midi_device_set_pre_input_process_func(device, usb_get_midi); |
| 806 | 933 | ||
| 807 | SetupHardware(); | 934 | // SetupHardware(); |
| 808 | sei(); | 935 | sei(); |
| 809 | } | 936 | } |
| 810 | 937 | ||
| @@ -969,16 +1096,23 @@ void cc_callback(MidiDevice * device, | |||
| 969 | uint8_t chan, uint8_t num, uint8_t val); | 1096 | uint8_t chan, uint8_t num, uint8_t val); |
| 970 | void sysex_callback(MidiDevice * device, | 1097 | void sysex_callback(MidiDevice * device, |
| 971 | uint16_t start, uint8_t length, uint8_t * data); | 1098 | uint16_t start, uint8_t length, uint8_t * data); |
| 1099 | |||
| 1100 | void setup_midi(void) | ||
| 1101 | { | ||
| 1102 | #ifdef MIDI_ADVANCED | ||
| 1103 | midi_init(); | ||
| 1104 | #endif | ||
| 1105 | midi_device_init(&midi_device); | ||
| 1106 | midi_device_set_send_func(&midi_device, usb_send_func); | ||
| 1107 | midi_device_set_pre_input_process_func(&midi_device, usb_get_midi); | ||
| 1108 | } | ||
| 972 | #endif | 1109 | #endif |
| 973 | 1110 | ||
| 974 | int main(void) __attribute__ ((weak)); | 1111 | int main(void) __attribute__ ((weak)); |
| 975 | int main(void) | 1112 | int main(void) |
| 976 | { | 1113 | { |
| 977 | |||
| 978 | #ifdef MIDI_ENABLE | 1114 | #ifdef MIDI_ENABLE |
| 979 | midi_device_init(&midi_device); | 1115 | setup_midi(); |
| 980 | midi_device_set_send_func(&midi_device, usb_send_func); | ||
| 981 | midi_device_set_pre_input_process_func(&midi_device, usb_get_midi); | ||
| 982 | #endif | 1116 | #endif |
| 983 | 1117 | ||
| 984 | setup_mcu(); | 1118 | setup_mcu(); |
| @@ -998,7 +1132,7 @@ int main(void) | |||
| 998 | // midi_send_noteoff(&midi_device, 0, 64, 127); | 1132 | // midi_send_noteoff(&midi_device, 0, 64, 127); |
| 999 | #endif | 1133 | #endif |
| 1000 | 1134 | ||
| 1001 | #ifdef BLUETOOTH_ENABLE | 1135 | #ifdef MODULE_ADAFRUIT_EZKEY |
| 1002 | serial_init(); | 1136 | serial_init(); |
| 1003 | #endif | 1137 | #endif |
| 1004 | 1138 | ||
| @@ -1029,7 +1163,7 @@ int main(void) | |||
| 1029 | 1163 | ||
| 1030 | print("Keyboard start.\n"); | 1164 | print("Keyboard start.\n"); |
| 1031 | while (1) { | 1165 | while (1) { |
| 1032 | #ifndef BLUETOOTH_ENABLE | 1166 | #if !defined(BLUETOOTH_ENABLE) |
| 1033 | while (USB_DeviceState == DEVICE_STATE_Suspended) { | 1167 | while (USB_DeviceState == DEVICE_STATE_Suspended) { |
| 1034 | print("[s]"); | 1168 | print("[s]"); |
| 1035 | suspend_power_down(); | 1169 | suspend_power_down(); |
| @@ -1039,20 +1173,36 @@ int main(void) | |||
| 1039 | } | 1173 | } |
| 1040 | #endif | 1174 | #endif |
| 1041 | 1175 | ||
| 1176 | keyboard_task(); | ||
| 1177 | |||
| 1042 | #ifdef MIDI_ENABLE | 1178 | #ifdef MIDI_ENABLE |
| 1043 | midi_device_process(&midi_device); | 1179 | midi_device_process(&midi_device); |
| 1044 | // MIDI_Task(); | 1180 | #ifdef MIDI_ADVANCED |
| 1181 | midi_task(); | ||
| 1182 | #endif | ||
| 1183 | #endif | ||
| 1184 | |||
| 1185 | #if defined(RGBLIGHT_ANIMATIONS) & defined(RGBLIGHT_ENABLE) | ||
| 1186 | rgblight_task(); | ||
| 1187 | #endif | ||
| 1188 | |||
| 1189 | #ifdef MODULE_ADAFRUIT_BLE | ||
| 1190 | adafruit_ble_task(); | ||
| 1045 | #endif | 1191 | #endif |
| 1046 | keyboard_task(); | ||
| 1047 | 1192 | ||
| 1048 | #ifdef VIRTSER_ENABLE | 1193 | #ifdef VIRTSER_ENABLE |
| 1049 | virtser_task(); | 1194 | virtser_task(); |
| 1050 | CDC_Device_USBTask(&cdc_device); | 1195 | CDC_Device_USBTask(&cdc_device); |
| 1051 | #endif | 1196 | #endif |
| 1052 | 1197 | ||
| 1198 | #ifdef RAW_ENABLE | ||
| 1199 | raw_hid_task(); | ||
| 1200 | #endif | ||
| 1201 | |||
| 1053 | #if !defined(INTERRUPT_CONTROL_ENDPOINT) | 1202 | #if !defined(INTERRUPT_CONTROL_ENDPOINT) |
| 1054 | USB_USBTask(); | 1203 | USB_USBTask(); |
| 1055 | #endif | 1204 | #endif |
| 1205 | |||
| 1056 | } | 1206 | } |
| 1057 | } | 1207 | } |
| 1058 | 1208 | ||
| @@ -1077,15 +1227,50 @@ void fallthrough_callback(MidiDevice * device, | |||
| 1077 | #endif | 1227 | #endif |
| 1078 | } | 1228 | } |
| 1079 | 1229 | ||
| 1230 | |||
| 1080 | void cc_callback(MidiDevice * device, | 1231 | void cc_callback(MidiDevice * device, |
| 1081 | uint8_t chan, uint8_t num, uint8_t val) { | 1232 | uint8_t chan, uint8_t num, uint8_t val) { |
| 1082 | //sending it back on the next channel | 1233 | //sending it back on the next channel |
| 1083 | midi_send_cc(device, (chan + 1) % 16, num, val); | 1234 | // midi_send_cc(device, (chan + 1) % 16, num, val); |
| 1084 | } | 1235 | } |
| 1085 | 1236 | ||
| 1086 | void sysex_callback(MidiDevice * device, | 1237 | #ifdef API_SYSEX_ENABLE |
| 1087 | uint16_t start, uint8_t length, uint8_t * data) { | 1238 | uint8_t midi_buffer[MIDI_SYSEX_BUFFER] = {0}; |
| 1088 | for (int i = 0; i < length; i++) | 1239 | #endif |
| 1089 | midi_send_cc(device, 15, 0x7F & data[i], 0x7F & (start + i)); | 1240 | |
| 1241 | void sysex_callback(MidiDevice * device, uint16_t start, uint8_t length, uint8_t * data) { | ||
| 1242 | #ifdef API_SYSEX_ENABLE | ||
| 1243 | // SEND_STRING("\n"); | ||
| 1244 | // send_word(start); | ||
| 1245 | // SEND_STRING(": "); | ||
| 1246 | // Don't store the header | ||
| 1247 | int16_t pos = start - 4; | ||
| 1248 | for (uint8_t place = 0; place < length; place++) { | ||
| 1249 | // send_byte(*data); | ||
| 1250 | if (pos >= 0) { | ||
| 1251 | if (*data == 0xF7) { | ||
| 1252 | // SEND_STRING("\nRD: "); | ||
| 1253 | // for (uint8_t i = 0; i < start + place + 1; i++){ | ||
| 1254 | // send_byte(midi_buffer[i]); | ||
| 1255 | // SEND_STRING(" "); | ||
| 1256 | // } | ||
| 1257 | const unsigned decoded_length = sysex_decoded_length(pos); | ||
| 1258 | uint8_t decoded[API_SYSEX_MAX_SIZE]; | ||
| 1259 | sysex_decode(decoded, midi_buffer, pos); | ||
| 1260 | process_api(decoded_length, decoded); | ||
| 1261 | return; | ||
| 1262 | } | ||
| 1263 | else if (pos >= MIDI_SYSEX_BUFFER) { | ||
| 1264 | return; | ||
| 1265 | } | ||
| 1266 | midi_buffer[pos] = *data; | ||
| 1267 | } | ||
| 1268 | // SEND_STRING(" "); | ||
| 1269 | data++; | ||
| 1270 | pos++; | ||
| 1271 | } | ||
| 1272 | #endif | ||
| 1090 | } | 1273 | } |
| 1274 | |||
| 1275 | |||
| 1091 | #endif | 1276 | #endif |
diff --git a/tmk_core/protocol/lufa/lufa.h b/tmk_core/protocol/lufa/lufa.h index aad08d640..a51573786 100644 --- a/tmk_core/protocol/lufa/lufa.h +++ b/tmk_core/protocol/lufa/lufa.h | |||
| @@ -49,7 +49,7 @@ | |||
| 49 | #include <LUFA/Drivers/USB/USB.h> | 49 | #include <LUFA/Drivers/USB/USB.h> |
| 50 | #include "host.h" | 50 | #include "host.h" |
| 51 | #ifdef MIDI_ENABLE | 51 | #ifdef MIDI_ENABLE |
| 52 | #include "midi.h" | 52 | #include "process_midi.h" |
| 53 | #endif | 53 | #endif |
| 54 | #ifdef __cplusplus | 54 | #ifdef __cplusplus |
| 55 | extern "C" { | 55 | extern "C" { |
| @@ -68,8 +68,19 @@ typedef struct { | |||
| 68 | } __attribute__ ((packed)) report_extra_t; | 68 | } __attribute__ ((packed)) report_extra_t; |
| 69 | 69 | ||
| 70 | #ifdef MIDI_ENABLE | 70 | #ifdef MIDI_ENABLE |
| 71 | void MIDI_Task(void); | 71 | void MIDI_Task(void); |
| 72 | MidiDevice midi_device; | 72 | MidiDevice midi_device; |
| 73 | #endif | ||
| 74 | |||
| 75 | #ifdef API_ENABLE | ||
| 76 | #include "api.h" | ||
| 77 | #endif | ||
| 78 | |||
| 79 | #ifdef API_SYSEX_ENABLE | ||
| 80 | #include "api_sysex.h" | ||
| 81 | // Allocate space for encoding overhead. | ||
| 82 | //The header and terminator are not stored to save a few bytes of precious ram | ||
| 83 | #define MIDI_SYSEX_BUFFER (API_SYSEX_MAX_SIZE + API_SYSEX_MAX_SIZE / 7 + (API_SYSEX_MAX_SIZE % 7 ? 1 : 0)) | ||
| 73 | #endif | 84 | #endif |
| 74 | 85 | ||
| 75 | // #if LUFA_VERSION_INTEGER < 0x120730 | 86 | // #if LUFA_VERSION_INTEGER < 0x120730 |
diff --git a/tmk_core/protocol/lufa/outputselect.c b/tmk_core/protocol/lufa/outputselect.c new file mode 100644 index 000000000..0df5d3b75 --- /dev/null +++ b/tmk_core/protocol/lufa/outputselect.c | |||
| @@ -0,0 +1,56 @@ | |||
| 1 | /* | ||
| 2 | Copyright 2017 Priyadi Iman Nurcahyo | ||
| 3 | This program is free software: you can redistribute it and/or modify | ||
| 4 | it under the terms of the GNU General Public License as published by | ||
| 5 | the Free Software Foundation, either version 2 of the License, or | ||
| 6 | (at your option) any later version. | ||
| 7 | This program is distributed in the hope that it will be useful, | ||
| 8 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 9 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 10 | GNU General Public License for more details. | ||
| 11 | You should have received a copy of the GNU General Public License | ||
| 12 | along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
| 13 | */ | ||
| 14 | |||
| 15 | #include "lufa.h" | ||
| 16 | #include "outputselect.h" | ||
| 17 | #ifdef MODULE_ADAFRUIT_BLE | ||
| 18 | #include "adafruit_ble.h" | ||
| 19 | #endif | ||
| 20 | |||
| 21 | uint8_t desired_output = OUTPUT_DEFAULT; | ||
| 22 | |||
| 23 | void set_output(uint8_t output) { | ||
| 24 | set_output_user(output); | ||
| 25 | desired_output = output; | ||
| 26 | } | ||
| 27 | |||
| 28 | __attribute__((weak)) | ||
| 29 | void set_output_user(uint8_t output) { | ||
| 30 | } | ||
| 31 | |||
| 32 | uint8_t auto_detect_output(void) { | ||
| 33 | if (USB_DeviceState == DEVICE_STATE_Configured) { | ||
| 34 | return OUTPUT_USB; | ||
| 35 | } | ||
| 36 | |||
| 37 | #ifdef MODULE_ADAFRUIT_BLE | ||
| 38 | if (adafruit_ble_is_connected()) { | ||
| 39 | return OUTPUT_BLUETOOTH; | ||
| 40 | } | ||
| 41 | #endif | ||
| 42 | |||
| 43 | #ifdef BLUETOOTH_ENABLE | ||
| 44 | return OUTPUT_BLUETOOTH; // should check if BT is connected here | ||
| 45 | #endif | ||
| 46 | |||
| 47 | return OUTPUT_NONE; | ||
| 48 | } | ||
| 49 | |||
| 50 | uint8_t where_to_send(void) { | ||
| 51 | if (desired_output == OUTPUT_AUTO) { | ||
| 52 | return auto_detect_output(); | ||
| 53 | } | ||
| 54 | return desired_output; | ||
| 55 | } | ||
| 56 | |||
diff --git a/tmk_core/protocol/lufa/outputselect.h b/tmk_core/protocol/lufa/outputselect.h new file mode 100644 index 000000000..28cc3298e --- /dev/null +++ b/tmk_core/protocol/lufa/outputselect.h | |||
| @@ -0,0 +1,40 @@ | |||
| 1 | /* | ||
| 2 | Copyright 2017 Priyadi Iman Nurcahyo | ||
| 3 | This program is free software: you can redistribute it and/or modify | ||
| 4 | it under the terms of the GNU General Public License as published by | ||
| 5 | the Free Software Foundation, either version 2 of the License, or | ||
| 6 | (at your option) any later version. | ||
| 7 | This program is distributed in the hope that it will be useful, | ||
| 8 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 9 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 10 | GNU General Public License for more details. | ||
| 11 | You should have received a copy of the GNU General Public License | ||
| 12 | along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
| 13 | */ | ||
| 14 | |||
| 15 | enum outputs { | ||
| 16 | OUTPUT_AUTO, | ||
| 17 | |||
| 18 | OUTPUT_NONE, | ||
| 19 | OUTPUT_USB, | ||
| 20 | OUTPUT_BLUETOOTH, | ||
| 21 | |||
| 22 | // backward compatibility | ||
| 23 | OUTPUT_USB_AND_BT | ||
| 24 | }; | ||
| 25 | |||
| 26 | /** | ||
| 27 | * backward compatibility for BLUETOOTH_ENABLE, send to BT and USB by default | ||
| 28 | */ | ||
| 29 | #ifndef OUTPUT_DEFAULT | ||
| 30 | #ifdef BLUETOOTH_ENABLE | ||
| 31 | #define OUTPUT_DEFAULT OUTPUT_USB_AND_BT | ||
| 32 | #else | ||
| 33 | #define OUTPUT_DEFAULT OUTPUT_AUTO | ||
| 34 | #endif | ||
| 35 | #endif | ||
| 36 | |||
| 37 | void set_output(uint8_t output); | ||
| 38 | void set_output_user(uint8_t output); | ||
| 39 | uint8_t auto_detect_output(void); | ||
| 40 | uint8_t where_to_send(void); \ No newline at end of file | ||
diff --git a/tmk_core/protocol/lufa/ringbuffer.hpp b/tmk_core/protocol/lufa/ringbuffer.hpp new file mode 100644 index 000000000..70a3c4881 --- /dev/null +++ b/tmk_core/protocol/lufa/ringbuffer.hpp | |||
| @@ -0,0 +1,66 @@ | |||
| 1 | #pragma once | ||
| 2 | // A simple ringbuffer holding Size elements of type T | ||
| 3 | template <typename T, uint8_t Size> | ||
| 4 | class RingBuffer { | ||
| 5 | protected: | ||
| 6 | T buf_[Size]; | ||
| 7 | uint8_t head_{0}, tail_{0}; | ||
| 8 | public: | ||
| 9 | inline uint8_t nextPosition(uint8_t position) { | ||
| 10 | return (position + 1) % Size; | ||
| 11 | } | ||
| 12 | |||
| 13 | inline uint8_t prevPosition(uint8_t position) { | ||
| 14 | if (position == 0) { | ||
| 15 | return Size - 1; | ||
| 16 | } | ||
| 17 | return position - 1; | ||
| 18 | } | ||
| 19 | |||
| 20 | inline bool enqueue(const T &item) { | ||
| 21 | static_assert(Size > 1, "RingBuffer size must be > 1"); | ||
| 22 | uint8_t next = nextPosition(head_); | ||
| 23 | if (next == tail_) { | ||
| 24 | // Full | ||
| 25 | return false; | ||
| 26 | } | ||
| 27 | |||
| 28 | buf_[head_] = item; | ||
| 29 | head_ = next; | ||
| 30 | return true; | ||
| 31 | } | ||
| 32 | |||
| 33 | inline bool get(T &dest, bool commit = true) { | ||
| 34 | auto tail = tail_; | ||
| 35 | if (tail == head_) { | ||
| 36 | // No more data | ||
| 37 | return false; | ||
| 38 | } | ||
| 39 | |||
| 40 | dest = buf_[tail]; | ||
| 41 | tail = nextPosition(tail); | ||
| 42 | |||
| 43 | if (commit) { | ||
| 44 | tail_ = tail; | ||
| 45 | } | ||
| 46 | return true; | ||
| 47 | } | ||
| 48 | |||
| 49 | inline bool empty() const { return head_ == tail_; } | ||
| 50 | |||
| 51 | inline uint8_t size() const { | ||
| 52 | int diff = head_ - tail_; | ||
| 53 | if (diff >= 0) { | ||
| 54 | return diff; | ||
| 55 | } | ||
| 56 | return Size + diff; | ||
| 57 | } | ||
| 58 | |||
| 59 | inline T& front() { | ||
| 60 | return buf_[tail_]; | ||
| 61 | } | ||
| 62 | |||
| 63 | inline bool peek(T &item) { | ||
| 64 | return get(item, false); | ||
| 65 | } | ||
| 66 | }; | ||
