CbmLitFindGlobalTracksIdeal.cxx

Go to the documentation of this file.

/* Copyright (C) 2009-2020 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#include "CbmLitFindGlobalTracksIdeal.h"
 
#include "CbmGlobalTrack.h"
#include "CbmMCTrack.h"
#include "CbmMatch.h"
#include "CbmTofHit.h"
#include "CbmTofPoint.h"
#include "CbmTrackMatchNew.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <iostream>
 
CbmLitFindGlobalTracksIdeal::CbmLitFindGlobalTracksIdeal()
  : fDet()
  ,
 
  fMCTracks(NULL)
  , fStsMatches(NULL)
  , fMuchMatches(NULL)
  , fTrdMatches(NULL)
  , fTofMCPoints(NULL)
  , fTofHits(NULL)
  , fTofHitsMatches(NULL)
  , fGlobalTracks(NULL)
  ,
 
  fMcStsMap()
  , fMcTrdMap()
  , fMcMuchMap()
  , fMcTofMap()
  ,
 
  fEventNo(0)
{
}
 
CbmLitFindGlobalTracksIdeal::~CbmLitFindGlobalTracksIdeal() {}
 
InitStatus CbmLitFindGlobalTracksIdeal::Init()
{
  fDet.DetermineSetup();
  std::cout << fDet.ToString();
 
  ReadDataBranches();
 
  return kSUCCESS;
}
 
void CbmLitFindGlobalTracksIdeal::Exec(Option_t* opt)
{
  if (fDet.GetDet(ECbmModuleId::kSts)) {
    FillTrackMap(fMcStsMap, fStsMatches);
  }
  if (fDet.GetDet(ECbmModuleId::kTrd)) {
    FillTrackMap(fMcTrdMap, fTrdMatches);
  }
  if (fDet.GetDet(ECbmModuleId::kMuch)) {
    FillTrackMap(fMcMuchMap, fMuchMatches);
  }
  if (fDet.GetDet(ECbmModuleId::kTof)) {
    FillMapTof();
  }
 
  CreateGlobalTracks();
 
  std::cout << "Event: " << fEventNo++ << std::endl;
}
 
void CbmLitFindGlobalTracksIdeal::Finish() {}
 
void CbmLitFindGlobalTracksIdeal::ReadDataBranches()
{
  FairRootManager* ioman = FairRootManager::Instance();
  if (NULL == ioman) {
    LOG(fatal) << GetName() << "::Init: CbmRootManager is not instantiated";
  }
 
  fMCTracks = (TClonesArray*) ioman->GetObject("MCTrack");
  if (NULL == fMCTracks) {
    LOG(fatal) << GetName() << "::Init: No MCTrack array!";
  }
 
  //STS data
  if (fDet.GetDet(ECbmModuleId::kSts)) {
    fStsMatches = (TClonesArray*) ioman->GetObject("StsTrackMatch");
    if (NULL == fStsMatches) {
      LOG(fatal) << GetName() << "::Init: No StsTrackMatch array!";
    }
  }
 
  //MUCH data
  if (fDet.GetDet(ECbmModuleId::kMuch)) {
    fMuchMatches = (TClonesArray*) ioman->GetObject("MuchTrackMatch");
    if (NULL == fMuchMatches) {
      LOG(fatal) << GetName() << "::Init: No MuchTrackMatch array!";
    }
  }
 
  //TRD data
  if (fDet.GetDet(ECbmModuleId::kTrd)) {
    fTrdMatches = (TClonesArray*) ioman->GetObject("TrdTrackMatch");
    if (NULL == fTrdMatches) {
      LOG(fatal) << GetName() << "::Init: No TrdTrackMatch array!";
    }
  }
 
  //TOF data
  if (fDet.GetDet(ECbmModuleId::kTof)) {
    fTofMCPoints = (TClonesArray*) ioman->GetObject("TofPoint");
    if (NULL == fTofMCPoints) {
      LOG(fatal) << GetName() << "::Init: No TofPoint array!";
    }
    fTofHits = (TClonesArray*) ioman->GetObject("TofHit");
    if (NULL == fTofHits) {
      LOG(fatal) << GetName() << "::Init: No TofHit array!";
    }
    fTofHitsMatches = (TClonesArray*) ioman->GetObject("TofHitMatch");
    if (NULL == fTofHitsMatches) {
      LOG(fatal) << GetName() << "::Init: No TofHitMatch array!";
    }
  }
 
  // Create and register CbmGlobalTrack array
  fGlobalTracks = new TClonesArray("CbmGlobalTrack", 100);
  ioman->Register("GlobalTrack", "Global", fGlobalTracks, IsOutputBranchPersistent("GlobalTrack"));
}
 
void CbmLitFindGlobalTracksIdeal::FillTrackMap(std::map<Int_t, Int_t>& mcMap, const TClonesArray* matches)
{
  mcMap.clear();
  Int_t nofTracks = matches->GetEntriesFast();
  for (Int_t iTrack = 0; iTrack < nofTracks; iTrack++) {
    CbmTrackMatchNew* trackMatch = (CbmTrackMatchNew*) matches->At(iTrack);
    if (trackMatch == NULL) {
      continue;
    }
    Int_t mcId = trackMatch->GetMatchedLink().GetIndex();
    if (mcId == -1) {
      continue;
    }
    mcMap.insert(std::pair<Int_t, Int_t>(mcId, iTrack));
  }
}
 
void CbmLitFindGlobalTracksIdeal::FillMapTof()
{
  fMcTofMap.clear();
  Int_t nofTofHits = fTofHits->GetEntriesFast();
  for (Int_t iTofHit = 0; iTofHit < nofTofHits; iTofHit++) {
    CbmTofHit* tofHit = (CbmTofHit*) fTofHits->At(iTofHit);
    if (tofHit == NULL) {
      continue;
    }
    CbmMatch* tofHitMatch = (CbmMatch*) fTofHitsMatches->At(iTofHit);
    if (tofHitMatch == NULL) {
      continue;
    }
    Int_t tofPointIndex   = tofHitMatch->GetMatchedLink().GetIndex();
    CbmTofPoint* tofPoint = (CbmTofPoint*) fTofMCPoints->At(tofPointIndex);
    if (tofPoint == NULL) {
      continue;
    }
    Int_t mcId = tofPoint->GetTrackID();
    if (mcId == -1) {
      continue;
    }
    fMcTofMap.insert(std::pair<Int_t, Int_t>(mcId, iTofHit));
  }
}
 
void CbmLitFindGlobalTracksIdeal::CreateGlobalTracks()
{
  fGlobalTracks->Clear();
  Int_t nGlobalTracks = 0;
  Int_t nofMCTracks   = fMCTracks->GetEntriesFast();
  for (Int_t iMCTrack = 0; iMCTrack < nofMCTracks; iMCTrack++) {
    CbmMCTrack* mcTrack = (CbmMCTrack*) fMCTracks->At(iMCTrack);
    if (mcTrack == NULL) {
      continue;
    }
    Int_t stsId = -1, trdId = -1, muchId = -1, tofId = -1;
    if (fDet.GetDet(ECbmModuleId::kSts) && (fMcStsMap.find(iMCTrack) != fMcStsMap.end())) {
      stsId = fMcStsMap[iMCTrack];
    }
    if (fDet.GetDet(ECbmModuleId::kTrd) && (fMcTrdMap.find(iMCTrack) != fMcTrdMap.end())) {
      trdId = fMcTrdMap[iMCTrack];
    }
    if (fDet.GetDet(ECbmModuleId::kMuch) && (fMcMuchMap.find(iMCTrack) != fMcMuchMap.end())) {
      muchId = fMcMuchMap[iMCTrack];
    }
    if (fDet.GetDet(ECbmModuleId::kTof) && (fMcTofMap.find(iMCTrack) != fMcTofMap.end())) {
      tofId = fMcTofMap[iMCTrack];
    }
 
    if (stsId == -1 && trdId == -1 && muchId == -1 && tofId == -1) {
      continue;
    }
 
    CbmGlobalTrack* globalTrack = new ((*fGlobalTracks)[nGlobalTracks++]) CbmGlobalTrack();
    globalTrack->SetStsTrackIndex(stsId);
    globalTrack->SetTrdTrackIndex(trdId);
    globalTrack->SetTofHitIndex(tofId);
    globalTrack->SetMuchTrackIndex(muchId);
  }
}
 
ClassImp(CbmLitFindGlobalTracksIdeal);