C++控制台的进度条类(支持多线),分享出来以供参考

// ---------------------------------------------------------------------------ProgressBarMthread.h
#ifndef PROGRESSBARMTHREAD
#define PROGRESSBARMTHREAD

#include <unordered_map>
#include <mutex>

class ProgressBarMthread
{
private:
	std::unordered_map<std::thread::id, int> mapX;
	int startX;
	std::mutex mut;

private:
	void moveCursor(int x, int y);
	void getCursorPosition(int& x, int& y);

public:
	ProgressBarMthread();

	void displayProgressBar(std::string title, int progress, int total, int barWidth = 100);

	~ProgressBarMthread();
};

#endif // !PROGRESSBARMTHREAD
// ---------------------------------------------------------------------------ProgressBarMthread.cpp
#include "ProgressBarMthread.h"

#include <iostream>

#ifdef _WIN32
	#include <windows.h>
#else
	#include <unistd.h>
	#include <term.h>
#endif

// 移动光标到指定位置
void ProgressBarMthread::moveCursor(int x, int y) {
    std::cout << "\033[" << x + 1 << ";" << y + 1 << "H";
}

// 获取光标位置的函数,Windows 版本
void ProgressBarMthread::getCursorPosition(int& x, int& y) {
#ifdef _WIN32
    CONSOLE_SCREEN_BUFFER_INFO csbi;
    GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &csbi);
    y = csbi.dwCursorPosition.X;
    x = csbi.dwCursorPosition.Y;
#else
	if (!cur_term) {
        int result;
        setupterm(nullptr, STDOUT_FILENO, &result);
        if (result <= 0) {
            return;
        }
    }
    y = tigetnum("cols");
    x = tigetnum("lines");
#end
}

ProgressBarMthread::ProgressBarMthread()
    : startX(0)
{
    int startY;
    getCursorPosition(startX, startY);
    if (startY != 0)
    {
        startX++;
    }
}

void ProgressBarMthread::displayProgressBar(std::string title, int progress, int total, int barWidth)
{
    std::lock_guard<std::mutex> lk(mut);

    auto tId = std::this_thread::get_id();

    if (mapX.find(tId) == mapX.end())
    {
        int x = startX + (int)mapX.size();
        mapX[tId] = x;
    }

    float percentage = static_cast<float>(progress) / total;
    int progressBarWidth = static_cast<int>(percentage * barWidth);

    moveCursor(mapX[tId], 0);
    std::cout << title << " ";
    std::cout << "[";
    for (int i = 0; i < barWidth; ++i) {
        if (i < progressBarWidth) {
            std::cout << "=";
        }
        else {
            std::cout << " ";
        }
    }
    std::cout << "] " << percentage * 100.0 << "%       ";
    std::cout.flush();
}

ProgressBarMthread::~ProgressBarMthread()
{
    int x = startX + (int)mapX.size();
    moveCursor(x, 0);
}
// ---------------------------------------------------------------------------main.cpp
// 使用示例
#include "ProgressBarMthread.h"

void threadFunc(ProgressBarMthread& proBar)
{
    for (int i = 0; i < 20; ++i)
    {
		// 随机睡眠一段时间,模拟大量数据处理
        std::random_device rd;
        std::mt19937 gen(rd());  
		std::uniform_int_distribution<int> distribution(0, 2);
        int x = distribution(gen);
        std::this_thread::sleep_for(std::chrono::seconds(x));
    
		// 显示进度
		proBar.displayProgressBar("title", i + 1, 20);
	}
}

int main() {
    
    std::cout << "befor befor" << std::endl;
    std::cout << "befor befor";

    {
        ProgressBarMthread proBar;
        
        std::thread t1(threadFunc, std::ref(proBar));
        std::thread t2(threadFunc, std::ref(proBar));
        std::thread t3(threadFunc, std::ref(proBar));
        t1.join();
        t2.join();
        t3.join();
    }

    std::cout << "after after" << std::endl;
    std::cout << "after after" << std::endl;

    return 0;
}

#c++##进度条#
全部评论

相关推荐

点赞 评论 收藏
分享
05-11 11:48
河南大学 Java
程序员牛肉:我是26届的双非。目前有两段实习经历,大三上去的美团,现在来字节了,做的是国际电商的营销业务。希望我的经历对你有用。 1.好好做你的CSDN,最好是直接转微信公众号。因为这本质上是一个很好的展示自己技术热情的证据。我当时也是烂大街项目(网盘+鱼皮的一个项目)+零实习去面试美团,但是当时我的CSDN阅读量超百万,微信公众号阅读量40万。面试的时候面试官就告诉我说觉得我对技术挺有激情的。可以看看我主页的美团面试面经。 因此花点时间好好做这个知识分享,最好是单拉出来搞一个板块。各大公司都极其看中知识落地的能力。 可以看看我的简历对于博客的描述。这个帖子里面有:https://www.nowcoder.com/discuss/745348200596324352?sourceSSR=users 2.实习经历有一些东西删除了,目前看来你的产出其实很少。有些内容其实很扯淡,最好不要保留。有一些点你可能觉得很牛逼,但是面试官眼里是减分的。 你还能负责数据库表的设计?这个公司得垃圾成啥样子,才能让一个实习生介入数据库表的设计,不要写这种东西。 一个公司的财务审批系统应该是很稳定的吧?为什么你去了才有RBAC权限设计?那这个公司之前是怎么处理权限分离的?这些东西看着都有点扯淡了。 还有就是使用Redis实现轻量级的消息队列?那为什么这一块不使用专业的MQ呢?为什么要使用redis,这些一定要清楚, 就目前看来,其实你的这个实习技术还不错。不要太焦虑。就是有一些内容有点虚了。可以考虑从PR中再投一点产出
投递美团等公司8个岗位
点赞 评论 收藏
分享
评论
4
4
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务