// topfshell.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 <e32hashtab.h>

#include <fshell/ioutils.h>

#include <fshell/memoryaccesscmd.h>

#include "sampler.h"

using namespace IoUtils;

class CCmdTop : public CMemoryAccessCommandBase

	{

public:

	static CCommandBase* NewLC();

	~CCmdTop();

private:

	CCmdTop();

	static TInt TimerCallback(TAny* aSelf);

	void TimerCallback();

	void UpdateL();

	struct SThreadData;

	static TInt SortReverse(const CCmdTop::SThreadData& aLeft, const CCmdTop::SThreadData& aRight);

private: // From CCommandBase.

	virtual const TDesC& Name() const;

	virtual void DoRunL();

	virtual void ArgumentsL(RCommandArgumentList& aArguments);

	virtual void OptionsL(RCommandOptionList& aOptions);

private:

	RBuf8 iReadBuf;

	struct SThreadData

		{

		TUint iId;

		TInt iNumSamples;

		TFullName iName; // Saves recalculating every time

		};

	RHashMap<TUint, SThreadData*> iSeenThreads; // Saves linear searching through iThreads for each sample (Maps thread id to SThreadData

	RPointerArray<SThreadData> iThreads;

	RSampler iSampler;

	CPeriodic* iTimer;

	TInt iRate;

	CTextBuffer* iBuffer;

	CTextFormatter* iFormatter;

	TInt iNumConsoleLines;

	TInt iNumLinesInLastUpdate;

	};

CCommandBase* CCmdTop::NewLC()

	{

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

	CleanupStack::PushL(self);

	self->BaseConstructL();

	return self;

	}

CCmdTop::~CCmdTop()

	{

	if (iTimer)

		{

		iTimer->Cancel();

		}

	delete iTimer;

	if (iSampler.Handle())

		{

		iSampler.Stop();

		iSampler.Close();

		User::FreeLogicalDevice(KSamplerName);

		}

	delete iBuffer;

	delete iFormatter;

	iReadBuf.Close();

	iThreads.ResetAndDestroy();

	iSeenThreads.Close();

	}

CCmdTop::CCmdTop()

	: CMemoryAccessCommandBase(EManualComplete), iRate(1000)

	{

	}

const TDesC& CCmdTop::Name() const

	{

	_LIT(KName, "top");	

	return KName;

	}

void CCmdTop::ArgumentsL(RCommandArgumentList& /*aArguments*/)

	{

	}

void CCmdTop::OptionsL(RCommandOptionList& aOptions)

	{

	aOptions.AppendIntL(iRate, _L("rate"));

	}

EXE_BOILER_PLATE(CCmdTop)

void CCmdTop::DoRunL()

	{

#ifdef FSHELL_MEMORY_ACCESS_SUPPORT

	LoadMemoryAccessL();

#endif

	TInt err = User::LoadLogicalDevice(KSamplerName);

	if (err != KErrNone && err != KErrAlreadyExists)

		{

		LeaveIfErr(err, _L("Couldn't load sampler ldd %S"), &KSamplerName);

		}

	iReadBuf.CreateL(iRate * 2 * sizeof(TUint32)); // go twice as big as needed just to be safe, allows for AO not getting to run on time

	iTimer = CPeriodic::NewL(CActive::EPriorityStandard);

	iTimer->Start(iRate*1000, iRate*1000, TCallBack(&CCmdTop::TimerCallback, this));

	LeaveIfErr(iSampler.Open(), _L("Couldn't open sampler"));

	iSampler.Reset(ETrue); // Parameter is ignored

	iSampler.Start(1000); // Always tell sampler to sample every millisecond, regardless of the frequency we update the UI

	TSize size;

	User::LeaveIfError(Stdout().GetScreenSize(size));

	iNumConsoleLines = size.iHeight;

	iBuffer = CTextBuffer::NewL(512);

	iFormatter = CTextFormatter::NewL(size.iWidth);

	Stdout().ClearScreen();

	UpdateL(); // Just so we display something on screen before the first update

	}

