CaloTopology/test/HGCalTopologyTester.cc

0001 #include <iostream>
0002 #include <string>
0003 #include <vector>
0004
0005 #include "FWCore/Framework/interface/Frameworkfwd.h"
0006 #include "FWCore/Framework/interface/one/EDAnalyzer.h"
0007 #include "FWCore/Framework/interface/Event.h"
0008 #include "FWCore/Framework/interface/EventSetup.h"
0009 #include "FWCore/Framework/interface/ESTransientHandle.h"
0010 #include "FWCore/Framework/interface/MakerMacros.h"
0011 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0012 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0013 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0014
0015 #include "Geometry/Records/interface/IdealGeometryRecord.h"
0016 #include "Geometry/HGCalCommonData/interface/HGCalDDDConstants.h"
0017 #include "Geometry/CaloTopology/interface/HGCalTopology.h"
0018 #include "DataFormats/ForwardDetId/interface/HFNoseDetId.h"
0019 #include "DataFormats/ForwardDetId/interface/HGCScintillatorDetId.h"
0020 #include "DataFormats/ForwardDetId/interface/HGCSiliconDetId.h"
0021
0022 class HGCalTopologyTester : public edm::one::EDAnalyzer<edm::one::WatchRuns> {
0023 public:
0024   explicit HGCalTopologyTester(const edm::ParameterSet&);
0025
0026   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0027
0028 private:
0029   void analyze(edm::Event const&, edm::EventSetup const&) override;
0030   void beginJob() override {}
0031   void beginRun(edm::Run const&, edm::EventSetup const&) override {}
0032   void endRun(edm::Run const&, edm::EventSetup const&) override {}
0033   void doTest(const HGCalTopology& topology);
0034
0035   // ----------member data ---------------------------
0036   const std::string detectorName_;
0037   const std::vector<int> type_, layer_, sec1_, sec2_, cell1_, cell2_;
0038   const edm::ESGetToken<HGCalTopology, IdealGeometryRecord> tokTopo_;
0039 };
0040
0041 HGCalTopologyTester::HGCalTopologyTester(const edm::ParameterSet& iC)
0042     : detectorName_(iC.getParameter<std::string>("detectorName")),
0043       type_(iC.getParameter<std::vector<int> >("types")),
0044       layer_(iC.getParameter<std::vector<int> >("layers")),
0045       sec1_(iC.getParameter<std::vector<int> >("sector1")),
0046       sec2_(iC.getParameter<std::vector<int> >("sector2")),
0047       cell1_(iC.getParameter<std::vector<int> >("cell1")),
0048       cell2_(iC.getParameter<std::vector<int> >("cell2")),
0049       tokTopo_{esConsumes<HGCalTopology, IdealGeometryRecord>(edm::ESInputTag{"", detectorName_})} {}
0050
0051 void HGCalTopologyTester::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
0052   edm::ParameterSetDescription desc;
0053   std::vector<int> types = {0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2};
0054   std::vector<int> layer = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18};
0055   std::vector<int> sec1 = {1, 1, 2, 2, 3, 3, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10};
0056   std::vector<int> sec2 = {3, 3, 3, 3, 2, 2, 6, 6, 6, 6, 3, 3, 8, 8, 9, 9, 3, 3};
0057   std::vector<int> cell1 = {0, 4, 12, 14, 18, 23, 1, 4, 7, 10, 13, 16, 0, 3, 6, 9, 12, 15};
0058   std::vector<int> cell2 = {0, 4, 0, 2, 23, 18, 1, 4, 7, 10, 13, 16, 0, 3, 6, 9, 12, 15};
0059   desc.add<std::string>("detectorName", "HGCalEESensitive");
0060   desc.add<std::vector<int> >("types", types);
0061   desc.add<std::vector<int> >("layers", layer);
0062   desc.add<std::vector<int> >("sector1", sec1);
0063   desc.add<std::vector<int> >("sector2", sec2);
0064   desc.add<std::vector<int> >("cell1", cell1);
0065   desc.add<std::vector<int> >("cell2", cell2);
0066   descriptions.add("hgcalTopologyTesterEE", desc);
0067 }
0068
0069 void HGCalTopologyTester::analyze(edm::Event const&, edm::EventSetup const& iSetup) {
0070   doTest(iSetup.getData(tokTopo_));
0071 }
0072
0073 void HGCalTopologyTester::doTest(const HGCalTopology& topology) {
0074   if (topology.waferHexagon8() || topology.tileTrapezoid()) {
0075     for (unsigned int i = 0; i < type_.size(); ++i) {
0076       DetId id;
0077       if (topology.isHFNose()) {
0078         id = HFNoseDetId(1, type_[i], layer_[i], sec1_[i], sec2_[i], cell1_[i], cell2_[i]);
0079       } else if (detectorName_ == "HGCalEESensitive") {
0080         id = HGCSiliconDetId(DetId::HGCalEE, 1, type_[i], layer_[i], sec1_[i], sec2_[i], cell1_[i], cell2_[i]);
0081       } else if (detectorName_ == "HGCalHESiliconSensitive") {
0082         id = HGCSiliconDetId(DetId::HGCalHSi, 1, type_[i], layer_[i], sec1_[i], sec2_[i], cell1_[i], cell2_[i]);
0083       } else if (detectorName_ == "HGCalHEScintillatorSensitive") {
0084         id = HGCScintillatorDetId(type_[i], layer_[i], sec1_[i], cell1_[i]);
0085       } else {
0086         break;
0087       }
0088       std::vector<DetId> ids = topology.neighbors(id);
0089       unsigned int k(0);
0090       if (topology.isHFNose()) {
0091         edm::LogVerbatim("HGCalGeom") << (HFNoseDetId)(id) << " has " << ids.size() << " neighbours:";
0092         for (const auto& idn : ids) {
0093           edm::LogVerbatim("HGCalGeom") << "[" << k << "] " << (HFNoseDetId)(idn);
0094           ++k;
0095         }
0096       } else if (id.det() == DetId::HGCalEE || id.det() == DetId::HGCalHSi) {
0097         edm::LogVerbatim("HGCalGeom") << (HGCSiliconDetId)(id) << " has " << ids.size() << " neighbours:";
0098         for (const auto& idn : ids) {
0099           edm::LogVerbatim("HGCalGeom") << "[" << k << "] " << (HGCSiliconDetId)(idn);
0100           ++k;
0101         }
0102       } else {
0103         edm::LogVerbatim("HGCalGeom") << (HGCScintillatorDetId)(id) << " has " << ids.size() << " neighbours:";
0104         for (const auto& idn : ids) {
0105           edm::LogVerbatim("HGCalGeom") << "[" << k << "] " << (HGCScintillatorDetId)(idn);
0106           ++k;
0107         }
0108       }
0109     }
0110   }
0111 }
0112
0113 //define this as a plug-in
0114 DEFINE_FWK_MODULE(HGCalTopologyTester);