/*

* 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 <txtrich.h>

#include <txtstyle.h>

#include "TXTMRTSR.H"

#include <e32test.h>

#include <gdi.h>

#include <conpics.h>											   						 

#include <s32mem.h>

#include <s32file.h>

#include <flddef.h>

#include <fldbltin.h>

#include <fldset.h>

GLDEF_C RTest test(_L("TRTCUSTM"));

LOCAL_D RFs theFs;

LOCAL_D CFileStore* TheStore;

LOCAL_D CParaFormatLayer* GlobalParaFormatLayer;

LOCAL_D CCharFormatLayer* GlobalCharFormatLayer;

LOCAL_D CTrapCleanup* TheTrapCleanup;

const TInt KTestCleanupStack=0x500;

class TStoreResolver : public MRichTextStoreResolver

	{

public:

	virtual const CStreamStore& StreamStoreL(TInt /*aPos*/)const {return *iStore;}

public:

	CStreamStore* iStore;

	};

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

class TTestFieldFactory : public MTextFieldFactory

	{

public:

	// from MTextFieldFactory

	virtual CTextField* NewFieldL(TUid aFieldType); 

	// Creates a field of the type specified

	// Returns NULL if it does not recognise/support the field type

	};

CTextField* TTestFieldFactory::NewFieldL(TUid aFieldType)

// Creates a field (in aHeader) of the type specified in aHeader

// 

	{

	CTextField* field=NULL;

	if (aFieldType==KDateTimeFieldUid)

		field = (CTextField*)new(ELeave) CDateTimeField();

	return field;

	}

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

LOCAL_D void WriteInlineL(RWriteStream& aStream,CRichText* aRichText)

	{

	aRichText->CancelInsertCharFormat();

	aRichText->ExternalizeFieldDataL(aStream);

	aRichText->ExternalizeStyleDataL(aStream);

	TBool hasMarkupData=aRichText->HasMarkupData();

	aStream.WriteUint8L((TUint8)hasMarkupData!=EFalse);

	if (hasMarkupData)

		aRichText->ExternalizeMarkupDataL(aStream);	

	aRichText->ExternalizePlainTextL(aStream);

	}

LOCAL_D void ReadInlineL(RReadStream& aStream,CRichText* aRichText)

	{

	aRichText->InternalizeFieldDataL(aStream);

	aRichText->InternalizeStyleDataL(aStream);

	TBool hasMarkupData=(TBool)aStream.ReadUint8L();

	if (hasMarkupData)

		aRichText->InternalizeMarkupDataL(aStream);	

	aRichText->InternalizePlainTextL(aStream);

	}

LOCAL_D TStreamId PerformSaveL(CRichText* aRichText)

//

	{

	RStoreWriteStream out;

	TStreamId id1=out.CreateLC(*TheStore);

	WriteInlineL(out,aRichText);

	delete aRichText;

	out.CommitL();

	CleanupStack::PopAndDestroy();  // out

	TheStore->CommitL();

	return id1;

	}

LOCAL_C CRichText* PerformLoadL(TStreamId aId)

//

	{

	CRichText* text=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	RStoreReadStream in;

	in.OpenLC(*TheStore,aId);

	TRAPD(ret,

	ReadInlineL(in,text));

	test(ret==KErrNone);

	CleanupStack::PopAndDestroy();  // in

	return text;

	}

LOCAL_C CStyleList* CreatePopulatedStyleList()

//

	{

	//

	// Create style aswell.

	CStyleList* list=CStyleList::NewL();

	CParagraphStyle* style1=CParagraphStyle::NewL(*GlobalParaFormatLayer,*GlobalCharFormatLayer);

	CParagraphStyle* style2=CParagraphStyle::NewL(*GlobalParaFormatLayer,*GlobalCharFormatLayer);

	CParagraphStyle* style3=CParagraphStyle::NewL(*GlobalParaFormatLayer,*GlobalCharFormatLayer);

	RParagraphStyleInfo info1(style1);

	RParagraphStyleInfo info2(style2);

	RParagraphStyleInfo info3(style3);

	list->AppendL(&info1);

	list->AppendL(&info2);

	list->AppendL(&info3);

	return list;

	}

_LIT(KTRtCustOutputFile,"c:\\etext\\TRTCUSTM.DAT");

LOCAL_C void CustomLoadSave()

	{

// Set up the framework

	theFs.Delete(KTRtCustOutputFile);

	theFs.MkDirAll(KTRtCustOutputFile);

	TheStore=CPermanentFileStore::CreateLC(theFs,KTRtCustOutputFile,EFileRead|EFileWrite);

	TheStore->SetTypeL(TheStore->Layout());

//

// Case (1) Rich text with owned style list. - no markup data

//

	test.Start(_L("RT + no markup + style list owned"));

	//

	// Setup the rich text

	CStyleList* list=CreatePopulatedStyleList();

	CRichText* richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list);

	TBool hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	//

	// Save the rich text

	TStreamId id1=PerformSaveL(richText1);

	//

	// Load the rich text

	CRichText* empty=NULL;

	TRAPD(ret,

	empty=PerformLoadL(id1));

	test(ret==KErrNone);

	hasMarkupData=empty->HasMarkupData();

	test(hasMarkupData);

	delete empty;

	empty=NULL;

