/*

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

#include <e32base.h>

#include <s32strm.h>

#include <txtfrmat.h>

#include <txtfmlyr.h>

#include <txtstyle.h>

#include <txtrich.h>

#include <txtuids.h>

#include <prnsetup.h>

#include <prnuids.h>

#include "WNGMODEL.H"

#include "WNGPANIC.H"

const TInt KDefaultTextGranularity=256;

EXPORT_C CWordModel* CWordModel::NewL(const MFieldFileNameInfo* aFileNameInfo,const MFieldNumPagesInfo* aFieldNumPagesInfo,const TDesC& aDriverPath)

/** Allocates and creates a new word processor engine, and performs printer initalisation.

@param aFileNameInfo Optional callback interface to get the current document's 

filename. This provides the file name for insertion into a field in the header 

or footer.

@param aFieldNumPagesInfo Optional callback interface to return the number 

of pages in the current document. This provides the total number of pages 

for insertion into a field in the header or footer.

@param aDriverPath Directory that contains printer driver files

@return New word processor engine */

	{

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

	CleanupStack::PushL(self);

	self->ConstructMinimalL(aFileNameInfo,aFieldNumPagesInfo,aDriverPath);

	CleanupStack::Pop();  // self

	return self;

	}

EXPORT_C CWordModel* CWordModel::NewL(const MFieldFileNameInfo* aFileNameInfo,const MFieldNumPagesInfo* aNumPagesInfo)

/** Allocates and creates a new word processor engine.

@param aFileNameInfo Optional callback interface to get the current document's 

filename. This provides the file name for insertion into a field in the header 

or footer.

@param aNumPagesInfo Optional callback interface to return the number 

of pages in the current document. This provides the total number of pages 

for insertion into a field in the header or footer.

@return New word processor engine */

	{

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

	CleanupStack::PushL(self);

	self->ConstructL(aFileNameInfo,aNumPagesInfo);

	CleanupStack::Pop();

	return self;

	}

EXPORT_C CWordModel::CWordModel()

/** Default constructor. */

	{}

void CWordModel::ConstructGlobalLayersL()

	{

	//

	// Create a default usable Normal style (the default layers)

	CParaFormat paraFormat;

	TParaFormatMask paraFormatMask;

	paraFormat.iSpaceAfterInTwips=100;

	paraFormatMask.SetAttrib(EAttSpaceAfter);

	paraFormat.iHorizontalAlignment=CParaFormat::ELeftAlign;

	paraFormatMask.SetAttrib(EAttAlignment);

	iParaFormatLayer=CParaFormatLayer::NewL(&paraFormat,paraFormatMask);

	//

	TCharFormat charFormat;

	TCharFormatMask charFormatMask;

	TInt typefaceAttributes=TTypeface::EProportional|TTypeface::ESerif;

	charFormat.iFontSpec.iTypeface.SetAttributes(typefaceAttributes);

	charFormat.iFontSpec.iHeight=200;

	charFormatMask.SetAttrib(EAttFontTypeface);

	charFormatMask.SetAttrib(EAttFontHeight);

	iCharFormatLayer=CCharFormatLayer::NewL(charFormat,charFormatMask);

	//

	iText=CRichText::NewL(iParaFormatLayer,iCharFormatLayer,

						CEditableText::ESegmentedStorage,

						KDefaultTextGranularity);	

	iPageTable=new(ELeave) CArrayFixFlat<TInt>(5);

	}

EXPORT_C void CWordModel::ConstructMinimalL(const MFieldFileNameInfo* aFileNameInfo,const MFieldNumPagesInfo* aFieldNumPagesInfo,const TDesC& aDriverPath)

// Minimal initialisation, to allow all components to function well.

// Used by winc developers.

// A printer device must be created by the caller.

//

/** Second-phase constructor, with printer initialisation.

@param aFileNameInfo Optional callback interface to get the current document's 

filename. This provides the file name for insertion into a field in the header 

or footer.

@param aFieldNumPagesInfo Optional callback interface to return the number 

of pages in the current document. This provides the total number of pages 

for insertion into a field in the header or footer.

@param aDriverPath Directory that contains printer driver files */

	{

	//

	// Create the uninitialised print setup object

	InitPrintSetupL(aFileNameInfo,aFieldNumPagesInfo,aDriverPath);

	//

	// Create global layers

	//

	ConstructGlobalLayersL();

	//

	// Create the default, empty style list.

	iStyleList=CStyleList::NewL();

	iStyleShortCutList=new(ELeave) CStyleShortCutList(3);

	//

	//

	iText->SetStyleListExternallyOwned(*iStyleList);

	}

