1 files changed, 99 insertions, 0 deletions
diff --git a/moses2/FF/LexicalReordering/LRModel.h b/moses2/FF/LexicalReordering/LRModel.h
new file mode 100644
index 000000000..0309d5386
--- /dev/null
+++ b/moses2/FF/LexicalReordering/LRModel.h
@@ -0,0 +1,99 @@
+/*
+ * LRModel.h
+ *
+ *  Created on: 23 Mar 2016
+ *      Author: hieu
+ */
+#pragma once
+#include <string>
+
+namespace Moses2
+{
+
+class MemPool;
+class Range;
+class Bitmap;
+class LRState;
+class LexicalReordering;
+
+class LRModel
+{
+public:
+  enum ModelType {
+    Monotonic, MSD, MSLR, LeftRight, None
+  };
+  enum Direction {
+    Forward, Backward, Bidirectional
+  };
+  enum Condition {
+    F, E, FE
+  };
+
+  enum ReorderingType {
+    M = 0, // monotonic
+    NM = 1, // non-monotonic
+    S = 1, // swap
+    D = 2, // discontinuous
+    DL = 2, // discontinuous, left
+    DR = 3, // discontinuous, right
+    R = 0, // right
+    L = 1, // left
+    MAX = 3, // largest possible
+    NONE = 4  // largest possible
+  };
+
+  LRModel(const std::string &modelType, LexicalReordering &ff);
+  virtual ~LRModel();
+
+  ModelType GetModelType() const {
+    return m_modelType;
+  }
+  Direction GetDirection() const {
+    return m_direction;
+  }
+  Condition GetCondition() const {
+    return m_condition;
+  }
+
+  bool IsPhraseBased() const {
+    return m_phraseBased;
+  }
+
+  bool CollapseScores() const {
+    return m_collapseScores;
+  }
+
+  size_t GetNumberOfTypes() const;
+
+  LexicalReordering*
+  GetScoreProducer() const {
+    return m_scoreProducer;
+  }
+
+  LRState *CreateLRState(MemPool &pool) const;
+
+  ReorderingType // for first phrase in phrase-based
+  GetOrientation(Range const& cur) const;
+
+  ReorderingType // for non-first phrases in phrase-based
+  GetOrientation(Range const& prev, Range const& cur) const;
+
+  ReorderingType // for HReorderingForwardState
+  GetOrientation(Range const& prev, Range const& cur, Bitmap const& cov) const;
+
+  ReorderingType // for HReorderingBackwarddState
+  GetOrientation(int const reoDistance) const;
+
+protected:
+
+  ModelType m_modelType;
+  bool m_phraseBased;
+  bool m_collapseScores;
+  Direction m_direction;
+  Condition m_condition;
+  LexicalReordering *m_scoreProducer;
+
+};
+
+} /* namespace Moses2 */
+