CbmL1GlobalTrackFinder.cxx

Go to the documentation of this file.

/* Copyright (C) 2019-2020 IKF-UFra, GSI
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Valentina Akishina [committer] */
 
/*
 *====================================================================
 *
 *  CBM Level 1 Reconstruction
 *
 *  Authors: V. Akishina
 *
 *  e-mail : v.akishina@gsi.de
 *
 *====================================================================
 *
 *  CbmL1GlobalTrackFinder source file
 *
 *====================================================================
 */
#include "CbmL1GlobalTrackFinder.h"
 
#include "CbmEvent.h"
#include "CbmGlobalTrack.h"
#include "CbmKfUtil.h"
#include "CbmMuchTrack.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "CbmTofTrack.h"
#include "CbmTrack.h"
#include "CbmTrdTrack.h"
#include "FairHit.h"
#include "FairMCPoint.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
#include "TDatabasePDG.h"
 
#include <vector>
 
using std::vector;
 
ClassImp(CbmL1GlobalTrackFinder)
 
  // -----   Default constructor   -------------------------------------------
  CbmL1GlobalTrackFinder::CbmL1GlobalTrackFinder()
  : fGlobalTracks(nullptr)
  , fStsTracks(nullptr)
  , fMuchTracks(nullptr)
  , fTrdTracks(nullptr)
  , fTofTracks(nullptr)
{
  fName = "Global Track Finder L1";
}
// -------------------------------------------------------------------------
 
 
// -----   Destructor   ----------------------------------------------------
CbmL1GlobalTrackFinder::~CbmL1GlobalTrackFinder() {}
// -------------------------------------------------------------------------
 
// -----   Public method Init   --------------------------------------------
void CbmL1GlobalTrackFinder::Init() {}
// -------------------------------------------------------------------------
 
 
// -----   Copy tracks to output array   -----------------------------------
Int_t CbmL1GlobalTrackFinder::CopyL1Tracks(CbmEvent* event)
{
  CbmL1* L1 = CbmL1::Instance();
  if (!L1) return 0;
  Int_t globalTrackIndex      = fGlobalTracks->GetEntriesFast();
  Int_t globalTrackIndexStart = globalTrackIndex;
 
  Int_t stsTrackIndex  = fStsTracks->GetEntriesFast();
  Int_t muchTrackIndex = fMuchTracks->GetEntriesFast();
  Int_t trdTrackIndex  = fTrdTracks->GetEntriesFast();
  Int_t tofTrackIndex  = fTofTracks->GetEntriesFast();
 
  LOG(debug) << "Copy L1 tracks : " << L1->fvRecoTracks.size() << " tracks in L1";
 
  for (vector<CbmL1Track>::iterator it = L1->fvRecoTracks.begin(); it != L1->fvRecoTracks.end(); ++it) {
    CbmL1Track& T = *it;
    //BEGIN add global track
    new ((*fGlobalTracks)[globalTrackIndex]) CbmGlobalTrack();
    if (event) event->AddData(ECbmDataType::kGlobalTrack, globalTrackIndex);
    CbmGlobalTrack* t = dynamic_cast<CbmGlobalTrack*>(fGlobalTracks->At(globalTrackIndex++));
    t->SetFlag(0);
    t->SetParamFirst(cbm::kf::ConvertTrackParam(T));
    t->SetParamLast(cbm::kf::ConvertTrackParam(T.TLast));
    t->SetChi2(T.GetChiSq());
    //   t->SetLength(T.length);
    t->SetNDF(T.GetNdf());
    t->SetChiSqTime(T.GetChiSqTime());
    t->SetNdfTime(T.GetNdfTime());
    t->SetPidHypo(T.GetQp() >= 0 ? 211 : -211);
    //     t->SetTime(T.Tpv[6]);
    //     t->SetTimeError(T.Cpv[20]);
    //END add global track
    //BEGIN create detector tracks if needed
    bool hasStsHits  = false;
    bool hasMuchHits = false;
    bool hasTrdHits  = false;
    bool hasTofHits  = false;
    for (vector<int>::iterator ih = it->Hits.begin(); ih != it->Hits.end(); ++ih) {
      const CbmL1HitId& h = L1->fvExternalHits[*ih];
 
      if (h.detId <= 1 && hasStsHits == false) {
        hasStsHits         = true;
        CbmStsTrack* track = new ((*fStsTracks)[stsTrackIndex]) CbmStsTrack();
        t->SetStsTrackIndex(stsTrackIndex);
        if (event) event->AddData(ECbmDataType::kStsTrack, stsTrackIndex);
        CbmL1TrackToCbmStsTrack(T, track);
        stsTrackIndex++;
      }
      if (h.detId == 2 && hasMuchHits == false) {
        hasMuchHits = true;
 
        CbmMuchTrack* track = new ((*fMuchTracks)[muchTrackIndex]) CbmMuchTrack();
        t->SetMuchTrackIndex(muchTrackIndex);
        if (event) event->AddData(ECbmDataType::kMuchTrack, muchTrackIndex);
        CbmL1TrackToCbmMuchTrack(T, track, h.detId);
        muchTrackIndex++;
      }
      if (h.detId == 3 && hasTrdHits == false) {
        hasTrdHits         = true;
        CbmTrdTrack* track = new ((*fTrdTracks)[trdTrackIndex]) CbmTrdTrack();
        t->SetTrdTrackIndex(trdTrackIndex);
        if (event) event->AddData(ECbmDataType::kTrdTrack, trdTrackIndex);
        CbmL1TrackToCbmTrdTrack(T, track, h.detId);
        trdTrackIndex++;
      }
      if (h.detId == 4 && hasTofHits == false) {
        hasTofHits         = true;
        CbmTofTrack* track = new ((*fTofTracks)[tofTrackIndex]) CbmTofTrack();
 
        t->SetTofTrackIndex(tofTrackIndex);
 
        if (event) event->AddData(ECbmDataType::kTofTrack, tofTrackIndex);
        CbmL1TrackToCbmTofTrack(T, track, h.detId);
        tofTrackIndex++;
 
        // if (event) event->AddData(ECbmDataType::kTofHit, h.hitId);
      }
    }
    //END create detector tracks if needed
  }
  return globalTrackIndex - globalTrackIndexStart;
}
// -------------------------------------------------------------------------
 
 
// -----   Public method CbmL1TrackToCbmTrack   ------------------------------------------
void CbmL1GlobalTrackFinder::CbmL1TrackToCbmTrack(CbmL1Track l1track, CbmTrack* track, int /*systemIdT*/)
{
  track->SetChiSq(l1track.GetChiSq());
  track->SetNDF(l1track.GetNdf());
  //track->SetPreviousTrackId(l1track.GetPreviousTrackId());//???
  //track->SetFlag(l1track.GetQuality());//???
 
  track->SetParamFirst(cbm::kf::ConvertTrackParam(l1track));
  track->SetParamLast(cbm::kf::ConvertTrackParam(l1track.TLast));
}
// -------------------------------------------------------------------------
 
