summaryrefslogtreecommitdiff
path: root/decoder/tdict.h
diff options
context:
space:
mode:
Diffstat (limited to 'decoder/tdict.h')
-rw-r--r--decoder/tdict.h30
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