1 files changed, 41 insertions, 0 deletions
diff --git a/decoder/lattice.h b/decoder/lattice.h
new file mode 100644
index 00000000..71589b92
--- /dev/null
+++ b/decoder/lattice.h
@@ -0,0 +1,41 @@
+#ifndef __LATTICE_H_
+#define __LATTICE_H_
+
+#include <string>
+#include <vector>
+#include "wordid.h"
+#include "array2d.h"
+
+class Lattice;
+struct LatticeTools {
+  static bool LooksLikePLF(const std::string &line);
+  static void ConvertTextToLattice(const std::string& text, Lattice* pl);
+  static void ConvertTextOrPLF(const std::string& text_or_plf, Lattice* pl);
+};
+
+struct LatticeArc {
+  WordID label;
+  double cost;
+  int dist2next;
+  LatticeArc() : label(), cost(), dist2next() {}
+  LatticeArc(WordID w, double c, int i) : label(w), cost(c), dist2next(i) {}
+};
+
+class Lattice : public std::vector<std::vector<LatticeArc> > {
+  friend void LatticeTools::ConvertTextOrPLF(const std::string& text_or_plf, Lattice* pl);
+ public:
+  Lattice() {}
+  explicit Lattice(size_t t, const std::vector<LatticeArc>& v = std::vector<LatticeArc>()) :
+   std::vector<std::vector<LatticeArc> >(t, v) {}
+  int Distance(int from, int to) const {
+    if (dist_.empty())
+      return (to - from);
+    return dist_(from, to);
+  }
+
+ private:
+  void ComputeDistances();
+  Array2D<int> dist_;
+};
+
+#endif