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");
}