- /* Copyright 2020 Christopher Courtney <drashna@live.com> (@drashna)
- *
- * 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 "tractyl_manuform.h"
- #include "transactions.h"
- #include <string.h>
- #include "drivers/sensors/pmw3360.h"
-
- #ifndef TRACKBALL_DPI_OPTIONS
- # define TRACKBALL_DPI_OPTIONS \
- { 1200, 1600, 2400 }
- # ifndef TRACKBALL_DPI_DEFAULT
- # define TRACKBALL_DPI_DEFAULT 1
- # endif
- #endif
- #ifndef TRACKBALL_DPI_DEFAULT
- # define TRACKBALL_DPI_DEFAULT 0
- #endif
-
- keyboard_config_t keyboard_config;
- uint16_t dpi_array[] = TRACKBALL_DPI_OPTIONS;
- #define DPI_OPTION_SIZE (sizeof(dpi_array) / sizeof(uint16_t))
-
- bool BurstState = false; // init burst state for Trackball module
- uint16_t MotionStart = 0; // Timer for accel, 0 is resting state
-
- __attribute__((weak)) void process_mouse_user(report_mouse_t* mouse_report, int8_t x, int8_t y) {
- mouse_report->x = x;
- mouse_report->y = y;
- }
-
- __attribute__((weak)) kb_pointer_data_t process_mouse(void) {
- kb_pointer_data_t temp_data = {.mouse_x = 0, .mouse_y = 0};
-
- report_pmw_t data = pmw_read_burst();
- if (data.isOnSurface && data.isMotion) {
- // Reset timer if stopped moving
- if (!data.isMotion) {
- if (MotionStart != 0) MotionStart = 0;
- return temp_data;
- }
-
- // Set timer if new motion
- if ((MotionStart == 0) && data.isMotion) {
- if (debug_mouse) dprintf("Starting motion.\n");
- MotionStart = timer_read();
- }
-
- if (debug_mouse) { dprintf("Delt] d: %d t: %u\n", abs(data.dx) + abs(data.dy), MotionStart); }
- if (debug_mouse) { dprintf("Pre ] X: %d, Y: %d\n", data.dx, data.dy); }
- #if defined(PROFILE_LINEAR)
- float scale = float(timer_elaspsed(MotionStart)) / 1000.0;
- data.dx *= scale;
- data.dy *= scale;
- #elif defined(PROFILE_INVERSE)
- // TODO
- #else
- // no post processing
- #endif
-
- // Wrap to HID size
- data.dx = constrain(data.dx, -127, 127);
- data.dy = constrain(data.dy, -127, 127);
- if (debug_mouse) dprintf("Cons] X: %d, Y: %d\n", data.dx, data.dy);
- // dprintf("Elapsed:%u, X: %f Y: %\n", i, pgm_read_byte(firmware_data+i));
-
- temp_data.mouse_x = -data.dx;
- temp_data.mouse_y = data.dy;
- }
- return temp_data;
- }
-
- bool process_record_kb(uint16_t keycode, keyrecord_t* record) {
- if (!process_record_user(keycode, record)) { return false; }
-
- #ifdef POINTING_DEVICE_ENABLE
- if (keycode == DPI_CONFIG && record->event.pressed) {
- if ((get_mods() | get_oneshot_mods()) & MOD_MASK_SHIFT) {
- keyboard_config.dpi_config = (keyboard_config.dpi_config - 1) % DPI_OPTION_SIZE;
- } else {
- keyboard_config.dpi_config = (keyboard_config.dpi_config + 1) % DPI_OPTION_SIZE;
- }
- eeconfig_update_kb(keyboard_config.raw);
- trackball_set_cpi(dpi_array[keyboard_config.dpi_config]);
- }
- #endif
-
- /* If Mousekeys is disabled, then use handle the mouse button
- * keycodes. This makes things simpler, and allows usage of
- * the keycodes in a consistent manner. But only do this if
- * Mousekeys is not enable, so it's not handled twice.
- */
- #ifndef MOUSEKEY_ENABLE
- if (IS_MOUSEKEY_BUTTON(keycode)) {
- report_mouse_t currentReport = pointing_device_get_report();
- if (record->event.pressed) {
- currentReport.buttons |= 1 << (keycode - KC_MS_BTN1);
- } else {
- currentReport.buttons &= ~(1 << (keycode - KC_MS_BTN1));
- }
- pointing_device_set_report(currentReport);
- pointing_device_send();
- }
- #endif
-
- return true;
- }
- __attribute__((weak)) void keyboard_pre_init_sync(void) {}
-
- // Hardware Setup
- void keyboard_pre_init_kb(void) {
- // debug_enable = true;
- // debug_matrix = true;
- // debug_mouse = true;
- // debug_encoder = true;
-
- /* Ground all output pins connected to ground. This provides additional
- * pathways to ground. If you're messing with this, know this: driving ANY
- * of these pins high will cause a short. On the MCU. Ka-blooey.
- */
-
- // This is the debug LED.
- #if defined(DEBUG_LED_PIN)
- setPinOutput(DEBUG_LED_PIN);
- writePin(DEBUG_LED_PIN, debug_enable);
- #endif
-
- keyboard_pre_init_sync();
- keyboard_pre_init_user();
- }
-
- #ifdef POINTING_DEVICE_ENABLE
- void pointing_device_init(void) {
- if (!is_keyboard_left()) {
- // initialize ball sensor
- pmw_spi_init();
- }
- trackball_set_cpi(dpi_array[keyboard_config.dpi_config]);
- }
-
- void pointing_device_task(void) {
- report_mouse_t mouse_report = pointing_device_get_report();
- kb_pointer_data_t data = {.mouse_x = mouse_report.x, .mouse_y = mouse_report.y};
- if (is_keyboard_left()) {
- if (is_keyboard_master()) {
- data = kb_pointer_sync_get();
- process_mouse_user(&mouse_report, data.mouse_x, data.mouse_y);
- }
- } else {
- data = process_mouse();
- if (!is_keyboard_master()) {
- kb_pointer_sync_send(data.mouse_x, data.mouse_y);
- } else {
- process_mouse_user(&mouse_report, data.mouse_x, data.mouse_y);
- }
- }
-
- pointing_device_set_report(mouse_report);
- pointing_device_send();
- }
- #endif
-
- void eeconfig_init_kb(void) {
- keyboard_config.dpi_config = TRACKBALL_DPI_DEFAULT;
- #ifdef POINTING_DEVICE_ENABLE
- trackball_set_cpi(dpi_array[keyboard_config.dpi_config]);
- #endif
- eeconfig_update_kb(keyboard_config.raw);
- eeconfig_init_user();
- }
-
- __attribute__((weak)) void matrix_init_sub_kb(void) {}
- void matrix_init_kb(void) {
- // is safe to just read DPI setting since matrix init
- // comes before pointing device init.
- keyboard_config.raw = eeconfig_read_kb();
- if (keyboard_config.dpi_config > DPI_OPTION_SIZE) { eeconfig_init_kb(); }
- matrix_init_sub_kb();
- matrix_init_user();
- }
-
- __attribute__((weak)) void matrix_scan_sub_kb(void) {}
- void matrix_scan_kb(void) {
- matrix_scan_sub_kb();
- matrix_scan_user();
- }
|