VertexFinderLCFIPlus.h

Go to the documentation of this file.

#ifndef  VERTEXFINDERLCFIPLUS_ANALYZERS_H
#define  VERTEXFINDERLCFIPLUS_ANALYZERS_H
 
#include <cmath>
#include <vector>
 
#include "ROOT/RVec.hxx"
#include "edm4hep/ReconstructedParticleData.h"
#include "edm4hep/TrackState.h"
 
#include "FCCAnalyses/ReconstructedParticle2Track.h"
#include "FCCAnalyses/VertexFitterSimple.h"
#include "FCCAnalyses/VertexingUtils.h"
 
#include "fastjet/JetDefinition.hh"
 
namespace FCCAnalyses{
namespace VertexFinderLCFIPlus{
 
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<VertexingUtils::FCCAnalysesVertex>> get_SV_jets( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> recoparticles,
                                             ROOT::VecOps::RVec<edm4hep::TrackState> thetracks,
                                             VertexingUtils::FCCAnalysesVertex PV,
                                             ROOT::VecOps::RVec<bool> isInPrimary,
                                             ROOT::VecOps::RVec<fastjet::PseudoJet> jets,
                                             std::vector<std::vector<int>> jet_consti,
                                             bool V0_rej=true,
                                             double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5. ) ;
  
  ROOT::VecOps::RVec<VertexingUtils::FCCAnalysesVertex> get_SV_event( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> recoparticles,
                                      ROOT::VecOps::RVec<edm4hep::TrackState> thetracks,
                                      VertexingUtils::FCCAnalysesVertex PV,
                                      ROOT::VecOps::RVec<bool> isInPrimary,
                                      bool V0_rej=true,
                                      double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5. ) ;
 
  ROOT::VecOps::RVec<VertexingUtils::FCCAnalysesVertex> get_SV_event( ROOT::VecOps::RVec<edm4hep::TrackState> np_tracks,
                                                                      ROOT::VecOps::RVec<edm4hep::TrackState> thetracks,
                                      VertexingUtils::FCCAnalysesVertex PV,
                                      bool V0_rej=true,
                                      double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5. ) ;
  
  ROOT::VecOps::RVec<int> VertexSeed_best( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
                       VertexingUtils::FCCAnalysesVertex PV,
                       double chi2_cut=9., double invM_cut=10.) ;
 
  ROOT::VecOps::RVec<int> addTrack_best( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
                     ROOT::VecOps::RVec<int> vtx_tr,
                     VertexingUtils::FCCAnalysesVertex PV,
                     double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5.) ;
 
  ROOT::VecOps::RVec<edm4hep::TrackState> V0rejection_tight( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
                                 VertexingUtils::FCCAnalysesVertex PV,
                                 bool V0_rej=true ) ;
 
  ROOT::VecOps::RVec<VertexingUtils::FCCAnalysesVertex> findSVfromTracks( ROOT::VecOps::RVec<edm4hep::TrackState> tracks_fin,
                                                                          const ROOT::VecOps::RVec<edm4hep::TrackState>& alltracks,
                                      VertexingUtils::FCCAnalysesVertex PV,
                                      double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5.) ;
 
  bool check_constraints( VertexingUtils::FCCAnalysesVertex vtx,
              ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
              VertexingUtils::FCCAnalysesVertex PV,
              bool seed=true,
              double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5.) ;
  
  ROOT::VecOps::RVec<bool> isV0( ROOT::VecOps::RVec<edm4hep::TrackState> np_tracks,
                 VertexingUtils::FCCAnalysesVertex PV,
                 bool tight = false ) ;
 
  
  VertexingUtils::FCCAnalysesV0 get_V0s( ROOT::VecOps::RVec<edm4hep::TrackState> np_tracks,
                     VertexingUtils::FCCAnalysesVertex PV,
                     bool tight,
                     double chi2_cut=9. ) ;
 
  VertexingUtils::FCCAnalysesV0 get_V0s( ROOT::VecOps::RVec<edm4hep::TrackState> np_tracks,
                     VertexingUtils::FCCAnalysesVertex PV,
                     double Ks_invM_low=0.493, double Ks_invM_high=0.503, double Ks_dis=0.5, double Ks_cosAng=0.999,
                     double Lambda_invM_low=1.111, double Lambda_invM_high=1.121, double Lambda_dis=0.5, double Lambda_cosAng=0.99995,
                     double Gamma_invM_low=0., double Gamma_invM_high=0.005, double Gamma_dis=9, double Gamma_cosAng=0.99995,
                     double chi2_cut=9. ) ;
 
  VertexingUtils::FCCAnalysesV0 get_V0s_jet( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> recoparticles,
                         ROOT::VecOps::RVec<edm4hep::TrackState> thetracks,
                         ROOT::VecOps::RVec<bool> isInPrimary,
                         ROOT::VecOps::RVec<fastjet::PseudoJet> jets,
                         std::vector<std::vector<int>> jet_consti,
                         VertexingUtils::FCCAnalysesVertex PV,
                         bool tight = true,
                         double chi2_cut=9. );
 
  ROOT::VecOps::RVec<double> get_V0candidate( VertexingUtils::FCCAnalysesVertex &V0_vtx,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tr_pair,
                          VertexingUtils::FCCAnalysesVertex PV,
                          bool chi2,
                          double chi2_cut=9. );
 
  ROOT::VecOps::RVec<double> constraints_Ks(bool tight) ;
  ROOT::VecOps::RVec<double> constraints_Lambda0(bool tight) ;
  ROOT::VecOps::RVec<double> constraints_Gamma(bool tight) ;
  //
  ROOT::VecOps::RVec<double> constraints_Ks(double invM_low, double invM_high, double dis, double cosAng) ;
  ROOT::VecOps::RVec<double> constraints_Lambda0(double invM_low, double invM_high, double dis, double cosAng) ;
  ROOT::VecOps::RVec<double> constraints_Gamma(double invM_low, double invM_high, double dis, double cosAng) ;
  //ROOT::VecOps::RVec<ROOT::VecOps::RVec<int>> VertexSeed_all( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
  //                                    VertexingUtils::FCCAnalysesVertex PV,
  //                                    double chi2_cut=9., double invM_cut=10.) ;
 
  //ROOT::VecOps::RVec<int> addTrack_multi( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
  //                        ROOT::VecOps::RVec<int> vtx_tr,
  //                        VertexingUtils::FCCAnalysesVertex PV,
  //                        double chi2_cut=9., double invM_cut=10., double chi2Tr_cut=5.) ;
 
 
}//end NS VertexFinderLCFIPlus
 
}//end NS FCCAnalyses
#endif