TrackerTFP/plugins/ProducerTFP.cc

0001 #include "FWCore/Framework/interface/stream/EDProducer.h"
0002 #include "FWCore/Framework/interface/Run.h"
0003 #include "FWCore/Framework/interface/EventSetup.h"
0004 #include "FWCore/Framework/interface/Event.h"
0005 #include "FWCore/Framework/interface/MakerMacros.h"
0006 #include "FWCore/Utilities/interface/EDGetToken.h"
0007 #include "FWCore/Utilities/interface/EDPutToken.h"
0008 #include "FWCore/Utilities/interface/ESGetToken.h"
0009 #include "FWCore/Utilities/interface/InputTag.h"
0010 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0011 #include "DataFormats/Common/interface/Handle.h"
0012 #include "DataFormats/Common/interface/OrphanHandle.h"
0013
0014 #include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
0015 #include "L1Trigger/TrackTrigger/interface/Setup.h"
0016 #include "L1Trigger/TrackerTFP/interface/DataFormats.h"
0017 #include "L1Trigger/TrackerTFP/interface/TrackQuality.h"
0018 #include "L1Trigger/TrackerTFP/interface/TrackFindingProcessor.h"
0019
0020 #include <string>
0021 #include <numeric>
0022 #include <vector>
0023 #include <deque>
0024
0025 namespace trackerTFP {
0026
0027   /*! \class  trackerTFP::ProducerTFP
0028    *  \brief  L1TrackTrigger final TFP output formatter
0029    *  \author Thomas Schuh
0030    *  \date   2023, June
0031    */
0032   class ProducerTFP : public edm::stream::EDProducer<> {
0033   public:
0034     explicit ProducerTFP(const edm::ParameterSet&);
0035     ~ProducerTFP() override = default;
0036
0037   private:
0038     void produce(edm::Event&, const edm::EventSetup&) override;
0039     // ED input token of stubs and tracks
0040     edm::EDGetTokenT<tt::StreamsTrack> edGetTokenTracks_;
0041     edm::EDGetTokenT<tt::StreamsStub> edGetTokenStubs_;
0042     // ED output token for accepted stubs and tracks
0043     edm::EDPutTokenT<tt::TTTracks> edPutTokenTTTracks_;
0044     edm::EDPutTokenT<tt::StreamsTrack> edPutTokenTracks_;
0045     // Setup token
0046     edm::ESGetToken<tt::Setup, tt::SetupRcd> esGetTokenSetup_;
0047     // DataFormats token
0048     edm::ESGetToken<DataFormats, DataFormatsRcd> esGetTokenDataFormats_;
0049     // TrackQuality token
0050     edm::ESGetToken<TrackQuality, DataFormatsRcd> esGetTokenTrackQuality_;
0051     // helper class to store configurations
0052     const tt::Setup* setup_ = nullptr;
0053     // helper class to extract structured data from tt::Frames
0054     const DataFormats* dataFormats_ = nullptr;
0055     // helper class to determine track quality
0056     const TrackQuality* trackQuality_ = nullptr;
0057   };
0058
0059   ProducerTFP::ProducerTFP(const edm::ParameterSet& iConfig) {
0060     const std::string& labelTracks = iConfig.getParameter<std::string>("InputLabelTFP");
0061     const std::string& labelStubs = iConfig.getParameter<std::string>("InputLabelTQ");
0062     const std::string& branchTracks = iConfig.getParameter<std::string>("BranchTracks");
0063     const std::string& branchTTTracks = iConfig.getParameter<std::string>("BranchTTTracks");
0064     const std::string& branchStubs = iConfig.getParameter<std::string>("BranchStubs");
0065     // book in- and output ED products
0066     edGetTokenTracks_ = consumes<tt::StreamsTrack>(edm::InputTag(labelTracks, branchTracks));
0067     edGetTokenStubs_ = consumes<tt::StreamsStub>(edm::InputTag(labelStubs, branchStubs));
0068     edPutTokenTTTracks_ = produces<tt::TTTracks>(branchTTTracks);
0069     edPutTokenTracks_ = produces<tt::StreamsTrack>(branchTracks);
0070     // book ES products
0071     esGetTokenSetup_ = esConsumes();
0072     esGetTokenDataFormats_ = esConsumes();
0073     esGetTokenTrackQuality_ = esConsumes();
0074   }
0075
0076   void ProducerTFP::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
0077     // helper class to store configurations
0078     const tt::Setup* setup = &iSetup.getData(esGetTokenSetup_);
0079     // helper class to extract structured data from tt::Frames
0080     const DataFormats* dataFormats = &iSetup.getData(esGetTokenDataFormats_);
0081     // helper class to determine track quality
0082     const TrackQuality* trackQuality = &iSetup.getData(esGetTokenTrackQuality_);
0083     // empty TFP products
0084     tt::TTTracks ttTracks;
0085     tt::StreamsTrack streamsTrack(setup->numRegions() * setup->tfpNumChannel());
0086     // read in TQ Products
0087     const tt::StreamsTrack& tracks = iEvent.get(edGetTokenTracks_);
0088     const tt::StreamsStub& stubs = iEvent.get(edGetTokenStubs_);
0089     // produce TTTracks
0090     TrackFindingProcessor tfp(setup, dataFormats, trackQuality);
0091     tfp.produce(tracks, stubs, ttTracks, streamsTrack);
0092     // put TTTRacks and produce TTTRackRefs
0093     const int nTrks = ttTracks.size();
0094     const edm::OrphanHandle<tt::TTTracks> oh = iEvent.emplace(edPutTokenTTTracks_, std::move(ttTracks));
0095     std::vector<TTTrackRef> ttTrackRefs;
0096     ttTrackRefs.reserve(nTrks);
0097     for (int iTrk = 0; iTrk < nTrks; iTrk++)
0098       ttTrackRefs.emplace_back(oh, iTrk);
0099     // replace old TTTrackRefs in streamsTrack with new TTTrackRefs
0100     tfp.produce(ttTrackRefs, streamsTrack);
0101     // put StreamsTrack
0102     iEvent.emplace(edPutTokenTracks_, std::move(streamsTrack));
0103   }
0104
0105 }  // namespace trackerTFP
0106
0107 DEFINE_FWK_MODULE(trackerTFP::ProducerTFP);