ReconstructedParticle2MC.h

Go to the documentation of this file.

 
#ifndef  RECONSTRUCTEDPARTICLE2MC_ANALYZERS_H
#define  RECONSTRUCTEDPARTICLE2MC_ANALYZERS_H
 
#include <cmath>
#include <vector>
 
#include "ROOT/RVec.hxx"
#include "edm4hep/ReconstructedParticleData.h"
#include "edm4hep/MCParticleData.h"
#include "podio/ObjectID.h"
#include "TLorentzVector.h"
 
namespace FCCAnalyses{
 
namespace ReconstructedParticle2MC{
 
  struct selRP_PDG {
    selRP_PDG(int arg_PDG, bool arg_chargedOnly);
    int m_PDG = 13 ;
    bool m_chargedOnly = true;
    std::vector<edm4hep::ReconstructedParticleData>  operator() (ROOT::VecOps::RVec<int> recind,
                                 ROOT::VecOps::RVec<int> mcind,
                                 ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                                 ROOT::VecOps::RVec<edm4hep::MCParticleData> mc) ;
  };
 
  struct selRP_PDG_index {
    selRP_PDG_index(int arg_PDG, bool arg_chargedOnly);
    int m_PDG = 13 ;
    bool m_chargedOnly = true;
    ROOT::VecOps::RVec<int>  operator() (ROOT::VecOps::RVec<int> recind,
                     ROOT::VecOps::RVec<int> mcind,
                     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                     ROOT::VecOps::RVec<edm4hep::MCParticleData> mc) ;
  };
 
 
  struct getRP2MC_p_func {
    ROOT::VecOps::RVec<float>  operator() (ROOT::VecOps::RVec<int> recin,
                       ROOT::VecOps::RVec<int> mcin,
                       ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                       ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
  };
 
  ROOT::VecOps::RVec<float> getRP2MC_p (ROOT::VecOps::RVec<int> recin,
                    ROOT::VecOps::RVec<int> mcin,
                    ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                    ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_px (ROOT::VecOps::RVec<int> recin,
                     ROOT::VecOps::RVec<int> mcin,
                     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                     ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_py (ROOT::VecOps::RVec<int> recin,
                     ROOT::VecOps::RVec<int> mcin,
                     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                     ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_pz (ROOT::VecOps::RVec<int> recin,
                     ROOT::VecOps::RVec<int> mcin,
                     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                     ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_mass (ROOT::VecOps::RVec<int> recin,
                       ROOT::VecOps::RVec<int> mcin,
                       ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                       ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_charge (ROOT::VecOps::RVec<int> recin,
                         ROOT::VecOps::RVec<int> mcin,
                         ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                         ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<float> getRP2MC_pdg (ROOT::VecOps::RVec<int> recin,
                      ROOT::VecOps::RVec<int> mcin,
                      ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                      ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<TLorentzVector> getRP2MC_tlv (ROOT::VecOps::RVec<int> recin,
                           ROOT::VecOps::RVec<int> mcin,
                           ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                           ROOT::VecOps::RVec<edm4hep::MCParticleData> mc);
 
  ROOT::VecOps::RVec<int> getRP2MC_index (ROOT::VecOps::RVec<int> recin,
                      ROOT::VecOps::RVec<int> mcin,
                      ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco);
 
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<int>> getRP2MC_indexVec (ROOT::VecOps::RVec<int> recin,
                                 ROOT::VecOps::RVec<int> mcin,
                                 ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco);
 
  ROOT::VecOps::RVec<int> getRP2MC_index_test (ROOT::VecOps::RVec<int> recin,
                           ROOT::VecOps::RVec<int> mcin,
                           ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                           ROOT::VecOps::RVec<edm4hep::MCParticleData> mc,
                           ROOT::VecOps::RVec<int> parents);
 
 
  ROOT::VecOps::RVec<int> getRP2MC_parentid (ROOT::VecOps::RVec<int> recin,
                         ROOT::VecOps::RVec<int> mcin,
                         ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                         ROOT::VecOps::RVec<edm4hep::MCParticleData> mc,
                         ROOT::VecOps::RVec<int> parents);
 
  std::vector<edm4hep::ReconstructedParticleData> selRP_ChargedHadrons ( ROOT::VecOps::RVec<int> recind,
                                     ROOT::VecOps::RVec<int> mcind,
                                     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                                     ROOT::VecOps::RVec<edm4hep::MCParticleData> mc) ;
 
  ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> selRP_matched_to_list( ROOT::VecOps::RVec<int>   mcParticles_indices,
                                        ROOT::VecOps::RVec<int> recind,
                                        ROOT::VecOps::RVec<int> mcind,
                                        ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                                        ROOT::VecOps::RVec<edm4hep::MCParticleData> mc) ;
 
  int getTrack2MC_index (  int track_index,
               ROOT::VecOps::RVec<int> recind,
               ROOT::VecOps::RVec<int> mcind,
               ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco);
 
}//end NS ReconstructedParticle2MC
 
}//end NS FCCAnalyses
#endif