mosesdecoder/moses2/MemPool.cpp

/*
 * MemPool.cpp
 *
 *  Created on: 28 Oct 2015
 *      Author: hieu
 */

#include <boost/foreach.hpp>
#include "MemPool.h"
#include "util/scoped.hh"
#include "legacy/Util2.h"

using namespace std;

namespace Moses2
{

MemPool::Page::Page(std::size_t vSize) :
  size(vSize)
{
  mem = (uint8_t*) util::MallocOrThrow(size);
  end = mem + size;
}

MemPool::Page::~Page()
{
  free(mem);
}
////////////////////////////////////////////////////
MemPool::MemPool(size_t initSize) :
  m_currSize(initSize), m_currPage(0)
{
  Page *page = new Page(m_currSize);
  m_pages.push_back(page);

  current_ = page->mem;
  //cerr << "new memory pool";
}

MemPool::~MemPool()
{
  //cerr << "delete memory pool" << endl;
  RemoveAllInColl(m_pages);
}

uint8_t *MemPool::More(std::size_t size)
{
  ++m_currPage;
  if (m_currPage >= m_pages.size()) {
    // add new page
    m_currSize <<= 1;
    std::size_t amount = std::max(m_currSize, size);

    Page *page = new Page(amount);
    m_pages.push_back(page);

    uint8_t *ret = page->mem;
    current_ = ret + size;
    return ret;
  } else {
    // use existing page
    Page &page = *m_pages[m_currPage];
    if (size <= page.size) {
      uint8_t *ret = page.mem;
      current_ = ret + size;
      return ret;
    } else {
      // recursive call More()
      return More(size);
    }
  }
}

void MemPool::Reset()
{
  m_currPage = 0;
  current_ = m_pages[0]->mem;
}

}