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.
		
		
		
		
		
			
		
			
				
					290 lines
				
				9.3 KiB
			
		
		
			
		
	
	
					290 lines
				
				9.3 KiB
			|   
											8 years ago
										 | /*
 | ||
|  |              LUFA Library | ||
|  |      Copyright (C) Dean Camera, 2017. | ||
|  | 
 | ||
|  |   dean [at] fourwalledcubicle [dot] com | ||
|  |            www.lufa-lib.org | ||
|  | */ | ||
|  | 
 | ||
|  | /*
 | ||
|  |   Copyright 2017  Dean Camera (dean [at] fourwalledcubicle [dot] com) | ||
|  | 
 | ||
|  |   Permission to use, copy, modify, distribute, and sell this | ||
|  |   software and its documentation for any purpose is hereby granted | ||
|  |   without fee, 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 disclaims 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 MouseHostWithParser demo. This file contains the main tasks of | ||
|  |  *  the demo and is responsible for the initial application hardware configuration. | ||
|  |  */ | ||
|  | 
 | ||
|  | #include "MouseHostWithParser.h"
 | ||
|  | 
 | ||
|  | /** Main program entry point. This routine configures the hardware required by the application, then
 | ||
|  |  *  enters a loop to run the application tasks in sequence. | ||
|  |  */ | ||
|  | int main(void) | ||
|  | { | ||
|  | 	SetupHardware(); | ||
|  | 
 | ||
|  | 	puts_P(PSTR(ESC_FG_CYAN "Mouse HID Parser Host Demo running.\r\n" ESC_FG_WHITE)); | ||
|  | 
 | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY); | ||
|  | 	GlobalInterruptEnable(); | ||
|  | 
 | ||
|  | 	for (;;) | ||
|  | 	{ | ||
|  | 		MouseHost_Task(); | ||
|  | 
 | ||
|  | 		USB_USBTask(); | ||
|  | 	} | ||
|  | } | ||
|  | 
 | ||
|  | /** Configures the board hardware and chip peripherals for the demo's functionality. */ | ||
|  | void SetupHardware(void) | ||
|  | { | ||
|  | #if (ARCH == ARCH_AVR8)
 | ||
|  | 	/* Disable watchdog if enabled by bootloader/fuses */ | ||
|  | 	MCUSR &= ~(1 << WDRF); | ||
|  | 	wdt_disable(); | ||
|  | 
 | ||
|  | 	/* Disable clock division */ | ||
|  | 	clock_prescale_set(clock_div_1); | ||
|  | #endif
 | ||
|  | 
 | ||
|  | 	/* Hardware Initialization */ | ||
|  | 	Serial_Init(9600, false); | ||
|  | 	LEDs_Init(); | ||
|  | 	USB_Init(); | ||
|  | 
 | ||
|  | 	/* Create a stdio stream for the serial port for stdin and stdout */ | ||
|  | 	Serial_CreateStream(NULL); | ||
|  | } | ||
|  | 
 | ||
|  | /** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 | ||
|  |  *  starts the library USB task to begin the enumeration and USB management process. | ||
|  |  */ | ||
|  | void EVENT_USB_Host_DeviceAttached(void) | ||
|  | { | ||
|  | 	puts_P(PSTR(ESC_FG_GREEN "Device Attached.\r\n" ESC_FG_WHITE)); | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING); | ||
|  | } | ||
|  | 
 | ||
|  | /** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 | ||
|  |  *  stops the library USB task management process. | ||
|  |  */ | ||
|  | void EVENT_USB_Host_DeviceUnattached(void) | ||
|  | { | ||
|  | 	puts_P(PSTR(ESC_FG_GREEN "Device Unattached.\r\n" ESC_FG_WHITE)); | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY); | ||
|  | } | ||
|  | 
 | ||
