/* 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文件中的前600個字符顯示出來,
   如果找不到原文件CH374HFT.C, 那么該程序將顯示C51子目錄下所有以CH374開頭的文件名,
   如果找不到C51子目錄, 那么該程序將顯示根目錄下的所有文件名,
   最后將程序ROM中的一個字符串寫入寫入新建的文件"NEWFILE.TXT"中
*/
/* CH374的INT#引腳采用查詢方式處理, 數據復制方式為"內部復制", 本程序適用于ATmega128單片機, 串口0輸出監控信息,9600bps */
/* 本例以字節為單位讀寫U盤文件,讀寫速度較扇區模式慢,但是由于字節模式讀寫文件不需要文件數據緩沖區FILE_DATA_BUF,
   所以總共只需要600字節的RAM,適用于單片機硬件資源有限、數據量小并且讀寫速度要求不高的系統 */

/* 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"

/* 本例用硬件標準SPI接口0產生CH374的4線SPI串口讀寫時序 */
/* 本例中的硬件連接方式如下(實際應用電路可以參照修改下述定義及子程序) */
/* Atmega128單片機的引腳  CH374芯片的引腳
         PB.0                 SCS#
         PB.1                 SCK
         PB.3                 SDO
         PB.2                 SDI     */

void CH374_PORT_INIT( )  /* 進行初始化 */
{
	PORTB |= 0x03;  /* 禁止SPI片選,設置SCS#默認為高電平 */
	DDRB |= 0x07;  /* 設置SCS#,SCK,SDI為輸出 */
	DDRB &= 0x08;  /* SDO輸入 */
/* 如果是硬件SPI接口,那么可使用mode3(CPOL=1&CPHA=1)或mode0(CPOL=0&CPHA=0),CH374在時鐘上升沿采樣輸入,下降沿輸出,數據位是高位在前 */
	SPCR = 0x5C;  /* 設置SPI模式3, DORD=0(MSB first), CPOL=1, CPHA=1, CH374也支持SPI模式0 */
}

static	UINT8	Spi374Exchange( UINT8 d )  /* 硬件SPI輸出且輸入8個位數據 */
{  /* 為了提高速度,可以將該子程序做成宏以減少子程序調用層次 */
	SPDR = d;  /* 先將數據寫入SPI數據寄存器,然后查詢SPI狀態寄存器以等待SPI字節傳輸完成 */
	while ( ( SPSR & 0x80 ) == 0 );  /* 查詢SPI狀態寄存器以等待SPI字節傳輸完成 */
	return( SPDR );  /* 先查詢SPI狀態寄存器以等待SPI字節傳輸完成,然后從SPI數據寄存器讀出數據 */
}

#define	Spi374OutByte( d )		Spi374Exchange( d )		/* SPI輸出8個位數據 */
#define	Spi374InByte( )			Spi374Exchange( 0xFF )	/* SPI輸入8個位數據 */

static	void	Spi374Start( UINT8 addr, UINT8 cmd )  /* SPI開始 */
{
	PORTB &= 0xFE;  /* SPI片選有效 */
	Spi374OutByte( addr );
	Spi374OutByte( cmd );
}

#define	Spi374Stop( )	{ PORTB |= 0x01; }  /* SPI結束 */
/*static	void	Spi374Stop( void )*/  /* SPI結束 */
/*{
	PORTB |= 0x01;*/	/* SPI片選無效 */
/*}*/

/*#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;
	UINT16	TotalCount;
	UINT8	*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" );
		while ( 1 ) {  /* 支持USB-HUB,更復雜的處理參考EXAM1和EXAM13 */
			CH374DelaymS( 10 );  /* 沒必要頻繁查詢 */
			if ( CH374DiskConnect( ) == ERR_SUCCESS ) break;  /* 查詢方式: 檢查磁盤是否連接并更新磁盤狀態,返回成功說明連接 */
		}
		LED_OUT_ACT( );  /* LED亮 */
		CH374DelaymS( 200 );  /* 延時,可選操作,有的USB存儲器需要幾十毫秒的延時 */

/* 檢查U盤是否準備好,有些U盤不需要這一步,但是某些U盤必須要執行這一步才能工作 */
		for ( i = 0; i < 5; i ++ ) {  /* 有的U盤總是返回未準備好,不過可以被忽略 */
			CH374DelaymS( 100 );
			printf( "Ready ?\n" );
			if ( CH374DiskReady( ) == ERR_SUCCESS ) break;  /* 查詢磁盤是否準備好 */
		}
		if ( CH374DiskStatus <= DISK_CONNECT ) continue;  /* 支持USB-HUB */

