/*

* Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:  

*

*/

#include "stdafx.h"

#include "STATDesktop.h"

#include "STATDesktopDlg.h"

#include "STATManageConnection.h"

#include "ScriptProgressMonitorImp.h"

#include <afxdlgs.h>	//required for CFileDialog

#include <process.h>

#ifdef _DEBUG

#define new DEBUG_NEW

#undef THIS_FILE

static char THIS_FILE[] = __FILE__;

#endif

//////////////////////////////////////////////////////////////////////////

// our thread-safe mechanism

CRITICAL_SECTION CriticalSection;

//////////////////////////////////////////////////////////////////////////

typedef struct _THREADPROC_DATA

{

	HWND windowHandle;

	int handle;

	PROC_RUNSCRIPT ptrRunScript;

}

	THREADPROC_DATA;

////////////////////////////////////////////////

// thread entry point

UINT WINAPI ThreadProc(LPVOID lpParameter)

{

	THREADPROC_DATA *data = reinterpret_cast<THREADPROC_DATA*>(lpParameter);

	HWND windowHandle = data->windowHandle;

	int handle = data->handle;

	PROC_RUNSCRIPT iptrRunScript = data->ptrRunScript;

	delete data;

	data = NULL;

	int	err =	GENERAL_FAILURE;

	{

		// Place this code in scope braces such that the

		// object is cleaned up before the end of the 

		// function.  The endthread methods will not call

		// destructors.

		ScriptProgressMonitorImp scriptMonitor( windowHandle );

		// We do not use the error but it is useful to get it for debugging

		// purposes.

		err =	(iptrRunScript)(handle, &scriptMonitor);

	}

	::_endthreadex(0);	// Use _endthreadex so as not to close thread handle.

	// Never get here.

	return 0;

}

/////////////////////////////////////////////////////////////////////////////

// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog

{

public:

	CAboutDlg(const char* aVersion);

// Dialog Data

	//{{AFX_DATA(CAboutDlg)

	enum { IDD = IDD_ABOUTBOX };

	//}}AFX_DATA

	// ClassWizard generated virtual function overrides

	//{{AFX_VIRTUAL(CAboutDlg)

	protected:

	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

	//}}AFX_VIRTUAL

// Implementation

protected:

	//{{AFX_MSG(CAboutDlg)

	virtual BOOL OnInitDialog();

	//}}AFX_MSG

	DECLARE_MESSAGE_MAP()

private:

	CString iVersion;

};

CAboutDlg::CAboutDlg(const char* aVersion) : CDialog(CAboutDlg::IDD)

{

	iVersion = CString(aVersion);

	//{{AFX_DATA_INIT(CAboutDlg)

	//}}AFX_DATA_INIT

}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)

{

	CDialog::DoDataExchange(pDX);

	//{{AFX_DATA_MAP(CAboutDlg)

	//}}AFX_DATA_MAP

}

CAboutDlg::OnInitDialog()

{

	CDialog::OnInitDialog();

	// set About Box details

	TCHAR tHeading[25];

	_stprintf(tHeading, _T("STAT Version: %s"), iVersion);

	GetDlgItem(IDC_ABOUTHEADING)->SetWindowText(tHeading);

	return TRUE;

}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)

	//{{AFX_MSG_MAP(CAboutDlg)

		// No message handlers

	//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CSTATDesktopDlg dialog

CSTATDesktopDlg::CSTATDesktopDlg(CWnd* pParent /*=NULL*/)

: CDialog(CSTATDesktopDlg::IDD, pParent),

	ihLib(NULL), iHandle(0), hThreadHandle((HANDLE)0), bRunning(false),

	iptrVersion(NULL), iptrGetError(NULL), iptrGetErrorText(NULL), iptrSetConnectionLogging(NULL),

	iptrCloseConnectionLogging(NULL), iptrConnect(NULL), iptrDisconnect(NULL),

	iptrSetCommandDelay(NULL), iptrSetCommandLogging(NULL),

	iptrSetImageVerification(NULL), iptrSendRawCommand(NULL),

	iptrOpenScriptFile(NULL), iptrRunScript(NULL),

	iptrSendCommandFile(NULL), iptrGetCommandCount(NULL), iptrGetCurrentCommandNumber(NULL),

	iptrStopProcessing(NULL),

	iptrGetSnapShot(NULL), iMessageReporter(NULL)

{

	//{{AFX_DATA_INIT(CSTATDesktopDlg)

	//}}AFX_DATA_INIT

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

}

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

void CSTATDesktopDlg::DoDataExchange(CDataExchange* pDX)

