CbmPVFinderKF.cxx

Go to the documentation of this file.

#include "CbmPVFinderKF.h"
 
#include "CbmKFPrimaryVertexFinder.h"
#include "CbmKFTrack.h"
#include "CbmKFVertex.h"
#include "CbmStsTrack.h"
#include "TClonesArray.h"
#include "math.h"
 
Int_t CbmPVFinderKF::FindPrimaryVertex(TClonesArray* tracks, CbmVertex* vertex)
{
 
    Int_t NTracks = tracks->GetEntries();
 
    CbmKFPrimaryVertexFinder Finder;
    CbmKFTrack* CloneArray = new CbmKFTrack[NTracks];
    for (Int_t i = 0; i < NTracks; i++) {
        CbmStsTrack* st = (CbmStsTrack*)tracks->At(i);
        Int_t NHits = st->GetNStsHits();
        if (NHits < 4)
            continue;
        if (st->GetFlag())
            continue;
        if (st->GetChi2() < 0. || st->GetChi2() > 3.5 * 3.5 * st->GetNDF())
            continue;
        CbmKFTrack& T = CloneArray[i];
        T.SetStsTrack(*st);
        if (!finite(T.GetTrack()[0]) || !finite(T.GetCovMatrix()[0]))
            continue;
        Finder.AddTrack(&T);
    }
    CbmKFVertex v;
    Finder.Fit(v);
    v.GetVertex(*vertex);
    delete[] CloneArray;
    return 0;
}