一个简单的c++内存池管理器实现（修改自intel ipp speech-codec）

利用c++实现一个简单的内存池管理器

头文件scratchmem.h

/*****************************************************************************
//
// INTEL CORPORATION PROPRIETARY INFORMATION
// This software is supplied under the terms of a license agreement or
// nondisclosure agreement with Intel Corporation and may not be copied
// or disclosed except in accordance with the terms of that agreement.
// Copyright (c) 2005-2007 Intel Corporation. All Rights Reserved.
//
// Intel(R) Integrated Performance Primitives
//
//     USCI - Unified Speech Codec Interface
//
// Purpose: Scratch memory managment header file.
***************************************************************************/

#ifndef __SCRATCHMEM_H__
#define __SCRATCHMEM_H__

typedef unsigned char   Ipp8u;
typedef unsigned short Ipp16u;
typedef unsigned int    Ipp32u;

typedef signed char    Ipp8s;
typedef signed short   Ipp16s;
typedef signed int     Ipp32s;

/* Define NULL pointer value */
#ifndef NULL
#ifdef __cplusplus
#define NULL    0
#else
#define NULL    ((void *)0)
#endif
#endif

#if defined(__ICC) || defined( __ICL ) || defined ( __ECL )
#define __INLINE static __inline
#elif defined( __GNUC__ )
#define __INLINE static __inline__
#else
#define __INLINE static
#endif

#if defined(__ICL ) || defined ( __ECL )
/* Intel C/C++ compiler bug for __declspec(align(8)) !!! */
#define __ALIGN(n) __declspec(align(16))
#define __ALIGN32 __declspec(align(32))
#else
#define __ALIGN(n)
#define __ALIGN32
#endif

#if (defined (_WIN64) || defined(linux64) || defined(linux32e)) && !defined(_WIN32_WCE)
typedef __INT64 Ipp64s;
__INLINE
Ipp64s IPP_INT_PTR( const void* ptr ) {
    union {
        void*   Ptr;
        Ipp64s Int;
    } dd;
    dd.Ptr = (void*)ptr;
    return dd.Int;
}
__INLINE
Ipp64u IPP_UINT_PTR( const void* ptr ) {
    union {
        void*    Ptr;
        Ipp64u   Int;
    } dd;
    dd.Ptr = (void*)ptr;
    return dd.Int;
}
#elif (defined(_WIN32) || defined(linux32)) && !defined(_WIN32_WCE)
__INLINE
Ipp32s IPP_INT_PTR( const void* ptr ) {
    union {
        void*   Ptr;
        Ipp32s Int;
    } dd;
    dd.Ptr = (void*)ptr;
    return dd.Int;
}

__INLINE
Ipp32u IPP_UINT_PTR( const void* ptr ) {
    union {
        void*   Ptr;
        Ipp32u Int;
    } dd;
    dd.Ptr = (void*)ptr;
    return dd.Int;
}
#else
#define IPP_INT_PTR( ptr ) ( (long)(ptr) )
#define IPP_UINT_PTR( ptr ) ( (Ipp32u long)(ptr) )
#endif

#define IPP_BYTES_TO_ALIGN(ptr, align) ((-(IPP_INT_PTR(ptr)&((align)-1)))&((align)-1))

#define IPP_ALIGNED_PTR(ptr, align) (void*)( (Ipp8s*)(ptr) + (IPP_BYTES_TO_ALIGN( ptr, align )) )

#define IPP_MALLOC_ALIGNED_BYTES 32

#define IPP_MALLOC_ALIGNED_0BYTES   0
#define IPP_MALLOC_ALIGNED_1BYTES   1
#define IPP_MALLOC_ALIGNED_8BYTES   8
#define IPP_MALLOC_ALIGNED_16BYTES 16
#define IPP_MALLOC_ALIGNED_32BYTES 32