/* 查詢磁盤物理容量 */
/*		printf( "DiskSize\n" );
		i = CH374DiskSize( );  
		mStopIfError( i );
		printf( "TotalSize = %u MB \n", (unsigned int)( mCmdParam.DiskSize.mDiskSizeSec * ( CH374vSectorSize / 512 ) >> 11 ) );  顯示為以MB為單位的容量
*/

/* 讀取原文件 */
		printf( "Open\n" );
		strcpy( (char *)mCmdParam.Open.mPathName, "/C51/CH374HFT.C" );  /* 文件名,該文件在C51子目錄下 */
		i = CH374FileOpen( );  /* 打開文件 */
		if ( i == ERR_MISS_DIR || i == ERR_MISS_FILE ) {  /* 沒有找到文件 */
/* 列出文件 */
			if ( i == ERR_MISS_DIR ) pCodeStr = (PUINT8)"/*";  /* C51子目錄不存在則列出根目錄下的文件 */
			else pCodeStr = (PUINT8)"/C51/CH374*";  /* CH374HFT.C文件不存在則列出\C51子目錄下的以CH374開頭的文件 */
			printf( "List file %s\n", pCodeStr );
			for ( c = 0; c < 254; c ++ ) {  /* 最多搜索前254個文件,如果要搜索大255個文件請參考EXAM1和EXAM0 */
				strcpy( (char *)mCmdParam.Open.mPathName, (char *)pCodeStr );  /* 搜索文件名,*為通配符,適用于所有文件或者子目錄 */
/* 另外一種枚舉方法是，將此處的c置為0xFF，然后將真正的枚舉序號存放于CH374vFileSize中，從而可以搜索大于254個文件，從0到2147483647個 */
				i = strlen( (char *)mCmdParam.Open.mPathName );
				mCmdParam.Open.mPathName[ i ] = c;  /* 根據字符串長度將結束符替換為搜索的序號,從0到254 */
				i = CH374FileOpen( );  /* 打開文件,如果文件名中含有通配符*,則為搜索文件而不打開 */
				if ( i == ERR_MISS_FILE ) break;  /* 再也搜索不到匹配的文件,已經沒有匹配的文件名 */
				if ( i == ERR_FOUND_NAME ) {  /* 搜索到與通配符相匹配的文件名,文件名及其完整路徑在命令緩沖區中 */
					printf( "  match file %03d#: %s\n", (unsigned int)c, mCmdParam.Open.mPathName );  /* 顯示序號和搜索到的匹配文件名或者子目錄名 */
					continue;  /* 繼續搜索下一個匹配的文件名,下次搜索時序號會加1 */
				}
				else {  /* 出錯 */
					mStopIfError( i );
					break;
				}
			}
		}
		else {  /* 找到文件或者出錯 */
			mStopIfError( i );
			TotalCount = 600;  /* 準備讀取總長度 */
			printf( "從文件中讀出的前%d個字符是:\n",TotalCount );
			while ( TotalCount ) {  /* 如果文件比較大,一次讀不完,可以再調用CH374ByteRead繼續讀取,文件指針自動向后移動 */
				if ( TotalCount > MAX_BYTE_IO ) c = MAX_BYTE_IO;  /* 剩余數據較多,限制單次讀寫的長度不能超過 sizeof( mCmdParam.ByteRead.mByteBuffer ) */
				else c = TotalCount;  /* 最后剩余的字節數 */
				mCmdParam.ByteRead.mByteCount = c;  /* 請求讀出幾十字節數據 */
				i = CH374ByteRead( );  /* 以字節為單位讀取數據塊,單次讀寫的長度不能超過MAX_BYTE_IO,第二次調用時接著剛才的向后讀 */
				mStopIfError( i );
				TotalCount -= mCmdParam.ByteRead.mByteCount;  /* 計數,減去當前實際已經讀出的字符數 */
				for ( i=0; i!=mCmdParam.ByteRead.mByteCount; i++ ) printf( "%c", mCmdParam.ByteRead.mByteBuffer[i] );  /* 顯示讀出的字符 */
				if ( mCmdParam.ByteRead.mByteCount < c ) {  /* 實際讀出的字符數少于要求讀出的字符數,說明已經到文件的結尾 */
					printf( "\n" );
					printf( "文件已經結束\n" );
					break;
				}
			}
/*	    如果希望從指定位置開始讀寫,可以移動文件指針
		mCmdParam.ByteLocate.mByteOffset = 608;  跳過文件的前608個字節開始讀寫
		CH374ByteLocate( );
		mCmdParam.ByteRead.mByteCount = 5;  讀取5個字節
		CH374ByteRead( );   直接讀取文件的第608個字節到612個字節數據,前608個字節被跳過

	    如果希望將新數據添加到原文件的尾部,可以移動文件指針
		CH374FileOpen( );
		mCmdParam.ByteLocate.mByteOffset = 0xffffffff;  移到文件的尾部
		CH374ByteLocate( );
		mCmdParam.ByteWrite.mByteCount = 13;  寫入13個字節的數據
		CH374ByteWrite( );   在原文件的后面添加數據,新加的13個字節接著原文件的尾部放置
		mCmdParam.ByteWrite.mByteCount = 2;  寫入2個字節的數據
		CH374ByteWrite( );   繼續在原文件的后面添加數據
		mCmdParam.ByteWrite.mByteCount = 0;  寫入0個字節的數據,實際上該操作用于通知程序庫更新文件長度
		CH374ByteWrite( );   寫入0字節的數據,用于自動更新文件的長度,所以文件長度增加15,如果不這樣做,那么執行CH374FileClose時也會自動更新文件長度
*/
			printf( "Close\n" );
			i = CH374FileClose( );  /* 關閉文件 */
			mStopIfError( i );
		}

