#include <avr/io.h>
#include <avr/wdt.h>
#include <avr/interrupt.h>  /* for sei() */
#include <util/delay.h>     /* for _delay_ms() */
#include <stdlib.h>
#include <avr/pgmspace.h>   /* required by usbdrv.h */

#include "usbdrv.h"
#include "oddebug.h"        /* This is also an example for using debug macros */
#include "requests.h"       /* The custom request numbers we use */
#include "uart.h"
//#include "sram.h"
#include "debug.h"
#include "crc.h"


#define STATE_IDLE 0
#define STATE_WRITE 1
#define BUFFER_SIZE 512

extern FILE uart_stdout;

uint32_t    rom_addr;
uint32_t    addr;
uint8_t     bytes_remaining = 0;
uint16_t    sync_errors = 0;
uint8_t     read_buffer[BUFFER_SIZE];
uint8_t     dataBuffer[4];  /* buffer must stay valid when usbFunctionSetup returns */
uint8_t     state = STATE_IDLE;

usbMsgLen_t usbFunctionSetup(uchar data[8]){
    
    usbRequest_t    *rq = (void *)data;
 	uint16_t crc = 0;
    uint8_t len = 0;
    if(rq->bRequest == CUSTOM_RQ_UPLOAD){ /* echo -- used for reliability tests */
        rom_addr = rq->wValue.word;
        rom_addr = rom_addr << 16;
        rom_addr = rom_addr | rq->wIndex.word;
        if (bytes_remaining){
            sync_errors++;
            printf("CUSTOM_RQ_UPLOAD Out of sync Addr=0x%lx remain=%i packet=%i sync_error=%i\n",rom_addr,bytes_remaining,rq->wLength.word,sync_errors );  
            len=0;
        }    
        bytes_remaining  = rq->wLength.word;
        //printf("CUSTOM_RQ_UPLOAD Addr=0x%lx Len=%i\n", rom_addr,bytes_remaining);  
        state = STATE_WRITE;
        len = 0xff;
    }else if(rq->bRequest == CUSTOM_RQ_DOWNLOAD){
        printf("CUSTOM_RQ_DOWNLOAD\n");
    }else if(rq->bRequest == CUSTOM_RQ_CRC_CHECK){
        rom_addr = rq->wValue.word;
        rom_addr = rom_addr << 16;
        rom_addr = rom_addr | rq->wIndex.word;
        printf("CUSTOM_RQ_CRC_CHECK Addr 0x%lx \n", rom_addr);  
    	crc = 0;
        cli();
        for (addr=0x000000; addr<rom_addr; addr+=BUFFER_SIZE) {
        	//sram_read_buffer(addr,read_buffer,BUFFER_SIZE);
    		crc = do_crc_update(crc,read_buffer,BUFFER_SIZE);
            //dump_packet(rom_addr,BUFFER_SIZE,read_buffer);
    		if (addr && addr%32768 == 0){
    		    printf("Addr: 0x%08lx CRC: %04x\n",addr,crc);
    			crc = 0;
    		}
        }
	    printf("Addr: 0x%08lx CRC: %04x\n",addr,crc);
        sei();
    }
    usbMsgPtr = dataBuffer;
    return len;   /* default for not implemented requests: return no data back to host */
}


uint8_t usbFunctionWrite(uint8_t *data, uint8_t len)
{
    if (len > bytes_remaining)
        len = bytes_remaining;
    bytes_remaining -= len;

    //printf("usbFunctionWrite addr=%lx len=%i remain=%i\n",rom_addr,len,bytes_remaining);
    cli();
    //sram_copy(rom_addr,data,len);
    sei();
    rom_addr +=len;
    //printf("usbFunctionWrite %lx %x\n",rom_addr,len);
    //state=STATE_IDLE;
    return len;
}


/* ------------------------------------------------------------------------- */

int main(void)
{
    uint8_t i;
    //wdt_enable(WDTO_1S);
    uart_init();
    stdout = &uart_stdout;
    //sram_init();
 	//printf("SRAM Init\n");
 	//spi_init();
 	//printf("SPI Init\n");
    usbInit();
 	printf("USB Init\n");
    usbDeviceDisconnect();  /* enforce re-enumeration, do this while interrupts are disabled! */
 	printf("USB disconnect\n");
    i = 10;
    while(--i){             /* fake USB disconnect for > 250 ms */
        wdt_reset();
        _delay_ms(1);
    }
    usbDeviceConnect();
 	printf("USB connect\n");
    sei();
 	printf("USB poll\n");
    for(;;){                /* main event loop */
        //wdt_reset();
        usbPoll();
    }
    return 0;
}

/* ------------------------------------------------------------------------- */