// Copyright (c) 1998-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 <s32file.h>

#include <e32test.h>

const TInt KTestCleanupStack=0x20;

// This is a path specification and should not be used as is

_LIT(KFileLocationSpec, "Z:\\STOR-TST\\T_SHAPE.DAT");

class CShape;

enum TShape {ENotAShape,ESquare,ECircle};

typedef CShape* (*TShapeCtor)();

class CShape : public CBase

	{

public:

	static TShapeCtor Ctor(TShape type);

//

	virtual void ExternalizeL(RWriteStream& aStream) const=0;

	virtual void InternalizeL(RReadStream& aStream)=0;

//

	TStreamId StoreL(CStreamStore& aStore) const;

	void RestoreL(const CStreamStore& aStore,TStreamId anId);

//

	virtual TPtrC Name() const=0;

	virtual TSize Extent() const=0;

	virtual TPoint Centre() const=0;

	};

class CSquare : public CShape

	{

public:

	static CShape* New();

	CSquare() {}

	CSquare(const TPoint& aCentre,TInt aSide);

	void ExternalizeL(RWriteStream& aStream) const;

	void InternalizeL(RReadStream& aStream);

	TPtrC Name() const;

	TSize Extent() const;

	TPoint Centre() const;

private:

	TRect iRect;

	};

class CCircle : public CShape

	{

public:

	static CShape* New();

	CCircle() {}

	CCircle(const TPoint& aCentre,TInt aRadius);

	void ExternalizeL(RWriteStream& aStream) const;

	void InternalizeL(RReadStream& aStream);

	TPtrC Name() const;

	TSize Extent() const;

	TPoint Centre() const;

private:

	TPoint iCentre;

	TInt iRadius;

	};

class TShapeHolder

	{

public:

	TShapeHolder();

	TShapeHolder(TShape aType,CShape* aShape);

//

	void ExternalizeL(RWriteStream& aStream) const;

	void InternalizeL(RReadStream& aStream);

	void StoreComponentsL(CStreamStore& aStore,CStoreMap& aMap) const;

	void RestoreComponentsL(const CStreamStore& aStore);

//

	void ExternalizeSerialL(RWriteStream& aStream) const;

	void InternalizeSerialL(RReadStream& aStream);

//

	TStreamId StoreL(CStreamStore& aStore) const;

	void RestoreL(const CStreamStore& aStore,TStreamId anId);

//

	CShape* Shape() const;

private:

	TShape iType;

	TSwizzle<CShape> iShape;

	};

LOCAL_D RTest test(_L("t_storshape"));

LOCAL_D CTrapCleanup* TheTrapCleanup;

LOCAL_D RFs TheFs;

//LOCAL_D CBufStore* TheStore;

LOCAL_D CFileStore* TheStore;

LOCAL_D RStoreWriteStream TheSink;

LOCAL_D RStoreReadStream TheSource;

//LOCAL_D RFileWriteStream TheSink;

//LOCAL_D RFileReadStream TheSource;

/**

@SYMTestCaseID          PDS-STORE-CT-4025

@SYMTestCaseDesc	    Basic test for CStoreMap Forget() and Unbind() API

@SYMTestPriority 	    High

@SYMTestActions  	    Unbind stream from Map, forget sgtream from map

@SYMTestExpectedResults map stream ID should be NUll

@SYMDEF                 DEF135804

*/

