JetFlavour.h

Go to the documentation of this file.

#ifndef SimDataFormats_JetMatching_JetFlavour_H
#define SimDataFormats_JetMatching_JetFlavour_H

#include <vector>
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/LorentzVector.h"

namespace reco {
  class JetFlavour {
  public:
    typedef math::XYZTLorentzVector LorentzVector;
    typedef math::XYZPoint Point;
    struct Leptons {
      int electron;
      int muon;
      int tau;

      Leptons() : electron(0), muon(0), tau(0) {}
    };

    JetFlavour(void) {}
    JetFlavour(const LorentzVector& lv, const Point& pt, int fl, const Leptons& le)
        : m_partonMomentum(lv), m_partonVertex(pt), m_flavour(fl), m_leptons(le) {}
    JetFlavour(const LorentzVector& lv, const Point& pt, int fl)
        : m_partonMomentum(lv), m_partonVertex(pt), m_flavour(fl) {}

    const LorentzVector getLorentzVector() const { return m_partonMomentum; }
    const Point getPartonVertex() const { return m_partonVertex; }
    const int getFlavour() const { return m_flavour; }
    const Leptons getLeptons() const { return m_leptons; }

  private:
    LorentzVector m_partonMomentum;
    Point m_partonVertex;  // is it needed?
    int m_flavour;
    Leptons m_leptons;
  };

}  // namespace reco
#endif