dl_memory_bank.h

浏览该文件的文档.

#pragma once
 
namespace dl
{
    template<typename T>
    class MemoryBank
    {
    public:
        MemoryBank() :
            _p{ nullptr },
            _cur{ 0 },
            _size{ 0 }
        {
        }
 
        void Clear()
        {
            //清空后显示析构
            for (size_t i = 0; i < _cur; ++i)
            {
                (_p + i)->~T();
            }
            _cur = 0;
        }
 
        template<typename... Args>
        T* Get(Args&& ...args)
        {
            _check();
            return new (_p + _cur++) T(args...);
        }

        T* begin() { return _p; }
        T* end() { return _p + _cur; }
 
        ~MemoryBank()
        {
            //需要显示析构
            for (size_t i = 0; i < _cur; ++i)
                (_p + i)->~T();
            free(_p);
        }
    private:
        T* _p;
        size_t _cur;//指向下一个空闲位置
        size_t _size;
 
        void _check()
        {
            //原有内存不会变化
            if (_cur >= _size)
            {
                _size = _size ? _size * 2 : 1;
                _p = (T*)realloc(_p, _size * sizeof(T));
            }
        }
    };
 
}