{

	CDialog::DoDataExchange(pDX);

	//{{AFX_DATA_MAP(CSTATDesktopDlg)

	DDX_Control(pDX, IDC_CONNECTION, m_Connection);

	DDX_Control(pDX, IDC_VIEW, m_ViewLog);

	DDX_Control(pDX, IDC_PROGRESS, m_Progress);

	DDX_Control(pDX, IDC_BROWSELOG, m_BrowseLog);

	DDX_Control(pDX, IDC_APPEND, m_Append);

	DDX_Control(pDX, IDC_BROWSEIMAGE, m_BrowseImage);

	DDX_Control(pDX, IDC_TIMEDELAY, m_Delay);

	DDX_Control(pDX, IDC_OUTPUT, m_Output);

	DDX_Control(pDX, IDC_IMAGEVERIFY, m_VerifyImage);

	DDX_Control(pDX, IDC_EDITLOGNAME, m_LogFileName);

	DDX_Control(pDX, IDC_DIFFLOGNAME, m_LogToFile);

	DDX_Control(pDX, IDC_IMAGEBOX, m_ImageBox);

	DDX_Control(pDX, IDC_SCRIPT, m_Script);

	DDX_Control(pDX, IDC_LOCATION, m_ImageLocation);

	DDX_Control(pDX, IDC_SPEEDSLIDER, m_SliderPos);

	DDX_Control(pDX, IDC_PERCENTAGEDIFF, m_PercentageDiff);

	//}}AFX_DATA_MAP

}

BEGIN_MESSAGE_MAP(CSTATDesktopDlg, CDialog)

	//{{AFX_MSG_MAP(CSTATDesktopDlg)

	ON_WM_SYSCOMMAND()

	ON_WM_PAINT()

	ON_WM_QUERYDRAGICON()

	ON_BN_CLICKED(IDC_BROWSE, OnBrowse)

	ON_BN_CLICKED(IDEDITSCRIPT, OnEditscript)

	ON_BN_CLICKED(IDEXIT, OnExit)

	ON_BN_CLICKED(IDRUN, OnRun)

	ON_BN_CLICKED(IDC_DIFFLOGNAME, OnLogToFile)

	ON_NOTIFY(NM_RELEASEDCAPTURE, IDC_SPEEDSLIDER, OnReleasedcaptureSpeedslider)

	ON_EN_CHANGE(IDC_TIMEDELAY, OnChangeTimedelay)

	ON_BN_CLICKED(IDC_BROWSEIMAGE, OnBrowseimage)

	ON_BN_CLICKED(IDC_IMAGEVERIFY, OnImageverify)

	ON_BN_CLICKED(IDC_BROWSELOG, OnBrowseLog)

	ON_BN_CLICKED(IDC_VIEW, OnView)

	ON_BN_CLICKED(IDC_MANAGE, OnManage)

	ON_WM_DESTROY()

	ON_MESSAGE(WM_DONE_COMMAND, OnDoneCommand)

	ON_MESSAGE(WM_DONE_SCRIPT, OnDoneScript)

	ON_MESSAGE(WM_SHOW_LOG_MESSAGE, OnLogMesage)

	//}}AFX_MSG_MAP

END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////

// CSTATDesktopDlg message handlers

BOOL CSTATDesktopDlg::OnInitDialog()

{

	CDialog::OnInitDialog();

	// IDM_ABOUTBOX must be in the system command range.

	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);

	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);

	if (pSysMenu != NULL)

	{

		CString strAboutMenu;

		strAboutMenu.LoadString(IDS_ABOUTBOX);

		if (!strAboutMenu.IsEmpty())

		{

			pSysMenu->AppendMenu(MF_SEPARATOR);

			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);

		}

	}

	SetIcon(m_hIcon, TRUE);			// Set big icon

	SetIcon(m_hIcon, FALSE);		// Set small icon

	// Load the STAT DLL library.

	static char	DLLPath[MAX_PATH];

#ifdef _DEBUG

	// Load the debug DLL from our current working folder.

	sprintf( DLLPath, "statd.dll" );

#else // _DEBUG

	// Load the current build of STAT DLL from the default 

	// STAT folder (probably '\epoc32\tools\STAT').

	sprintf( DLLPath, "%s\\%s", DLLFolder, DLLName );

