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;
 | 
						|
}
 |