PFDisplacedVertexCandidateProducer.cc

Go to the documentation of this file.

 
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "RecoParticleFlow/PFTracking/interface/PFDisplacedVertexCandidateFinder.h"
 
class PFDisplacedVertexCandidateProducer : public edm::stream::EDProducer<> {
public:
  explicit PFDisplacedVertexCandidateProducer(const edm::ParameterSet&);
 
  ~PFDisplacedVertexCandidateProducer() override;
 
  void produce(edm::Event&, const edm::EventSetup&) override;
 
private:
  edm::EDGetTokenT<reco::TrackCollection> inputTagTracks_;
 
  edm::EDGetTokenT<reco::VertexCollection> inputTagMainVertex_;
  edm::EDGetTokenT<reco::BeamSpot> inputTagBeamSpot_;
 
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> magneticFieldToken_;
 
  bool verbose_;
 
  PFDisplacedVertexCandidateFinder pfDisplacedVertexCandidateFinder_;
};
 
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(PFDisplacedVertexCandidateProducer);
 
using namespace std;
using namespace edm;
 
PFDisplacedVertexCandidateProducer::PFDisplacedVertexCandidateProducer(const edm::ParameterSet& iConfig)
    : magneticFieldToken_(esConsumes()) {
  // --- Setup input collection names --- //
  inputTagTracks_ = consumes<reco::TrackCollection>(iConfig.getParameter<InputTag>("trackCollection"));
 
  inputTagMainVertex_ = consumes<reco::VertexCollection>(iConfig.getParameter<InputTag>("mainVertexLabel"));
 
  inputTagBeamSpot_ = consumes<reco::BeamSpot>(iConfig.getParameter<InputTag>("offlineBeamSpotLabel"));
 
  verbose_ = iConfig.getUntrackedParameter<bool>("verbose");
 
  bool debug = iConfig.getUntrackedParameter<bool>("debug");
 
  // ------ Algo Parameters ------ //
 
  // Distance of minimal approach below which
  // two tracks are considered as linked together
  double dcaCut = iConfig.getParameter<double>("dcaCut");
 
  // Do not reconstruct vertices wich are
  // too close to the beam pipe
  double primaryVertexCut = iConfig.getParameter<double>("primaryVertexCut");
 
  //maximum distance between the DCA Point and the inner hit of the track
  double dcaPInnerHitCut = iConfig.getParameter<double>("dcaPInnerHitCut");
 
  edm::ParameterSet ps_trk = iConfig.getParameter<edm::ParameterSet>("tracksSelectorParameters");
 
  // Collection to be produced
  produces<reco::PFDisplacedVertexCandidateCollection>();
 
  // Vertex Finder parameters  -----------------------------------
  pfDisplacedVertexCandidateFinder_.setDebug(debug);
  pfDisplacedVertexCandidateFinder_.setParameters(dcaCut, primaryVertexCut, dcaPInnerHitCut, ps_trk);
}
 
PFDisplacedVertexCandidateProducer::~PFDisplacedVertexCandidateProducer() {}
 
void PFDisplacedVertexCandidateProducer::produce(Event& iEvent, const EventSetup& iSetup) {
  LogDebug("PFDisplacedVertexCandidateProducer")
      << "START event: " << iEvent.id().event() << " in run " << iEvent.id().run() << endl;
 
  // Prepare and fill useful event information for the Finder
  auto const& theMagField = &iSetup.getData(magneticFieldToken_);
 
  Handle<reco::TrackCollection> trackCollection;
  iEvent.getByToken(inputTagTracks_, trackCollection);
 
  Handle<reco::VertexCollection> mainVertexHandle;
  iEvent.getByToken(inputTagMainVertex_, mainVertexHandle);
 
  Handle<reco::BeamSpot> beamSpotHandle;
  iEvent.getByToken(inputTagBeamSpot_, beamSpotHandle);
 
  pfDisplacedVertexCandidateFinder_.setPrimaryVertex(mainVertexHandle, beamSpotHandle);
  pfDisplacedVertexCandidateFinder_.setInput(trackCollection, theMagField);
 
  // Run the finder
  pfDisplacedVertexCandidateFinder_.findDisplacedVertexCandidates();
 
  if (verbose_) {
    ostringstream str;
    str << pfDisplacedVertexCandidateFinder_ << endl;
    cout << pfDisplacedVertexCandidateFinder_ << endl;
    LogInfo("PFDisplacedVertexCandidateProducer") << str.str() << endl;
  }
 
  std::unique_ptr<reco::PFDisplacedVertexCandidateCollection> pOutputDisplacedVertexCandidateCollection(
      pfDisplacedVertexCandidateFinder_.transferVertexCandidates());
 
  iEvent.put(std::move(pOutputDisplacedVertexCandidateCollection));
 
  LogDebug("PFDisplacedVertexCandidateProducer")
      << "STOP event: " << iEvent.id().event() << " in run " << iEvent.id().run() << endl;
}