CbmEcalTrackExtrapolationKF.cxx

Go to the documentation of this file.

 
#include <iostream>
 
#include "TClonesArray.h"
 
#include "CbmEcalInf.h"
#include "CbmEcalStructure.h"
#include "CbmEcalTrackExtrapolationKF.h"
#include "CbmGlobalTrack.h"
#include "CbmKFTrack.h"
#include "CbmMCTrack.h"
#include "CbmStsKFTrackFitter.h"
#include "CbmStsTrack.h"
#include "FairRootManager.h"
#include "FairTrackParam.h"
 
using namespace std;
 
// -----   Default constructor   -------------------------------------------
CbmEcalTrackExtrapolationKF::CbmEcalTrackExtrapolationKF()
  : CbmEcalTrackExtrapolation(1)
  , fTrackParamArray(NULL)
  , fSTSArray(NULL)
  , iEx(-1) {
  ;
}
// -------------------------------------------------------------------------
 
 
// -----   Standard constructor   ------------------------------------------
CbmEcalTrackExtrapolationKF::CbmEcalTrackExtrapolationKF(Int_t verbose)
  : CbmEcalTrackExtrapolation(verbose)
  , fTrackParamArray(NULL)
  , fSTSArray(NULL)
  , iEx(-1) {
  ;
}
// -------------------------------------------------------------------------
 
 
// -----   Destructor   ----------------------------------------------------
CbmEcalTrackExtrapolationKF::~CbmEcalTrackExtrapolationKF() { ; }
// -------------------------------------------------------------------------
 
 
// -----   Public method Init   --------------------------------------------
void CbmEcalTrackExtrapolationKF::Init() {
  CbmEcalTrackExtrapolation::Init();
  //Get and check FairRootManager
  FairRootManager* ioman = FairRootManager::Instance();
  if (!ioman)
    Fatal("CbmEcalTrackExtrapolationKF::Init()",
          "Can't instantise Rootmanager");
 
  //Get STS track array
  fSTSArray = (TClonesArray*) ioman->GetObject("StsTrack");
  if (!fSTSArray)
    cout << "-W- CbmEcalTrackExtrapolationKF::Init: No sts track array!"
         << endl;
}
// -------------------------------------------------------------------------
 
 
// -----   Public method DoExtrapolate   ------------------------------------------
Int_t CbmEcalTrackExtrapolationKF::DoExtrapolate(
  TClonesArray* gTrackArray,
  TClonesArray* fTrackParamArray_) {
  if (!fTrackParamArray_) {
    cout << "-E- CbmEcalTrackExtrapolationKF::DoExtrapolate: ";
    cout << "TrackParamArray missing! " << endl;
    return -1;
  }
 
  if (!gTrackArray) {
    cout << "-E- CbmEcalTrackExtrapolationKF::DoExtrapolate: ";
    cout << "Track Array missing! " << endl;
    return -1;
  }
  Int_t nTr = 0;
  Int_t i   = 0;
  Int_t n   = gTrackArray->GetEntriesFast();
  CbmGlobalTrack* tr;
  CbmKFTrack kfTr;
  FairTrackParam trpar;
  for (; i < n; i++) {
    Map()[nTr] = -1111;
    tr         = (CbmGlobalTrack*) gTrackArray->At(i);
    if (!tr) continue;
    if (tr->GetStsTrackIndex() < 0 || tr->GetTrdTrackIndex() < 0) continue;
    kfTr.SetTrackParam(*(const_cast<FairTrackParam*>(tr->GetParamLast())));
    kfTr.Extrapolate(Str()->GetEcalInf()->GetZPos());
    kfTr.GetTrackParam(trpar);
    Map()[nTr] = i;
    new ((*fTrackParamArray_)[nTr++]) FairTrackParam(trpar);
  }
 
  return nTr;
}
// -------------------------------------------------------------------------
 
 
// -----   Public method Finish   ------------------------------------------
void CbmEcalTrackExtrapolationKF::Finish() { ; }
//-----------------------------------------------------------------------------------
 
ClassImp(CbmEcalTrackExtrapolationKF)