keyboard/hhkb_rn42/rn42/rn42_task.c

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