migrate CaloTruthCellsProducer to use esConsumes

cms-sw · Sep 29, 2021 · 97ceaf1 · 97ceaf1
1 parent 0344ed8
commit 97ceaf1
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/L1Trigger/L1THGCalUtilities/plugins/CaloTruthCellsProducer.cc b/L1Trigger/L1THGCalUtilities/plugins/CaloTruthCellsProducer.cc
@@ -31,6 +31,7 @@ class CaloTruthCellsProducer : public edm::stream::EDProducer<> {
   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
 private:
+  void beginRun(const edm::Run&, const edm::EventSetup&) override;
   void produce(edm::Event&, edm::EventSetup const&) override;
 
   std::unordered_map<uint32_t, double> makeHitMap(edm::Event const&,
@@ -45,6 +46,8 @@ class CaloTruthCellsProducer : public edm::stream::EDProducer<> {
   edm::EDGetTokenT<std::vector<PCaloHit>> simHitsTokenEE_;
   edm::EDGetTokenT<std::vector<PCaloHit>> simHitsTokenHEfront_;
   edm::EDGetTokenT<std::vector<PCaloHit>> simHitsTokenHEback_;
+  edm::ESGetToken<HGCalTriggerGeometryBase, CaloGeometryRecord> triggerGeomToken_;
+  edm::ESHandle<HGCalTriggerGeometryBase> triggerGeomHandle_;
 
   HGCalClusteringDummyImpl dummyClustering_;
   HGCalShowerShape showerShape_;
@@ -60,6 +63,7 @@ CaloTruthCellsProducer::CaloTruthCellsProducer(edm::ParameterSet const& config)
       simHitsTokenEE_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsEE"))),
       simHitsTokenHEfront_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsHEfront"))),
       simHitsTokenHEback_(consumes<std::vector<PCaloHit>>(config.getParameter<edm::InputTag>("simHitsHEback"))),
+      triggerGeomToken_(esConsumes<HGCalTriggerGeometryBase, CaloGeometryRecord, edm::Transition::BeginRun>()),
       dummyClustering_(config.getParameterSet("dummyClustering")) {
   produces<CaloToCellsMap>();
   produces<l1t::HGCalClusterBxCollection>();
@@ -70,6 +74,10 @@ CaloTruthCellsProducer::CaloTruthCellsProducer(edm::ParameterSet const& config)
 
 CaloTruthCellsProducer::~CaloTruthCellsProducer() {}
 
+void CaloTruthCellsProducer::beginRun(const edm::Run& /*run*/, const edm::EventSetup& es) {
+  triggerGeomHandle_ = es.getHandle(triggerGeomToken_);
+}
+
 void CaloTruthCellsProducer::produce(edm::Event& event, edm::EventSetup const& setup) {
   edm::Handle<CaloParticleCollection> caloParticlesHandle;
   event.getByToken(caloParticlesToken_, caloParticlesHandle);
@@ -79,13 +87,11 @@ void CaloTruthCellsProducer::produce(edm::Event& event, edm::EventSetup const& s
   event.getByToken(triggerCellsToken_, triggerCellsHandle);
   auto const& triggerCells(*triggerCellsHandle);
 
-  edm::ESHandle<HGCalTriggerGeometryBase> geometryHandle;
-  setup.get<CaloGeometryRecord>().get(geometryHandle);
-  auto const& geometry(*geometryHandle);
+  auto const& geometry(*triggerGeomHandle_);
 
-  dummyClustering_.setGeometry(geometryHandle.product());
-  showerShape_.setGeometry(geometryHandle.product());
-  triggerTools_.setGeometry(geometryHandle.product());
+  dummyClustering_.setGeometry(triggerGeomHandle_.product());
+  showerShape_.setGeometry(triggerGeomHandle_.product());
+  triggerTools_.setGeometry(triggerGeomHandle_.product());
 
   std::unordered_map<uint32_t, CaloParticleRef> tcToCalo;