#include <stdio.h>
#include <string.h>

#include <At89x52.h>

#include "khal.h"
#include "kmain.h"
#include "kd12ci.h"
#include "kchap9.h"
#include "kprotodma.h"
//=====================================

/*
//*************************************************************************
// USB protocol function pointer arrays
//*************************************************************************
*/
code void (*StandardDeviceRequest[])(void) =
{
	get_status,
	clear_feature,
	reserved,
	set_feature,
	reserved,
	set_address,
	get_descriptor,
	reserved,
	get_configuration,
	set_configuration,
	get_interface,
	set_interface,
	reserved,
	reserved,
	reserved,
	reserved
};

code void (*VendorDeviceRequest[])(void) =
{
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	reserved,
	read_write_register,
	reserved,
	reserved,
	reserved
};



/*
//*************************************************************************
//  Public static data
//*************************************************************************
*/
extern EPPFLAGS bEPPflags;
extern unsigned long ClockTicks;

extern unsigned char idata GenEpBuf[];
extern IO_REQUEST idata ioRequest;

extern unsigned char ioSize, ioCount;
extern unsigned char xdata MainEpBuf[];
extern unsigned char idata EpBuf[];
//------------
extern void fn_usb_isr(void);
//-----------

CONTROL_XFER ControlData;
unsigned char bNoRAM;


//==================================================
/*
  PCON:		SMOD | - | - | - | GF1|GF0|PD |IDL|
  TCON:		TF1  | - |TF0| - |IE1 |IT1|IE0|IT0|
    
*/
/* Configure Timer 0
   - Mode                  = 1
   - Interrupt                   = ENABLED
   - Clock Source                = INTERNAL
   - Enable Gating Control    = DISABLED
*/


void init_timer0(void)
{
	TMOD &= 0XF0;                /* clear Timer 0   */
	TMOD  |= 0X1;
	TL0 = 0X0;                   /* value set by user    */
	TH0 = 0X0;                   /* value set by user  */
	ET0 = 1;     		     /* IE.1*/
	TR0 = 1;                     /* TCON.4 start timer  */
	PT0 = 1;

	//EA = 1;
}


void init_port()
{
	P0 = 0xFF;
	P1 = 0xFF;
	P2 = 0xFF;
	P3 = 0xFF;
	MCU_D12CS = 0x0;	//D12 Select
	D12SUSPD = 0;		//D12 not in Suspend
}

/*Serial Port */
/*Mode            = 1  /8-bit UART
  Serial Port Interrupt    = Disabled         */
/*Receive         = Enabled   */
/*Auto Addressing    = Disabled   */
void init_serial(void)
{
	SCON = 0X52;		/* SCON: SM0|SM1|SM2|REN|TB8|RB8|TI|RI		*/
	PCON = 0X80 | PCON;	/* PCON: SMOD | - | - | - | GF1|GF0|PD |IDL|    */
	TMOD = 0X20;		/* TMOD: GATE|C/T|M1|M0|GATE|C/T|M1|M0 */
	TCON = 0x69;    	/* TCON: TF1|TR1|TF0|TR0|IE1|IT1|IE0|IT0 */
	TH1 = 0xF3;		/*TH1=0xF3: */
}
//----------------------------
/*
void help_devreq(unsigned char typ, unsigned char req)
{
	typ >>= 5;

	if(typ == USB_STANDARD_REQUEST) {
	}
	else {
		if(bEPPflags.bits.verbose)
			printf("Request Type = %s, bRequest = 0x%bx.\n", _NAME_USB_REQUEST_TYPE[typ],
				req);
	}
}
*/

//=================================================================
void disconnect_USB(void)
{
	// Initialize D12 configuration
	D12_SetMode(D12_NOLAZYCLOCK, D12_SETTOONE | D12_CLOCK_12M);
}

void connect_USB(void)
{
	// reset event flags

	DISABLE;
	bEPPflags.value = 0;
	ENABLE;

	// V2.1 enable normal+sof interrupt
	D12_SetDMA(D12_ENDP4INTENABLE | D12_ENDP5INTENABLE);

	// Initialize D12 configuration

	//D12_SetMode(D12_NOLAZYCLOCK ,D12_SETTOONE | D12_CLOCK_12M );
	D12_SetMode(D12_NOLAZYCLOCK | D12_SOFTCONNECT,D12_SETTOONE | D12_CLOCK_12M );

}


void reconnect_USB(void)
{
	unsigned long clk_cnt;

	MCU_LED0 = 0;
	MCU_LED1 = 0;

	// Pull-down D12's SUSPEND pin
	// Disable 74HCT123 pulse generation before disconnect
	// Release D12's SUSPEND pin after receiving bus reset from host
	D12SUSPD = 0;
	disconnect_USB();

	//printf("Wait for 1 second ...\n");

	clk_cnt = ClockTicks;
	while(ClockTicks < clk_cnt + 20)
		;

	connect_USB();

	MCU_LED0 = 1;
	MCU_LED1 = 1;

}
//------------------------------
void stall_ep0(void)
{
	D12_SetEndpointStatus(0, 1);
	D12_SetEndpointStatus(1, 1);
}

