决定还是两个都看,原来那个tinystl只注释,新的手打。
allocate的设计,我只能说,是精妙到姥姥家了,太精妙了!
但是也太难了…………
allocator.h
#ifndef _ALLOCATOR_H_
#define _ALLOCATOR_H_
#include "Alloc.h"
#include "Construct.h"
#include <cassert>
#include <new>
namespace TinySTL{
/*
**空间配置器,以变量数目为单位分配
*/
template<class T>
class allocator{
public:
typedef T value_type;
typedef T* pointer;
typedef const T* const_pointer;
typedef T& reference;
typedef const T& const_reference;
typedef size_t size_type;
typedef ptrdiff_t difference_type;
public:
static T *allocate();//只分配一个T的内存空间
static T *allocate(size_t n);//分配n个
static void deallocate(T *ptr);//释放
static void deallocate(T *ptr, size_t n);
//给定指针构建对象
static void construct(T *ptr);
static void construct(T *ptr, const T& value);
//析构
static void destroy(T *ptr);
static void destroy(T *first, T *last);
};
template<class T>
T *allocator<T>::allocate(){
return static_cast<T *>(alloc::allocate(sizeof(T)));
}
template<class T>
T *allocator<T>::allocate(size_t n){
if (n == 0) return 0;
return static_cast<T *>(alloc::allocate(sizeof(T) * n));
}
template<class T>
void allocator<T>::deallocate(T *ptr){
alloc::deallocate(static_cast<void *>(ptr), sizeof(T));
}
template<class T>
void allocator<T>::deallocate(T *ptr, size_t n){
if (n == 0) return;
alloc::deallocate(static_cast<void *>(ptr), sizeof(T)* n);
}
template<class T>
void allocator<T>::construct(T *ptr){
new(ptr)T();
}
template<class T>
void allocator<T>::construct(T *ptr, const T& value){
new(ptr)T(value);
}
template<class T>
void allocator<T>::destroy(T *ptr){
ptr->~T();
}
template<class T>
void allocator<T>::destroy(T *first, T *last){
for (; first != last; ++first){
first->~T();
}
}
}
#endif
alloc.h
#ifndef _ALLOC_H_
#define _ALLOC_H_
#include <cstdlib>
namespace TinySTL{
/*
**空间配置器,以字节数为单位分配
**内部使用
*/
class alloc{
private:
//所谓小型区块链就是说list里面一共有128/8=16个区块,每个区块地址依次加8。
//也就是每一块的内存大小依次是8,16,24/32……120,128
enum EAlign{ ALIGN = 8};//小型区块的上调边界
enum EMaxBytes{ MAXBYTES = 128};//小型区块的上限,超过的区块由malloc分配
enum ENFreeLists{ NFREELISTS = (EMaxBytes::MAXBYTES / EAlign::ALIGN)};//free-lists的个数
enum ENObjs{ NOBJS = 20};//每次增加的节点数
private:
//free-lists的节点构,使用union构造,节省内存,柔性数组
union obj{
union obj *next;//下一个区块的地址
char client[1];//用char去申请内存
};
static obj *free_list[ENFreeLists::NFREELISTS];
private:
//静态变量初始化一次
static char *start_free;//内存池起始位置
static char *end_free;//内存池结束位置
static size_t heap_size;//堆区大小
private:
//位运算求2^n次方的倍数
static size_t ROUND_UP(size_t bytes){
return ((bytes + EAlign::ALIGN - 1) & ~(EAlign::ALIGN - 1));
}
//根据区块大小,决定使用第n号free-list,n从0开始计算
static size_t FREELIST_INDEX(size_t bytes){
return (((bytes)+EAlign::ALIGN - 1) / EAlign::ALIGN - 1);
}
//返回一个大小为n的对象,并可能加入大小为n的其他区块到free-list
static void *refill(size_t n);
//配置一大块空间,可容纳nobjs个大小为size的区块
//如果配置nobjs个区块有所不便,nobjs可能会降低
static char *chunk_alloc(size_t size, size_t& nobjs);
public:
static void *allocate(size_t bytes);
static void deallocate(void *ptr, size_t bytes);
static void *reallocate(void *ptr, size_t old_sz, size_t new_sz);
};
}
#endif
alloc.cpp
#include "../Alloc.h"
namespace TinySTL{
//对静态变量初始化
char *alloc::start_free = 0;
char *alloc::end_free = 0;
size_t alloc::heap_size = 0;
//obj *free_list[ENFreeLists::NFREELISTS]每个前面都加了作用域
alloc::obj *alloc::free_list[alloc::ENFreeLists::NFREELISTS] = {
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
};
//成员函数
void *alloc::allocate(size_t bytes){//分配内存
if (bytes > EMaxBytes::MAXBYTES){//申请内存大于预设定内存
return malloc(bytes);//调用malloc
}
size_t index = FREELIST_INDEX(bytes);//计算索引号
obj *list = free_list[index];
if (list){//此list还有空间给我们
free_list[index] = list->next;//移向下一个内存
return list;//返回地址
}
else{//此list没有足够的空间,需要从内存池里面取空间
return refill(ROUND_UP(bytes));
}
}
void alloc::deallocate(void *ptr, size_t bytes){
if (bytes > EMaxBytes::MAXBYTES){
free(ptr);
}
else{
size_t index = FREELIST_INDEX(bytes);
obj *node = static_cast<obj *>(ptr);
node->next = free_list[index];
free_list[index] = node;
}
}
void *alloc::reallocate(void *ptr, size_t old_sz, size_t new_sz){
deallocate(ptr, old_sz);
ptr = allocate(new_sz);
return ptr;
}
//返回一个大小为n的对象,并且有时候会为适当的free list增加节点
//假设bytes已经上调为8的倍数
void *alloc::refill(size_t bytes){
size_t nobjs = ENObjs::NOBJS;//一次取20个
//从内存池里取
char *chunk = chunk_alloc(bytes, nobjs);
obj **my_free_list = 0;
obj *result = 0;
obj *current_obj = 0, *next_obj = 0;
if (nobjs == 1){//取出的空间只够一个对象使用
return chunk;
}
else{
my_free_list = free_list + FREELIST_INDEX(bytes);
result = (obj *)(chunk);
*my_free_list = next_obj = (obj *)(chunk + bytes);
//将取出的多余的空间加入到相应的free list里面去
for (int i = 1;; ++i){
current_obj = next_obj;
next_obj = (obj *)((char *)next_obj + bytes);
if (nobjs - 1 == i){
current_obj->next = 0;
break;
}
else{
current_obj->next = next_obj;
}
}
return result;
}
}
//假设bytes已经上调为8的倍数
char *alloc::chunk_alloc(size_t bytes, size_t& nobjs){
char *result = 0;
size_t total_bytes = bytes * nobjs;//总字节
size_t bytes_left = end_free - start_free;//实际字节
if (bytes_left >= total_bytes){//内存池剩余空间完全满足需要
result = start_free;
start_free = start_free + total_bytes;//更新起点
return result;
}
else if (bytes_left >= bytes){//内存池剩余空间不能完全满足需要,但足够供应一个或以上的区块
nobjs = bytes_left / bytes;
total_bytes = nobjs * bytes;
result = start_free;
start_free += total_bytes;//榨取能提供的
return result;
}
else{//内存池剩余空间连一个区块的大小都无法提供
size_t bytes_to_get = 2 * total_bytes + ROUND_UP(heap_size >> 4);//除了两倍多分配一点
if (bytes_left > 0){//heap_size什么时候用?
obj **my_free_list = free_list + FREELIST_INDEX(bytes_left);//把剩下的榨取了转移到新内存
((obj *)start_free)->next = *my_free_list;//start_free被强制转换为obj型,然后把list地址赋值
*my_free_list = (obj *)start_free;//相当于加了个中间变量
}
start_free = (char *)malloc(bytes_to_get);//需要的空间
if (!start_free){//heap空间不足,补充内存池
obj **my_free_list = 0, *p = 0;//
for (int i = 0; i <= EMaxBytes::MAXBYTES; i += EAlign::ALIGN){//
my_free_list = free_list + FREELIST_INDEX(i);//寻找尚有未用的区块
p = *my_free_list;
if (p != 0){
*my_free_list = p->next;
start_free = (char *)p;
end_free = start_free + i;
return chunk_alloc(bytes, nobjs);//递归调用自己
}
}
end_free = 0;
//这似乎应该再去调用一次一级适配器
}
heap_size += bytes_to_get;
end_free = start_free + bytes_to_get;
return chunk_alloc(bytes, nobjs);//递归调用自己,修正数据
}
}
}