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