2021-06-26 15:24:07 +02:00
|
|
|
// SPDX-License-Identifier: Apache-2.0
|
2019-10-05 18:58:52 +02:00
|
|
|
/*
|
|
|
|
|
* Copyright (C) 2019 Gwenhael Goavec-Merou <gwenhael.goavec-merou@trabucayre.com>
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
#include <stdlib.h>
|
2021-02-24 13:31:55 +01:00
|
|
|
#include <chrono>
|
|
|
|
|
#include <string>
|
2019-10-05 18:58:52 +02:00
|
|
|
#include "progressBar.hpp"
|
2019-11-20 07:51:37 +01:00
|
|
|
#include "display.hpp"
|
2019-10-05 18:58:52 +02:00
|
|
|
|
2021-01-30 07:57:49 +01:00
|
|
|
ProgressBar::ProgressBar(std::string mess, int maxValue, int progressLen,
|
|
|
|
|
bool quiet): _mess(mess), _maxValue(maxValue),
|
|
|
|
|
_progressLen(progressLen), _quiet(quiet), _first(true)
|
2019-10-05 18:58:52 +02:00
|
|
|
{
|
2021-02-24 13:31:55 +01:00
|
|
|
last_time = std::chrono::system_clock::now();
|
2019-10-05 18:58:52 +02:00
|
|
|
}
|
2020-12-24 23:35:05 +01:00
|
|
|
void ProgressBar::display(int value, char force)
|
2019-10-05 18:58:52 +02:00
|
|
|
{
|
2021-01-30 07:57:49 +01:00
|
|
|
if (_quiet) {
|
|
|
|
|
if (_first) {
|
|
|
|
|
printInfo(_mess + ": ", false);
|
|
|
|
|
_first = false;
|
|
|
|
|
}
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2021-02-24 13:31:55 +01:00
|
|
|
std::chrono::time_point<std::chrono::system_clock> this_time;
|
|
|
|
|
this_time = std::chrono::system_clock::now();
|
|
|
|
|
std::chrono::duration<double> diff = this_time - last_time;
|
|
|
|
|
|
|
|
|
|
if (!force && diff.count() < 1)
|
2020-12-24 23:35:05 +01:00
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
last_time = this_time;
|
2019-10-05 18:58:52 +02:00
|
|
|
float percent = ((float)value * 100.0f)/(float)_maxValue;
|
|
|
|
|
float nbEq = (percent * (float) _progressLen)/100.0f;
|
|
|
|
|
|
2019-11-20 07:51:37 +01:00
|
|
|
printInfo("\r" + _mess + ": [", false);
|
2019-10-05 18:58:52 +02:00
|
|
|
for (int z=0; z < nbEq; z++) {
|
|
|
|
|
fputc('=', stderr);
|
|
|
|
|
}
|
|
|
|
|
fprintf(stderr, "%*s", (int)(_progressLen-nbEq), "");
|
2021-07-11 11:30:02 +02:00
|
|
|
char perc_str[11];
|
|
|
|
|
snprintf(perc_str, sizeof(perc_str), "] %3.2f%%", percent);
|
|
|
|
|
printInfo(perc_str, false);
|
2019-10-05 18:58:52 +02:00
|
|
|
}
|
|
|
|
|
void ProgressBar::done()
|
|
|
|
|
{
|
2021-01-30 07:57:49 +01:00
|
|
|
if (_quiet) {
|
|
|
|
|
printSuccess("Done");
|
|
|
|
|
} else {
|
|
|
|
|
display(_maxValue, true);
|
|
|
|
|
printSuccess("\nDone");
|
|
|
|
|
}
|
2019-11-20 07:51:37 +01:00
|
|
|
}
|
|
|
|
|
void ProgressBar::fail()
|
|
|
|
|
{
|
2021-01-30 07:57:49 +01:00
|
|
|
if (_quiet) {
|
2021-02-24 13:31:55 +01:00
|
|
|
printError("Fail");
|
2021-01-30 07:57:49 +01:00
|
|
|
} else {
|
|
|
|
|
display(_maxValue, true);
|
|
|
|
|
printError("\nFail");
|
|
|
|
|
}
|
2019-10-05 18:58:52 +02:00
|
|
|
}
|