MCParticle.h

Go to the documentation of this file.

 
#ifndef  MCPARTICLE_ANALYZERS_H
#define  MCPARTICLE_ANALYZERS_H
 
#include <cmath>
#include <vector>
 
#include "TLorentzVector.h"
#include "ROOT/RVec.hxx"
#include "edm4hep/MCParticleData.h"
#include "edm4hep/ParticleIDData.h"
#include "edm4hep/Vector3f.h"
#include "edm4hep/Vector3d.h"
#include "edm4hep/Vector2i.h"
 
 
namespace FCCAnalyses{
 
namespace MCParticle{
 
  struct filter_pdgID {
    filter_pdgID(int arg_pdgid, bool arg_abs);
    float m_pdgid; //> Generator pdgid
    bool m_abs;//> Use absolute value for pdgig
    bool  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
  struct sel_pt {
    sel_pt(float arg_min_pt);
    float m_min_pt = 20; //> transverse momentum threshold [GeV]
    ROOT::VecOps::RVec<edm4hep::MCParticleData>  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
  struct sel_genStatus {
    sel_genStatus(int arg_status);
    float m_status = 1; //> Generator status
    ROOT::VecOps::RVec<edm4hep::MCParticleData>  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
  struct sel_pdgID {
    sel_pdgID(int arg_pdg, bool arg_chargeconjugate);
    int m_pdg = 13;
    bool m_chargeconjugate = true;
    ROOT::VecOps::RVec<edm4hep::MCParticleData>  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
  struct get_tree{
    get_tree(int arg_index);
    float m_index; //> MC Particle index to build the tree from
    ROOT::VecOps::RVec<int> operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in, ROOT::VecOps::RVec<int> ind);
  };
 
  struct get_decay {
    get_decay(int arg_mother, int arg_daughters, bool arg_inf);
    int m_mother = 0; //> mother pdg id
    int m_daughters = 0;//> daughters pdg id
    bool m_inf = false;//> boolean to check if the pdgid is below a value rather than equal
    bool  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in, ROOT::VecOps::RVec<int> ind);
  };
 
  struct get_EventPrimaryVertex {
    get_EventPrimaryVertex( int arg_genstatus  );
    int m_genstatus = 21;   // Pythia8  code of the incoming particles of the hardest subprocess
    TVector3  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
  struct get_EventPrimaryVertexP4 {
    get_EventPrimaryVertexP4();
    int m_genstatus = 21;   // Pythia8  code of the incoming particles of the hardest subprocess
    TLorentzVector  operator() (ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
  };
 
 
  struct get_indices{
    get_indices( int pdg_mother, std::vector<int> pdg_daughters, bool stableDaughters, bool chargeConjugateMother, bool chargeConjugateDaughters, bool inclusiveDecay ) ;
    int m_pdg_mother;
    std::vector<int> m_pdg_daughters;
    bool m_stableDaughters;
    bool m_chargeConjugateMother;
    bool m_chargeConjugateDaughters;
    bool m_inclusiveDecay;
    ROOT::VecOps::RVec<int>   operator() ( ROOT::VecOps::RVec<edm4hep::MCParticleData> in , ROOT::VecOps::RVec<int> ind);
  };
 
  struct get_indices_ExclusiveDecay: get_indices{
     get_indices_ExclusiveDecay( int pdg_mother, std::vector<int> pdg_daughters, bool stableDaughters, bool chargeConjugate ) ;
  };
 
  ROOT::VecOps::RVec<int>  get_indices_MotherByIndex( int imother,
                              std::vector<int> m_pdg_daughters,
                              bool m_stableDaughters,
                              bool m_chargeConjugateDaughters,
                              bool m_inclusiveDecay,
                              ROOT::VecOps::RVec<edm4hep::MCParticleData> in ,
                              ROOT::VecOps::RVec<int> ind);
 
   ROOT::VecOps::RVec<int>  get_indices_ExclusiveDecay_MotherByIndex( int imother,
                                     std::vector<int> m_pdg_daughters,
                                     bool m_stableDaughters,
                                                         ROOT::VecOps::RVec<edm4hep::MCParticleData> in ,
                                     ROOT::VecOps::RVec<int> ind);
 
 
  ROOT::VecOps::RVec<int> get_parentid(ROOT::VecOps::RVec<int> mcind, ROOT::VecOps::RVec<edm4hep::MCParticleData> mc, ROOT::VecOps::RVec<int> parents);
 
  ROOT::VecOps::RVec<float> get_time(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_pdg(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_genStatus(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_simStatus(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<edm4hep::Vector3d> get_vertex(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_vertex_x(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_vertex_y(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_vertex_z(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<edm4hep::Vector3d> get_endPoint(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<edm4hep::Vector3d> get_endPoint(ROOT::VecOps::RVec<edm4hep::MCParticleData> in, ROOT::VecOps::RVec<int> ind );
 
  ROOT::VecOps::RVec<float> get_endPoint_x(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_endPoint_y(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_endPoint_z(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_pt(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_p(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_px(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_py(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_pz(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_eta(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_y(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_theta(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_phi(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_e(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_mass(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> get_charge(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<TLorentzVector> get_tlv(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<edm4hep::MCParticleData> mergeParticles(ROOT::VecOps::RVec<edm4hep::MCParticleData> x, ROOT::VecOps::RVec<edm4hep::MCParticleData> y);
 
  int get_n(ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  ROOT::VecOps::RVec<float> AngleBetweenTwoMCParticles( ROOT::VecOps::RVec<edm4hep::MCParticleData> p1, ROOT::VecOps::RVec<edm4hep::MCParticleData> p2 );
 
  std::vector<int> get_list_of_stable_particles_from_decay( int i, ROOT::VecOps::RVec<edm4hep::MCParticleData> in, ROOT::VecOps::RVec<int> ind) ;
 
  std::vector<int> get_list_of_particles_from_decay( int i,
                         ROOT::VecOps::RVec<edm4hep::MCParticleData> in,
                         ROOT::VecOps::RVec<int> ind);
 
  edm4hep::MCParticleData sel_byIndex( int idx, ROOT::VecOps::RVec<edm4hep::MCParticleData> in);
 
  std::vector<int> list_of_stable_particles_from_decay( int i, ROOT::VecOps::RVec<edm4hep::MCParticleData> in, ROOT::VecOps::RVec<int> ind) ;
  std::vector<int> list_of_particles_from_decay( int i,
                                                 ROOT::VecOps::RVec<edm4hep::MCParticleData> in,
                                                 ROOT::VecOps::RVec<int> ind);
 
 
  int get_lepton_origin(int idx,
                        const ROOT::VecOps::RVec<edm4hep::MCParticleData> &in,
                        const ROOT::VecOps::RVec<int> &ind);
 
  int get_lepton_origin(const edm4hep::MCParticleData &p,
                        const ROOT::VecOps::RVec<edm4hep::MCParticleData> &in,
                        const ROOT::VecOps::RVec<int> &ind);
 
  ROOT::VecOps::RVec<int> get_leptons_origin(const ROOT::VecOps::RVec<edm4hep::MCParticleData> &particles,
                                             const ROOT::VecOps::RVec<edm4hep::MCParticleData> &in,
                                             const ROOT::VecOps::RVec<int> &ind);
 
 
}//end NS MCParticle
 
}//end NS FCCAnalyses
#endif