EXPORT_C void CWordModel::ConstructL(const MFieldFileNameInfo* aFileNameInfo,const MFieldNumPagesInfo* aNumPagesInfo)

/** Second-phase constructor.

@param aFileNameInfo Callback interface to get the current document's filename.

@param aNumPagesInfo Callback interface to return the number of pages in the current 

document. */

	{

	//

	// Create the uninitialised print setup object

	InitPrintSetupL( aFileNameInfo, aNumPagesInfo, KDefaultPrinterDriverPath );

	//

	// Create a default usable Normal style (the default layers)

	ConstructGlobalLayersL();

/*

	iStyleList=NULL;

	iStyleShortCutList=NULL;

	iStyleList=CStyleList::NewL();

	iStyleShortCutList=new(ELeave) CStyleShortCutList(3);

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

		{

		CParagraphStyle* style=CParagraphStyle::NewL(*iParaFormatLayer,*iCharFormatLayer);

		RParagraphStyleInfo info(style);

		iStyleList->AppendL(&info);

		}

	iStyleList->At(0).iStyle->SetType(KSystemParagraphStyleUid);

	iStyleList->At(0).iStyle->iName=_L("Heading 1");

	iStyleList->At(1).iStyle->SetType(KSystemParagraphStyleUid);

	iStyleList->At(1).iStyle->iName=_L("Heading 2");

	iStyleList->At(2).iStyle->SetType(KSystemParagraphStyleUid);

	iStyleList->At(2).iStyle->iName=_L("Heading 3");

	iStyleShortCutList->AppendL('1');

	iStyleShortCutList->AppendL('2');

	iStyleShortCutList->AppendL('3');

	iNormalStyleShortCut='N';

*/

//	iText->SetStyleListExternallyOwned(*iStyleList);

	}

void CWordModel::InitPrintSetupL(const MFieldFileNameInfo* aFileNameInfo,const MFieldNumPagesInfo* aNumPagesInfo,const TDesC& aDriverPath)

// Creates a default printer device and does some initialisation

// A real printer device is not constructed here.

//

    { 

	iPrintSetup=CPrintSetup::NewL();

	iPrintSetup->AddPrinterDriverDirL(aDriverPath);

	iPrintSetup->Header()->SetFileNameInfo(CONST_CAST(MFieldFileNameInfo&,*aFileNameInfo));

	iPrintSetup->Footer()->SetFileNameInfo(CONST_CAST(MFieldFileNameInfo&,*aFileNameInfo));

	iPrintSetup->Header()->SetNumPagesInfo(CONST_CAST(MFieldNumPagesInfo&,*aNumPagesInfo));

	iPrintSetup->Footer()->SetNumPagesInfo(CONST_CAST(MFieldNumPagesInfo&,*aNumPagesInfo));

	iPrintSetup->iNumOfFirstPage=1;

	}

EXPORT_C CWordModel::~CWordModel()

/** Destructor. */

	{

	delete iPrintSetup;

	delete iStyleList;

	delete iStyleShortCutList;

	delete iPageTable;

	delete iText;

	delete iParaFormatLayer;

	delete iCharFormatLayer;

	}

EXPORT_C void CWordModel::StoreL(CStreamStore& aStore,CStreamDictionary& aStreamDic,const TAny* aSecurity)const

/** Stores the engine.

This function writes the engine data to a stream store, and records the streams 

used in a stream dictionary.

@param aStore Store to write to

@param aStreamDic Stream dictionary to write to

@param aSecurity Optional security object. If this is specified, the stored 

text stream is encrypted */

	{

	TStreamId id=StoreStylesL(aStore);

	aStreamDic.AssignL(KUidParagraphStyleStream,id);

	id=iPrintSetup->StoreL(aStore);

	aStreamDic.AssignL(KUidPrintSetupStream,id);

	id=StoreTextDataL(aStore,aSecurity);

	aStreamDic.AssignL(KUidEditableTextStream,id);

	id=StoreFieldDataL(aStore);

	aStreamDic.AssignL(KUidTextFieldStream,id);

	id=StoreMarkupDataL(aStore);

	aStreamDic.AssignL(KUidTextMarkupStream,id);

	}

EXPORT_C void CWordModel::RestoreL(const CStreamStore& aStore,const CStreamDictionary& aStreamDic,

								   const TAny* aSecurity,const MFieldFileNameInfo* aFileNameInfo,

								   const MFieldNumPagesInfo* aNumPagesInfo,

									MPictureFactory* aPictureFactory)

