wehub-resource-sync
50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
/*!
|
|
* Copyright (c) 2023-2025 by Contributors
|
|
* \file support/progress_bar.h
|
|
* \brief A simple progress bar in C++.
|
|
*/
|
|
#ifndef MLC_LLM_SUPPORT_PROGRESS_BAR_H_
|
|
#define MLC_LLM_SUPPORT_PROGRESS_BAR_H_
|
|
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
namespace mlc {
|
|
namespace llm {
|
|
|
|
class ProgressBar {
|
|
public:
|
|
explicit ProgressBar(int total, int width = 100) : total(total), width(width), cur(0) {}
|
|
|
|
void Progress() {
|
|
if (cur < total) {
|
|
++cur;
|
|
}
|
|
int bar_width = width - 2; // Adjust for borders
|
|
int completed = static_cast<int>(static_cast<float>(cur) / total * bar_width);
|
|
int remaining = bar_width - completed;
|
|
std::cout << "[" //
|
|
<< std::string(completed, '=') //
|
|
<< ">" //
|
|
<< std::string(remaining, ' ') //
|
|
<< "] " //
|
|
<< " [" << cur << "/" << total << "]";
|
|
if (cur < total) {
|
|
std::cout << "\r";
|
|
std::cout.flush();
|
|
} else {
|
|
std::cout << std::endl; // Move to the next line after the progress bar is complete
|
|
}
|
|
}
|
|
|
|
private:
|
|
int total;
|
|
int width;
|
|
int cur;
|
|
};
|
|
|
|
} // namespace llm
|
|
} // namespace mlc
|
|
|
|
#endif // MLC_LLM_SUPPORT_PROGRESS_BAR_H_
|