// Copyright (c) 1997-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 "T_PerfDbCreator.h"

// Group constants

const TInt   KGroups=10;

const TInt   KGroupSize=25;

const TText * const KGroupNames[KGroups]=

	{

	_S("Family"),

	_S("Friends"),

	_S("Work"),

	_S("Group 4"),

	_S("Group 5"),

	_S("Group 6"),

	_S("Group 7"),

	_S("Group 8"),

	_S("Group 9"),

	_S("Group 10")

	};

//************************************************************************************************************

//   CPerfDbCreator

//************************************************************************************************************

CPerfDbCreator* CPerfDbCreator::NewL()

   {

	CPerfDbCreator* self = new(ELeave)CPerfDbCreator();

	CleanupStack::PushL(self);

	self->ConstructL();

	CleanupStack::Pop(self);

	return self;

   } 

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

CPerfDbCreator::CPerfDbCreator()

:  iSeed(KSeed)

   {

   }

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

void CPerfDbCreator::ConstructL()

   {

   }

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

CPerfDbCreator::~CPerfDbCreator()

   {

	if (iContactsDb != NULL)

	   delete iContactsDb;

	if (iTemplate != NULL)

	   delete iTemplate;

   }

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

// The contacts to be duplicated are not really random

void CPerfDbCreator::GenContactsL(TContactLevel aLevel, TInt aSingles, TInt aDuplicates)

   {

   TInt            i, dupDiv, duplicated=0;

   TReal           dupDivReal;

   TContactItemId  contactItemId;

   CContactItem*   contactItem;

   if (aDuplicates != 0)

      {

      Math::Round(dupDivReal, (TReal)aSingles/(TReal)aDuplicates, 0);

      dupDiv= (TInt) dupDivReal;

      }

   else

      dupDiv=0xffff;

   for (i=0; i < aSingles; i++)

      {

      contactItemId = AddContactL(aLevel);

      // duplicate every dupDiv contact, but do not exceed aDuplicates

      if ((duplicated<aDuplicates) && ((i % dupDiv) == 0)) 

         {

         duplicated++;

         contactItem = iContactsDb->ReadContactL(contactItemId);

         iContactsDb->AddNewContactL(*contactItem);

         delete contactItem;

         }

      // Compact after every KCompactEvery added contact

      if ((i % KCompactEvery) == 0)

         iContactsDb->CompactL();

      }

   }

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

void CPerfDbCreator::GenDbL(TUserProfile aProfile, TInt aCount, TInt aDupRatio,  TBool aAddSeedPrefix, const TDesC &aFileName, TBool aWithGroups)

   {

   TReal  levelContacts[KLevels], levelDuplicates[KLevels];

   TReal  totalContacts, totalDuplicates;

   TInt   i;

   iAddSeedPrefix=aAddSeedPrefix;

   iSeed=KSeed;

   iContactsDb = CContactDatabase::ReplaceL(aFileName);

   // Template will be used later, to add new contacts

   iTemplate=static_cast<CContactTemplate*>(iContactsDb->ReadContactL(iContactsDb->TemplateId()));

   // Calculate number of contacts and duplicates for each level

   totalDuplicates= (aCount*aDupRatio)/100/2;

   totalContacts=   aCount-totalDuplicates;

   for (i=0; i<KLevels; i++)

      {

      levelContacts[i]=(totalContacts*KRatios[aProfile][i])/100;

      levelDuplicates[i]=(totalDuplicates*KRatios[aProfile][i])/100;

      Math::Round(levelContacts[i], levelContacts[i], 0);

      Math::Round(levelDuplicates[i], levelDuplicates[i], 0);

      }

   // Generate contacts and duplicates for each level

	for (i=0; i<KLevels; i++)

		{

		GenContactsL(static_cast<TContactLevel>(i), (TInt) levelContacts[i], (TInt) levelDuplicates[i]);

		}

	// Create groups

	if (aWithGroups)

		{

		for (i = 0; i < KGroups; i++)

			{

			AddGroupL(TPtrC(KGroupNames[i]));

			}

		}

   // And clanup

	delete iContactsDb;

	iContactsDb = NULL;

	delete iTemplate;

	iTemplate = NULL;

   }

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

void CPerfDbCreator::GenAllTheSameDbL(TContactLevel aLevel, TInt aCount, TBool aAddSeedPrefix, const TDesC &aFileName)

   {

   TInt            i;

   TContactItemId  contactItemId;

   CContactItem*   contactItem;   

   iAddSeedPrefix=aAddSeedPrefix;

   iSeed=KSeed;

   iContactsDb = CContactDatabase::ReplaceL(aFileName);

   // Template will be used later, to add new contacts

   iTemplate=static_cast<CContactTemplate*>(iContactsDb->ReadContactL(iContactsDb->TemplateId()));

   // Generate database

   contactItemId = AddContactL(aLevel); // contact 0

   for (i=1; i < aCount; i++)          // contacts 1,2,3...

      {

      contactItem = iContactsDb->ReadContactL(contactItemId);

      iContactsDb->AddNewContactL(*contactItem);

      delete contactItem;

      // Compact after every KCompactEvery added contact

      if ((i % KCompactEvery) == 0)

        iContactsDb->CompactL();

      }

   // And clanup

	delete iContactsDb;

	iContactsDb = NULL;

	delete iTemplate;

	iTemplate = NULL;   

   }

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

