JetFlavourInfo.h

Go to the documentation of this file.

#ifndef SimDataFormats_JetMatching_JetFlavourInfo_H
#define SimDataFormats_JetMatching_JetFlavourInfo_H

#include <vector>
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"

namespace reco {
  class JetFlavourInfo {
  public:
    JetFlavourInfo(void) : m_hadronFlavour(0), m_partonFlavour(0) {}
    JetFlavourInfo(const int hadronFlavour, const int partonFlavour)
        : m_hadronFlavour(hadronFlavour), m_partonFlavour(partonFlavour) {}
    JetFlavourInfo(const GenParticleRefVector& bHadrons,
                   const GenParticleRefVector& cHadrons,
                   const GenParticleRefVector& partons,
                   const GenParticleRefVector& leptons,
                   const int hadronFlavour,
                   const int partonFlavour)
        : m_bHadrons(bHadrons),
          m_cHadrons(cHadrons),
          m_partons(partons),
          m_leptons(leptons),
          m_hadronFlavour(hadronFlavour),
          m_partonFlavour(partonFlavour) {}

    const GenParticleRefVector& getbHadrons() const { return m_bHadrons; }
    const GenParticleRefVector& getcHadrons() const { return m_cHadrons; }
    const GenParticleRefVector& getPartons() const { return m_partons; }
    const GenParticleRefVector& getLeptons() const { return m_leptons; }
    const int getHadronFlavour() const { return m_hadronFlavour; }
    const int getPartonFlavour() const { return m_partonFlavour; }

    void setHadronFlavour(const int hadronFlavour) { m_hadronFlavour = hadronFlavour; }
    void setPartonFlavour(const int partonFlavour) { m_partonFlavour = partonFlavour; }

  private:
    GenParticleRefVector m_bHadrons;
    GenParticleRefVector m_cHadrons;
    GenParticleRefVector m_partons;
    GenParticleRefVector m_leptons;
    int m_hadronFlavour;
    int m_partonFlavour;
  };

}  // namespace reco
#endif