ViewVC Help

View File | Revision Log | Show Annotations | View Changeset | Root Listing

root/OpenMD/trunk/src/utils/ProgressBar.cpp

Comparing trunk/src/utils/ProgressBar.cpp (file contents):
Revision 1413 by gezelter, Mon Mar 22 19:21:22 2010 UTC vs.
Revision 1939 by gezelter, Thu Oct 31 18:18:57 2013 UTC

#	Line 35 | Line 35
35		*
36		* [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).
37		* [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).
38	<	* [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).
39	<	* [4]  Vardeman, Stocker & Gezelter, in progress (2010).
38	>	* [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 234107 (2008).
39	>	* [4]  Kuang & Gezelter,  J. Chem. Phys. 133, 164101 (2010).
40	>	* [5]  Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41		*/
42
42	–	#include <string>
43	–	#include <sstream>
44	–	#include <sys/ioctl.h>
43		#ifdef IS_MPI
44		#include <mpi.h>
45	<	#endif //is_mpi
45	>	#endif
46	>
47	>	#include <iostream>
48	>	#include <cstdlib>
49	>
50	>	#ifdef _MSC_VER
51	>	#include <Windows.h>
52	>	#include <stdio.h>
53	>	#include <io.h>
54	>	#define isatty _isatty
55	>	#define fileno _fileno
56	>	#else
57	>	#include <cstdio>
58	>	#include <sys/ioctl.h>
59	>	#include <unistd.h>
60	>	#endif
61	>
62		#include "utils/ProgressBar.hpp"
63
64	+	using namespace std;
65	+
66		namespace OpenMD {
67
68		const char * progressSpinner_ = "|/-\\";
#	Line 58 | Line 74 | namespace OpenMD {
74		#ifdef IS_MPI
75		if (MPI::COMM_WORLD.Get_rank() == 0) {
76		#endif
77	<	printf("\n");
78	<	fflush(stdout);
77	>	cout << endl;
78	>	cout.flush();
79		#ifdef IS_MPI
80		}
81		#endif
#	Line 70 | Line 86 | namespace OpenMD {
86		}
87
88		void ProgressBar::update() {
89	<	int width = 80;
89	>
90		#ifdef IS_MPI
91		if (MPI::COMM_WORLD.Get_rank() == 0) {
92		#endif
93	<	#ifndef IS_MPI
94	<	// get the window width:
95	<	struct ttysize ts;
96	<	ioctl(0, TIOCGWINSZ, &ts);
81	<	width = ts.ts_cols;
82	<	#endif
93	>
94	>	// only do the progress bar if we are actually running in a tty:
95	>	if (isatty(fileno(stdout))  && (getenv("SGE_TASK_ID")==NULL)) {
96	>	// get the window width:
97
98	<	// We'll use:
99	<	// 31 characters for the completion estimate,
100	<	//  6 for the % complete,
101	<	//  2 characters for the open and closing brackets.
98	>	int width = 0;
99	>	#ifdef _MSC_VER
100	>	CONSOLE_SCREEN_BUFFER_INFO csbi;
101	>	HANDLE hConsole = GetStdHandle( STD_OUTPUT_HANDLE );
102	>	int ret = GetConsoleScreenBufferInfo(hConsole, &csbi);
103	>	if(ret) {
104	>	width = csbi.dwSize.X - 1;
105	>	}
106	>	#else
107	>	struct winsize w;
108	>	ioctl(fileno(stdout), TIOCGWINSZ, &w);
109	>	width = w.ws_col;
110	>	#endif
111
112	<	int avail = width - 31 - 6 - 2;
112	>	// handle the case when the width is returned as a nonsensical value.
113	>	if (width <= 0) width = 80;
114
115	<	++iteration_;
116	<
117	<	if (maximum_ > 0.0) {
118	<	// we know the maximum, so
119	<	// draw a progress bar
120	<
97	<	RealType percent = value_ * 100.0 / maximum_;
98	<	int hashes = int(percent * avail / 100.0);
99	<	std::string progressbar;
100	<	progressbar.assign(hashes, '#');
115	>	// We'll use:
116	>	// 31 characters for the completion estimate,
117	>	//  6 for the % complete,
118	>	//  2 characters for the open and closing brackets.
119	>
120	>	int avail = width - 31 - 6 - 2;
121
122	<	// add the spinner to the end of the progress bar:
103	<	progressbar += progressSpinner_[iteration_ & 3];
122	>	++iteration_;
123
124	<	// compute the best estimate of the ending time:
106	<	time_t current_ = time(NULL);
107	<	time_t end_ = start_ + (current_ - start_) * (100.0/percent);
108	<	struct tm * ender = localtime(&end_);
109	<	char buffer[24];
110	<	strftime(buffer, 24, "%a %b %d @ %I:%M %p", ender);
124	>	if (maximum_ > 0.0) {
125
126	<	std::stringstream fmt;
127	<	fmt << "\r%3d%% [%-" << avail << "s] Estimate: %s";
128	<	std::string st = fmt.str();
126	>	// we know the maximum, so draw a progress bar
127	>
128	>	RealType percent = value_ * 100.0 / maximum_;
129	>	int hashes = int(percent * avail / 100.0);
130	>
131	>	// compute the best estimate of the ending time:
132	>	time_t current_ = time(NULL);
133	>	time_t end_ = static_cast<time_t>(start_ + (current_ - start_) *
134	>	(100.0/percent) );
135	>	struct tm * ender = localtime(&end_);
136	>	char buffer[22];
137	>	strftime(buffer, 22, "%a %b %d @ %I:%M %p", ender);
138
139	<	printf(st.c_str(), int(percent),
140	<	progressbar.c_str(),
141	<	buffer);
142	<
143	<	} else {
144	<	// we don't know the maximum, so we can't draw a progress bar
145	<	int center = (iteration_ % 48) + 1; // 50 spaces, minus 2
146	<	std::string before;
147	<	std::string after;
148	<	before.assign(std::max(center - 2, 0), ' ');
149	<	after.assign(std::min(center + 2, 50), ' ');
150	<
151	<	printf("\r[%s###%s]            ",
152	<	before.c_str(), after.c_str());
153	<	}
154	<	fflush(stdout);
139	>	#ifdef _MSC_VER
140	>	csbi.dwCursorPosition.X = 0;
141	>	SetConsoleCursorPosition(hConsole, csbi.dwCursorPosition);
142	>	#else
143	>	cout << '\r';
144	>	#endif
145	>	cout.width(3);
146	>	cout << right << int(percent);
147	>	cout.width(3);
148	>	cout << "% [";
149	>	cout.fill('#');
150	>	if (hashes+1 < avail) {
151	>	cout.width(hashes+1);
152	>	cout << progressSpinner_[iteration_ & 3];
153	>	} else {
154	>	cout.width(avail);
155	>	cout << '#';
156	>	}
157	>	cout.fill(' ');
158	>	if (avail - hashes - 1  > 0) {
159	>	cout.width(avail - hashes - 1);
160	>	cout << ' ';
161	>	}
162	>	cout.width(11);
163	>	cout << "] Estimate:";
164	>	cout.width(22);
165	>	cout << buffer;
166	>
167	>	}
168	>	cout.flush();
169	>	}
170		#ifdef IS_MPI
171		}
172		#endif

Comparing trunk/src/utils/ProgressBar.cpp (property svn:keywords):
Revision 1413 by gezelter, Mon Mar 22 19:21:22 2010 UTC vs.
Revision 1939 by gezelter, Thu Oct 31 18:18:57 2013 UTC

#	Line 0 | Line 1
1	+	Author Id Revision Date

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines