/*

* 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 "TSTCLIPB.H"

#include <txtstyle.h>

#include <gdi.h>

#include <conpics.h>											   						 

#include <flddef.h>

#include <fldbltin.h>

#include <s32mem.h>

#include <s32file.h>

#include "../incp/T_PMLPAR.H"

const TInt KTestCleanupStack=0x500;

LOCAL_D RFs theFs;

LOCAL_D CParaFormatLayer* GlobalParaFormatLayer;

LOCAL_D CCharFormatLayer* GlobalCharFormatLayer;

LOCAL_D MPictureFactory* GlobalPictureFactory;

LOCAL_D CTrapCleanup* TheTrapCleanup;

LOCAL_D CParser* TheParser;

LOCAL_D RFs TheSession;

//

	{

	TRAPD(ret,

	TheParser=CParser::NewL());

	CRichText* text=NULL;

	TRAP(ret,

	text=TheParser->ParseL(aFileName));

	delete GlobalParaFormatLayer;

	delete GlobalCharFormatLayer;

	GlobalParaFormatLayer=(CParaFormatLayer*)text->GlobalParaFormatLayer();

	GlobalCharFormatLayer=(CCharFormatLayer*)text->GlobalCharFormatLayer();

	delete TheParser;

	TheParser=NULL;

	return text;

	}

// Test CRichText construction, (and index), forcing a leave error at each

// possible stage of the process.

//

	{

	CParaFormatLayer* paraLayer=CParaFormatLayer::NewL();

	CCharFormatLayer* charLayer=CCharFormatLayer::NewL();

	TInt failRate=0;

	CRichText* doc=NULL;

	for (failRate=1;;failRate++)

		{

		__UHEAP_RESET;

		__UHEAP_SETFAIL(RHeap::EDeterministic,failRate);

		__UHEAP_MARK;

		TRAPD(ret,doc=CRichText::NewL(paraLayer,charLayer));

		if (ret!=KErrNone)

			{

			__UHEAP_MARKEND;

			test(doc==NULL);

			}

		else

			{

			test(doc!=NULL);

//			test(!doc->HasMarkupData());

			delete doc;

			__UHEAP_MARKEND;

			break;

			}

		}

	__UHEAP_RESET;

	delete paraLayer;

	delete charLayer;

	TBuf<36> answer;

	answer.Format(_L("        #allocs for full c'tion: %d\n"),failRate-1);

	}

	{

// Set up the framework

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

	TFileName file=_L("z:\\test\\app-framework\\etext\\climb.txt");

	/*TInt count=*/richText1->ImportTextFileL(0,file,CPlainText::EOrganiseByParagraph);

	//

	// Insert some picture components

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

	CXzePicture* pic2=CXzePicture::NewL('l');

	CXzePicture* pic3=CXzePicture::NewL('l');

	TPictureHeader hdr1;

	TPictureHeader hdr2;

	TPictureHeader hdr3;

	hdr1.iPictureType = KUidXzePictureType;

	hdr2.iPictureType = KUidXzePictureType;

	hdr3.iPictureType = KUidXzePictureType;

	hdr1.iPicture = pic1;

	hdr2.iPicture = pic2;

	hdr3.iPicture = pic3;

	richText1->InsertL(richText1->DocumentLength(),hdr3);

	richText1->InsertL(richText1->DocumentLength(),hdr2);

	richText1->InsertL(richText1->DocumentLength(),hdr1);

	//

	// Apply some random formatting

	TCharFormat charFormat; TCharFormatMask charMask;

	charFormat.iFontSpec.iFontStyle.SetStrokeWeight(EStrokeWeightBold);

	charMask.SetAttrib(EAttFontStrokeWeight);

	richText1->ApplyCharFormatL(charFormat,charMask,10,37);

//

// Copy to clipboard normally.

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

	CClipboard* writeBoard=CClipboard::NewForWritingLC(TheSession);

	richText1->CopyToStoreL(writeBoard->Store(),writeBoard->StreamDictionary(),0,richText1->DocumentLength()-1);  // forces virtual trailing phrase

	writeBoard->CommitL();

	CleanupStack::PopAndDestroy();  // writeboard

	writeBoard=NULL;

	delete richText1;

	richText1=NULL;

//

// Paste from clipboard normally.

	__UHEAP_MARK;

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

	MDemPictureFactory pictureFactory;

	empty->SetPictureFactory(&pictureFactory,NULL);

	CClipboard* readBoard=CClipboard::NewForReadingLC(TheSession);

	TRAPD(ret,

	empty->PasteFromStoreL(readBoard->Store(),readBoard->StreamDictionary(),0));

	test(ret==KErrNone);

	delete empty;

	empty=NULL;

	CleanupStack::PopAndDestroy();  // readBoard

	readBoard=NULL;

	__UHEAP_MARKEND;

//

