#include "PAD.h"
  

  /** Main program entry point. This routine contains the overall program flow, including initial
   *  setup of all components and the main program loop.
   */

  int main(void){

  	SetupHardware();
  

  	LEDs_SetAllLEDs(LEDS_LED1 | LEDS_LED2);

  	GlobalInterruptEnable();
  

  	for (;;){
  		USB_USBTask();

  		PAD_Task();
  	}
  }

  uint8_t led = 0;

  void PAD_Task(void){
  	if (USB_DeviceState != DEVICE_STATE_Configured)
  		return;
  	LEDs_SetAllLEDs(0);
  
  	// receive from 328p & send to PC
  	/* Notre convention :
  	 *
  	 *  328p -> 12u2 : 0bXYZZZZZZ
  	 * X = 0 -> les Z sont les états de chaque boutons
  	 * X = 1 & Y = 0 -> les Z correspondent à la valeur X du joystick
  	 * X = 1 & Y = 1 -> les Z correspondent à la valeur Y du joystick
  	 * 
  	 * Cas 0b00ZZZZZZ : (boutons)
  	 *  12u2 -> PC (endpoint button 1) : 0b000ZZZZZ
  	 * Z : état d'un bouton
  	
  	 * Cas 0b1YZZZZZZ : (joystick)

  	 *  12u2 -> PC (endpoint button 2) : 0bX0YYYYYY
  	 * X = 0 -> valeur en X du joystick
  	 * X = 1 -> valeur en Y du joystick

  	 */
  	if(Serial_IsCharReceived()){
  		uint8_t byte = Serial_ReceiveByte();
  		if((byte & 0x80) == 0){ // bouttons
  			Endpoint_SelectEndpoint(PAD_IN_EP_BUTTON1);
  			if (Endpoint_IsINReady() && Endpoint_IsReadWriteAllowed()){

  				LEDs_SetAllLEDs(LEDS_LED1); //LED Tx

  				Endpoint_Write_8(byte);
  				Endpoint_ClearIN();

  			}

  		}else if((byte & 0x40) == 0){ // joystick X

  			uint8_t toSend = (byte) & 0x3F;

  			Endpoint_SelectEndpoint(PAD_IN_EP_BUTTON2);
  			if (Endpoint_IsINReady() && Endpoint_IsReadWriteAllowed()){

  				LEDs_SetAllLEDs(LEDS_LED1); //LED Tx
  				Endpoint_Write_8(toSend);

  				Endpoint_ClearIN();

  			}
  		}else{ // joystick Y

  			uint8_t toSend = ((byte) & 0x3F) | 0x80;

  			Endpoint_SelectEndpoint(PAD_IN_EP_BUTTON2);
  			if (Endpoint_IsINReady() && Endpoint_IsReadWriteAllowed()){

  				LEDs_SetAllLEDs(LEDS_LED1); //LED Tx
  				Endpoint_Write_8(toSend);

  				Endpoint_ClearIN();

  			}

  		}
  	}

  
  	// receive from PC & send to 328p
  	/* Notre convention :
  	 *
  	 *  PC (endpoint led 1) -> 16u2 : 0b000XXXXX
  	 * X = état d'une LED (LEDs 8 9 10 11 12)
  	 *
  	 *  PC (endpoint led 2) -> 16u2 : 0b0000000X
  	 * X = état d'une LED (LED 13)
  	 * 
  	 *  16u2 -> 328p : 0bXYYYYYYY

  	 * X = 0 -> pour les LEDs 8 9 10 11 12 13 (Y : les données)

  	 */
  	Endpoint_SelectEndpoint(PAD_OUT_EP_LED1);
  	if (Endpoint_IsOUTReceived() && Endpoint_IsReadWriteAllowed()){

  		LEDs_SetAllLEDs(LEDS_LED2); //LED Rx

  		uint8_t byte = Endpoint_Read_8();

  		led = (led & 0x20) | byte;
  		Serial_SendByte(led);

  		Endpoint_ClearOUT();
  	}

  	Endpoint_SelectEndpoint(PAD_OUT_EP_LED2);

  	if (Endpoint_IsOUTReceived() && Endpoint_IsReadWriteAllowed()){

  		LEDs_SetAllLEDs(LEDS_LED2); //LED Rx

  		uint8_t byte = Endpoint_Read_8();

  		led = (led & 0x1F) | byte << 5;
  		Serial_SendByte(led);

  		Endpoint_ClearOUT();
  	}

  
  	//LED Reset
  	LEDs_SetAllLEDs(0);

  }
  
  /** Configures the board hardware and chip peripherals for the project'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 */
  	USB_Init();

  	LEDs_Init();

  	Serial_Init(9600, 0);
  }

  // endpoints creation

  void EVENT_USB_Device_ConfigurationChanged(void){

  	/* Setup HID Report Endpoints */
  	Endpoint_ConfigureEndpoint(PAD_OUT_EP_LED1  , EP_TYPE_INTERRUPT, PAD_EPSIZE, 1);

  	Endpoint_ConfigureEndpoint(PAD_OUT_EP_LED2  , EP_TYPE_INTERRUPT, PAD_EPSIZE, 1);

  	Endpoint_ConfigureEndpoint(PAD_IN_EP_BUTTON1, EP_TYPE_INTERRUPT, PAD_EPSIZE, 1);
  	Endpoint_ConfigureEndpoint(PAD_IN_EP_BUTTON2, EP_TYPE_INTERRUPT, PAD_EPSIZE, 1);

  }