BmnCounterTask.cxx

Go to the documentation of this file.

#include "BmnCounterTask.h"
 
#include "FairRootManager.h"
 
#include <iostream>
 
#define ANSI_COLOR_RED   "\x1b[91m"
#define ANSI_COLOR_BLUE  "\x1b[94m"
#define ANSI_COLOR_RESET "\x1b[0m"
 
BmnCounter::BmnCounter()
: fEvHead(NULL),
  fNEvents(0),
  fIEvent(0),
  fRunId(0)
{}
 
BmnCounter::BmnCounter(Long64_t n)
: fEvHead(NULL),
  fNEvents(n),
  fIEvent(0),
  fRunId(0)
{}
 
BmnCounter::~BmnCounter()
{}
 
InitStatus BmnCounter::Init()
{
    FairRootManager* ioman = FairRootManager::Instance();
    if (!ioman)
    {
        Fatal("Init", "FairRootManager is not instantiated");
        return kERROR;
    }
    fEvHead = (BmnEventHeader*) ioman->GetObject("EventHeader");
    if (!fEvHead)
        cout << "WARNING! No EventHeader array!!!" << endl;
 
    if (ioman->CheckMaxEventNo(fNEvents) < fNEvents)
        fNEvents = ioman->CheckMaxEventNo(fNEvents);
 
    return kSUCCESS;
}
 
void BmnCounter::Exec(Option_t* opt) {
 
    fRunId = (fEvHead) ? fEvHead->GetRunId() : 0;
 
    printf(ANSI_COLOR_BLUE "RUN-" ANSI_COLOR_RESET);
    printf(ANSI_COLOR_RED "%d" ANSI_COLOR_RESET, fRunId);
    printf(ANSI_COLOR_BLUE ": [");
 
    Float_t progress = fIEvent * 1.0 / fNEvents;
    Int_t barWidth = 70;
 
    Int_t pos = barWidth * progress;
    for (Int_t i = 0; i < barWidth; ++i) {
        if (i < pos) printf("=");
        else if (i == pos) printf(">");
        else printf(" ");
    }
 
    printf("] " ANSI_COLOR_RESET);
    printf(ANSI_COLOR_RED "%d%%\r" ANSI_COLOR_RESET, Int_t(progress * 100.0 + 0.5));
    cout.flush();
    fIEvent++;
}
 
void BmnCounter::Finish() {
    printf("\n");
}