|  | /** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 | ||
|  |  *  enumerated by the host and is now ready to be used by the application. | ||
|  |  */ | ||
|  | void EVENT_USB_Host_DeviceEnumerationComplete(void) | ||
|  | { | ||
|  | 	puts_P(PSTR("Getting Config Data.\r\n")); | ||
|  | 
 | ||
|  | 	uint8_t ErrorCode; | ||
|  | 
 | ||
|  | 	/* Get and process the configuration descriptor data */ | ||
|  | 	if ((ErrorCode = ProcessConfigurationDescriptor()) != SuccessfulConfigRead) | ||
|  | 	{ | ||
|  | 		if (ErrorCode == ControlError) | ||
|  | 		  puts_P(PSTR(ESC_FG_RED "Control Error (Get Configuration).\r\n")); | ||
|  | 		else | ||
|  | 		  puts_P(PSTR(ESC_FG_RED "Invalid Device.\r\n")); | ||
|  | 
 | ||
|  | 		printf_P(PSTR(" -- Error Code: %d\r\n" ESC_FG_WHITE), ErrorCode); | ||
|  | 
 | ||
|  | 		LEDs_SetAllLEDs(LEDMASK_USB_ERROR); | ||
|  | 		return; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/* Set the device configuration to the first configuration (rarely do devices use multiple configurations) */ | ||
|  | 	if ((ErrorCode = USB_Host_SetDeviceConfiguration(1)) != HOST_SENDCONTROL_Successful) | ||
|  | 	{ | ||
|  | 		printf_P(PSTR(ESC_FG_RED "Control Error (Set Configuration).\r\n" | ||
|  | 		                         " -- Error Code: %d\r\n" ESC_FG_WHITE), ErrorCode); | ||
|  | 
 | ||
|  | 		LEDs_SetAllLEDs(LEDMASK_USB_ERROR); | ||
|  | 		return; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	printf_P(PSTR("Processing HID Report (Size %d Bytes).\r\n"), HIDReportSize); | ||
|  | 
 | ||
|  | 	/* Get and process the device's first HID report descriptor */ | ||
|  | 	if ((ErrorCode = GetHIDReportData()) != ParseSuccessful) | ||
|  | 	{ | ||
|  | 		puts_P(PSTR(ESC_FG_RED "Report Parse Error.\r\n")); | ||
|  | 
 | ||
|  | 		if (!(HIDReportInfo.TotalReportItems)) | ||
|  | 			puts_P(PSTR("Not a valid Mouse." ESC_FG_WHITE)); | ||
|  | 		else | ||
|  | 			printf_P(PSTR(" -- Error Code: %d\r\n" ESC_FG_WHITE), ErrorCode); | ||
|  | 
 | ||
|  | 		LEDs_SetAllLEDs(LEDMASK_USB_ERROR); | ||
|  | 		USB_Host_SetDeviceConfiguration(0); | ||
|  | 		return; | ||
|  | 	} | ||
|  | 
 | ||
|  | 	puts_P(PSTR("Mouse Enumerated.\r\n")); | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_READY); | ||
|  | } | ||
|  | 
 | ||
|  | /** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */ | ||
|  | void EVENT_USB_Host_HostError(const uint8_t ErrorCode) | ||
|  | { | ||
|  | 	USB_Disable(); | ||
|  | 
 | ||
|  | 	printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n" | ||
|  | 	                         " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode); | ||
|  | 
 | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_ERROR); | ||
|  | 	for(;;); | ||
|  | } | ||
|  | 
 | ||
|  | /** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 | ||
|  |  *  enumerating an attached USB device. | ||
|  |  */ | ||
|  | void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode, | ||
|  |                                             const uint8_t SubErrorCode) | ||
|  | { | ||
|  | 	printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n" | ||
|  | 	                         " -- Error Code %d\r\n" | ||
|  | 	                         " -- Sub Error Code %d\r\n" | ||
|  | 	                         " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState); | ||
|  | 
 | ||
|  | 	LEDs_SetAllLEDs(LEDMASK_USB_ERROR); | ||
|  | } | ||
|  | 
 | ||
