RecoTauVertexAssociator.h

Go to the documentation of this file.

#ifndef RecoTauTag_RecoTau_RecoTauVertexAssociator_h
#define RecoTauTag_RecoTau_RecoTauVertexAssociator_h

/* RecoTauVertexAssociator
 *
 * Authors: Evan K. Friis, Christian Veelken, UC Davis
 *          Michalis Bachtis, UW Madison
 *
 * The associatedVertex member function retrieves the vertex from the event
 * associated to a given tau.  This class is configured using a cms.PSet.
 *
 * The required arguments are:
 *  o primaryVertexSrc - InputTag with the vertex collection
 *  o useClosestPV - Use the "closest to lead track in z" to find the vertex.
 *
 * The setEvent method must be called at least once per event.
 *
 */

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/EDConsumerBase.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Common/interface/Handle.h"
#include "RecoTauTag/RecoTau/interface/RecoTauQualityCuts.h"

#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include <map>

// Forward declarations
namespace edm {
  class ParameterSet;
  class Event;
}

namespace reco {
  class PFTau;
  class PFJet;
}

namespace reco { namespace tau {

class RecoTauVertexAssociator {
  public:
    enum Algorithm {
      kHighestPtInEvent,
      kClosestDeltaZ,
      kHighestWeigtForLeadTrack,
      kCombined
    };

    RecoTauVertexAssociator (const edm::ParameterSet& pset,  edm::ConsumesCollector&& iC);
    virtual ~RecoTauVertexAssociator (){}
    reco::VertexRef associatedVertex(const PFJet& tau) const;
    reco::VertexRef associatedVertex(const PFJetRef jet) const;
    reco::VertexRef associatedVertex(const PFTau& tau) const;
    void setEvent(const edm::Event& evt);
    reco::TrackBaseRef getLeadTrack(const PFJet& jet) const;
    //    std::map<const PFJet*,reco::VertexRef> Employees;
    // containers for holding vertices associated to jets
    std::map<const reco::PFJet*,reco::VertexRef> *JetToVertexAssociation;
    int  myEventNumber;


  private:
    std::vector<reco::VertexRef> vertices_;
    edm::InputTag vertexTag_;
    Algorithm algo_;
    //PJ adding quality cuts
    RecoTauQualityCuts qcuts_;
    bool recoverLeadingTrk;

    edm::EDGetTokenT<reco::VertexCollection> vx_token;

};

} /* tau */ } /* reco */

#endif /* end of include guard: RecoTauTag_RecoTau_RecoTauVertexAssociator_h */