diff options
Diffstat (limited to 'quantum/process_keycode/process_joystick.c')
| -rw-r--r-- | quantum/process_keycode/process_joystick.c | 168 | 
1 files changed, 168 insertions, 0 deletions
| diff --git a/quantum/process_keycode/process_joystick.c b/quantum/process_keycode/process_joystick.c new file mode 100644 index 0000000000..c12f756854 --- /dev/null +++ b/quantum/process_keycode/process_joystick.c @@ -0,0 +1,168 @@ +#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(pin_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(pin_t pin, uint16_t restoreState) { +#if defined(PROTOCOL_LUFA) +    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; +} | 
