/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_14/src/RecoBTau/JetTagMVALearning/plugins/EventProgress.cc

Go to the documentation of this file.

#include <time.h>

#include <iostream>
#include <iomanip>
#include <limits>
#include <cmath>

#include "EventProgress.h"

EventProgress::EventProgress() :
        nEvents(0), startTime(time(NULL)), lastUpdate(0), displayWidth(0)
{
}

EventProgress::EventProgress(unsigned long _nEvents) :
        nEvents(_nEvents), startTime(time(NULL)), lastUpdate(0),
        displayWidth(int(std::log(nEvents + 0.5) / M_LN10 + 1))
{
}

EventProgress::~EventProgress()
{
        std::cout << "                            \r" << std::flush;
}

void EventProgress::update(unsigned long event)
{
        time_t currentTime = time(0);
        if (currentTime == lastUpdate)
                return;

        lastUpdate = currentTime;

        if (displayWidth)
                std::cout << "Event " << std::setw(displayWidth) << event;
        else
                std::cout << "Event " << event;

        if (nEvents) {
                unsigned long eta = (event >= 10) ? ((currentTime - startTime) * (nEvents - event) / event)
                                                  : std::numeric_limits<unsigned long>::max();

                std::cout << " (" << std::setw(2) << (event * 100 / nEvents) << "%), ETA ";
                if (eta >= 6000)
                        std::cout << "##:##\r";
                else
                        std::cout << std::setw(2) << (eta / 60) << ":"
                                  << std::setfill('0') << std::setw(2) << (eta % 60);
        }

        std::cout << std::setfill(' ')
                  << std::resetiosflags(std::ios::fixed)
                  << std::resetiosflags(std::ios::scientific)
                  << std::resetiosflags(std::ios::floatfield)
                  << std::setprecision(8)
                  << "\r" << std::flush;
}