Minor updates

This commit is contained in:
Pranav Srinivas Kumar
2019-12-03 10:46:25 -06:00
parent ec303ef957
commit 1e6df03be6

View File

@@ -5,13 +5,14 @@
#include <mutex> #include <mutex>
#include <thread> #include <thread>
template <typename T>
class progress_bar { class progress_bar {
std::string _name{"Running"}; std::string _name{"Running"};
size_t _bar_width{100}; size_t _bar_width{80};
std::string _start{"|"}; std::string _start{"|"};
std::string _end{"|"}; std::string _end{"|"};
std::mutex _mutex; std::mutex _mutex;
std::atomic<size_t> _progress{0}; T _progress{T()};
public: public:
@@ -19,20 +20,16 @@ public:
std::cout << std::endl; std::cout << std::endl;
} }
void increment(size_t value) { void increment(T value) {
std::unique_lock<std::mutex> lock{_mutex}; std::unique_lock<std::mutex> lock{_mutex};
_progress = value; _progress = value / 100.0;
std::cout << _name << " ["; std::cout << _name << " [";
for (size_t i = 0; i < _bar_width; i++) { T pos = _progress * static_cast<T>(_bar_width);
if (i < value) std::cout << "#"; for (size_t i = 0; i < _bar_width; ++i) {
else if (i == value) std::cout << ">"; if (i < pos) std::cout << '#';
else if (i == pos) std::cout << ">";
else std::cout << " "; else std::cout << " ";
} }
std::cout << "] " << int(_progress) << " %\r"; std::cout << "] " << static_cast<int>(value) << " %\r" << std::flush;
std::cout.flush();
}
size_t current() {
return _progress;
} }
}; };