added time estimates to the progress bar

David Mayerich
1 parent df98143c
Showing 1 changed file with 31 additions and 1 deletions Show diff stats
stim/ui/progressbar.h
@@ -5,7 +5,28 @@
 #include <sstream>
 using namespace std;
-static void rtsProgressBar(unsigned int percent)
+void stimPrintTime(unsigned long long s) {
+	char buffer[26];
+	std::cout << std::fixed << std::showpoint << std::setprecision(1);
+	if (s >= 60) {											//if there are more than 60 seconds
+		unsigned long long m = s / 60;						//get the number of minutes
+		s = s - m * 60;										//get the number of remaining seconds
+		if (m > 60) {										//if more than 60 minutes
+			unsigned long long h = m / 60;
+			m = m - h * 60;									//get the number of remaining minutes
+			if (h > 24) {									//if there are more than 24 hours
+				unsigned long long d = h / 24;
+				h = h - d * 24;								//get the number of remaining hours
+				std::cout << " " << d << "d";
+			}
+			std::cout << " " << h << "h";
+		}
+		std::cout << " " << m << "m";
+	}
+	std::cout << " " << s << "s";
+}
+
+static void rtsProgressBar(unsigned int percent, double elapsed_s = 0)
 {
 	//std::cout<<percent<<std::endl;
 	stringstream bar;
@@ -26,6 +47,15 @@ static void rtsProgressBar(unsigned int percent)
 	bar<<"]";
 	cout << "\r"; // carriage return back to beginning of line
 	cout << bar.str() << " " << slash[x] << " " << percent << " %"; // print the bars and percentage
+	if (elapsed_s > 0) {
+		stimPrintTime(elapsed_s);
+		if (percent > 0) {
+			std::cout << " / ";
+			double s_per_percent = elapsed_s / percent;
+			double total = s_per_percent * 100;
+			stimPrintTime(total);
+		}
+	}
 	cout.flush();
 	x++; // increment to make the slash appear to rotate
 	if(x == 4)