Add option to use SiStrip quality DB information (disabled by default)

trackreco · Jul 6, 2021 · 19205b8 · 19205b8
1 parent ce66870
commit 19205b8
Showing 1 changed file with 28 additions and 20 deletions.
diff --git a/RecoTracker/MkFit/plugins/MkFitEventOfHitsProducer.cc b/RecoTracker/MkFit/plugins/MkFitEventOfHitsProducer.cc
@@ -41,9 +41,10 @@ class MkFitEventOfHitsProducer : public edm::global::EDProducer<> {
   const edm::EDGetTokenT<MkFitClusterIndexToHit> pixelClusterIndexToHitToken_;
   const edm::EDGetTokenT<MkFitClusterIndexToHit> stripClusterIndexToHitToken_;
   const edm::ESGetToken<MkFitGeometry, TrackerRecoGeometryRecord> mkFitGeomToken_;
-  const edm::ESGetToken<SiStripQuality, SiStripQualityRcd> qualityToken_;
-  const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geomToken_;
+  edm::ESGetToken<SiStripQuality, SiStripQualityRcd> qualityToken_;
+  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geomToken_;
   const edm::EDPutTokenT<MkFitEventOfHits> putToken_;
+  const bool useStripStripQualityDB_;
 };
 
 MkFitEventOfHitsProducer::MkFitEventOfHitsProducer(edm::ParameterSet const& iConfig)
@@ -52,15 +53,20 @@ MkFitEventOfHitsProducer::MkFitEventOfHitsProducer(edm::ParameterSet const& iCon
       pixelClusterIndexToHitToken_{consumes(iConfig.getParameter<edm::InputTag>("pixelHits"))},
       stripClusterIndexToHitToken_{consumes(iConfig.getParameter<edm::InputTag>("stripHits"))},
       mkFitGeomToken_{esConsumes()},
-      qualityToken_{esConsumes()},
-      geomToken_{esConsumes()},
-      putToken_{produces<MkFitEventOfHits>()} {}
+      putToken_{produces<MkFitEventOfHits>()},
+      useStripStripQualityDB_{iConfig.getParameter<bool>("useStripStripQualityDB")} {
+	if (useStripStripQualityDB_) {
+	  qualityToken_ = esConsumes();
+	  geomToken_ = esConsumes();
+	}
+      }
 
 void MkFitEventOfHitsProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
   edm::ParameterSetDescription desc;
 
   desc.add("pixelHits", edm::InputTag{"mkFitSiPixelHits"});
   desc.add("stripHits", edm::InputTag{"mkFitSiStripHits"});
+  desc.add("useStripStripQualityDB", false)->setComment("Use SiStrip quality DB information");
 
   descriptions.addWithDefaultLabel(desc);
 }
@@ -73,22 +79,24 @@ void MkFitEventOfHitsProducer::produce(edm::StreamID iID, edm::Event& iEvent, co
   auto eventOfHits = std::make_unique<mkfit::EventOfHits>(mkFitGeom.trackerInfo());
   mkfit::StdSeq::Cmssw_LoadHits_Begin(*eventOfHits, {&pixelHits.hits(), &stripHits.hits()});
 
-  std::vector<mkfit::DeadVec> deadvectors(mkFitGeom.layerNumberConverter().nLayers());
-  const auto& siStripQuality = iSetup.getData(qualityToken_);
-  const auto& trackerGeom = iSetup.getData(geomToken_);
-  const auto& badStrips = siStripQuality.getBadComponentList();
-  for (const auto& bs : badStrips) {
-    const auto& surf = trackerGeom.idToDet(DetId(bs.detid))->surface();
-    const DetId detid(bs.detid);
-    bool isBarrel = (mkFitGeom.topology()->side(detid) == static_cast<unsigned>(TrackerDetSide::Barrel));
-    const auto ilay = mkFitGeom.mkFitLayerNumber(detid);
-    //dump content of deadmodules.h in standalone setup
-    // std::cout << "deadvectors["<<ilay<<"].push_back({"<<surf.phiSpan().first<<","<<surf.phiSpan().second<<","
-    // 		<<(isBarrel ? surf.zSpan().first : surf.rSpan().first)<<","<<(isBarrel ? surf.zSpan().second : surf.rSpan().second)<<"});"<<std::endl;
-    deadvectors[ilay].push_back({surf.phiSpan().first,surf.phiSpan().second,
-	  (isBarrel ? surf.zSpan().first : surf.rSpan().first),(isBarrel ? surf.zSpan().second : surf.rSpan().second)});
+  if(useStripStripQualityDB_) {
+    std::vector<mkfit::DeadVec> deadvectors(mkFitGeom.layerNumberConverter().nLayers());
+    const auto& siStripQuality = iSetup.getData(qualityToken_);
+    const auto& trackerGeom = iSetup.getData(geomToken_);
+    const auto& badStrips = siStripQuality.getBadComponentList();
+    for (const auto& bs : badStrips) {
+      const auto& surf = trackerGeom.idToDet(DetId(bs.detid))->surface();
+      const DetId detid(bs.detid);
+      bool isBarrel = (mkFitGeom.topology()->side(detid) == static_cast<unsigned>(TrackerDetSide::Barrel));
+      const auto ilay = mkFitGeom.mkFitLayerNumber(detid);
+      //dump content of deadmodules.h in standalone setup
+      // std::cout << "deadvectors["<<ilay<<"].push_back({"<<surf.phiSpan().first<<","<<surf.phiSpan().second<<","
+      // 		<<(isBarrel ? surf.zSpan().first : surf.rSpan().first)<<","<<(isBarrel ? surf.zSpan().second : surf.rSpan().second)<<"});"<<std::endl;
+      deadvectors[ilay].push_back({surf.phiSpan().first,surf.phiSpan().second,
+	    (isBarrel ? surf.zSpan().first : surf.rSpan().first),(isBarrel ? surf.zSpan().second : surf.rSpan().second)});
+    }
+    mkfit::StdSeq::LoadDeads(*eventOfHits, deadvectors);
   }
-  mkfit::StdSeq::LoadDeads(*eventOfHits, deadvectors);
 
   fill(iEvent.get(pixelClusterIndexToHitToken_).hits(), *eventOfHits, mkFitGeom);
   fill(iEvent.get(stripClusterIndexToHitToken_).hits(), *eventOfHits, mkFitGeom);