qmk_firmware_custom/drivers/qwiic/joystiic.c

/* Copyright 2018 Jack Humbert <jack.humb@gmail.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "joystiic.h"
#include "print.h"
#include "action.h"

#define JOYSTIIC_DEFAULT_ADDR       0x20
#define JOYSTIIC_COMMAND_HORIZONTAL 0x00
#define JOYSTIIC_COMMAND_VERTICAL   0x02
#define JOYSTIIC_COMMAND_BUTTON     0x04

#define JOYSTIIC_CENTER 512
#define JOYSTIIC_DEADZONE 200

uint16_t joystiic_horizontal;
uint16_t joystiic_vertical;
bool joystiic_button;

uint8_t joystiic_tx[1];
uint8_t joystiic_rx_horizontal[2];
uint8_t joystiic_rx_vertical[2];
uint8_t joystiic_rx_button[1];

bool joystiic_triggered[5] = {0};

void joystiic_init(void) {
  i2c_init();
  i2c_start(JOYSTIIC_DEFAULT_ADDR);
}

void joystiic_update(uint16_t horizontal, uint16_t vertical, bool button) {
  joystiic_update_kb(horizontal, vertical, button);
  joystiic_update_user(horizontal, vertical, button);
}

__attribute__ ((weak))
void joystiic_update_kb(uint16_t horizontal, uint16_t vertical, bool button) { }

__attribute__ ((weak))
void joystiic_update_user(uint16_t horizontal, uint16_t vertical, bool button) { }

void joystiic_trigger(uint8_t trigger, bool active) {
  joystiic_trigger_kb(trigger, active);
  joystiic_trigger_user(trigger, active);
}

__attribute__ ((weak))
void joystiic_trigger_kb(uint8_t trigger, bool active) { }

__attribute__ ((weak))
void joystiic_trigger_user(uint8_t trigger, bool active) { }

void joystiic_trigger_if_not(uint8_t trigger, bool active) {
  if (joystiic_triggered[trigger] != active) {
    joystiic_triggered[trigger] = active;
    joystiic_trigger(trigger, active);
  }
}

void joystiic_task(void) {
  // get horizontal axis
  joystiic_tx[0] = JOYSTIIC_COMMAND_HORIZONTAL;

  if (MSG_OK != i2c_transmit_receive(JOYSTIIC_DEFAULT_ADDR << 1,
    joystiic_tx, 1,
    joystiic_rx_horizontal, 2
  )) {
    printf("error hori\n");
  }

  joystiic_horizontal = ((uint16_t)joystiic_rx_horizontal[0] << 8) | joystiic_rx_horizontal[1];

  joystiic_trigger_if_not(JOYSTIIC_LEFT, joystiic_horizontal > (JOYSTIIC_CENTER + JOYSTIIC_DEADZONE));
  joystiic_trigger_if_not(JOYSTIIC_RIGHT, joystiic_horizontal < (JOYSTIIC_CENTER - JOYSTIIC_DEADZONE));

  // get vertical axis
  joystiic_tx[0] = JOYSTIIC_COMMAND_VERTICAL;
  if (MSG_OK != i2c_transmit_receive(JOYSTIIC_DEFAULT_ADDR << 1,
    joystiic_tx, 1,
    joystiic_rx_vertical, 2
  )) {
    printf("error vert\n");
  }

  joystiic_vertical = ((uint16_t)joystiic_rx_vertical[0] << 8) | joystiic_rx_vertical[1];

  joystiic_trigger_if_not(JOYSTIIC_UP, joystiic_vertical > (JOYSTIIC_CENTER + JOYSTIIC_DEADZONE));
  joystiic_trigger_if_not(JOYSTIIC_DOWN, joystiic_vertical < (JOYSTIIC_CENTER - JOYSTIIC_DEADZONE));

  // get button press
  joystiic_tx[0] = JOYSTIIC_COMMAND_BUTTON;
  if (MSG_OK != i2c_transmit_receive(JOYSTIIC_DEFAULT_ADDR << 1,
    joystiic_tx, 1,
    joystiic_rx_button, 1
  )) {
    printf("error vert\n");
  }

  joystiic_button = !joystiic_rx_button[0];

  joystiic_trigger_if_not(JOYSTIIC_PRESS, joystiic_button);

  joystiic_update(joystiic_horizontal, joystiic_vertical, joystiic_button);

  //printf("%d\n", joystiic[0]);

  // SEND_STRING("H: ");
  // send_word(joystiic_rx_horizontal[0]);
  // tap_code(KC_SPACE);
  // send_word(joystiic_rx_horizontal[1]);
  // tap_code(KC_SPACE);
}