| #include <cassert> |
| #include <iostream> |
| #include "Ngram.h" |
|
|
| using namespace std; |
| Vocab vocab; |
| Ngram* ngram = NULL; |
|
|
| extern "C" { |
|
|
| void srilm_init(const char* fname, int order) { |
| cerr << "Loading " << order << "-gram LM: " << fname << endl; |
| File file(fname, "r", 0); |
| assert(file); |
| ngram = new Ngram(vocab, order); |
| ngram->read(file, false); |
| cerr << "Done\n"; |
| } |
|
|
| int srilm_getvoc(const char* word) { |
| return vocab.getIndex((VocabString)word); |
| } |
|
|
| float srilm_wordprob(int w, int* context) { |
| return (float)ngram->wordProb(w, (VocabIndex*)context); |
| } |
|
|
| } |
|
|
|
|
