114 lines
2.9 KiB
C++
114 lines
2.9 KiB
C++
#include "ProgressBar.h"
|
|
#include <algorithm>
|
|
#include <future>
|
|
#include <chrono>
|
|
#include <format.h>
|
|
#include <iostream>
|
|
#include <boost/algorithm/clamp.hpp>
|
|
|
|
using std::string;
|
|
|
|
ProgressForwarder::ProgressForwarder(std::function<void(double progress)> callback) :
|
|
callback(callback)
|
|
{}
|
|
|
|
void ProgressForwarder::reportProgress(double value) {
|
|
callback(value);
|
|
}
|
|
|
|
ProgressMerger::ProgressMerger(ProgressSink& sink) :
|
|
sink(sink)
|
|
{}
|
|
|
|
ProgressSink& ProgressMerger::addSink(double weight) {
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
|
|
totalWeight += weight;
|
|
int sinkIndex = weightedValues.size();
|
|
weightedValues.push_back(0);
|
|
forwarders.push_back(ProgressForwarder([weight, sinkIndex, this](double progress) {
|
|
weightedValues[sinkIndex] = progress * weight;
|
|
report();
|
|
}));
|
|
return forwarders.back();
|
|
}
|
|
|
|
void ProgressMerger::report() {
|
|
std::lock_guard<std::mutex> lock(mutex);
|
|
|
|
if (totalWeight != 0) {
|
|
double weightedSum = 0;
|
|
for (double weightedValue : weightedValues) {
|
|
weightedSum += weightedValue;
|
|
}
|
|
double progress = weightedSum / totalWeight;
|
|
sink.reportProgress(progress);
|
|
} else {
|
|
sink.reportProgress(0);
|
|
}
|
|
}
|
|
|
|
ProgressBar::ProgressBar() {
|
|
updateLoopFuture = std::async(std::launch::async, &ProgressBar::updateLoop, this);
|
|
}
|
|
|
|
ProgressBar::~ProgressBar() {
|
|
done = true;
|
|
updateLoopFuture.wait();
|
|
}
|
|
|
|
void ProgressBar::reportProgress(double value) {
|
|
// Make sure value is in [0..1] range
|
|
value = boost::algorithm::clamp(value, 0.0, 1.0);
|
|
|
|
currentProgress = value;
|
|
}
|
|
|
|
void ProgressBar::updateLoop() {
|
|
const int blockCount = 20;
|
|
const std::chrono::milliseconds animationInterval(1000 / 8);
|
|
const string animation = "|/-\\";
|
|
|
|
while (!done) {
|
|
int progressBlockCount = static_cast<int>(currentProgress * blockCount);
|
|
int percent = static_cast<int>(currentProgress * 100);
|
|
string text = fmt::format("[{0}{1}] {2:3}% {3}",
|
|
string(progressBlockCount, '#'), string(blockCount - progressBlockCount, '-'),
|
|
percent,
|
|
animation[animationIndex++ % animation.size()]);
|
|
updateText(text);
|
|
|
|
std::this_thread::sleep_for(animationInterval);
|
|
}
|
|
|
|
updateText("");
|
|
}
|
|
|
|
void ProgressBar::updateText(const string& text) {
|
|
// Get length of common portion
|
|
int commonPrefixLength = 0;
|
|
int commonLength = std::min(currentText.size(), text.size());
|
|
while (commonPrefixLength < commonLength && text[commonPrefixLength] == currentText[commonPrefixLength]) {
|
|
commonPrefixLength++;
|
|
}
|
|
|
|
// Construct output string
|
|
string output;
|
|
|
|
// ... backtrack to the first differing character
|
|
output.append(currentText.size() - commonPrefixLength, '\b');
|
|
|
|
// ... add new suffix
|
|
output.append(text, commonPrefixLength, text.size() - commonPrefixLength);
|
|
|
|
// ... if the new text is shorter than the old one: delete overlapping characters
|
|
int overlapCount = currentText.size() - text.size();
|
|
if (overlapCount > 0) {
|
|
output.append(overlapCount, ' ');
|
|
output.append(overlapCount, '\b');
|
|
}
|
|
|
|
std::cerr << output;
|
|
currentText = text;
|
|
}
|