/* 2004.06.05
****************************************
**  Copyright  (C)  W.ch  1999-2004   **
**  Web:  http://www.winchiphead.com  **
****************************************
**  USB Host File Interface for CH374 **
**  TC2.0@PC, ICCAVR_6.31@AVR         **
****************************************
*/
/* CH374 主機文件系統接口 */
/* 支持: FAT12/FAT16/FAT32 */

/* AVR單片機C語言的U盤文件讀寫示例程序 */
/* 該程序將U盤中的/C51/CH374HFT.C文件中的小寫字母轉成大寫字母后, 寫到新建的文件NEWFILE.TXT中,
   如果找不到原文件CH374HFT.C, 那么該程序將顯示C51子目錄下所有以CH374開頭的文件名, 并新建NEWFILE.TXT文件并寫入提示信息,
   如果找不到C51子目錄, 那么該程序將顯示根目錄下的所有文件名, 并新建NEWFILE.TXT文件并寫入提示信息
*/
/* CH374的INT#引腳采用查詢方式處理, 數據復制方式為"內部復制", 本程序適用于ATmega128單片機, 串口0輸出監控信息,9600bps */

/* ICCAVR -v -Wp -I\ICC\INCLUDE -Wf -Mavr_enhanced -Wl -L\ICC\LIB -m -bfunc_lit:0x8C.0x20000 -bdata:0x0100.0x0FFF -dram_end:0x0FFF -dhwstk_size:40 -ucrtatmega.o CH374HFB.C CH374HFB.A */

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

/* 以下定義的詳細說明請看CH374HF9.H文件 */
#define LIB_CFG_INT_EN			0		/* CH374的INT#引腳連接方式,0為"查詢方式",1為"中斷方式" */

#define CH374_INT_WIRE			( PINB & 0x10 )	/* PINB.4, CH374的中斷線INT#引腳,連接CH374的INT#引腳,用于查詢中斷狀態 */
/* 如果未連接CH374的中斷引腳,那么應該去掉上述定義,自動使用寄存器查詢方式 */

/*#define DISK_BASE_BUF_LEN		2048*/	/* 默認的磁盤數據緩沖區大小為512字節,建議選擇為2048甚至4096以支持某些大扇區的U盤,為0則禁止在.H文件中定義緩沖區并由應用程序在pDISK_BASE_BUF中指定 */
/* 如果需要復用磁盤數據緩沖區以節約RAM,那么可將DISK_BASE_BUF_LEN定義為0以禁止在.H文件中定義緩沖區,而由應用程序在調用CH375Init之前將與其它程序合用的緩沖區起始地址置入pDISK_BASE_BUF變量 */

#define NO_DEFAULT_CH374_F_ENUM		1		/* 未調用CH374FileEnumer程序故禁止以節約代碼 */
#define NO_DEFAULT_CH374_F_QUERY	1		/* 未調用CH374FileQuery程序故禁止以節約代碼 */
#define NO_DEFAULT_CH374_RESET		1		/* 未調用CH374Reset程序故禁止以節約代碼 */

#include "..\CH374HFB.H"

UINT8	my_buffer[ 0x0800 ];			/* 外部RAM的文件數據緩沖區 */
/* 如果準備使用雙緩沖區交替讀寫,那么可以在參數中指定緩沖區起址 */

/* 有些AVR單片機提供開放系統總線,那么直接將CH374掛在其系統總線上,以8位I/O方式進行讀寫 */
/* 雖然Atmega128提供系統總線,不過本例假定不開放系統總線,所以用I/O引腳模擬產生CH374的并口讀寫時序 */
/* 本例中的硬件連接方式如下(實際應用電路可以參照修改下述3個并口讀寫子程序) */
/*    單片機的引腳     CH374芯片的引腳
       PINB.4                INT#
       PORTB.3               A0
       PORTB.2               CS#
       PORTB.1               WR#
       PORTB.0               RD#
      PORTA(8位端口)        D7-D0       */

void CH374_PORT_INIT( )  /* 由于使用通用I/O模擬并口讀寫時序,所以進行初始化 */
{
	DDRA = 0x00;  /* 設置8位并口為輸入 */
	PORTB = 0x07;  /* 設置CS,WR,RD默認為高電平 */
	DDRB = 0x0F;  /* 設置CS,WR,RD,A0為輸出,設置INT#為輸入 */
}