/** Restores the engine.

@param aStore Store to read from

@param aStreamDic Stream dictionary to read from

@param aSecurity Optional security object. This is required to read text from 

an encrypted stream.

@param aFileNameInfo An optional object implementing the MFieldFileNameInfo 

interface. This provides the file name for insertion into a field in the header 

or footer. 

@param aNumPagesInfo An optional object implementing the MFieldNumPagesInfo 

interface. This provides the total number of pages for insertion into a field 

in the header or footer. 

@param aPictureFactory An optional object implementing the picture factory 

interface. This is required if the header or footer contains pictures which 

should be restored. */

	{

	TStreamId id=aStreamDic.At(KUidParagraphStyleStream);

	__ASSERT_ALWAYS(id!=KNullStreamId,User::Leave(KErrCorrupt));

	RestoreStylesL(aStore,id);

	id=aStreamDic.At(KUidPrintSetupStream);	

	__ASSERT_ALWAYS(id!=KNullStreamId,User::Leave(KErrCorrupt));

	iPrintSetup->RestoreL(aStore,id,CONST_CAST(MFieldFileNameInfo*,aFileNameInfo),CONST_CAST(MFieldNumPagesInfo*,aNumPagesInfo),aPictureFactory);

	id=aStreamDic.At(KUidEditableTextStream);

	__ASSERT_ALWAYS(id!=KNullStreamId,User::Leave(KErrCorrupt));

	RestoreTextDataL(aStore,id,aSecurity);

	id=aStreamDic.At(KUidTextFieldStream);

	if (id!=KNullStreamId)

		RestoreFieldDataL(aStore,id);

	iText->SetStyleListExternallyOwned(*iStyleList);  // forces restore with style references

	id=aStreamDic.At(KUidTextMarkupStream);	

	if (id!=KNullStreamId)

		RestoreMarkupDataL(aStore,id);

	}

EXPORT_C void CWordModel::RestoreMinimalL(const CStreamStore& aStore,const CStreamDictionary& aStreamDic,

										  const TAny* aSecurity,const MFieldFileNameInfo* /*aFileNameInfo*/,

										  const MFieldNumPagesInfo* /*aNumPagesInfo*/,

										  MPictureFactory* /*aPictureFactory*/)

/** Restores the engine, without restoring the print setup stream.

This allows restoring without loading a printer driver. Header and footer information 

will not be available.

@param aStore Store to read from

@param aStreamDic Stream dictionary to read from

@param aSecurity Optional security object. This is required to read text from 

an encrypted stream.

@param aFileNameInfo Unused

@param aNumPagesInfo Unused

@param aPictureFactory Unused */

	{

	TStreamId id=aStreamDic.At(KUidParagraphStyleStream);

	__ASSERT_ALWAYS(id!=KNullStreamId,User::Leave(KErrCorrupt));

	RestoreStylesL(aStore,id);

	id=aStreamDic.At(KUidEditableTextStream);

	__ASSERT_ALWAYS(id!=KNullStreamId,User::Leave(KErrCorrupt));

	RestoreTextDataL(aStore,id,aSecurity);

	id=aStreamDic.At(KUidTextFieldStream);

	if (id!=KNullStreamId)

		RestoreFieldDataL(aStore,id);

	iText->SetStyleListExternallyOwned(*iStyleList);  // forces restore with style references

	id=aStreamDic.At(KUidTextMarkupStream);	

	if (id!=KNullStreamId)

		RestoreMarkupDataL(aStore,id);

	}

TStreamId CWordModel::StoreFieldDataL(CStreamStore& aStore)const

//

	{

	const TInt fieldCount=iText->FieldCount();

	if (fieldCount<=0)

		return KNullStreamId;

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

	iText->StoreFieldComponentsL(aStore,*map);

	//

	RStoreWriteStream stream(*map);

	TStreamId id=stream.CreateLC(aStore);

	iText->ExternalizeFieldDataL(stream);

	stream.CommitL();

	//

	map->Reset();

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

	return id;

	}

void CWordModel::RestoreFieldDataL(const CStreamStore& aStore,TStreamId aId)

// Assumes a valid stream id

// 	

	{

	__ASSERT_DEBUG(aId!=KNullStreamId,User::Leave(KErrCorrupt));

	RStoreReadStream stream;

	stream.OpenLC(aStore,aId);

	//

	iText->InternalizeFieldDataL(stream);

	CleanupStack::PopAndDestroy();  // stream

	//

	iText->RestoreFieldComponentsL(aStore);

	}

TStreamId CWordModel::StoreTextDataL(CStreamStore& aStore,const TAny* aSecurity)const

	{

	CStreamStore* store = &aStore;

	//

	RStoreWriteStream stream;

	TStreamId id=stream.CreateLC(*store);

	iText->ExternalizePlainTextL(stream);

	stream.CommitL();

	CleanupStack::PopAndDestroy();  // stream

	//

	if (aSecurity)

		CleanupStack::PopAndDestroy();  // secure store

	return id;

	}

