diff options
Diffstat (limited to 'decoder/tdict.h')
-rw-r--r-- | decoder/tdict.h | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/decoder/tdict.h b/decoder/tdict.h new file mode 100644 index 00000000..31f66367 --- /dev/null +++ b/decoder/tdict.h @@ -0,0 +1,30 @@ +#ifndef _TDICT_H_ +#define _TDICT_H_ + +#include <string> +#include <vector> +#include "wordid.h" + +class Vocab; + +struct TD { + static Vocab* dict_; + static void ConvertSentence(const std::string& sent, std::vector<WordID>* ids); + static void GetWordIDs(const std::vector<std::string>& strings, std::vector<WordID>* ids); + static std::string GetString(const std::vector<WordID>& str); + static int AppendString(const WordID& w, int pos, int bufsize, char* buffer) { + const char* word = TD::Convert(w); + const char* const end_buf = buffer + bufsize; + char* dest = buffer + pos; + while(dest < end_buf && *word) { + *dest = *word; + ++dest; + ++word; + } + return (dest - buffer); + } + static WordID Convert(const std::string& s); + static const char* Convert(const WordID& w); +}; + +#endif |