//---------------
void on_exit(void)
{
 
}
//----------------------------------
void check_key_LED(void)
{
	static unsigned char c, last_key = 0xf;

	c = MCU_SWM0 & MCU_SWM1;
	c &= 0x0f;
	if (c != last_key) {
		D12_WriteEndpoint(3, 1, &c);
	}

	last_key = c;

	if(bEPPflags.bits.ep1_rxdone) {
		DISABLE;
		bEPPflags.bits.ep1_rxdone = 0;
		ENABLE;
		MCU_LED0 = !(GenEpBuf[3] & 0x1);
		MCU_LED1 = !(GenEpBuf[3] & 0x2);
	}
}
//--------------------------
void init_unconfig(void)
{
	//unsigned char i;

	D12_SetEndpointEnable(0);	/* Disable all endpoints but EPP0. */
}

void init_config(void)
{
	D12_SetEndpointEnable(1);	/* Enable  generic/iso endpoints. */
}
//--------------------------

void single_transmit(unsigned char * buf, unsigned char len)
{
	if( len <= EP0_PACKET_SIZE) {
		D12_WriteEndpoint(1, len, buf);
	}
}

void code_transmit(unsigned char code * pRomData, unsigned short len)
{
	ControlData.wCount = 0;
	if(ControlData.wLength > len)
		ControlData.wLength = len;

	ControlData.pData = pRomData;
	if( ControlData.wLength >= EP0_PACKET_SIZE) {	// 如果發送代碼大于包長度
		D12_WriteEndpoint(1, EP0_PACKET_SIZE, ControlData.pData);
		ControlData.wCount += EP0_PACKET_SIZE;

		DISABLE;
		bEPPflags.bits.control_state = USB_TRANSMIT;
		ENABLE;
	}
	else {						//發送代碼小于包長度
		D12_WriteEndpoint(1, ControlData.wLength, pRomData);
		ControlData.wCount += ControlData.wLength;
		DISABLE;
		bEPPflags.bits.control_state = USB_IDLE;
		ENABLE;
	}
}

//---------------------------
void setup_dma()
{
	ioSize = ioRequest.uSize;
	ioCount = 0;

	DISABLE;
	bEPPflags.bits.dma_state = DMA_RUNNING;
	ENABLE;

	single_transmit(0, 0);

	if(ioRequest.bCommand & 0x1) {
		if(ioSize > 64 || ioSize == 0) {
			//if(bNoRAM)
				D12_WriteEndpoint(5, 64, EpBuf);
			//else
			//	D12_WriteEndpoint(5, 64, MainEpBuf);
			ioCount += 64;
		}
		else {
			//if(bNoRAM)
				D12_WriteEndpoint(5, ioSize, EpBuf);
			//else
			//	D12_WriteEndpoint(5, ioSize, MainEpBuf);
			ioCount += ioSize;
		}
	}
}

//------------------------------
void control_handler()
{
	unsigned char type, req;
	
	type = ControlData.DeviceRequest.bmRequestType & USB_REQUEST_TYPE_MASK;	//0x60
	req = ControlData.DeviceRequest.bRequest & USB_REQUEST_MASK;		//0x0F

	//help_devreq(type, req); // print out device request

	if (type == USB_STANDARD_REQUEST)
		(*StandardDeviceRequest[req])();
	else if (type == USB_VENDOR_REQUEST)
		(*VendorDeviceRequest[req])();
	else
		stall_ep0();
	
}


//-------------------------------------------

//-----------xugy------begin-------------

//-----------xugy------end---------------


//-------------------------------------------

void main()
{
//	unsigned char key, i;
	//bNoRam=1;
	
	init_port();
	init_serial();
	init_timer0(); 
	EA = 1; 
	//for test
	MCU_LED0=0;
	MCU_LED1=0;
	MCU_LED0=1;
	MCU_LED1=1;
	//--------------
	printf("Re-connect PDIUSBD12 evaluation board to USB.\n");
	IE0=0;
	IT0=0;
	EX0=1;
	//--------------------------
//	MCU_D12RST = 0;
//	MCU_D12RST = 1;
	//--------------------
	
	MCU_D12CS = 0x1;

		bNoRAM = TRUE;

	MCU_D12CS = 0x0;

		reconnect_USB();

	while( TRUE ){
	   SBUF= 0xAA;
		if (bEPPflags.bits.timer){
			DISABLE;
			bEPPflags.bits.timer = 0;
			ENABLE;

			if(bEPPflags.bits.configuration)
				check_key_LED();
		}

		if (bEPPflags.bits.bus_reset) {
			DISABLE;
			bEPPflags.bits.bus_reset = 0;
			ENABLE;
			// Release D12's SUSPEND pin after bus reset
			// Enable 74HCT123 pulse generation before disconnect
			///D12SUSPD = 1;
			D12SUSPD=0;
		} // if bus reset

		if (bEPPflags.bits.suspend) {
			DISABLE;
			bEPPflags.bits.suspend= 0;
			ENABLE;

			if(D12SUSPD == 1) {
				D12SUSPD = 0;
				P0 = 0xFF;
				P1 = 0xFF;
				P2 = 0xFF;
				P3 = 0xFF;
				D12_SetDMA(0xC3);
				D12SUSPD = 1;
				PCON |= 0x02;
				while (1);
		
			}
		
		} // if suspend change

		if (bEPPflags.bits.setup_packet){
			DISABLE;
			bEPPflags.bits.setup_packet = 0;
			ENABLE;
			control_handler();
			//D12SUSPD = 1;
			D12SUSPD=0;
		} // if setup_packet
		
		if(bEPPflags.bits.setup_dma != 0) {
			DISABLE;
			bEPPflags.bits.setup_dma --;
			ENABLE;
			setup_dma();
		} // if setup_dma
		
	} // Main Loop

    on_exit();
	//=======================================	
 
}