keyboard/hhkb_rn42/rn42/rn42_task.c

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