#endif // _DEBUG

	ihLib =	::LoadLibrary( DLLPath );

	if( ihLib  == NULL )

	{

		::AfxMessageBox( "Failed to load dynamic library.", MB_ICONERROR );

		PostMessage( WM_CLOSE, 0, 0 );

	}

	else

	{

		iptrVersion =	reinterpret_cast<PROC_VERSION>(::GetProcAddress( ihLib, ProcVersion ));

		iptrGetError =	reinterpret_cast<PROC_GETERROR>(::GetProcAddress( ihLib, ProcGetError ));

		iptrGetErrorText =	reinterpret_cast<PROC_GETERRORTEXT>(::GetProcAddress( ihLib, ProcGetErrorText ));

		iptrSetConnectionLogging =	reinterpret_cast<PROC_SETCONNECTIONLOGGING>(::GetProcAddress( ihLib, ProcSetConnectionLogging ));

		iptrCloseConnectionLogging =	reinterpret_cast<PROC_CLOSECONNECTIONLOGGING>(::GetProcAddress( ihLib, ProcCloseConnectionLogging ));

		iptrConnect =	reinterpret_cast<PROC_CONNECT>(::GetProcAddress( ihLib, ProcConnect ));

		iptrDisconnect =	reinterpret_cast<PROC_DISCONNECT>(::GetProcAddress( ihLib, ProcDisconnect ));

		iptrSetCommandDelay =	reinterpret_cast<PROC_SETCOMMANDDELAY>(::GetProcAddress( ihLib, ProcSetCommandDelay ));

		iptrSetCommandLogging =	reinterpret_cast<PROC_SETCOMMANDLOGGING>(::GetProcAddress( ihLib, ProcSetCommandLogging ));

		iptrSetImageVerification =	reinterpret_cast<PROC_SETIMAGEVERIFICATION>(::GetProcAddress( ihLib, ProcSetImageVerification ));

		iptrOpenScriptFile =	reinterpret_cast<PROC_OPENSCRIPTFILE>(::GetProcAddress( ihLib, ProcOpenScriptFile ));

		iptrRunScript =	reinterpret_cast<PROC_RUNSCRIPT>(::GetProcAddress( ihLib, ProcRunScript ));

		iptrSendRawCommand =	reinterpret_cast<PROC_SENDRAWCOMMAND>(::GetProcAddress( ihLib, ProcSendRawCommand ));

		iptrSendCommandFile =	reinterpret_cast<PROC_SENDCOMMANDFILE>(::GetProcAddress( ihLib, ProcSendCommandFile ));

		iptrGetCommandCount =	reinterpret_cast<PROC_GETCOMMANDCOUNT>(::GetProcAddress( ihLib, ProcGetCommandCount ));

		iptrGetCurrentCommandNumber =	reinterpret_cast<PROC_GETCURRENTCOMMANDNUMBER>(::GetProcAddress( ihLib, ProcGetCurrentCommandNumber ));

		iptrStopProcessing =	reinterpret_cast<PROC_STOPPROCESSING>(::GetProcAddress( ihLib, ProcStopProcessing ));

		iptrGetSnapShot =	reinterpret_cast<PROC_GETSNAPSHOT>(::GetProcAddress( ihLib, ProcGetSnapShot ));

		if( ( NULL == iptrVersion ) ||

			( NULL == iptrGetError ) ||

			( NULL == iptrGetErrorText ) ||

			( NULL == iptrSetConnectionLogging ) ||

			( NULL == iptrCloseConnectionLogging ) ||

			( NULL == iptrConnect ) ||

			( NULL == iptrDisconnect ) ||

			( NULL == iptrSetCommandDelay ) ||

			( NULL == iptrSetCommandLogging ) ||

			( NULL == iptrSetImageVerification ) ||

			( NULL == iptrOpenScriptFile ) ||

			( NULL == iptrRunScript ) ||

			( NULL == iptrSendRawCommand ) ||

			( NULL == iptrSendCommandFile ) ||

			( NULL == iptrGetCommandCount ) ||

			( NULL == iptrGetCurrentCommandNumber ) ||

			( NULL == iptrStopProcessing ) )

		{

			::AfxMessageBox( "Failed to load a function address.", MB_ICONERROR );

			PostMessage( WM_CLOSE, 0, 0 );

		}

	}

	iMessageReporter =	new MessageReporterImp( GetSafeHwnd() );

	iLastVerify = 0;

	iLastLog = 0;

	statIniFile.SetIniFileName(STAT_INI_NAME);

	// read in previous settings

	ReadSettings();

	InitializeCriticalSection(&CriticalSection);

	return TRUE;  // return TRUE  unless you set the focus to a control

}

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

void CSTATDesktopDlg::OnSysCommand(UINT nID, LPARAM lParam)

{

	if ((nID & 0xFFF0) == IDM_ABOUTBOX)

	{

		CAboutDlg dlgAbout(iptrVersion());

		dlgAbout.DoModal();

	}

	else

		CDialog::OnSysCommand(nID, lParam);

}

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

// If you add a minimize button to your dialog, you will need the code below

//  to draw the icon.  For MFC applications using the document/view model,

//  this is automatically done for you by the framework.

void CSTATDesktopDlg::OnPaint() 

{

	if (IsIconic())

	{

		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle

		int cxIcon = GetSystemMetrics(SM_CXICON);

		int cyIcon = GetSystemMetrics(SM_CYICON);

		CRect rect;

		GetClientRect(&rect);

		int x = (rect.Width() - cxIcon + 1) / 2;

		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon

		dc.DrawIcon(x, y, m_hIcon);

	}

	else

		CDialog::OnPaint();

}

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