TContactItemId CPerfDbCreator::AddContactL(TContactLevel aLevel)

	{

	CPerfTestContact* contact=CPerfTestContact::NewLC(*iTemplate);

	HBufC* buf=HBufC::NewLC(256);

	TPtr bufPtr=buf->Des();

   // Always in the contact

 	SetRandomAlphaString(bufPtr,32, iAddSeedPrefix);

  	contact->SetFirstNameL(bufPtr);

  	SetRandomAlphaString(bufPtr,32, iAddSeedPrefix);

  	contact->SetLastNameL(bufPtr);

  	SetRandomNumericString(bufPtr,24);

  	contact->SetHomePhoneL(bufPtr);

   if (aLevel == ELevel2)

	   {

      SetRandomAlphaNumericString(bufPtr,32);

	   contact->SetHomeStreetL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetHomeTownL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetHomeCountyL(bufPtr);

	   SetRandomAlphaNumericString(bufPtr,16);

	   contact->SetHomePostCodeL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetHomeCountryL(bufPtr);

	   }

   if (aLevel >= ELevel2)

      {

      SetRandomAlphaNumericString(bufPtr,48);

	   contact->SetHomeEmailAddressL(bufPtr);	   

      }

   if (aLevel == ELevel3)

	   {	   

      SetRandomAlphaString(bufPtr,32, iAddSeedPrefix);

	   contact->SetCompanyNameL(bufPtr);

      SetRandomAlphaNumericString(bufPtr,32);

	   contact->SetWorkStreetL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetWorkTownL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetWorkCountyL(bufPtr);

	   SetRandomAlphaNumericString(bufPtr,16);

	   contact->SetWorkPostCodeL(bufPtr);

	   SetRandomAlphaString(bufPtr,32);

	   contact->SetWorkCountryL(bufPtr);	   

	   }

	if (aLevel >= ELevel3)

	   {

	   SetRandomNumericString(bufPtr,24);

	   contact->SetWorkMobileL(bufPtr);

	   SetRandomNumericString(bufPtr,24);

	   contact->SetWorkFaxL(bufPtr);	   

	   SetRandomNumericString(bufPtr,24);

	   contact->SetHomeMobileL(bufPtr);	   

	   }

   if (aLevel >= ELevel4)

	   {

      SetRandomNumericString(bufPtr,24);

	   contact->SetWorkPhoneL(bufPtr);

   	SetRandomAlphaNumericString(bufPtr,48);

	   contact->SetWorkEmailAddressL(bufPtr);

      // Picture is not used for sorting, so do not

      // grenerate any random data.	   

	   HBufC8* buf8=HBufC8::NewLC(2048);

	   TPtr8   bufPtr8=buf8->Des();

	   contact->SetPictureL(bufPtr8);

	   CleanupStack::PopAndDestroy(1);

	   }

	TContactItemId id = iContactsDb->AddNewContactL(contact->ContactItem());

	CleanupStack::PopAndDestroy(2); // buf, contact.

	return id;

	}

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

void CPerfDbCreator::AddGroupL(const TDesC & aGroupLabel)

	{

	CContactGroup * group = static_cast<CContactGroup *>(iContactsDb->CreateContactGroupLC(aGroupLabel));

 	// Add a number of existing items to the group

 	TInt itemCount = iContactsDb->CountL();

 	TInt contactId;

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

 		{

 		contactId = Math::Rand(iSeed) % itemCount;

 		iContactsDb->AddContactToGroupL(contactId, group->Id());

 		}

 	CleanupStack::PopAndDestroy(group);

	}

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

void CPerfDbCreator::SetRandomAlphaString(TDes& aBuf,TInt aLength, TBool aWithSeed)

	{

	aBuf.SetLength(aLength);

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

		{

		aBuf[ii]=RandomAlphaCharCode();

		}

   if (aWithSeed)

      aBuf.Replace(0,8, iTextSeed);

	}

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

void CPerfDbCreator::SetRandomNumericString(TDes& aBuf,TInt aLength, TBool aWithSeed)

	{

	aBuf.SetLength(aLength);

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

		{

		aBuf[ii]=RandomNumericCharCode();

		}

   if (aWithSeed)

      aBuf.Replace(0,8, iTextSeed);		

	}

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

