1 files changed, 55 insertions, 0 deletions
diff --git a/vest/scorer.h b/vest/scorer.h
new file mode 100644
index 00000000..83d4db4c
--- /dev/null
+++ b/vest/scorer.h
@@ -0,0 +1,55 @@
+#ifndef SCORER_H_
+#define SCORER_H_
+
+#include <vector>
+#include <string>
+
+#include "wordid.h"
+
+class ViterbiEnvelope;
+class ErrorSurface;
+class Hypergraph;  // needed for alignment
+
+enum ScoreType { IBM_BLEU, NIST_BLEU, Koehn_BLEU, TER, BLEU_minus_TER_over_2, SER, AER };
+ScoreType ScoreTypeFromString(const std::string& st);
+
+class Score {
+ public:
+  virtual ~Score();
+  virtual float ComputeScore() const = 0;
+  virtual void ScoreDetails(std::string* details) const = 0;
+  virtual void PlusEquals(const Score& rhs) = 0;
+  virtual void Subtract(const Score& rhs, Score* res) const = 0;
+  virtual Score* GetZero() const = 0;
+  virtual bool IsAdditiveIdentity() const = 0; // returns true if adding this delta
+                                      // to another score results in no score change
+				      // under any circumstances
+  virtual void Encode(std::string* out) const = 0;
+};
+
+class SentenceScorer {
+ public:
+  virtual ~SentenceScorer();
+  void ComputeErrorSurface(const ViterbiEnvelope& ve, ErrorSurface* es, const ScoreType type, const Hypergraph& hg) const;
+  virtual Score* ScoreCandidate(const std::vector<WordID>& hyp) const = 0;
+  virtual const std::string* GetSource() const;
+  static Score* CreateScoreFromString(const ScoreType type, const std::string& in);
+  static SentenceScorer* CreateSentenceScorer(const ScoreType type,
+    const std::vector<std::vector<WordID> >& refs,
+    const std::string& src = "");
+};
+
+class DocScorer {
+ public:
+  ~DocScorer();
+  DocScorer(
+    const ScoreType type,
+    const std::vector<std::string>& ref_files,
+    const std::string& src_file = "");
+  int size() const { return scorers_.size(); }
+  const SentenceScorer* operator[](size_t i) const { return scorers_[i]; }
+ private:
+  std::vector<SentenceScorer*> scorers_;
+};
+
+#endif