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 "timer.h"
  11 #include "command.h"
  12 #include "battery.h"
  13
  14 static bool config_mode = false;
  15 static bool force_usb = false;
  16
  17 static void status_led(bool on)
  18 {
  19     if (on) {
  20         DDRE  |=  (1<<6);
  21         PORTE &= ~(1<<6);
  22     } else {
  23         DDRE  |=  (1<<6);
  24         PORTE |=  (1<<6);
  25     }
  26 }
  27
  28 void rn42_task_init(void)
  29 {
  30     battery_init();
  31 }
  32
  33 void rn42_task(void)
  34 {
  35     int16_t c;
  36     if (config_mode) {
  37         // Config mode: print output from RN-42
  38         while ((c = serial_recv2()) != -1) {
  39             // without flow control it'll fail to receive data when flooded
  40             xprintf("%c", c);
  41         }
  42     } else {
  43         // Raw mode: interpret output report of LED state
  44         while ((c = serial_recv2()) != -1) {
  45             // LED Out report: 0xFE, 0x02, 0x01, <leds>
  46             // To get the report over UART set bit3 with SH, command.
  47             static enum {LED_INIT, LED_FE, LED_02, LED_01} state = LED_INIT;
  48             xprintf("%02X\n", c);
  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                     // TODO: move to rn42.c and make accessible with keyboard_leds()
  64                     xprintf("LED status: %02X\n", 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     /* Low voltage alert */
  85     if (battery_status() == LOW_VOLTAGE) {
  86         battery_led(LED_ON);
  87     } else {
  88         battery_led(LED_CHARGER);
  89     }
  90
  91     /* Connection monitor */
  92     if (rn42_linked()) {
  93         status_led(true);
  94     } else {
  95         status_led(false);
  96     }
  97 }
  98
  99
 100
 101 /******************************************************************************
 102  * Command
 103  ******************************************************************************/
 104 bool command_extra(uint8_t code)
 105 {
 106     uint32_t t;
 107     uint16_t b;
 108     static host_driver_t *prev_driver = &rn42_driver;
 109     switch (code) {
 110         case KC_H:
 111         case KC_SLASH: /* ? */
 112             print("\n\n----- Bluetooth RN-42 Help -----\n");
 113             print("Del: enter/exit config mode(auto_connect/disconnect)\n");
 114             print("i:   RN-42 info\n");
 115             print("b:   battery voltage\n");
 116
 117             if (config_mode) {
 118                 return true;
 119             } else {
 120                 print("u:   Force USB mode\n");
 121                 return false;   // to display default command help
 122             }
 123         case KC_DELETE:
 124             if (rn42_autoconnecting()) {
 125                 prev_driver = host_get_driver();
 126                 clear_keyboard();
 127                 _delay_ms(500);
 128                 host_set_driver(&rn42_config_driver);   // null driver; not to send a key to host
 129                 rn42_disconnect();
 130                 print("\nRN-42: disconnect\n");
 131                 print("Enter config mode\n");
 132                 print("type $$$ to start and + for local echo\n");
 133                 command_state = CONSOLE;
 134                 config_mode = true;
 135             } else {
 136                 rn42_autoconnect();
 137                 print("\nRN-42: auto_connect\n");
 138                 print("Exit config mode\n");
 139                 command_state = ONESHOT;
 140                 config_mode = false;
 141                 //clear_keyboard();
 142                 host_set_driver(prev_driver);
 143             }
 144             return true;
 145         case KC_U:
 146             if (config_mode) return false;
 147             if (force_usb) {
 148                 print("Auto mode\n");
 149                 force_usb = false;
 150             } else {
 151                 print("USB mode\n");
 152                 force_usb = true;
 153                 clear_keyboard();
 154                 host_set_driver(&lufa_driver);
 155             }
 156             return true;
 157         case KC_I:
 158             print("\n----- RN-42 info -----\n");
 159             xprintf("protocol: %s\n", (host_get_driver() == &rn42_driver) ? "RN-42" : "LUFA");
 160             xprintf("force_usb: %X\n", force_usb);
 161             xprintf("rn42_autoconnecting(): %X\n", rn42_autoconnecting());
 162             xprintf("rn42_linked(): %X\n", rn42_linked());
 163             xprintf("rn42_rts(): %X\n", rn42_rts());
 164             xprintf("config_mode: %X\n", config_mode);
 165             xprintf("VBUS: %X\n", USBSTA&(1<<VBUS));
 166             xprintf("battery_charging: %X\n", battery_charging());
 167             xprintf("battery_status: %X\n", battery_status());
 168             return true;
 169         case KC_B:
 170             // battery monitor
 171             t = timer_read32()/1000;
 172             b = battery_voltage();
 173             xprintf("BAT: %umV\t", b);
 174             xprintf("%02u:",   t/3600);
 175             xprintf("%02u:",   t%3600/60);
 176             xprintf("%02u\n",  t%60);
 177             return true;
 178         default:
 179             if (config_mode)
 180                 return true;
 181             else
 182                 return false;   // exec default command
 183     }
 184     return true;
 185 }
 186
 187 static uint8_t code2asc(uint8_t code);
 188 bool command_console_extra(uint8_t code)
 189 {
 190     switch (code) {
 191         default:
 192             rn42_putc(code2asc(code));
 193             return true;
 194     }
 195     return false;
 196 }
 197
 198 // convert keycode into ascii charactor
 199 static uint8_t code2asc(uint8_t code)
 200 {
 201     bool shifted = (get_mods() & (MOD_BIT(KC_LSHIFT)|MOD_BIT(KC_RSHIFT))) ? true : false;
 202     switch (code) {
 203         case KC_A: return (shifted ? 'A' : 'a');
 204         case KC_B: return (shifted ? 'B' : 'b');
 205         case KC_C: return (shifted ? 'C' : 'c');
 206         case KC_D: return (shifted ? 'D' : 'd');
 207         case KC_E: return (shifted ? 'E' : 'e');
 208         case KC_F: return (shifted ? 'F' : 'f');
 209         case KC_G: return (shifted ? 'G' : 'g');
 210         case KC_H: return (shifted ? 'H' : 'h');
 211         case KC_I: return (shifted ? 'I' : 'i');
 212         case KC_J: return (shifted ? 'J' : 'j');
 213         case KC_K: return (shifted ? 'K' : 'k');
 214         case KC_L: return (shifted ? 'L' : 'l');
 215         case KC_M: return (shifted ? 'M' : 'm');
 216         case KC_N: return (shifted ? 'N' : 'n');
 217         case KC_O: return (shifted ? 'O' : 'o');
 218         case KC_P: return (shifted ? 'P' : 'p');
 219         case KC_Q: return (shifted ? 'Q' : 'q');
 220         case KC_R: return (shifted ? 'R' : 'r');
 221         case KC_S: return (shifted ? 'S' : 's');
 222         case KC_T: return (shifted ? 'T' : 't');
 223         case KC_U: return (shifted ? 'U' : 'u');
 224         case KC_V: return (shifted ? 'V' : 'v');
 225         case KC_W: return (shifted ? 'W' : 'w');
 226         case KC_X: return (shifted ? 'X' : 'x');
 227         case KC_Y: return (shifted ? 'Y' : 'y');
 228         case KC_Z: return (shifted ? 'Z' : 'z');
 229         case KC_1: return (shifted ? '!' : '1');
 230         case KC_2: return (shifted ? '@' : '2');
 231         case KC_3: return (shifted ? '#' : '3');
 232         case KC_4: return (shifted ? '$' : '4');
 233         case KC_5: return (shifted ? '%' : '5');
 234         case KC_6: return (shifted ? '^' : '6');
 235         case KC_7: return (shifted ? '&' : '7');
 236         case KC_8: return (shifted ? '*' : '8');
 237         case KC_9: return (shifted ? '(' : '9');
 238         case KC_0: return (shifted ? ')' : '0');
 239         case KC_ENTER: return '\n';
 240         case KC_ESCAPE: return 0x1B;
 241         case KC_BSPACE: return '\b';
 242         case KC_TAB: return '\t';
 243         case KC_SPACE: return ' ';
 244         case KC_MINUS: return (shifted ? '_' : '-');
 245         case KC_EQUAL: return (shifted ? '+' : '=');
 246         case KC_LBRACKET: return (shifted ? '{' : '[');
 247         case KC_RBRACKET: return (shifted ? '}' : ']');
 248         case KC_BSLASH: return (shifted ? '|' : '\\');
 249         case KC_NONUS_HASH: return (shifted ? '|' : '\\');
 250         case KC_SCOLON: return (shifted ? ':' : ';');
 251         case KC_QUOTE: return (shifted ? '"' : '\'');
 252         case KC_GRAVE: return (shifted ? '~' : '`');
 253         case KC_COMMA: return (shifted ? '<' : ',');
 254         case KC_DOT: return (shifted ? '>' : '.');
 255         case KC_SLASH: return (shifted ? '?' : '/');
 256         case KC_DELETE: return '\0';    // Delete to disconnect
 257         default: return ' ';
 258     }
 259 }