/*

* 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 MEMSPYDRIVERHEAPWALKER_H

#define MEMSPYDRIVERHEAPWALKER_H

// System includes

#include <e32cmn.h>

#include <kern_priv.h>

// Shared includes

#include <memspy/driver/memspydriverenumerationsshared.h>

// User includes

#include "MemSpyDriverHeap.h"

#include "MemSpyDriverHeapStatistics.h"

#include "heaputils.h"

using namespace LtkUtils;

// Heap walker observer interface - can be used to make a record of each cell

class MMemSpyHeapWalkerObserver

    {

public:

    virtual TBool HandleHeapCell(TMemSpyDriverCellType aCellType, TAny* aCellAddress, TInt aLength, TInt aNestingLevel, TInt aAllocNumber) = 0;

    virtual void HandleHeapWalkInit() = 0;

    };

// Heap walker - allows in-place walking of any heap

class RMemSpyDriverHeapWalker

    {

public:

	RMemSpyDriverHeapWalker(RMemSpyDriverRHeapBase& aHeap, MMemSpyHeapWalkerObserver* aObserver=NULL);

public: // API

    TInt Traverse();

    void CopyStatsTo( TMemSpyHeapStatisticsRHeap& aStats );

    void SetObserver( MMemSpyHeapWalkerObserver* aObserver );

    inline void SetPrintDebug() { iPrintDebug = ETrue; }

    inline const TMemSpyHeapWalkStatistics& Stats() const { return iStats; }

private: // Internal methods

	static TBool CellCallback(RAllocatorHelper& aHelper, TAny* aContext, RAllocatorHelper::TExtendedCellType aCellType, TLinAddr aCellAddress, TInt aLength);

	TBool DoCellCallback(RAllocatorHelper& aHelper, RAllocatorHelper::TExtendedCellType aCellType, TLinAddr aCellAddress, TInt aLength);

    TBool NotifyCell( TMemSpyDriverCellType aType, TAny* aCellAddress, TInt aLength, TInt aNestingLevel = -1, TInt aAllocNumber = -1 );

    //

    void UpdateStats( TMemSpyDriverCellType aType, TAny* aCellAddress, TInt aLength, TInt aNestingLevel, TInt aAllocNumber );

    void InitialiseStats();

    void FinaliseStats();

    void PrintStats();

    //

    inline TBool PrintDebug() const { return iPrintDebug; }

private:

    RMemSpyDriverRHeapBase& iHeap;

    TBool iPrintDebug;

    MMemSpyHeapWalkerObserver* iObserver;

    TMemSpyHeapWalkStatistics iStats;

    };

#endif