adding phase2 IT rechit dqm module

DQM/SiTrackerPhase2/plugins/BuildFile.xml

@@ -1,12 +1,14 @@
 <use name="FWCore/Framework"/>
 <use name="FWCore/ParameterSet"/>
 <use name="DataFormats/SiPixelDigi"/>
+<use name="DataFormats/Common"/>
 <use name="DataFormats/DetId"/>
+<use name="DataFormats/TrackerRecHit2D"/>
 <use name="Geometry/Records"/>
 <use name="Geometry/CommonDetUnit"/>
 <use name="Geometry/TrackerGeometryBuilder"/>
 <use name="DQMServices/Core"/>
-<use name="DataFormats/Common"/>
 <use name="boost_system"/>
 <use name="rootcling"/>
 <flags EDM_PLUGIN="1"/>
+<flags CXXFLAGS="-g"/>

DQM/SiTrackerPhase2/plugins/Phase2ITMonitorRecHit.cc

@@ -0,0 +1,266 @@
+// Package:    Phase2ITMonitorRecHit
+// Class:      Phase2ITMonitorRecHit
+//
+/**\class Phase2ITMonitorRecHit Phase2ITMonitorRecHit.cc 
+ Description:  Plugin for Phase2 RecHit validation
+*/
+//
+// Author: Shubhi Parolia, Suvankar Roy Chowdhury
+// Date: July 2020
+//
+// system include files
+#include <memory>
+#include "DQM/SiTrackerPhase2/plugins/Phase2ITMonitorRecHit.h"
+#include "DQM/SiTrackerPhase2/interface/TrackerPhase2DQMUtil.h"
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+#include "FWCore/Framework/interface/ESWatcher.h"
+#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
+#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
+#include "Geometry/CommonDetUnit/interface/GeomDet.h"
+#include "Geometry/CommonDetUnit/interface/TrackerGeomDet.h"
+#include "Geometry/CommonDetUnit/interface/PixelGeomDetUnit.h"
+#include "Geometry/CommonDetUnit/interface/PixelGeomDetType.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
+#include "DataFormats/GeometrySurface/interface/LocalError.h"
+#include "DataFormats/GeometryVector/interface/LocalPoint.h"
+// DQM Histograming
+#include "DQMServices/Core/interface/MonitorElement.h"
+//
+// constructors
+//
+Phase2ITMonitorRecHit::Phase2ITMonitorRecHit(const edm::ParameterSet& iConfig):
+  config_(iConfig),
+  tokenRecHitsIT_(consumes<SiPixelRecHitCollection>(iConfig.getParameter<edm::InputTag>("rechitsSrc")))
+{      
+  edm::LogInfo("Phase2ITMonitorRecHit") << ">>> Construct Phase2ITMonitorRecHit ";
+}
+
+Phase2ITMonitorRecHit::~Phase2ITMonitorRecHit() {
+  edm::LogInfo("Phase2ITMonitorRecHit") << ">>> Destroy Phase2ITMonitorRecHit ";
+}
+// -- Analyze
+void Phase2ITMonitorRecHit::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
+
+  // Get the geometry
+  edm::ESHandle<TrackerGeometry> geomHandle;
+  iSetup.get<TrackerDigiGeometryRecord>().get(geomHandle);
+  const TrackerGeometry* tkGeom = &(*geomHandle);
+
+  edm::ESHandle<TrackerTopology> tTopoHandle;
+  iSetup.get<TrackerTopologyRcd>().get(tTopoHandle);
+  const TrackerTopology* tTopo = tTopoHandle.product();
+
+  fillITHistos(iEvent, tTopo, tkGeom);
+}
+
+void Phase2ITMonitorRecHit::fillITHistos(const edm::Event& iEvent, 
+					 const TrackerTopology* tTopo, 
+					 const TrackerGeometry* tkGeom) {
+  // Get the RecHits
+  edm::Handle<SiPixelRecHitCollection> rechits;
+  iEvent.getByToken(tokenRecHitsIT_, rechits);
+  if(!rechits.isValid())  return;
+  std::map<std::string, unsigned int>  nrechitLayerMap;
+  unsigned long int nTotrechitsinevt = 0;
+  // Loop over modules
+  SiPixelRecHitCollection::const_iterator DSViter; 
+  for (DSViter = rechits->begin(); DSViter != rechits->end(); ++DSViter) {
+    // Get the detector id
+    unsigned int rawid(DSViter->detId());
+    DetId detId(rawid);
+    // Get the geomdet
+    const GeomDetUnit* geomDetunit(tkGeom->idToDetUnit(detId));
+    if (!geomDetunit)
+      continue;
+    std::string key = Phase2TkUtil::getITHistoId(detId.rawId(), tTopo);
+    nTotrechitsinevt += DSViter->size();
+    if(nrechitLayerMap.find(key) == nrechitLayerMap.end()) {
+	nrechitLayerMap.insert(std::make_pair(key, DSViter->size())); 
+    } else {
+      nrechitLayerMap[key] += DSViter->size();
+    }
+
+    edmNew::DetSet<SiPixelRecHit>::const_iterator rechitIt;
+    //loop over rechits for a single detId
+    for(rechitIt = DSViter->begin(); rechitIt != DSViter->end(); ++rechitIt) {
+      LocalPoint lp = rechitIt->localPosition();
+      Global3DPoint globalPos = geomDetunit->surface().toGlobal(lp);
+      //in mm
+      double gx = globalPos.x()*10.;
+      double gy = globalPos.y()*10.;
+      double gz = globalPos.z()*10.;
+      double gr = globalPos.perp()*10.;
+      //Fill global positions
+      if(geomDetunit->subDetector() == GeomDetEnumerators::SubDetector::P2PXB) {
+	globalXY_barrel_->Fill(gx, gy);
+	globalRZ_barrel_->Fill(gz, gr);
+      } else if(geomDetunit->subDetector() == GeomDetEnumerators::SubDetector::P2PXEC) {
+	globalXY_endcap_->Fill(gx, gy);
+	globalRZ_endcap_->Fill(gz, gr);
+      }
+      //layer wise histo
+      if(layerMEs_[key].clusterSizeX) layerMEs_[key].clusterSizeX->Fill(rechitIt->cluster()->sizeX());
+      if(layerMEs_[key].clusterSizeY) layerMEs_[key].clusterSizeY->Fill(rechitIt->cluster()->sizeY());
+      if(layerMEs_[key].globalPosXY) layerMEs_[key].globalPosXY->Fill(gx, gy); 
+      if(layerMEs_[key].globalPosRZ) layerMEs_[key].globalPosRZ->Fill(gz, gr); 
+      if(layerMEs_[key].localPosXY) layerMEs_[key].localPosXY->Fill(lp.x(), lp.y()); 
+      if(layerMEs_[key].posX) layerMEs_[key].posX->Fill(lp.x());
+      if(layerMEs_[key].posY) layerMEs_[key].posX->Fill(lp.y());
+      if(layerMEs_[key].posX) layerMEs_[key].poserrX->Fill(rechitIt->localPositionError().xx());
+      if(layerMEs_[key].posY) layerMEs_[key].poserrY->Fill(rechitIt->localPositionError().yy());
+    }//end loop over rechits of a detId
+  } //End loop over DetSetVector
+
+  //fill nRecHits per event
+  numberRecHits_->Fill(nTotrechitsinevt);
+  //fill nRecHit counter per layer
+  for(auto& lme : nrechitLayerMap) 
+    if(layerMEs_[lme.first].numberRecHits) 
+      layerMEs_[lme.first].numberRecHits->Fill(lme.second);
+}
+void Phase2ITMonitorRecHit::bookHistograms(DQMStore::IBooker& ibooker,
+						 edm::Run const& iRun,
+						 edm::EventSetup const& iSetup) {
+
+  std::string top_folder = config_.getParameter<std::string>("TopFolderName");
+  ibooker.cd();
+  edm::LogInfo("Phase2ITMonitorRecHit") << " Booking Histograms in : " << top_folder;
+  std::string dir = top_folder;
+  ibooker.setCurrentFolder(dir);
+  std::stringstream HistoName;
+  //Global histos for IT
+  HistoName.str("");
+  HistoName << "NumberRecHits";
+  numberRecHits_ = Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalNumberRecHits"),
+						HistoName.str(),
+						ibooker);
+  HistoName.str("");
+  HistoName << "Global_Position_XY_IT_barrel";
+  globalXY_barrel_ = Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionXY_PXB"),
+						  HistoName.str(),
+						  ibooker);
+  HistoName.str("");
+  HistoName << "Global_Position_RZ_IT_barrel";
+  globalRZ_barrel_ = Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionRZ_PXB"),
+						  HistoName.str(),
+						  ibooker);
+  HistoName.str("");
+  HistoName << "Global_Position_XY_IT_endcap";
+  globalXY_endcap_ = Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionXY_PXEC"),
+						  HistoName.str(),
+						  ibooker);
+  HistoName.str("");
+  HistoName << "Global_Position_RZ_IT_endcap";
+  globalRZ_endcap_ = Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionRZ_PXEC"),
+						  HistoName.str(),
+						  ibooker);
+
+  //Now book layer wise histos
+  edm::ESWatcher<TrackerDigiGeometryRecord> theTkDigiGeomWatcher;
+  edm::ESHandle<TrackerTopology> tTopoHandle;
+  iSetup.get<TrackerTopologyRcd>().get(tTopoHandle);
+  const TrackerTopology* const tTopo = tTopoHandle.product();
+  if (theTkDigiGeomWatcher.check(iSetup)) {
+    edm::ESHandle<TrackerGeometry> geom_handle;
+    iSetup.get<TrackerDigiGeometryRecord>().get(geomType_, geom_handle);
+    const TrackerGeometry* tGeom = geom_handle.product();
+    for (auto const& det_u : tGeom->detUnits()) {
+      //Always check TrackerNumberingBuilder before changing this part
+      if(!(det_u->subDetector() == GeomDetEnumerators::SubDetector::P2PXB 
+	   || det_u->subDetector() == GeomDetEnumerators::SubDetector::P2PXEC))
+	continue;
+      unsigned int detId_raw = det_u->geographicalId().rawId();
+      edm::LogInfo("Phase2ITMonitorRecHit") << "Detid:" << detId_raw
+					     <<"\tsubdet=" << det_u->subDetector()
+					     << "\t key=" << Phase2TkUtil::getITHistoId(detId_raw, tTopo)
+					     << std::endl;
+      bookLayerHistos(ibooker, detId_raw, tTopo, dir);
+    }
+  }
+}
+// -- Book Layer Histograms
+void Phase2ITMonitorRecHit::bookLayerHistos(DQMStore::IBooker& ibooker,
+					     unsigned int det_id,
+					     const TrackerTopology* tTopo,
+					     std::string& subdir) {  
+  std::string key = Phase2TkUtil::getITHistoId(det_id, tTopo);
+  if(key.empty())  return;
+  if (layerMEs_.find(key) == layerMEs_.end()) {
+    ibooker.cd();
+    RecHitME local_histos;
+    std::ostringstream histoName;
+    ibooker.setCurrentFolder(subdir+"/"+key);
+    std::cout<< "Setting subfolder>>>" << subdir << "\t" << key << std::endl;
+    edm::LogInfo("Phase2ITMonitorRecHit") << " Booking Histograms in : " << (subdir+"/"+key);
+    histoName.str("");
+    histoName << "Number_RecHits";
+    local_histos.numberRecHits = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("LocalNumberRecHits"),
+				   histoName.str(),
+				   ibooker);
+    histoName.str("");
+    histoName << "RecHit_X";
+    local_histos.posX = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("RecHitPosX"),
+				   histoName.str(),
+				   ibooker);
+
+    histoName.str("");
+    histoName << "RecHit_Y";
+    local_histos.posY = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("RecHitPosY"),
+				   histoName.str(),
+				   ibooker);
+
+    histoName.str("");
+    histoName << "RecHit_X_error";
+    local_histos.poserrX = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("RecHitPosErrorX"),
+				   histoName.str(),
+				   ibooker);
+
+    histoName.str("");
+    histoName << "RecHit_Y_error";
+    local_histos.poserrY = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("RecHitPosErrorY"),
+				   histoName.str(),
+				   ibooker);
+
+    histoName.str("");
+    histoName << "Cluster_SizeX";
+    local_histos.clusterSizeX = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("LocalClusterSizeX"),
+				   histoName.str(),
+				   ibooker);
+    histoName.str("");
+    histoName << "Cluster_SizeY";
+    local_histos.clusterSizeY = 
+      Phase2TkUtil::book1DFromPSet(config_.getParameter<edm::ParameterSet>("LocalClusterSizeY"),
+				   histoName.str(),
+				   ibooker);
+
+    histoName.str("");
+    histoName << "Global_Position_XY"; 
+    local_histos.globalPosXY = 
+      Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionXY_perlayer"),
+				   histoName.str(),
+				   ibooker);    
+
+   histoName.str("");
+    histoName << "Global_Position_RZ"; 
+    local_histos.globalPosRZ = 
+      Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("GlobalPositionRZ_perlayer"),
+				   histoName.str(),
+				   ibooker);    
+    histoName.str("");
+    histoName << "Local_Position_XY" ;
+    local_histos.localPosXY = 
+      Phase2TkUtil::book2DFromPSet(config_.getParameter<edm::ParameterSet>("LocalPositionXY"),
+				   histoName.str(),
+				   ibooker);
+    layerMEs_.insert(std::make_pair(key, local_histos));
+  }
+}  
+//define this as a plug-in
+DEFINE_FWK_MODULE(Phase2ITMonitorRecHit);

