原代码是linux下运行，修改后可在window下运行。
#ifndef TQDM_H
#define TQDM_H
#include <chrono>
#include <ctime>
#include <numeric>
#include <ios>
#include <string>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <math.h>
#include <algorithm>
#include <io.h>
class tqdm {
private:
// time, iteration counters and deques for rate calculations
std::chrono::time_point<std::chrono::system_clock> t_first = std::chrono::system_clock::now();
std::chrono::time_point<std::chrono::system_clock> t_old = std::chrono::system_clock::now();
int n_old = 0;
std::vector<double> deq_t;
std::vector<int> deq_n;
int nupdates = 0;
int total_ = 0;
int period = 1;
unsigned int smoothing = 50;
bool use_ema = true;
float alpha_ema = 0.1;
std::vector<const char*> bars = { " ", "▏", "▎", "▍", "▌", "▋", "▊", "▉", "█" };
bool in_screen = (system("test $STY") == 0);
bool in_tmux = (system("test $TMUX") == 0);
bool is_tty = _isatty(1);
bool use_colors = true;
bool color_transition = true;
int width = 40;
std::string right_pad = "▏";
std::string label = "";
void hsv_to_rgb(float h, float s, float v, int& r, int& g, int& b) {
if (s < 1e-6) {
v *= 255.;
r = v; g = v; b = v;
}
int i = (int)(h*6.0);
float f = (h*6.) - i;
int p = (int)(255.0*(v*(1. - s)));
int q = (int)(255.0*(v*(1. - s * f)));
int t = (int)(255.0*(v*(1. - s * (1. - f))));
v *= 255;
i %= 6;
int vi = (int)v;
if (i == 0) { r = vi; g = t; b = p; }
else if (i == 1) { r = q; g = vi; b = p; }
else if (i == 2) { r = p; g = vi; b = t; }
else if (i == 3) { r = p; g = q; b = vi; }
else if (i == 4) { r = t; g = p; b = vi; }
else if (i == 5) { r = vi; g = p; b = q; }
}
public:
tqdm() {
if (in_screen) {
set_theme_basic();
color_transition = false;
}
else if (in_tmux) {
color_transition = false;
}
}
void reset() {
t_first = std::chrono::system_clock::now();
t_old = std::chrono::system_clock::now();
n_old = 0;
deq_t.clear();
deq_n.clear();
period = 1;
nupdates = 0;
total_ = 0;
label = "";
}
void set_theme_line() { bars = { "─", "─", "─", "╾", "╾", "╾", "╾", "━", "═" }; }
void set_theme_circle() { bars = { " ", "◓", "◑", "◒", "◐", "◓", "◑", "◒", "#" }; }
void set_theme_braille() { bars = { " ", "⡀", "⡄", "⡆", "⡇", "⡏", "⡟", "⡿", "⣿" }; }
void set_theme_braille_spin() { bars = { " ", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠇", "⠿" }; }
void set_theme_vertical() { bars = { "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█", "█" }; }
void set_theme_basic() {
bars = { " ", " ", " ", " ", " ", " ", " ", " ", "#" };
right_pad = "|";
}
void set_label(std::string label_) { label = label_; }
void disable_colors() {
color_transition = false;
use_colors = false;
}
void finish() {
progress(total_, total_);
printf("\n");
fflush(stdout);
}
void progress(int curr, int tot) {
if (is_tty && (curr%period == 0)) {
total_ = tot;
nupdates++;
auto now = std::chrono::system_clock::now();
double dt = ((std::chrono::duration<double>)(now - t_old)).count();
double dt_tot = ((std::chrono::duration<double>)(now - t_first)).count();
int dn = curr - n_old;
n_old = curr;
t_old = now;
if (deq_n.size() >= smoothing) deq_n.erase(deq_n.begin());
if (deq_t.size() >= smoothing) deq_t.erase(deq_t.begin());
deq_t.push_back(dt);
deq_n.push_back(dn);
double avgrate = 0.;
if (use_ema) {
avgrate = deq_n[0] / deq_t[0];
for (unsigned int i = 1; i < deq_t.size(); i++) {
double r = 1.0*deq_n[i] / deq_t[i];
avgrate = alpha_ema * r + (1.0 - alpha_ema)*avgrate;
}
}
else {
double dtsum = std::accumulate(deq_t.begin(), deq_t.end(), 0.);
int dnsum = std::accumulate(deq_n.begin(), deq_n.end(), 0.);
avgrate = dnsum / dtsum;
}
// learn an appropriate period length to avoid spamming stdout
// and slowing down the loop, shoot for ~25Hz and smooth over 3 seconds
if (nupdates > 10) {
period = (int)(std::min(std::max((1.0 / 25)*curr / dt_tot, 1.0), 5e5));
smoothing = 25 * 3;
}
double peta = (tot - curr) / avgrate;
double pct = (double)curr / (tot*0.01);
if ((tot - curr) <= period) {
pct = 100.0;
avgrate = tot / dt_tot;
curr = tot;
peta = 0;
}
double fills = ((double)curr / tot * width);
int ifills = (int)fills;
printf("\015 ");
if (use_colors) {
if (color_transition) {
// red (hue=0) to green (hue=1/3)
int r = 255, g = 255, b = 255;
hsv_to_rgb(0.0 + 0.01*pct / 3, 0.65, 1.0, r, g, b);
printf("\033[38;2;%d;%d;%dm ", r, g, b);
}
else {
printf("\033[32m ");
}
}
for (int i = 0; i < ifills; i++) std::cout << bars[8];
if (!in_screen && (curr != tot)) printf("%s", bars[(int)(8.0*(fills - ifills))]);
for (int i = 0; i < width - ifills - 1; i++) std::cout << bars[0];
printf("%s ", right_pad.c_str());
if (use_colors) printf("\033[1m\033[31m");
printf("%4.1f%% ", pct);
if (use_colors) printf("\033[34m");
std::string unit = "Hz";
double div = 1.;
if (avgrate > 1e6) {
unit = "MHz"; div = 1.0e6;
}
else if (avgrate > 1e3) {
unit = "kHz"; div = 1.0e3;
}
printf("[%4d/%4d | %3.1f %s | %.0fs<%.0fs] ", curr, tot, avgrate / div, unit.c_str(), dt_tot, peta);
printf("%s ", label.c_str());
if (use_colors) printf("\033[0m\033[32m\033[0m\015 ");
if ((tot - curr) > period) fflush(stdout);
}
}
};
#endif主函数：
#include "MyClass.h"
#include <windows.h>
int main() {
int N = 2000;
tqdm bar;
std::cout << "Overhead of loop only:" << std::endl;
for (int i = 0; i < 100000000; i++) {
bar.progress(i, 100000000);
}
bar.finish();
std::cout << "Basic:" << std::endl;
bar.reset();
bar.set_theme_basic();
for (int i = 0; i < N; i++) {
bar.progress(i, N);
Sleep(1000);
}
bar.finish();
std::cout << "Braille:" << std::endl;
bar.reset();
bar.set_theme_braille();
for (int i = 0; i < N; i++) {
bar.progress(i, N);
Sleep(300);
}
bar.finish();
std::cout << "Line:" << std::endl;
bar.reset();
bar.set_theme_line();
for (int i = 0; i < N; i++) {
bar.progress(i, N);
Sleep(300);
}
bar.finish();
std::cout << "Circles:" << std::endl;
bar.reset();
bar.set_theme_circle();
for (int i = 0; i < N; i++) {
bar.progress(i, N);
Sleep(300);
}
bar.finish();
bar.reset();
std::cout << "Vertical bars:" << std::endl;
bar.reset();
bar.set_theme_vertical();
for (int i = 0; i < N; i++) {
bar.progress(i, N);
Sleep(3000);
}
bar.finish();
return 0;
}
源码