// Paste from clipboard with OOM

	__UHEAP_MARK;

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

	for (TInt ii=1; ;ii++)

		{

		readBoard=CClipboard::NewForReadingLC(TheSession);

		__UHEAP_SETFAIL(RHeap::EDeterministic,ii);

		TRAP(ret,empty->PasteFromStoreL(readBoard->Store(),readBoard->StreamDictionary(),0));

		__UHEAP_RESET;

		empty->Reset();

		CleanupStack::PopAndDestroy();

		if (ret==KErrNone)

			break;

		}

	readBoard=NULL;

	//

	delete empty;

	empty=NULL;

	TheSession.Close();

	__UHEAP_MARKEND;

	}

// Test case for defect HA-282 - simple case

//

	{

	__UHEAP_MARK;

// Set up the framework

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

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

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

	//

	// Format the first paragraph

	TCharFormat charFormat; TCharFormatMask charMask;

	charFormat.iFontSpec.iFontStyle.SetStrokeWeight(EStrokeWeightBold);

	charMask.SetAttrib(EAttFontStrokeWeight);

	richText1->ApplyCharFormatL(charFormat,charMask,0,6);

//

// Copy to clipboard normally.

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

	CClipboard* writeBoard=CClipboard::NewForWritingLC(TheSession);

	richText1->CopyToStoreL(writeBoard->Store(),writeBoard->StreamDictionary(),2,6);  // forces virtual trailing phrase

	writeBoard->CommitL();

	CleanupStack::PopAndDestroy();  // writeboard

	writeBoard=NULL;

//

// Paste from clipboard normally.

//	__UHEAP_MARK;

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

	empty->AppendTakingSolePictureOwnershipL(*richText1);

	CClipboard* readBoard=CClipboard::NewForReadingLC(TheSession);

	TRAPD(ret,

	empty->PasteFromStoreL(readBoard->Store(),readBoard->StreamDictionary(),empty->DocumentLength()));

	test(ret==KErrNone);

	delete empty;

	empty=NULL;

	CleanupStack::PopAndDestroy();  // readBoard

	readBoard=NULL;

//	__UHEAP_MARKEND;

//

// Paste from clipboard with OOM

	__UHEAP_MARK;

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

	empty->AppendTakingSolePictureOwnershipL(*richText1);

	delete richText1;

	richText1=NULL;

	for (TInt ii=1; ;ii++)

		{

		readBoard=CClipboard::NewForReadingLC(TheSession);

		__UHEAP_SETFAIL(RHeap::EDeterministic,ii);

		TRAP(ret,empty->PasteFromStoreL(readBoard->Store(),readBoard->StreamDictionary(),empty->DocumentLength()));

		__UHEAP_RESET;

//		empty->Reset();

		CleanupStack::PopAndDestroy();

		if (ret==KErrNone)

			break;

		}

	readBoard=NULL;

	//

	delete empty;

	empty=NULL;

	__UHEAP_MARKEND;

	TheSession.Close();

	__UHEAP_MARKEND;

	}

	{

	aRichText->ExternalizeStyleDataL(aStream);

	aRichText->ExternalizeMarkupDataL(aStream);	

	aRichText->ExternalizePlainTextL(aStream);

	}

	{

	aRichText->InternalizeStyleDataL(aStream);

	aRichText->InternalizeMarkupDataL(aStream);	

	aRichText->InternalizePlainTextL(aStream);

	}

