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;
};

}
#endif