KFParticleFinder.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#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 = nullptr, bool isPrimary = 0,
                                 std::vector<float> *vMotherTopoChi2Ndf = nullptr, const float *secCuts = nullptr, const float massMotherPDG = 0,
                                 const float massMotherPDGSigma = 0, std::vector<KFParticle> *vMotherPrim = nullptr, std::vector<KFParticle> *vMotherSec = nullptr);

  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 = nullptr, const float *PCut = nullptr);

  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 = nullptr, bool isPrimary = 0,
                               std::vector<float> *ChiToPrimVtx = nullptr, std::vector<KFParticle> *vHyperonPrim = nullptr, float hyperonPrimMass = 0, float hyperonPrimMassErr = 0,
                               std::vector<KFParticle> *vHyperonSec = nullptr);

  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 = nullptr, 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 */