Prints the estimated remaining time.

diff --git a/flatland.cc b/flatland.cc
index 75e9862..075e073 100644 (file)
--- a/flatland.cc
+++ b/flatland.cc
@@ -31,6 +31,7 @@
 #include <errno.h>
 #include <string.h>
 #include <sys/stat.h>
+#include <sys/time.h>
 
 using namespace std;
 
@@ -191,6 +192,10 @@ int main(int argc, char **argv) {
     abort();
   }
 
+  struct timeval start_time, current_time;
+
+  gettimeofday(&start_time, 0);
+
   for(int n = 0; n < nb_sequences; n++) {
 
     Universe *universe;
@@ -217,10 +222,16 @@ int main(int argc, char **argv) {
     }
 
     if((n+1)%100 == 0) {
+      gettimeofday(&current_time, 0);
+      int estimated_remaining_time =
+        ((nb_sequences - n) * (current_time.tv_sec - start_time.tv_sec)) / n;
       cout << "Created "
            << n+1 << "/" << nb_sequences << " sequences in "
            << data_dir
-           << "." << endl;
+           << " (~"
+           << estimated_remaining_time/60 << "min"
+           << estimated_remaining_time%60 << "s remaining)."
+           << endl;
     }
 
     do {