Particle.hh

Go to the documentation of this file.

 
#pragma once
 
#include "fastjet/PseudoJet.hh"
 
#include "HEJ/PDG_codes.hh"
 
namespace HEJ {
 
  struct Particle {
    ParticleID type;
    fastjet::PseudoJet p;
 
    double rapidity() const{
      return p.rapidity();
    }
    double perp() const{
      return p.perp();
    }
    double px() const{
      return p.px();
    }
    double py() const{
      return p.py();
    }
    double pz() const{
      return p.pz();
    }
    double E() const{
      return p.E();
    }
    double m() const{
      return p.m();
    }
  };
 
 
  struct rapidity_less{
    template<class FourVector>
    bool operator()(FourVector const & p1, FourVector const & p2){
      return p1.rapidity() < p2.rapidity();
    }
  };
 
 
  struct pz_less{
    template<class FourVector>
    bool operator()(FourVector const & p1, FourVector const & p2){
      return p1.pz() < p2.pz();
    }
  };
 
 
  inline
  std::vector<fastjet::PseudoJet> to_PseudoJet(
      std::vector<Particle> const & v
  ){
    std::vector<fastjet::PseudoJet> result;
    for(auto && sp: v) result.emplace_back(sp.p);
    return result;
  }
 
  inline
  bool is_parton(Particle const & p){
    return is_parton(p.type);
  }
 
  inline bool is_AWZH_boson(Particle const & particle){
    return is_AWZH_boson(particle.type);
  }
 
  inline
  std::vector<Particle> filter_partons(
      std::vector<Particle> const & v
  ){
    std::vector<Particle> result;
    result.reserve(v.size());
    std::copy_if(
        begin(v), end(v), std::back_inserter(result),
        [](Particle const & p){ return is_parton(p); }
    );
    return result;
  }
}