Demos/Device/ClassDriver/KeyboardMouse/KeyboardMouse.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2010.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6       www.fourwalledcubicle.com
   7 */
   8
   9 /*
  10   Copyright 2010  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11   Copyright 2010  Denver Gingerich (denver [at] ossguy [dot] com)
  12
  13   Permission to use, copy, modify, distribute, and sell this
  14   software and its documentation for any purpose is hereby granted
  15   without fee, provided that the above copyright notice appear in
  16   all copies and that both that the copyright notice and this
  17   permission notice and warranty disclaimer appear in supporting
  18   documentation, and that the name of the author not be used in
  19   advertising or publicity pertaining to distribution of the
  20   software without specific, written prior permission.
  21
  22   The author disclaim all warranties with regard to this
  23   software, including all implied warranties of merchantability
  24   and fitness.  In no event shall the author be liable for any
  25   special, indirect or consequential damages or any damages
  26   whatsoever resulting from loss of use, data or profits, whether
  27   in an action of contract, negligence or other tortious action,
  28   arising out of or in connection with the use or performance of
  29   this software.
  30 */
  31
  32 /** \file
  33  *
  34  *  Main source file for the KeyboardMouse demo. This file contains the main tasks of
  35  *  the demo and is responsible for the initial application hardware configuration.
  36  */
  37
  38 #include "KeyboardMouse.h"
  39
  40 /** Buffer to hold the previously generated Keyboard HID report, for comparison purposes inside the HID class driver. */
  41 uint8_t PrevKeyboardHIDReportBuffer[sizeof(USB_KeyboardReport_Data_t)];
  42
  43 /** Buffer to hold the previously generated Mouse HID report, for comparison purposes inside the HID class driver. */
  44 uint8_t PrevMouseHIDReportBuffer[sizeof(USB_MouseReport_Data_t)];
  45
  46 /** LUFA HID Class driver interface configuration and state information. This structure is
  47  *  passed to all HID Class driver functions, so that multiple instances of the same class
  48  *  within a device can be differentiated from one another. This is for the keyboard HID
  49  *  interface within the device.
  50  */
  51 USB_ClassInfo_HID_Device_t Keyboard_HID_Interface =
  52         {
  53                 .Config =
  54                         {
  55                                 .InterfaceNumber              = 0,
  56
  57                                 .ReportINEndpointNumber       = KEYBOARD_IN_EPNUM,
  58                                 .ReportINEndpointSize         = HID_EPSIZE,
  59                                 .ReportINEndpointDoubleBank   = false,
  60
  61                                 .PrevReportINBuffer           = PrevKeyboardHIDReportBuffer,
  62                                 .PrevReportINBufferSize       = sizeof(PrevKeyboardHIDReportBuffer),
  63                         },
  64         };
  65
  66 /** LUFA HID Class driver interface configuration and state information. This structure is
  67  *  passed to all HID Class driver functions, so that multiple instances of the same class
  68  *  within a device can be differentiated from one another. This is for the mouse HID
  69  *  interface within the device.
  70  */
  71 USB_ClassInfo_HID_Device_t Mouse_HID_Interface =
  72         {
  73                 .Config =
  74                         {
  75                                 .InterfaceNumber              = 1,
  76
  77                                 .ReportINEndpointNumber       = MOUSE_IN_EPNUM,
  78                                 .ReportINEndpointSize         = HID_EPSIZE,
  79
  80                                 .PrevReportINBuffer           = PrevMouseHIDReportBuffer,
  81                                 .PrevReportINBufferSize       = sizeof(PrevMouseHIDReportBuffer),
  82                         },
  83         };
  84
  85 /** Main program entry point. This routine contains the overall program flow, including initial
  86  *  setup of all components and the main program loop.
  87  */
  88 int main(void)
  89 {
  90         SetupHardware();
  91
  92         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  93
  94         for (;;)
  95         {
  96                 HID_Device_USBTask(&Keyboard_HID_Interface);
  97                 HID_Device_USBTask(&Mouse_HID_Interface);
  98                 USB_USBTask();
  99         }
 100 }
 101
 102 /** Configures the board hardware and chip peripherals for the demo's functionality. */
 103 void SetupHardware()
 104 {
 105         /* Disable watchdog if enabled by bootloader/fuses */
 106         MCUSR &= ~(1 << WDRF);
 107         wdt_disable();
 108
 109         /* Disable clock division */
 110         clock_prescale_set(clock_div_1);
 111
 112         /* Hardware Initialization */
 113         Joystick_Init();
 114         LEDs_Init();
 115         USB_Init();
 116 }
 117
 118 /** Event handler for the library USB Connection event. */
 119 void EVENT_USB_Device_Connect(void)
 120 {
 121     LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 122 }
 123
 124 /** Event handler for the library USB Disconnection event. */
 125 void EVENT_USB_Device_Disconnect(void)
 126 {
 127     LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 128 }
 129
 130 /** Event handler for the library USB Configuration Changed event. */
 131 void EVENT_USB_Device_ConfigurationChanged(void)
 132 {
 133         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 134
 135         if (!(HID_Device_ConfigureEndpoints(&Keyboard_HID_Interface)))
 136           LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 137
 138         if (!(HID_Device_ConfigureEndpoints(&Mouse_HID_Interface)))
 139           LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 140
 141         USB_Device_EnableSOFEvents();
 142 }
 143
 144 /** Event handler for the library USB Unhandled Control Request event. */
 145 void EVENT_USB_Device_UnhandledControlRequest(void)
 146 {
 147         HID_Device_ProcessControlRequest(&Keyboard_HID_Interface);
 148         HID_Device_ProcessControlRequest(&Mouse_HID_Interface);
 149 }
 150
 151 /** Event handler for the USB device Start Of Frame event. */
 152 void EVENT_USB_Device_StartOfFrame(void)
 153 {
 154         HID_Device_MillisecondElapsed(&Keyboard_HID_Interface);
 155         HID_Device_MillisecondElapsed(&Mouse_HID_Interface);
 156 }
 157
 158 /** HID class driver callback function for the creation of HID reports to the host.
 159  *
 160  *  \param[in] HIDInterfaceInfo  Pointer to the HID class interface configuration structure being referenced
 161  *  \param[in,out] ReportID  Report ID requested by the host if non-zero, otherwise callback should set to the generated report ID
 162  *  \param[in] ReportType  Type of the report to create, either REPORT_ITEM_TYPE_In or REPORT_ITEM_TYPE_Feature
 163  *  \param[out] ReportData  Pointer to a buffer where the created report should be stored
 164  *  \param[out] ReportSize  Number of bytes written in the report (or zero if no report is to be sent
 165  *
 166  *  \return Boolean true to force the sending of the report, false to let the library determine if it needs to be sent
 167  */
 168 bool CALLBACK_HID_Device_CreateHIDReport(USB_ClassInfo_HID_Device_t* const HIDInterfaceInfo, uint8_t* const ReportID,
 169                                          const uint8_t ReportType, void* ReportData, uint16_t* ReportSize)
 170 {
 171         uint8_t JoyStatus_LCL    = Joystick_GetStatus();
 172         uint8_t ButtonStatus_LCL = Buttons_GetStatus();
 173
 174         /* Determine which interface must have its report generated */
 175         if (HIDInterfaceInfo == &Keyboard_HID_Interface)
 176         {
 177                 USB_KeyboardReport_Data_t* KeyboardReport = (USB_KeyboardReport_Data_t*)ReportData;
 178
 179                 /* If first board button not being held down, no keyboard report */
 180                 if (!(ButtonStatus_LCL & BUTTONS_BUTTON1))
 181                   return 0;
 182
 183                 if (JoyStatus_LCL & JOY_UP)
 184                   KeyboardReport->KeyCode[0] = 0x04; // A
 185                 else if (JoyStatus_LCL & JOY_DOWN)
 186                   KeyboardReport->KeyCode[0] = 0x05; // B
 187
 188                 if (JoyStatus_LCL & JOY_LEFT)
 189                   KeyboardReport->KeyCode[0] = 0x06; // C
 190                 else if (JoyStatus_LCL & JOY_RIGHT)
 191                   KeyboardReport->KeyCode[0] = 0x07; // D
 192
 193                 if (JoyStatus_LCL & JOY_PRESS)
 194                   KeyboardReport->KeyCode[0] = 0x08; // E
 195
 196                 *ReportSize = sizeof(USB_KeyboardReport_Data_t);
 197                 return false;
 198         }
 199         else
 200         {
 201                 USB_MouseReport_Data_t* MouseReport = (USB_MouseReport_Data_t*)ReportData;
 202
 203                 /* If first board button being held down, no mouse report */
 204                 if (ButtonStatus_LCL & BUTTONS_BUTTON1)
 205                   return 0;
 206
 207                 if (JoyStatus_LCL & JOY_UP)
 208                   MouseReport->Y = -1;
 209                 else if (JoyStatus_LCL & JOY_DOWN)
 210                   MouseReport->Y =  1;
 211
 212                 if (JoyStatus_LCL & JOY_LEFT)
 213                   MouseReport->X = -1;
 214                 else if (JoyStatus_LCL & JOY_RIGHT)
 215                   MouseReport->X =  1;
 216
 217                 if (JoyStatus_LCL & JOY_PRESS)
 218                   MouseReport->Button  = (1 << 0);
 219
 220                 *ReportSize = sizeof(USB_MouseReport_Data_t);
 221                 return true;
 222         }
 223 }
 224
 225 /** HID class driver callback function for the processing of HID reports from the host.
 226  *
 227  *  \param[in] HIDInterfaceInfo  Pointer to the HID class interface configuration structure being referenced
 228  *  \param[in] ReportID  Report ID of the received report from the host
 229  *  \param[in] ReportData  Pointer to a buffer where the created report has been stored
 230  *  \param[in] ReportSize  Size in bytes of the received HID report
 231  */
 232 void CALLBACK_HID_Device_ProcessHIDReport(USB_ClassInfo_HID_Device_t* const HIDInterfaceInfo, const uint8_t ReportID,
 233                                           const void* ReportData, const uint16_t ReportSize)
 234 {
 235         if (HIDInterfaceInfo == &Keyboard_HID_Interface)
 236         {
 237                 uint8_t  LEDMask   = LEDS_NO_LEDS;
 238                 uint8_t* LEDReport = (uint8_t*)ReportData;
 239
 240                 if (*LEDReport & 0x01) // NUM Lock
 241                   LEDMask |= LEDS_LED1;
 242
 243                 if (*LEDReport & 0x02) // CAPS Lock
 244                   LEDMask |= LEDS_LED3;
 245
 246                 if (*LEDReport & 0x04) // SCROLL Lock
 247                   LEDMask |= LEDS_LED4;
 248
 249                 LEDs_SetAllLEDs(LEDMask);
 250         }
 251 }