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