You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							390 lines
						
					
					
						
							12 KiB
						
					
					
				
			
		
		
	
	
							390 lines
						
					
					
						
							12 KiB
						
					
					
				/*
 | 
						|
             LUFA Library
 | 
						|
     Copyright (C) Dean Camera, 2009.
 | 
						|
              
 | 
						|
  dean [at] fourwalledcubicle [dot] com
 | 
						|
      www.fourwalledcubicle.com
 | 
						|
*/
 | 
						|
 | 
						|
/*
 | 
						|
  Copyright 2009  Denver Gingerich (denver [at] ossguy [dot] com)
 | 
						|
      Based on code by Dean Camera (dean [at] fourwalledcubicle [dot] com)
 | 
						|
 | 
						|
  Permission to use, copy, modify, and distribute this software
 | 
						|
  and its documentation for any purpose and without fee is hereby
 | 
						|
  granted, provided that the above copyright notice appear in all
 | 
						|
  copies and that both that the copyright notice and this
 | 
						|
  permission notice and warranty disclaimer appear in supporting
 | 
						|
  documentation, and that the name of the author not be used in
 | 
						|
  advertising or publicity pertaining to distribution of the
 | 
						|
  software without specific, written prior permission.
 | 
						|
 | 
						|
  The author disclaim all warranties with regard to this
 | 
						|
  software, including all implied warranties of merchantability
 | 
						|
  and fitness.  In no event shall the author be liable for any
 | 
						|
  special, indirect or consequential damages or any damages
 | 
						|
  whatsoever resulting from loss of use, data or profits, whether
 | 
						|
  in an action of contract, negligence or other tortious action,
 | 
						|
  arising out of or in connection with the use or performance of
 | 
						|
  this software.
 | 
						|
*/
 | 
						|
 | 
						|
/** \file
 | 
						|
 *
 | 
						|
 *  Main source file for the Keyboard demo. This file contains the main tasks of the demo and
 | 
						|
 *  is responsible for the initial application hardware configuration.
 | 
						|
 */
 | 
						|
 
 | 
						|
#include "Keyboard.h"
 | 
						|
 | 
						|
/* Global Variables */
 | 
						|
/** Indicates what report mode the host has requested, true for normal HID reporting mode, false for special boot
 | 
						|
 *  protocol reporting mode.
 | 
						|
 */
 | 
						|
bool UsingReportProtocol = true;
 | 
						|
 | 
						|
/** Current Idle period. This is set by the host via a Set Idle HID class request to silence the device's reports
 | 
						|
 *  for either the entire idle duration, or until the report status changes (e.g. the user presses a key).
 | 
						|
 */
 | 
						|
uint16_t IdleCount = 500;
 | 
						|
 | 
						|
/** Current Idle period remaining. When the IdleCount value is set, this tracks the remaining number of idle
 | 
						|
 *  milliseconds. This is separate to the IdleCount timer and is incremented and compared as the host may request 
 | 
						|
 *  the current idle period via a Get Idle HID class request, thus its value must be preserved.
 | 
						|
 */
 | 
						|
uint16_t IdleMSRemaining = 0;
 | 
						|
 | 
						|
 | 
						|
/** Main program entry point. This routine configures the hardware required by the application, then
 | 
						|
 *  starts the scheduler to run the USB management task.
 | 
						|
 */
 | 
						|
