|
#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); |
|
} |
|
|
|
} |
|
|
|
|
