PuppiCandidate.h

Go to the documentation of this file.

#ifndef CommonTools_PileupAlgos_PuppiCandidate
#define CommonTools_PileupAlgos_PuppiCandidate

#include "fastjet/PseudoJet.hh"

// Extension of fastjet::PseudoJet that caches eta and some other info
// Puppi uses register to decide between NHs, PV CHs, and PU CHs.
class PuppiCandidate : public fastjet::PseudoJet {
public:
  using fastjet::PseudoJet::PseudoJet;
  double pseudorapidity() const {
    _ensure_valid_eta();
    return _eta;
  }
  double eta() const { return pseudorapidity(); }
  void _ensure_valid_eta() const {
    if (_eta == fastjet::pseudojet_invalid_rap)
      _eta = fastjet::PseudoJet::pseudorapidity();
  }
  void set_info(int puppi_register) { _puppi_register = puppi_register; }
  inline int puppi_register() const { return _puppi_register; }

private:
  // variable names in fastjet style
  mutable double _eta = fastjet::pseudojet_invalid_rap;
  int _puppi_register;
};

#endif