chunk_alloc.hpp

Go to the documentation of this file.

#ifndef __STAN__MEMORY__CHUNK_ALLOC_HPP__
#define __STAN__MEMORY__CHUNK_ALLOC_HPP__
 
#include <cstdlib>
#include <cstddef>
#include <sstream>
#include <stdexcept>
#include <stdint.h> // FIXME: replace with cstddef?
#include <vector>
 
#define DEFAULT_INITIAL_NCHUNKS (1 << 8)
 
namespace stan { 
  namespace memory { 
    template<typename T, size_t Tnchunks_per_block = DEFAULT_INITIAL_NCHUNKS>
    class chunk_alloc {
    private:
      std::vector<char*> blocks_; // storage for blocks, may be bigger than cur_block_
      size_t cur_block_;          // index into blocks_ for next alloc
      size_t used_;          // index into blocks_[cur_block_] for next alloc
    public:
      
      
      chunk_alloc() :
      blocks_(1, eight_byte_aligned_malloc(sizeof(T)*Tnchunks_per_block)),
      used_(0)
      {
        if (!blocks_[0])
          throw std::bad_alloc();  // no msg allowed in bad_alloc ctor
      }
      
      ~chunk_alloc() { 
        // free ALL blocks
        for (size_t i = 0; i < blocks_.size(); ++i)
          if (blocks_[i])
            free(blocks_[i]);
      }
      
      inline void* alloc() {
        char *result;
        if (unlikely(used_ >= Tnchunks_per_block)) {
          used_ = 0;
          cur_block_++;
        }
        if (unlikely(cur_block_ >= blocks_.size())) {
          result = eight_byte_aligned_malloc(Tnchunks_per_block*sizeof(T));
          if (!result)
            throw std::bad_alloc(); // no msg allowed in bad_alloc ctor
          blocks_.push_back(result);
        }
        result = blocks_[cur_block_] + sizeof(T)*used_;
        ++used_;
        
        return (void*)result;
      }
      
      inline void recover_all() {
        cur_block_ = 0;
        used_ = 0;
      }
      
      inline void free_all() {
        // frees all BUT the first (index 0) block
        for (size_t i = 1; i < blocks_.size(); ++i)
          if (blocks_[i])
            free(blocks_[i]);
        blocks_.resize(1); 
        recover_all();
      }
    };
  }
}
#endif