#define IPP_ALIGNED_ARRAY(align,arrtype,arrname,arrlength)\
arrtype arrname##AlignedArrBuff[(arrlength)+IPP_MALLOC_ALIGNED_##align##BYTES/sizeof(arrtype)];\
arrtype *arrname = (arrtype*)IPP_ALIGNED_PTR(arrname##AlignedArrBuff,align)

__INLINE void* GetMemory(Ipp32s arrlen, Ipp32s sizeOfElem, Ipp8s **CurPtr)
{
   void *ret;

   ret = (void*)IPP_ALIGNED_PTR(*CurPtr,sizeOfElem);
   *CurPtr += (arrlen+1)*sizeOfElem;

   return ret;
}

__INLINE void* GetAlignMemory(Ipp32s align, Ipp32s arrlen, Ipp32s sizeOfElem, Ipp8s **CurPtr)
{
   void *ret;

   ret = (void*)IPP_ALIGNED_PTR(*CurPtr,align);
   *CurPtr += (arrlen+align/sizeOfElem)*sizeOfElem;

   return ret;
}

typedef struct _ScratchMem_Obj {
    Ipp8s *base;
    Ipp8s *CurPtr;
    Ipp32s *VecPtr;
    Ipp32s   offset;
}ScratchMem_Obj;

   #define LOCAL_ALIGN_ARRAY(align,arrtype,arrname,arrlength,obj)\
      arrtype *arrname = (arrtype *)GetAlignMemory(align,arrlength,sizeof(arrtype),&(obj)->Mem.CurPtr)

   #define LOCAL_ARRAY(arrtype,arrname,arrlength,obj)\
      arrtype *arrname = (arrtype *)GetMemory(arrlength,sizeof(arrtype),&(obj)->Mem.CurPtr)

   #define LOCAL_ARRAY_FREE(arrtype,arrname,arrlength,obj)\
      arrname=NULL;\
      (obj)->Mem.CurPtr -= ((arrlength)+1)*sizeof(arrtype)

   #define LOCAL_ALIGN_ARRAY_FREE(align,arrtype,arrname,arrlength,obj)\
      arrname=NULL;\
      (obj)->Mem.CurPtr -= ((arrlength)+IPP_MALLOC_ALIGNED_##align##BYTES/sizeof(arrtype))*sizeof(arrtype)

   #define CLEAR_SCRATCH_MEMORY(obj)\
      (obj)->Mem.CurPtr = (obj)->Mem.base

   #define OPEN_SCRATCH_BLOCK(obj)\
      (obj)->Mem.VecPtr[(obj)->Mem.offset] = IPP_INT_PTR((obj)->Mem.CurPtr);\
      (obj)->Mem.offset++

   #define CLOSE_SCRATCH_BLOCK(obj)\
      (obj)->Mem.offset--;\
      (obj)->Mem.CurPtr = (Ipp8s *)(obj)->Mem.VecPtr[(obj)->Mem.offset]

#ifdef CONST
   #undef CONST
#endif

/*#if (_IPP_ARCH == _IPP_ARCH_XSC)
   #define CONST
#else
#define CONST const
#endif*/
#define CONST

#define SOCKET_SCRATCH_MEMORY_SIZE        (5120+40)
#define Media_SOCKETFUN(type,name,arg) extern type name arg

typedef enum{
    APIMedia_StsBadCodecType     =   -5,
   APIMedia_StsNotInitialized   =   -4,
   APIMedia_StsBadArgErr        =   -3,
   APIMedia_StsDeactivated      =   -2,
   APIMedia_StsErr              =   -1,
   APIMedia_StsNoErr            =    0
}APIMedia_Status;

#endif /* __SCRATCHMEM_H__ */

实现文件scratchmem.cpp：

声明一个结构体，用于内存管理

struct BufferManager{
ScratchMem_Obj      Mem;//内存管理对象
};

//内存管理器
BufferManager *G_BufferManager = NULL;

//获取定义的内存块Size

APIMedia_Status apiMedia_ScratchMemoryAlloc(Ipp32s *pCodecSize) {
    if(NULL==pCodecSize)
        return APIMedia_StsBadArgErr;
    *pCodecSize = SOCKET_SCRATCH_MEMORY_SIZE;
    return APIMedia_StsNoErr;
}

//初始化内存管理器

APIMedia_Status apiMedia_InitBuff(BufferManager* manageObj, Ipp8s *buff)
{
if(NULL==manageObj) return APIMedia_StsBadArgErr;
if(NULL==buff) return APIMedia_StsBadArgErr;

if(buff)   manageObj->Mem.base = buff; // otherwise reinit
else if (manageObj->Mem.base == NULL) return APIMedia_StsNotInitialized;
manageObj->Mem.CurPtr = manageObj->Mem.base;
manageObj->Mem.VecPtr = (Ipp32s *)(manageObj->Mem.base+SOCKET_SCRATCH_MEMORY_SIZE);

return APIMedia_StsNoErr;
}

//申请内存

Ipp8u* ippsMalloc_8u (int len){
return (Ipp8u*)malloc(len);
}

int main( int argc, char * argv[] )
{

//初始化内存池
int eSize;
Ipp8u* _buffer;
eSize = sizeof(BufferManager);
G_BufferManager = (BufferManager *)ippsMalloc_8u(eSize);

apiMedia_ScratchMemoryAlloc(&eSize);
_buffer = ippsMalloc_8u(eSize);
memset(_buffer, 0, eSize);
apiMedia_InitBuff(G_BufferManager, (Ipp8s *)_buffer);

//向内存管理器申请内存同时声明一个变量buff

char G_ReadBuffer[] = {"intel ipp is good"};

int len = sizeof(G_ReadBuffer);

LOCAL_ARRAY(Ipp8u, buff, len, G_BufferManager);
memcpy(buff, G_ReadBuffer, len);

printf("test buffer manager %s", buff);

//归还内存

LOCAL_ARRAY_FREE(Ipp8u, buff, len, G_BufferManager);

}

转载于:https://my.oschina.net/u/2332347/blog/637844