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.cc | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 klm/alone/vocab.cc

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

diff --git a/klm/alone/vocab.cc b/klm/alone/vocab.cc
new file mode 100644
index 00000000..ffe55301
--- /dev/null
+++ b/klm/alone/vocab.cc
@@ -0,0 +1,19 @@
+#include "alone/vocab.hh"
+
+#include "lm/virtual_interface.hh"
+#include "util/string_piece.hh"
+
+namespace alone {
+
+Vocab::Vocab(const lm::base::Vocabulary &backing) : backing_(backing), end_sentence_(FindOrAdd("</s>")) {}
+
+const std::pair<const std::string, lm::WordIndex> &Vocab::FindOrAdd(const StringPiece &str) {
+  Map::const_iterator i(FindStringPiece(map_, str));
+  if (i != map_.end()) return *i;
+  std::pair<std::string, lm::WordIndex> to_ins;
+  to_ins.first.assign(str.data(), str.size());
+  to_ins.second = backing_.Index(str);
+  return *map_.insert(to_ins).first;
+}
+
+} // namespace alone
-- 
cgit v1.2.3