1 files changed, 74 insertions, 0 deletions
diff --git a/contrib/moses2/SCFG/nbest/KBestExtractor.cpp b/contrib/moses2/SCFG/nbest/KBestExtractor.cpp
new file mode 100644
index 000000000..87b9c86d5
--- /dev/null
+++ b/contrib/moses2/SCFG/nbest/KBestExtractor.cpp
@@ -0,0 +1,74 @@
+/*
+ * KBestExtractor.cpp
+ *
+ *  Created on: 2 Aug 2016
+ *      Author: hieu
+ */
+#include <boost/foreach.hpp>
+#include <sstream>
+#include "KBestExtractor.h"
+#include "../Manager.h"
+#include "../Hypothesis.h"
+#include "../Stacks.h"
+#include "../Stack.h"
+#include "../Sentence.h"
+#include "../../System.h"
+#include "../../Scores.h"
+#include "../../legacy/Util2.h"
+
+using namespace std;
+
+namespace Moses2
+{
+//bool g_debug = false;
+
+namespace SCFG
+{
+/////////////////////////////////////////////////////////////
+KBestExtractor::KBestExtractor(const SCFG::Manager &mgr)
+:m_mgr(mgr)
+{
+
+}
+
+KBestExtractor::~KBestExtractor()
+{
+}
+
+void KBestExtractor::OutputToStream(std::stringstream &strm)
+{
+	//cerr << "1" << flush;
+	const Stack &lastStack = m_mgr.GetStacks().GetLastStack();
+	UTIL_THROW_IF2(lastStack.GetColl().size() != 1, "Only suppose to be 1 hypo coll in last stack");
+	UTIL_THROW_IF2(lastStack.GetColl().begin()->second == NULL, "NULL hypo collection");
+
+	const Hypotheses &hypos = lastStack.GetColl().begin()->second->GetSortedAndPrunedHypos();
+	UTIL_THROW_IF2(hypos.size() != 1, "Only suppose to be 1 hypo in collection");
+	const HypothesisBase *hypo = hypos[0];
+
+	const ArcLists &arcLists = m_mgr.arcLists;
+	const ArcList &arcList = arcLists.GetArcList(hypo);
+	NBests &nbests = m_nbestColl.GetOrCreateNBests(m_mgr, arcList);
+
+	size_t ind = 0;
+	while (nbests.Extend(m_mgr, m_nbestColl, ind)) {
+		const NBest &deriv = nbests.Get(ind);
+		strm << m_mgr.GetTranslationId() << " ||| ";
+		//cerr << "1" << flush;
+		strm << deriv.GetStringExclSentenceMarkers();
+		//cerr << "2" << flush;
+		strm << " ||| ";
+		deriv.GetScores().OutputBreakdown(strm, m_mgr.system);
+		//cerr << "3" << flush;
+		strm << "||| ";
+		strm << deriv.GetScores().GetTotalScore();
+		//cerr << "4" << flush;
+
+		strm << endl;
+
+		++ind;
+	}
+}
+
+}
+} /* namespace Moses2 */