Fix #44: --num-iter option ignored

The input stream was not set back to beginning after end iterator was reached, so that the second iteration was immediately at the end.
HITS-AIN · Dec 9, 2019 · a1f8460 · a1f8460
1 parent d0b3b46
commit a1f8460
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/SelfOrganizingMapLib/DataIterator.h b/src/SelfOrganizingMapLib/DataIterator.h
@@ -114,6 +114,7 @@ class DataIterator
                 static_cast<std::streamsize>(layout.size() * sizeof(T)));
             ++count;
         } else {
+            is.seekg(0, is.beg);
             end_flag = true;
         }
     }

diff --git a/src/SelfOrganizingMapLib/DataIteratorShuffled.h b/src/SelfOrganizingMapLib/DataIteratorShuffled.h
@@ -125,6 +125,7 @@ class DataIteratorShuffled
                 static_cast<std::streamsize>(layout.size() * sizeof(T)));
             ++cur_random_list;
         } else {
+            is.seekg(0, is.beg);
             end_flag = true;
         }
     }