#include <e32std.h>

#include <e32test.h>

#include <e32math.h>

#include <cntdb.h>

#include <cntitem.h>

#include <cntfldst.h>

#include "t_utils2.h"

#include "t_bench.h"

//

// Configuration.

//

#define __TxEST

#define __VxERBOSE

//

// Constants.

//

_LIT(KTestName,"t_bench");

_LIT(KDbFileName,"c:contacts.cdb");

_LIT(KLogFileName,"t_bench.log");

_LIT(KBackSpace,"\x08");

_LIT(KVisualCounterFormat,"%S%d");

_LIT(KTimeProfileFormat,"Time taken (secs)");

_LIT(KSystemMemoryFormat,"System memory used (Kb)");

_LIT(KThreadMemoryFormat,"This thread memory used (Kb)");

_LIT(KFileSizeText,"File size (bytes)");

_LIT(KAverageTimeText,"Average time taken (secs)");

_LIT(KResultFormatTInt,"Result #%d.%d %S [%d]");

_LIT(KResultFormatTReal,"Result #%d.%d %S [%f]");

_LIT(KThreadNameFormat,"T_BenchCli%2d");

_LIT(KPhoneMatchFail,"6666666666666666");

_LIT(KNameMatchFail,"xxxxxxxxxxxxxxxx");

_LIT(KAsyncFindMatch,"a");

_LIT(KNewLine,"\n");

_LIT(KTextDefSeparator,"\t");

#ifdef __VERBOSE

_LIT(KContactSummaryFormat,"%d\t%S\t%S\t%S");

_LIT(KAscendingSortOrder,"ascending");

_LIT(KDescendingSortOrder,"descending");

_LIT(KProfileResult,"Took %d\tmicrosecs for %d iteration(s)");

#endif

const TInt KNumSortClients=10;

const TInt KClientStackSize=KDefaultStackSize;

const TInt KClientHeapSize=0x20000;

#ifdef __TEST

const TInt KNumTypicalContacts=10;

const TInt KNumAtypicalContacts=2;

const TInt KNumTypicalContactsCompact=0;

const TInt KNumAtypicalContactsCompact=0;

#else

const TInt KNumTypicalContacts=100;

const TInt KNumAtypicalContacts=50;

const TInt KNumTypicalContactsCompact=150;

const TInt KNumAtypicalContactsCompact=50;

#endif // __TEST

//

// CBenchMarker.

//

CBenchMarker::CBenchMarker() : iTest(KTestName),iVisualCounter(-1),iNumTypicalContacts(-1),iNumAtypicalContacts(-1),iNumTypicalContactsCompact(-1),iNumAtypicalContactsCompact(-1)

	{

	}

CBenchMarker::~CBenchMarker()

	{

	delete iGenerator;

	delete iDb;

	delete iTextDef;

	delete iLog;

	iTest.Close();

	iFs.Close();

	}

void CBenchMarker::ConstructL(const TDesC& aCommandLine)

	{

	User::LeaveIfError(iFs.Connect());

	DecodeCommandLineL(aCommandLine);

	iLog=CLog::NewL(iTest,iLogFileName);

	iGenerator=CRandomContactGenerator::NewL();

	}