void Write374Index( UINT8 mCmd )		/* 向索引端口寫入索引地址 */
{
/*	*(volatile unsigned char *)CH374_IDX_PORT_ADDR = mCmd;  通過并口直接讀寫CH374而非普通I/O模擬 */
	PORTB |= 0x08;  /* 輸出A0=1 */
	PORTA = mCmd;  /* 向CH374的并口輸出數據 */
	DDRA = 0xFF;  /* 并口D0-D7輸出 */
	PORTB &= 0xF9;  /* 輸出有效寫控制信號, 寫CH374芯片的命令端口, A0=1; CS=0; WR=0; RD=1; */
	DDRA = 0xFF;  /* 該操作無意義,僅作延時,CH374要求讀寫脈沖寬度大于100nS */
	PORTB |= 0x07;  /* 輸出無效的控制信號, 完成操作CH374芯片, A0=1; CS=1; WR=1; RD=1; */
	DDRA = 0x00;  /* 禁止數據輸出 */
	PORTB &= 0xF7;  /* 輸出A0=0; 可選操作 */
}

void Write374Data( UINT8 mData )		/* 向數據端口寫入數據,索引地址自動加1 */
{
/*	*(volatile unsigned char *)CH374_DAT_PORT_ADDR = mData;  通過并口直接讀寫CH374而非普通I/O模擬 */
	PORTA = mData;  /* 向CH374的并口輸出數據 */
	DDRA = 0xFF;  /* 并口D0-D7輸出 */
	PORTB &= 0xF1;  /* 輸出有效寫控制信號, 寫CH374芯片的數據端口, A0=0; CS=0; WR=0; RD=1; */
	DDRA = 0xFF;  /* 該操作無意義,僅作延時,CH374要求讀寫脈沖寬度大于100nS */
	PORTB |= 0x07;  /* 輸出無效的控制信號, 完成操作CH374芯片, A0=0; CS=1; WR=1; RD=1; */
	DDRA = 0x00;  /* 禁止數據輸出 */
}

UINT8 Read374Data( void )			/* 從數據端口讀出數據,索引地址自動加1 */
{
	UINT8	mData;
/*	mData = *(volatile unsigned char *)CH374_DAT_PORT_ADDR;  通過并口直接讀寫CH374而非普通I/O模擬 */
	DDRA = 0x00;  /* 數據輸入 */
	PORTB &= 0xF2;  /* 輸出有效讀控制信號, 讀CH374芯片的數據端口, A0=0; CS=0; WR=1; RD=0; */
	DDRA = 0x00;  /* 該操作無意義,僅作延時,CH374要求讀寫脈沖寬度大于100nS */
	mData = PINA;  /* 從CH374的并口PA輸入數據 */
	PORTB |= 0x07;  /* 輸出無效的控制信號, 完成操作CH374芯片, A0=0; CS=1; WR=1; RD=1; */
	return( mData );
}

#include "PARA.C"	/* 并口時序 */
/*#include "SPI.C"*/    /* SPI時序 */

/* 在PB.7連接一個LED用于監控演示程序的進度,低電平LED亮 */
#define LED_OUT_INIT( )		{ PORTB |= 0x80; DDRB |= 0x80; }	/* PORTB.7 高電平為輸出方向 */
#define LED_OUT_ACT( )		{ PORTB &= 0x7F; }	/* PORTB.7 低電平驅動LED顯示 */
#define LED_OUT_INACT( )	{ PORTB |= 0x80; }	/* PORTB.7 低電平驅動LED顯示 */

/* 檢查操作狀態,如果錯誤則顯示錯誤代碼并停機 */
void	mStopIfError( UINT8 iError )
{
	if ( iError == ERR_SUCCESS ) return;  /* 操作成功 */
	printf( "Error: %02X\n", (UINT16)iError );  /* 顯示錯誤 */
/* 遇到錯誤后,應該分析錯誤碼以及CH374DiskStatus狀態,例如調用CH374DiskConnect查詢當前U盤是否連接,如果U盤已斷開那么就重新等待U盤插上再操作,
   建議出錯后的處理步驟:
   1、調用一次CH374DiskReady,成功則繼續操作,例如Open,Read/Write等,在CH374DiskReady中會自動調用CH374DiskConnect，不必另外調用
   2、如果CH374DiskReady不成功,那么強行將CH374DiskStatus置為DISK_CONNECT狀態,然后從頭開始操作(等待U盤連接CH374DiskConnect，CH374DiskReady等) */
	while ( 1 ) {
		LED_OUT_ACT( );  /* LED閃爍 */
		CH374DelaymS( 100 );
		LED_OUT_INACT( );
		CH374DelaymS( 100 );
	}
}

/* 為printf和getkey輸入輸出初始化串口 */
extern int _textmode;
void	mInitSTDIO( )
{
	UBRR0H = 0;
	UBRR0L = 103;  /* 9600bps@16MHz */
	UCSR0B = 0x18; /* BIT(RXEN) | BIT(TXEN); */
	UCSR0C = 0x06; /* BIT(UCSZ1) | BIT(UCSZ0); */
	_textmode = 1;
}

