From 173910593bf6bf3dc52902f99a683560d8c73942 Mon Sep 17 00:00:00 2001
From: Kenneth Heafield <github@kheafield.com>
Date: Wed, 12 Sep 2012 15:07:44 +0100
Subject: Add the alone stuff, using a wrapper to the edge class.

---
 klm/alone/vocab.hh | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 klm/alone/vocab.hh

(limited to 'klm/alone/vocab.hh')

diff --git a/klm/alone/vocab.hh b/klm/alone/vocab.hh
new file mode 100644
index 00000000..3ac0f542
--- /dev/null
+++ b/klm/alone/vocab.hh
@@ -0,0 +1,34 @@
+#ifndef ALONE_VOCAB__
+#define ALONE_VOCAB__
+
+#include "lm/word_index.hh"
+#include "util/string_piece.hh"
+
+#include <boost/functional/hash/hash.hpp>
+#include <boost/unordered_map.hpp>
+
+#include <string>
+
+namespace lm { namespace base { class Vocabulary; } }
+
+namespace alone {
+
+class Vocab {
+  public:
+    explicit Vocab(const lm::base::Vocabulary &backing);
+
+    const std::pair<const std::string, lm::WordIndex> &FindOrAdd(const StringPiece &str);
+
+    const std::pair<const std::string, lm::WordIndex> &EndSentence() const { return end_sentence_; }
+
+  private:
+    typedef boost::unordered_map<std::string, lm::WordIndex> Map;
+    Map map_;
+
+    const lm::base::Vocabulary &backing_;
+
+    const std::pair<const std::string, lm::WordIndex> &end_sentence_;
+};
+
+} // namespace alone
+#endif // ALONE_VCOAB__
-- 
cgit v1.2.3