axmol/cocos/2d/renderer/RenderCommandPool.h

85 lines
1.8 KiB
C++

//
// RenderCommandPool.h
// cocos2d_libs
//
// Created by Huabing on 11/28/13.
//
//
#ifndef __CC_RENDERCOMMANDPOOL_H__
#define __CC_RENDERCOMMANDPOOL_H__
#include <set>
#include <list>
#include "CCPlatformMacros.h"
NS_CC_BEGIN
template <class T>
class RenderCommandPool
{
public:
RenderCommandPool()
{
}
~RenderCommandPool()
{
// if( 0 != _usedPool.size())
// {
// CCLOG("All RenderCommand should not be used when Pool is released!");
// }
_freePool.clear();
for (typename std::list<T*>::iterator iter = _allocatedPoolBlocks.begin(); iter != _allocatedPoolBlocks.end(); ++iter)
{
delete[] *iter;
*iter = nullptr;
}
_allocatedPoolBlocks.clear();
}
public:
T* generateCommand()
{
T* result = nullptr;
if(_freePool.empty())
{
AllocateCommands();
}
result = _freePool.front();
_freePool.pop_front();
//_usedPool.insert(result);
return result;
}
void pushBackCommand(T* ptr)
{
// if(_usedPool.find(ptr) == _usedPool.end())
// {
// CCLOG("push Back Wrong command!");
// return;
// }
_freePool.push_back(ptr);
//_usedPool.erase(ptr);
}
private:
void AllocateCommands()
{
static const int COMMANDS_ALLOCATE_BLOCK_SIZE = 32;
T* commands = new T[COMMANDS_ALLOCATE_BLOCK_SIZE];
_allocatedPoolBlocks.push_back(commands);
for(int index = 0; index < COMMANDS_ALLOCATE_BLOCK_SIZE; ++index)
{
_freePool.push_back(commands+index);
}
}
private:
std::list<T*> _allocatedPoolBlocks;
std::list<T*> _freePool;
//std::set<T*> _usedPool;
};
NS_CC_END
#endif