void CBenchMarker::DecodeCommandLineL(const TDesC& aCommandLine)

	{

	_LIT(KArgContactsToAdd,"-a");

	_LIT(KArgContactsToCompact,"-c");

	_LIT(KArgDatabaseFileName,"-d");

	_LIT(KArgNoDataGenerationTests,"-g");

	_LIT(KArgLogFile,"-l");

	_LIT(KArgHelp,"-h");

	TLex cl(aCommandLine);

	while (cl.Remainder().Length()>0)

		{

		TPtrC token=cl.NextToken();

		if (token.CompareF(KArgContactsToAdd)==0)

			{

			cl.SkipSpace();

			User::LeaveIfError(cl.Val(iNumTypicalContacts));

			cl.SkipSpace();

			User::LeaveIfError(cl.Val(iNumAtypicalContacts));

			}

		else if (token.CompareF(KArgContactsToCompact)==0)

			{

			cl.SkipSpace();

			User::LeaveIfError(cl.Val(iNumTypicalContactsCompact));

			cl.SkipSpace();

			User::LeaveIfError(cl.Val(iNumAtypicalContactsCompact));

			}

		else if (token.CompareF(KArgDatabaseFileName)==0)

			{

			iDbFileName=cl.NextToken();

			}

		else if (token.CompareF(KArgNoDataGenerationTests)==0)

			{

			iNoDataGenerationTests=ETrue;

			}

		else if (token.CompareF(KArgLogFile)==0)

			{

			iLogFileName=cl.NextToken();

			}

		else if (token.CompareF(KArgHelp)==0)

			{

			iTest.Printf(_L("Usage t_bench [options]\n\n"));

			iTest.Printf(_L("-a <num_typical> <num_atypical> (default: %d %d)\n"),KNumTypicalContacts,KNumAtypicalContacts);

			iTest.Printf(_L("-c <num_typical_comp> <num_atypical_comp> (default: %d %d)\n"),KNumTypicalContactsCompact,KNumAtypicalContactsCompact);

			iTest.Printf(_L("-d <db_name> (default: %S)\n"),&KDbFileName);

			iTest.Printf(_L("-l <log_name> (default: %S)\n"),&KLogFileName);

			iTest.Printf(_L("-g skip data generation tests\n"));

			iTest.Printf(_L("\n<hit any key>"));

			iTest.Getch();

			User::Leave(KErrNone);

			}

		}

	if (iDbFileName.Length()==0)

		{

		iDbFileName=KDbFileName;

		}

	if (iLogFileName.Length()==0)

		{

		iLogFileName=KLogFileName;

		}

	if (iNoDataGenerationTests)

		{

		TUint att;

		if (iFs.Att(iDbFileName,att)==KErrNotFound)

			{

			iTest.Printf(_L("-g switch ignored, because %S not found\n"),&iDbFileName);

			iNoDataGenerationTests=EFalse;

			}

		}

	if (iNumTypicalContacts<0)

		{

		iNumTypicalContacts=KNumTypicalContacts;

		}

	if (iNumAtypicalContacts<0)

		{

		iNumAtypicalContacts=KNumAtypicalContacts;

		}

	if (iNumTypicalContactsCompact<0)

		{

		iNumTypicalContactsCompact=KNumTypicalContactsCompact;

		}

	if (iNumAtypicalContactsCompact<0)

		{

		iNumAtypicalContactsCompact=KNumAtypicalContactsCompact;

		}

	}

void CBenchMarker::RunL()

	{

	iTest.Printf(_L("Starting CntModel benchmarks...\n")); // This forces the console to get loaded under WINS - would otherwise scew profiles.

	for (TInt i=0;i<ENumTests;++i)

		{

		StartProfile();

		TInt timeMinorNumber=0;

		TRAPD(err,timeMinorNumber=DoTestL(TTest(i)));

		if (err)

			{

			iLog->LogLine(_L("Test %d left with %d"),i,err);

			break;

			}

		LogResult(i,timeMinorNumber,KTimeProfileFormat,EndProfile());

		User::CompressAllHeaps();

		TMemoryInfoV1Buf memoryBuf;

		UserHal::MemoryInfo(memoryBuf);

		TMemoryInfoV1 memory(memoryBuf());

		const TInt totalK=memory.iTotalRamInBytes>>10;

		const TInt freeK=memory.iFreeRamInBytes>>10;

		const TInt usedK=totalK-freeK;

		LogResult(i,timeMinorNumber+1,KSystemMemoryFormat,usedK);

		TInt allocSize;

		User::Heap().AllocSize(allocSize);;

		allocSize>>=10;

		LogResult(i,timeMinorNumber+2,KThreadMemoryFormat,allocSize);

		}

	iTest.Printf(_L("Dump log to comm port? (y/n) "));

	TKeyCode key=iTest.Getch();

	if (key=='y' || key=='Y')

		{

writelog:

		iLog->WriteLogToCommPortL();

		}

	iTest.Printf(_L("\nAgain? (y/n)"));

	key=iTest.Getch();

	if (key=='y' || key=='Y')

		{

		goto writelog;

		}

	}