LOCAL_C void testExtraStoreMapAPIsL()

	{

	test.Next(_L("@SYMTestCaseID PDS-STORE-CT-4025"));

	CShape* shape=new(ELeave) CCircle(TPoint(70,80),40);

	CleanupStack::PushL(shape);

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

	TStreamId id = shape->StoreL(*TheStore);

	map->BindL(shape,id);

	test(id == map->At(shape));

	//Unbind the twizzle(shape) and test to make sure it is unbinded

	map->Unbind(shape);

	test(KNullStreamId == map->At(shape));

	map->BindL(shape,id);

	test(shape == map->Label(id));

	//Forget the stream id and test to make sure it is forgotten

	map->Forget(id);

	test(shape != map->Label(id));

	CleanupStack::PopAndDestroy(2,shape);

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1200

@SYMTestCaseDesc	    Shape streaming test

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for streaming of different shapes

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testShapesL()

    {

	test.Start(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1200 Shape streaming "));

	TParsePtrC parse(KFileLocationSpec);

//

//	TheStore=CDirectFileStore::ReplaceLC(TheFs,KTestFile,EFileRead|EFileWrite);

	TheStore=CPermanentFileStore::ReplaceLC(TheFs,parse.NameAndExt(),EFileRead|EFileWrite);

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

//

	RStoreWriteStream snk;

	TStreamId id=snk.CreateL(*TheStore);

	TShapeHolder hldr(ESquare,new(ELeave) CSquare(TPoint(20,30),40));

	hldr.ExternalizeSerialL(snk);

	delete hldr.Shape();

	hldr=TShapeHolder(ECircle,new(ELeave) CCircle(TPoint(70,80),40));

	hldr.ExternalizeSerialL(snk);

	delete hldr.Shape();

	snk.Close();

	RStoreReadStream src;

	src.OpenL(*TheStore,id);

	hldr.InternalizeSerialL(src);

	delete hldr.Shape();

	hldr.InternalizeSerialL(src);

	delete hldr.Shape();

	src.Close();

//

	hldr=TShapeHolder(ESquare,new(ELeave) CSquare(TPoint(20,30),40));

	id=hldr.StoreL(*TheStore);

	delete hldr.Shape();

	hldr.RestoreL(*TheStore,id);

	delete hldr.Shape();

//

	CShape* shape=new(ELeave) CCircle(TPoint(70,80),40);

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

	TStreamId id2 = shape->StoreL(*TheStore);

	testExtraStoreMapAPIsL();

	map->BindL(shape,id2);

	snk=RStoreWriteStream(*map);

	id=snk.CreateL(*TheStore);

	snk<<shape;

	snk.Close();

	delete shape;

	src.OpenL(*TheStore,id);

	src>>id;

	src.Close();

	shape=new(ELeave) CCircle;

	shape->RestoreL(*TheStore,id);

	delete map;

	TRAPD(r,shape->RestoreL(*TheStore,id));

	test(r==KErrNotFound);

	delete shape;

//

	CleanupStack::PopAndDestroy();

    }

//

// Prepare the test directory.

//

LOCAL_C void setupTestDirectory()

    {

	TInt r=TheFs.Connect();

	test(r==KErrNone);

//

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	r=TheFs.MkDir(parse.DriveAndPath());

	test(r==KErrNone||r==KErrAlreadyExists);

	r=TheFs.SetSessionPath(parse.DriveAndPath());

	test(r==KErrNone);

	}

//

// Initialise the cleanup stack.

//

LOCAL_C void setupCleanup()

    {

	TheTrapCleanup=CTrapCleanup::New();

	test(TheTrapCleanup!=NULL);

	TRAPD(r,\

		{\

		for (TInt i=KTestCleanupStack;i>0;i--)\

			CleanupStack::PushL((TAny*)1);\

		test(r==KErrNone);\

		CleanupStack::Pop(KTestCleanupStack);\

		});

	test(r==KErrNone);

	}

LOCAL_C void DeleteDataFile(const TDesC& aFullName)

	{

	RFs fsSession;

	TInt err = fsSession.Connect();

	if(err == KErrNone)

		{

		TEntry entry;

		if(fsSession.Entry(aFullName, entry) == KErrNone)

			{

			RDebug::Print(_L("Deleting \"%S\" file.\n"), &aFullName);

			err = fsSession.SetAtt(aFullName, 0, KEntryAttReadOnly);

			if(err != KErrNone)

				{

				RDebug::Print(_L("Error %d changing \"%S\" file attributes.\n"), err, &aFullName);

				}

			err = fsSession.Delete(aFullName);

			if(err != KErrNone)

				{

				RDebug::Print(_L("Error %d deleting \"%S\" file.\n"), err, &aFullName);

				}

			}

		fsSession.Close();

		}

	else

		{

		RDebug::Print(_L("Error %d connecting file session. File: %S.\n"), err, &aFullName);

		}

	}

//

// Test the streaming framework.

//

GLDEF_C TInt E32Main()

    {

	test.Title();

	setupTestDirectory();

	setupCleanup();

	__UHEAP_MARK;

//

	TRAPD(r,testShapesL());

	test(r==KErrNone);

	//deletion of data files must be before call to .End() - DEF047652

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	::DeleteDataFile(parse.FullName());

	test.End();

//

	__UHEAP_MARKEND;

	delete TheTrapCleanup;

	TheFs.Close();

	test.Close();

	return 0;

    }

TShapeCtor CShape::Ctor(TShape type)

	{

	switch (type)

		{

	case ESquare:

		return &CSquare::New;

	case ECircle:

		return &CCircle::New;

	default:

		return NULL;

		}

	}

TStreamId CShape::StoreL(CStreamStore& aStore) const

	{

	RStoreWriteStream stream;

	TStreamId id=stream.CreateLC(aStore);

	ExternalizeL(stream);

	stream.CommitL();

	CleanupStack::PopAndDestroy();

	return id;

	}

void CShape::RestoreL(const CStreamStore& aStore,TStreamId anId)

	{

	RStoreReadStream stream;

	stream.OpenLC(aStore,anId);

	InternalizeL(stream);

	CleanupStack::PopAndDestroy();

	}

CShape* CSquare::New()

	{

	return new CSquare;

	}

CSquare::CSquare(const TPoint& aCentre,TInt aSide)

	{

	TInt offset=aSide/2;

	iRect.iTl.iX=aCentre.iX-offset;

	iRect.iTl.iY=aCentre.iY-offset;

	iRect.iBr.iX=iRect.iTl.iX+aSide;

	iRect.iBr.iY=iRect.iTl.iY+aSide;

	}

void CSquare::ExternalizeL(RWriteStream& aStream) const

	{

	aStream.WriteUint32L(iRect.iTl.iX);

	aStream.WriteUint32L(iRect.iTl.iY);

	aStream.WriteUint32L(iRect.iBr.iX);

	aStream.WriteUint32L(iRect.iBr.iY);

	}

void CSquare::InternalizeL(RReadStream& aStream)

	{

	iRect.iTl.iX=aStream.ReadUint32L();

	iRect.iTl.iY=aStream.ReadUint32L();

	iRect.iBr.iX=aStream.ReadUint32L();

	iRect.iBr.iY=aStream.ReadUint32L();

	}

TPtrC CSquare::Name() const

	{

	return _L("Square");

	}

TSize CSquare::Extent() const

	{

	return (iRect.iBr-iRect.iTl).AsSize();

	}

TPoint CSquare::Centre() const

	{

	return TPoint((iRect.iBr.iX-iRect.iTl.iX)/2,(iRect.iBr.iY-iRect.iTl.iY)/2);

	}

CShape* CCircle::New()

	{

	return new CCircle;

	}

CCircle::CCircle(const TPoint& aCentre,TInt aRadius)

	: iCentre(aCentre),iRadius(aRadius)

	{}

void CCircle::ExternalizeL(RWriteStream& aStream) const

	{

	aStream.WriteUint32L(iCentre.iX);

	aStream.WriteUint32L(iCentre.iY);

	aStream.WriteUint32L(iRadius);

	}

void CCircle::InternalizeL(RReadStream& aStream)

	{

	iCentre.iX=aStream.ReadUint32L();

	iCentre.iY=aStream.ReadUint32L();

	iRadius=aStream.ReadUint32L();

	}

TPtrC CCircle::Name() const

	{

	return _L("Circle");

	}

TSize CCircle::Extent() const

	{

	TInt diameter=iRadius*2;

	return TSize(diameter,diameter);

	}

TPoint CCircle::Centre() const

	{

	return iCentre;

	}

TShapeHolder::TShapeHolder()

	: iType(ENotAShape),iShape(NULL)

	{}

TShapeHolder::TShapeHolder(TShape aType,CShape* aShape)

	: iType(aType),iShape(aShape)

	{

	__ASSERT_DEBUG((iType==ENotAShape)==(aShape==NULL),User::Panic(_L("gargl"),0));

	}

void TShapeHolder::ExternalizeL(RWriteStream& aStream) const

	{

	aStream.WriteUint8L(iType);

	aStream<<iShape;

	}

void TShapeHolder::InternalizeL(RReadStream& aStream)

	{

	TShape type=TShape(aStream.ReadUint8L());

	if ((type==ENotAShape)!=(CShape::Ctor(type)==NULL))

		User::Leave(KErrCorrupt); // representation violation!!!

//

	aStream>>iShape;

	iType=type;

	}

void TShapeHolder::StoreComponentsL(CStreamStore& aStore,CStoreMap& aMap) const

	{

	if (iShape!=NULL)

		{

		TStreamId id=iShape->StoreL(aStore);

		aMap.BindL(iShape,id);

		}

	}

void TShapeHolder::RestoreComponentsL(const CStreamStore& aStore)

	{

	TShapeCtor ctor=CShape::Ctor(iType);

	CShape* shape=NULL;

	if (ctor!=NULL)

		{

		User::LeaveIfNull(shape=(*ctor)());

		CleanupStack::PushL(shape);

		shape->RestoreL(aStore,iShape.AsId());

		CleanupStack::Pop();

		}

	iShape=shape;

	}

void TShapeHolder::ExternalizeSerialL(RWriteStream& aStream) const

	{

	aStream.WriteUint8L(iType);

	if (iShape!=NULL)

		aStream<<*iShape;

	}

void TShapeHolder::InternalizeSerialL(RReadStream& aStream)

	{

	TShape type=TShape(aStream.ReadUint8L());

	TShapeCtor ctor=CShape::Ctor(type);

	if ((type==ENotAShape)!=(ctor==NULL))

		User::Leave(1832); // representation violation!!!

//

	CShape* shape=NULL;

	if (ctor!=NULL)

		{

		User::LeaveIfNull(shape=(*ctor)());

		CleanupStack::PushL(shape);

		aStream>>*shape;

		CleanupStack::Pop();

		}

	iType=type;

	iShape=shape;

	}

TStreamId TShapeHolder::StoreL(CStreamStore& aStore) const

	{

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

	StoreComponentsL(aStore,*map);

//

	RStoreWriteStream stream(*map);

	TStreamId id=stream.CreateLC(aStore);

	ExternalizeL(stream);

	stream.CommitL();

//

	map->Reset();

	CleanupStack::PopAndDestroy(2);

	return id;

	}

void TShapeHolder::RestoreL(const CStreamStore& aStore,TStreamId anId)

	{

	RStoreReadStream stream;

	stream.OpenLC(aStore,anId);

	InternalizeL(stream);

	CleanupStack::PopAndDestroy();

//

	RestoreComponentsL(aStore);

	}

CShape* TShapeHolder::Shape() const

	{

	return iShape;

	}