From df6536e08ab48365ff5a45c3be9021876bebce41 Mon Sep 17 00:00:00 2001
From: Fabio Cossutti <fabio.cossutti@ts.infn.it>
Date: Sun, 28 Jun 2020 12:22:06 +0200
Subject: [PATCH] Add MTD to the digi dump

---
 SimGeneral/Debugging/test/SimDigiDumper.cc    | 48 +++++++++++++++++++
 SimGeneral/Debugging/test/SimDigiDumper.h     |  5 ++
 .../Debugging/test/runSimDigiDumper_cfg.py    |  6 ++-
 3 files changed, 57 insertions(+), 2 deletions(-)

diff --git a/SimGeneral/Debugging/test/SimDigiDumper.cc b/SimGeneral/Debugging/test/SimDigiDumper.cc
index 5d3afd116d3f5..4a7c3113499c3 100644
--- a/SimGeneral/Debugging/test/SimDigiDumper.cc
+++ b/SimGeneral/Debugging/test/SimDigiDumper.cc
@@ -39,6 +39,8 @@ SimDigiDumper::SimDigiDumper(const edm::ParameterSet& iPSet) {
   MuCSCStripSrc_ = consumes<CSCStripDigiCollection>(iPSet.getParameter<edm::InputTag>("MuCSCStripSrc"));
   MuCSCWireSrc_ = consumes<CSCWireDigiCollection>(iPSet.getParameter<edm::InputTag>("MuCSCWireSrc"));
   MuRPCSrc_ = consumes<RPCDigiCollection>(iPSet.getParameter<edm::InputTag>("MuRPCSrc"));
+  BTLSrc_ = consumes<BTLDigiCollection>(iPSet.getParameter<edm::InputTag>("BTLSrc"));
+  ETLSrc_ = consumes<ETLDigiCollection>(iPSet.getParameter<edm::InputTag>("ETLSrc"));
 
   // TODO(proper responsible): update the cout, for sure not my
   // business.
@@ -148,6 +150,52 @@ void SimDigiDumper::analyze(const edm::Event& iEvent, const edm::EventSetup& iSe
     }
   }
 
+  // BTL
+  bool isBTL = true;
+  edm::Handle<BTLDigiCollection> BTLDigi;
+  const BTLDigiCollection* BTLdigis = 0;
+  iEvent.getByToken(BTLSrc_, BTLDigi);
+  if (!BTLDigi.isValid()) {
+    std::cout << "Unable to find BTLDigi in event!" << std::endl;
+  } else {
+    BTLdigis = BTLDigi.product();
+    if (BTLDigi->size() == 0)
+      isBTL = false;
+    std::cout << "Barrel Timing Layer, digi multiplicity = " << BTLDigi->size() << std::endl;
+
+    if (isBTL) {
+      // loop over digis
+      for (unsigned int digis = 0; digis < BTLDigi->size(); ++digis) {
+        BTLDataFrame btldf = (*BTLdigis)[digis];
+        std::cout << btldf.id().rawId() << std::endl;
+        btldf.print();
+      }
+    }
+  }
+
+  // ETL
+  bool isETL = true;
+  edm::Handle<ETLDigiCollection> ETLDigi;
+  const ETLDigiCollection* ETLdigis = 0;
+  iEvent.getByToken(ETLSrc_, ETLDigi);
+  if (!ETLDigi.isValid()) {
+    std::cout << "Unable to find ETLDigi in event!" << std::endl;
+  } else {
+    ETLdigis = ETLDigi.product();
+    if (ETLDigi->size() == 0)
+      isETL = false;
+    std::cout << "Endcap Timing Layer, digi multiplicity = " << ETLDigi->size() << std::endl;
+
+    if (isETL) {
+      // loop over digis
+      for (unsigned int digis = 0; digis < ETLDigi->size(); ++digis) {
+        ETLDataFrame etldf = (*ETLdigis)[digis];
+        std::cout << etldf.id().rawId() << std::endl;
+        etldf.print();
+      }
+    }
+  }
+
   // HBHE
   bool isHBHE = true;
   edm::Handle<HBHEDigiCollection> hbhe;
diff --git a/SimGeneral/Debugging/test/SimDigiDumper.h b/SimGeneral/Debugging/test/SimDigiDumper.h
index 5ec3011ae7a1c..864a8b7cd364a 100644
--- a/SimGeneral/Debugging/test/SimDigiDumper.h
+++ b/SimGeneral/Debugging/test/SimDigiDumper.h
@@ -26,6 +26,8 @@
 #include "DataFormats/CSCDigi/interface/CSCWireDigiCollection.h"
 // muon RPC info
 #include "DataFormats/RPCDigi/interface/RPCDigiCollection.h"
+// BTL/ETL info
+#include "DataFormats/FTLDigi/interface/FTLDigiCollections.h"
 
 #include <vector>
 
@@ -43,6 +45,9 @@ class SimDigiDumper : public edm::EDAnalyzer {
   edm::EDGetTokenT<EEDigiCollection> ECalEESrc_;
   edm::EDGetTokenT<ESDigiCollection> ECalESSrc_;
 
+  edm::EDGetTokenT<BTLDigiCollection> BTLSrc_;
+  edm::EDGetTokenT<ETLDigiCollection> ETLSrc_;
+
   edm::EDGetTokenT<HBHEDigiCollection> HCalDigi_;
   edm::EDGetTokenT<HODigiCollection> HCalHODigi_;
   edm::EDGetTokenT<HFDigiCollection> HCalHFDigi_;
diff --git a/SimGeneral/Debugging/test/runSimDigiDumper_cfg.py b/SimGeneral/Debugging/test/runSimDigiDumper_cfg.py
index f17e40ed6efff..c227f28f43bca 100644
--- a/SimGeneral/Debugging/test/runSimDigiDumper_cfg.py
+++ b/SimGeneral/Debugging/test/runSimDigiDumper_cfg.py
@@ -16,14 +16,16 @@
     MuCSCStripSrc = cms.InputTag("simMuonCSCDigis","MuonCSCStripDigi"),
     MuDTSrc = cms.InputTag("simMuonDTDigis"),
     HCalDigi = cms.InputTag("simHcalDigis"),
-    ZdcDigi = cms.InputTag("simHcalUnsuppressedDigis"),                        
+    ZdcDigi = cms.InputTag("simHcalUnsuppressedDigis"),
     MuCSCWireSrc = cms.InputTag("simMuonCSCDigis","MuonCSCWireDigi"),
     ECalEESrc = cms.InputTag("simEcalDigis","eeDigis"),
     SiStripSrc = cms.InputTag("simSiStripDigis","ZeroSuppressed"),
     SiPxlSrc = cms.InputTag("simSiPixelDigis"),
     ECalEBSrc = cms.InputTag("simEcalDigis","ebDigis"),
     ECalESSrc = cms.InputTag("simEcalPreshowerDigis"),
-    MuRPCSrc = cms.InputTag("simMuonRPCDigis")
+    MuRPCSrc = cms.InputTag("simMuonRPCDigis"),
+    BTLSrc = cms.InputTag("mix","FTLBarrel"),
+    ETLSrc = cms.InputTag("mix","FTLEndcap"),
 )
 
 process.p1 = cms.Path(process.prod)