TInt CBenchMarker::DoTestL(TTest aTest)

	{

	TInt timeMinorNumber=0;

	iLog->Log(_L("Test   #%d   "),aTest);

	TPtrC match;

	switch (aTest)

		{

		case EOpenNewDb:

			iLog->LogLine(_L("Open new database"));

			if (iNoDataGenerationTests)

				{

				iDb=CContactDatabase::OpenL(iDbFileName);

				}

			else

				{

				iDb=CContactDatabase::ReplaceL(iDbFileName);

				LogResult(aTest,0,KFileSizeText,DbFileSize());

				timeMinorNumber=1;

				}

			iGenerator->SetDbL(*iDb);

			break;

		case ECreateSecondaryClients:

			iLog->LogLine(_L("Create %d secondary client threads"),KNumSecondaryClients);

			CreateSecondaryClients();

			break;

		case EAddContacts:

			iLog->LogLine(_L("Add %d typical and %d atypical contacts"),iNumTypicalContacts,iNumAtypicalContacts);

			if (!iNoDataGenerationTests)

				{

				LogResult(aTest,0,KAverageTimeText,AddContactsL());

				LogResult(aTest,1,KFileSizeText,DbFileSize());

				timeMinorNumber=2;

				}

			break;

		case ECompactDb:

			iLog->LogLine(_L("Compact database"));

			if (!iNoDataGenerationTests)

				{

				iDb->CompactL();

				LogResult(aTest,0,KFileSizeText,DbFileSize());

				timeMinorNumber=1;

				}

			break;

		case EAddAndCompact:

			iLog->LogLine(_L("Add %d typical and %d atypical contacts (compact after each)"),iNumTypicalContactsCompact,iNumAtypicalContactsCompact);

			if (!iNoDataGenerationTests)

				{

				LogResult(aTest,0,KAverageTimeText,AddContactsCompactL());

				LogResult(aTest,1,KFileSizeText,DbFileSize());

				timeMinorNumber=2;

				}

			break;

		case EGetSortedItems:

			iLog->LogLine(_L("Get sorted id array by firstname, lastname, company name"));

			DoSortL();

			break;

		case ELogContactSummary:

			iLog->LogLine(_L("Log contact summary"));

			LogContactSummaryL(EFalse);

			break;

		case ELogContactSummaryFast:

			iLog->LogLine(_L("Log contact summary using TContactTextDefItem"));

			LogContactSummaryL(ETrue);

			break;

		case EOpenRContactView:

			iLog->LogLine(_L("Open RContactView"));

			OpenRContactViewL();

			break;

		case ESetSortOrderOfRContactView:

			iLog->LogLine(_L("Set order of RContactView"));

			SetSortOrderOfRContactViewL();

			break;

		case EGetSortOrderOfRContactView:

			iLog->LogLine(_L("Get order of RContactView"));

			GetSortOrderOfRContactViewL();

			break;

		case ELogContactSummaryWithRContactView:

			iLog->LogLine(_L("Log contact summary using RContactView"));

			LogContactSummaryFromRContactViewL(EFalse);

			break;

		case ELogContactSummaryFastWithRContactView:

			iLog->LogLine(_L("Log contact summary using RContactView and TContactTextDefItem"));

			LogContactSummaryFromRContactViewL(ETrue);

			break;

		case EFindWithRContactView:

			iLog->LogLine(_L("Find TContactItemId from RContactView"));

			FindInRContactViewL();

			break;

		case ECloseRContactView:

			iLog->LogLine(_L("Close RContactView"));

#ifdef __USE_NEW_INTERFACES

			iSortedIdList.Close();

#endif

			break;

		case EFindFromLargeFieldSetSuceed:

			match.Set(iGenerator->NameMatch());

			iLog->LogLine(_L("FindLC %S using large field set (will suceed)"),&match);

			FindFromLargeFieldSetL(match);

			break;

		case EFindFromLargeFieldSetFail:

			iLog->LogLine(_L("FindLC %S using large field set (will fail)"),&KNameMatchFail);

			FindFromLargeFieldSetL(KNameMatchFail);

			break;

		case EFindFromSmallFieldSetSuceed:

			match.Set(iGenerator->NameMatch());

			iLog->LogLine(_L("FindLC %S using small field set (will suceed)"),&match);

			FindFromSmallFieldSetL(match);

			break;

		case EFindFromSmallFieldSetFail:

			iLog->LogLine(_L("FindLC %S using small field set (will fail)"),&KNameMatchFail);

			FindFromSmallFieldSetL(KNameMatchFail);

			break;

		case EFindPhoneNumberSuceed:

			match.Set(iGenerator->PhoneMatch());

			iLog->LogLine(_L("FindLC %S using just phone field (will suceed)"),&match);

			FindPhoneNumberL(match);

			break;

		case EFindPhoneNumberFail:

			iLog->LogLine(_L("FindLC %S using just phone field (will fail)"),&KPhoneMatchFail);

			FindPhoneNumberL(KPhoneMatchFail);

			break;

		case EFindEmailAddressSuceed:

			match.Set(iGenerator->EmailMatch());

			iLog->LogLine(_L("FindLC %S using just email field (will suceed)"),&match);

			FindEmailAddressL(match);

			break;

		case EFindEmailAddressFail:

			iLog->LogLine(_L("FindLC %S using just email field (will fail)"),&KNameMatchFail);

			FindEmailAddressL(KNameMatchFail);

			break;

		case EFindAsyncFromLargeFieldSet:

			iLog->LogLine(_L("FindAsyncL %S using large field set"),&KAsyncFindMatch);

			FindAsyncFromLargeFieldSetL(KAsyncFindMatch);

			break;

		case EFindAsyncFromSmallFieldSet:

			iLog->LogLine(_L("FindAsyncL %S using small field set"),&KAsyncFindMatch);

			FindAsyncFromSmallFieldSetL(KAsyncFindMatch);

			break;

		case ECloseDb:

			iLog->LogLine(_L("Close database"));

			delete iDb;

			iDb=NULL;

			break;

		case ECloseSecondaryClients:

			iLog->LogLine(_L("Close secondary clients"));

			CloseSecondaryClients();

			break;

		case EOpenExistingDb:

			iLog->LogLine(_L("Open existing database"));

			iDb=CContactDatabase::OpenL(iDbFileName);

			break;

		case EMultiClientSort:

			iLog->LogLine(_L("%d client simultaneous sort"),KNumSortClients);

			CreateSortClients();

			break;

		case ENumTests:

			ASSERT(EFalse);

			break;

		}

	return timeMinorNumber;

	}

