ngram base dist

1 files changed, 25 insertions, 0 deletions

diff --git a/gi/pf/ngram_base.h b/gi/pf/ngram_base.h
new file mode 100644
index 00000000..4ea999f3
--- /dev/null
+++ b/gi/pf/ngram_base.h
@@ -0,0 +1,25 @@
+#ifndef _NGRAM_BASE_H_
+#define _NGRAM_BASE_H_
+
+#include <string>
+#include <vector>
+#include "trule.h"
+#include "wordid.h"
+#include "prob.h"
+
+struct FixedNgramBaseImpl;
+struct FixedNgramBase {
+  FixedNgramBase(const std::string& lmfname);
+  ~FixedNgramBase();
+  prob_t StringProbability(const std::vector<WordID>& s) const;
+
+  prob_t operator()(const TRule& rule) const {
+    return StringProbability(rule.e_);
+  }
+
+ private:
+  FixedNgramBaseImpl* impl;
+
+};
+
+#endif