aboutsummaryrefslogtreecommitdiff
path: root/iwrap/main.c
diff options
context:
space:
mode:
Diffstat (limited to 'iwrap/main.c')
-rw-r--r--iwrap/main.c378
1 files changed, 378 insertions, 0 deletions
diff --git a/iwrap/main.c b/iwrap/main.c
new file mode 100644
index 000000000..a552afb67
--- /dev/null
+++ b/iwrap/main.c
@@ -0,0 +1,378 @@
1/*
2Copyright 2011 Jun Wako <wakojun@gmail.com>
3
4This program is free software: you can redistribute it and/or modify
5it under the terms of the GNU General Public License as published by
6the Free Software Foundation, either version 2 of the License, or
7(at your option) any later version.
8
9This program is distributed in the hope that it will be useful,
10but WITHOUT ANY WARRANTY; without even the implied warranty of
11MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12GNU General Public License for more details.
13
14You should have received a copy of the GNU General Public License
15along with this program. If not, see <http://www.gnu.org/licenses/>.
16*/
17#include <stdint.h>
18#include <avr/interrupt.h>
19#include <avr/io.h>
20//#include <avr/wdt.h>
21#include "wd.h" // in order to use watchdog in interrupt mode
22#include <avr/sleep.h>
23#include <util/delay.h>
24#include <avr/power.h>
25#include "keyboard.h"
26#include "matrix.h"
27#include "host.h"
28#include "iwrap.h"
29#ifdef HOST_VUSB
30# include "vusb.h"
31# include "usbdrv.h"
32#endif
33#include "uart.h"
34#include "suart.h"
35#include "timer.h"
36#include "debug.h"
37#include "usb_keycodes.h"
38#include "command.h"
39
40
41static void sleep(uint8_t term);
42static bool console(void);
43static uint8_t console_command(uint8_t c);
44static uint8_t key2asc(uint8_t key);
45
46
47/*
48static void set_prr(void)
49{
50 power_adc_disable();
51 power_spi_disable();
52 power_twi_disable();
53#ifndef TIMER_H
54 //power_timer0_disable(); // used in timer.c
55#endif
56 power_timer1_disable();
57 power_timer2_disable();
58}
59*/
60
61/*
62static void pullup_pins(void)
63{
64 // DDRs are set to 0(input) by default.
65#ifdef PORTA
66 PORTA = 0xFF;
67#endif
68 PORTB = 0xFF;
69 PORTC = 0xFF;
70 PORTD = 0xFF;
71#ifdef PORTE
72 PORTE = 0xFF;
73#endif
74#ifdef PORTE
75 PORTF = 0xFF;
76#endif
77}
78*/
79
80
81#ifdef HOST_VUSB
82static void disable_vusb(void)
83{
84 // disable interrupt & disconnect to prevent host from enumerating
85 USB_INTR_ENABLE &= ~(1 << USB_INTR_ENABLE_BIT);
86 usbDeviceDisconnect();
87}
88
89static void enable_vusb(void)
90{
91 USB_INTR_ENABLE |= (1 << USB_INTR_ENABLE_BIT);
92 usbDeviceConnect();
93}
94
95static void init_vusb(void)
96{
97 uint8_t i = 0;
98
99 usbInit();
100 disable_vusb();
101 /* fake USB disconnect for > 250 ms */
102 while(--i){
103 _delay_ms(1);
104 }
105 enable_vusb();
106}
107#endif
108
109void change_driver(host_driver_t *driver)
110{
111 host_clear_keyboard_report();
112 host_swap_keyboard_report();
113 host_clear_keyboard_report();
114 host_send_keyboard_report();
115 _delay_ms(1000);
116 host_set_driver(driver);
117}
118
119
120static bool sleeping = false;
121static bool insomniac = false; // TODO: should be false for power saving
122static uint16_t last_timer = 0;
123
124int main(void)
125{
126 MCUSR = 0;
127 clock_prescale_set(clock_div_1);
128 WD_SET(WD_OFF);
129
130 // power saving: the result is worse than nothing... why?
131 //pullup_pins();
132 //set_prr();
133
134 print_enable = true;
135 debug_enable = false;
136
137#ifdef HOST_VUSB
138 disable_vusb();
139#endif
140 uart_init(115200);
141 keyboard_init();
142 print("\nSend BREAK for UART Console Commands.\n");
143
144 // TODO: move to iWRAP/suart file
145 print("suart init\n");
146 // suart init
147 // PC4: Tx Output IDLE(Hi)
148 PORTC |= (1<<4);
149 DDRC |= (1<<4);
150 // PC5: Rx Input(pull-up)
151 PORTC |= (1<<5);
152 DDRC &= ~(1<<5);
153 // suart receive interrut(PC5/PCINT13)
154 PCMSK1 = 0b00100000;
155 PCICR = 0b00000010;
156
157 host_set_driver(iwrap_driver());
158
159 print("iwrap_init()\n");
160 iwrap_init();
161 iwrap_call();
162
163 last_timer = timer_read();
164 while (true) {
165#ifdef HOST_VUSB
166 if (host_get_driver() == vusb_driver())
167 usbPoll();
168#endif
169 keyboard_proc();
170#ifdef HOST_VUSB
171 if (host_get_driver() == vusb_driver())
172 vusb_transfer_keyboard();
173#endif
174 if (matrix_is_modified() || console()) {
175 last_timer = timer_read();
176 sleeping = false;
177 } else if (!sleeping && timer_elapsed(last_timer) > 4000) {
178 sleeping = true;
179 iwrap_check_connection();
180 }
181
182 if (host_get_driver() == iwrap_driver()) {
183 if (sleeping && !insomniac) {
184 _delay_ms(1); // wait for UART to send
185 iwrap_sleep();
186 sleep(WDTO_60MS);
187 }
188 }
189 }
190}
191
192static void sleep(uint8_t term)
193{
194 WD_SET(WD_IRQ, term);
195
196 cli();
197 set_sleep_mode(SLEEP_MODE_PWR_DOWN);
198 sleep_enable();
199 sleep_bod_disable();
200 sei();
201 sleep_cpu();
202 sleep_disable();
203
204 WD_SET(WD_OFF);
205}
206
207ISR(WDT_vect)
208{
209 // wake up
210}
211
212static bool console(void)
213{
214 // Send to Bluetoot module WT12
215 static bool breaked = false;
216 if (!uart_available())
217 return false;
218 else {
219 uint8_t c;
220 c = uart_getchar();
221 uart_putchar(c);
222 switch (c) {
223 case 0x00: // BREAK signal
224 if (!breaked) {
225 print("break(? for help): ");
226 breaked = true;
227 }
228 break;
229 case '\r':
230 uart_putchar('\n');
231 iwrap_buf_send();
232 break;
233 case '\b':
234 iwrap_buf_del();
235 break;
236 default:
237 if (breaked) {
238 print("\n");
239 console_command(c);
240 breaked = false;
241 } else {
242 iwrap_buf_add(c);
243 }
244 break;
245 }
246 return true;
247 }
248}
249
250uint8_t command_extra()
251{
252 return console_command(key2asc(host_get_first_key()));
253}
254
255static uint8_t console_command(uint8_t c)
256{
257 switch (c) {
258 case 'h':
259 case '?':
260 print("\nCommands for Bluetooth(WT12/iWRAP):\n");
261 print("r: reset. software reset by watchdog\n");
262 print("i: insomniac. prevent KB from sleeping\n");
263 print("c: iwrap_call. CALL for BT connection.\n");
264#ifdef HOST_VUSB
265 print("u: USB mode. switch to USB.\n");
266 print("w: BT mode. switch to Bluetooth.\n");
267#endif
268 print("k: kill first connection.\n");
269 print("Del: unpair first pairing.\n");
270 print("\n");
271 return 0;
272 case 'r':
273 print("reset\n");
274 WD_AVR_RESET();
275 return 1;
276 case 'i':
277 insomniac = !insomniac;
278 if (insomniac)
279 print("insomniac\n");
280 else
281 print("not insomniac\n");
282 return 1;
283 case 'c':
284 print("iwrap_call()\n");
285 iwrap_call();
286 return 1;
287#ifdef HOST_VUSB
288 case 'u':
289 print("USB mode\n");
290 init_vusb();
291 change_driver(vusb_driver());
292 //iwrap_kill();
293 //iwrap_sleep();
294 // disable suart receive interrut(PC5/PCINT13)
295 PCMSK1 &= ~(0b00100000);
296 PCICR &= ~(0b00000010);
297 return 1;
298 case 'w':
299 print("iWRAP mode\n");
300 change_driver(iwrap_driver());
301 disable_vusb();
302 // enable suart receive interrut(PC5/PCINT13)
303 PCMSK1 |= 0b00100000;
304 PCICR |= 0b00000010;
305 return 1;
306#endif
307 case 'k':
308 print("kill\n");
309 iwrap_kill();
310 return 1;
311 case 0x7F: // DELETE
312 print("unpair\n");
313 iwrap_unpair();
314 return 1;
315 }
316 return 0;
317}
318
319// convert keycode into ascii charactor
320static uint8_t key2asc(uint8_t key)
321{
322 switch (key) {
323 case KB_A: return 'a';
324 case KB_B: return 'b';
325 case KB_C: return 'c';
326 case KB_D: return 'd';
327 case KB_E: return 'e';
328 case KB_F: return 'f';
329 case KB_G: return 'g';
330 case KB_H: return 'h';
331 case KB_I: return 'i';
332 case KB_J: return 'j';
333 case KB_K: return 'k';
334 case KB_L: return 'l';
335 case KB_M: return 'm';
336 case KB_N: return 'n';
337 case KB_O: return 'o';
338 case KB_P: return 'p';
339 case KB_Q: return 'q';
340 case KB_R: return 'r';
341 case KB_S: return 's';
342 case KB_T: return 't';
343 case KB_U: return 'u';
344 case KB_V: return 'v';
345 case KB_W: return 'w';
346 case KB_X: return 'x';
347 case KB_Y: return 'y';
348 case KB_Z: return 'z';
349 case KB_1: return '1';
350 case KB_2: return '2';
351 case KB_3: return '3';
352 case KB_4: return '4';
353 case KB_5: return '5';
354 case KB_6: return '6';
355 case KB_7: return '7';
356 case KB_8: return '8';
357 case KB_9: return '9';
358 case KB_0: return '0';
359 case KB_ENTER: return '\n';
360 case KB_ESCAPE: return 0x1B;
361 case KB_BSPACE: return '\b';
362 case KB_TAB: return '\t';
363 case KB_SPACE: return ' ';
364 case KB_MINUS: return '-';
365 case KB_EQUAL: return '=';
366 case KB_LBRACKET: return '[';
367 case KB_RBRACKET: return ']';
368 case KB_BSLASH: return '\\';
369 case KB_NONUS_HASH: return '\\';
370 case KB_SCOLON: return ';';
371 case KB_QUOTE: return '\'';
372 case KB_GRAVE: return '`';
373 case KB_COMMA: return ',';
374 case KB_DOT: return '.';
375 case KB_SLASH: return '/';
376 default: return 0x00;
377 }
378}
generated by cgit v1.2.3 (git 2.25.1) at 2025-11-21 23:33:29 +0000