//

// Case (2) Rich text with referenced style list. - no markup data

//

	test.Next(_L("RT + no markup + style list externally owned"));

	//

	// Setup the rich text

	list=CreatePopulatedStyleList();

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list,CEditableText::ESegmentedStorage,2);

	richText1->SetStyleListExternallyOwned(ETrue);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	//

	// Save the rich text

	id1=PerformSaveL(richText1);

	//

	// Load the rich text

	empty=PerformLoadL(id1);

	hasMarkupData=empty->HasMarkupData();

	test(!hasMarkupData);

	TInt paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	test(list->Count()==3);  // the style list is now externally owned.

	delete empty;

	test(list->Count()==3);  // the style list should not have been destroyed by the rich text

	delete list;

	empty=NULL;

//

// Case (3) Rich text with referenced style list. - with markup

//

	test.Next(_L("RT + markup + style list externally owned"));

	// Setup the rich text

	list=CreatePopulatedStyleList();

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list);

	richText1->SetStyleListExternallyOwned(ETrue);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	CParaFormat* paraFormat=CParaFormat::NewLC();

	TParaFormatMask paraMask;

	paraFormat->iHorizontalAlignment=CParaFormat::ERightAlign;

	paraMask.SetAttrib(EAttAlignment);

	richText1->ApplyParaFormatL(paraFormat,paraMask,8,0);

	CleanupStack::PopAndDestroy();  // paraFormat

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	//

	// Save the rich text

	id1=PerformSaveL(richText1);

	//

	// Load the rich text

	empty=PerformLoadL(id1);

	hasMarkupData=empty->HasMarkupData();

	test(hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	test(list->Count()==3);  // the style list is now externally owned.

	delete empty;

	test(list->Count()==3);  // the style list should not have been destroyed by the rich text

	delete list;

	empty=NULL;

//

// Case (4) Rich text with no style list. - no effective markup

//

	test.Next(_L("RT + delete picture + no style list"));

	// Setup the rich text

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	//

	// Create & insert a picture

	CXzePicture* pic1=CXzePicture::NewL('o');

	TPictureHeader header1;

	TSize size;

	pic1->GetSizeInTwips(size);

	header1.iPictureType=KUidXzePictureType;

	header1.iPicture=pic1;

	header1.iSize=size;

	richText1->InsertL(3,header1);

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	richText1->DeleteL(3,1);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	//

	// Save the rich text

	id1=PerformSaveL(richText1);

	//

	// Load the rich text

	empty=PerformLoadL(id1);

	hasMarkupData=empty->HasMarkupData();

	test(!hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	delete empty;

	empty=NULL;

//

// Case (5) Rich text with SetInsertCharFormat()

//

	test.Next(_L("RT + SetInsertCharFormatL()"));

	// Setup the rich text

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	TCharFormat charFormat;

	TCharFormatMask charMask;

	charFormat.iFontPresentation.iStrikethrough=EStrikethroughOn;

	charMask.SetAttrib(EAttFontStrikethrough);

	richText1->SetInsertCharFormatL(charFormat,charMask,3);

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	//

	// Save the rich text

	id1=PerformSaveL(richText1);

	//

	// Load the rich text

	empty=PerformLoadL(id1);

	hasMarkupData=empty->HasMarkupData();

	test(!hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	delete empty;

	empty=NULL;

//

// Case (6) Rich text with components - default re/store used.

//

	test.Next(_L("RT + components"));

	// Setup the rich text

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	//

	// Create & insert some fields

	TTestFieldFactory factory;

	richText1->SetFieldFactory(&factory);

	CTextField* field=NULL;

	CTextField* field2=NULL;

	TRAP(ret,

	field=factory.NewFieldL(KDateTimeFieldUid));	test(ret==KErrNone);

	TRAP(ret,

	field2=factory.NewFieldL(KDateTimeFieldUid));    test(ret==KErrNone);

	TRAP(ret,

	richText1->InsertFieldL(0,field,KDateTimeFieldUid));  test(ret==KErrNone);

	TRAP(ret,

	richText1->UpdateFieldL(0));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->InsertFieldL(0,field2,KDateTimeFieldUid));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->UpdateFieldL(0));

	test(ret==KErrNone);

	//

	// Create & insert a picture

	pic1=CXzePicture::NewL('o');

	pic1->GetSizeInTwips(size);

	header1.iPictureType=KUidXzePictureType;

	header1.iPicture=pic1;

	header1.iSize=size;

	richText1->InsertL(0,header1);

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	//

	// Save the rich text

	TStreamId head=richText1->StoreL(*TheStore);

	delete richText1;

	//

	// Load the rich text

	empty=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	empty->SetFieldFactory(&factory);

	empty->RestoreL(*TheStore,head);

	//

	// Create the correct factories

	TStoreResolver storeResolver;

	storeResolver.iStore=TheStore;

	MDemPictureFactory* pictureFactory=new(ELeave) MDemPictureFactory;

	empty->SetPictureFactory(pictureFactory,&storeResolver);

	CXzePicture* rPic=(CXzePicture*)empty->PictureHandleL(0);

	test(rPic->iLabel=='o');

	//

	hasMarkupData=empty->HasMarkupData();

	test(hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	delete empty;

	empty=NULL;

	delete pictureFactory;

//

// Case (7) Rich text with fields with referenced style list with markup

//

	test.Next(_L("RT + fields + markup + style list"));

	// Setup the rich text

	list=CreatePopulatedStyleList();

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list);

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	paraFormat=CParaFormat::NewLC();

	paraMask.ClearAll();

	paraFormat->iHorizontalAlignment=CParaFormat::ERightAlign;

	paraMask.SetAttrib(EAttAlignment);

	richText1->ApplyParaFormatL(paraFormat,paraMask,8,0);

	CleanupStack::PopAndDestroy();  // paraFormat

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	//

	// Now add a text field to this.

	richText1->SetFieldFactory(&factory);

	field=NULL;

	TRAP(ret,

	field=factory.NewFieldL(KDateTimeFieldUid));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->InsertFieldL(0,field,KDateTimeFieldUid));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->UpdateFieldL(0));

	test(ret==KErrNone);

	//

	// Save the rich text

	//

	// 1st the components.

	CStoreMap* map=CStoreMap::NewLC(*TheStore);

	richText1->StoreFieldComponentsL(*TheStore,*map);

	//

	RStoreWriteStream out(*map);

	id1=out.CreateLC(*TheStore);

	//

	richText1->CancelInsertCharFormat();

	richText1->ExternalizeFieldDataL(out);

	richText1->ExternalizeStyleDataL(out);

	hasMarkupData=richText1->HasMarkupData();

	out.WriteUint8L((TUint8)hasMarkupData!=EFalse);

	if (hasMarkupData)

		richText1->ExternalizeMarkupDataL(out);	

	richText1->ExternalizePlainTextL(out);

	//

	delete richText1;

	out.CommitL();

	//

	map->Reset();

	CleanupStack::PopAndDestroy(2);  // map,out

	TheStore->CommitL();

	//

	// Load the rich text

	empty=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	empty->SetFieldFactory(&factory);

	RStoreReadStream in;

	in.OpenLC(*TheStore,id1);

	empty->InternalizeFieldDataL(in);

	empty->InternalizeStyleDataL(in);

	hasMarkupData=(TBool)in.ReadUint8L();

	if (hasMarkupData)

		empty->InternalizeMarkupDataL(in);	

	empty->InternalizePlainTextL(in);

	CleanupStack::PopAndDestroy();  // in

	//	

	empty->RestoreFieldComponentsL(*TheStore);

	hasMarkupData=empty->HasMarkupData();

	test(hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	test(empty->StyleList()->Count()==3);  // the style list is now externally owned.

	delete empty;

	empty=NULL;

//

// Case (8) Rich text with fields with referenced style list with markup - default store

//

	test.Next(_L("RT + fields + markup + style list - default store"));

	// Setup the rich text

	list=CreatePopulatedStyleList();

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list);

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	paraFormat=CParaFormat::NewLC();

	paraMask.ClearAll();

	paraFormat->iHorizontalAlignment=CParaFormat::ERightAlign;

	paraMask.SetAttrib(EAttAlignment);

	richText1->ApplyParaFormatL(paraFormat,paraMask,8,0);

	CleanupStack::PopAndDestroy();  // paraFormat

	hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);

	//

	// Now add a text field to this.

	richText1->SetFieldFactory(&factory);

	field=NULL;

	TRAP(ret,

	field=factory.NewFieldL(KDateTimeFieldUid));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->InsertFieldL(0,field,KDateTimeFieldUid));

	test(ret==KErrNone);

	TRAP(ret,

	richText1->UpdateFieldL(0));

	test(ret==KErrNone);

	//

	// Save the rich text

	id1=richText1->StoreL(*TheStore);

	delete richText1;

	//

	// Load the rich text

	empty=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	empty->SetFieldFactory(&factory);

	empty->RestoreL(*TheStore,id1);

	//

	hasMarkupData=empty->HasMarkupData();

	test(hasMarkupData);

	paragraphCount=empty->ParagraphCount();

	test(paragraphCount==3);

	TEtextComponentInfo ci=empty->ComponentInfo();	

	test(ci.iFieldCount==1);

	test(ci.iStyleCount==3);

	delete empty;

	empty=NULL;	//

//

// Case (9) Rich text with no components whatsoever - default store

//

	test.Next(_L("RT + no markup whatsoever - default store"));

	richText1=CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer);

	hasMarkupData=richText1->HasMarkupData();

	test(!hasMarkupData);

	richText1->InsertL(0,_L("hello"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);

	richText1->InsertL(richText1->DocumentLength(),_L("there"));

	richText1->InsertL(richText1->DocumentLength(),CEditableText::EParagraphDelimiter);