int main(void)
 | 
						|
{
 | 
						|
	SetupHardware();
 | 
						|
	
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 | 
						|
 | 
						|
	for (;;)
 | 
						|
	{
 | 
						|
		HID_Task();
 | 
						|
		USB_USBTask();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Configures the board hardware and chip peripherals for the demo's functionality. */
 | 
						|
void SetupHardware(void)
 | 
						|
{
 | 
						|
	/* Disable watchdog if enabled by bootloader/fuses */
 | 
						|
	MCUSR &= ~(1 << WDRF);
 | 
						|
	wdt_disable();
 | 
						|
 | 
						|
	/* Disable clock division */
 | 
						|
	clock_prescale_set(clock_div_1);
 | 
						|
 | 
						|
	/* Hardware Initialization */
 | 
						|
	Joystick_Init();
 | 
						|
	LEDs_Init();
 | 
						|
	USB_Init();
 | 
						|
	
 | 
						|
	/* Millisecond timer initialization, with output compare interrupt enabled for the idle timing */
 | 
						|
	OCR0A  = 0x7D;
 | 
						|
	TCCR0A = (1 << WGM01);
 | 
						|
	TCCR0B = ((1 << CS01) | (1 << CS00));
 | 
						|
	TIMSK0 = (1 << OCIE0A);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_Connect event. This indicates that the device is enumerating via the status LEDs and
 | 
						|
 *  starts the library USB task to begin the enumeration and USB management process.
 | 
						|
 */
 | 
						|
void EVENT_USB_Connect(void)
 | 
						|
{
 | 
						|
	/* Indicate USB enumerating */
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 | 
						|
 | 
						|
	/* Default to report protocol on connect */
 | 
						|
	UsingReportProtocol = true;
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_Disconnect event. This indicates that the device is no longer connected to a host via
 | 
						|
 *  the status LEDs.
 | 
						|
 */
 | 
						|
void EVENT_USB_Disconnect(void)
 | 
						|
{
 | 
						|
	/* Indicate USB not ready */
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_ConfigurationChanged event. This is fired when the host sets the current configuration
 | 
						|
 *  of the USB device after enumeration, and configures the keyboard device endpoints.
 | 
						|
 */
 | 
						|
void EVENT_USB_ConfigurationChanged(void)
 | 
						|
{	
 | 
						|
	/* Indicate USB connected and ready */
 | 
						|
	LEDs_SetAllLEDs(LEDMASK_USB_READY);
 | 
						|
 | 
						|
	/* Setup Keyboard Keycode Report Endpoint */
 | 
						|
	if (!(Endpoint_ConfigureEndpoint(KEYBOARD_EPNUM, EP_TYPE_INTERRUPT,
 | 
						|
		                             ENDPOINT_DIR_IN, KEYBOARD_EPSIZE,
 | 
						|
	                                 ENDPOINT_BANK_SINGLE)))
 | 
						|
	{
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* Setup Keyboard LED Report Endpoint */
 | 
						|
	if (!(Endpoint_ConfigureEndpoint(KEYBOARD_LEDS_EPNUM, EP_TYPE_INTERRUPT,
 | 
						|
		                             ENDPOINT_DIR_OUT, KEYBOARD_EPSIZE,
 | 
						|
	                                 ENDPOINT_BANK_SINGLE)))
 | 
						|
	{
 | 
						|
		LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Event handler for the USB_UnhandledControlPacket event. This is used to catch standard and class specific
 | 
						|
 *  control requests that are not handled internally by the USB library (including the HID commands, which are
 | 
						|
 *  all issued via the control endpoint), so that they can be handled appropriately for the application.
 | 
						|
 */
 | 
						|
void EVENT_USB_UnhandledControlPacket(void)
 | 
						|
{
 | 
						|
	/* Handle HID Class specific requests */
 | 
						|
	switch (USB_ControlRequest.bRequest)
 | 
						|
	{
 | 
						|
		case REQ_GetReport:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{
 | 
						|
				USB_KeyboardReport_Data_t KeyboardReportData;
 | 
						|
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
	
 | 
						|
				/* Create the next keyboard report for transmission to the host */
 | 
						|
				CreateKeyboardReport(&KeyboardReportData);
 | 
						|
 | 
						|
				/* Write the report data to the control endpoint */
 | 
						|
				Endpoint_Write_Control_Stream_LE(&KeyboardReportData, sizeof(KeyboardReportData));
 | 
						|
				
 | 
						|
				/* Finalize the stream transfer to send the last packet or clear the host abort */
 | 
						|
				Endpoint_ClearOUT();
 | 
						|
			}
 | 
						|
		
 | 
						|
			break;
 | 
						|
		case REQ_SetReport:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
				
 | 
						|
				/* Wait until the LED report has been sent by the host */
 | 
						|
				while (!(Endpoint_IsOUTReceived()));
 | 
						|
 | 
						|
				/* Read in the LED report from the host */
 | 
						|
				uint8_t LEDStatus = Endpoint_Read_Byte();
 | 
						|
 | 
						|
				/* Process the incoming LED report */
 | 
						|
				ProcessLEDReport(LEDStatus);
 | 
						|
			
 | 
						|
				/* Clear the endpoint data */
 | 
						|
				Endpoint_ClearOUT();
 | 
						|
 | 
						|
				/* Acknowledge status stage */
 | 
						|
				while (!(Endpoint_IsINReady()));
 | 
						|
				Endpoint_ClearIN();
 | 
						|
			}
 | 
						|
			
 | 
						|
			break;
 | 
						|
		case REQ_GetProtocol:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
				
 | 
						|
				/* Write the current protocol flag to the host */
 | 
						|
				Endpoint_Write_Byte(UsingReportProtocol);
 | 
						|
				
 | 
						|
				/* Send the flag to the host */
 | 
						|
				Endpoint_ClearIN();
 | 
						|
 | 
						|
				/* Acknowledge status stage */
 | 
						|
				while (!(Endpoint_IsOUTReceived()));
 | 
						|
				Endpoint_ClearOUT();
 | 
						|
			}
 | 
						|
			
 | 
						|
			break;
 | 
						|
		case REQ_SetProtocol:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
 | 
						|
				/* Set or clear the flag depending on what the host indicates that the current Protocol should be */
 | 
						|
				UsingReportProtocol = (USB_ControlRequest.wValue != 0);
 | 
						|
 | 
						|
				/* Acknowledge status stage */
 | 
						|
				while (!(Endpoint_IsINReady()));
 | 
						|
				Endpoint_ClearIN();
 | 
						|
			}
 | 
						|
			
 | 
						|
			break;
 | 
						|
		case REQ_SetIdle:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
				
 | 
						|
				/* Get idle period in MSB */
 | 
						|
				IdleCount = (USB_ControlRequest.wValue >> 8);
 | 
						|
				
 | 
						|
				/* Acknowledge status stage */
 | 
						|
				while (!(Endpoint_IsINReady()));
 | 
						|
				Endpoint_ClearIN();
 | 
						|
			}
 | 
						|
			
 | 
						|
			break;
 | 
						|
		case REQ_GetIdle:
 | 
						|
			if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
 | 
						|
			{		
 | 
						|
				Endpoint_ClearSETUP();
 | 
						|
				
 | 
						|
				/* Write the current idle duration to the host */
 | 
						|
				Endpoint_Write_Byte(IdleCount);
 | 
						|
				
 | 
						|
				/* Send the flag to the host */
 | 
						|
				Endpoint_ClearIN();
 | 
						|
 | 
						|
				/* Acknowledge status stage */
 | 
						|
				while (!(Endpoint_IsOUTReceived()));
 | 
						|
				Endpoint_ClearOUT();
 | 
						|
			}
 | 
						|
 | 
						|
			break;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** ISR for the timer 0 compare vector. This ISR fires once each millisecond, and increments the
 | 
						|
 *  scheduler elapsed idle period counter when the host has set an idle period.
 | 
						|
 */
 | 
						|
ISR(TIMER0_COMPA_vect, ISR_BLOCK)
 | 
						|
{
 | 
						|
	/* One millisecond has elapsed, decrement the idle time remaining counter if it has not already elapsed */
 | 
						|
	if (IdleMSRemaining)
 | 
						|
	  IdleMSRemaining--;
 | 
						|
}
 | 
						|
 | 
						|
/** Fills the given HID report data structure with the next HID report to send to the host.
 | 
						|
 *
 | 
						|
 *  \param ReportData  Pointer to a HID report data structure to be filled
 | 
						|
 */
 | 
						|
void CreateKeyboardReport(USB_KeyboardReport_Data_t* ReportData)
 | 
						|
{
 | 
						|
	uint8_t JoyStatus_LCL = Joystick_GetStatus();
 | 
						|
 | 
						|
	/* Clear the report contents */
 | 
						|
	memset(ReportData, 0, sizeof(USB_KeyboardReport_Data_t));
 | 
						|
 | 
						|
	if (JoyStatus_LCL & JOY_UP)
 | 
						|
	  ReportData->KeyCode[0] = 0x04; // A
 | 
						|
	else if (JoyStatus_LCL & JOY_DOWN)
 | 
						|
	  ReportData->KeyCode[0] = 0x05; // B
 | 
						|
 | 
						|
	if (JoyStatus_LCL & JOY_LEFT)
 | 
						|
	  ReportData->KeyCode[0] = 0x06; // C
 | 
						|
	else if (JoyStatus_LCL & JOY_RIGHT)
 | 
						|
	  ReportData->KeyCode[0] = 0x07; // D
 | 
						|
 | 
						|
	if (JoyStatus_LCL & JOY_PRESS)
 | 
						|
	  ReportData->KeyCode[0] = 0x08; // E
 | 
						|
}
 | 
						|
 | 
						|
/** Processes a received LED report, and updates the board LEDs states to match.
 | 
						|
 *
 | 
						|
 *  \param LEDReport  LED status report from the host
 | 
						|
 */
 | 
						|
void ProcessLEDReport(uint8_t LEDReport)
 | 
						|
{
 | 
						|
	uint8_t LEDMask = LEDS_LED2;
 | 
						|
	
 | 
						|
	if (LEDReport & 0x01) // NUM Lock
 | 
						|
	  LEDMask |= LEDS_LED1;
 | 
						|
	
 | 
						|
	if (LEDReport & 0x02) // CAPS Lock
 | 
						|
	  LEDMask |= LEDS_LED3;
 | 
						|
 | 
						|
	if (LEDReport & 0x04) // SCROLL Lock
 | 
						|
	  LEDMask |= LEDS_LED4;
 | 
						|
 | 
						|
	/* Set the status LEDs to the current Keyboard LED status */
 | 
						|
	LEDs_SetAllLEDs(LEDMask);
 | 
						|
}
 | 
						|
 | 
						|
/** Sends the next HID report to the host, via the keyboard data endpoint. */
 | 
						|
void SendNextReport(void)
 | 
						|
{
 | 
						|
	static USB_KeyboardReport_Data_t PrevKeyboardReportData;
 | 
						|
	USB_KeyboardReport_Data_t        KeyboardReportData;
 | 
						|
	bool                             SendReport = true;
 | 
						|
	
 | 
						|
	/* Create the next keyboard report for transmission to the host */
 | 
						|
	CreateKeyboardReport(&KeyboardReportData);
 | 
						|
	
 | 
						|
	/* Check to see if the report data has changed - if so a report MUST be sent */
 | 
						|
	SendReport = (memcmp(&PrevKeyboardReportData, &KeyboardReportData, sizeof(USB_KeyboardReport_Data_t)) != 0);
 | 
						|
	
 | 
						|
	/* Save the current report data for later comparison to check for changes */
 | 
						|
	PrevKeyboardReportData = KeyboardReportData;
 | 
						|
	
 | 
						|
	/* Check if the idle period is set and has elapsed */
 | 
						|
	if ((IdleCount != HID_IDLE_CHANGESONLY) && (!(IdleMSRemaining)))
 | 
						|
	{
 | 
						|
		/* Reset the idle time remaining counter, must multiply by 4 to get the duration in milliseconds */
 | 
						|
		IdleMSRemaining = (IdleCount << 2);
 | 
						|
		
 | 
						|
		/* Idle period is set and has elapsed, must send a report to the host */
 | 
						|
		SendReport = true;
 | 
						|
	}
 | 
						|
	
 | 
						|
	/* Select the Keyboard Report Endpoint */
 | 
						|
	Endpoint_SelectEndpoint(KEYBOARD_EPNUM);
 | 
						|
 | 
						|
	/* Check if Keyboard Endpoint Ready for Read/Write and if we should send a new report */
 | 
						|
	if (Endpoint_IsReadWriteAllowed() && SendReport)
 | 
						|
	{
 | 
						|
		/* Write Keyboard Report Data */
 | 
						|
		Endpoint_Write_Stream_LE(&KeyboardReportData, sizeof(KeyboardReportData));
 | 
						|
		
 | 
						|
		/* Finalize the stream transfer to send the last packet */
 | 
						|
		Endpoint_ClearIN();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Reads the next LED status report from the host from the LED data endpoint, if one has been sent. */
 | 
						|
void ReceiveNextReport(void)
 | 
						|
{
 | 
						|
	/* Select the Keyboard LED Report Endpoint */
 | 
						|
	Endpoint_SelectEndpoint(KEYBOARD_LEDS_EPNUM);
 | 
						|
 | 
						|
	/* Check if Keyboard LED Endpoint contains a packet */
 | 
						|
	if (Endpoint_IsOUTReceived())
 | 
						|
	{
 | 
						|
		/* Check to see if the packet contains data */
 | 
						|
		if (Endpoint_IsReadWriteAllowed())
 | 
						|
		{
 | 
						|
			/* Read in the LED report from the host */
 | 
						|
			uint8_t LEDReport = Endpoint_Read_Byte();
 | 
						|
 | 
						|
			/* Process the read LED report from the host */
 | 
						|
			ProcessLEDReport(LEDReport);
 | 
						|
		}
 | 
						|
 | 
						|
		/* Handshake the OUT Endpoint - clear endpoint and ready for next report */
 | 
						|
		Endpoint_ClearOUT();
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Function to manage HID report generation and transmission to the host, when in report mode. */
 | 
						|
void HID_Task(void)
 | 
						|
{
 | 
						|
	/* Check if the USB system is connected to a host */
 | 
						|
	if (USB_IsConnected)
 | 
						|
	{
 | 
						|
		/* Send the next keypress report to the host */
 | 
						|
		SendNextReport();
 | 
						|
		
 | 
						|
		/* Process the LED report sent from the host */
 | 
						|
		ReceiveNextReport();
 | 
						|
	}
 | 
						|
}
 |