Demos/Host/ClassDriver/PrinterHost/PrinterHost.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2011.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6            www.lufa-lib.org
   7 */
   8
   9 /*
  10   Copyright 2011  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, distribute, and sell this
  13   software and its documentation for any purpose is hereby granted
  14   without fee, provided that the above copyright notice appear in
  15   all copies and that both that the copyright notice and this
  16   permission notice and warranty disclaimer appear in supporting
  17   documentation, and that the name of the author not be used in
  18   advertising or publicity pertaining to distribution of the
  19   software without specific, written prior permission.
  20
  21   The author disclaim all warranties with regard to this
  22   software, including all implied warranties of merchantability
  23   and fitness.  In no event shall the author be liable for any
  24   special, indirect or consequential damages or any damages
  25   whatsoever resulting from loss of use, data or profits, whether
  26   in an action of contract, negligence or other tortious action,
  27   arising out of or in connection with the use or performance of
  28   this software.
  29 */
  30
  31 /** \file
  32  *
  33  *  Main source file for the PrinterHost demo. This file contains the main tasks of
  34  *  the demo and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "PrinterHost.h"
  38
  39 /** LUFA Printer Class driver interface configuration and state information. This structure is
  40  *  passed to all Printer Class driver functions, so that multiple instances of the same class
  41  *  within a device can be differentiated from one another.
  42  */
  43 USB_ClassInfo_PRNT_Host_t Printer_PRNT_Interface =
  44         {
  45                 .Config =
  46                         {
  47                                 .DataINPipeNumber       = 1,
  48                                 .DataINPipeDoubleBank   = false,
  49
  50                                 .DataOUTPipeNumber      = 2,
  51                                 .DataOUTPipeDoubleBank  = false,
  52                         },
  53         };
  54
  55
  56 /** Main program entry point. This routine configures the hardware required by the application, then
  57  *  enters a loop to run the application tasks in sequence.
  58  */
  59 int main(void)
  60 {
  61         SetupHardware();
  62
  63         puts_P(PSTR(ESC_FG_CYAN "Printer Host Demo running.\r\n" ESC_FG_WHITE));
  64
  65         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  66         sei();
  67
  68         for (;;)
  69         {
  70                 switch (USB_HostState)
  71                 {
  72                         case HOST_STATE_Addressed:
  73                                 LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
  74
  75                                 uint16_t ConfigDescriptorSize;
  76                                 uint8_t  ConfigDescriptorData[512];
  77
  78                                 if (USB_Host_GetDeviceConfigDescriptor(1, &ConfigDescriptorSize, ConfigDescriptorData,
  79                                                                        sizeof(ConfigDescriptorData)) != HOST_GETCONFIG_Successful)
  80                                 {
  81                                         puts_P(PSTR("Error Retrieving Configuration Descriptor.\r\n"));
  82                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
  83                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
  84                                         break;
  85                                 }
  86
  87                                 if (PRNT_Host_ConfigurePipes(&Printer_PRNT_Interface,
  88                                                              ConfigDescriptorSize, ConfigDescriptorData) != PRNT_ENUMERROR_NoError)
  89                                 {
  90                                         puts_P(PSTR("Attached Device Not a Valid Printer Class Device.\r\n"));
  91                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
  92                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
  93                                         break;
  94                                 }
  95
  96                                 if (USB_Host_SetDeviceConfiguration(1) != HOST_SENDCONTROL_Successful)
  97                                 {
  98                                         puts_P(PSTR("Error Setting Device Configuration.\r\n"));
  99                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 100                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 101                                         break;
 102                                 }
 103
 104                                 if (PRNT_Host_SetBidirectionalMode(&Printer_PRNT_Interface) != HOST_SENDCONTROL_Successful)
 105                                 {
 106                                         puts_P(PSTR("Error Setting Bidirectional Mode.\r\n"));
 107                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 108                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 109                                         break;
 110                                 }
 111
 112                                 puts_P(PSTR("Printer Device Enumerated.\r\n"));
 113                                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 114                                 USB_HostState = HOST_STATE_Configured;
 115                                 break;
 116                         case HOST_STATE_Configured:
 117                                 LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 118
 119                                 puts_P(PSTR("Retrieving Device ID...\r\n"));
 120
 121                                 char DeviceIDString[300];
 122                                 if (PRNT_Host_GetDeviceID(&Printer_PRNT_Interface, DeviceIDString,
 123                                                           sizeof(DeviceIDString)) != HOST_SENDCONTROL_Successful)
 124                                 {
 125                                         puts_P(PSTR("Error Getting Device ID.\r\n"));
 126                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 127                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 128                                         break;
 129                                 }
 130
 131                                 printf_P(PSTR("Device ID: %s.\r\n"), DeviceIDString);
 132
 133                                 char     TestPageData[] = "\033%-12345X\033E" "LUFA PCL Test Page" "\033E\033%-12345X";
 134                                 uint16_t TestPageLength = strlen(TestPageData);
 135
 136                                 printf_P(PSTR("Sending Test Page (%d bytes)...\r\n"), TestPageLength);
 137
 138                                 if (PRNT_Host_SendData(&Printer_PRNT_Interface, &TestPageData, TestPageLength) != PIPE_RWSTREAM_NoError)
 139                                 {
 140                                         puts_P(PSTR("Error Sending Page Data.\r\n"));
 141                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 142                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 143                                         break;
 144                                 }
 145
 146                                 puts_P(PSTR("Test Page Sent.\r\n"));
 147
 148                                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 149                                 USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 150                                 break;
 151                 }
 152
 153                 PRNT_Host_USBTask(&Printer_PRNT_Interface);
 154                 USB_USBTask();
 155         }
 156 }
 157
 158 /** Configures the board hardware and chip peripherals for the demo's functionality. */
 159 void SetupHardware(void)
 160 {
 161         /* Disable watchdog if enabled by bootloader/fuses */
 162         MCUSR &= ~(1 << WDRF);
 163         wdt_disable();
 164
 165         /* Disable clock division */
 166         clock_prescale_set(clock_div_1);
 167
 168         /* Hardware Initialization */
 169         Serial_Init(9600, false);
 170         LEDs_Init();
 171         USB_Init();
 172
 173         /* Create a stdio stream for the serial port for stdin and stdout */
 174         Serial_CreateStream(NULL);
 175 }
 176
 177 /** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 178  *  starts the library USB task to begin the enumeration and USB management process.
 179  */
 180 void EVENT_USB_Host_DeviceAttached(void)
 181 {
 182         puts_P(PSTR("Device Attached.\r\n"));
 183         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 184 }
 185
 186 /** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 187  *  stops the library USB task management process.
 188  */
 189 void EVENT_USB_Host_DeviceUnattached(void)
 190 {
 191         puts_P(PSTR("\r\nDevice Unattached.\r\n"));
 192         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 193 }
 194
 195 /** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 196  *  enumerated by the host and is now ready to be used by the application.
 197  */
 198 void EVENT_USB_Host_DeviceEnumerationComplete(void)
 199 {
 200         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 201 }
 202
 203 /** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */
 204 void EVENT_USB_Host_HostError(const uint8_t ErrorCode)
 205 {
 206         USB_Disable();
 207
 208         printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n"
 209                                  " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode);
 210
 211         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 212         for(;;);
 213 }
 214
 215 /** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 216  *  enumerating an attached USB device.
 217  */
 218 void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode,
 219                                             const uint8_t SubErrorCode)
 220 {
 221         printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n"
 222                                  " -- Error Code %d\r\n"
 223                                  " -- Sub Error Code %d\r\n"
 224                                  " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState);
 225
 226         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 227 }
 228