码迷,mamicode.com
首页 > 其他好文 > 详细

标准库Allocator的简易实现(二)

时间:2014-10-02 20:34:33      阅读:264      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   color   使用   ar   sp   div   c   

自己实现Allocator并不难,其实只需要改变allocate和deallocate,来实现自己的内存分配策略。

 

下面是一个std::allocator的模拟实现

#ifndef ALLOCATOR_HPP
#define ALLOCATOR_HPP

#include <stddef.h>
#include <limits>

template <typename T>
class Allocator
{
public:
    typedef size_t size_type;
    typedef ptrdiff_t difference_type;
    typedef T*  pointer;
    typedef const T* const_pointer;
    typedef T& reference;
    typedef const T& const_reference;
    typedef T value_type;

    //Allocator::rebind<T2>::other
    template <typename V>
    struct rebind
    {
        typedef Allocator<V> other;
    };

    pointer address(reference value) const { return &value; }
    const_pointer address(const_reference value) const { return &value; }

    Allocator() throw() {   }
    Allocator(const Allocator &) throw() {  }
    //不同类型的allcator可以相互复制
    template <typename V> Allocator(const Allocator<V> &other) { } 
    ~Allocator() throw() {  }

    //最多可以分配的数目
    size_type max_size() const throw()
    { return std::numeric_limits<size_type>::max() / sizeof(T); }

    //分配内存,返回该类型的指针
    pointer allocate(size_type num)
    { return (pointer)(::operator new(num * sizeof(T))); }

    //执行构造函数,构建一个对象
    void construct(pointer p, const T &value)
    { new ((void*)p) T(value); }

    //销毁对象
    void destroy(pointer p)
    { p->~T(); }

    //释放内存
    void deallocate(pointer p, size_type num)
    { ::operator delete((void *)p); }
};

//这两个运算符不需要friend声明
template <typename T, typename V>
bool operator==(const Allocator<T> &, const Allocator<V> &) throw()
{ return true; }

template <typename T, typename V>
bool operator!=(const Allocator<T> &, const Allocator<V> &) throw()
{ return false; }


#endif

这里注意rebind的实现,如果需要使用Test的分配器分配其他类型,就可以这样:

Allocator<Test>::rebind<Test2>::other alloc;

测试代码如下:

#include "Allocator.hpp"
#include <string>
#include <vector>
using namespace std;


int main(int argc, char const *argv[])
{
    vector<string, Allocator<string> > vec(10, "haha");

    vec.push_back("foo");
    vec.push_back("bar");

    //Allocator<Test>::rebind<Test2>::other alloc;

    return 0;
}

标准库Allocator的简易实现(二)

标签:des   style   blog   color   使用   ar   sp   div   c   

原文地址:http://www.cnblogs.com/inevermore/p/4004261.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!