placeholder for chart decoding

1 files changed, 59 insertions, 0 deletions

diff --git a/moses/ChartTranslationOptions.cpp b/moses/ChartTranslationOptions.cpp
index 196becc01..bcfca72a5 100644
--- a/moses/ChartTranslationOptions.cpp
+++ b/moses/ChartTranslationOptions.cpp
@@ -70,6 +70,11 @@ float ChartTranslationOptions::CalcEstimateOfBestScore(
 
 void ChartTranslationOptions::Evaluate(const InputType &input, const InputPath &inputPath)
 {
+  SetInputPath(&inputPath);
+  if (StaticData::Instance().GetPlaceholderFactor() != NOT_FOUND) {
+	  CreateSourceRuleFromInputPath();
+  }
+
   CollType::iterator iter;
   for (iter = m_collection.begin(); iter != m_collection.end(); ++iter) {
     ChartTranslationOption &transOpt = **iter;
@@ -79,4 +84,58 @@ void ChartTranslationOptions::Evaluate(const InputType &input, const InputPath &
 
 }
 
+void ChartTranslationOptions::SetInputPath(const InputPath *inputPath)
+{
+  CollType::iterator iter;
+  for (iter = m_collection.begin(); iter != m_collection.end(); ++iter) {
+	ChartTranslationOption &transOpt = **iter;
+	transOpt.SetInputPath(inputPath);
+  }
+}
+
+void ChartTranslationOptions::CreateSourceRuleFromInputPath()
+{
+	if (m_collection.size() == 0) {
+		return;
+	}
+
+	const InputPath *inputPath = m_collection.front()->GetInputPath();
+	CHECK(inputPath);
+	std::vector<const Word*> &ruleSourceFromInputPath = inputPath->AddRuleSourceFromInputPath();
+
+	size_t chartCellIndex = 0;
+	const ChartCellLabel *chartCellLabel = (chartCellIndex < m_stackVec.size()) ? m_stackVec[chartCellIndex] : NULL;
+
+	size_t ind = 0;
+	for (size_t sourcePos = m_wordsRange->GetStartPos(); sourcePos <= m_wordsRange->GetEndPos(); ++sourcePos, ++ind) {
+		if (chartCellLabel) {
+			if (sourcePos == chartCellLabel->GetCoverage().GetEndPos()) {
+				// end of child range. push an empty word to denote non-term
+				ruleSourceFromInputPath.push_back(NULL);
+				++chartCellIndex;
+				chartCellLabel = (chartCellIndex < m_stackVec.size()) ? m_stackVec[chartCellIndex] : NULL;
+			}
+			else if (sourcePos >= chartCellLabel->GetCoverage().GetStartPos()) {
+				// in the range of child hypo. do nothing
+			}
+			else {
+				// not yet reached child range. add word
+				ruleSourceFromInputPath.push_back(&inputPath->GetPhrase().GetWord(ind));
+			}
+		}
+		else {
+			// no child in sight. add word
+			ruleSourceFromInputPath.push_back(&inputPath->GetPhrase().GetWord(ind));
+		}
+	}
+
+	// save it to each trans opt
+  CollType::iterator iter;
+  for (iter = m_collection.begin(); iter != m_collection.end(); ++iter) {
+	ChartTranslationOption &transOpt = **iter;
+	transOpt.SetSourceRuleFromInputPath(&ruleSourceFromInputPath);
+  }
+
+}
+
 }