TrackingParticleNumberOfLayersProducer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
 
#include "SimGeneral/TrackingAnalysis/interface/TrackingParticleNumberOfLayers.h"
 
class TrackingParticleNumberOfLayersProducer : public edm::global::EDProducer<> {
public:
  TrackingParticleNumberOfLayersProducer(const edm::ParameterSet &iConfig);
 
  static void fillDescriptions(edm::ConfigurationDescriptions &descriptions);
 
  void produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override;
 
private:
  edm::EDGetTokenT<TrackingParticleCollection> tpToken_;
  std::vector<edm::EDGetTokenT<std::vector<PSimHit>>> simHitTokens_;
};
 
TrackingParticleNumberOfLayersProducer::TrackingParticleNumberOfLayersProducer(const edm::ParameterSet &iConfig)
    : tpToken_(consumes<TrackingParticleCollection>(iConfig.getParameter<edm::InputTag>("trackingParticles"))) {
  for (const auto &tag : iConfig.getParameter<std::vector<edm::InputTag>>("simHits")) {
    simHitTokens_.push_back(consumes<std::vector<PSimHit>>(tag));
  }
 
  produces<edm::ValueMap<unsigned int>>("trackerLayers");
  produces<edm::ValueMap<unsigned int>>("pixelLayers");
  produces<edm::ValueMap<unsigned int>>("stripStereoLayers");
}
 
void TrackingParticleNumberOfLayersProducer::fillDescriptions(edm::ConfigurationDescriptions &descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("trackingParticles", edm::InputTag("mix", "MergedTrackTruth"));
 
  desc.add<std::vector<edm::InputTag>>("simHits",
                                       {edm::InputTag("g4SimHits", "TrackerHitsPixelBarrelLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsPixelBarrelHighTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsPixelEndcapLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsPixelEndcapHighTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTIBLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTIBHighTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTIDLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTIDHighTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTOBLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTOBHighTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTECLowTof"),
                                        edm::InputTag("g4SimHits", "TrackerHitsTECHighTof")});
 
  descriptions.add("trackingParticleNumberOfLayersProducer", desc);
}
 
void TrackingParticleNumberOfLayersProducer::produce(edm::StreamID,
                                                     edm::Event &iEvent,
                                                     const edm::EventSetup &iSetup) const {
  edm::Handle<TrackingParticleCollection> htps;
  iEvent.getByToken(tpToken_, htps);
 
  TrackingParticleNumberOfLayers algo(iEvent, simHitTokens_);
  auto ret = algo.calculate(htps, iSetup);
  iEvent.put(std::move(std::get<TrackingParticleNumberOfLayers::nTrackerLayers>(ret)), "trackerLayers");
  iEvent.put(std::move(std::get<TrackingParticleNumberOfLayers::nPixelLayers>(ret)), "pixelLayers");
  iEvent.put(std::move(std::get<TrackingParticleNumberOfLayers::nStripMonoAndStereoLayers>(ret)), "stripStereoLayers");
}
 
DEFINE_FWK_MODULE(TrackingParticleNumberOfLayersProducer);