#include <avr/io.h> #include <avr/wdt.h> #include <avr/power.h> #include <util/delay.h> // LUFA #include "lufa.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 */ #if (F_CPU == 8000000) clock_prescale_set(clock_div_2); // 16MHz crystal divided by 2 #else clock_prescale_set(clock_div_1); #endif // Leonardo needs. Without this USB device is not recognized. USB_Disable(); USB_Init(); // for Console_Task USB_Device_EnableSOFEvents(); print_set_sendchar(sendchar); } 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(); /* NOTE: Don't insert time consuming job here. * It'll cause unclear initialization failure when DFU reset(worm start). */ sei(); /* Some keyboards bootup quickly and cannot be initialized with this startup wait.*/ // wait for startup of sendchar routine while (USB_DeviceState != DEVICE_STATE_Configured) ; if (debug_enable) { _delay_ms(1000); } debug("init: done\n"); for (;;) { keyboard_task(); #if !defined(INTERRUPT_CONTROL_ENDPOINT) // LUFA Task for control request USB_USBTask(); #endif } return 0; }