// -----   Public method CbmL1TrackToCbmStsTrack   ------------------------------------------
void CbmL1GlobalTrackFinder::CbmL1TrackToCbmStsTrack(CbmL1Track l1track, CbmStsTrack* track)
{
  CbmL1Track& T = l1track;
  CbmL1* L1     = CbmL1::Instance();
 
  for (vector<int>::iterator ih = T.Hits.begin(); ih != T.Hits.end(); ++ih) {
    CbmL1HitId& h = L1->fvExternalHits[*ih];
    if (h.detId == 0) {
      track->AddMvdHit(h.hitId);
    }
    else if (h.detId == 1) {
      track->AddStsHit(h.hitId);
    }
  }
 
  track->SetFlag(0);
  track->SetChiSq(T.GetChiSq());
  track->SetNDF(T.GetNdf());
  track->SetPidHypo(T.GetQp() >= 0. ? 211 : -211);
  track->SetStartTime(T.Tpv.GetTime());
  track->SetStartTimeError(T.Tpv.GetTimeError());
  track->SetFirstHitTime(T.GetTime());
  track->SetFirstHitTimeError(T.GetTimeError());
  track->SetLastHitTime(T.TLast.GetTime());
  track->SetLastHitTimeError(T.TLast.GetTimeError());
 
  track->SetParamFirst(cbm::kf::ConvertTrackParam(T));
  track->SetParamLast(cbm::kf::ConvertTrackParam(T.TLast));
}
// -------------------------------------------------------------------------
 
