|| || || || || || || ||

---

ProgressPrinter.h

Go to the documentation of this file.

#ifndef Util_ProgressPrinter_h
#define Util_ProgressPrinter_h

#include "Basis/CmdOptions.h"
#include "Link/Mpi/MpiFuncs.h"

/* will print
/ name [========          ]  45%
                             ^ percentage
       ^ progress bar
  ^ name of this progress
^ spinning mark
*/

namespace Impala
{
namespace Util
{

class ProgressPrinter
{
public:
    ProgressPrinter(std::ostream& console) :
        mConsole(console)
    {
        mDoShow = CmdOptions::GetInstance().GetBool("showProgress", true);
        if (Link::Mpi::MpiUsed())
            mDoShow = false;
        mRange = 100;
        mLinewidth = 79;
        mMinimumBarLength = 20;
    }

    void Start(const std::string& name, int range)
    {
        if (mDoShow)
        {
            mSpinner = 0;
            mName = name;
            mRange = range;
            if(mName.size() > mLinewidth - 10 - mMinimumBarLength)
                mName = mName.substr(0, mLinewidth - 10 - mMinimumBarLength - 3) + "...";
            mProgressBarLength = mLinewidth - 10 - mName.size();
            WriteLine(0);
        }
    }

    void PrintProgress(int progress)
    {
        if (mDoShow)
        {
            EraseLine();
            WriteLine(progress);
        }
    }

    void Done()
    {
        if (mDoShow)
        {
            EraseLine();
            WriteLine(mRange);
            mConsole << "\n";
            mConsole.flush();
        }
    }

private:
    void WriteLine(int progress)
    {
        switch(mSpinner)
        {
        case 0: mConsole << '/'; mSpinner=1; break;
        case 1: mConsole << '-'; mSpinner=2; break;
        case 2: mConsole << '\\'; mSpinner=3; break;
        case 3: mConsole << '|'; mSpinner=0; break;
        }
        mConsole << ' ' << mName << " [";
        int length = mProgressBarLength * progress / mRange;
        mConsole << std::string(length, '=') << std::string(mProgressBarLength - length, ' ');
        std::string percentage = Impala::MakeString((int)(100 * ((double)progress / (double)mRange)));
        mConsole << "] " << std::string(3-percentage.size(), ' ') << percentage << '%';
        mConsole.flush();
    }

    void EraseLine()
    {
        if (Link::Mpi::MpiUsed())
            return;
        mConsole << std::string(mLinewidth,'\b');
        mConsole.flush();
    }

    std::ostream &mConsole;
    int mRange;
    int mProgressBarLength;
    std::string mName;
    int mLinewidth;
    int mMinimumBarLength;
    int mSpinner;

    bool mDoShow;
};

}//namespace Util
}//namespace Impala

#endif //Util_ProgressPrinter_h

---