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;
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

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

void PFDisplacedVertexCandidateProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  // The track collection use for the fitting. May be any collection.
  // The only condition is that it shall contain the hit pattern information
  desc.add<edm::InputTag>("trackCollection", {"generalTracks"});
  // verbosity
  desc.addUntracked<bool>("verbose", false);
  // Debug flag
  desc.addUntracked<bool>("debug", false);
  // maximum dca distance for two tracks to be linked
  desc.add<double>("dcaCut", 0.5);
  // minimum distance of secondary vertex with respect to the primary
  desc.add<double>("primaryVertexCut", 1.8);
  // maximum distance between the DCA Point and the inner hit of the track
  // not used for the moment
  desc.add<double>("dcaPInnerHitCut", 1000.0);
  // Primary vertex information used for dxy calculation
  desc.add<edm::InputTag>("mainVertexLabel", {"offlinePrimaryVertices", ""});
  desc.add<edm::InputTag>("offlineBeamSpotLabel", {"offlineBeamSpot", ""});
  // Tracks preselection to reduce the combinatorics in PFDisplacedVertexCandidates
  // this cuts are repeated then in a smarter way in the PFDisplacedVertexFinder
  // be sure you are consistent between them.
  {
    edm::ParameterSetDescription pset;
    // selection parameters for secondary tracks
    pset.add<double>("nChi2_max", 5.);
    pset.add<double>("pt_min", 0.2);
    // if the tracks is not a good candidate to be a secondary (dxy cut) restrict in minimal pt
    // this cut reduce drastically the combinatorics. It is very useful to reduce the
    // PFDisplacedVertex timing
    pset.add<double>("pt_min_prim", 0.8);
    pset.add<double>("dxy", 0.2);
    desc.add<edm::ParameterSetDescription>("tracksSelectorParameters", pset);
  }
  descriptions.add("particleFlowDisplacedVertexCandidate", desc);
}

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