mirror of
https://github.com/mfulz/qmk_firmware.git
synced 2025-10-31 21:32:31 +01:00
887e96cbb2
Add one analog axis to the onekey:joystick keymap. Fix pin state save and restore during joystick analog read for STM32 MCUs.
169 lines
6.3 KiB
C
169 lines
6.3 KiB
C
#include "joystick.h"
|
|
#include "process_joystick.h"
|
|
|
|
#include "analog.h"
|
|
|
|
#include <string.h>
|
|
#include <math.h>
|
|
|
|
bool process_joystick_buttons(uint16_t keycode, keyrecord_t *record);
|
|
|
|
bool process_joystick(uint16_t keycode, keyrecord_t *record) {
|
|
if (process_joystick_buttons(keycode, record) && (joystick_status.status & JS_UPDATED) > 0) {
|
|
send_joystick_packet(&joystick_status);
|
|
joystick_status.status &= ~JS_UPDATED;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
__attribute__((weak))
|
|
void joystick_task(void) {
|
|
if (process_joystick_analogread() && (joystick_status.status & JS_UPDATED)) {
|
|
send_joystick_packet(&joystick_status);
|
|
joystick_status.status &= ~JS_UPDATED;
|
|
}
|
|
}
|
|
|
|
bool process_joystick_buttons(uint16_t keycode, keyrecord_t *record) {
|
|
if (keycode < JS_BUTTON0 || keycode > JS_BUTTON_MAX) {
|
|
return true;
|
|
} else {
|
|
if (record->event.pressed) {
|
|
joystick_status.buttons[(keycode - JS_BUTTON0) / 8] |= 1 << (keycode % 8);
|
|
} else {
|
|
joystick_status.buttons[(keycode - JS_BUTTON0) / 8] &= ~(1 << (keycode % 8));
|
|
}
|
|
|
|
joystick_status.status |= JS_UPDATED;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
uint16_t savePinState(uint32_t pin) {
|
|
#ifdef __AVR__
|
|
uint8_t pinNumber = pin & 0xF;
|
|
return ((PORTx_ADDRESS(pin) >> pinNumber) & 0x1) << 1 | ((DDRx_ADDRESS(pin) >> pinNumber) & 0x1);
|
|
#elif defined(PROTOCOL_CHIBIOS)
|
|
/*
|
|
The pin configuration is backed up in the following format :
|
|
bit 15 9 8 7 6 5 4 3 2 1 0
|
|
|unused|ODR|IDR|PUPDR|OSPEEDR|OTYPER|MODER|
|
|
*/
|
|
return (( PAL_PORT(pin)->MODER >> (2*PAL_PAD(pin))) & 0x3)
|
|
| (((PAL_PORT(pin)->OTYPER >> (1*PAL_PAD(pin))) & 0x1) << 2)
|
|
| (((PAL_PORT(pin)->OSPEEDR >> (2*PAL_PAD(pin))) & 0x3) << 3)
|
|
| (((PAL_PORT(pin)->PUPDR >> (2*PAL_PAD(pin))) & 0x3) << 5)
|
|
| (((PAL_PORT(pin)->IDR >> (1*PAL_PAD(pin))) & 0x1) << 7)
|
|
| (((PAL_PORT(pin)->ODR >> (1*PAL_PAD(pin))) & 0x1) << 8);
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
void restorePinState(uint32_t pin, uint16_t restoreState) {
|
|
#ifdef __AVR__
|
|
uint8_t pinNumber = pin & 0xF;
|
|
PORTx_ADDRESS(pin) = (PORTx_ADDRESS(pin) & ~_BV(pinNumber)) | (((restoreState >> 1) & 0x1) << pinNumber);
|
|
DDRx_ADDRESS(pin) = (DDRx_ADDRESS(pin) & ~_BV(pinNumber)) | ((restoreState & 0x1) << pinNumber);
|
|
#elif defined(PROTOCOL_CHIBIOS)
|
|
PAL_PORT(pin)->MODER = (PAL_PORT(pin)->MODER & ~(0x3<< (2*PAL_PAD(pin)))) | (restoreState & 0x3) << (2*PAL_PAD(pin));
|
|
PAL_PORT(pin)->OTYPER = (PAL_PORT(pin)->OTYPER & ~(0x1<< (1*PAL_PAD(pin)))) | ((restoreState>>2) & 0x1) << (1*PAL_PAD(pin));
|
|
PAL_PORT(pin)->OSPEEDR= (PAL_PORT(pin)->OSPEEDR & ~(0x3<< (2*PAL_PAD(pin)))) | ((restoreState>>3) & 0x3) << (2*PAL_PAD(pin));
|
|
PAL_PORT(pin)->PUPDR = (PAL_PORT(pin)->PUPDR & ~(0x3<< (2*PAL_PAD(pin)))) | ((restoreState>>5) & 0x3) << (2*PAL_PAD(pin));
|
|
PAL_PORT(pin)->IDR = (PAL_PORT(pin)->IDR & ~(0x1<< (1*PAL_PAD(pin)))) | ((restoreState>>7) & 0x1) << (1*PAL_PAD(pin));
|
|
PAL_PORT(pin)->ODR = (PAL_PORT(pin)->ODR & ~(0x1<< (1*PAL_PAD(pin)))) | ((restoreState>>8) & 0x1) << (1*PAL_PAD(pin));
|
|
#else
|
|
return;
|
|
#endif
|
|
}
|
|
|
|
__attribute__((weak)) bool process_joystick_analogread() { return process_joystick_analogread_quantum(); }
|
|
|
|
bool process_joystick_analogread_quantum() {
|
|
#if JOYSTICK_AXES_COUNT > 0
|
|
for (int axis_index = 0; axis_index < JOYSTICK_AXES_COUNT; ++axis_index) {
|
|
if (joystick_axes[axis_index].input_pin == JS_VIRTUAL_AXIS) {
|
|
continue;
|
|
}
|
|
|
|
// save previous input pin status as well
|
|
uint16_t inputSavedState = savePinState(joystick_axes[axis_index].input_pin);
|
|
|
|
// disable pull-up resistor
|
|
writePinLow(joystick_axes[axis_index].input_pin);
|
|
|
|
// if pin was a pull-up input, we need to uncharge it by turning it low
|
|
// before making it a low input
|
|
setPinOutput(joystick_axes[axis_index].input_pin);
|
|
|
|
wait_us(10);
|
|
|
|
// save and apply output pin status
|
|
uint16_t outputSavedState = 0;
|
|
if (joystick_axes[axis_index].output_pin != JS_VIRTUAL_AXIS) {
|
|
// save previous output pin status
|
|
outputSavedState = savePinState(joystick_axes[axis_index].output_pin);
|
|
|
|
setPinOutput(joystick_axes[axis_index].output_pin);
|
|
writePinHigh(joystick_axes[axis_index].output_pin);
|
|
}
|
|
|
|
uint16_t groundSavedState = 0;
|
|
if (joystick_axes[axis_index].ground_pin != JS_VIRTUAL_AXIS) {
|
|
// save previous output pin status
|
|
groundSavedState = savePinState(joystick_axes[axis_index].ground_pin);
|
|
|
|
setPinOutput(joystick_axes[axis_index].ground_pin);
|
|
writePinLow(joystick_axes[axis_index].ground_pin);
|
|
}
|
|
|
|
wait_us(10);
|
|
|
|
setPinInput(joystick_axes[axis_index].input_pin);
|
|
|
|
wait_us(10);
|
|
|
|
# if defined(__AVR__) || defined(PROTOCOL_CHIBIOS)
|
|
int16_t axis_val = analogReadPin(joystick_axes[axis_index].input_pin);
|
|
# else
|
|
// default to resting position
|
|
int16_t axis_val = joystick_axes[axis_index].mid_digit;
|
|
# endif
|
|
|
|
//test the converted value against the lower range
|
|
int32_t ref = joystick_axes[axis_index].mid_digit;
|
|
int32_t range = joystick_axes[axis_index].min_digit;
|
|
int32_t ranged_val = ((axis_val - ref)* -127)/(range - ref) ;
|
|
|
|
if (ranged_val > 0){
|
|
//the value is in the higher range
|
|
range = joystick_axes[axis_index].max_digit;
|
|
ranged_val = ((axis_val - ref)* 127)/(range - ref);
|
|
}
|
|
|
|
//clamp the result in the valid range
|
|
ranged_val = ranged_val<-127 ? -127 : ranged_val;
|
|
ranged_val = ranged_val>127 ? 127 : ranged_val;
|
|
|
|
if (ranged_val != joystick_status.axes[axis_index]) {
|
|
joystick_status.axes[axis_index] = ranged_val;
|
|
joystick_status.status |= JS_UPDATED;
|
|
}
|
|
|
|
// restore output, ground and input status
|
|
if (joystick_axes[axis_index].output_pin != JS_VIRTUAL_AXIS) {
|
|
restorePinState(joystick_axes[axis_index].output_pin, outputSavedState);
|
|
}
|
|
if (joystick_axes[axis_index].ground_pin != JS_VIRTUAL_AXIS) {
|
|
restorePinState(joystick_axes[axis_index].ground_pin, groundSavedState);
|
|
}
|
|
|
|
restorePinState(joystick_axes[axis_index].input_pin, inputSavedState);
|
|
}
|
|
|
|
#endif
|
|
return true;
|
|
}
|