123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307 |
- /*
- Copyright 2013 Jun Wako <[email protected]>
-
- 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 "host.h"
- #include "report.h"
- #include "debug.h"
- #include "action_util.h"
- #include "timer.h"
-
- static inline void add_key_byte(uint8_t code);
- static inline void del_key_byte(uint8_t code);
- #ifdef NKRO_ENABLE
- static inline void add_key_bit(uint8_t code);
- static inline void del_key_bit(uint8_t code);
- #endif
-
- static uint8_t real_mods = 0;
- static uint8_t weak_mods = 0;
-
- #ifdef USB_6KRO_ENABLE
- #define RO_ADD(a, b) ((a + b) % REPORT_KEYS)
- #define RO_SUB(a, b) ((a - b + REPORT_KEYS) % REPORT_KEYS)
- #define RO_INC(a) RO_ADD(a, 1)
- #define RO_DEC(a) RO_SUB(a, 1)
- static int8_t cb_head = 0;
- static int8_t cb_tail = 0;
- static int8_t cb_count = 0;
- #endif
-
- // TODO: pointer variable is not needed
- //report_keyboard_t keyboard_report = {};
- report_keyboard_t *keyboard_report = &(report_keyboard_t){};
-
- #ifndef NO_ACTION_ONESHOT
- static int8_t oneshot_mods = 0;
- #if (defined(ONESHOT_TIMEOUT) && (ONESHOT_TIMEOUT > 0))
- static int16_t oneshot_time = 0;
- #endif
- #endif
-
-
- void send_keyboard_report(void) {
- keyboard_report->mods = real_mods;
- keyboard_report->mods |= weak_mods;
- #ifndef NO_ACTION_ONESHOT
- if (oneshot_mods) {
- #if (defined(ONESHOT_TIMEOUT) && (ONESHOT_TIMEOUT > 0))
- if (TIMER_DIFF_16(timer_read(), oneshot_time) >= ONESHOT_TIMEOUT) {
- dprintf("Oneshot: timeout\n");
- clear_oneshot_mods();
- }
- #endif
- keyboard_report->mods |= oneshot_mods;
- if (has_anykey()) {
- clear_oneshot_mods();
- }
- }
- #endif
- host_keyboard_send(keyboard_report);
- }
-
- /* key */
- void add_key(uint8_t key)
- {
- #ifdef NKRO_ENABLE
- if (keyboard_nkro) {
- add_key_bit(key);
- return;
- }
- #endif
- add_key_byte(key);
- }
-
- void del_key(uint8_t key)
- {
- #ifdef NKRO_ENABLE
- if (keyboard_nkro) {
- del_key_bit(key);
- return;
- }
- #endif
- del_key_byte(key);
- }
-
- void clear_keys(void)
- {
- // not clear mods
- for (int8_t i = 1; i < REPORT_SIZE; i++) {
- keyboard_report->raw[i] = 0;
- }
- }
-
-
- /* modifier */
- uint8_t get_mods(void) { return real_mods; }
- void add_mods(uint8_t mods) { real_mods |= mods; }
- void del_mods(uint8_t mods) { real_mods &= ~mods; }
- void set_mods(uint8_t mods) { real_mods = mods; }
- void clear_mods(void) { real_mods = 0; }
-
- /* weak modifier */
- uint8_t get_weak_mods(void) { return weak_mods; }
- void add_weak_mods(uint8_t mods) { weak_mods |= mods; }
- void del_weak_mods(uint8_t mods) { weak_mods &= ~mods; }
- void set_weak_mods(uint8_t mods) { weak_mods = mods; }
- void clear_weak_mods(void) { weak_mods = 0; }
-
- /* Oneshot modifier */
- #ifndef NO_ACTION_ONESHOT
- void set_oneshot_mods(uint8_t mods)
- {
- oneshot_mods = mods;
- #if (defined(ONESHOT_TIMEOUT) && (ONESHOT_TIMEOUT > 0))
- oneshot_time = timer_read();
- #endif
- }
- void clear_oneshot_mods(void)
- {
- oneshot_mods = 0;
- #if (defined(ONESHOT_TIMEOUT) && (ONESHOT_TIMEOUT > 0))
- oneshot_time = 0;
- #endif
- }
- #endif
-
-
-
-
- /*
- * inspect keyboard state
- */
- uint8_t has_anykey(void)
- {
- uint8_t cnt = 0;
- for (uint8_t i = 1; i < REPORT_SIZE; i++) {
- if (keyboard_report->raw[i])
- cnt++;
- }
- return cnt;
- }
-
- uint8_t has_anymod(void)
- {
- return bitpop(real_mods);
- }
-
- uint8_t get_first_key(void)
- {
- #ifdef NKRO_ENABLE
- if (keyboard_nkro) {
- uint8_t i = 0;
- for (; i < REPORT_BITS && !keyboard_report->nkro.bits[i]; i++)
- ;
- return i<<3 | biton(keyboard_report->nkro.bits[i]);
- }
- #endif
- #ifdef USB_6KRO_ENABLE
- uint8_t i = cb_head;
- do {
- if (keyboard_report->keys[i] != 0) {
- break;
- }
- i = RO_INC(i);
- } while (i != cb_tail);
- return keyboard_report->keys[i];
- #else
- return keyboard_report->keys[0];
- #endif
- }
-
-
-
- /* local functions */
- static inline void add_key_byte(uint8_t code)
- {
- #ifdef USB_6KRO_ENABLE
- int8_t i = cb_head;
- int8_t empty = -1;
- if (cb_count) {
- do {
- if (keyboard_report->keys[i] == code) {
- return;
- }
- if (empty == -1 && keyboard_report->keys[i] == 0) {
- empty = i;
- }
- i = RO_INC(i);
- } while (i != cb_tail);
- if (i == cb_tail) {
- if (cb_tail == cb_head) {
- // buffer is full
- if (empty == -1) {
- // pop head when has no empty space
- cb_head = RO_INC(cb_head);
- cb_count--;
- }
- else {
- // left shift when has empty space
- uint8_t offset = 1;
- i = RO_INC(empty);
- do {
- if (keyboard_report->keys[i] != 0) {
- keyboard_report->keys[empty] = keyboard_report->keys[i];
- keyboard_report->keys[i] = 0;
- empty = RO_INC(empty);
- }
- else {
- offset++;
- }
- i = RO_INC(i);
- } while (i != cb_tail);
- cb_tail = RO_SUB(cb_tail, offset);
- }
- }
- }
- }
- // add to tail
- keyboard_report->keys[cb_tail] = code;
- cb_tail = RO_INC(cb_tail);
- cb_count++;
- #else
- int8_t i = 0;
- int8_t empty = -1;
- for (; i < REPORT_KEYS; i++) {
- if (keyboard_report->keys[i] == code) {
- break;
- }
- if (empty == -1 && keyboard_report->keys[i] == 0) {
- empty = i;
- }
- }
- if (i == REPORT_KEYS) {
- if (empty != -1) {
- keyboard_report->keys[empty] = code;
- }
- }
- #endif
- }
-
- static inline void del_key_byte(uint8_t code)
- {
- #ifdef USB_6KRO_ENABLE
- uint8_t i = cb_head;
- if (cb_count) {
- do {
- if (keyboard_report->keys[i] == code) {
- keyboard_report->keys[i] = 0;
- cb_count--;
- if (cb_count == 0) {
- // reset head and tail
- cb_tail = cb_head = 0;
- }
- if (i == RO_DEC(cb_tail)) {
- // left shift when next to tail
- do {
- cb_tail = RO_DEC(cb_tail);
- if (keyboard_report->keys[RO_DEC(cb_tail)] != 0) {
- break;
- }
- } while (cb_tail != cb_head);
- }
- break;
- }
- i = RO_INC(i);
- } while (i != cb_tail);
- }
- #else
- for (uint8_t i = 0; i < REPORT_KEYS; i++) {
- if (keyboard_report->keys[i] == code) {
- keyboard_report->keys[i] = 0;
- }
- }
- #endif
- }
-
- #ifdef NKRO_ENABLE
- static inline void add_key_bit(uint8_t code)
- {
- if ((code>>3) < REPORT_BITS) {
- keyboard_report->nkro.bits[code>>3] |= 1<<(code&7);
- } else {
- dprintf("add_key_bit: can't add: %02X\n", code);
- }
- }
-
- static inline void del_key_bit(uint8_t code)
- {
- if ((code>>3) < REPORT_BITS) {
- keyboard_report->nkro.bits[code>>3] &= ~(1<<(code&7));
- } else {
- dprintf("del_key_bit: can't del: %02X\n", code);
- }
- }
- #endif
|