CbmEcalTrackExtrapolationKF.cxx

Go to the documentation of this file.

 
#include "CbmEcalTrackExtrapolationKF.h"
 
#include "CbmEcalInf.h"
#include "CbmEcalStructure.h"
#include "CbmGlobalTrack.h"
#include "CbmKFTrack.h"
#include "CbmMCTrack.h"
#include "CbmStsKFTrackFitter.h"
#include "CbmStsTrack.h"
#include "FairRootManager.h"
#include "FairTrackParam.h"
#include "TClonesArray.h"
 
#include <iostream>
 
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()
{
    ;
}
//-----------------------------------------------------------------------------------