// -----   Public method CbmL1TrackToCbmMuchTrack   ------------------------------------------
void CbmL1GlobalTrackFinder::CbmL1TrackToCbmMuchTrack(CbmL1Track l1track, CbmMuchTrack* track, int systemIdT)
{
 
  CbmL1Track& T = l1track;
 
  CbmL1* L1 = CbmL1::Instance();
 
  for (vector<int>::iterator ih = T.Hits.begin(); ih != T.Hits.end(); ++ih) {
    CbmL1HitId& h = L1->fvExternalHits[*ih];
    if (h.detId != systemIdT) continue;
    track->AddHit(h.hitId, kMUCHPIXELHIT);
  }
  track->SetChiSq(T.GetChiSq());
  track->SetNDF(T.GetNdf());
  //track->SetPreviousTrackId(T->GetPreviousTrackId());//???
  //track->SetFlag(T->GetQuality());//???
 
  track->SetParamFirst(cbm::kf::ConvertTrackParam(T));
  track->SetParamLast(cbm::kf::ConvertTrackParam(T.TLast));
}
// -------------------------------------------------------------------------
 
// -----   Public method CbmL1TrackToCbmTrdTrack   ------------------------------------------
void CbmL1GlobalTrackFinder::CbmL1TrackToCbmTrdTrack(CbmL1Track l1track, CbmTrdTrack* track, int systemIdT)
{
  CbmL1Track& T = l1track;
 
  CbmL1* L1 = CbmL1::Instance();
 
  for (vector<int>::iterator ih = T.Hits.begin(); ih != T.Hits.end(); ++ih) {
    CbmL1HitId& h = L1->fvExternalHits[*ih];
    if (h.detId != systemIdT) continue;
    track->AddHit(h.hitId, kTRDHIT);
  }
 
  track->SetChiSq(T.GetChiSq());
  track->SetNDF(T.GetNdf());
  //track->SetPreviousTrackId(T->GetPreviousTrackId());//???
  //track->SetFlag(T->GetQuality());//???
 
  track->SetParamFirst(cbm::kf::ConvertTrackParam(T.Tpv));
  track->SetParamLast(cbm::kf::ConvertTrackParam(T.TLast));
}
// -------------------------------------------------------------------------
 
// -----   Public method CbmL1TrackToCbmTofTrack   ------------------------------------------
void CbmL1GlobalTrackFinder::CbmL1TrackToCbmTofTrack(CbmL1Track l1track, CbmTofTrack* track, int systemIdT)
{
  CbmL1Track& T = l1track;
 
  CbmL1* L1 = CbmL1::Instance();
 
  for (vector<int>::iterator ih = T.Hits.begin(); ih != T.Hits.end(); ++ih) {
    CbmL1HitId& h = L1->fvExternalHits[*ih];
    if (h.detId != systemIdT) continue;
    track->AddHit(h.hitId, kTOFHIT);
  }
  track->SetChiSq(T.GetChiSq());
  track->SetNDF(T.GetNdf());
  //track->SetPreviousTrackId(T->GetPreviousTrackId());//???
  //track->SetFlag(T->GetQuality());//???
 
  track->SetParamFirst(cbm::kf::ConvertTrackParam(T));
  track->SetParamLast(cbm::kf::ConvertTrackParam(T.TLast));
}
// -------------------------------------------------------------------------
 
// -----   Public method DoFind   ------------------------------------------
Int_t CbmL1GlobalTrackFinder::DoFind()
{
 
  if (!fTracks) {
    LOG(error) << "-E- CbmL1GlobalTrackFinder::DoFind: "
               << "Track array missing! ";
    return -1;
  }
 
  CbmL1* L1 = CbmL1::Instance();
  if (!L1) return 0;
 
  L1->Reconstruct();
  int ntracks = CopyL1Tracks();
 
  return ntracks;
}
// -------------------------------------------------------------------------
 
 
// -----   Track finding in one event   ------------------------------------
Int_t CbmL1GlobalTrackFinder::FindTracks(CbmEvent* event)
{
 
  CbmL1* l1 = CbmL1::Instance();
  if (!l1) return 0;
 
  l1->Reconstruct(event);
  int nTracks = CopyL1Tracks(event);
 
  return nTracks;
}
// -------------------------------------------------------------------------
 
 
// -------------------------------------------------------------------------
void CbmL1GlobalTrackFinder::SetDefaultParticlePDG(int pdg)
{
  CbmL1* l1 = CbmL1::Instance();
  if (!l1 || !l1->fpAlgo) {
    LOG(fatal) << "L1 instance doesn't exist or is not initialised";
    return;
  }
  auto* p = TDatabasePDG::Instance()->GetParticle(pdg);
  if (!p) {
    LOG(fatal) << "Particle with pdg " << pdg << " doesn't exist";
    return;
  }
  l1->fpAlgo->SetDefaultParticleMass(p->Mass());
}