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

}
#endif