mosesdecoder/contrib/moses2/SCFG/Manager.h

/*
 * Manager.h
 *
 *  Created on: 23 Oct 2015
 *      Author: hieu
 */

#pragma once

#include <queue>
#include <cstddef>
#include <string>
#include <deque>
#include "../ManagerBase.h"
#include "Stacks.h"
#include "InputPaths.h"
#include "Misc.h"

namespace Moses2
{

namespace SCFG
{
class SymbolBind;
class TargetPhraseImpl;
class SymbolBindElement;

class Manager: public Moses2::ManagerBase
{
public:
  Manager(System &sys, const TranslationTask &task, const std::string &inputStr,
      long translationId);

  virtual ~Manager();
  void Decode();
  std::string OutputBest() const;
  std::string OutputNBest();
  std::string OutputTransOpt();

  const InputPaths &GetInputPaths() const
  { return m_inputPaths; }

  QueueItemRecycler &GetQueueItemRecycler()
  { return m_queueItemRecycler; }

  const Stacks &GetStacks() const
  { return m_stacks; }

protected:
  Stacks m_stacks;
  SCFG::InputPaths m_inputPaths;

  void InitActiveChart(SCFG::InputPath &path);
  void Lookup(SCFG::InputPath &path);
  void LookupUnary(SCFG::InputPath &path);
  void Decode(SCFG::InputPath &path, Stack &stack);

  void ExpandHypo(
      const SCFG::InputPath &path,
      const SCFG::SymbolBind &symbolBind,
      const SCFG::TargetPhraseImpl &tp,
      Stack &stack);

  bool IncrPrevHypoIndices(
      Vector<size_t> &prevHyposIndices,
      size_t ind,
      const std::vector<const SymbolBindElement*> ntEles);

  // cube pruning
  Queue m_queue;
  SeenPositions m_seenPositions;

  QueueItemRecycler m_queueItemRecycler;

  void CreateQueue(
      const SCFG::InputPath &path,
      const SymbolBind &symbolBind,
      const SCFG::TargetPhrases &tps);
};

}
}