TInt CCmdTop::TimerCallback(TAny* aSelf)

	{

	static_cast<CCmdTop*>(aSelf)->TimerCallback();

	return 0;

	}

void CCmdTop::TimerCallback()

	{

	iReadBuf.Zero();

	TRequestStatus stat;

	iSampler.Read(iReadBuf, stat);

	User::WaitForRequest(stat); // API used to be async, it's now sync but I haven't changed the interface

	// First off clear all the iNumSamples (as that was for the previous sampling period)

	for (TInt i = 0; i < iThreads.Count(); i++)

		{

		iThreads[i]->iNumSamples = 0;

		}

	TInt numSamples = iReadBuf.Length() / sizeof(TUint32);

	TUint32* buf = (TUint32*)iReadBuf.Ptr();

	for (TInt i = 0; i < numSamples; i++)

		{

		TUint32 threadId = buf[i];

		SThreadData** thread = iSeenThreads.Find(threadId);

		if (thread)

			{

			(*thread)->iNumSamples++;

			}

		else

			{

			// New thread id

			SThreadData* thread = new SThreadData;

			if (!thread) continue;

			TInt err = iThreads.Append(thread);

			if (err)

				{

				delete thread;

				continue;

				}

			err = iSeenThreads.Insert(threadId, thread);

			if (err)

				{

				iThreads.Remove(iThreads.Count()-1);

				delete thread;

				continue;

				}

			thread->iNumSamples = 1;

			thread->iId = threadId;

			RThread rthread;

#ifdef FSHELL_MEMORY_ACCESS_SUPPORT

			err = iMemAccess.RThreadForceOpen(rthread, threadId);

#else

			err = rthread.Open(TThreadId(threadId));

#endif

			if (err == KErrNone)

				{

				thread->iName = rthread.FullName();

				//PrettyName(EListThread, thread->iName);

				rthread.Close();

				}

			else

				{

				thread->iName = _L("Tid: ");

				thread->iName.AppendNum(threadId);

				}

			}

		}

	TRAP_IGNORE(UpdateL());

	}

TInt CCmdTop::SortReverse(const CCmdTop::SThreadData& aLeft, const CCmdTop::SThreadData& aRight)

	{

	// This sorts largest number of samples first 

	TInt res = - (aLeft.iNumSamples - aRight.iNumSamples);

	if (res == 0) return aLeft.iId - aRight.iId; // make sure there is a total ordering in event of tie break

	return res;

	}

void CCmdTop::UpdateL()

	{

	// First, sort iThreads

	iThreads.Sort(TLinearOrder<SThreadData>(&SortReverse));

	TInt numSamples = iReadBuf.Size() / sizeof(TUint32);

	iBuffer->Zero();

	iFormatter->Zero();

	iBuffer->AppendL(_L("Tid\tThread name\tCPU usage\r\n"));

	for (TInt i = 0; i < iThreads.Count() && i < iNumConsoleLines-2; i++) // minus one for title, one for last line of screen which I can't seem to figure out how to make use of

		{

		SThreadData& thread = *iThreads[i];

		TReal percent = 100 * thread.iNumSamples / (TReal)numSamples;

		iBuffer->AppendFormatL(_L("%d\t%S\t%00.2f%%\r\n"), thread.iId, &thread.iName, percent);

		}

	Stdout().SetCursorHeight(0);

	iFormatter->TabulateL(0, 1, iBuffer->Descriptor(), ETruncateLongestColumn);

	User::LeaveIfError(Stdout().SetCursorPosAbs(TPoint(0, 0)));

	Stdout().Write(iFormatter->Descriptor());

	TInt numLines = 1 + iThreads.Count(); // plus 1 for the title line

	TInt numOldLines = iNumLinesInLastUpdate - numLines;

	while (numOldLines > 0)

		{

		Stdout().ClearToEndOfLine();

		Stdout().SetCursorPosRel(TPoint(0, 1));

		--numOldLines;

		}

	iNumLinesInLastUpdate = numLines;

	}