Merge remote-tracking branch 'upstream/master'

1 files changed, 4 insertions, 11 deletions

diff --git a/extractor/alignment.cc b/extractor/alignment.cc
index b187c03a..2278c825 100644
--- a/extractor/alignment.cc
+++ b/extractor/alignment.cc
@@ -23,8 +23,8 @@ Alignment::Alignment(const string& filename) {
     boost::split(items, line, boost::is_any_of(" -"));
     vector<pair<int, int>> alignment;
     alignment.reserve(items.size() / 2);
-    for (size_t i = 0; i < items.size(); i += 2) {
-      alignment.push_back(make_pair(stoi(items[i]), stoi(items[i + 1])));
+    for (size_t i = 1; i < items.size(); i += 2) {
+      alignment.push_back(make_pair(stoi(items[i - 1]), stoi(items[i])));
     }
     alignments.push_back(alignment);
   }
@@ -39,15 +39,8 @@ vector<pair<int, int>> Alignment::GetLinks(int sentence_index) const {
   return alignments[sentence_index];
 }
 
-void Alignment::WriteBinary(const fs::path& filepath) {
-  FILE* file = fopen(filepath.string().c_str(), "w");
-  int size = alignments.size();
-  fwrite(&size, sizeof(int), 1, file);
-  for (vector<pair<int, int>> alignment: alignments) {
-    size = alignment.size();
-    fwrite(&size, sizeof(int), 1, file);
-    fwrite(alignment.data(), sizeof(pair<int, int>), size, file);
-  }
+bool Alignment::operator==(const Alignment& other) const {
+  return alignments == other.alignments;
 }
 
 } // namespace extractor