#ifndef DICT_H_ #define DICT_H_ #include <cassert> #include <cstring> #include <string> #include <vector> #include "hash.h" #include "wordid.h" class Dict { typedef //HASH_MAP<std::string, WordID, boost::hash<std::string> > Map; HASH_MAP<std::string, WordID> Map; public: Dict() : b0_("<bad0>") { HASH_MAP_EMPTY(d_,"<bad1>"); words_.reserve(1000); } inline int max() const { return words_.size(); } static bool is_ws(char x) { return (x == ' ' || x == '\t'); } inline void ConvertWhitespaceDelimitedLine(const std::string& line, std::vector<int>* out) { size_t cur = 0; size_t last = 0; int state = 0; out->clear(); while(cur < line.size()) { if (is_ws(line[cur++])) { if (state == 0) continue; out->push_back(Convert(line.substr(last, cur - last - 1))); state = 0; } else { if (state == 1) continue; last = cur - 1; state = 1; } } if (state == 1) out->push_back(Convert(line.substr(last, cur - last))); } inline WordID Convert(const std::string& word, bool frozen = false) { Map::iterator i = d_.find(word); if (i == d_.end()) { if (frozen) return 0; words_.push_back(word); d_[word] = words_.size(); return words_.size(); } else { return i->second; } } inline WordID Convert(const std::vector<std::string>& words, bool frozen = false) { return Convert(toString(words), frozen); } static inline std::string toString(const std::vector<std::string>& words) { std::string word= ""; for (std::vector<std::string>::const_iterator it=words.begin(); it != words.end(); ++it) { if (it != words.begin()) word += " ||| "; word += *it; } return word; } inline const std::string& Convert(const WordID& id) const { if (id == 0) return b0_; assert(id <= (int)words_.size()); return words_[id-1]; } void AsVector(const WordID& id, std::vector<std::string>* results) const; void clear() { words_.clear(); d_.clear(); } private: const std::string b0_; std::vector<std::string> words_; Map d_; }; #endif