int	main( ) {
	UINT8	i, c, SecCount;
	UINT16	NewSize, count;  /* 因為演示板的RAM容量只有32KB,所以NewSize限制為16位,實際上如果文件大于32256字節,應該分幾次讀寫并且將NewSize改為UINT32以便累計 */
	PUINT8	pCodeStr;
	CH374_PORT_INIT( );
	LED_OUT_INIT( );
	LED_OUT_ACT( );  /* 開機后LED亮一下以示工作 */
	CH374DelaymS( 100 );  /* 延時100毫秒 */
	LED_OUT_INACT( );
	mInitSTDIO( );  /* 為了讓計算機通過串口監控演示過程 */
	printf( "Start\n" );

#if DISK_BASE_BUF_LEN == 0
	pDISK_BASE_BUF = &my_buffer[0];  /* 不在.H文件中定義CH374的專用緩沖區,而是用緩沖區指針指向其它應用程序的緩沖區便于合用以節約RAM */
#endif

	i = CH374LibInit( );  /* 初始化CH374程序庫和CH374芯片,操作成功返回0 */
	mStopIfError( i );
/* 其它電路初始化 */

	while ( 1 ) {
		printf( "Wait Udisk\n" );

#ifdef UNSUPPORT_USB_HUB
/* 如果不需要支持USB-HUB,那么等待U盤插入的程序與CH374相似,都是通過CH374DiskConnect查詢連接,已連接則通過CH374DiskReady等待就緒,然后讀寫 */
		while ( CH374DiskStatus < DISK_CONNECT ) {  /* 查詢CH374中斷并更新中斷狀態,等待U盤插入 */
			CH374DiskConnect( );
			CH374DelaymS( 50 );  /* 沒必要頻繁查詢 */
		}
		LED_OUT_ACT( );  /* LED亮 */
		CH374DelaymS( 200 );  /* 延時,可選操作,有的USB存儲器需要幾十毫秒的延時 */

/* 對于檢測到USB設備的,最多等待100*50mS,主要針對有些MP3太慢,對于檢測到USB設備并且連接DISK_MOUNTED的,最多等待5*50mS,主要針對DiskReady不過的 */
		for ( i = 0; i < 100; i ++ ) {  /* 最長等待時間,100*50mS */
			CH374DelaymS( 50 );
			printf( "Ready ?\n" );
			if ( CH374DiskReady( ) == ERR_SUCCESS ) break;  /* 查詢磁盤是否準備好 */
			if ( CH374DiskStatus < DISK_CONNECT ) break;  /* 檢測到斷開,重新檢測并計時 */
			if ( CH374DiskStatus >= DISK_MOUNTED && i > 5 ) break;  /* 有的U盤總是返回未準備好,不過可以忽略,只要其建立連接MOUNTED且嘗試5*50mS */
		}
		if ( CH374DiskStatus < DISK_CONNECT ) {  /* 檢測到斷開,重新檢測并計時 */
			printf( "Device gone\n" );
			continue;  /* 重新等待 */
		}
		if ( CH374DiskStatus < DISK_MOUNTED ) {  /* 未知USB設備,例如USB鍵盤、打印機等 */
			printf( "Unknown device\n" );
			goto UnknownUsbDevice;
		}
#else
/* 如果需要支持USB-HUB,那么必須參考本例中下面的等待程序 */
		while ( 1 ) {  /* 支持USB-HUB */
			CH374DelaymS( 50 );  /* 沒必要頻繁查詢 */
			if ( CH374DiskConnect( ) == ERR_SUCCESS ) {  /* 查詢方式: 檢查磁盤是否連接并更新磁盤狀態,返回成功說明連接 */
				CH374DelaymS( 200 );  /* 延時,可選操作,有的USB存儲器需要幾十毫秒的延時 */

/* 對于檢測到USB設備的,最多等待100*50mS,主要針對有些MP3太慢,對于檢測到USB設備并且連接DISK_MOUNTED的,最多等待5*50mS,主要針對DiskReady不過的 */
				for ( i = 0; i < 100; i ++ ) {  /* 最長等待時間,100*50mS */
					CH374DelaymS( 50 );
					printf( "Ready ?\n" );
					if ( CH374DiskReady( ) == ERR_SUCCESS ) break;  /* 查詢磁盤是否準備好 */
					if ( CH374DiskStatus < DISK_CONNECT ) {  /* 檢測到斷開,重新檢測并計時 */
						printf( "Device gone\n" );
						break;  /* 重新等待 */
					}
					if ( CH374DiskStatus >= DISK_MOUNTED && i > 5 ) break;  /* 有的U盤總是返回未準備好,不過可以忽略,只要其建立連接MOUNTED且嘗試5*50mS */