JetClusteringUtils.h

Go to the documentation of this file.

 
#ifndef JETCLUSTERINGUTILS_ANALYZERS_H
#define JETCLUSTERINGUTILS_ANALYZERS_H
 
#include "Math/Vector4D.h"
#include "ROOT/RVec.hxx"
 
#include "FastJet/JetClustering.h"
#include "fastjet/JetDefinition.hh"
 
#include "TRandom3.h"
 
namespace FCCAnalyses {
  namespace JetClusteringUtils {
    const int Nmax_dmerge = 10;
 
    std::vector<fastjet::PseudoJet> set_pseudoJets(const ROOT::VecOps::RVec<float>& px,
                                                   const ROOT::VecOps::RVec<float>& py,
                                                   const ROOT::VecOps::RVec<float>& pz,
                                                   const ROOT::VecOps::RVec<float>& e);
 
    std::vector<fastjet::PseudoJet> set_pseudoJets_xyzm(const ROOT::VecOps::RVec<float>& px,
                                                        const ROOT::VecOps::RVec<float>& py,
                                                        const ROOT::VecOps::RVec<float>& pz,
                                                        const ROOT::VecOps::RVec<float>& m);
 
    ROOT::VecOps::RVec<fastjet::PseudoJet> get_pseudoJets(const JetClustering::FCCAnalysesJet& in);
 
    std::vector<std::vector<int>> get_constituents(const JetClustering::FCCAnalysesJet& in);
 
    float get_exclusive_dmerge(const JetClustering::FCCAnalysesJet& in, int n);
 
    float get_exclusive_dmerge_max(const JetClustering::FCCAnalysesJet& in, int n);
 
    ROOT::VecOps::RVec<float> get_px(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_py(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_pz(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_e(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_pt(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_p(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_m(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_eta(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_phi(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    ROOT::VecOps::RVec<float> get_phi_std(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
 
    ROOT::VecOps::RVec<float> get_theta(const ROOT::VecOps::RVec<fastjet::PseudoJet>& in);
 
    struct sel_pt {
      sel_pt(float arg_min_pt);
      float m_min_pt = 1.; //> transverse momentum threshold [GeV]
      ROOT::VecOps::RVec<fastjet::PseudoJet>  operator() (ROOT::VecOps::RVec<fastjet::PseudoJet> in);
    };
 
    JetClustering::FCCAnalysesJet initialise_FCCAnalysesJet();
 
    JetClustering::FCCAnalysesJet build_FCCAnalysesJet(const std::vector<fastjet::PseudoJet>& in,
                                                       const std::vector<float>& dmerge,
                                                       const std::vector<float>& dmerge_max);
 
    std::vector<fastjet::PseudoJet> build_jets(fastjet::ClusterSequence& cs, int exclusive, float cut, int sorted);
 
    bool check(unsigned int n, int exclusive, float cut);
 
    fastjet::RecombinationScheme recomb_scheme(int recombination);
 
    std::vector<float> exclusive_dmerge(fastjet::ClusterSequence& cs, int do_dmarge_max);
 
    // build the resonance from 2 <fastjet::PseudoJet> objects. Keep the closest to the mass given as input
    struct resonanceBuilder {
      float m_resonance_mass;
      resonanceBuilder(float arg_resonance_mass);
      ROOT::VecOps::RVec<fastjet::PseudoJet> operator()(ROOT::VecOps::RVec<fastjet::PseudoJet> legs);
    };
 
    struct recoilBuilder {
      recoilBuilder(float arg_sqrts);
      float m_sqrts = 240.0;
      double operator() (ROOT::VecOps::RVec<fastjet::PseudoJet> in);
    };
  }  // namespace JetClusteringUtils
}  // namespace FCCAnalyses
 
#endif