void CBenchMarker::StartProfile()

	{

	CCntTest::ProfileReset(0,1); // reset profiles 0-1

	CCntTest::ProfileStart(0);

	}

TReal CBenchMarker::EndProfile()

	{

	CCntTest::ProfileEnd(0);

	TCntProfile profile[1];

	CCntTest::ProfileResult(profile,0,1);

	return TReal(profile[0].iTime/1000000.0);

	}

void CBenchMarker::StartAverageProfile()

	{

	iAverageProfileCounter=0;

	iNumAverageProfiles=0;

	CCntTest::ProfileReset(1,1);

	CCntTest::ProfileStart(1);

	}

void CBenchMarker::UpdateAverageProfile(TInt aNumIterations)

	{

	CCntTest::ProfileEnd(1);

	TCntProfile profile[1];

	CCntTest::ProfileResult(profile,1,1);

	iAverageProfileCounter+=profile[0].iTime;

	iNumAverageProfiles+=aNumIterations;

	CCntTest::ProfileReset(1,1);

	CCntTest::ProfileStart(1);

#ifdef __VERBOSE

	iLog->LogLineNoEcho(KProfileResult,profile[0].iTime,aNumIterations);	

#endif

	}

TReal CBenchMarker::EndAverageProfile()

	{

	CCntTest::ProfileEnd(1);

	if (iNumAverageProfiles>0)

		{

		return TReal((iAverageProfileCounter/iNumAverageProfiles)/1000000.0);

		}

	return TReal(0.0);

	}

TReal CBenchMarker::AddContactsL()

	{

	iTest.Printf(_L("Adding "));

	StartAverageProfile();

	TInt ii;

	for (ii=0;ii<iNumTypicalContacts;++ii)

		{

		IncVisualCounter();

		iGenerator->AddTypicalRandomContactL();

		UpdateAverageProfile(1);

		}

	for (ii=0;ii<iNumAtypicalContacts;++ii)

		{

		IncVisualCounter();

		iGenerator->AddAtypicalRandomContactL();

		UpdateAverageProfile(1);

		}

	EndVisualCounter();

	return EndAverageProfile();

	}

TReal CBenchMarker::AddContactsCompactL()

	{

	iTest.Printf(_L("Adding "));

	StartAverageProfile();

	TInt ii;

	for (ii=0;ii<iNumTypicalContactsCompact;++ii)

		{

		IncVisualCounter();

		iGenerator->AddTypicalRandomContactL();

		iDb->CompactL();

		UpdateAverageProfile(1);

		}

	for (ii=0;ii<iNumAtypicalContactsCompact;++ii)

		{

		IncVisualCounter();

		iGenerator->AddAtypicalRandomContactL();

		iDb->CompactL();

		UpdateAverageProfile(1);

		}

	EndVisualCounter();

	return EndAverageProfile();

	}

#ifdef __USE_NEW_INTERFACES

void CBenchMarker::LogContactSummaryFromRContactViewL(TBool aFaster)

	{

	iTest.Printf(_L("Logging "));

	const TInt numContacts(iSortedIdList.Count());

	for (TInt ii=0;ii<numContacts;++ii)

		{

		IncVisualCounter();

		LogContactSummaryL(aFaster,iSortedIdList.GetL(ii));

		}

	EndVisualCounter();