DQM/SiTrackerPhase2/plugins/Phase2ITMonitorRecHit.h

@@ -0,0 +1,68 @@
+#ifndef _Validation_SiTrackerPhase2V_Phase2ITMonitorRecHit_h
+#define _Validation_SiTrackerPhase2V_Phase2ITMonitorRecHit_h
+#include <iostream>
+#include <map>
+#include <vector>
+#include <algorithm>
+#include "FWCore/Framework/interface/Frameworkfwd.h"
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+#include "FWCore/ServiceRegistry/interface/Service.h"
+#include "FWCore/Framework/interface/ESHandle.h"
+#include "FWCore/Utilities/interface/InputTag.h"
+#include "DataFormats/DetId/interface/DetId.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+#include "DataFormats/Common/interface/DetSetVectorNew.h"
+#include "DataFormats/Common/interface/Handle.h"
+#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
+#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
+#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
+#include "DQMServices/Core/interface/DQMEDAnalyzer.h"
+#include "DQMServices/Core/interface/DQMStore.h"
+
+
+// DQM Histograming
+class TrackerTopology;
+class TrackerGeometry;
+
+class Phase2ITMonitorRecHit : public DQMEDAnalyzer {
+public:
+  explicit Phase2ITMonitorRecHit(const edm::ParameterSet&);
+  ~Phase2ITMonitorRecHit() override;
+  void bookHistograms(DQMStore::IBooker& ibooker, edm::Run const& iRun, edm::EventSetup const& iSetup) override;
+  void analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) override;
+
+ private:
+  void fillITHistos(const edm::Event& iEvent, 
+		    const TrackerTopology* tTopo, 
+		    const TrackerGeometry* tkGeom);
+
+  void bookLayerHistos(DQMStore::IBooker& ibooker, unsigned int det_id, const TrackerTopology* tTopo, std::string& subdir);
+
+  edm::ParameterSet config_;
+  std::string geomType_;
+  const edm::EDGetTokenT<SiPixelRecHitCollection> tokenRecHitsIT_;
+
+  MonitorElement* numberRecHits_;
+  MonitorElement* globalXY_barrel_;
+  MonitorElement* globalXY_endcap_;
+  MonitorElement* globalRZ_barrel_;
+  MonitorElement* globalRZ_endcap_;
+
+  struct RecHitME {
+    MonitorElement* numberRecHits = nullptr;
+    MonitorElement* globalPosXY = nullptr;
+    MonitorElement* globalPosRZ = nullptr;
+    MonitorElement* localPosXY = nullptr;
+    MonitorElement* posX = nullptr;
+    MonitorElement* posY = nullptr;
+    MonitorElement* poserrX = nullptr;
+    MonitorElement* poserrY = nullptr;
+    MonitorElement* clusterSizeX = nullptr;
+    MonitorElement* clusterSizeY = nullptr;
+
+  };
+  std::map<std::string, RecHitME>  layerMEs_;
+};
+#endif