GenJet.h

Go to the documentation of this file.

#ifndef JetReco_GenJet_h
#define JetReco_GenJet_h

#include "DataFormats/JetReco/interface/Jet.h"

namespace reco {
  class GenParticle;

  class GenJet : public Jet {
  public:
    struct Specific {
      Specific()
          : m_EmEnergy(0),
            m_HadEnergy(0),
            m_InvisibleEnergy(0),
            m_AuxiliaryEnergy(0),
            m_ChargedHadronEnergy(0),
            m_NeutralHadronEnergy(0),
            m_ChargedEmEnergy(0),
            m_NeutralEmEnergy(0),
            m_MuonEnergy(0),
            m_ChargedHadronMultiplicity(0),
            m_NeutralHadronMultiplicity(0),
            m_ChargedEmMultiplicity(0),
            m_NeutralEmMultiplicity(0),
            m_MuonMultiplicity(0) {}

      float m_EmEnergy;
      float m_HadEnergy;
      float m_InvisibleEnergy;
      float m_AuxiliaryEnergy;

      float m_ChargedHadronEnergy;
      float m_NeutralHadronEnergy;
      float m_ChargedEmEnergy;
      float m_NeutralEmEnergy;
      float m_MuonEnergy;
      int m_ChargedHadronMultiplicity;
      int m_NeutralHadronMultiplicity;
      int m_ChargedEmMultiplicity;
      int m_NeutralEmMultiplicity;
      int m_MuonMultiplicity;
    };

    GenJet() {}

    GenJet(const LorentzVector& fP4,
           const Point& fVertex,
           const Specific& fSpecific,
           const Jet::Constituents& fConstituents);
    GenJet(const LorentzVector& fP4, const Point& fVertex, const Specific& fSpecific);

    GenJet(const LorentzVector& fP4, const Specific& fSpecific, const Jet::Constituents& fConstituents);

    ~GenJet() override{};
    float emEnergy() const { return m_specific.m_EmEnergy; };
    float hadEnergy() const { return m_specific.m_HadEnergy; };
    float invisibleEnergy() const { return m_specific.m_InvisibleEnergy; };
    float auxiliaryEnergy() const { return m_specific.m_AuxiliaryEnergy; };

    // PF-like definitions
    float chargedHadronEnergy() const { return m_specific.m_ChargedHadronEnergy; }
    float neutralHadronEnergy() const { return m_specific.m_NeutralHadronEnergy; }
    float chargedEmEnergy() const { return m_specific.m_ChargedEmEnergy; }
    float neutralEmEnergy() const { return m_specific.m_NeutralEmEnergy; }
    float muonEnergy() const { return m_specific.m_MuonEnergy; }
    int chargedHadronMultiplicity() const { return m_specific.m_ChargedHadronMultiplicity; }
    int neutralHadronMultiplicity() const { return m_specific.m_NeutralHadronMultiplicity; }
    int chargedEmMultiplicity() const { return m_specific.m_ChargedEmMultiplicity; }
    int neutralEmMultiplicity() const { return m_specific.m_NeutralEmMultiplicity; }
    int muonMultiplicity() const { return m_specific.m_MuonMultiplicity; }

    float detectorEta(float fZVertex) const;

    static const GenParticle* genParticle(const reco::Candidate* fConstituent);
    virtual const GenParticle* getGenConstituent(unsigned fIndex) const;
    virtual std::vector<const GenParticle*> getGenConstituents() const;

    // block accessors

    const Specific& getSpecific() const { return m_specific; }

    void setSpecific(const Specific& spec) { m_specific = spec; }

    GenJet* clone() const override;

    std::string print() const override;

  private:
    bool overlap(const Candidate&) const override;

    // Data members
    //Variables specific to to the GenJet class
    Specific m_specific;
  };
}  // namespace reco
// temporary fix before include_checcker runs globally
#include "DataFormats/JetReco/interface/GenJetCollection.h"  //INCLUDECHECKER:SKIP
#endif