output alignments into file down. Alignments for n-best todo

1 files changed, 16 insertions, 1 deletions

diff --git a/moses/AlignmentInfo.cpp b/moses/AlignmentInfo.cpp
index 4dc0497e3..a75c5f58d 100644
--- a/moses/AlignmentInfo.cpp
+++ b/moses/AlignmentInfo.cpp
@@ -17,6 +17,7 @@
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 ***********************************************************************/
 #include <algorithm>
+#include <set>
 #include "util/check.hh"
 #include "AlignmentInfo.h"
 #include "TypeDef.h"
@@ -92,7 +93,21 @@ std::vector< const std::pair<size_t,size_t>* > AlignmentInfo::GetSortedAlignment
   return ret;
   
 }
-  
+
+std::vector<size_t> AlignmentInfo::GetSourceIndex2PosMap() const
+{
+  std::set<size_t> sourcePoses;
+
+  CollType::const_iterator iter;
+  for (iter = m_collection.begin(); iter != m_collection.end(); ++iter) {
+    size_t sourcePos = iter->first;
+    sourcePoses.insert(sourcePos);
+  }
+
+  std::vector<size_t> ret(sourcePoses.begin(), sourcePoses.end());
+  return ret;
+}
+
 std::ostream& operator<<(std::ostream &out, const AlignmentInfo &alignmentInfo)
 {
   AlignmentInfo::const_iterator iter;