crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding check for redirect and non-terminal calls.

Browse source
This commit is contained in:
Patrick Shriwise 2018-10-29 23:00:25 -05:00
parent ddbac3683b
commit 416ab13b14
1 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/progress_bar.cpp
View file

@ -5,8 +5,26 @@
#include <iostream>
#include <iomanip>
#ifdef UNIX
#include <unistd.h>
#endif
#define BAR_WIDTH 72
#ifdef UNIX
int check_isatty(int fd) {
return isatty(fd);
}
#endif
bool is_terminal() {
#ifdef UNIX
return check_isatty(STDOUT_FILENO) != 0;
#else
return false;
#endif
}
ProgressBar::ProgressBar() {
bar = "";
set_value(0.0);
@ -14,6 +32,9 @@ ProgressBar::ProgressBar() {
void
ProgressBar::set_value(double val) {
if (!is_terminal()) return;
// set the bar percentage
if (val >= 100.0) {
bar.append("100");
@ -28,19 +49,19 @@ ProgressBar::set_value(double val) {
bar.append("% |");
// remaining width of the bar
int remain = BAR_WIDTH - bar.size() - 1;
int remain = BAR_WIDTH - bar.size() - 2;
// set the bar width
if (val >= 100.0) {
bar.append(remain, '=');
} else {
int width = (int)(65*val/100);
int width = (int)((double)remain*val/100);
bar.append(width, '=');
bar.append(1, '>');
bar.append(remain-width-1, ' ');
}
bar.append("|");
bar.append("|+");
// write the bar
std::cout << '\r' << bar << std::flush;