void CWordModel::RestoreTextDataL(const CStreamStore& aStore,TStreamId aId,const TAny* aSecurity)

	{

	__ASSERT_DEBUG(aId!=KNullStreamId,User::Leave(KErrCorrupt));

	const CStreamStore* store=&aStore;

	//

	RStoreReadStream stream;

	stream.OpenLC(*store,aId);

	//

	iText->InternalizePlainTextL(stream);

	CleanupStack::PopAndDestroy();  // stream

	//

	if (aSecurity)

		CleanupStack::PopAndDestroy();  // secure store

	}

TStreamId CWordModel::StoreMarkupDataL(CStreamStore& aStore)const

// The markup data must be stored in a distinct stream, and will never be encrypted.

// Only the text content will be encrypted.

// ie, If the container doc is passworded, that password will NOT apply to all embedded objects.

//

	{

	const TBool hasMarkupData=iText->HasMarkupData();

	if (!hasMarkupData)

		return KNullStreamId;

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

	iText->StoreMarkupComponentsL(aStore,*map);

	//

	RStoreWriteStream stream(*map);

	TStreamId id=stream.CreateLC(aStore);

	iText->ExternalizeMarkupDataL(stream);

	stream.CommitL();

	//

	map->Reset();

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

	return id;

	}

void CWordModel::RestoreMarkupDataL(const CStreamStore& aStore,TStreamId aId)

//

//

	{

	__ASSERT_DEBUG(aId!=KNullStreamId,User::Leave(KErrCorrupt));

	RStoreReadStream stream;

	stream.OpenLC(aStore,aId);

	//

	iText->InternalizeMarkupDataL(stream);

	CleanupStack::PopAndDestroy();  // strean

	}

TStreamId CWordModel::StoreStylesL(CStreamStore& aStore)const

// Store the style list, and the table of short cut keys.

// (The short-cut table is mapped directly on to the style list)

// Also stores the normal (global layers).

//

	{

	RStoreWriteStream stream;

	TStreamId id=stream.CreateLC(aStore);

	//

	stream<< *iText->GlobalParaFormatLayer();  // The "normal" style

	stream<< *iText->GlobalCharFormatLayer();  //

	stream.WriteUint32L(iNormalStyleShortCut);  // The "normal" short cut

	//

	TInt shortCutCount=(iStyleShortCutList) ? iStyleShortCutList->Count() : 0;

	stream.WriteUint8L((TUint8)shortCutCount);

	for (TInt nn=0;nn<shortCutCount;nn++)

		stream.WriteUint32L((*iStyleShortCutList)[nn]);

	stream<< *iStyleList;

	//

	stream.CommitL();

	CleanupStack::PopAndDestroy();  // stream

	return id;

	}

void CWordModel::RestoreStylesL(const CStreamStore& aStore,TStreamId aId)

// Restore the style list and style short cut key list from the specified stream.

//

	{

	RStoreReadStream stream;

	stream.OpenLC(aStore,aId);

	//	

	delete iParaFormatLayer;

	delete iCharFormatLayer;

	iParaFormatLayer=NULL;

	iCharFormatLayer=NULL;

	iParaFormatLayer=CParaFormatLayer::NewL(stream);

	iCharFormatLayer=CCharFormatLayer::NewL(stream);

	iNormalStyleShortCut=stream.ReadUint32L();

	//

	TUint8 shortCutCount=stream.ReadUint8L();

	if (iStyleList)

		iStyleList->Reset();

	//

	if (iStyleShortCutList)

		iStyleShortCutList->Reset();  // loading new document from within word

	else

		iStyleShortCutList=new(ELeave) CStyleShortCutList(Max(1,(TInt)shortCutCount));  // loading document from shell

	for (TInt mm=0;mm<shortCutCount;mm++)

		{

		TChar shortcut=stream.ReadUint32L();

		iStyleShortCutList->AppendL(shortcut);

		}

	if (iStyleList)

		iStyleList->InternalizeL(stream,iParaFormatLayer,iCharFormatLayer);

	else

		iStyleList=CStyleList::NewL(stream,iParaFormatLayer,iCharFormatLayer);

	//

	iText->SetGlobalParaFormat(iParaFormatLayer);

	iText->SetGlobalCharFormat(iCharFormatLayer);

	//

	CleanupStack::PopAndDestroy();  // stream

	//__ASSERT_DEBUG(iStyleList->Count()==iStyleShortCutList->Count(),Panic(EStyleIntegrityError));

	//remove assertion so word engine supports wp apps with no short cuts but 1 (global) style

	}