CbmL1PFFitter.h

Go to the documentation of this file.

/*
 *=====================================================
 *
 *  CBM Level 1 Reconstruction
 *
 *  Authors: M.Zyzak
 *
 *  e-mail :
 *
 *=====================================================
 *
 *  Finds Particles: Lambdas, K0
 *
 */
 
#ifndef _CbmL1PFFitter_h_
#define _CbmL1PFFitter_h_
 
#include "CbmL1Def.h"
 
#include <vector>
 
class CbmL1Track;
class CbmStsTrack;
class L1TrackPar;
class L1UMeasurementInfo;
class L1MaterialInfo;
class L1Station;
class L1FieldRegion;
class CbmKFVertex;
 
class CbmL1PFFitter
{
  public:
    CbmL1PFFitter();
    ~CbmL1PFFitter();
 
    void AddMaterial(L1TrackPar& T, fvec radThick, fvec qp0, fvec& mass2, fvec& w);
    void AddPipeMaterial(L1TrackPar& T, fvec qp0, fvec& mass2, fvec& w);
    void FilterFirst(L1TrackPar& track, fvec& x, fvec& y, fvec& w, L1Station& st);
    void FilterLast(L1TrackPar& track, fvec& x, fvec& y, fvec& w, L1Station& st);
    void Filter(L1TrackPar& T, L1UMeasurementInfo& info, fvec& u, fvec& w);
 
    void Fit(std::vector<CbmL1Track>& Tracks, fvec mass = 0.1395679f);
 
    void CalculateFieldRegion(std::vector<CbmL1Track>& Tracks, std::vector<L1FieldRegion>& Field);
    void GetChiToVertex(std::vector<CbmL1Track>& Tracks, std::vector<float>& chiToVtx, CbmKFVertex& primVtx);
 
    // functions for fitting CbmStsTrack
    void Fit(std::vector<CbmStsTrack>& Tracks, int pidHypo = 211);
    void CalculateFieldRegion(std::vector<CbmStsTrack>& Tracks, std::vector<L1FieldRegion>& Field);
    void GetChiToVertex(std::vector<CbmStsTrack>& Tracks,
                        std::vector<L1FieldRegion>& field,
                        std::vector<float>& chiToVtx,
                        CbmKFVertex& primVtx,
                        float chiPrim = -1);
};
#endif