2019-12-03 09:54:50 -06:00
|
|
|
#pragma once
|
|
|
|
|
#include <iostream>
|
|
|
|
|
#include <string>
|
|
|
|
|
#include <atomic>
|
|
|
|
|
#include <mutex>
|
|
|
|
|
#include <thread>
|
|
|
|
|
|
2019-12-03 10:46:25 -06:00
|
|
|
template <typename T>
|
2019-12-03 09:54:50 -06:00
|
|
|
class progress_bar {
|
2019-12-03 10:02:46 -06:00
|
|
|
std::string _name{"Running"};
|
2019-12-03 10:46:25 -06:00
|
|
|
size_t _bar_width{80};
|
2019-12-03 09:54:50 -06:00
|
|
|
std::string _start{"|"};
|
|
|
|
|
std::string _end{"|"};
|
|
|
|
|
std::mutex _mutex;
|
2019-12-03 10:46:25 -06:00
|
|
|
T _progress{T()};
|
2019-12-03 09:54:50 -06:00
|
|
|
|
|
|
|
|
public:
|
|
|
|
|
|
|
|
|
|
~progress_bar() {
|
|
|
|
|
std::cout << std::endl;
|
|
|
|
|
}
|
|
|
|
|
|
2019-12-03 10:46:25 -06:00
|
|
|
void increment(T value) {
|
2019-12-03 09:54:50 -06:00
|
|
|
std::unique_lock<std::mutex> lock{_mutex};
|
2019-12-03 10:46:25 -06:00
|
|
|
_progress = value / 100.0;
|
2019-12-03 10:02:46 -06:00
|
|
|
std::cout << _name << " [";
|
2019-12-03 10:46:25 -06:00
|
|
|
T pos = _progress * static_cast<T>(_bar_width);
|
|
|
|
|
for (size_t i = 0; i < _bar_width; ++i) {
|
|
|
|
|
if (i < pos) std::cout << '#';
|
|
|
|
|
else if (i == pos) std::cout << ">";
|
2019-12-03 09:54:50 -06:00
|
|
|
else std::cout << " ";
|
|
|
|
|
}
|
2019-12-03 10:46:25 -06:00
|
|
|
std::cout << "] " << static_cast<int>(value) << " %\r" << std::flush;
|
2019-12-03 09:54:50 -06:00
|
|
|
}
|
|
|
|
|
};
|