• window下tqdm进度条

    原代码是linux下运行，修改后可在window下运行。

    1. #ifndef TQDM_H
    2. #define TQDM_H
    3.  
    4. #include <chrono>
    5. #include <ctime>
    6. #include <numeric>
    7. #include <ios>
    8. #include <string>
    9. #include <cstdlib>
    10. #include <iostream>
    11. #include <vector>
    12. #include <math.h>
    13. #include <algorithm>
    14. #include <io.h>
    15. class tqdm {
    16. private:
    17. 	// time, iteration counters and deques for rate calculations
    18. 	std::chrono::time_point<std::chrono::system_clock> t_first = std::chrono::system_clock::now();
    19. 	std::chrono::time_point<std::chrono::system_clock> t_old = std::chrono::system_clock::now();
    20. 	int n_old = 0;
    21. 	std::vector<double> deq_t;
    22. 	std::vector<int> deq_n;
    23. 	int nupdates = 0;
    24. 	int total_ = 0;
    25. 	int period = 1;
    26. 	unsigned int smoothing = 50;
    27. 	bool use_ema = true;
    28. 	float alpha_ema = 0.1;
    29.  
    30. 	std::vector<const char*> bars = { " ", "▏", "▎", "▍", "▌", "▋", "▊", "▉", "█" };
    31.  
    32. 	bool in_screen = (system("test $STY") == 0);
    33. 	bool in_tmux = (system("test $TMUX") == 0);
    34. 	bool is_tty = _isatty(1);
    35. 	bool use_colors = true;
    36. 	bool color_transition = true;
    37. 	int width = 40;
    38.  
    39. 	std::string right_pad = "▏";
    40. 	std::string label = "";
    41.  
    42. 	void hsv_to_rgb(float h, float s, float v, int& r, int& g, int& b) {
    43. 		if (s < 1e-6) {
    44. 			v *= 255.;
    45. 			r = v; g = v; b = v;
    46. 		}
    47. 		int i = (int)(h*6.0);
    48. 		float f = (h*6.) - i;
    49. 		int p = (int)(255.0*(v*(1. - s)));
    50. 		int q = (int)(255.0*(v*(1. - s * f)));
    51. 		int t = (int)(255.0*(v*(1. - s * (1. - f))));
    52. 		v *= 255;
    53. 		i %= 6;
    54. 		int vi = (int)v;
    55. 		if (i == 0) { r = vi; g = t;  b = p; }
    56. 		else if (i == 1) { r = q;  g = vi; b = p; }
    57. 		else if (i == 2) { r = p;  g = vi; b = t; }
    58. 		else if (i == 3) { r = p;  g = q;  b = vi; }
    59. 		else if (i == 4) { r = t;  g = p;  b = vi; }
    60. 		else if (i == 5) { r = vi; g = p;  b = q; }
    61. 	}
    62.  
    63. public:
    64. 	tqdm() {
    65. 		if (in_screen) {
    66. 			set_theme_basic();
    67. 			color_transition = false;
    68. 		}
    69. 		else if (in_tmux) {
    70. 			color_transition = false;
    71. 		}
    72. 	}
    73.  
    74. 	void reset() {
    75. 		t_first = std::chrono::system_clock::now();
    76. 		t_old = std::chrono::system_clock::now();
    77. 		n_old = 0;
    78. 		deq_t.clear();
    79. 		deq_n.clear();
    80. 		period = 1;
    81. 		nupdates = 0;
    82. 		total_ = 0;
    83. 		label = "";
    84. 	}
    85.  
    86. 	void set_theme_line() { bars = { "─", "─", "─", "╾", "╾", "╾", "╾", "━", "═" }; }
    87. 	void set_theme_circle() { bars = { " ", "◓", "◑", "◒", "◐", "◓", "◑", "◒", "#" }; }
    88. 	void set_theme_braille() { bars = { " ", "⡀", "⡄", "⡆", "⡇", "⡏", "⡟", "⡿", "⣿" }; }
    89. 	void set_theme_braille_spin() { bars = { " ", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠇", "⠿" }; }
    90. 	void set_theme_vertical() { bars = { "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█", "█" }; }
    91. 	void set_theme_basic() {
    92. 		bars = { " ", " ", " ", " ", " ", " ", " ", " ", "#" };
    93. 		right_pad = "|";
    94. 	}
    95. 	void set_label(std::string label_) { label = label_; }
    96. 	void disable_colors() {
    97. 		color_transition = false;
    98. 		use_colors = false;
    99. 	}
    100.  
    101. 	void finish() {
    102. 		progress(total_, total_);
    103. 		printf("\n");
    104. 		fflush(stdout);
    105. 	}
    106. 	void progress(int curr, int tot) {
    107. 		if (is_tty && (curr%period == 0)) {
    108. 			total_ = tot;
    109. 			nupdates++;
    110. 			auto now = std::chrono::system_clock::now();
    111. 			double dt = ((std::chrono::duration<double>)(now - t_old)).count();
    112. 			double dt_tot = ((std::chrono::duration<double>)(now - t_first)).count();
    113. 			int dn = curr - n_old;
    114. 			n_old = curr;
    115. 			t_old = now;
    116. 			if (deq_n.size() >= smoothing) deq_n.erase(deq_n.begin());
    117. 			if (deq_t.size() >= smoothing) deq_t.erase(deq_t.begin());
    118. 			deq_t.push_back(dt);
    119. 			deq_n.push_back(dn);
    120.  
    121. 			double avgrate = 0.;
    122. 			if (use_ema) {
    123. 				avgrate = deq_n[0] / deq_t[0];
    124. 				for (unsigned int i = 1; i < deq_t.size(); i++) {
    125. 					double r = 1.0*deq_n[i] / deq_t[i];
    126. 					avgrate = alpha_ema * r + (1.0 - alpha_ema)*avgrate;
    127. 				}
    128. 			}
    129. 			else {
    130. 				double dtsum = std::accumulate(deq_t.begin(), deq_t.end(), 0.);
    131. 				int dnsum = std::accumulate(deq_n.begin(), deq_n.end(), 0.);
    132. 				avgrate = dnsum / dtsum;
    133. 			}
    134.  
    135. 			// learn an appropriate period length to avoid spamming stdout
    136. 			// and slowing down the loop, shoot for ~25Hz and smooth over 3 seconds
    137. 			if (nupdates > 10) {
    138. 				period = (int)(std::min(std::max((1.0 / 25)*curr / dt_tot, 1.0), 5e5));
    139. 				smoothing = 25 * 3;
    140. 			}
    141. 			double peta = (tot - curr) / avgrate;
    142. 			double pct = (double)curr / (tot*0.01);
    143. 			if ((tot - curr) <= period) {
    144. 				pct = 100.0;
    145. 				avgrate = tot / dt_tot;
    146. 				curr = tot;
    147. 				peta = 0;
    148. 			}
    149.  
    150. 			double fills = ((double)curr / tot * width);
    151. 			int ifills = (int)fills;
    152.  
    153. 			printf("\015 ");
    154. 			if (use_colors) {
    155. 				if (color_transition) {
    156. 					// red (hue=0) to green (hue=1/3)
    157. 					int r = 255, g = 255, b = 255;
    158. 					hsv_to_rgb(0.0 + 0.01*pct / 3, 0.65, 1.0, r, g, b);
    159. 					printf("\033[38;2;%d;%d;%dm ", r, g, b);
    160. 				}
    161. 				else {
    162. 					printf("\033[32m ");
    163. 				}
    164. 			}
    165. 			for (int i = 0; i < ifills; i++) std::cout << bars[8];
    166. 			if (!in_screen && (curr != tot)) printf("%s", bars[(int)(8.0*(fills - ifills))]);
    167. 			for (int i = 0; i < width - ifills - 1; i++) std::cout << bars[0];
    168. 			printf("%s ", right_pad.c_str());
    169. 			if (use_colors) printf("\033[1m\033[31m");
    170. 			printf("%4.1f%% ", pct);
    171. 			if (use_colors) printf("\033[34m");
    172.  
    173. 			std::string unit = "Hz";
    174. 			double div = 1.;
    175. 			if (avgrate > 1e6) {
    176. 				unit = "MHz"; div = 1.0e6;
    177. 			}
    178. 			else if (avgrate > 1e3) {
    179. 				unit = "kHz"; div = 1.0e3;
    180. 			}
    181. 			printf("[%4d/%4d | %3.1f %s | %.0fs<%.0fs] ", curr, tot, avgrate / div, unit.c_str(), dt_tot, peta);
    182. 			printf("%s ", label.c_str());
    183. 			if (use_colors) printf("\033[0m\033[32m\033[0m\015 ");
    184.  
    185. 			if ((tot - curr) > period) fflush(stdout);
    186.  
    187. 		}
    188. 	}
    189. };
    190. #endif

    主函数： 

    1. #include "MyClass.h"
    2. #include <windows.h>
    3. int main() {
    4.  
    5. 	int N = 2000;
    6. 	tqdm bar;
    7.  
    8. 	std::cout << "Overhead of loop only:" << std::endl;
    9. 	for (int i = 0; i < 100000000; i++) {
    10. 		bar.progress(i, 100000000);
    11. 	}
    12. 	bar.finish();
    13.  
    14.  
    15. 	std::cout << "Basic:" << std::endl;
    16. 	bar.reset();
    17. 	bar.set_theme_basic();
    18. 	for (int i = 0; i < N; i++) {
    19. 		bar.progress(i, N);
    20. 		Sleep(1000);
    21. 	}
    22. 	bar.finish();
    23.  
    24. 	std::cout << "Braille:" << std::endl;
    25. 	bar.reset();
    26. 	bar.set_theme_braille();
    27. 	for (int i = 0; i < N; i++) {
    28. 		bar.progress(i, N);
    29. 		Sleep(300);
    30. 	}
    31. 	bar.finish();
    32.  
    33. 	std::cout << "Line:" << std::endl;
    34. 	bar.reset();
    35. 	bar.set_theme_line();
    36. 	for (int i = 0; i < N; i++) {
    37. 		bar.progress(i, N);
    38. 		Sleep(300);
    39. 	}
    40. 	bar.finish();
    41.  
    42. 	std::cout << "Circles:" << std::endl;
    43. 	bar.reset();
    44. 	bar.set_theme_circle();
    45. 	for (int i = 0; i < N; i++) {
    46. 		bar.progress(i, N);
    47. 		Sleep(300);
    48. 	}
    49. 	bar.finish();
    50.  
    51. 	bar.reset();
    52. 	std::cout << "Vertical bars:" << std::endl;
    53. 	bar.reset();
    54. 	bar.set_theme_vertical();
    55. 	for (int i = 0; i < N; i++) {
    56. 		bar.progress(i, N);
    57. 		Sleep(3000);
    58. 	}
    59. 	bar.finish();
    60.  
    61. 	return 0;
    62. }

    源码 

  • 相关阅读:
    Himall验证帮助类判断当前时间是否在指定的时间段内、是否是数值(包括整数和小数)
    c实现mp4解封装
    lua 判空的坑
    ELK 与 EFK的介绍和对比
    美国IP代理如何获取？适用于哪些场景？
    MySQL备份与恢复
    [激光原理与应用-20]：《激光原理与技术》-6- 谐振腔的结构、作用、工作原理
    Java基于SSM框架的教室预约申请管理系统 毕业设计
    HTML5:七天学会基础动画网页6
    自己动手从零写桌面操作系统GrapeOS系列教程——11.MBR介绍
  • 原文地址：https://blog.csdn.net/qq_35054151/article/details/140377766