Puppi.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#include "fastjet/internal/base.hh"
#include "fastjet/PseudoJet.hh"
 
#ifndef JETRECTOOLS_PUPPI_h
#define JETRECTOOLS_PUPPI_h
 
//This class can be used to pass in otherchi2 values for each PFO
class PuppiUserInfo: public fastjet::PseudoJet::UserInfoBase{
 public:
  PuppiUserInfo();
  PuppiUserInfo(const std::vector<double>&);
  PuppiUserInfo(double);
 
  std::vector<double> otherChi2Vec;
};
 
//=================================================================================================================================
 
class Puppi{
 
 public:
  Puppi(double R0, double Rmin, double beta, double centralPTCutOffset, double centralPTCutSlope, double forwardPTCutOffset, double forwardPTCutSlope, double etaBoundary);
 
  void setParticles(const std::vector<fastjet::PseudoJet>& chargedHS, const std::vector<fastjet::PseudoJet>& chargedPU, const std::vector<fastjet::PseudoJet>& neutral, const std::vector<fastjet::PseudoJet>& forward, int nPU);
 
  // Returns the weight that would be applied to a particle (assuming it is neutral or forward)
  double getWeight(const fastjet::PseudoJet& pfo);
 
  // Returns the value of alpha for this particle
  double getAlpha(const fastjet::PseudoJet& pfo);
 
  double getMedian();
  double getRMS();
 
 private:
 
  double getChi2(const fastjet::PseudoJet& pfo);
 
  void findAlphaMedianAndRMS();
 
  std::vector<fastjet::PseudoJet> m_chargedHS;
  std::vector<fastjet::PseudoJet> m_chargedPU;
  std::vector<fastjet::PseudoJet> m_neutral;
  std::vector<fastjet::PseudoJet> m_forward;
 
  std::vector< std::vector< fastjet::PseudoJet >* > m_allParticles;
 
  double m_R0;
  double m_Rmin;
  double m_beta;
  double m_centralPTCutOffset;
  double m_centralPTCutSlope;
  double m_forwardPTCutOffset;
  double m_forwardPTCutSlope;
  double m_etaBoundary;
  bool m_includeCentralNeutralsInAlpha;
 
  double m_median;
  double m_rms;
 
  double m_nPV;
 
};
 
#endif 
//FASTJET_END_NAMESPACE