CbmFitGlobalTracks.cxx

Go to the documentation of this file.

/* Copyright (C) 2006-2015 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dmytro Kresan, Denis Bertini [committer], Florian Uhlig */
 
// ------------------------------------------------------------------
// -----                    CbmFitGlobalTracks                  -----
// -----             Created 06/03/2006 by D.Kresan             -----
// ------------------------------------------------------------------
#include "CbmFitGlobalTracks.h"
 
#include "CbmGlobalTrack.h"
#include "CbmGlobalTrackFitter.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using std::cout;
using std::endl;
 
//___________________________________________________________________
//
// CbmFitGlobalTracks
//
// Task for global track fitting. Uses algorithm, provided by concre-
// te implementation of CbmGlobalTrackFitter class. This concrete en-
// gine is to be set in standard constructor or by UseFitter method.
//
 
 
// ------------------------------------------------------------------
CbmFitGlobalTracks::CbmFitGlobalTracks() : FairTask(), fVerbose(0), fFitter(NULL), fArrayGlbTrack(NULL) {}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
CbmFitGlobalTracks::CbmFitGlobalTracks(const char* name, Int_t verbose, CbmGlobalTrackFitter* fitter)
  : FairTask(name, verbose)
  , fVerbose(verbose)
  , fFitter(fitter)
  , fArrayGlbTrack(NULL)
{
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
CbmFitGlobalTracks::~CbmFitGlobalTracks()
{
  // Destructor
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
InitStatus CbmFitGlobalTracks::Init()
{
  // Initialisation of the task. Initialise fitter
  if (NULL == fFitter) {
    cout << "-E- CbmFitGlobalTracks::Init : "
         << "no track fitter selected!" << endl;
    return kERROR;
  }
  fFitter->Init();
  // Get pointer to the ROOT manager
  FairRootManager* rootMgr = FairRootManager::Instance();
  if (NULL == rootMgr) {
    cout << "-E- CbmFitGlobalTracks::Init : "
         << "CBM ROOT manager is not instantiated!" << endl;
    return kERROR;
  }
  fArrayGlbTrack = (TClonesArray*) rootMgr->GetObject("GlobalTrack");
  if (NULL == fArrayGlbTrack) {
    cout << "-W- CbmFitGlobalTracks::Init : "
         << "no global track array!" << endl;
  }
  return kSUCCESS;
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
void CbmFitGlobalTracks::Exec(Option_t*)
{
  // Task execution. Call DoFit of the fitter for each global track
  if (NULL != fArrayGlbTrack) {
    CbmGlobalTrack* glbTrack;
    for (Int_t iGlbTrack = 0; iGlbTrack < fArrayGlbTrack->GetEntriesFast(); iGlbTrack++) {
      // Get pointer to the global track
      glbTrack = (CbmGlobalTrack*) fArrayGlbTrack->At(iGlbTrack);
      if (NULL == glbTrack) continue;
      // Fit the global track
      fFitter->DoFit(glbTrack);
    }
  }
}
// ------------------------------------------------------------------
 
 
// ------------------------------------------------------------------
void CbmFitGlobalTracks::Finish()
{
  // Finish of the task execution
}
// ------------------------------------------------------------------
 
 
ClassImp(CbmFitGlobalTracks);