common/host.c

   1 /*
   2 Copyright 2011,2012 Jun Wako <wakojun@gmail.com>
   3
   4 This program is free software: you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation, either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <stdint.h>
  19 #include <avr/interrupt.h>
  20 #include "keycode.h"
  21 #include "host.h"
  22 #include "util.h"
  23 #include "debug.h"
  24
  25
  26 #ifdef NKRO_ENABLE
  27 bool keyboard_nkro = false;
  28 #endif
  29
  30 report_keyboard_t *keyboard_report = &(report_keyboard_t){};
  31 report_mouse_t mouse_report = {};
  32
  33
  34 static host_driver_t *driver;
  35 static uint16_t last_system_report = 0;
  36 static uint16_t last_consumer_report = 0;
  37
  38 static inline void add_key_byte(uint8_t code);
  39 static inline void del_key_byte(uint8_t code);
  40 #ifdef NKRO_ENABLE
  41 static inline void add_key_bit(uint8_t code);
  42 static inline void del_key_bit(uint8_t code);
  43 #endif
  44
  45
  46 void host_set_driver(host_driver_t *d)
  47 {
  48     driver = d;
  49 }
  50
  51 host_driver_t *host_get_driver(void)
  52 {
  53     return driver;
  54 }
  55
  56 uint8_t host_keyboard_leds(void)
  57 {
  58     if (!driver) return 0;
  59     return (*driver->keyboard_leds)();
  60 }
  61 /* send report */
  62 void host_keyboard_send(report_keyboard_t *report)
  63 {
  64     if (!driver) return;
  65     (*driver->send_keyboard)(report);
  66
  67     if (debug_keyboard) {
  68         dprint("keyboard_report: ");
  69         for (uint8_t i = 0; i < REPORT_SIZE; i++) {
  70             dprintf("%02X ", keyboard_report->raw[i]);
  71         }
  72         dprint("\n");
  73     }
  74 }
  75
  76 void host_mouse_send(report_mouse_t *report)
  77 {
  78     if (!driver) return;
  79     (*driver->send_mouse)(report);
  80 }
  81
  82 void host_system_send(uint16_t report)
  83 {
  84     if (report == last_system_report) return;
  85     last_system_report = report;
  86
  87     if (!driver) return;
  88     (*driver->send_system)(report);
  89 }
  90
  91 void host_consumer_send(uint16_t report)
  92 {
  93     if (report == last_consumer_report) return;
  94     last_consumer_report = report;
  95
  96     if (!driver) return;
  97     (*driver->send_consumer)(report);
  98 }
  99
 100
 101
 102 /* keyboard report utils */
 103 void host_add_key(uint8_t key)
 104 {
 105 #ifdef NKRO_ENABLE
 106     if (keyboard_nkro) {
 107         add_key_bit(key);
 108         return;
 109     }
 110 #endif
 111     add_key_byte(key);
 112 }
 113
 114 void host_del_key(uint8_t key)
 115 {
 116 #ifdef NKRO_ENABLE
 117     if (keyboard_nkro) {
 118         del_key_bit(key);
 119         return;
 120     }
 121 #endif
 122     del_key_byte(key);
 123 }
 124
 125 void host_clear_keys(void)
 126 {
 127     // not clea  mods
 128     for (int8_t i = 1; i < REPORT_SIZE; i++) {
 129         keyboard_report->raw[i] = 0;
 130     }
 131 }
 132
 133 uint8_t host_get_mods(void)
 134 {
 135     return keyboard_report->mods;
 136 }
 137
 138 void host_add_mods(uint8_t mods)
 139 {
 140     keyboard_report->mods |= mods;
 141 }
 142
 143 void host_del_mods(uint8_t mods)
 144 {
 145     keyboard_report->mods &= ~mods;
 146 }
 147
 148 void host_set_mods(uint8_t mods)
 149 {
 150     keyboard_report->mods = mods;
 151 }
 152
 153 void host_clear_mods(void)
 154 {
 155     keyboard_report->mods = 0;
 156 }
 157
 158 uint8_t host_has_anykey(void)
 159 {
 160     uint8_t cnt = 0;
 161     for (uint8_t i = 1; i < REPORT_SIZE; i++) {
 162         if (keyboard_report->raw[i])
 163             cnt++;
 164     }
 165     return cnt;
 166 }
 167
 168 uint8_t host_has_anymod(void)
 169 {
 170     return bitpop(keyboard_report->mods);
 171 }
 172
 173 uint8_t host_get_first_key(void)
 174 {
 175 #ifdef NKRO_ENABLE
 176     if (keyboard_nkro) {
 177         uint8_t i = 0;
 178         for (; i < REPORT_BITS && !keyboard_report->nkro.bits[i]; i++)
 179             ;
 180         return i<<3 | biton(keyboard_report->nkro.bits[i]);
 181     }
 182 #endif
 183     return keyboard_report->keys[0];
 184 }
 185
 186 void host_send_keyboard_report(void)
 187 {
 188     if (!driver) return;
 189     host_keyboard_send(keyboard_report);
 190 }
 191
 192 uint8_t host_mouse_in_use(void)
 193 {
 194     return (mouse_report.buttons | mouse_report.x | mouse_report.y | mouse_report.v | mouse_report.h);
 195 }
 196
 197 uint16_t host_last_sysytem_report(void)
 198 {
 199     return last_system_report;
 200 }
 201
 202 uint16_t host_last_consumer_report(void)
 203 {
 204     return last_consumer_report;
 205 }
 206
 207 static inline void add_key_byte(uint8_t code)
 208 {
 209     int8_t i = 0;
 210     int8_t empty = -1;
 211     for (; i < REPORT_KEYS; i++) {
 212         if (keyboard_report->keys[i] == code) {
 213             break;
 214         }
 215         if (empty == -1 && keyboard_report->keys[i] == 0) {
 216             empty = i;
 217         }
 218     }
 219     if (i == REPORT_KEYS) {
 220         if (empty != -1) {
 221             keyboard_report->keys[empty] = code;
 222         }
 223     }
 224 }
 225
 226 static inline void del_key_byte(uint8_t code)
 227 {
 228     for (uint8_t i = 0; i < REPORT_KEYS; i++) {
 229         if (keyboard_report->keys[i] == code) {
 230             keyboard_report->keys[i] = 0;
 231         }
 232     }
 233 }
 234
 235 #ifdef NKRO_ENABLE
 236 static inline void add_key_bit(uint8_t code)
 237 {
 238     if ((code>>3) < REPORT_BITS) {
 239         keyboard_report->nkro.bits[code>>3] |= 1<<(code&7);
 240     } else {
 241         dprintf("add_key_bit: can't add: %02X\n", code);
 242     }
 243 }
 244
 245 static inline void del_key_bit(uint8_t code)
 246 {
 247     if ((code>>3) < REPORT_BITS) {
 248         keyboard_report->nkro.bits[code>>3] &= ~(1<<(code&7));
 249     } else {
 250         dprintf("del_key_bit: can't del: %02X\n", code);
 251     }
 252 }
 253 #endif