mosesdecoder/contrib/other-builds/moses2/Recycler.h

/*
 * Recycler.h
 *
 *  Created on: 2 Jan 2016
 *      Author: hieu
 */
#pragma once

#include <cstddef>
#include <deque>
#include <vector>

namespace Moses2
{

template<typename T>
class Recycler
{
public:
  Recycler() :
      m_currInd(0)
  {
  }
  virtual ~Recycler()
  {
  }

  T Get()
  {
    if (!m_coll.empty()) {
      T &obj = m_coll.back();
      m_coll.pop_back();
      return obj;
    }
    else if (m_currInd) {
      --m_currInd;
      T &obj = m_all[m_currInd];
      return obj;
    }
    else {
      return NULL;
    }
  }

  void Clear()
  {
    m_coll.clear();
    m_currInd = m_all.size();
  }

  void Keep(const T& val)
  {
    m_all.push_back(val);
  }

  void Recycle(const T& val)
  {
    m_coll.push_back(val);
  }

protected:
  // all objects we're looking after
  std::vector<T> m_all;

  // pointer to the object that's just been given out.
  // to give out another obj, must decrement THEN give out
  size_t m_currInd;

  // objects that have been give back to us
  std::deque<T> m_coll;
};

} /* namespace Moses2 */