keyboards/hhkb/rn42/rn42_task.c

   1 #include <stdint.h>
   2 #include <string.h>
   3 #include <avr/pgmspace.h>
   4 #include <avr/eeprom.h>
   5 #include "keycode.h"
   6 #include "serial.h"
   7 #include "host.h"
   8 #include "action.h"
   9 #include "action_util.h"
  10 #include "lufa.h"
  11 #include "rn42_task.h"
  12 #include "print.h"
  13 #include "debug.h"
  14 #include "timer.h"
  15 #include "wait.h"
  16 #include "command.h"
  17 #include "battery.h"
  18 #include "keycode_config.h"
  19
  20 extern keymap_config_t keymap_config;
  21
  22 static bool config_mode = false;
  23 static bool force_usb = false;
  24
  25 static void status_led(bool on)
  26 {
  27     if (on) {
  28         DDRE  |=  (1<<6);
  29         PORTE &= ~(1<<6);
  30     } else {
  31         DDRE  |=  (1<<6);
  32         PORTE |=  (1<<6);
  33     }
  34 }
  35
  36 void rn42_task_init(void)
  37 {
  38     battery_init();
  39 #ifdef NKRO_ENABLE
  40     rn42_nkro_last = keymap_config.nkro;
  41 #endif
  42 }
  43
  44 void rn42_task(void)
  45 {
  46     int16_t c;
  47     // Raw mode: interpret output report of LED state
  48     while ((c = rn42_getc()) != -1) {
  49         // LED Out report: 0xFE, 0x02, 0x01, <leds>
  50         // To get the report over UART set bit3 with SH, command.
  51         static enum {LED_INIT, LED_FE, LED_02, LED_01} state = LED_INIT;
  52         switch (state) {
  53             case LED_INIT:
  54                 if (c == 0xFE) state = LED_FE;
  55                 else {
  56                     if (0x0 <= c && c <= 0x7f) xprintf("%c", c);
  57                     else xprintf(" %02X", c);
  58                 }
  59                 break;
  60             case LED_FE:
  61                 if (c == 0x02) state = LED_02;
  62                 else           state = LED_INIT;
  63                 break;
  64             case LED_02:
  65                 if (c == 0x01) state = LED_01;
  66                 else           state = LED_INIT;
  67                 break;
  68             case LED_01:
  69                 dprintf("LED status: %02X\n", c);
  70                 rn42_set_leds(c);
  71                 state = LED_INIT;
  72                 break;
  73             default:
  74                 state = LED_INIT;
  75         }
  76     }
  77
  78     /* Switch between USB and Bluetooth */
  79     if (!config_mode) { // not switch while config mode
  80         if (!force_usb && !rn42_rts()) {
  81             if (host_get_driver() != &rn42_driver) {
  82                 clear_keyboard();
  83 #ifdef NKRO_ENABLE
  84                 rn42_nkro_last = keymap_config.nkro;
  85                 keymap_config.nkro = false;
  86 #endif
  87                 host_set_driver(&rn42_driver);
  88             }
  89         } else {
  90             if (host_get_driver() != &lufa_driver) {
  91                 clear_keyboard();
  92 #ifdef NKRO_ENABLE
  93                 keymap_config.nkro = rn42_nkro_last;
  94 #endif
  95                 host_set_driver(&lufa_driver);
  96             }
  97         }
  98     }
  99
 100
 101     static uint16_t prev_timer = 0;
 102     uint16_t e = timer_elapsed(prev_timer);
 103     if (e > 1000) {
 104         /* every second */
 105         prev_timer += e/1000*1000;
 106
 107         /* Low voltage alert */
 108         uint8_t bs = battery_status();
 109         if (bs == LOW_VOLTAGE) {
 110             battery_led(LED_ON);
 111         } else {
 112             battery_led(LED_CHARGER);
 113         }
 114
 115         /* every minute */
 116         uint32_t t = timer_read32()/1000;
 117         if (t%60 == 0) {
 118             uint16_t v = battery_voltage();
 119             uint8_t h = t/3600;
 120             uint8_t m = t%3600/60;
 121             uint8_t s = t%60;
 122             dprintf("%02u:%02u:%02u\t%umV\n", h, m, s, v);
 123             /* TODO: xprintf doesn't work for this.
 124             xprintf("%02u:%02u:%02u\t%umV\n", (t/3600), (t%3600/60), (t%60), v);
 125             */
 126         }
 127     }
 128
 129
 130     /* Connection monitor */
 131     if (!rn42_rts() && rn42_linked()) {
 132         status_led(true);
 133     } else {
 134         status_led(false);
 135     }
 136 }
 137
 138
 139
 140 /******************************************************************************
 141  * Command
 142  ******************************************************************************/
 143 static host_driver_t *prev_driver = &rn42_driver;
 144
 145 static void enter_command_mode(void)
 146 {
 147     prev_driver = host_get_driver();
 148     clear_keyboard();
 149     host_set_driver(&rn42_config_driver);   // null driver; not to send a key to host
 150     rn42_disconnect();
 151     while (rn42_linked()) ;
 152
 153     print("Entering config mode ...\n");
 154     wait_ms(1100);          // need 1 sec
 155     SEND_COMMAND("$$$");
 156     wait_ms(600);           // need 1 sec
 157     rn42_print_response();
 158     const char *s = SEND_COMMAND("v\r\n");
 159     if (strncmp("v", s, 1) != 0) SEND_COMMAND("+\r\n"); // local echo on
 160 }
 161
 162 static void exit_command_mode(void)
 163 {
 164     print("Exiting config mode ...\n");
 165     SEND_COMMAND("---\r\n");    // exit
 166
 167     rn42_autoconnect();
 168     clear_keyboard();
 169     host_set_driver(prev_driver);
 170 }
 171
 172 static void init_rn42(void)
 173 {
 174     // RN-42 configure
 175     if (!config_mode) enter_command_mode();
 176     SEND_COMMAND("SF,1\r\n");  // factory defaults
 177     SEND_COMMAND("S-,TmkBT\r\n");
 178     SEND_COMMAND("SS,Keyboard/Mouse\r\n");
 179     SEND_COMMAND("SM,4\r\n");  // auto connect(DTR)
 180     SEND_COMMAND("SW,8000\r\n");   // Sniff disable
 181     SEND_COMMAND("S~,6\r\n");   // HID profile
 182     SEND_COMMAND("SH,003C\r\n");   // combo device, out-report, 4-reconnect
 183     SEND_COMMAND("SY,FFF4\r\n");   // transmit power -12
 184     SEND_COMMAND("R,1\r\n");
 185     if (!config_mode) exit_command_mode();
 186 }
 187
 188 #if 0
 189 // Switching connections
 190 // NOTE: Remote Address doesn't work in the way manual says.
 191 // EEPROM address for link store
 192 #define RN42_LINK0  (uint8_t *)128
 193 #define RN42_LINK1  (uint8_t *)140
 194 #define RN42_LINK2  (uint8_t *)152
 195 #define RN42_LINK3  (uint8_t *)164
 196 static void store_link(uint8_t *eeaddr)
 197 {
 198     enter_command_mode();
 199     SEND_STR("GR\r\n"); // remote address
 200     const char *s = rn42_gets(500);
 201     if (strcmp("GR", s) == 0) s = rn42_gets(500);   // ignore local echo
 202     xprintf("%s(%d)\r\n", s, strlen(s));
 203     if (strlen(s) == 12) {
 204         for (int i = 0; i < 12; i++) {
 205             eeprom_write_byte(eeaddr+i, *(s+i));
 206             dprintf("%c ", *(s+i));
 207         }
 208         dprint("\r\n");
 209     }
 210     exit_command_mode();
 211 }
 212
 213 static void restore_link(const uint8_t *eeaddr)
 214 {
 215     enter_command_mode();
 216     SEND_COMMAND("SR,Z\r\n");   // remove remote address
 217     SEND_STR("SR,");            // set remote address from EEPROM
 218     for (int i = 0; i < 12; i++) {
 219         uint8_t c = eeprom_read_byte(eeaddr+i);
 220         rn42_putc(c);
 221         dprintf("%c ", c);
 222     }
 223     dprintf("\r\n");
 224     SEND_COMMAND("\r\n");
 225     SEND_COMMAND("R,1\r\n");    // reboot
 226     exit_command_mode();
 227 }
 228
 229 static const char *get_link(uint8_t * eeaddr)
 230 {
 231     static char s[13];
 232     for (int i = 0; i < 12; i++) {
 233         uint8_t c = eeprom_read_byte(eeaddr+i);
 234         s[i] = c;
 235     }
 236     s[12] = '\0';
 237     return s;
 238 }
 239 #endif
 240
 241 static void pairing(void)
 242 {
 243     enter_command_mode();
 244     SEND_COMMAND("SR,Z\r\n");   // remove remote address
 245     SEND_COMMAND("R,1\r\n");    // reboot
 246     exit_command_mode();
 247 }
 248
 249 bool command_extra(uint8_t code)
 250 {
 251     uint32_t t;
 252     uint16_t b;
 253     switch (code) {
 254         case KC_H:
 255         case KC_SLASH: /* ? */
 256             print("\n\n----- Bluetooth RN-42 Help -----\n");
 257             print("i:       RN-42 info\n");
 258             print("b:       battery voltage\n");
 259             print("Del:     enter/exit RN-42 config mode\n");
 260             print("Slck:    RN-42 initialize\n");
 261 #if 0
 262             print("1-4:     restore link\n");
 263             print("F1-F4:   store link\n");
 264 #endif
 265             print("p:       pairing\n");
 266
 267             if (config_mode) {
 268                 return true;
 269             } else {
 270                 print("u:       toggle Force USB mode\n");
 271                 return false;   // to display default command help
 272             }
 273         case KC_P:
 274             pairing();
 275             return true;
 276 #if 0
 277         /* Store link address to EEPROM */
 278         case KC_F1:
 279             store_link(RN42_LINK0);
 280             return true;
 281         case KC_F2:
 282             store_link(RN42_LINK1);
 283             return true;
 284         case KC_F3:
 285             store_link(RN42_LINK2);
 286             return true;
 287         case KC_F4:
 288             store_link(RN42_LINK3);
 289             return true;
 290         /* Restore link address to EEPROM */
 291         case KC_1:
 292             restore_link(RN42_LINK0);
 293             return true;
 294         case KC_2:
 295             restore_link(RN42_LINK1);
 296             return true;
 297         case KC_3:
 298             restore_link(RN42_LINK2);
 299             return true;
 300         case KC_4:
 301             restore_link(RN42_LINK3);
 302             return true;
 303 #endif
 304         case KC_I:
 305             print("\n----- RN-42 info -----\n");
 306             xprintf("protocol: %s\n", (host_get_driver() == &rn42_driver) ? "RN-42" : "LUFA");
 307             xprintf("force_usb: %X\n", force_usb);
 308             xprintf("rn42: %s\n", rn42_rts() ? "OFF" : (rn42_linked() ? "CONN" : "ON"));
 309             xprintf("rn42_autoconnecting(): %X\n", rn42_autoconnecting());
 310             xprintf("config_mode: %X\n", config_mode);
 311             xprintf("USB State: %s\n",
 312                     (USB_DeviceState == DEVICE_STATE_Unattached) ? "Unattached" :
 313                     (USB_DeviceState == DEVICE_STATE_Powered) ? "Powered" :
 314                     (USB_DeviceState == DEVICE_STATE_Default) ? "Default" :
 315                     (USB_DeviceState == DEVICE_STATE_Addressed) ? "Addressed" :
 316                     (USB_DeviceState == DEVICE_STATE_Configured) ? "Configured" :
 317                     (USB_DeviceState == DEVICE_STATE_Suspended) ? "Suspended" : "?");
 318             xprintf("battery: ");
 319             switch (battery_status()) {
 320                 case FULL_CHARGED:  xprintf("FULL"); break;
 321                 case CHARGING:      xprintf("CHARG"); break;
 322                 case DISCHARGING:   xprintf("DISCHG"); break;
 323                 case LOW_VOLTAGE:   xprintf("LOW"); break;
 324                 default:            xprintf("?"); break;
 325             };
 326             xprintf("\n");
 327             xprintf("RemoteWakeupEnabled: %X\n", USB_Device_RemoteWakeupEnabled);
 328             xprintf("VBUS: %X\n", USBSTA&(1<<VBUS));
 329             t = timer_read32()/1000;
 330             uint8_t d = t/3600/24;
 331             uint8_t h = t/3600;
 332             uint8_t m = t%3600/60;
 333             uint8_t s = t%60;
 334             xprintf("uptime: %02u %02u:%02u:%02u\n", d, h, m, s);
 335 #if 0
 336             xprintf("LINK0: %s\r\n", get_link(RN42_LINK0));
 337             xprintf("LINK1: %s\r\n", get_link(RN42_LINK1));
 338             xprintf("LINK2: %s\r\n", get_link(RN42_LINK2));
 339             xprintf("LINK3: %s\r\n", get_link(RN42_LINK3));
 340 #endif
 341             return true;
 342         case KC_B:
 343             // battery monitor
 344             t = timer_read32()/1000;
 345             b = battery_voltage();
 346             xprintf("BAT: %umV\t", b);
 347             xprintf("%02u:",   t/3600);
 348             xprintf("%02u:",   t%3600/60);
 349             xprintf("%02u\n",  t%60);
 350             return true;
 351         case KC_U:
 352             if (config_mode) return false;
 353             if (force_usb) {
 354                 print("Auto mode\n");
 355                 force_usb = false;
 356             } else {
 357                 print("USB mode\n");
 358                 force_usb = true;
 359             }
 360             return true;
 361         case KC_DELETE:
 362             /* RN-42 Command mode */
 363             if (rn42_autoconnecting()) {
 364                 enter_command_mode();
 365
 366                 command_state = CONSOLE;
 367                 config_mode = true;
 368             } else {
 369                 exit_command_mode();
 370
 371                 command_state = ONESHOT;
 372                 config_mode = false;
 373             }
 374             return true;
 375         case KC_SCROLLLOCK:
 376             init_rn42();
 377             return true;
 378 #ifdef NKRO_ENABLE
 379         case KC_N:
 380             if (host_get_driver() != &lufa_driver) {
 381                 // ignored unless USB mode
 382                 return true;
 383             }
 384             return false;
 385 #endif
 386         default:
 387             if (config_mode)
 388                 return true;
 389             else
 390                 return false;   // yield to default command
 391     }
 392     return true;
 393 }
 394
 395 /*
 396  * RN-42 Command mode
 397  * sends charactors to the module
 398  */
 399 static uint8_t code2asc(uint8_t code);
 400 bool command_console_extra(uint8_t code)
 401 {
 402     rn42_putc(code2asc(code));
 403     return true;
 404 }
 405
 406 // convert keycode into ascii charactor
 407 static uint8_t code2asc(uint8_t code)
 408 {
 409     bool shifted = (get_mods() & (MOD_BIT(KC_LSHIFT)|MOD_BIT(KC_RSHIFT))) ? true : false;
 410     switch (code) {
 411         case KC_A: return (shifted ? 'A' : 'a');
 412         case KC_B: return (shifted ? 'B' : 'b');
 413         case KC_C: return (shifted ? 'C' : 'c');
 414         case KC_D: return (shifted ? 'D' : 'd');
 415         case KC_E: return (shifted ? 'E' : 'e');
 416         case KC_F: return (shifted ? 'F' : 'f');
 417         case KC_G: return (shifted ? 'G' : 'g');
 418         case KC_H: return (shifted ? 'H' : 'h');
 419         case KC_I: return (shifted ? 'I' : 'i');
 420         case KC_J: return (shifted ? 'J' : 'j');
 421         case KC_K: return (shifted ? 'K' : 'k');
 422         case KC_L: return (shifted ? 'L' : 'l');
 423         case KC_M: return (shifted ? 'M' : 'm');
 424         case KC_N: return (shifted ? 'N' : 'n');
 425         case KC_O: return (shifted ? 'O' : 'o');
 426         case KC_P: return (shifted ? 'P' : 'p');
 427         case KC_Q: return (shifted ? 'Q' : 'q');
 428         case KC_R: return (shifted ? 'R' : 'r');
 429         case KC_S: return (shifted ? 'S' : 's');
 430         case KC_T: return (shifted ? 'T' : 't');
 431         case KC_U: return (shifted ? 'U' : 'u');
 432         case KC_V: return (shifted ? 'V' : 'v');
 433         case KC_W: return (shifted ? 'W' : 'w');
 434         case KC_X: return (shifted ? 'X' : 'x');
 435         case KC_Y: return (shifted ? 'Y' : 'y');
 436         case KC_Z: return (shifted ? 'Z' : 'z');
 437         case KC_1: return (shifted ? '!' : '1');
 438         case KC_2: return (shifted ? '@' : '2');
 439         case KC_3: return (shifted ? '#' : '3');
 440         case KC_4: return (shifted ? '$' : '4');
 441         case KC_5: return (shifted ? '%' : '5');
 442         case KC_6: return (shifted ? '^' : '6');
 443         case KC_7: return (shifted ? '&' : '7');
 444         case KC_8: return (shifted ? '*' : '8');
 445         case KC_9: return (shifted ? '(' : '9');
 446         case KC_0: return (shifted ? ')' : '0');
 447         case KC_ENTER: return '\n';
 448         case KC_ESCAPE: return 0x1B;
 449         case KC_BSPACE: return '\b';
 450         case KC_TAB: return '\t';
 451         case KC_SPACE: return ' ';
 452         case KC_MINUS: return (shifted ? '_' : '-');
 453         case KC_EQUAL: return (shifted ? '+' : '=');
 454         case KC_LBRACKET: return (shifted ? '{' : '[');
 455         case KC_RBRACKET: return (shifted ? '}' : ']');
 456         case KC_BSLASH: return (shifted ? '|' : '\\');
 457         case KC_NONUS_HASH: return (shifted ? '|' : '\\');
 458         case KC_SCOLON: return (shifted ? ':' : ';');
 459         case KC_QUOTE: return (shifted ? '"' : '\'');
 460         case KC_GRAVE: return (shifted ? '~' : '`');
 461         case KC_COMMA: return (shifted ? '<' : ',');
 462         case KC_DOT: return (shifted ? '>' : '.');
 463         case KC_SLASH: return (shifted ? '?' : '/');
 464         case KC_DELETE: return '\0';    // Delete to disconnect
 465         default: return ' ';
 466     }
 467 }