Demos/Host/ClassDriver/PrinterHost/PrinterHost.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2013.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6            www.lufa-lib.org
   7 */
   8
   9 /*
  10   Copyright 2013  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 disclaims 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                                 .DataINPipe             =
  48                                         {
  49                                                 .Address        = (PIPE_DIR_IN  | 1),
  50                                                 .Banks          = 1,
  51                                         },
  52                                 .DataOUTPipe            =
  53                                         {
  54                                                 .Address        = (PIPE_DIR_OUT | 2),
  55                                                 .Banks          = 1,
  56                                         },
  57                         },
  58         };
  59
  60
  61 /** Main program entry point. This routine configures the hardware required by the application, then
  62  *  enters a loop to run the application tasks in sequence.
  63  */
  64 int main(void)
  65 {
  66         SetupHardware();
  67
  68         puts_P(PSTR(ESC_FG_CYAN "Printer Host Demo running.\r\n" ESC_FG_WHITE));
  69
  70         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  71         GlobalInterruptEnable();
  72
  73         for (;;)
  74         {
  75                 PrinterHost_Task();
  76
  77                 PRNT_Host_USBTask(&Printer_PRNT_Interface);
  78                 USB_USBTask();
  79         }
  80 }
  81
  82 /** Configures the board hardware and chip peripherals for the demo's functionality. */
  83 void SetupHardware(void)
  84 {
  85         /* Disable watchdog if enabled by bootloader/fuses */
  86         MCUSR &= ~(1 << WDRF);
  87         wdt_disable();
  88
  89         /* Disable clock division */
  90         clock_prescale_set(clock_div_1);
  91
  92         /* Hardware Initialization */
  93         Serial_Init(9600, false);
  94         LEDs_Init();
  95         USB_Init();
  96
  97         /* Create a stdio stream for the serial port for stdin and stdout */
  98         Serial_CreateStream(NULL);
  99 }
 100
 101 /** Task to manage an enumerated USB printer once connected, to display device
 102  *  information and print a test PCL page.
 103  */
 104 void PrinterHost_Task(void)
 105 {
 106         if (USB_HostState != HOST_STATE_Configured)
 107           return;
 108
 109         LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 110
 111         puts_P(PSTR("Retrieving Device ID...\r\n"));
 112
 113         char DeviceIDString[300];
 114         if (PRNT_Host_GetDeviceID(&Printer_PRNT_Interface, DeviceIDString,
 115                                   sizeof(DeviceIDString)) != HOST_SENDCONTROL_Successful)
 116         {
 117                 puts_P(PSTR("Error Getting Device ID.\r\n"));
 118                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 119                 USB_Host_SetDeviceConfiguration(0);
 120                 return;
 121         }
 122
 123         printf_P(PSTR("Device ID: %s.\r\n"), DeviceIDString);
 124
 125         char     TestPageData[] = "\033%-12345X\033E" "LUFA PCL Test Page" "\033E\033%-12345X";
 126         uint16_t TestPageLength = strlen(TestPageData);
 127
 128         printf_P(PSTR("Sending Test Page (%d bytes)...\r\n"), TestPageLength);
 129
 130         if (PRNT_Host_SendData(&Printer_PRNT_Interface, &TestPageData, TestPageLength) != PIPE_RWSTREAM_NoError)
 131         {
 132                 puts_P(PSTR("Error Sending Page Data.\r\n"));
 133                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 134                 USB_Host_SetDeviceConfiguration(0);
 135                 return;
 136         }
 137
 138         puts_P(PSTR("Test Page Sent.\r\n"));
 139
 140         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 141         USB_Host_SetDeviceConfiguration(0);
 142 }
 143
 144 /** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 145  *  starts the library USB task to begin the enumeration and USB management process.
 146  */
 147 void EVENT_USB_Host_DeviceAttached(void)
 148 {
 149         puts_P(PSTR("Device Attached.\r\n"));
 150         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 151 }
 152
 153 /** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 154  *  stops the library USB task management process.
 155  */
 156 void EVENT_USB_Host_DeviceUnattached(void)
 157 {
 158         puts_P(PSTR("\r\nDevice Unattached.\r\n"));
 159         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 160 }
 161
 162 /** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 163  *  enumerated by the host and is now ready to be used by the application.
 164  */
 165 void EVENT_USB_Host_DeviceEnumerationComplete(void)
 166 {
 167         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 168
 169         uint16_t ConfigDescriptorSize;
 170         uint8_t  ConfigDescriptorData[512];
 171
 172         if (USB_Host_GetDeviceConfigDescriptor(1, &ConfigDescriptorSize, ConfigDescriptorData,
 173                                                sizeof(ConfigDescriptorData)) != HOST_GETCONFIG_Successful)
 174         {
 175                 puts_P(PSTR("Error Retrieving Configuration Descriptor.\r\n"));
 176                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 177                 return;
 178         }
 179
 180         if (PRNT_Host_ConfigurePipes(&Printer_PRNT_Interface,
 181                                      ConfigDescriptorSize, ConfigDescriptorData) != PRNT_ENUMERROR_NoError)
 182         {
 183                 puts_P(PSTR("Attached Device Not a Valid Printer Class Device.\r\n"));
 184                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 185                 return;
 186         }
 187
 188         if (USB_Host_SetDeviceConfiguration(1) != HOST_SENDCONTROL_Successful)
 189         {
 190                 puts_P(PSTR("Error Setting Device Configuration.\r\n"));
 191                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 192                 return;
 193         }
 194
 195         if (PRNT_Host_SetBidirectionalMode(&Printer_PRNT_Interface) != HOST_SENDCONTROL_Successful)
 196         {
 197                 puts_P(PSTR("Error Setting Bidirectional Mode.\r\n"));
 198                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 199                 USB_Host_SetDeviceConfiguration(0);
 200                 return;
 201         }
 202
 203         puts_P(PSTR("Printer Device Enumerated.\r\n"));
 204         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 205 }
 206
 207 /** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */
 208 void EVENT_USB_Host_HostError(const uint8_t ErrorCode)
 209 {
 210         USB_Disable();
 211
 212         printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n"
 213                                  " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode);
 214
 215         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 216         for(;;);
 217 }
 218
 219 /** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 220  *  enumerating an attached USB device.
 221  */
 222 void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode,
 223                                             const uint8_t SubErrorCode)
 224 {
 225         printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n"
 226                                  " -- Error Code %d\r\n"
 227                                  " -- Sub Error Code %d\r\n"
 228                                  " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState);
 229
 230         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 231 }
 232