#include "44b.h"
#include "44blib.h"
#include "usb.h"
#include "usb_inc.h"
#include "usb_dbg.h"
#include "usb_d12_isr.h"

///////////////////////////////////////////////////////////////////////////
extern CONTROL_XFER 	ControlData;
extern EPPFLAGS 		bEPPflags;
extern unsigned char 	GenEpBuf[EP1_PACKET_SIZE];
extern IO_REQUEST 		ioRequest;

/////////////////////////////////////////////////////////////////////
void bus_reset(void)
{
}

void ep0_rxdone(void)
{
	unsigned char 	ep_last, i, pos;
	
	ep_last = D12_ReadLastTransactionStatus(0); // Clear interrupt flag

	if (1/*ep_last & D12_SETUPPACKET*/ /*bEPPflags.bits.control_state == USB_IDLE*/) 
	{
		ControlData.wLength = 0;
		ControlData.wCount = 0;
		
		if( D12_ReadEndpoint(0, (unsigned char *)(&(ControlData.DeviceRequest)),
			sizeof(ControlData.DeviceRequest)) != sizeof(DEVICE_REQUEST) ) {
			
			D12_SetEndpointStatus(0, 1);
			D12_SetEndpointStatus(1, 1);
			bEPPflags.bits.control_state = USB_IDLE;
			
			return;
		}
		
		// Acknowledge setup here to unlock in/out endp
		D12_AcknowledgeEndpoint(0);
		D12_AcknowledgeEndpoint(1);
		
		ControlData.wLength = ControlData.DeviceRequest.wLength;
		ControlData.wCount = 0;
		
		if (ControlData.DeviceRequest.bmRequestType & (unsigned char)USB_ENDPOINT_DIRECTION_MASK) 
		{
			bEPPflags.bits.setup_packet = 1;
			bEPPflags.bits.control_state = USB_IDLE;		/* get command */
		}
		else 
		{
			if (ControlData.DeviceRequest.wLength == 0) 
			{
				bEPPflags.bits.setup_packet = 1;
				bEPPflags.bits.control_state = USB_IDLE;		/* set command */
			}
			else 
			{
				if(ControlData.DeviceRequest.wLength > MAX_CONTROLDATA_SIZE) 
				{
					// it is a error package, send NAK to host!!!
					
					bEPPflags.bits.control_state = USB_IDLE;
					D12_SetEndpointStatus(0, 1);
					D12_SetEndpointStatus(1, 1);
				}
				else 
				{
					// receive control package next time.
					
					bEPPflags.bits.control_state = USB_RECEIVE;	/* set command with OUT token */
				}
			} // set command with data
		} // else set command
	} // if setup packet
	else if (bEPPflags.bits.control_state == USB_RECEIVE) 
	{
		i =	D12_ReadEndpoint(0, ControlData.dataBuffer + ControlData.wCount,
			EP0_PACKET_SIZE);
		
		ControlData.wCount += i;
		if( i != EP0_PACKET_SIZE || ControlData.wCount >= ControlData.wLength) 
		{
			bEPPflags.bits.setup_packet = 1;
			bEPPflags.bits.control_state = USB_IDLE;
		}
	}
	else 
	{
		bEPPflags.bits.control_state = USB_IDLE;
	}
}

void ep0_txdone(void)
{
	short i = ControlData.wLength - ControlData.wCount;

	D12_ReadLastTransactionStatus(1); // Clear interrupt flag

	if (bEPPflags.bits.control_state != USB_TRANSMIT) 
		return;

	if( i >= EP0_PACKET_SIZE) 
	{
		D12_WriteEndpoint(1, ControlData.pData + ControlData.wCount, EP0_PACKET_SIZE);
		ControlData.wCount += EP0_PACKET_SIZE;

		bEPPflags.bits.control_state = USB_TRANSMIT;
	}
	else if( i != 0) 
	{
		D12_WriteEndpoint(1, ControlData.pData + ControlData.wCount, i);
		ControlData.wCount += i;

		bEPPflags.bits.control_state = USB_IDLE;
	}
	else if (i == 0)
	{
		D12_WriteEndpoint(1, 0, 0); // Send zero packet at the end ???

		bEPPflags.bits.control_state = USB_IDLE;
	}
}

void ep0_txkick(char* p_data, int len)
{

}

void dma_eot(void)
{
	if(bEPPflags.bits.dma_state == DMA_PENDING)
		bEPPflags.bits.setup_dma = 1;
	else
		bEPPflags.bits.dma_state = DMA_IDLE;
}

void ep1_txdone(void)
{
	D12_ReadLastTransactionStatus(3); /* Clear interrupt flag */
}

void ep1_txkick(char* p_data, int len)
{

}

// 接收數據
void ep1_rxdone(void)
{
	unsigned char i;
	unsigned char len;

	D12_ReadLastTransactionStatus(2); 		/* Clear interrupt flag */

	len = D12_ReadEndpoint(2, GenEpBuf, sizeof(GenEpBuf));
	
	USB_DbgPrint(("%x\n", *GenEpBuf));
	
	D12_WriteEndpoint(3, GenEpBuf, len);	// loop to ep1

	if(len != 0)
		bEPPflags.bits.ep1_rxdone = 1;
}

void main_txkick(char* p_data, int len)
{

}

void main_txdone(void)
{
	D12_ReadLastTransactionStatus(5); /* Clear interrupt flag */
}

void main_rxdone(void)
{
	unsigned char 	len;
	unsigned char	buf[1024];

	D12_ReadLastTransactionStatus(4); 	/* Clear interrupt flag */
	len = D12_ReadMainEndpoint( buf );	
	D12_WriteEndpoint(5, buf, len);		// loop to ep2
}