|  | /** Task to read and process the HID report descriptor and HID reports from the device and display the
 | ||
|  |  *  results onto the board LEDs. | ||
|  |  */ | ||
|  | void MouseHost_Task(void) | ||
|  | { | ||
|  | 	if (USB_HostState != HOST_STATE_Configured) | ||
|  | 	  return; | ||
|  | 
 | ||
|  | 	/* Select and unfreeze mouse data pipe */ | ||
|  | 	Pipe_SelectPipe(MOUSE_DATA_IN_PIPE); | ||
|  | 	Pipe_Unfreeze(); | ||
|  | 
 | ||
|  | 	/* Check to see if a packet has been received */ | ||
|  | 	if (Pipe_IsINReceived()) | ||
|  | 	{ | ||
|  | 		/* Check if data has been received from the attached mouse */ | ||
|  | 		if (Pipe_IsReadWriteAllowed()) | ||
|  | 		{ | ||
|  | 			/* Create buffer big enough for the report */ | ||
|  | 			uint8_t MouseReport[Pipe_BytesInPipe()]; | ||
|  | 
 | ||
|  | 			/* Load in the mouse report */ | ||
|  | 			Pipe_Read_Stream_LE(MouseReport, Pipe_BytesInPipe(), NULL); | ||
|  | 
 | ||
|  | 			/* Process the read in mouse report from the device */ | ||
|  | 			ProcessMouseReport(MouseReport); | ||
|  | 		} | ||
|  | 
 | ||
|  | 		/* Clear the IN endpoint, ready for next data packet */ | ||
|  | 		Pipe_ClearIN(); | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/* Freeze mouse data pipe */ | ||
|  | 	Pipe_Freeze(); | ||
|  | } | ||
|  | 
 | ||
|  | /** Processes a read HID report from an attached mouse, extracting out elements via the HID parser results
 | ||
|  |  *  as required and displays movement and button presses on the board LEDs. | ||
|  |  * | ||
|  |  *  \param[in] MouseReport  Pointer to a HID report from an attached mouse device | ||
|  |  */ | ||
|  | void ProcessMouseReport(uint8_t* MouseReport) | ||
|  | { | ||
|  | 	uint8_t LEDMask = LEDS_NO_LEDS; | ||
|  | 
 | ||
|  | 	/* Check each HID report item in turn, looking for mouse X/Y/button reports */ | ||
|  | 	for (uint8_t ReportNumber = 0; ReportNumber < HIDReportInfo.TotalReportItems; ReportNumber++) | ||
|  | 	{ | ||
|  | 		/* Create a temporary item pointer to the next report item */ | ||
|  | 		HID_ReportItem_t* ReportItem = &HIDReportInfo.ReportItems[ReportNumber]; | ||
|  | 
 | ||
|  | 		bool FoundData; | ||
|  | 
 | ||
|  | 		if ((ReportItem->Attributes.Usage.Page        == USAGE_PAGE_BUTTON) && | ||
|  | 			(ReportItem->ItemType                     == HID_REPORT_ITEM_In)) | ||
|  | 		{ | ||
|  | 			/* Get the mouse button value */ | ||
|  | 			FoundData = USB_GetHIDReportItemInfo(MouseReport, ReportItem); | ||
|  | 
 | ||
|  | 			/* For multi-report devices - if the requested data was not in the issued report, continue */ | ||
|  | 			if (!(FoundData)) | ||
|  | 			  continue; | ||
|  | 
 | ||
|  | 			/* If button is pressed, all LEDs are turned on */ | ||
|  | 			if (ReportItem->Value) | ||
|  | 			  LEDMask = LEDS_ALL_LEDS; | ||
|  | 		} | ||
|  | 		else if ((ReportItem->Attributes.Usage.Page   == USAGE_PAGE_GENERIC_DCTRL) && | ||
|  | 				 (ReportItem->Attributes.Usage.Usage  == USAGE_SCROLL_WHEEL)       && | ||
|  | 				 (ReportItem->ItemType                == HID_REPORT_ITEM_In)) | ||
|  | 		{ | ||
|  | 			/* Get the mouse wheel value if it is contained within the current
 | ||
|  | 			 * report, if not, skip to the next item in the parser list | ||
|  | 			 */ | ||
|  | 			if (!(USB_GetHIDReportItemInfo(MouseReport, ReportItem))) | ||
|  | 			  continue; | ||
|  | 
 | ||
|  | 			int16_t WheelDelta = HID_ALIGN_DATA(ReportItem, int16_t); | ||
|  | 
 | ||
|  | 			if (WheelDelta) | ||
|  | 			  LEDMask = (LEDS_LED1 | LEDS_LED2 | ((WheelDelta > 0) ? LEDS_LED3 : LEDS_LED4)); | ||
|  | 		} | ||
|  | 		else if ((ReportItem->Attributes.Usage.Page   == USAGE_PAGE_GENERIC_DCTRL) && | ||
|  | 				 ((ReportItem->Attributes.Usage.Usage == USAGE_X)                  || | ||
|  | 				  (ReportItem->Attributes.Usage.Usage == USAGE_Y))                 && | ||
|  | 				 (ReportItem->ItemType                == HID_REPORT_ITEM_In)) | ||
|  | 		{ | ||
|  | 			/* Get the mouse relative position value */ | ||
|  | 			FoundData = USB_GetHIDReportItemInfo(MouseReport, ReportItem); | ||
|  | 
 | ||
|  | 			/* For multi-report devices - if the requested data was not in the issued report, continue */ | ||
|  | 			if (!(FoundData)) | ||
|  | 			  continue; | ||
|  | 
 | ||
|  | 			int16_t DeltaMovement = HID_ALIGN_DATA(ReportItem, int16_t); | ||
|  | 
 | ||
|  | 			/* Check to see if a (non-zero) delta movement has been indicated */ | ||
|  | 			if (DeltaMovement) | ||
|  | 			{ | ||
|  | 				/* Determine if the report is for the X or Y delta movement, light LEDs as appropriate */ | ||
|  | 				if (ReportItem->Attributes.Usage.Usage == USAGE_X) | ||
|  | 				  LEDMask |= ((DeltaMovement > 0) ? LEDS_LED1 : LEDS_LED2); | ||
|  | 				else | ||
|  | 				  LEDMask |= ((DeltaMovement > 0) ? LEDS_LED3 : LEDS_LED4); | ||
|  | 			} | ||
|  | 		} | ||
|  | 	} | ||
|  | 
 | ||
|  | 	/* Display the button information on the board LEDs */ | ||
|  | 	LEDs_SetAllLEDs(LEDMask); | ||
|  | } | ||
|  | 
 |