starting placeholder

1 files changed, 39 insertions, 2 deletions

diff --git a/moses/InputPath.cpp b/moses/InputPath.cpp
index 09fed6588..9059b03bb 100644
--- a/moses/InputPath.cpp
+++ b/moses/InputPath.cpp
@@ -3,6 +3,9 @@
 #include "TargetPhraseCollection.h"
 #include "StaticData.h"
 #include "TypeDef.h"
+#include "AlignmentInfo.h"
+
+using namespace std;
 
 namespace Moses
 {
@@ -60,8 +63,42 @@ void InputPath::SetTargetPhrases(const PhraseDictionary &phraseDictionary
                                  , const TargetPhraseCollection *targetPhrases
                                  , const void *ptNode)
 {
-  std::pair<const TargetPhraseCollection*, const void*> value(targetPhrases, ptNode);
-  m_targetPhrases[&phraseDictionary] = value;
+  FactorType placeholderFactor = StaticData::Instance().GetPlaceholderFactor();
+  if (targetPhrases == NULL || placeholderFactor == NOT_FOUND || m_placeholders.size() == 0) {
+    // use all of the target phrase given
+    std::pair<const TargetPhraseCollection*, const void*> value(targetPhrases, ptNode);
+    m_targetPhrases[&phraseDictionary] = value;
+  }
+  else {
+    // filter out target phrases with alignments that are not 1-to-1 with placeholder
+	m_copiedSet.push_back(TargetPhraseCollection());
+	TargetPhraseCollection &newTargetPhrases = m_copiedSet.back();
+    std::pair<const TargetPhraseCollection*, const void*> value(&newTargetPhrases, ptNode);
+    m_targetPhrases[&phraseDictionary] = value;
+
+    TargetPhraseCollection::const_iterator iter;
+    for (iter = targetPhrases->begin(); iter != targetPhrases->end(); ++iter) {
+    	TargetPhrase *targetPhrase = *iter;
+    	const AlignmentInfo &alignments = targetPhrase->GetAlignTerm();
+    	bool ok = IsCompatibleWithPlaceholders(alignments);
+
+    	if (ok) {
+    		newTargetPhrases.Add(targetPhrase);
+    	}
+    }
+  }
+}
+
+bool InputPath::IsCompatibleWithPlaceholders(const AlignmentInfo &alignments) const
+{
+  for (size_t i = 0; i < m_placeholders.size(); ++i) {
+	  size_t sourcePos = m_placeholders[i];
+	  set<size_t> targetPos = alignments.GetAlignmentsForSource(sourcePos);
+	  if (targetPos.size() != 1) {
+		  return false;
+	  }
+  }
+  return true;
 }
 
 std::ostream& operator<<(std::ostream& out, const InputPath& obj)