JetExtendedAssociation.h

Go to the documentation of this file.

#ifndef JetReco_JetExtendedAssociation_h
#define JetReco_JetExtendedAssociation_h

#include "DataFormats/Common/interface/AssociationVector.h"
#include "DataFormats/JetReco/interface/JetCollection.h"
#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/Common/interface/RefVector.h"
#include "DataFormats/Math/interface/LorentzVector.h"

namespace fwlite {
  class Event;
}

namespace reco {
  namespace JetExtendedAssociation {
    class JetExtendedData;
    typedef math::PtEtaPhiELorentzVectorF LorentzVector;
    typedef reco::JetExtendedAssociation::JetExtendedData Value;
    typedef std::vector<Value> Values;
    typedef edm::AssociationVector<reco::JetRefBaseProd, Values> Container;
    typedef Container::value_type value_type;
    typedef Container::transient_vector_type transient_vector_type;
    typedef edm::Ref <Container> Ref;
    typedef edm::RefProd <Container> RefProd;
    typedef edm::RefVector <Container> RefVector;


    int tracksAtVertexNumber (const Container&, const reco::JetBaseRef&);
    int tracksAtVertexNumber (const Container&, const reco::Jet&);
    const LorentzVector& tracksAtVertexP4 (const Container&, const reco::JetBaseRef&);
    const LorentzVector& tracksAtVertexP4 (const Container&, const reco::Jet&);
    int tracksAtCaloNumber (const Container&, const reco::JetBaseRef&);
    int tracksAtCaloNumber (const Container&, const reco::Jet&);
    const LorentzVector& tracksAtCaloP4 (const Container&, const reco::JetBaseRef&);
    const LorentzVector& tracksAtCaloP4 (const Container&, const reco::Jet&);

    bool setValue (Container&, const reco::JetBaseRef&, const JetExtendedData&);
    bool setValue (Container*, const reco::JetBaseRef&, const JetExtendedData&);
    const JetExtendedData& getValue (const Container&, const reco::JetBaseRef&);
    const JetExtendedData& getValue (const Container&, const reco::Jet&);
    std::vector<reco::JetBaseRef > allJets (const Container&);
    bool hasJet (const Container&, const reco::JetBaseRef&);
    bool hasJet (const Container&, const reco::Jet&);

    class JetExtendedData {
    public:
      JetExtendedData ();
      ~JetExtendedData () {}
      int mTracksAtVertexNumber;
      LorentzVector mTracksAtVertexP4;
      int mTracksAtCaloNumber;
      LorentzVector mTracksAtCaloP4;
    };
  }
}

#endif