BmnCheckMCTask.cxx

Go to the documentation of this file.

#include "BmnCheckMCTask.h"
 
BmnCheckMCTask::BmnCheckMCTask(Long64_t nevents, Int_t minHits, Int_t code, vector<Int_t> outCodes)
{
    fMCTracksBranchName = "MCTrack";
    fGemPointsBranchName = "StsPoint";
    fSilPointsBranchName = "SiliconPoint";
    fCSCPointsBranchName = "CSCPoint";
    fPVertexName = "BmnVertex";
    fEvHeaderName = "DstEventHeader.";
    fMCEvHeaderName = "MCEventHeader.";
    fMCTracks = nullptr;
    fGemPoints = nullptr;
    fSilPoints = nullptr;
    fCSCPoints = nullptr;
    fRunSimInst = nullptr;
    fPVertexShow = nullptr;
    fEvHeaderShow = nullptr;
    fMCEvHeader = nullptr;
    fMinHits = minHits;
    fPDGCode = code;
    fPDGOutCodes = outCodes;
    nMaxValidEvents = nevents;
    nValidEvents = 0;
    nVertices = 0;
}
 
BmnCheckMCTask::~BmnCheckMCTask() {}
 
InitStatus BmnCheckMCTask::Init()
{
    if (fVerbose > 0)
        printf("Recon Checker Init\n");
 
    FairRootManager* ioman = FairRootManager::Instance();
    if (NULL == ioman)
        Fatal("Init", "FairRootManager is not instantiated");
    fMCTracks = (TClonesArray*)ioman->GetObject(fMCTracksBranchName);
    fGemPoints = (TClonesArray*)ioman->GetObject(fGemPointsBranchName);
    fSilPoints = (TClonesArray*)ioman->GetObject(fSilPointsBranchName);
    fCSCPoints = (TClonesArray*)ioman->GetObject(fCSCPointsBranchName);
    fPVertexShow = (TClonesArray*)ioman->GetObject(fPVertexName);
    fEvHeaderShow = (DstEventHeader*)ioman->GetObject(fEvHeaderName);
    //    fMCEvHeader = (FairMCEventHeader*) ioman->GetObject(fMCEvHeaderName);
 
    return kSUCCESS;
}
 
void BmnCheckMCTask::Exec(Option_t* option)
{
    if (!fMCEvHeader) {   // surely it should be in the init (if it would worked with FairRunSim =( )
        FairRootManager* ioman = FairRootManager::Instance();
        fMCEvHeader = (FairMCEventHeader*)ioman->GetObject(fMCEvHeaderName);
    }
    if (fVerbose > 1)
        printf("\nBmnCheckMCTask::Exec eventID = %u  GetSaveEvent() = %d\n", fEvHeaderShow->GetEventId(),
               fRunSimInst->GetSaveEvent());
    if (fMCTracks == nullptr)
        return;
    for (Int_t iTrack = 0; iTrack < fPVertexShow->GetEntriesFast(); iTrack++) {
        CbmVertex* v = (CbmVertex*)fPVertexShow->UncheckedAt(iTrack);
        fVZ = v->GetZ();
        //        if (fVerbose > 1)
        //            printf(" Read PV ( %f %f %f)\n", v->GetX(), v->GetY(), v->GetZ());
    }
    fRunSimInst->SetSaveEvent(kFALSE);
    if (fVZ < CutExistentVertex) {
        return;
    }
    if (Abs(fVZ) > Abs(CutValidZ)) {
        nVertices++;
        return;
    }
    if (BmnRecoTools::IsReconstructable(fMCTracks, fGemPoints, fSilPoints, fCSCPoints, fPDGCode, fPDGOutCodes,
                                        fMinHits))
    {
        fRunSimInst->SetSaveEvent(kTRUE);
        nValidEvents++;
        nVertices++;
        fMCEvHeader->SetEventID(fEvHeaderShow->GetEventId());
        if (fVerbose > 1)
            printf("Recon\n");
        if (nValidEvents >= nMaxValidEvents && nMaxValidEvents != 0)
            fRunSimInst->GetMCApp()->StopMCRun();
    } else {
        if (fVerbose > 1)
            printf("NonRec\n");
    }
    //    if (fVerbose > 1)
    //        printf("\n%s : exec  SET = %d\n", typeid (*this).name(), fRunSimInst->GetSaveEvent());
    if (fVerbose > 1)
        printf("\nValid events: %lld\n", nValidEvents);
}
 
void BmnCheckMCTask::Finish()
{
    printf("\nBmnCheckMCTask: Found: \n\t%lld vertices, \n\t%lld valid events\n", nVertices, nValidEvents);
}
 
void BmnCheckMCTask::FinishEvent()
{
    if (fVerbose > 1)
        printf("\n%s : finish event\n", typeid(*this).name());
}