void CPerfDbCreator::SetRandomAlphaNumericString(TDes& aBuf,TInt aLength, TBool aWithSeed)

	{

	aBuf.SetLength(aLength);

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

		{

		aBuf[ii]=RandomAlphaNumbericCharCode();

		}

   if (aWithSeed)

      aBuf.Replace(0,8, iTextSeed);		

	}

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

TText CPerfDbCreator::RandomAlphaCharCode()

	{

	return RandomCharCode('A','z',' ');

	}

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

TText CPerfDbCreator::RandomNumericCharCode()

	{

	return RandomCharCode('0','9',' ');

	}

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

TText CPerfDbCreator::RandomAlphaNumbericCharCode()

	{

	return RandomCharCode('0','z',' ');

	}

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

TText CPerfDbCreator::RandomCharCode(TText aLowerBound,TText aUpperBound,TText aException)

	{

	TText charCode=0;

	do

		{

		charCode=RandomCharCode(aLowerBound,aUpperBound);

		}

		while (charCode==aException);

	return charCode;

	}

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

TText CPerfDbCreator::RandomCharCode(TText aLowerBound,TText aUpperBound)

	{

	TText charCode=STATIC_CAST(TText,(Math::Rand(iSeed)%(aUpperBound-aLowerBound))+aLowerBound);

	ASSERT(charCode>=aLowerBound && charCode<=aUpperBound);

    iTextSeed.NumFixedWidthUC((TUint) iSeed, EHex,8);   

	return charCode;

	}

//************************************************************************************************************

//   CPerfTestContact

//************************************************************************************************************

CPerfTestContact* CPerfTestContact::NewL(CContactItem& aSeedContact)

	{

	CPerfTestContact* self=CPerfTestContact::NewLC(aSeedContact);

	CleanupStack::Pop();

	return self;

	}

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

CPerfTestContact* CPerfTestContact::NewLC(CContactItem& aSeedContact)

	{

	CPerfTestContact* self=new(ELeave) CPerfTestContact();

	CleanupStack::PushL(self);

	self->ConstructL(aSeedContact);

	return self;

	}

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

CPerfTestContact::CPerfTestContact()

	{

	}

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

void CPerfTestContact::ConstructL(CContactItem& aSeedContact)

	{

	iContactItem=CContactCard::NewL(&aSeedContact);

	}

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

CPerfTestContact::~CPerfTestContact()

	{

	delete iContactItem;

	}

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

CContactItem& CPerfTestContact::ContactItem()

	{

	return *iContactItem;

	}

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

const TPtrC CPerfTestContact::FirstNameL()

	{

	return TextFieldL(KUidContactFieldGivenName);

	}

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

const TPtrC CPerfTestContact::LastNameL()

	{

	return TextFieldL(KUidContactFieldFamilyName);

	}

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

const TPtrC CPerfTestContact::CompanyNameL()

	{

	return TextFieldL(KUidContactFieldCompanyName);

	}

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

const TPtrC CPerfTestContact::JobTitleL()

	{

	return TextFieldL(KUidContactFieldJobTitle);

	}

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

const TPtrC CPerfTestContact::WorkPhoneL()

	{

	CContentType* contentType=CContentType::NewL();

	CleanupStack::PushL(contentType);

	contentType->AddFieldTypeL(KUidContactFieldPhoneNumber);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapWORK);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapVOICE);

	const TPtrC text=TextFieldL(*contentType);

	CleanupStack::PopAndDestroy(); // contentType.

	return text;

	}

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

const TPtrC CPerfTestContact::WorkMobileL()

	{

	CContentType* contentType=CContentType::NewL();

	CleanupStack::PushL(contentType);

	contentType->AddFieldTypeL(KUidContactFieldPhoneNumber);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapWORK);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapVOICE);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapCELL);

	const TPtrC text=TextFieldL(*contentType);

	CleanupStack::PopAndDestroy(); // contentType.

	return text;

	}

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

const TPtrC CPerfTestContact::HomePhoneL()

	{

	CContentType* contentType=CContentType::NewL();

	CleanupStack::PushL(contentType);

	contentType->AddFieldTypeL(KUidContactFieldPhoneNumber);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapHOME);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapVOICE);

	const TPtrC text=TextFieldL(*contentType);

	CleanupStack::PopAndDestroy(); // contentType.

	return text;

	}

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

const TPtrC CPerfTestContact::HomeMobileL()

	{

	CContentType* contentType=CContentType::NewL();

	CleanupStack::PushL(contentType);

	contentType->AddFieldTypeL(KUidContactFieldPhoneNumber);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapHOME);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapVOICE);

	contentType->AddFieldTypeL(KUidContactFieldVCardMapCELL);

	const TPtrC text=TextFieldL(*contentType);

	CleanupStack::PopAndDestroy(); // contentType.

	return text;

	}