#include <nitro.h>
#include "WDS.h"

// ヒープ初期化関数
static void InitializeAllocateSystem(void);

// アクセスポイント情報のデバッグ表示用関数
static void DumpWDSApInfo( WDSApInfo *apinfo );

void VBlankIntr(void);

// WDS非同期関数のコールバック関数たち
// コールバック関数内ではアプリケーションのステート変数g_appstateを変更する
void WDS_Initialize_CB(void *arg);
void WDS_StartScan_CB(void *arg);
void WDS_EndScan_CB(void *arg);
void WDS_End_CB(void *arg);

// アプリケーションを制御するステートの列挙型
typedef enum AppState {
	APP_STATE_WDSINIT,
	APP_STATE_WDSWAITINIT,
	APP_STATE_WDSSCAN,
	APP_STATE_WDSWAITSCAN,
	APP_STATE_WDSCOMPLETESCAN,
	APP_STATE_WDSENDSCAN,
	APP_STATE_WDSWAITENDSCAN,
	APP_STATE_WDSCOMPLETEENDSCAN,
        APP_STATE_WDSEND,
	APP_STATE_WDSWAITEND,
	APP_STATE_WDSCOMPLETEEND
} AppState;

// アプリケーションを制御するステート変数
static AppState g_appstate = APP_STATE_WDSINIT;

// 受け取ったビーコン情報を格納する変数(この配列をランチャー経由でホットスポットチャンネルに渡す)
static WDSBriefApInfo briefapinfo[WDS_APINFO_MAX];

// 時間計測用
static OSTick wdsBegin;

/*---------------------------------------------------------------------------*
  Name:         NitroMain

  Description:  main

  Arguments:    None

  Returns:      None
 *---------------------------------------------------------------------------*/
void NitroMain(void)
{
	u8 *wdsSysBuf;
	OSTick wdsScanBeginTick = 0;
	
	// 各種初期化処理
	OS_Init();
	OS_InitTick();
	OS_InitAlarm();

	//---- interrupt setting
	OS_SetIrqFunction(OS_IE_V_BLANK, VBlankIntr);
	OS_EnableIrqMask(OS_IE_V_BLANK);
	OS_EnableIrq();
	GX_VBlankIntr(TRUE);

	InitializeAllocateSystem();
	
	// WDSライブラリが使用するバッファを確保(32バイトアラインメントしている必要がある)
	wdsSysBuf = OS_Alloc( WDS_GetWorkAreaSize() );
	if( wdsSysBuf == NULL)
	{
		OS_Panic("OS_Alloc Failed");
	}
	
	// メインループ
	while( 1 ) {
		OSTick now;
		int i;
		
		OS_WaitVBlankIntr();
		switch( g_appstate )
		{
		case APP_STATE_WDSINIT:
			// イニシャルステート
			
			// WDSライブラリの初期化関数を呼び出し、その非同期処理の完了を待つ
			OS_Printf("*** WDS_Initialize\n");
			g_appstate = APP_STATE_WDSWAITINIT;
			wdsBegin = OS_GetTick();
			if( WDS_Initialize( wdsSysBuf, WDS_Initialize_CB, 0 ) == 0 )
			{
				OS_Printf("WDS_Initialize successed\n");
			}
			else {
				OS_TPrintf("WDS_Initialize failed");
				g_appstate = APP_STATE_WDSCOMPLETEEND;
			}
			break;
			OS_Printf("*** WDS_Initialize waiting asyncronous process\n");
		case APP_STATE_WDSWAITINIT:
		case APP_STATE_WDSWAITSCAN:
		case APP_STATE_WDSWAITENDSCAN:
		case APP_STATE_WDSWAITEND:
			// 非同期処理の完了を待つステート群
			
			// コールバック関数が呼び出され、ステートが変更されるのを待てばよい
			break;
		case APP_STATE_WDSSCAN:
			// 初期化が完了した直後か、ビーコンスキャン完了時に引き続きスキャンを行う場合に入ってくるステート
			
			//OS_Printf("*** WDS_StartScan\n");
			// ビーコンスキャン非同期処理を開始する
			g_appstate = APP_STATE_WDSWAITSCAN;
			wdsBegin = OS_GetTick();
			if( WDS_StartScan( WDS_StartScan_CB ) == 0 )
			{
				if( wdsScanBeginTick == 0 )
					wdsScanBeginTick = OS_GetTick();
			}
			else {
				OS_TPrintf("WDS_EndScan failed");
				g_appstate = APP_STATE_WDSEND;
			}
			break;
		case APP_STATE_WDSCOMPLETESCAN:
			// スキャン完了後に入ってくるステート
			
			// 一回のスキャンではビーコンを取れないことが多いので、2秒間ビーコン受信を繰り返す
			now = OS_GetTick();
			if( OS_TicksToMilliSeconds(now - wdsScanBeginTick) < 2000 )
			{
				// 再スキャンのためにスキャン開始ステートに移行
				g_appstate	= APP_STATE_WDSSCAN;
			}
			else {
				// スキャン終了ステートに移行
				g_appstate	= APP_STATE_WDSENDSCAN;
			}
			break;
		case APP_STATE_WDSENDSCAN:
			// スキャンを終了させる際に入ってくるステート
			OS_Printf("*** WDS_EndScan\n");
			
			// スキャンを終了させる非同期処理を開始する
			wdsBegin = OS_GetTick();
			g_appstate = APP_STATE_WDSWAITENDSCAN;
			if( WDS_EndScan( WDS_EndScan_CB ) == 0 )
			{
				OS_Printf("WDS_EndScan successed\n");
			}
			else {
				OS_TPrintf("WDS_EndScan failed");
				g_appstate = APP_STATE_WDSEND;
			}
			break;
		case APP_STATE_WDSCOMPLETEENDSCAN:
			// スキャン終了非同期処理が終わった際に入って来るステート
			OS_Printf("*** WDS_GetApInfoAll\n");
			if( WDS_GetApInfoAll( briefapinfo ) != 0 )
			{
				OS_Panic("WDS_GetApInfoAll failed\n");
			}
			for( i = 0 ; i < WDS_APINFO_MAX ; i++ )
			{
				if( briefapinfo[i].isvalid == TRUE )
				{
					OS_TPrintf("rssi: %d\n", briefapinfo[i].rssi);
					DumpWDSApInfo( &briefapinfo[i].apinfo );
				}
			}
			g_appstate = APP_STATE_WDSEND;
			break;
                case APP_STATE_WDSEND:	
			// WDSライブラリを終了し、無線ハードの電源を落とす非同期処理を開始する
			OS_Printf("*** WDS_End\n");
			g_appstate = APP_STATE_WDSWAITEND;
			wdsBegin = OS_GetTick();
			if( WDS_End( WDS_End_CB ) == 0 )
			{
				OS_Printf("WDS_End successed\n");
			}
			else {
				OS_Panic("WDS_End failed");
			}
			break;
		case APP_STATE_WDSCOMPLETEEND:
			// WDSライブラリの解放処理が完了した際に入って来るステート
			
			OS_TPrintf("WDS completed\n");
			OS_Terminate();
		}
	}
}

