1 files changed, 129 insertions, 0 deletions
diff --git a/moses2/TranslationModel/PhraseTable.h b/moses2/TranslationModel/PhraseTable.h
new file mode 100644
index 000000000..ef40c06a4
--- /dev/null
+++ b/moses2/TranslationModel/PhraseTable.h
@@ -0,0 +1,129 @@
+/*
+ * PhraseTable.h
+ *
+ *  Created on: 23 Oct 2015
+ *      Author: hieu
+ */
+#pragma once
+#include <string>
+#include <boost/unordered_map.hpp>
+#include "../Word.h"
+#include "../HypothesisColl.h"
+#include "../FF/StatelessFeatureFunction.h"
+#include "../legacy/Util2.h"
+
+namespace Moses2
+{
+
+class System;
+class InputPathsBase;
+class InputPath;
+class Manager;
+class TargetPhrases;
+class Range;
+
+namespace SCFG
+{
+class InputPath;
+class Stacks;
+class Manager;
+class ActiveChartEntry;
+}
+
+////////////////////////////////////////////////////////////////////////
+class PhraseTable: public StatelessFeatureFunction
+{
+public:
+  int decodeGraphBackoff;
+
+  PhraseTable(size_t startInd, const std::string &line);
+  virtual ~PhraseTable();
+
+  virtual void SetParameter(const std::string& key, const std::string& value);
+  virtual void Lookup(const Manager &mgr, InputPathsBase &inputPaths) const;
+  virtual TargetPhrases *Lookup(const Manager &mgr, MemPool &pool,
+                                InputPath &inputPath) const;
+
+  void SetPtInd(size_t ind) {
+    m_ptInd = ind;
+  }
+
+  size_t GetPtInd() const {
+    return m_ptInd;
+  }
+
+  bool SatisfyBackoff(const Manager &mgr, const InputPath &path) const;
+
+  virtual void
+  EvaluateInIsolation(MemPool &pool, const System &system, const Phrase<Moses2::Word> &source,
+                      const TargetPhraseImpl &targetPhrase, Scores &scores,
+                      SCORE &estimatedScore) const;
+
+  virtual void
+  EvaluateInIsolation(MemPool &pool, const System &system, const Phrase<SCFG::Word> &source,
+                      const TargetPhrase<SCFG::Word> &targetPhrase, Scores &scores,
+                      SCORE &estimatedScore) const;
+
+  // scfg
+  virtual void InitActiveChart(
+    MemPool &pool,
+    const SCFG::Manager &mgr,
+    SCFG::InputPath &path) const = 0;
+
+  virtual void Lookup(
+    MemPool &pool,
+    const SCFG::Manager &mgr,
+    size_t maxChartSpan,
+    const SCFG::Stacks &stacks,
+    SCFG::InputPath &path) const = 0;
+
+  virtual void LookupUnary(MemPool &pool,
+                           const SCFG::Manager &mgr,
+                           const SCFG::Stacks &stacks,
+                           SCFG::InputPath &path) const;
+
+protected:
+  std::string m_path;
+  size_t m_ptInd; // in the order that it is list in [feature], NOT order of [mapping]
+  size_t m_tableLimit;
+  std::vector<FactorType> m_input, m_output;
+
+  // cache
+  size_t m_maxCacheSize; // 0 = no caching
+
+  struct CacheCollEntry2 {
+    TargetPhrases *tpsPtr;
+    clock_t clock;
+  };
+
+  // scfg
+  virtual void LookupNT(
+    MemPool &pool,
+    const SCFG::Manager &mgr,
+    const Moses2::Range &subPhraseRange,
+    const SCFG::InputPath &prevPath,
+    const SCFG::Stacks &stacks,
+    SCFG::InputPath &outPath) const;
+
+  virtual void LookupGivenWord(
+    MemPool &pool,
+    const SCFG::Manager &mgr,
+    const SCFG::InputPath &prevPath,
+    const SCFG::Word &wordSought,
+    const Moses2::Hypotheses *hypos,
+    const Moses2::Range &subPhraseRange,
+    SCFG::InputPath &outPath) const;
+
+  virtual void LookupGivenNode(
+    MemPool &pool,
+    const SCFG::Manager &mgr,
+    const SCFG::ActiveChartEntry &prevEntry,
+    const SCFG::Word &wordSought,
+    const Moses2::Hypotheses *hypos,
+    const Moses2::Range &subPhraseRange,
+    SCFG::InputPath &outPath) const = 0;
+
+};
+
+}
+