// Copyright (c) 2007-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 <mmf/server/mmffile.h>

 #include "customaviplaycontroller.h"

 #include "tsu_mmf_avictrltestids.h"

 #define KInitIncompleteCode KMaxTInt

 _LIT(KTestAviFile, "c:\\mm\\mmf\\testfiles\\avictrl\\aud_vid.avi");

 /**

 *

 * NewL

 *

 */

 CMMFController* CCustomAviPlayController::NewL()

 	{

     CCustomAviPlayController* self = new(ELeave) CCustomAviPlayController;

 	CleanupStack::PushL(self);

 	self->ConstructL();

 	CleanupStack::Pop( self );

 	return STATIC_CAST( CMMFController*, self );

 	}

 /**

 *

 * CCustomAviPlayController

 *

 */

 CCustomAviPlayController::~CCustomAviPlayController()

 	{

 	if (iDataSource)

 		{

 		delete iDataSource;

 		}

 	if (iDataSink)

 		{

 		delete iDataSink;

 		}

     }

 /**

 *

 * CCustomAviPlayController

 *

 */

 CCustomAviPlayController::CCustomAviPlayController() :	CAviPlayController()

 	{

 	}

 /**

 *

 * Override base class ResetL

 *

 */

 void CCustomAviPlayController::ResetL()

 	{

 	CAviPlayController::ResetL();

 	if (iDataSource)

 		{

 		delete iDataSource;

 		iDataSource = NULL;

 		}

 	if (iDataSink)

 		{

 		delete iDataSink;

 		iDataSink = NULL;

 		}

 	}

 /**

 *

 * Override base class PrimeL

 *

 */

 void CCustomAviPlayController::PrimeL(TMMFMessage& aMessage)

 	{

 	if (iMemAllocPrimeTestEnabled)

 		{

 		__UHEAP_SETFAIL(RHeap::EFailNext ,iMemAllocPrimeFailCount);

 		TRAPD(err, CAviPlayController::PrimeL(aMessage));

 		__UHEAP_SETFAIL(RHeap::ENone ,0);

 		User::LeaveIfError(err);

 		}

 	else

 		{

 		CAviPlayController::PrimeL(aMessage);

 		}

 	}

 /**

 *

 * Override base class CustomCommand

 *

 */

 void CCustomAviPlayController::CustomCommand(TMMFMessage& aMessage)

 	{

 	//[ check if the command is for the custom plugin

 	// otherwise pass it on to the real audio controller ]

 	/*if (iPrimingMessage)

 		{

 		delete iPrimingMessage;

 		iPrimingMessage = NULL;

 		}

 	TRAPD(err, iPrimingMessage = CCustomMMFMessageHolder::NewL(aMessage));

 	if (err != KErrNone)

 		{

 		aMessage.Complete(err);

 		return;

 		}*/

 	if (aMessage.Destination().InterfaceId() == KMemoryAllocationTestId)

 		{

 		switch (aMessage.Function())

 			{

 		case KMemoryAllocTest1:

 			DoMemoryAllocTest1(aMessage);

 			break;

 		case KMemoryAllocTest2:

 			DoPrepareMemoryAllocTest2(aMessage);

 			break;

 		case KMemoryAllocTest3:

 			DoMemoryAllocTest3(aMessage);

 			break;

 		case KOOMPrimeTestEnable:

 			DoSetPrimetTestEnable(aMessage);

 			break;

 		case KOOMPrimeTestFailCount:

 			DoSetPrimetTestFailCount(aMessage);

 			break;

 			}

 		}

 	else

 		{

 		//[ let the plugin process the message ]

 		CAviPlayController::CustomCommand(aMessage);

 		}

 	}

 static void CleanupController(TAny* ptr)

 	{

 	CCustomAviPlayController* controller = STATIC_CAST(CCustomAviPlayController*, ptr);

 	TRAP_IGNORE(controller->ResetL());

 	}

 /**

 *

 * ConstructL

 *

 */

 void CCustomAviPlayController::ConstructL()

 	{

 	CAviPlayController::ConstructL();

 	}

 /**

 *

 * DoMemoryAllocTest1

 *

 */

 void CCustomAviPlayController::DoMemoryAllocTest1(TMMFMessage& aMessage)

 	{

 	//[ it is a alloc memory test command ]

 	// [ new algorithm adopted from M&G database ]

 	//

 	TInt errorCode = KErrNone;

 	TInt failCount  = 1;

 	TBool completed  = EFalse;

 	TBool badResult  = EFalse;

 	TBool reachedEnd = EFalse; 

 	for( ; ; )

 		{

 		__UHEAP_SETFAIL(RHeap::EFailNext ,failCount);	// Leavescan will think __UHEAP_SETFAIL is a leaving function due to the macro ending in 'L'. Does not leave

 		//NB: Do not use __MM_HEAP_MARK macro's in this test, the cleaning up the CustomMMFAudioController

 		//is insufficient to remove all possible allocated memory from the framework.

 		//CMMFControllerProxyServer::DoStartThreadL has been updated to do heap checking for 

 		//the whole server thread.

 		//[ run a scenario of the major api 

 		//functions which alloc memory in the controller]

 		TRAP( errorCode, 			

 			TMMFFileConfig fileConfig; // video file 

 			fileConfig().iPath = KTestAviFile;

 			//[ lets manufacture a source and sink ]

 			//[Create the source]

 			MDataSource* source = MDataSource::NewSourceL(KUidMmfFileSource, fileConfig);

 			CleanupDeletePushL(source);

 			//[ Create the sink ]

 			MDataSink* sink = MDataSink::NewSinkL(KUidMmfAudioOutput, KNullDesC8);

 			CleanupDeletePushL(sink);

 			// Use a cleanup item to stop & reset the controller and so remove the data 

 			// sources/sinks automatically when this funtion leaves. This can be done

 			// before the sinks/sources are added.

 			TCleanupItem cleanupItem(CleanupController, this);

 			CleanupStack::PushL(cleanupItem);

 			//[ add the source ]

 			CAviPlayController::AddDataSourceL(*source);

 			//[ add the sink ]

 			CAviPlayController::AddDataSinkL(*sink);

 			CleanupStack::PopAndDestroy(3);// source, sink, cleanupItem

 		);

 		if( errorCode == KErrNone )

 			{

 			// [ check we have passed through all allocs in the test]

 			TAny* testAlloc = User::Alloc(1);

 			if( testAlloc == NULL )

 				{

 				reachedEnd = ETrue;

 				failCount -= 1; // Failcount of 1 equates to 0 successful allocs, etc.

 				}

 			else 

 				{

 				User::Free( testAlloc );

 				}

 			completed = reachedEnd || badResult;

 			}

 		else if( errorCode != KErrNoMemory && errorCode != KErrCorrupt) 

 			{

 			// [ we failed for some reason other than memory

 			//     allocation, so fail the test ]

 			completed = ETrue;

 			badResult = ETrue;

 			}

 		__UHEAP_SETFAIL(RHeap::ENone ,0);

 		// [ exit the loop ]

 		if( completed )

 			{

 			break;

 			}

 		failCount +=1;

 		}

 	if(!badResult)

 		{

 		aMessage.Complete(KErrNone);

 		}

 	else

 		{

 		aMessage.Complete(errorCode);

 		}

 	}

 /**

 *

 * DoMemoryAllocTest2

 *

 */

 void CCustomAviPlayController::DoPrepareMemoryAllocTest2(TMMFMessage& aMessage)

 	{

 	TRAPD(err, 

 		TMMFFileConfig fileConfig; // video file 

 		fileConfig().iPath = KTestAviFile;

 		// Use a cleanup item to stop & reset the controller and so remove the data 

 		// sources/sinks automatically when this funtion leaves. This can be done

 		// before the sinks/sources are added.

 		TCleanupItem cleanupItem(CleanupController, this);

 		CleanupStack::PushL(cleanupItem);

 		//[ lets manufacture a source and sink ]

 		//[Create the source]

 		if (iDataSource)

 			{

 			delete iDataSource;

 			iDataSource = NULL;

 			}

 		iDataSource = MDataSource::NewSourceL(KUidMmfFileSource, fileConfig);

 		//[ Create the sink ]

 		if (iDataSink)

 			{

 			delete iDataSink;

 			iDataSink = NULL;

 			}

 		iDataSink = MDataSink::NewSinkL(KUidMmfAudioOutput, KNullDesC8);

 		//[ add the source ]

 		CAviPlayController::AddDataSourceL(*iDataSource);

 		//[ add the sink ]

 		CAviPlayController::AddDataSinkL(*iDataSink);

 		CleanupStack::Pop(1);	// cleanupItem

 		);

 	aMessage.Complete(err);

 	}

 /**

 *

 * DoMemoryAllocTest3

 *

 */

 void CCustomAviPlayController::DoMemoryAllocTest3(TMMFMessage& aMessage)

 	{

 	__UHEAP_SETFAIL(RHeap::EFailNext ,iMemAllocPrimeFailCount);

 	TRAPD(err, 

 		// Use a cleanup item to stop & reset the controller and so remove the data 

 		// sources/sinks automatically when this funtion leaves. This can be done

 		// before the sinks/sources are added.

 		TCleanupItem cleanupItem(CleanupController, this);

 		CleanupStack::PushL(cleanupItem);

 		//[ play ]

 		CAviPlayController::PlayL();

 		//[ stop ]

 		CAviPlayController::StopL();

 		//[ reset ]	

 		CAviPlayController::ResetL();

 		CleanupStack::PopAndDestroy(1);	// cleanupItem

 		);

 	__UHEAP_SETFAIL(RHeap::ENone ,0);

 	aMessage.Complete(err);

 	}

 /**

 *

 * DoSetPrimetTestEnable

 *

 */

 void CCustomAviPlayController::DoSetPrimetTestEnable(TMMFMessage& aMessage)

 	{

 	TPckgBuf<TBool> primeTestEnable;

 	TInt err = aMessage.ReadData1FromClient(primeTestEnable);

 	aMessage.Complete(err);

 	iMemAllocPrimeTestEnabled = primeTestEnable();

 	}

 /**

 *

 * DoSetPrimetTestFailCount

 *

 */

 void CCustomAviPlayController::DoSetPrimetTestFailCount(TMMFMessage& aMessage)

 	{

 	TPckgBuf<TInt> primeTestFailCount;

 	TInt err = aMessage.ReadData1FromClient(primeTestFailCount);

 	aMessage.Complete(err);

 	iMemAllocPrimeFailCount = primeTestFailCount();

 	}