HCURSOR CSTATDesktopDlg::OnQueryDragIcon()

{

	return (HCURSOR) m_hIcon;

}

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

void CSTATDesktopDlg::OnBrowse() 

{

	// use the folder of the current script as a starting point

	m_Script.GetWindowText(szBuffer, sizeof(szBuffer));

	CString buf = szBuffer;

	if (!buf.IsEmpty())

	{

		int index = buf.ReverseFind('\\');

		if (index != -1)

		{

			CString folder = buf.Left(index);

			SetCurrentDirectory(folder.GetBuffer(0));

		}

	}

	CFileDialog dlgStatFileOpen(TRUE, 

								NULL, 

								NULL, 

								OFN_FILEMUSTEXIST | OFN_PATHMUSTEXIST, 

								_T("Script Files (*.txt)|*.txt|All Files (*.*)|*.*||"));

	// new script file to use

	if (dlgStatFileOpen.DoModal() == IDOK)	//when ok is pressed

		m_Script.SetWindowText(dlgStatFileOpen.GetPathName());

}

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

//Frees memory and releases COM stuff etc...

void CSTATDesktopDlg::OnCancel() 

{

	OnOK();

}

void CSTATDesktopDlg::OnExit() 

{

	SaveSettings();

	OnOK();

}

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

//when the Run button is pressed

void CSTATDesktopDlg::OnRun()

{

	if (bRunning)

	{

		CWaitCursor oWait;

		Message("Signalling current script to stop.  Please wait...");

		(iptrStopProcessing)(iHandle);

	}

	else

	{

		m_Output.ResetContent();

		Message("*** STAT LOG ***");

		RunIt();

	}

}

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

// main work happens here

void CSTATDesktopDlg::RunIt()

{

	// We should use GetData( true ) here to update all the internal data with the

	// text in the dialog controls instead of GetWindowText.

	CWaitCursor oWait;

	int ret = ITS_OK;

	bRunning = true;

	// make sure we have something to work with

	(*szBuffer) = (TCHAR)0;

	m_Script.GetWindowText(szBuffer, sizeof(szBuffer));

	if (!(*szBuffer))

	{

		CleanUp(_T("One or more required fields in the dialog are empty"));

		return;

	}

	// initialise back end

	if((ret = ConnectToEngine()) != ITS_OK)

	{

		CString msg;

		if (ret == E_NOCONNECTION)

			msg = _T("No connection specified");

		else

			msg = (iptrGetError)(iHandle);

		CleanUp(msg);

		return;

	}

	// turn on logging

	if(m_LogToFile.GetCheck())

	{

		m_LogFileName.GetWindowText(szBuffer, sizeof(szBuffer));

		CString cBuffer = szBuffer;

		if (!cBuffer.IsEmpty() && (_tcscmp(szBuffer, _T("<date-time>")) != 0))

			MessageFormat("Logging to %s Append: %d...", szBuffer, m_Append.GetCheck());

		else

		{

			if (cBuffer.IsEmpty())

				m_LogFileName.SetWindowText(_T("<date-time>"));

			else

				cBuffer.Empty();

			MessageFormat("Logging to default file, Append: %d...", m_Append.GetCheck());

		}

		ret = (iptrSetCommandLogging)(iHandle, cBuffer, iMessageReporter,

										EVerbose, (m_Append.GetCheck() > 0), NULL, NULL );

		if (ret != LOG_FILE_OK)

		{

			CleanUp(_T("Logging could not be set"));

			return;

		}

	}

	else

	{

		CString cBuffer;

		ret = (iptrSetCommandLogging)(iHandle, NULL, iMessageReporter,

										EVerbose, (m_Append.GetCheck() > 0), NULL, NULL );

		if (ret != LOG_FILE_OK)

		{

			CleanUp(_T("Logging could not be set"));

			return;

		}

	}

	// set the script execution speed

	GetDlgItem(IDC_TIMEDELAY)->GetWindowText(szBuffer, sizeof(szBuffer));

	(iptrSetCommandDelay)(iHandle, _ttoi(szBuffer));

	MessageFormat("Command delay set to %d...", _ttoi(szBuffer));

	// image verification

	if(m_VerifyImage.GetCheck() && !SetImageVerification())

	{

		CleanUp(_T("Image verification could not be set"));

		return;

	}

	// get list of commands in script

	m_Script.GetWindowText(szBuffer, sizeof(szBuffer));

	int iCount = 0;

	ret = (iptrGetCommandCount)(iHandle, szBuffer, &iCount);

	if(ret != ITS_OK)	//attempt to open as commands if file cannot be opened

	{

		CleanUp((iptrGetErrorText)(iHandle, ret));

		return;

	}

	// set the progress bar