From 45c9efc7d558dbe160056f02e74df1fee5d2d0e5 Mon Sep 17 00:00:00 2001
From: Kenneth Heafield <github@kheafield.com>
Date: Tue, 11 Sep 2012 14:30:16 +0100
Subject: Add search library to cdec (not used yet)

---
 klm/search/word.hh | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 klm/search/word.hh

(limited to 'klm/search/word.hh')

diff --git a/klm/search/word.hh b/klm/search/word.hh
new file mode 100644
index 00000000..e7a15be9
--- /dev/null
+++ b/klm/search/word.hh
@@ -0,0 +1,47 @@
+#ifndef SEARCH_WORD__
+#define SEARCH_WORD__
+
+#include "lm/word_index.hh"
+
+#include <boost/functional/hash.hpp>
+
+#include <string>
+#include <utility>
+
+namespace search {
+
+class Word {
+  public:
+    // Construct a non-terminal.
+    Word() : entry_(NULL) {}
+
+    explicit Word(const std::pair<const std::string, lm::WordIndex> &entry) {
+      entry_ = &entry;
+    }
+
+    // Returns true for two non-terminals even if their labels are different (since we don't care about labels).
+    bool operator==(const Word &other) const {
+      return entry_ == other.entry_;
+    }
+
+    bool Terminal() const { return entry_ != NULL; }
+
+    const std::string &String() const { return entry_->first; }
+
+    lm::WordIndex Index() const { return entry_->second; }
+
+  protected:
+    friend size_t hash_value(const Word &word);
+
+    const std::pair<const std::string, lm::WordIndex> *Entry() const { return entry_; }
+
+  private:
+    const std::pair<const std::string, lm::WordIndex> *entry_;
+};
+
+inline size_t hash_value(const Word &word) {
+  return boost::hash_value(word.Entry());
+}
+
+} // namespace search
+#endif // SEARCH_WORD__
-- 
cgit v1.2.3