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 "CommonTools/Utils/interface/StringCutObjectSelector.h"

#include "DataFormats/Provenance/interface/RunLumiEventNumber.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& jet) const;
    reco::VertexRef associatedVertex(const PFTau& tau, bool useJet=false) const;
    reco::VertexRef associatedVertex(const TrackBaseRef& track) const;

    void setEvent(const edm::Event& evt);
    reco::TrackBaseRef getLeadTrack(const PFJet& jet) const;

  private:
    edm::InputTag vertexTag_;
    bool vxTrkFiltering_;
    StringCutObjectSelector<reco::Vertex>* vertexSelector_;
    std::vector<reco::VertexRef> selectedVertices_;
    std::string algorithm_;
    Algorithm algo_;
    //PJ adding quality cuts
    RecoTauQualityCuts* qcuts_;
    bool recoverLeadingTrk_;
    enum { kLeadTrack, kLeadPFCand, kMinLeadTrackOrPFCand, kFirstTrack };
    int leadingTrkOrPFCandOption_;
    edm::EDGetTokenT<reco::VertexCollection> vxToken_;
    // containers for holding vertices associated to jets
    std::map<const reco::PFJet*, reco::VertexRef>* jetToVertexAssociation_;
    edm::EventNumber_t lastEvent_;    
    int verbosity_;
};

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

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