// spinlock.cpp
// 
// Copyright (c) 2008 - 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include <fshell/memoryaccesscmd.h>
#include <fshell/ioutils.h>
#include <HAL.H>

using namespace IoUtils;

class CCmdSpinlock : public CMemoryAccessCommandBase
	{
public:
	static CCommandBase* NewLC();
	~CCmdSpinlock();
private:
	CCmdSpinlock();
private: // From CCommandBase.
	virtual const TDesC& Name() const;
	virtual void DoRunL();
	virtual void ArgumentsL(RCommandArgumentList& aArguments);
	virtual void OptionsL(RCommandOptionList& aOptions);
private:
	TInt iStartIn;
	TInt iDuration;
	TInt iLoad;
	TInt iPriority;
	volatile TInt iCounter;
	};


CCommandBase* CCmdSpinlock::NewLC()
	{
	CCmdSpinlock* self = new(ELeave) CCmdSpinlock();
	CleanupStack::PushL(self);
	self->BaseConstructL();
	return self;
	}

CCmdSpinlock::~CCmdSpinlock()
	{
	}

CCmdSpinlock::CCmdSpinlock()
	{
	}

const TDesC& CCmdSpinlock::Name() const
	{
	_LIT(KName, "spinlock");
	return KName;
	}

void CCmdSpinlock::ArgumentsL(RCommandArgumentList& aArguments)
	{
	aArguments.AppendIntL(iLoad, _L("load"));
	}

void CCmdSpinlock::OptionsL(RCommandOptionList& aOptions)
	{
	aOptions.AppendIntL(iStartIn, _L("start-after"));
	aOptions.AppendIntL(iDuration, _L("duration"));
	aOptions.AppendIntL(iPriority, _L("priority"));
	}

EXE_BOILER_PLATE(CCmdSpinlock)

void CCmdSpinlock::DoRunL()
	{
	if (iPriority)
		{
#ifdef FSHELL_MEMORY_ACCESS_SUPPORT
		LoadMemoryAccessL();

		RThread me;
		CleanupClosePushL(me);
		LeaveIfErr(me.Open(me.Id()), _L("Couldn't open myself (!?!)"));
		LeaveIfErr(iMemAccess.SetThreadPriority(me, iPriority), _L("Couldn't set priority"));
		CleanupStack::PopAndDestroy(&me);
#else
		PrintWarning(_L("--priority is not supported if memoryaccess is not available."));
#endif
		}

	if (iStartIn)
		{
		Printf(_L("Starting in %d seconds...\r\n"), iStartIn);
		User::After(iStartIn * 1000000);
		}

	if (iDuration)
		{
		Printf(_L("Starting to spin for %d seconds... (Press CTRL-C to exit earlier)\r\n"), iDuration);
		}
	else
		{
		Printf(_L("Starting to spin... (Press CTRL-C to exit)\r\n"));
		}

	// We have ~300,000 cycles per millisecond to play with. We will do some amount of work then wait a millisecond.
	// This is pretty arbitrary.

	TInt tickPeriod;
	HAL::Get(HAL::ENanoTickPeriod, tickPeriod);

	TInt endTime; // in nkern ticks
	if (iDuration)
		{
		endTime = User::NTickCount() + (iDuration * 1000000)/tickPeriod;
		}
	else
		{
		endTime = KMaxTInt;
		}

	if (iLoad)
		{
		//TODO this loop needs sorting out. The User::After is a bit broken.
		TInt n = iLoad * 1000;
		while (User::NTickCount() < (TUint)endTime)
			{
			for (iCounter = 0; iCounter < n; iCounter++)
				{
				// Let's do an exec call just for fun
				UserSvr::DebugMask(0);
				}

			User::After(1000); // Wait 1 ms
			}
		}
	else
		{
		// Just spin
		while (User::NTickCount() < (TUint)endTime)
			{
			}
		}
	}