/*
* Copyright (c) 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:  
*
*/


#ifndef PROFILER_ITT_SAMPLER_H
#define PROFILER_ITT_SAMPLER_H

#include "GeneralsConfig.h"

#include <kern_priv.h>

#include <piprofiler/ProfilerGenericClassesKrn.h>
#include <piprofiler/ProfilerTraces.h>
#include "GppSamplerImpl.h"

/*
 *	
 *	ITT sampler definition
 *	
 */

class IttSamplerImpl
{
public:
	IttSamplerImpl();
	~IttSamplerImpl();

	TInt	SampleImpl(TUint32 pc,TUint32 sampleNum);
	TBool	SampleNeeded(TUint32 sampleNum);
	TInt 	CreateFirstSample();
	void	Reset();

	TUint8* itt_sample;
	TBool							iSampleProcesses;
private:

	TUint8					internalData[256];
	TPtr8*					ptrToData;
	TInt					currentLibCount;
	TInt					currentProcCount;
	
	TBuf8<64>				iVersionData;
	SDblQue* 				codeSegList;
};

template <int BufferSize>
class DProfilerIttSampler : public DProfilerGenericSampler<BufferSize>
{
public:
	DProfilerIttSampler(struct TProfilerGppSamplerData* gppSamplerDataIn);
	~DProfilerIttSampler();

	void	Sample();
	TInt	Reset(DProfilerSampleStream* aStream, TUint32 aSyncOffset);
	TInt	PostSample();
	TBool	PostSampleNeeded();

private:
    struct TProfilerGppSamplerData*	    gppSamplerData;
	IttSamplerImpl					ittSampler;
	TBool							sampleInProgress;
	TBool							sampleNeeded;
};

/*  
 *	
 *	ITT sampler implementation
 *	
 */ 

template <int BufferSize>
DProfilerIttSampler<BufferSize>::DProfilerIttSampler(struct TProfilerGppSamplerData* gppSamplerDataIn) :
	DProfilerGenericSampler<BufferSize>(PROFILER_ITT_SAMPLER_ID)
{
	this->gppSamplerData = (struct TProfilerGppSamplerData*)gppSamplerDataIn;
	this->sampleInProgress = false;
	LOGSTRING2("CProfilerIttSampler<%d>::CProfilerIttSampler",BufferSize);	
}

template <int BufferSize>
TInt DProfilerIttSampler<BufferSize>::Reset(DProfilerSampleStream* aStream, TUint32 aSyncOffset)
{
	LOGSTRING2("CProfilerIttSampler<%d>::Reset - calling superclass reset",BufferSize);
	DProfilerGenericSampler<BufferSize>::Reset(aStream);
	LOGSTRING2("CProfilerIttSampler<%d>::Reset - called superclass reset",BufferSize);	
	TInt length = this->ittSampler.CreateFirstSample();
	this->iSampleBuffer->AddSample((TUint8*)&length,1);
	this->iSampleBuffer->AddSample(ittSampler.itt_sample,length);
	this->sampleInProgress = false;
	this->sampleNeeded = false;

	this->ittSampler.Reset();

	return KErrNone;
}

template <int BufferSize> 
TInt DProfilerIttSampler<BufferSize>::PostSample()
{
	if(sampleNeeded)
	{
		this->sampleNeeded = false;

		LOGSTRING3("CProfilerIttSampler<%d>::PostSample - state %d",BufferSize,this->sampleBuffer->GetBufferStatus());

		
		TInt length = this->ittSampler.SampleImpl(this->gppSamplerData->lastPcValue,
									        		  this->gppSamplerData->sampleNumber);
		if(length != 0)
		{		
			LOGSTRING("ITT sampler PostSample - starting to sample");

			while(length > 0)
			{
                this->iSampleBuffer->AddSample(ittSampler.itt_sample,length);
				length = this->ittSampler.SampleImpl( this->gppSamplerData->lastPcValue,
									        		  this->gppSamplerData->sampleNumber );	
				if(length == 0) 
				{
					LOGSTRING("MEM sampler PostSample - all samples generated!");
				}
			}
			LOGSTRING("ITT sampler PostSample - finished sampling");
		}
		this->sampleInProgress = false;
	}
	
	// finally perform superclass postsample
	TInt i = this->DProfilerGenericSampler<BufferSize>::PostSample();
	return i;
}

template <int BufferSize> 
TBool DProfilerIttSampler<BufferSize>::PostSampleNeeded()
{
	LOGSTRING3("CProfilerMemSampler<%d>::PostSampleNeeded - state %d",BufferSize,this->sampleBuffer->GetBufferStatus());

	TUint32 status = this->iSampleBuffer->iBufferStatus;

	if(status == DProfilerSampleBuffer::BufferCopyAsap || status == DProfilerSampleBuffer::BufferFull || this->sampleNeeded == true)
	{
		return true;
	}
	
	return false;
}

template <int BufferSize>
void DProfilerIttSampler<BufferSize>::Sample()
{
	LOGSTRING2("CProfilerIttSampler<%d>::Sample",BufferSize);	
	
	//#ifdef ITT_TEST
	LOGSTRING2("CProfilerIttSampler<%d>::Sample",BufferSize);	
	
	if(ittSampler.SampleNeeded(this->gppSamplerData->sampleNumber) && this->sampleInProgress == false) 
	{
		this->sampleInProgress = true;
		this->sampleNeeded = true;

		LOGSTRING2("CProfilerIttSampler<%d>::Sample - sample needed",BufferSize);	
	}	

	LOGSTRING2("CProfilerIttSampler<%d>::Sample",BufferSize);

	return;
}

template <int BufferSize>
DProfilerIttSampler<BufferSize>::~DProfilerIttSampler()
{
	LOGSTRING2("CProfilerIttSampler<%d>::~CProfilerIttSampler",BufferSize);		
}
#endif