TrackAssociatorEDProducer.cc

Go to the documentation of this file.

//
// Original Author:  Stefano Magni
//         Created:  Fri Mar  9 10:52:11 CET 2007
//
//
 
// system include files
#include <memory>
#include <string>
 
// user include files
#include "FWCore/Framework/interface/global/EDProducer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
 
#include "FWCore/Framework/interface/ESHandle.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "SimDataFormats/Associations/interface/TrackToTrackingParticleAssociator.h"
 
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
 
#include "FWCore/Utilities/interface/EDGetToken.h"
 
//
// class decleration
//
 
class TrackAssociatorEDProducer : public edm::global::EDProducer<> {
public:
  explicit TrackAssociatorEDProducer(const edm::ParameterSet &);
  ~TrackAssociatorEDProducer() override;
 
private:
  void produce(edm::StreamID, edm::Event &, const edm::EventSetup &) const override;
 
  bool theIgnoremissingtrackcollection;
 
  edm::EDGetTokenT<TrackingParticleCollection> TPCollectionToken_;
  edm::EDGetTokenT<edm::View<reco::Track>> trackCollectionToken_;
  edm::EDGetTokenT<reco::TrackToTrackingParticleAssociator> associatorToken_;
};
 
TrackAssociatorEDProducer::TrackAssociatorEDProducer(const edm::ParameterSet &pset)
    : theIgnoremissingtrackcollection(pset.getUntrackedParameter<bool>("ignoremissingtrackcollection", false)) {
  produces<reco::SimToRecoCollection>();
  produces<reco::RecoToSimCollection>();
 
  TPCollectionToken_ = consumes<TrackingParticleCollection>(pset.getParameter<edm::InputTag>("label_tp"));
  trackCollectionToken_ = consumes<edm::View<reco::Track>>(pset.getParameter<edm::InputTag>("label_tr"));
  associatorToken_ = consumes<reco::TrackToTrackingParticleAssociator>(pset.getParameter<edm::InputTag>("associator"));
}
 
TrackAssociatorEDProducer::~TrackAssociatorEDProducer() {}
 
//
// member functions
//
 
// ------------ method called to produce the data  ------------
void TrackAssociatorEDProducer::produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const {
  using namespace edm;
 
  edm::Handle<reco::TrackToTrackingParticleAssociator> theAssociator;
  iEvent.getByToken(associatorToken_, theAssociator);
 
  Handle<TrackingParticleCollection> TPCollection;
  iEvent.getByToken(TPCollectionToken_, TPCollection);
 
  Handle<edm::View<reco::Track>> trackCollection;
  bool trackAvailable = iEvent.getByToken(trackCollectionToken_, trackCollection);
 
  std::unique_ptr<reco::RecoToSimCollection> rts;
  std::unique_ptr<reco::SimToRecoCollection> str;
 
  if (theIgnoremissingtrackcollection && !trackAvailable) {
    // the track collection is not in the event and we're being told to ignore
    // this. do not output anything to the event, other wise this would be
    // considered as inefficiency.
  } else {
    // associate tracks
    LogTrace("TrackValidator") << "Calling associateRecoToSim method"
                               << "\n";
    reco::RecoToSimCollection recSimColl = theAssociator->associateRecoToSim(trackCollection, TPCollection);
 
    LogTrace("TrackValidator") << "Calling associateSimToReco method"
                               << "\n";
    reco::SimToRecoCollection simRecColl = theAssociator->associateSimToReco(trackCollection, TPCollection);
 
    rts = std::make_unique<reco::RecoToSimCollection>(recSimColl);
    str = std::make_unique<reco::SimToRecoCollection>(simRecColl);
 
    iEvent.put(std::move(rts));
    iEvent.put(std::move(str));
  }
}
 
// define this as a plug-in
DEFINE_FWK_MODULE(TrackAssociatorEDProducer);