/*

* 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:  Declaration of the class RAnalyzeToolAllocator.

*

*/

#ifndef ANALYZETOOLALLOCATOR_H

#define ANALYZETOOLALLOCATOR_H

// INCLUDES

#include <u32std.h>

#include "codeblock.h"

#include "threadstack.h"

#include "analyzetoolmemoryallocator.h"

#include <analyzetool/atstorageserverclnt.h>

#include <analyzetool/analyzetool.h>

#include <analyzetool/atcommon.h>

// CLASS DECLARATION

/**

*  Class which overloads the RAlloctor functions and provides access to 

*  the overloaded functions  

*/

class RAnalyzeToolAllocator : public RAnalyzeToolMemoryAllocator

    {

    public:

        /**

        * C++ default constructor.

        * @param aNotFirst Is this first thread using this heap

        * @param aStorageServer A reference to the 

        *                       <code>RATStorageServer</code> which is 

        *                       used to store kernel events

        * @param aCodeblocks A reference to array of code segments

        * @param aMutex A reference to mutex for schedule access to the 

        *                   shared resources

        * @param aProcessId A reference to the observed process id

        * @param aAnalyzeTool Reference to device driver

        * @param aStorageServerOpen Variable to check if Storage server is connected

        * @param aLogOption The logging option for storage server

        * @param aAllocCallStackSize Max number of stored callstack items when memory allocated

        * @param aFreeCallStackSize Max number of stored callstack items when memory freed

        */

        RAnalyzeToolAllocator( TBool aNotFirst,

                               RATStorageServer& aStorageServer, 

                               RArray<TCodeblock>& aCodeblocks, 

                               RMutex& aMutex, 

                               TUint aProcessId,

                               RAnalyzeTool& aAnalyzeTool,

                               TBool aStorageServerOpen,

                               TUint32 aLogOption,

                               TUint32 aAllocCallStackSize,

                               TUint32 aFreeCallStackSize );

        /**

        * Destructor.

        */                     

        ~RAnalyzeToolAllocator();

        /**

        * Allocates a cell of specified size from the heap.

        * @param aSize The size of the cell to be allocated from the heap. 

        * @return TAny* A pointer to the allocated cell.

        */

        TAny* Alloc( TInt aSize );

        /**

        * Frees the specified cell and returns it to the heap.

        * @param aPtr A pointer to a cell to be freed.

        */

        void Free( TAny* aPtr );

        /**

        * Increases or decreases the size of an existing cell.

        * @param aPtr A pointer to the cell to be reallocated.

        * @param aSize The new size of the cell. This may be bigger 

        *              or smaller than the size of the original cell.

        * @param aMode Flags controlling the reallocation.

        * @return TAny* A pointer to the reallocated cell. This may be the 

        *               same as the original pointer supplied through aCell.

        */

        //lint --e{1735} suppress "Virtual function has default parameter"

        TAny* ReAlloc( TAny* aPtr, TInt aSize, TInt aMode = 0 );

        /**

        * Gets the length of the available space in the specified 

        * allocated cell.

        * @param aCell A pointer to the allocated cell.

        * @return TInt The length of the available space in the allocated cell.

        */

        TInt AllocLen( const TAny* aCell ) const;

    #ifndef __KERNEL_MODE__

        /**

        * Opens this heap for shared access. Opening the heap increases 

        * the heap's access count by one.

        */

        TInt Open();

        /**

        * Closes this shared heap. Closing the heap decreases the heap's 

        * access count by one.

        */

        void Close();

        /**

        * The function frees excess committed space from the top of the heap.

        * The size of the heap is never reduced below the minimum size 

        * specified during creation of the heap.

        * @return TInt The space reclaimed. If no space can be reclaimed, 

                       then this value is zero.

        */

        TInt Compress();

        /**

        * Frees all allocated cells on this heap. 

        */

        void Reset();

        /**

        * Gets the number of cells allocated on this heap, and 

        * the total space allocated to them.

        * @param aTotalAllocSize On return, contains the total 

        *                        space allocated to the cells.

        * @return TInt The number of cells allocated on this heap.

        */

        TInt AllocSize( TInt& aTotalAllocSize ) const;

        /**

        * Gets the total free space currently available on the heap and the 

        * space available in the largest free block. The space available 

        * represents the total space which can be allocated. Note that 

        * compressing the heap may reduce the total free space available 

        * and the space available in the largest free block.

        * @param aBiggestBlock On return, contains the space available 

        *                      in the largest free block on the heap.

        * @return TInt The total free space currently available on the heap.

        */

        TInt Available( TInt& aBiggestBlock ) const;

    #endif

        /**

        * Invocates specified debug funtionality.

        * @param aFunc The debug function

        * @param a1 Debug function specific paramenter.

        * @param a2 Debug function specific paramenter.

        * @return TInt Returns KErrNone, if successful otherwise one 

        *              of the other system-wide error codes.

        */

        //lint --e{1735} suppress "Virtual function has default parameter"

        TInt DebugFunction( TInt aFunc, TAny* a1 = NULL, TAny* a2 = NULL );

    protected:

        /**

        * Extension function

        * @param aExtensionId The extension id

        * @param a0 Extension specific paramenter.

        * @param a1 Extension specific paramenter.

        * @return TInt Returns KErrNone, if successful otherwise one 

        *              of the other system-wide error codes. 

        */

        TInt Extension_( TUint aExtensionId, TAny*& a0, TAny* a1 );

    public: // from RAnalyzeToolMemoryAllocator

        /**

        * Installs the RTraceAllocator allocator

        */

        void Uninstall();

        /**

        * Shares the heap

        */

        void ShareHeap();

    private:

        /**

        * Find the current thread which is using the heap

        * @param aStackStart A reference where the stack start is stored

        * @return TBool ETrue if a thread can be found, EFalse otherwise

        */

        TBool FindCurrentThreadStack( TUint32& aStackStart );

    private: 

        /* Handle to the storage server*/

        RATStorageServer& iStorageServer;

        /* A reference to codeblocks of the observed process */            

        RArray<TCodeblock>& iCodeblocks;

        /* The mutex for serializing access to the shared resources */

        RMutex& iMutex;

        /* The process id */

        TUint iProcessId;

        /* Array for storing the callstack */

        TFixedArray<TUint32, KATMaxCallstackLength> iCallStack;

        /* Array for storing the reallocation callstack */

        TFixedArray <TUint32, KATMaxCallstackLength> iReCallStack;

        /* Array for storing the reallocation callstack */	

        TFixedArray<TUint32, KATMaxFreeCallstackLength> iFreeCallStack;

        /* Array of threads using this heap */

        RArray<TThreadStack> iThreadArray;

        RAnalyzeTool& iAnalyzeTool;

        /* A flag for indicating that the RATStorageServer is open */

        TBool iStorageServerOpen;

        /* Log option */

        TUint32 iLogOption;

        /* Max items on stored call stack when memory allocated */

        TUint32 iAllocMaxCallStack;

        /* Max items on stored call stack when memory freed */

        TUint32 iFreeMaxCallStack;

    };

#endif // ANALYZETOOLALLOCATOR_H

// End of File