mosesdecoder/moses2/TrellisPaths.h

/*
 * TrellisPaths.h
 *
 *  Created on: 16 Mar 2016
 *      Author: hieu
 */
#pragma once

#include <vector>
#include <queue>
#include "PhraseBased/TrellisPath.h"

namespace Moses2
{

template<typename T>
struct CompareTrellisPath
{
  bool operator()(const T* pathA, const T* pathB) const
  {
    return (pathA->GetFutureScore() < pathB->GetFutureScore());
  }
};

template<typename T>
class TrellisPaths
{
public:
  TrellisPaths() {}

  virtual ~TrellisPaths()
  {
    while (!empty()) {
      T *path = Get();
      delete path;
    }
  }

  bool empty() const
  {
    return m_coll.empty();
  }

  //! add a new entry into collection
  void Add(T *trellisPath)
  {
    m_coll.push(trellisPath);
  }

  T *Get()
  {
    T *top = m_coll.top();

    // Detach
    m_coll.pop();
    return top;
  }

  size_t GetSize() const
  { return m_coll.size(); }

protected:
  typedef std::priority_queue<T*, std::vector<T*>,
      CompareTrellisPath<T> > CollectionType;
  CollectionType m_coll;
};

} /* namespace Moses2 */