#ifdef EN_DISK_WRITE  /* 子程序庫支持寫操作 */
/* 產生新文件 */
		printf( "Create\n" );
		strcpy( (char *)mCmdParam.Create.mPathName, "/NEWFILE.TXT" );  /* 新文件名,在根目錄下,中文文件名 */
		i = CH374FileCreate( );  /* 新建文件并打開,如果文件已經存在則先刪除后再新建 */
		mStopIfError( i );
		printf( "Write\n" );
		pCodeStr = (PUINT8)"Note: \xd\xa這個程序是以字節為單位進行U盤文件讀寫,單片機只需要有600字節的RAM\xd\xa";
		while( 1 ) {  /* 分多次寫入文件數據 */
			for ( i=0; i<MAX_BYTE_IO; i++ ) {
				c = *pCodeStr;
				mCmdParam.ByteWrite.mByteBuffer[i] = c;
				if ( c == 0 ) break;  /* 源字符串結束 */
				pCodeStr++;
			}
			if ( i == 0 ) break;  /* 源字符串結束,完成寫文件 */
			mCmdParam.ByteWrite.mByteCount = i;  /* 寫入數據的字符數,單次讀寫的長度不能超過MAX_BYTE_IO,第二次調用時接著剛才的向后寫 */
			i = CH374ByteWrite( );  /* 向文件寫入數據 */
			mStopIfError( i );
		}
/*		printf( "Modify\n" );
		mCmdParam.Modify.mFileAttr = 0xff;   輸入參數: 新的文件屬性,為0FFH則不修改
		mCmdParam.Modify.mFileTime = 0xffff;   輸入參數: 新的文件時間,為0FFFFH則不修改,使用新建文件產生的默認時間
		mCmdParam.Modify.mFileDate = MAKE_FILE_DATE( 2004, 5, 18 );  輸入參數: 新的文件日期: 2004.05.18
		mCmdParam.Modify.mFileSize = 0xffffffff;   輸入參數: 新的文件長度,以字節為單位寫文件應該由程序庫關閉文件時自動更新長度,所以此處不修改
		i = CH374FileModify( );   修改當前文件的信息,修改日期
		mStopIfError( i );
*/
		printf( "Close\n" );
		mCmdParam.Close.mUpdateLen = 1;  /* 自動計算文件長度,以字節為單位寫文件,建議讓程序庫關閉文件以便自動更新文件長度 */
		i = CH374FileClose( );
		mStopIfError( i );

/* 刪除某文件 */
/*		printf( "Erase\n" );
		strcpy( (char *)mCmdParam.Create.mPathName, "/OLD" );  將被刪除的文件名,在根目錄下
		i = CH374FileErase( );  刪除文件并關閉
		if ( i != ERR_SUCCESS ) printf( "Error: %02X\n", (UINT16)i );  顯示錯誤
*/

/* 查詢磁盤信息 */
/*		printf( "Disk\n" );
		i = CH374DiskQuery( );
		mStopIfError( i );
		printf( "Fat=%d, Total=%ld, Free=%ld\n", (UINT16)mCmdParam.Query.mDiskFat, mCmdParam.Query.mTotalSector, mCmdParam.Query.mFreeSector );
*/
#endif
		printf( "Take out\n" );
		while ( 1 ) {  /* 支持USB-HUB */
			CH374DelaymS( 10 );  /* 沒必要頻繁查詢 */
			if ( CH374DiskConnect( ) != ERR_SUCCESS ) break;  /* 查詢方式: 檢查磁盤是否連接并更新磁盤狀態,返回成功說明連接 */
		}
		LED_OUT_INACT( );  /* LED滅 */
		CH374DelaymS( 200 );
	}
}