BmnSimpleProgressBar.h

Go to the documentation of this file.

/*
    BM@N alignment routine
    BM@N experiment at NICA complex, JINR, 2025
 
    Department: Math & Soft Group of HEP lab
    Author:     Igor Polev, polev@jinr.ru
 
    BmnSimpleProgressBar class header-only implementation
    Primitive progress bar with some thread-safety
*/
 
#ifndef BMNSIMPLEPROGRESSBAR_H
#define BMNSIMPLEPROGRESSBAR_H
 
#include <iostream>
#include <mutex>
#include <string>
 
class BmnSimpleProgressBar
{
  private:
    bool show{false};
    int done{0};
    std::string bar{"\r [                                                  ] "};   // with 50 spaces
    std::mutex mtx;
 
    inline void Print() { std::cout << bar << done << "%\r" << std::flush; }
 
  public:
    inline BmnSimpleProgressBar() = default;
    inline void Init();
    inline void Tick();
    inline void Clear();
};
 
inline void BmnSimpleProgressBar::Clear()
{
    show = false;
    std::cout << "\r                                                          \r" << std::flush;
}
 
inline void BmnSimpleProgressBar::Init()
{
    done = 0;
    for (int i = 3; i < 53; i++)
        bar[i] = ' ';
    Clear();
    Print();
    show = true;
}
 
inline void BmnSimpleProgressBar::Tick()
{
    if (!show)
        return;
 
    std::lock_guard<std::mutex> lock{mtx};
    if (done >= 100) {
        show = false;
        return;
    }
    if (done % 2)
        bar[done / 2 + 3] = '=';
    else
        bar[done / 2 + 3] = '-';
    Print();
    done++;
}
 
#endif   // BMNSIMPLEPROGRESSBAR_H