Local/plugins/CTPPSDiamondRecHitProducer.cc

0001 /****************************************************************************
0002  *
0003  * This is a part of PPS offline software.
0004  * Authors:
0005  *   Laurent Forthomme (laurent.forthomme@cern.ch)
0006  *   Nicola Minafra (nicola.minafra@cern.ch)
0007  *
0008  ****************************************************************************/
0009
0010 #include <memory>
0011
0012 #include "FWCore/Framework/interface/Frameworkfwd.h"
0013 #include "FWCore/Framework/interface/stream/EDProducer.h"
0014 #include "FWCore/Framework/interface/Event.h"
0015 #include "FWCore/Framework/interface/MakerMacros.h"
0016 #include "FWCore/Framework/interface/ESHandle.h"
0017 #include "FWCore/Framework/interface/ESWatcher.h"
0018
0019 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0020 #include "FWCore/Utilities/interface/StreamID.h"
0021 #include "FWCore/Utilities/interface/ESGetToken.h"
0022
0023 #include "DataFormats/Common/interface/DetSetVector.h"
0024 #include "DataFormats/Common/interface/DetSet.h"
0025
0026 #include "DataFormats/CTPPSDetId/interface/CTPPSDiamondDetId.h"
0027 #include "DataFormats/CTPPSDigi/interface/CTPPSDiamondDigi.h"
0028 #include "DataFormats/CTPPSReco/interface/CTPPSDiamondRecHit.h"
0029
0030 #include "RecoPPS/Local/interface/CTPPSDiamondRecHitProducerAlgorithm.h"
0031
0032 #include "Geometry/Records/interface/VeryForwardRealGeometryRecord.h"
0033 #include "CondFormats/DataRecord/interface/PPSTimingCalibrationRcd.h"
0034 #include "CondFormats/DataRecord/interface/PPSTimingCalibrationLUTRcd.h"
0035
0036 class CTPPSDiamondRecHitProducer : public edm::stream::EDProducer<> {
0037 public:
0038   explicit CTPPSDiamondRecHitProducer(const edm::ParameterSet&);
0039
0040   static void fillDescriptions(edm::ConfigurationDescriptions&);
0041
0042 private:
0043   void produce(edm::Event&, const edm::EventSetup&) override;
0044
0045   edm::EDGetTokenT<edm::DetSetVector<CTPPSDiamondDigi> > digiToken_;
0046   edm::ESGetToken<PPSTimingCalibration, PPSTimingCalibrationRcd> timingCalibrationToken_;
0047   edm::ESGetToken<PPSTimingCalibrationLUT, PPSTimingCalibrationLUTRcd> timingCalibrationLUTToken_;
0048   edm::ESGetToken<CTPPSGeometry, VeryForwardRealGeometryRecord> geometryToken_;
0049
0050   /// A watcher to detect timing calibration changes.
0051   edm::ESWatcher<PPSTimingCalibrationRcd> calibWatcher_;
0052   edm::ESWatcher<PPSTimingCalibrationLUTRcd> lutWatcher_;
0053
0054   bool applyCalib_;
0055   CTPPSDiamondRecHitProducerAlgorithm algo_;
0056 };
0057
0058 CTPPSDiamondRecHitProducer::CTPPSDiamondRecHitProducer(const edm::ParameterSet& iConfig)
0059     : digiToken_(consumes<edm::DetSetVector<CTPPSDiamondDigi> >(iConfig.getParameter<edm::InputTag>("digiTag"))),
0060       geometryToken_(esConsumes<CTPPSGeometry, VeryForwardRealGeometryRecord>()),
0061       applyCalib_(iConfig.getParameter<bool>("applyCalibration")),
0062       algo_(iConfig) {
0063   if (applyCalib_) {
0064     timingCalibrationToken_ = esConsumes<PPSTimingCalibration, PPSTimingCalibrationRcd>(
0065         edm::ESInputTag(iConfig.getParameter<std::string>("timingCalibrationTag")));
0066     timingCalibrationLUTToken_ = esConsumes<PPSTimingCalibrationLUT, PPSTimingCalibrationLUTRcd>();
0067   }
0068   produces<edm::DetSetVector<CTPPSDiamondRecHit> >();
0069 }
0070
0071 void CTPPSDiamondRecHitProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
0072   auto pOut = std::make_unique<edm::DetSetVector<CTPPSDiamondRecHit> >();
0073
0074   // get the digi collection
0075   const auto& digis = iEvent.get(digiToken_);
0076
0077   if (!digis.empty()) {
0078     if (applyCalib_ && (calibWatcher_.check(iSetup) or lutWatcher_.check(iSetup)))
0079       algo_.setCalibration(iSetup.getData(timingCalibrationToken_), iSetup.getData(timingCalibrationLUTToken_));
0080
0081     // produce the rechits collection
0082     algo_.build(iSetup.getData(geometryToken_), digis, *pOut);
0083   }
0084
0085   iEvent.put(std::move(pOut));
0086 }
0087
0088 void CTPPSDiamondRecHitProducer::fillDescriptions(edm::ConfigurationDescriptions& descr) {
0089   edm::ParameterSetDescription desc;
0090
0091   desc.add<edm::InputTag>("digiTag", edm::InputTag("ctppsDiamondRawToDigi", "TimingDiamond"))
0092       ->setComment("input digis collection to retrieve");
0093   desc.add<std::string>("timingCalibrationTag", ":PPSDiamondTimingCalibration")
0094       ->setComment("input tag for timing calibrations retrieval");
0095   desc.add<double>("timeSliceNs", 25.0 / 1024.0)
0096       ->setComment("conversion constant between HPTDC timing bin size and nanoseconds");
0097   desc.add<bool>("applyCalibration", true)->setComment("switch on/off the timing calibration");
0098
0099   descr.add("ctppsDiamondRecHits", desc);
0100 }
0101
0102 DEFINE_FWK_MODULE(CTPPSDiamondRecHitProducer);