UserHooksUtils.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PYTHIA8_USERHOOKS_USERHOOKSUTILS_H
#define PYTHIA8_USERHOOKS_USERHOOKSUTILS_H
#include "Pythia8/Event.h"
#include <stdexcept>
 
namespace Pythia8_UserHooks{
  
  inline double pTProj(const Pythia8::Particle &leg, const Pythia8::Particle &comparison){
    return leg.px()*comparison.px() + leg.py()*comparison.py() + leg.pz()*comparison.pz();
  }
  
  inline double pTProj(size_t legIndex, size_t comparedIndex, const Pythia8::Event &evt){
    return pTProj(evt[legIndex], evt[comparedIndex]);
  }
  
  inline double pT2Leg(const Pythia8::Particle &leg, const Pythia8::Particle &comparison){
    double proj = pTProj(leg, comparison);
    return leg.pAbs2() - (proj*proj) / comparison.pAbs2();
  }
  
  inline double pT2Leg(size_t legIndex, size_t comparedIndex, const Pythia8::Event &evt){
    return pT2Leg(evt[legIndex], evt[comparedIndex]);
  }
  
  inline double pTLeg(const Pythia8::Particle &leg, const Pythia8::Particle &comparison){
    return std::sqrt(pT2Leg(leg, comparison));
  }
  
  inline double pTLeg(size_t legIndex, size_t comparedIndex, const Pythia8::Event &evt){
    return std::sqrt(pT2Leg(legIndex, comparedIndex, evt));
  }
  
  inline size_t findLastEmission(const Pythia8::Event &evt, int status){
    size_t emission = evt.size() - 1;
    
    while(emission != 0){
      if (evt[emission].isFinal() && evt[emission].status() == status) return emission;
      
      --emission;
    }
    
    return 0;
  }
  
  inline size_t findLastISREmission(const Pythia8::Event &evt){
    size_t emission = findLastEmission(evt, 43);
    if(emission == 0) throw std::runtime_error("findLastISREmission:: Could not find ISR emission");
    return emission;
  }
  
  inline size_t findLastFSREmission(const Pythia8::Event &evt){
    size_t emission = findLastEmission(evt, 51);
    if(emission == 0) throw std::runtime_error("findLastFSREmission:: Could not find FSR emission");
    return emission;
  }
  
  inline size_t findLastISRRadiator(const Pythia8::Event &evt){
    size_t radiator = findLastEmission(evt, -41);
    if(radiator == 0) throw std::runtime_error("findLastISRRadiator:: Could not find ISR radiator");
    return radiator;
  }
  
  inline size_t findLastFSRRadiator(const Pythia8::Event &evt){
    size_t emitted = findLastFSREmission(evt);
    return evt[emitted].mother1();
  }
  
}
 
#endif