KFParticleFinder.h

Go to the documentation of this file.

#ifndef KFParticleFinder_h
#define KFParticleFinder_h

#include "KFParticle.h"
#include "KFParticleSIMD.h"
#include "KFPTrack.h"

#include <vector>

#ifdef NonhomogeniousField
class L1FieldRegion;
#endif
class KFParticleSIMD;

class KFParticleFinder
{
 public:

  KFParticleFinder();
  ~KFParticleFinder() {};


    static void FindParticles(std::vector<KFPTrack>& vRTracks, std::vector<float>& ChiToPrimVtx,
#ifdef NonhomogeniousField
                                           std::vector<L1FieldRegion>& vField,
#endif
                               std::vector<KFParticle>& Particles,
                               KFParticleSIMD& PrimVtx, const std::vector<int>& vTrackPDG, const float cuts[2][3] = DefaultCuts);

    static void ExtrapolateToPV(std::vector<KFParticle>& vParticles, KFParticleSIMD& PrimVtx);
    static fvec GetChi2BetweenParticles(KFParticleSIMD &p1, KFParticleSIMD &p2);

    static void Find2DaughterDecay(std::vector<KFPTrack>& vTracks,
#ifdef NonhomogeniousField
                                   const std::vector<L1FieldRegion>& vField,
#endif
                                   std::vector<KFParticle>& Particles,
                                   const int DaughterNegPDG,
                                   const int DaughterPosPDG,
                                   const int MotherPDG,
                                   std::vector<short>& idNeg,
                                   std::vector<short>& idPos,
                                   KFParticleSIMD& PrimVtx,
                                   const float* cuts = 0,
                                   bool isPrimary = 0,
                                   std::vector<float>* vMotherTopoChi2Ndf = 0,
                                   const float* secCuts = 0,
                                   const float massMotherPDG = 0,
                                   const float massMotherPDGSigma = 0,
                                   std::vector<KFParticle>* vMotherPrim = 0,
                                   std::vector<KFParticle>* vMotherSec = 0);

    static void Find2DaughterDecay(std::vector<KFPTrack>& vTracks,
#ifdef NonhomogeniousField
                                   const std::vector<L1FieldRegion>& vField,
#endif
                                   std::vector<KFParticle>& Particles,
                                   const int DaughterNegPDG,
                                   const int DaughterPosPDG,
                                   const int MotherPDG,
                                   std::vector<short>& idNeg,
                                   std::vector<short>& idPos,
                                   KFParticleSIMD& PrimVtx,
                                   const float* cuts,
                                   bool isPrimary,
                                   const float PtCut,
                                   const float Chi2PrimCut = -100.f,
                                   std::vector<float>* ChiToPrimVtx = 0,
                                   const float* PCut = 0);

  static void FindTrackV0Decay(const int MotherPDG,
                               std::vector<KFParticle>& Particles,
                               std::vector<KFParticle>& vV0,
                               std::vector<KFPTrack>& vTracks,
#ifdef NonhomogeniousField
                               const std::vector<L1FieldRegion>& field,
#endif
                               const int DaughterPDG,
                               std::vector<short>& idTrack,
                               KFParticleSIMD& PrimVtx,
                               const float* cuts = 0,
                               bool isPrimary = 0,
                               std::vector<float>* ChiToPrimVtx = 0,
                               std::vector<KFParticle>* vHyperonPrim = 0,
                               float hyperonPrimMass = 0,
                               float hyperonPrimMassErr = 0,
                               std::vector<KFParticle>* vHyperonSec = 0);

  static void FindHyperons(int PDG,
                           KFParticleSIMD vDaughters[2],
                           std::vector<int>& daughterIds,
                           std::vector<KFParticle>& vLambdaSec,
                           std::vector<KFParticle>& vHyperon,
                           KFParticleSIMD& PrimVtx,
                           const float *cuts = 0,
                           int startIndex=0);

    static void FindDMesLambdac(std::vector<KFPTrack>& vTracks,
#ifdef NonhomogeniousField
                                const std::vector<L1FieldRegion>& vField,
#endif
                                std::vector<KFParticle>& Particles,
                                const int DaughterPDG[5], //pi, K_b, pi_b, K, p
                                const int MotherPDG[8],
                                std::vector<short>* idTrack[5], //pi, K_b, pi_b, K, p
                                KFParticleSIMD& PrimVtx,
                                const float cuts[8][8],
                                std::vector<float> ChiToPrimVtx);

    static void CombineTrackPart(std::vector<KFPTrack>& vTracks,
#ifdef NonhomogeniousField
                                 const std::vector<L1FieldRegion>& vField,
#endif
                                 std::vector<KFParticle>& Particles,
                                 KFParticle& part,
                                 const int DaughterPDG,
                                 const int MotherPDG,
                                 std::vector<short>& id,
                                 const float* cuts,
                                 const unsigned short startIndex = 0,
                                 const bool IsSamePart = 0);

  static void SelectParticleCandidates(std::vector<KFParticle>& Particles,
                                       std::vector<KFParticle>& vCandidates,
                                       KFParticleSIMD& PrimVtx,
                                       const float cuts[5]);

 private:

  static const float DefaultCuts[2][3];
};

#endif /* !KFParticleFinder_h */