133 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include <avr/io.h>
 | |
| #include <avr/wdt.h>
 | |
| #include <avr/power.h>
 | |
| #include <util/delay.h>
 | |
| 
 | |
| // USB HID host
 | |
| #include "Usb.h"
 | |
| #include "usbhub.h"
 | |
| #include "hid.h"
 | |
| #include "hidboot.h"
 | |
| #include "parser.h"
 | |
| 
 | |
| // LUFA
 | |
| #include "lufa.h"
 | |
| 
 | |
| #include "timer.h"
 | |
| #include "sendchar.h"
 | |
| #include "debug.h"
 | |
| #include "keyboard.h"
 | |
| #include "led.h"
 | |
| 
 | |
| 
 | |
| /* LED ping configuration */
 | |
| #define TMK_LED
 | |
| //#define LEONARDO_LED
 | |
| #if defined(TMK_LED)
 | |
| // For TMK converter and Teensy
 | |
| #define LED_TX_INIT    (DDRD  |=  (1<<6))
 | |
| #define LED_TX_ON      (PORTD |=  (1<<6))
 | |
| #define LED_TX_OFF     (PORTD &= ~(1<<6))
 | |
| #define LED_TX_TOGGLE  (PORTD ^=  (1<<6))
 | |
| #elif defined(LEONARDO_LED)
 | |
| // For Leonardo(TX LED)
 | |
| #define LED_TX_INIT    (DDRD  |=  (1<<5))
 | |
| #define LED_TX_ON      (PORTD &= ~(1<<5))
 | |
| #define LED_TX_OFF     (PORTD |=  (1<<5))
 | |
| #define LED_TX_TOGGLE  (PORTD ^=  (1<<5))
 | |
| #else
 | |
| #define LED_TX_INIT
 | |
| #define LED_TX_ON
 | |
| #define LED_TX_OFF
 | |
| #define LED_TX_TOGGLE
 | |
| #endif
 | |
| 
 | |
| 
 | |
| static void LUFA_setup(void)
 | |
| {
 | |
|     /* Disable watchdog if enabled by bootloader/fuses */
 | |
|     MCUSR &= ~(1 << WDRF);
 | |
|     wdt_disable();
 | |
| 
 | |
|     /* Disable clock division */
 | |
|     clock_prescale_set(clock_div_1);
 | |
| 
 | |
|     // Leonardo needs. Without this USB device is not recognized.
 | |
|     USB_Disable();
 | |
| 
 | |
|     USB_Init();
 | |
| 
 | |
|     // for Console_Task
 | |
|     USB_Device_EnableSOFEvents();
 | |
|     print_set_sendchar(sendchar);
 | |
| }
 | |
| 
 | |
| 
 | |
| 
 | |
| /*
 | |
|  * USB Host Shield HID keyboard
 | |
|  */
 | |
| USB usb_host;
 | |
| USBHub hub1(&usb_host);
 | |
| HIDBoot<HID_PROTOCOL_KEYBOARD>    kbd(&usb_host);
 | |
| KBDReportParser kbd_parser;
 | |
| 
 | |
| 
 | |
| void led_set(uint8_t usb_led)
 | |
| {
 | |
|     kbd.SetReport(0, 0, 2, 0, 1, &usb_led);
 | |
| }
 | |
| 
 | |
| 
 | |
| 
 | |
| int main(void)
 | |
| {
 | |
|     // LED for debug
 | |
|     LED_TX_INIT;
 | |
|     LED_TX_ON;
 | |
| 
 | |
|     debug_enable = true;
 | |
|     debug_keyboard = true;
 | |
| 
 | |
|     host_set_driver(&lufa_driver);
 | |
|     keyboard_init();
 | |
| 
 | |
|     LUFA_setup();
 | |
| 
 | |
|     // USB Host Shield setup
 | |
|     usb_host.Init();
 | |
|     kbd.SetReportParser(0, (HIDReportParser*)&kbd_parser);
 | |
| 
 | |
|     /* NOTE: Don't insert time consuming job here.
 | |
|      * It'll cause unclear initialization failure when DFU reset(worm start).
 | |
|      */
 | |
|     sei();
 | |
| 
 | |
|     // wait for startup of sendchar routine
 | |
|     while (USB_DeviceState != DEVICE_STATE_Configured) ;
 | |
|     if (debug_enable) {
 | |
|         _delay_ms(1000);
 | |
|     }
 | |
| 
 | |
|     debug("init: done\n");
 | |
| 
 | |
| uint16_t timer;
 | |
|     for (;;) {
 | |
|         keyboard_task();
 | |
| 
 | |
| timer = timer_read();
 | |
|         usb_host.Task();
 | |
| timer = timer_elapsed(timer);
 | |
| if (timer > 100) {
 | |
|     debug("host.Task: "); debug_hex16(timer);  debug("\n");
 | |
| }
 | |
| 
 | |
| #if !defined(INTERRUPT_CONTROL_ENDPOINT)
 | |
|         // LUFA Task for control request
 | |
|         USB_USBTask();
 | |
| #endif
 | |
|     }
 | |
| 
 | |
|     return 0;
 | |
| }
 |