FairMCModuleDraw.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                        FairMCModuleDraw source file           -----
// -----                                                               -----
// -------------------------------------------------------------------------
#include "FairMCModuleDraw.h"
 
#include "BmnZdcPoint.h"
#include "CbmMCTrack.h"
#include "FairLogger.h"
#include "FairRootManager.h"
#include "TEvePointSet.h"
#include "TString.h"
 
#include <iostream>
using namespace std;
 
// -----   Default constructor   -------------------------------------------
FairMCModuleDraw::FairMCModuleDraw()
    : FairTask("FairMCModuleDraw", 0)
    , fVerbose(0)
    , fEventManager(nullptr)
    , fPointList(nullptr)
    , fMCTracks(nullptr)
    , fColor(0)
    , fStyle(0)
{}
 
// -----   Standard constructor   ------------------------------------------
FairMCModuleDraw::FairMCModuleDraw(const char* name, Color_t color, Style_t mstyle, Int_t iVerbose)
    : FairTask(name, iVerbose)
    , fVerbose(iVerbose)
    , fEventManager(nullptr)
    , fPointList(nullptr)
    , fMCTracks(nullptr)
    , fColor(color)
    , fStyle(mstyle)
{}
 
// -------------------------------------------------------------------------
InitStatus FairMCModuleDraw::Init()
{
    if (fVerbose > 1)
        cout << "FairMCModuleDraw::Init()" << endl;
 
    FairRootManager* fManager = FairRootManager::Instance();
 
    fPointList = (TClonesArray*)fManager->GetObject(GetName());
    if (fPointList == 0) {
        LOG(error) << "FairMCModuleDraw::Init() branch " << GetName() << " not found! Task will be deactivated";
        SetActive(kFALSE);
        return kERROR;
    }
    if (fVerbose > 2)
        cout << "FairMCModuleDraw::Init() get point list" << fPointList << endl;
 
    fMCTracks = (TClonesArray*)fManager->GetObject("MCTrack");
 
    fEventManager = MpdEventManager::Instance();
    if (fVerbose > 2)
        cout << "FairMCModuleDraw::Init() get instance of MpdEventManager " << endl;
 
    // create array for ZDC modules' visibility
    fEventManager->isZDCModule = new bool[104];
 
    return kSUCCESS;
}
 
void FairMCModuleDraw::Exec(Option_t* option)
{
    if (!IsActive())
        return Reset();
 
    bool* pZDCModuleNumber = fEventManager->isZDCModule;
    Int_t npoints = fPointList->GetEntriesFast();
    for (int i = 0; i < npoints; i++) {
        BmnZdcPoint* p = (BmnZdcPoint*)fPointList->At(i);
 
        CbmMCTrack* pMCTrack = (CbmMCTrack*)fMCTracks->At(p->GetTrackID());
        if (!pMCTrack) {
            cout << "MCTrack is NULL for selected index" << endl;
            continue;
        }
 
        // filter only forward points
        Double32_t start_z = pMCTrack->GetStartZ();
        // cout<<"StartZ: "<<start_z<<cout<<endl;
        if (start_z < 1000.0)
            pZDCModuleNumber[p->GetCopyMother()] = true;
    }
}
 
TVector3 FairMCModuleDraw::GetVector(TObject* obj)
{
    FairMCPoint* p = (FairMCPoint*)obj;
    return TVector3(p->GetX(), p->GetY(), p->GetZ());
}
 
TObject* FairMCModuleDraw::GetValue(TObject* obj, Int_t i)
{
    return new TNamed(Form("Point %d", i), "");
}
 
// -----   Destructor   ----------------------------------------------------
FairMCModuleDraw::~FairMCModuleDraw()
{
    delete[] fEventManager->isZDCModule;
}
 
// -------------------------------------------------------------------------
void FairMCModuleDraw::SetParContainers() {}
 
/** Action after each event**/
void FairMCModuleDraw::Finish() {}
 
// -------------------------------------------------------------------------
void FairMCModuleDraw::Reset()
{
    bool* pZDCModuleNumber = fEventManager->isZDCModule;
    for (int i = 0; i < 104; i++)
        pZDCModuleNumber[i] = false;
}