361 lines
17 KiB
C

#include QMK_KEYBOARD_H
#include "drivers/sensors/pimoroni_trackball.h"
#include "secret.h"
#ifndef SECRET_1
#define SECRET_1 secret
#endif
#ifndef SECRET_2
#define SECRET_2 secret
#endif
#ifndef EMAIL_1
#define EMAIL_1 email
#endif
#ifndef EMAIL_2
#define EMAIL_2 email
#endif
enum custom_keycodes {
DRAG_SCROLL = QK_USER_0,
BALL_HUE,
BALL_SAT,
BALL_VAL,
BALL_SPEED,
SAVE_CONF,
};
#ifdef LAYOUT_split_3x6_3_ex2
const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
[0] = LAYOUT_split_3x6_3_ex2(
//,--------------------------------------------------------------. ,--------------------------------------------------------------.
KC_TAB, KC_Q, KC_W, KC_E, KC_R, KC_T, KC_LCTL, KC_RCTL, KC_Y, KC_U, KC_I, KC_O, KC_P, KC_BSPC,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LCTL, KC_A, KC_S, KC_D, KC_F, KC_G, KC_LALT, KC_RALT, KC_H, KC_J, KC_K, KC_L, KC_SCLN, KC_QUOT,
//|--------+--------+--------+--------+--------+--------+--------' `--------+--------+--------+--------+--------+--------+--------|
KC_LSFT, KC_Z, KC_X, KC_C, KC_V, KC_B, KC_N, KC_M, KC_COMM, KC_DOT, KC_SLSH, KC_ESC,
//|--------+--------+--------+--------+--------+--------+--------. ,--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, TL_LOWR, KC_SPC, KC_ENT, TL_UPPR, KC_RGUI
//`--------------------------' `--------------------------'
),
[1] = LAYOUT_split_3x6_3_ex2(
//,--------------------------------------------------------------. ,--------------------------------------------------------------.
KC_TAB, KC_1, KC_2, KC_3, KC_4, KC_5, KC_LCTL, KC_RCTL, KC_6, KC_7, KC_8, KC_9, KC_0, KC_BSPC,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LCTL, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_LALT, KC_RALT, KC_LEFT, KC_DOWN, KC_UP,KC_RIGHT, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------' `--------+--------+--------+--------+--------+--------+--------|
KC_LSFT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------. ,--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RGUI
//`--------------------------' `--------------------------'
),
[2] = LAYOUT_split_3x6_3_ex2(
//,--------------------------------------------------------------. ,--------------------------------------------------------------.
KC_TAB, KC_EXLM, KC_AT, KC_HASH, KC_DLR, KC_PERC, KC_LCTL, KC_RCTL, KC_CIRC, KC_AMPR, KC_ASTR, KC_LPRN, KC_RPRN, KC_BSPC,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LCTL, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_LALT, KC_RALT, KC_MINS, KC_EQL, KC_LBRC, KC_RBRC, KC_BSLS, KC_GRV,
//|--------+--------+--------+--------+--------+--------+--------' `--------+--------+--------+--------+--------+--------+--------|
KC_LSFT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_UNDS, KC_PLUS, KC_LCBR, KC_RCBR, KC_PIPE, KC_TILD,
//|--------+--------+--------+--------+--------+--------+--------. ,--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RGUI
//`--------------------------' `--------------------------'
),
[3] = LAYOUT_split_3x6_3_ex2(
//,--------------------------------------------------------------. ,--------------------------------------------------------------.
QK_BOOT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
RGB_TOG, RGB_HUI, RGB_SAI, RGB_VAI, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------' `--------+--------+--------+--------+--------+--------+--------|
RGB_MOD, RGB_HUD, RGB_SAD, RGB_VAD, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------. ,--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RGUI
//`--------------------------' `--------------------------'
)
};
#else
const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
[0] = LAYOUT_split_3x6_3(
//,-----------------------------------------------------. ,-----------------------------------------------------.
KC_TAB, KC_Q, KC_W, KC_E, KC_R, KC_T, KC_Y, KC_U, KC_I, KC_O, KC_P, KC_BSPC,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LCTL, KC_A, KC_S, KC_D, KC_F, KC_G, KC_H, KC_J, KC_K, KC_L, KC_SCLN, KC_QUOT,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LSFT, KC_Z, KC_X, KC_C, KC_V, KC_B, KC_N, KC_M, KC_COMM, KC_DOT, KC_SLSH, KC_ESC,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, TL_LOWR, KC_SPC, KC_ENT, TL_UPPR, KC_RALT
//`--------------------------' `--------------------------'
),
[1] = LAYOUT_split_3x6_3(
//,-----------------------------------------------------. ,-----------------------------------------------------.
KC_TAB, KC_1, KC_2, KC_3, KC_4, KC_5, KC_6, KC_7, KC_8, KC_9, KC_0, KC_BSPC,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LCTL, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_LEFT, KC_DOWN, KC_UP,KC_RIGHT, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LSFT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RALT
//`--------------------------' `--------------------------'
),
[2] = LAYOUT_split_3x6_3(
//,-----------------------------------------------------. ,-----------------------------------------------------.
KC_TAB, KC_EXLM, KC_AT, KC_HASH, KC_DLR, KC_PERC, KC_CIRC, KC_AMPR, KC_ASTR, KC_LPRN, KC_RPRN, KC_BSPC,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LCTL, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_MINS, KC_EQL, KC_LBRC, KC_RBRC, KC_BSLS, KC_GRV,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
KC_LSFT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, KC_UNDS, KC_PLUS, KC_LCBR, KC_RCBR, KC_PIPE, KC_TILD,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RALT
//`--------------------------' `--------------------------'
),
[3] = LAYOUT_split_3x6_3(
//,-----------------------------------------------------. ,-----------------------------------------------------.
QK_BOOT, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
RGB_TOG, RGB_HUI, RGB_SAI, RGB_VAI, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------|
RGB_MOD, RGB_HUD, RGB_SAD, RGB_VAD, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX, XXXXXXX,
//|--------+--------+--------+--------+--------+--------+--------| |--------+--------+--------+--------+--------+--------+--------|
KC_LGUI, _______, KC_SPC, KC_ENT, _______, KC_RALT
//`--------------------------' `--------------------------'
)
};
#endif
bool set_scrolling = false;
// Modify these values to adjust the scrolling speed
#define SCROLL_DIVISOR_H 8.0
#define SCROLL_DIVISOR_V 8.0
// Variables to store accumulated scroll values
float scroll_accumulated_h = 0;
float scroll_accumulated_v = 0;
void pointing_device_init_user(void) {
set_auto_mouse_layer(5);
set_auto_mouse_enable(true);
}
/* Trackball related stuff */
static bool hue_mode_enabled = 0;
static bool sat_mode_enabled = 0;
static bool val_mode_enabled = 0;
static bool speed_mode_enabled = 0;
typedef union {
uint32_t raw;
struct {
uint8_t tb_hue :8;
uint8_t tb_value :8;
uint8_t tb_saturation :8;
};
} user_config_t;
user_config_t user_config;
static uint8_t tb_hue = 1;
static uint8_t tb_saturation = 255;
static uint8_t tb_value = 255;
#ifndef MIN
# define MIN(a, b) ((a) < (b) ? (a) : (b))
#endif
void trackball_set_hsv(uint8_t hue, uint8_t sat, uint8_t brightness) {
RGB rgb = hsv_to_rgb((HSV){hue, sat, brightness});
uint8_t white = MIN(rgb.r, MIN(rgb.g, rgb.b));
rgb.r -= white;
rgb.g -= white;
rgb.b -= white;
pimoroni_trackball_set_rgbw(rgb.r, rgb.g, rgb.b, white);
}
void keyboard_post_init_user(void) {
user_config.raw = eeconfig_read_user();
tb_hue = user_config.tb_hue;
tb_saturation = user_config.tb_saturation;
tb_value = user_config.tb_value;
trackball_set_hsv(tb_hue, tb_saturation,tb_value);
}
void eeconfig_init_user(void) {
user_config.raw = 0;
user_config.tb_hue = 80;
user_config.tb_saturation = 80;
user_config.tb_value = 80;
eeconfig_update_user(user_config.raw);
}
void suspend_power_down_user(void) {
trackball_set_hsv(0, 0, 0);
}
void suspend_wakeup_init_user(void) {
trackball_set_hsv(tb_hue, tb_saturation,tb_value);
}
bool is_mouse_record_user(uint16_t keycode, keyrecord_t* record) {
switch(keycode) {
case DRAG_SCROLL:
case BALL_HUE:
case BALL_SAT:
case BALL_VAL:
case BALL_SPEED:
return true;
default:
return false;
}
}
bool process_record_user(uint16_t keycode, keyrecord_t *record) {
switch (keycode) {
case DRAG_SCROLL:
// Toggle set_scrolling when DRAG_SCROLL key is pressed or released
set_scrolling = record->event.pressed;
break;
case BALL_HUE:
hue_mode_enabled = record->event.pressed;
break;
case BALL_SAT:
sat_mode_enabled = record->event.pressed;
break;
case BALL_VAL:
val_mode_enabled = record->event.pressed;
break;
case BALL_SPEED:
speed_mode_enabled = record->event.pressed;
break;
case SAVE_CONF:
user_config.tb_value = tb_value;
user_config.tb_hue = tb_hue;
user_config.tb_saturation = tb_saturation;
eeconfig_update_user(user_config.raw);
break;
case C(KC_PAGE_UP): // for changing tabs
case C(KC_PAGE_DOWN): // for changing tabs
case A(KC_LEFT): // for back nav
case A(KC_RIGHT): // for forward nav
auto_mouse_keyevent(record->event.pressed);
break;
default:
break;
}
return true;
}
report_mouse_t pointing_device_task_user(report_mouse_t mouse_report) {
// Check if drag scrolling is active
if (set_scrolling) {
// Calculate and accumulate scroll values based on mouse movement and divisors
scroll_accumulated_h += (float)mouse_report.x / SCROLL_DIVISOR_H;
scroll_accumulated_v += (float)mouse_report.y / SCROLL_DIVISOR_V;
// Assign integer parts of accumulated scroll values to the mouse report
mouse_report.h = (int8_t)scroll_accumulated_h;
mouse_report.v = (int8_t)scroll_accumulated_v;
// Update accumulated scroll values by subtracting the integer parts
scroll_accumulated_h -= (int8_t)scroll_accumulated_h;
scroll_accumulated_v -= (int8_t)scroll_accumulated_v;
// Clear the X and Y values of the mouse report
mouse_report.x = 0;
mouse_report.y = 0;
} else if (speed_mode_enabled) {
mouse_report.x = mouse_report.x * 2;
mouse_report.y = mouse_report.y * 2;
} else if (hue_mode_enabled) {
// adjust ball hue
tb_hue += mouse_report.y;
trackball_set_hsv(tb_hue | 1, tb_saturation, tb_value);
// Clear the X and Y values of the mouse report
mouse_report.x = 0;
mouse_report.y = 0;
} else if (sat_mode_enabled) {
// adjust ball saturation
tb_saturation += mouse_report.y * 2;
trackball_set_hsv(tb_hue, tb_saturation | 1, tb_value);
// Clear the X and Y values of the mouse report
mouse_report.x = 0;
mouse_report.y = 0;
} else if (val_mode_enabled) {
// adjust ball value
tb_value += mouse_report.y * 2;
trackball_set_hsv(tb_hue, tb_saturation, tb_value | 1);
// Clear the X and Y values of the mouse report
mouse_report.x = 0;
mouse_report.y = 0;
}
return mouse_report;
}
// Function to handle layer changes and disable drag scrolling when not in AUTO_MOUSE_DEFAULT_LAYER
layer_state_t layer_state_set_user(layer_state_t state) {
uint8_t layer = get_highest_layer(state);
if (layer != AUTO_MOUSE_DEFAULT_LAYER) {
set_scrolling = false;
hue_mode_enabled = false;
sat_mode_enabled = false;
val_mode_enabled = false;
speed_mode_enabled = false;
}
switch (layer) {
case 1:
trackball_set_hsv(tb_hue + 40, tb_saturation, tb_value > 60 ? tb_value - 60 : 0);
break;
case 2:
// numbers + nav
trackball_set_hsv(tb_hue + 80, tb_saturation, tb_value > 60 ? tb_value - 60 : 0);
break;
case 3:
// symbols
trackball_set_hsv(tb_hue + 80, tb_saturation, tb_value > 60 ? tb_value - 60 : 0);
break;
case 4:
// media
trackball_set_hsv(tb_hue + 80, tb_saturation, tb_value > 60 ? tb_value - 60 : 0);
break;
case 5:
// mouse layer
trackball_set_hsv(tb_hue, tb_saturation, tb_value);
break;
default:
trackball_set_hsv(tb_hue, tb_saturation, tb_value > 60 ? tb_value - 60 : 0);
break;
}
return state;
}
void leader_end_user(void) {
if (leader_sequence_two_keys(KC_D, KC_A)) {
SEND_STRING(SECRET_1);
} else if (leader_sequence_three_keys(KC_T, KC_A, KC_O)) {
SEND_STRING(SECRET_2);
} else if (leader_sequence_two_keys(KC_E, KC_O)) {
SEND_STRING(EMAIL_1);
} else if (leader_sequence_two_keys(KC_E, KC_S)) {
SEND_STRING(EMAIL_2);
}
}
#ifdef ENCODER_MAP_ENABLE
const uint16_t PROGMEM encoder_map[][NUM_ENCODERS][NUM_DIRECTIONS] = {
[0] = { ENCODER_CCW_CW(RGB_MOD, RGB_RMOD), ENCODER_CCW_CW(RGB_HUI, RGB_HUD), ENCODER_CCW_CW(RGB_VAI, RGB_VAD), ENCODER_CCW_CW(RGB_SAI, RGB_SAD), },
[1] = { ENCODER_CCW_CW(RGB_MOD, RGB_RMOD), ENCODER_CCW_CW(RGB_HUI, RGB_HUD), ENCODER_CCW_CW(RGB_VAI, RGB_VAD), ENCODER_CCW_CW(RGB_SAI, RGB_SAD), },
[2] = { ENCODER_CCW_CW(RGB_MOD, RGB_RMOD), ENCODER_CCW_CW(RGB_HUI, RGB_HUD), ENCODER_CCW_CW(RGB_VAI, RGB_VAD), ENCODER_CCW_CW(RGB_SAI, RGB_SAD), },
[3] = { ENCODER_CCW_CW(RGB_MOD, RGB_RMOD), ENCODER_CCW_CW(RGB_HUI, RGB_HUD), ENCODER_CCW_CW(RGB_VAI, RGB_VAD), ENCODER_CCW_CW(RGB_SAI, RGB_SAD), },
};
#endif