Make sure hitmap is filled also without RECO

Author: lecriste

Fireworks/Calo/interface/FWHeatmapProxyBuilderTemplate.h

@@ -44,7 +44,7 @@ class FWHeatmapProxyBuilderTemplate : public FWSimpleProxyBuilder {
   // ---------- member functions ---------------------------
 
 protected:
-  const std::unordered_map<DetId, const HGCRecHit*>* hitmap;
+  std::unordered_map<DetId, const HGCRecHit*>* hitmap = new std::unordered_map<DetId, const HGCRecHit*>;
 
   static constexpr uint8_t gradient_steps = 9;
   static constexpr uint8_t gradient[3][gradient_steps] = {{static_cast<uint8_t>(0.2082 * 255),
@@ -93,11 +93,41 @@ class FWHeatmapProxyBuilderTemplate : public FWSimpleProxyBuilder {
 
   void build(const FWEventItem* iItem, TEveElementList* product, const FWViewContext* vc) override {
     if (item()->getConfig()->template value<bool>("Heatmap")) {
+      hitmap->clear();
+
+      edm::Handle<HGCRecHitCollection> recHitHandleEE;
+      edm::Handle<HGCRecHitCollection> recHitHandleFH;
+      edm::Handle<HGCRecHitCollection> recHitHandleBH;
+
       const edm::EventBase* event = iItem->getEvent();
 
-      edm::Handle<std::unordered_map<DetId, const HGCRecHit*>> hitMapHandle;
-      event->getByLabel(edm::InputTag("hgcalRecHitMapProducer"), hitMapHandle);
-      hitmap = &*hitMapHandle;
+      event->getByLabel(edm::InputTag("HGCalRecHit", "HGCEERecHits"), recHitHandleEE);
+      event->getByLabel(edm::InputTag("HGCalRecHit", "HGCHEFRecHits"), recHitHandleFH);
+      event->getByLabel(edm::InputTag("HGCalRecHit", "HGCHEBRecHits"), recHitHandleBH);
+
+      if (recHitHandleEE.isValid()) {
+        const auto& rechitsEE = *recHitHandleEE;
+
+        for (unsigned int i = 0; i < rechitsEE.size(); ++i) {
+          (*hitmap)[rechitsEE[i].detid().rawId()] = &rechitsEE[i];
+        }
+      }
+
+      if (recHitHandleFH.isValid()) {
+        const auto& rechitsFH = *recHitHandleFH;
+
+        for (unsigned int i = 0; i < rechitsFH.size(); ++i) {
+          (*hitmap)[rechitsFH[i].detid().rawId()] = &rechitsFH[i];
+        }
+      }
+
+      if (recHitHandleBH.isValid()) {
+        const auto& rechitsBH = *recHitHandleBH;
+
+        for (unsigned int i = 0; i < rechitsBH.size(); ++i) {
+          (*hitmap)[rechitsBH[i].detid().rawId()] = &rechitsBH[i];
+        }
+      }
     }
 
     FWSimpleProxyBuilder::build(iItem, product, vc);