diff options
Diffstat (limited to 'users/pcewing/key_repeater.c')
| -rw-r--r-- | users/pcewing/key_repeater.c | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/users/pcewing/key_repeater.c b/users/pcewing/key_repeater.c new file mode 100644 index 0000000000..6c77d45b5f --- /dev/null +++ b/users/pcewing/key_repeater.c @@ -0,0 +1,168 @@ +/* + * Copyright 2022 Paul Ewing + * + * 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 "key_repeater.h" +#include <stdlib.h> +#include QMK_KEYBOARD_H + +enum key_repeater_state { + KR_DISABLED, + KR_WAITING, + KR_BUTTON_DOWN, + + __KR_STATE_COUNT +}; + +struct key_repeater_t { + int key; + uint32_t key_duration_min; + uint32_t key_duration_max; + uint32_t wait_duration_min; + uint32_t wait_duration_max; + + enum key_repeater_state state; + uint32_t previous_button_down; + uint32_t previous_button_up; + uint32_t key_duration; + uint32_t wait_duration; +}; + +// Utility functions +uint32_t get_rand(uint32_t min, uint32_t max); + +// State handler function table +void kr_waiting(struct key_repeater_t *kr); +void kr_button_down(struct key_repeater_t *kr); + +typedef void (*kr_state_handler)(struct key_repeater_t *kr); +static kr_state_handler kr_state_handlers[] = { + [KR_DISABLED] = NULL, + [KR_WAITING] = kr_waiting, + [KR_BUTTON_DOWN] = kr_button_down, +}; + +struct key_repeater_t* kr_new(struct key_repeater_config_t* cfg) { + struct key_repeater_t* kr = (struct key_repeater_t*)malloc(sizeof(struct key_repeater_t)); + + if (cfg) { + kr->key = cfg->key; + kr->key_duration_min = cfg->key_duration_min; + kr->key_duration_max = cfg->key_duration_max; + kr->wait_duration_min = cfg->wait_duration_min; + kr->wait_duration_max = cfg->wait_duration_max; + } else { + kr->key = KC_NO; + kr->key_duration_min = 0; + kr->key_duration_max = 0; + kr->wait_duration_min = 0; + kr->wait_duration_max = 0; + } + + kr->state = KR_DISABLED; + kr->previous_button_down = 0; + kr->previous_button_up = 0; + kr->key_duration = 0; + kr->wait_duration = 0; + + return kr; +} + +void kr_free(struct key_repeater_t **kr) { + if (kr && *kr) { + free(*kr); + *kr = NULL; + } +} + +void kr_enable(struct key_repeater_t *kr) { + if (!kr || kr->key == KC_NO) { + return; + } + + if (kr->state != KR_DISABLED) { + return; + } + kr->state = KR_WAITING; + kr->previous_button_down = 0; + kr->previous_button_up = 0; + kr->key_duration = 0; + kr->wait_duration = 0; +} + +void kr_disable(struct key_repeater_t *kr) { + if (!kr || kr->key == KC_NO) { + return; + } + + if (kr->state == KR_BUTTON_DOWN) { + unregister_code(kr->key); + } + kr->state = KR_DISABLED; +} + +void kr_poll(struct key_repeater_t *kr) { + if (!kr || kr->key == KC_NO) { + return; + } + + kr_state_handler handler = kr_state_handlers[kr->state]; + if (handler) { + (handler)(kr); + } +} + +void kr_waiting(struct key_repeater_t *kr) { + if (!kr || kr->key == KC_NO) { + return; + } + + uint32_t now = timer_read32(); + if (now > (kr->previous_button_up + kr->wait_duration)) { + kr->state = KR_BUTTON_DOWN; + kr->previous_button_down = now; + if (kr->key_duration_min == kr->key_duration_max) { + kr->key_duration = kr->key_duration_min; + } else { + kr->key_duration = get_rand(kr->key_duration_min, kr->key_duration_max); + } + register_code(kr->key); + } +} + +void kr_button_down(struct key_repeater_t *kr) { + if (!kr || kr->key == KC_NO) { + return; + } + + uint32_t now = timer_read32(); + if (now > (kr->previous_button_down + kr->key_duration)) { + kr->state = KR_WAITING; + kr->previous_button_up = now; + if (kr->wait_duration_min == kr->wait_duration_max) { + kr->wait_duration = kr->wait_duration_min; + } else { + kr->wait_duration = get_rand(kr->wait_duration_min, kr->wait_duration_max); + } + unregister_code(kr->key); + } +} + +// Return a random number between min and max; assumes that the random number +// generator has already been seeded +uint32_t get_rand(uint32_t min, uint32_t max) { + return (rand() % (max - min)) + min; +} |