// アクセスポイント情報のデバッグ表示用関数
static void DumpWDSApInfo( WDSApInfo *apinfo )
{
	int i;
	char buf[256];
	
	OS_TPrintf( "================================\n" );
	// SSID
	MI_CpuCopy8( apinfo->ssid, buf, WDS_SSID_BUF_SIZE) ;
	buf[WDS_SSID_BUF_SIZE] = 0x00;
	OS_TPrintf( "SSID: %s\n", buf );
	
	// APNUM
	MI_CpuCopy8( apinfo->apnum, buf, WDS_APNUM_BUF_SIZE) ;
	buf[WDS_APNUM_BUF_SIZE] = 0x00;
	OS_TPrintf( "APNUM: %s\n", buf );
	
	// CHANNEL
	OS_TPrintf( "channel: %d\n", apinfo->channel );
	
	// ENCRYPTFLAG
	OS_TPrintf( "encryptmethod: %d\n", apinfo->encryptflag);
	
	// WEPKEY
	OS_TPrintf( "WEPKEY: " );
	for( i = 0 ; i < WDS_WEPKEY_BUF_SIZE ; i++ )
	{
		OS_TPrintf( "%02x", apinfo->wepkey[i] );
	}
	OS_TPrintf( "\n" );
	OS_TPrintf( "================================\n" );
}

static void VBlankIntr(void)
{
	OS_SetIrqCheckFlag(OS_IE_V_BLANK);
}

// ヒープ初期化関数
static void InitializeAllocateSystem(void)
{
    void *tempLo;
    OSHeapHandle hh;

    tempLo = OS_InitAlloc(OS_ARENA_MAIN, OS_GetMainArenaLo(), OS_GetMainArenaHi(), 1);
    OS_SetArenaLo(OS_ARENA_MAIN, tempLo);
    hh = OS_CreateHeap(OS_ARENA_MAIN, OS_GetMainArenaLo(), OS_GetMainArenaHi());
    if (hh < 0)
    {
        OS_Panic("ARM9: Fail to create heap...\n");
    }
    hh = OS_SetCurrentHeap(OS_ARENA_MAIN, hh);
}

static void WDS_Initialize_CB(void *arg)
{
#pragma unused(arg)
	OS_TPrintf("WDS_Initialize_CB\n");
	g_appstate = APP_STATE_WDSSCAN;
	OS_TPrintf("Elapsed time = %llu\n", OS_TicksToMicroSeconds( OS_GetTick() - wdsBegin ) );
}

static void WDS_StartScan_CB(void *arg)
{
#pragma unused(arg)
	OS_TPrintf("WDS_StartScan_CB\n");
	g_appstate = APP_STATE_WDSCOMPLETESCAN;
	OS_TPrintf("Elapsed time = %llu\n", OS_TicksToMicroSeconds( OS_GetTick() - wdsBegin ) );
}

static void WDS_EndScan_CB(void *arg)
{
#pragma unused(arg)
	OS_TPrintf("WDS_EndScan_CB\n");
	g_appstate = APP_STATE_WDSCOMPLETEENDSCAN;
	OS_TPrintf("Elapsed time = %llu\n", OS_TicksToMicroSeconds( OS_GetTick() - wdsBegin ) );
}

static void WDS_End_CB(void *arg)
{
#pragma unused(arg)
	OS_TPrintf("WDS_End_CB\n");
	g_appstate = APP_STATE_WDSCOMPLETEEND;
	OS_TPrintf("Elapsed time = %llu\n", OS_TicksToMicroSeconds( OS_GetTick() - wdsBegin ) );
}