STL 源码解析中构建的小型空间适配器

/******************************************************************************* File name: jjalloc Description: stl 源码解析中构建的小型空间适配器 *******************************************************************************/ #ifndef _JJALLOC_ #define _JJALLOC_ #include <new> // for placement new #include <cstddef> // for ptrdiff_t, size_t #include <cstdlib> // for exit() #include <climits> // for UINT_MAX #include <iostream> // for cerr namespace JJ { template <class T> inline T* _allocate(ptrdiff_t size, T*) // typedef int ptrdiff_t; { set_new_handler(0); // 用来抛出bad_alloc异常 T* tmp = (T*)(::operator new((size_t)(size * sizeof(T)))); // typedef unsigned int size_t; if(tmp == 0) { cerr << "out of memory" << endl; exit(1); } return tmp; } template <class T> inline void _deallocate(T* buffer) { ::operator delete(buffer); } template <class T1, class T2> inline void _construct(T1* p, const T2& value) { new(p) T1(value); // placement new, invoke ctor of T1 } template <class T> inline void _destroy(T* ptr) { ptr->~T(); } 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; // rebind allocator of type U template <class U> struct rebind { typedef allocator<U> other; }; pointer allocate(size_type n, const void* hint=0) { return _allocate((difference_type)n, (pointer)0); } void deallocate(pointer p, size_type n) { _deallocate(p); } void construct(pointer p, const T& value) { _construct(p, value); } void destroy(pointer p) { _destroy(p); } pointer address(reference x) { return (pointer)&x; } const_pointer const_address(const_reference x) { return (const_pointer)&x; } size_type max_size() const { return size_type(UINT_MAX/sizeof(T)); } }; }; #endif    

 

 

 // FileName: jjalloc.cpp // Description: jjaaloc的测试 #include "jjalloc.h" #include <vector> #include <iostream> using namespace std; int main() { int iA[5] = {0, 1, 2, 3, 4}; unsigned int i; vector<int, JJ::allocator<int> > iv(iA, iA+5); for(i = 0; i < iv.size(); i++) { cout << iv[i] << endl; } return 0; }

 

 

需要理解的是这几个函数 

set_new_handler(0); // 用来抛出bad_alloc异常

在vc6.0中找到set_new_handler()定义处：

  // new standard header

  typedef void (__cdecl *new_handler)();
  new_handler __cdecl set_new_handler(new_handler) _THROW0();

可见 new_handler 是一个(__cdecl void *)(void) 的函数指针即：指向一个参数和返回值都为空的函数指针

而 set_new_handle() 是一个输入并返回 new_handler 的函数 

_THROW0() 表示不会抛出异常

      new_handle 将在new出错的时候时候被调用，去处理出错的指针，set_new_handle()返回值是没有调用前就已经存在的旧的处理函数指针。默认的new_handle 会抛出bad_alloc异常，new_handle=NULL也会抛出bad_alloc异常。set_new_handle()设置自定义的new_handle 

::operator new((size_t)(size * sizeof(T)))

使用全局的new申请到size个T空间

 

 new(p) T2(value)；   // placement new是用来实现定位构造的

new的第三种用法，即 在p指针指向的位置够找一个T2(value)；  

void main()
{
   char s[sizeof(A)];
   A* p = (A*)s;
   new(p) A(3); //p->A::A(3);  如果没有这句，相当于没有调用拷贝构造函数
   p->Say();
}