/*
 *  spreadspace avr utils
 *
 *
 *  Copyright (C) 2013-2016 Christian Pointner <equinox@spreadspace.org>
 *
 *  This file is part of spreadspace avr utils.
 *
 *  spreadspace avr utils is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  any later version.
 *
 *  spreadspace avr utils is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with spreadspace avr utils. If not, see <http://www.gnu.org/licenses/>.
 */


#include <avr/io.h>
#include <avr/wdt.h>
#include <avr/interrupt.h>
#include <avr/power.h>
#include <stdio.h>

#include "serialio.h"
#include "led.h"

/*
             LUFA Library
     Copyright (C) Dean Camera, 2014.

  dean [at] fourwalledcubicle [dot] com
           www.lufa-lib.org
*/
#include <LUFA/Version.h>
#include <LUFA/Drivers/USB/USB.h>
#include "lufa-descriptor-usbserialmidi.h"

USB_ClassInfo_CDC_Device_t VirtualSerial_CDC_Interface =
  {
    .Config =
      {
        .ControlInterfaceNumber         = INTERFACE_ID_CDC_CCI,
        .DataINEndpoint                 =
          {
            .Address                = CDC_TX_EPADDR,
            .Size                   = CDC_TXRX_EPSIZE,
            .Banks                  = 1,
          },
        .DataOUTEndpoint                =
          {
            .Address                = CDC_RX_EPADDR,
            .Size                   = CDC_TXRX_EPSIZE,
            .Banks                  = 1,
          },
        .NotificationEndpoint           =
          {
            .Address                = CDC_NOTIFICATION_EPADDR,
            .Size                   = CDC_NOTIFICATION_EPSIZE,
            .Banks                  = 1,
          },
      },
  };


USB_ClassInfo_MIDI_Device_t Keyboard_MIDI_Interface =
	{
		.Config =
			{
				.StreamingInterfaceNumber = INTERFACE_ID_AudioStream,
				.DataINEndpoint           =
					{
						.Address          = MIDI_STREAM_IN_EPADDR,
						.Size             = MIDI_STREAM_EPSIZE,
						.Banks            = 1,
					},
				.DataOUTEndpoint          =
					{
						.Address          = MIDI_STREAM_OUT_EPADDR,
						.Size             = MIDI_STREAM_EPSIZE,
						.Banks            = 1,
					},
			},
	};


void EVENT_USB_Device_ConfigurationChanged(void)
{
  CDC_Device_ConfigureEndpoints(&VirtualSerial_CDC_Interface);
  MIDI_Device_ConfigureEndpoints(&Keyboard_MIDI_Interface);
}

void EVENT_USB_Device_ControlRequest(void)
{
  CDC_Device_ProcessControlRequest(&VirtualSerial_CDC_Interface);
  MIDI_Device_ProcessControlRequest(&Keyboard_MIDI_Interface);
}
/* end LUFA CDC-ACM MIDI specific definitions*/

#include "util.h"
#include "led.h"

static FILE usb_stream;

#define KEY_PIN PINB
#define KEY_PORT PORTB
#define KEY_DDR DDRB
#define KEY_BIT 0

#define KEY_LP_MAX 255

void key_init(void)
{
  KEY_DDR = (1<<KEY_BIT);
  KEY_PORT = (1<<KEY_BIT);
}

void handle_cmd(uint8_t cmd)
{
  switch(cmd) {
  case '0': led_off(); break;
  case '1': led_on(); break;
  case 't': led_toggle(); break;
  case 'r': reset2bootloader(); break;
  default: printf("error\r\n"); return;
  }
  printf("ok\r\n");
}

int main(void)
{
  MCUSR &= ~(1 << WDRF);
  wdt_disable();

  cpu_init();
  led_init();
  key_init();
  USB_Init();
  sei();

  CDC_Device_CreateStream(&VirtualSerial_CDC_Interface,&usb_stream);
  stdin = stdout = stderr = &usb_stream;

  MIDI_EventPacket_t MIDIEventOn = (MIDI_EventPacket_t)
    {
      .Event       = MIDI_EVENT(0, MIDI_COMMAND_NOTE_ON),
      .Data1       = MIDI_COMMAND_NOTE_ON | 0,
      .Data2       = 0,
      .Data3       = 0x7F,
    };
  MIDI_EventPacket_t MIDIEventOff = (MIDI_EventPacket_t)
    {
      .Event       = MIDI_EVENT(0, MIDI_COMMAND_NOTE_OFF),
      .Data1       = MIDI_COMMAND_NOTE_OFF | 0,
      .Data2       = 0,
      .Data3       = 0,
    };

  static uint8_t last_state = (1<<KEY_BIT);
  static uint8_t lp_cnt = 0;
  for(;;) {
    int16_t BytesReceived = CDC_Device_BytesReceived(&VirtualSerial_CDC_Interface);
    while(BytesReceived > 0) {
      int ReceivedByte = fgetc(stdin);
      if(ReceivedByte != EOF) {
        handle_cmd(ReceivedByte);
      }
      BytesReceived--;
    }

    MIDI_EventPacket_t ReceivedMIDIEvent;
    while(MIDI_Device_ReceiveEventPacket(&Keyboard_MIDI_Interface, &ReceivedMIDIEvent)) {
      led_on();
      printf("MIDI: %02X %02X %02X\r\n", ReceivedMIDIEvent.Data1, ReceivedMIDIEvent.Data2, ReceivedMIDIEvent.Data3);
      led_off();
    }

    uint8_t state = KEY_PIN & (1<<KEY_BIT);
    if(state != last_state)
      lp_cnt++;
    else
      lp_cnt += lp_cnt ? -1 : 0;

    if(lp_cnt >= KEY_LP_MAX) {
      MIDI_EventPacket_t* MIDIEvent = state ? &MIDIEventOff : &MIDIEventOn;
      MIDI_Device_SendEventPacket(&Keyboard_MIDI_Interface, MIDIEvent);
      MIDI_Device_Flush(&Keyboard_MIDI_Interface);

      printf("Key %s\r\n", state ? "released" : "pressed");

      last_state = state;
      lp_cnt = 0;
    }


#if defined(USB_SERIES_4_AVR) || defined(USB_SERIES_6_AVR) || defined(USB_SERIES_7_AVR)
    if(USB_DeviceState == DEVICE_STATE_Unattached && USB_VBUS_GetStatus()) {
      USB_Disable();
      USB_Init();
    }
#endif

    CDC_Device_USBTask(&VirtualSerial_CDC_Interface);
    MIDI_Device_USBTask(&Keyboard_MIDI_Interface);
    USB_USBTask();
  }
}