CbmGlobalTrackFitterIdeal.cxx

Go to the documentation of this file.

/* Copyright (C) 2006-2012 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dmytro Kresan, Denis Bertini [committer], Florian Uhlig */
 
// ------------------------------------------------------------------
// -----                  CbmGlobalTrackFitterIdeal             -----
// -----              Created 07/03/2006 by D.Kresan            -----
// ------------------------------------------------------------------
#include "CbmGlobalTrackFitterIdeal.h"
 
#include "CbmGlobalTrack.h"
#include "CbmStsTrack.h"
#include "CbmTrdTrack.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using std::cout;
using std::endl;
 
//___________________________________________________________________
//
// CbmGlobalTrackFitterIdeal
//
// Concrete implementation of the global track fitting algorithm, ba-
// sed on the MC information
//
 
 
// ------------------------------------------------------------------
CbmGlobalTrackFitterIdeal::CbmGlobalTrackFitterIdeal()
  : CbmGlobalTrackFitter()
  , fArrayStsTrack(NULL)
  , fArrayTrdTrack(NULL)
{
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
CbmGlobalTrackFitterIdeal::~CbmGlobalTrackFitterIdeal()
{
  // Destructor
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
void CbmGlobalTrackFitterIdeal::Init()
{
  // Initialisation
 
  // Get pointer to the ROOT I/O manager
  FairRootManager* rootMgr = FairRootManager::Instance();
  if (NULL == rootMgr) {
    cout << "-E- CbmGlobalTrackFitterIdeal::Init :"
         << " ROOT manager is not instantiated" << endl;
    return;
  }
  // Get track arrays
  fArrayStsTrack = (TClonesArray*) rootMgr->GetObject("StsTrack");
  if (NULL == fArrayStsTrack) {
    cout << "-W- CbmGlobalTrackFitterIdeal::Init : "
         << "no STS track array!" << endl;
  }
  fArrayTrdTrack = (TClonesArray*) rootMgr->GetObject("TrdTrack");
  if (NULL == fArrayTrdTrack) {
    cout << "-W- CbmGlobalTrackFitterIdeal::Init : "
         << "no TRD track array!" << endl;
  }
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
void CbmGlobalTrackFitterIdeal::DoFit(CbmGlobalTrack* glbTrack)
{
  // Implementation of the algorithm
  if (NULL == glbTrack) return;
  if (NULL == fArrayStsTrack || NULL == fArrayTrdTrack) return;
 
  if (glbTrack->GetStsTrackIndex() < 0 && glbTrack->GetTrdTrackIndex() < 0) {
    return;
  }
  else if (glbTrack->GetStsTrackIndex() >= 0 && glbTrack->GetTrdTrackIndex() < 0) {
    CbmStsTrack* stsTrack = (CbmStsTrack*) fArrayStsTrack->At(glbTrack->GetStsTrackIndex());
    if (NULL == stsTrack) {
      return;
    }
    glbTrack->SetParamFirst(stsTrack->GetParamFirst());
    glbTrack->SetParamLast(stsTrack->GetParamLast());
  }
  else if (glbTrack->GetStsTrackIndex() < 0 && glbTrack->GetTrdTrackIndex() >= 0) {
    CbmTrdTrack* trdTrack = (CbmTrdTrack*) fArrayTrdTrack->At(glbTrack->GetTrdTrackIndex());
    if (NULL == trdTrack) {
      return;
    }
    glbTrack->SetParamFirst(trdTrack->GetParamFirst());
    glbTrack->SetParamLast(trdTrack->GetParamLast());
  }
  else {
    CbmStsTrack* stsTrack = (CbmStsTrack*) fArrayStsTrack->At(glbTrack->GetStsTrackIndex());
    if (NULL == stsTrack) {
      return;
    }
    CbmTrdTrack* trdTrack = (CbmTrdTrack*) fArrayTrdTrack->At(glbTrack->GetTrdTrackIndex());
    if (NULL == trdTrack) {
      return;
    }
    glbTrack->SetParamFirst(stsTrack->GetParamFirst());
    glbTrack->SetParamLast(trdTrack->GetParamLast());
  }
}
// ------------------------------------------------------------------
 
 
ClassImp(CbmGlobalTrackFitterIdeal);