//

	{

	//

	// 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(KTRichOutputFile,"c:\\etext\\TRICH.DAT");

	{

// Set up the framework

	theFs.Delete(KTRichOutputFile);

	theFs.MkDirAll(KTRichOutputFile);

	CFileStore* store = CPermanentFileStore::CreateLC(theFs,KTRichOutputFile,EFileRead|EFileWrite);

	store->SetTypeL(store->Layout());

	CStyleList* list=CreatePopulatedStyleList();

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

	TBool hasMarkupData=richText1->HasMarkupData();

	test(hasMarkupData);  // because of owned style list.

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

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

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

	RStoreWriteStream out;

	TStreamId id1 =  out.CreateLC(*store);

// Write out inline normally

	WriteInlineL(out,richText1);

	delete richText1;

	out.CommitL();

	CleanupStack::PopAndDestroy();  // out

	store->CommitL();

	RStoreReadStream in;

	in.OpenLC(*store,id1);

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

// Read in inline normally:

	TRAPD(ret,ReadInlineL(in,empty));

	test(ret==KErrNone);

	CleanupStack::PopAndDestroy();  // in

	delete empty;

	empty=NULL;

// Read in inline with OOM:

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

	for (TInt jj = 1; ;++jj)

		{

		RStoreReadStream in2;

		in2.OpenLC(*store,id1);

		__UHEAP_SETFAIL(RHeap::EDeterministic,jj);

		TRAPD(ret,ReadInlineL(in2,empty));

		__UHEAP_RESET;

		CleanupStack::PopAndDestroy();  // in2

		if (ret!=KErrNone)

			empty->Reset();

		else

			{

			test(empty->HasMarkupData());

			break;

			}

		}

// Write out inline with OOM:

	CStyleList* list2=CreatePopulatedStyleList();

	CRichText* richText2 = CRichText::NewL(GlobalParaFormatLayer,GlobalCharFormatLayer,*list2);

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

	TStreamId i = store->ExtendL();

	for (TInt ii = 1; ;++ii)

		{

		RStoreWriteStream out2;

		out2.ReplaceLC(*store,i);	

		__UHEAP_SETFAIL(RHeap::EDeterministic,ii);

		TRAP(ret,WriteInlineL(out2,richText2));

		__UHEAP_RESET;

		if (ret==KErrNone)

			{

			out2.CommitL();

			CleanupStack::PopAndDestroy();  // out2

			delete richText2;

			break;		

			}

		else

			CleanupStack::PopAndDestroy();  // out2

		}

	store->CommitL();

// Read in inline normally following a write out with OOM

	RStoreReadStream in3;

	in3.OpenLC(*store,i);

	empty->Reset();

	TRAP(ret,ReadInlineL(in3,empty));	// unhandled exception

	test(ret==KErrNone);

	CleanupStack::PopAndDestroy();

	empty->Reset();

//

//

//

	// Set up new component framework

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

	CXzePicture* pic2=CXzePicture::NewL('l');

	CXzePicture* pic3=CXzePicture::NewL('l');

	TPictureHeader hdr1;

	TPictureHeader hdr2;

	TPictureHeader hdr3;

	hdr1.iPictureType = KUidXzePictureType;

	hdr2.iPictureType = KUidXzePictureType;

	hdr3.iPictureType = KUidXzePictureType;

	hdr1.iPicture = pic1;

	hdr2.iPicture = pic2;

	hdr3.iPicture = pic3;

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

	richText3->InsertL(0,hdr3);

	richText3->InsertL(0,hdr2);

	richText3->InsertL(0,hdr1);

	richText3->InsertL(2,_L("hello"));

// Write outline normally:

	TStreamId id2 = richText3->StoreL(*store);

	store->CommitL();

	delete richText3;

// Read outline normally:

	empty->Reset();

	TRAP(ret,empty->RestoreL(*store,id2));

	test(ret==KErrNone);

	empty->Reset();

// Read outline with OOM:

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

	for (TInt kk = 1; ;++kk)

		{

		__UHEAP_SETFAIL(RHeap::EDeterministic,kk);

		//

		TRAP(ret,inText2->RestoreL(*store,id2));

		//

		__UHEAP_RESET;

		if (ret!=KErrNone)

			inText2->Reset();

		else

			{

			delete inText2;

			break;

			}

		}

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

	CXzePicture* pic4=CXzePicture::NewL('P');

	CXzePicture* pic5=CXzePicture::NewL('P');

	CXzePicture* pic6=CXzePicture::NewL('P');

	TPictureHeader hdr4;

	TPictureHeader hdr5;

	TPictureHeader hdr6;

	hdr4.iPictureType=KUidXzePictureType;

	hdr5.iPictureType=KUidXzePictureType;

	hdr6.iPictureType=KUidXzePictureType;

	hdr4.iPicture=pic4;

	hdr5.iPicture=pic5;

	hdr6.iPicture=pic6;

	richText4->InsertL(0,hdr4);

	richText4->InsertL(0,hdr5);

	richText4->InsertL(0,hdr6);

	richText4->InsertL(1,_L("hello"));

// Wtite outline with OOM:

	TStreamId id3(0);

	for (TInt mm=1; ;++mm)

		{

		__UHEAP_SETFAIL(RHeap::EDeterministic,mm);

		TRAPD(ret,id3 = richText4->StoreL(*store));

		if (ret==KErrNone)

			{

			__UHEAP_RESET;

			delete richText4;

			store->CommitL();

			break;		

			}

		}

// Read outline normally folliwng a write with OOM:

	empty->Reset();

	TRAP(ret,empty->RestoreL(*store,id3));

	test(ret==KErrNone);

	CleanupStack::PopAndDestroy();  // store

	delete empty;

	}

// Simply invokes a method to run the text (& component's) onvariants.

//

	{

	__UHEAP_RESET;

	CParaFormat* paraFormat=CParaFormat::NewLC();

	aText->GetParagraphFormatL(paraFormat,0);  // forces CRichTextIndex::__DbgTextInvariant() to run.

	CleanupStack::PopAndDestroy();  // paraFormat

	}

// Tests basic editing functions under OOM conditions.

//

	{

//

//	Insert a single character

	const TInt insertCharLoop=10;

	const TInt KSmallTextBufferSize=1;

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

	richText1->SetPictureFactory(NULL,NULL);  // forces index generation

	for (TInt mm=1;mm<=insertCharLoop;mm++)

		{

		__UHEAP_SETFAIL(RHeap::EDeterministic,mm);

		for (TInt nn=1;nn<=insertCharLoop;nn++)

			{

			TRAPD(ret,richText1->InsertL(richText1->DocumentLength(),'x'));

			if (ret!=KErrNone)

				{

				test(richText1->DocumentLength()+1==nn);

				CheckTextIntegrity(richText1);

				break;

				}

			}

		__UHEAP_RESET;