1 files changed, 285 insertions, 0 deletions
diff --git a/contrib/moses2/Scores.cpp b/contrib/moses2/Scores.cpp
new file mode 100644
index 000000000..b6e731807
--- /dev/null
+++ b/contrib/moses2/Scores.cpp
@@ -0,0 +1,285 @@
+/*
+ * Scores.cpp
+ *
+ *  Created on: 23 Oct 2015
+ *      Author: hieu
+ */
+
+#include <boost/foreach.hpp>
+#include <vector>
+#include <cstddef>
+#include <stdio.h>
+#include "Scores.h"
+#include "Weights.h"
+#include "System.h"
+#include "FF/FeatureFunction.h"
+#include "FF/FeatureFunctions.h"
+#include "legacy/Util2.h"
+
+using namespace std;
+
+namespace Moses2
+{
+
+Scores::Scores(const System &system, MemPool &pool, size_t numScores) :
+    m_total(0)
+{
+  if (system.options.nbest.nbest_size) {
+    m_scores = new (pool.Allocate<SCORE>(numScores)) SCORE[numScores];
+    Init<SCORE>(m_scores, numScores, 0);
+  }
+  else {
+    m_scores = NULL;
+  }
+}
+
+Scores::Scores(const System &system, MemPool &pool, size_t numScores,
+    const Scores &origScores) :
+    m_total(origScores.m_total)
+{
+  if (system.options.nbest.nbest_size) {
+    m_scores = new (pool.Allocate<SCORE>(numScores)) SCORE[numScores];
+    memcpy(m_scores, origScores.m_scores, sizeof(SCORE) * numScores);
+  }
+  else {
+    m_scores = NULL;
+  }
+}
+
+Scores::~Scores()
+{
+
+}
+
+const SCORE *Scores::GetScores(const FeatureFunction &featureFunction) const
+{
+  assert(m_scores);
+  size_t ffStartInd = featureFunction.GetStartInd();
+  const SCORE &scores = m_scores[ffStartInd];
+  return &scores;
+}
+
+void Scores::Reset(const System &system)
+{
+  if (system.options.nbest.nbest_size) {
+    size_t numScores = system.featureFunctions.GetNumScores();
+    Init<SCORE>(m_scores, numScores, 0);
+  }
+  m_total = 0;
+}
+
+void Scores::PlusEquals(const System &system,
+    const FeatureFunction &featureFunction, const SCORE &score)
+{
+  assert(featureFunction.GetNumScores() == 1);
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  if (system.options.nbest.nbest_size) {
+    m_scores[ffStartInd] += score;
+  }
+  SCORE weight = weights[ffStartInd];
+  m_total += score * weight;
+}
+
+void Scores::PlusEquals(const System &system,
+    const FeatureFunction &featureFunction, const SCORE &score, size_t offset)
+{
+  assert(offset < featureFunction.GetNumScores());
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  if (system.options.nbest.nbest_size) {
+    m_scores[ffStartInd + offset] += score;
+  }
+  SCORE weight = weights[ffStartInd + offset];
+  m_total += score * weight;
+}
+
+void Scores::PlusEquals(const System &system,
+    const FeatureFunction &featureFunction, const std::vector<SCORE> &scores)
+{
+  assert(scores.size() == featureFunction.GetNumScores());
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  for (size_t i = 0; i < scores.size(); ++i) {
+    SCORE incrScore = scores[i];
+    if (system.options.nbest.nbest_size) {
+      m_scores[ffStartInd + i] += incrScore;
+    }
+    //cerr << "ffStartInd=" << ffStartInd << " " << i << endl;
+    SCORE weight = weights[ffStartInd + i];
+    m_total += incrScore * weight;
+  }
+}
+
+void Scores::PlusEquals(const System &system,
+    const FeatureFunction &featureFunction, SCORE scores[])
+{
+  //assert(scores.size() == featureFunction.GetNumScores());
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  for (size_t i = 0; i < featureFunction.GetNumScores(); ++i) {
+    SCORE incrScore = scores[i];
+    if (system.options.nbest.nbest_size) {
+      m_scores[ffStartInd + i] += incrScore;
+    }
+    //cerr << "ffStartInd=" << ffStartInd << " " << i << endl;
+    SCORE weight = weights[ffStartInd + i];
+    m_total += incrScore * weight;
+  }
+}
+
+void Scores::PlusEquals(const System &system, const Scores &other)
+{
+  size_t numScores = system.featureFunctions.GetNumScores();
+  if (system.options.nbest.nbest_size) {
+    for (size_t i = 0; i < numScores; ++i) {
+      m_scores[i] += other.m_scores[i];
+    }
+  }
+  m_total += other.m_total;
+}
+
+void Scores::MinusEquals(const System &system, const Scores &other)
+{
+  size_t numScores = system.featureFunctions.GetNumScores();
+  if (system.options.nbest.nbest_size) {
+    for (size_t i = 0; i < numScores; ++i) {
+      m_scores[i] -= other.m_scores[i];
+    }
+  }
+  m_total -= other.m_total;
+}
+
+void Scores::Assign(const System &system,
+    const FeatureFunction &featureFunction, const SCORE &score)
+{
+  assert(featureFunction.GetNumScores() == 1);
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+
+  if (system.options.nbest.nbest_size) {
+    assert(m_scores[ffStartInd] == 0);
+    m_scores[ffStartInd] = score;
+  }
+  SCORE weight = weights[ffStartInd];
+  m_total += score * weight;
+
+}
+
+void Scores::Assign(const System &system,
+    const FeatureFunction &featureFunction, const std::vector<SCORE> &scores)
+{
+  assert(scores.size() == featureFunction.GetNumScores());
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  for (size_t i = 0; i < scores.size(); ++i) {
+    SCORE incrScore = scores[i];
+
+    if (system.options.nbest.nbest_size) {
+      assert(m_scores[ffStartInd + i] == 0);
+      m_scores[ffStartInd + i] = incrScore;
+    }
+    //cerr << "ffStartInd=" << ffStartInd << " " << i << endl;
+    SCORE weight = weights[ffStartInd + i];
+    m_total += incrScore * weight;
+  }
+}
+
+void Scores::CreateFromString(const std::string &str,
+    const FeatureFunction &featureFunction, const System &system,
+    bool transformScores)
+{
+  vector<SCORE> scores = Tokenize<SCORE>(str);
+  if (transformScores) {
+    std::transform(scores.begin(), scores.end(), scores.begin(),
+        TransformScore);
+    std::transform(scores.begin(), scores.end(), scores.begin(), FloorScore);
+  }
+
+  /*
+   std::copy(scores.begin(),scores.end(),
+   std::ostream_iterator<SCORE>(cerr," "));
+   */
+
+  PlusEquals(system, featureFunction, scores);
+}
+
+std::string Scores::Debug(const System &system) const
+{
+  stringstream out;
+  out << "total=" << m_total;
+
+  if (system.options.nbest.nbest_size) {
+    out << ", ";
+    BOOST_FOREACH(const FeatureFunction *ff, system.featureFunctions.GetFeatureFunctions()){
+      out << ff->GetName() << "= ";
+      for (size_t i = ff->GetStartInd(); i < (ff->GetStartInd() + ff->GetNumScores()); ++i) {
+        out << m_scores[i] << " ";
+      }
+    }
+  }
+
+  return out.str();
+}
+
+void Scores::OutputBreakdown(std::ostream &out, const System &system) const
+{
+  if (system.options.nbest.nbest_size) {
+    BOOST_FOREACH(const FeatureFunction *ff, system.featureFunctions.GetFeatureFunctions()){
+      if (ff->IsTuneable()) {
+        out << ff->GetName() << "= ";
+        for (size_t i = ff->GetStartInd(); i < (ff->GetStartInd() + ff->GetNumScores()); ++i) {
+          out << m_scores[i] << " ";
+        }
+      }
+    }
+  }
+}
+
+// static functions to work out estimated scores
+SCORE Scores::CalcWeightedScore(const System &system,
+    const FeatureFunction &featureFunction, SCORE scores[])
+{
+  SCORE ret = 0;
+
+  const Weights &weights = system.weights;
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  for (size_t i = 0; i < featureFunction.GetNumScores(); ++i) {
+    SCORE incrScore = scores[i];
+
+    //cerr << "ffStartInd=" << ffStartInd << " " << i << endl;
+    SCORE weight = weights[ffStartInd + i];
+    ret += incrScore * weight;
+  }
+
+  return ret;
+}
+
+SCORE Scores::CalcWeightedScore(const System &system,
+    const FeatureFunction &featureFunction, SCORE score)
+{
+  const Weights &weights = system.weights;
+  assert(featureFunction.GetNumScores() == 1);
+
+  size_t ffStartInd = featureFunction.GetStartInd();
+  SCORE weight = weights[ffStartInd];
+  SCORE ret = score * weight;
+
+  return ret;
+}
+
+}
+