logo

qmk_firmware

custom branch of QMK firmware git clone https://anongit.hacktivis.me/git/qmk_firmware.git

qp_comms_i2c.c (3067B)

    

  1. // Copyright 2022 Nick Brassel (@tzarc)
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  4. #ifdef QUANTUM_PAINTER_I2C_ENABLE
  6. #    include "i2c_master.h"
  7. #    include "qp_comms_i2c.h"
  9. ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
  10. // Helpers
  12. static uint32_t qp_comms_i2c_send_raw(painter_device_t device, const void *data, uint32_t byte_count) {
  13.     painter_driver_t *     driver       = (painter_driver_t *)device;
  14.     qp_comms_i2c_config_t *comms_config = (qp_comms_i2c_config_t *)driver->comms_config;
  15.     i2c_status_t           res          = i2c_transmit(comms_config->chip_address << 1, data, byte_count, I2C_TIMEOUT);
  16.     if (res < 0) {
  17.         return 0;
  18.     }
  19.     return byte_count;
  20. }
  22. ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
  23. // Base I2C support
  25. bool qp_comms_i2c_init(painter_device_t device) {
  26.     i2c_init();
  27.     return true;
  28. }
  30. bool qp_comms_i2c_start(painter_device_t device) {
  31.     return true;
  32. }
  34. uint32_t qp_comms_i2c_send_data(painter_device_t device, const void *data, uint32_t byte_count) {
  35.     return qp_comms_i2c_send_raw(device, data, byte_count);
  36. }
  38. void qp_comms_i2c_stop(painter_device_t device) {}
  40. ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
  41. // Command+Data I2C support
  43. static const uint8_t cmd_byte  = 0x00;
  44. static const uint8_t data_byte = 0x40;
  46. void qp_comms_i2c_cmddata_send_command(painter_device_t device, uint8_t cmd) {
  47.     uint8_t buf[2] = {cmd_byte, cmd};
  48.     qp_comms_i2c_send_raw(device, &buf, 2);
  49. }
  51. uint32_t qp_comms_i2c_cmddata_send_data(painter_device_t device, const void *data, uint32_t byte_count) {
  52.     uint8_t buf[1 + byte_count];
  53.     buf[0] = data_byte;
  54.     memcpy(&buf[1], data, byte_count);
  55.     if (qp_comms_i2c_send_raw(device, buf, sizeof(buf)) != sizeof(buf)) {
  56.         return 0;
  57.     }
  58.     return byte_count;
  59. }
  61. void qp_comms_i2c_bulk_command_sequence(painter_device_t device, const uint8_t *sequence, size_t sequence_len) {
  62.     uint8_t buf[32];
  63.     for (size_t i = 0; i < sequence_len;) {
  64.         uint8_t command   = sequence[i];
  65.         uint8_t delay     = sequence[i + 1];
  66.         uint8_t num_bytes = sequence[i + 2];
  67.         buf[0]            = cmd_byte;
  68.         buf[1]            = command;
  69.         memcpy(&buf[2], &sequence[i + 3], num_bytes);
  70.         qp_comms_i2c_send_raw(device, buf, num_bytes + 2);
  71.         if (delay > 0) {
  72.             wait_ms(delay);
  73.         }
  74.         i += (3 + num_bytes);
  75.     }
  76. }
  78. const painter_comms_with_command_vtable_t i2c_comms_cmddata_vtable = {
  79.     .base =
  80.         {
  81.             .comms_init  = qp_comms_i2c_init,
  82.             .comms_start = qp_comms_i2c_start,
  83.             .comms_send  = qp_comms_i2c_cmddata_send_data,
  84.             .comms_stop  = qp_comms_i2c_stop,
  85.         },
  86.     .send_command          = qp_comms_i2c_cmddata_send_command,
  87.     .bulk_command_sequence = qp_comms_i2c_bulk_command_sequence,
  88. };
  90. #endif // QUANTUM_PAINTER_I2C_ENABLE