TrackFindingTracklet/plugins/ProducerDR.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
0013 #include "L1Trigger/TrackTrigger/interface/Setup.h"
0014 #include "L1Trigger/TrackerTFP/interface/DataFormats.h"
0015 #include "L1Trigger/TrackFindingTracklet/interface/ChannelAssignment.h"
0016 #include "L1Trigger/TrackFindingTracklet/interface/DR.h"
0017 #include "SimDataFormats/Associations/interface/TTTypes.h"
0018
0019 #include <string>
0020 #include <vector>
0021 #include <deque>
0022 #include <iterator>
0023 #include <cmath>
0024 #include <numeric>
0025
0026 using namespace std;
0027 using namespace edm;
0028 using namespace trackerTFP;
0029 using namespace tt;
0030
0031 namespace trklet {
0032
0033   /*! \class  trklet::ProducerDR
0034    *  \brief  Emulates removal of duplicated TTTracks f/w
0035    *  \author Thomas Schuh
0036    *  \date   2023, Feb
0037    */
0038   class ProducerDR : public stream::EDProducer<> {
0039   public:
0040     explicit ProducerDR(const ParameterSet&);
0041     ~ProducerDR() override {}
0042
0043   private:
0044     void beginRun(const Run&, const EventSetup&) override;
0045     void produce(Event&, const EventSetup&) override;
0046     virtual void endJob() {}
0047     // ED input token of Tracks
0048     EDGetTokenT<StreamsTrack> edGetTokenTracks_;
0049     // ED input token of Stubs
0050     EDGetTokenT<StreamsStub> edGetTokenStubs_;
0051     // ED output token for stubs
0052     EDPutTokenT<StreamsStub> edPutTokenAcceptedStubs_;
0053     EDPutTokenT<StreamsStub> edPutTokenLostStubs_;
0054     // ED output token for tracks
0055     EDPutTokenT<StreamsTrack> edPutTokenAcceptedTracks_;
0056     EDPutTokenT<StreamsTrack> edPutTokenLostTracks_;
0057     // Setup token
0058     ESGetToken<Setup, SetupRcd> esGetTokenSetup_;
0059     // DataFormats token
0060     ESGetToken<DataFormats, DataFormatsRcd> esGetTokenDataFormats_;
0061     // ChannelAssignment token
0062     ESGetToken<ChannelAssignment, ChannelAssignmentRcd> esGetTokenChannelAssignment_;
0063     // configuration
0064     ParameterSet iConfig_;
0065     // helper class to store configurations
0066     const Setup* setup_ = nullptr;
0067     // helper class to extract structured data from tt::Frames
0068     const DataFormats* dataFormats_ = nullptr;
0069     // helper class to assign tracks to channel
0070     const ChannelAssignment* channelAssignment_ = nullptr;
0071   };
0072
0073   ProducerDR::ProducerDR(const ParameterSet& iConfig) : iConfig_(iConfig) {
0074     const string& label = iConfig.getParameter<string>("LabelDRin");
0075     const string& branchAcceptedStubs = iConfig.getParameter<string>("BranchAcceptedStubs");
0076     const string& branchAcceptedTracks = iConfig.getParameter<string>("BranchAcceptedTracks");
0077     const string& branchLostStubs = iConfig.getParameter<string>("BranchLostStubs");
0078     const string& branchLostTracks = iConfig.getParameter<string>("BranchLostTracks");
0079     // book in- and output ED products
0080     edGetTokenTracks_ = consumes<StreamsTrack>(InputTag(label, branchAcceptedTracks));
0081     edGetTokenStubs_ = consumes<StreamsStub>(InputTag(label, branchAcceptedStubs));
0082     edPutTokenAcceptedStubs_ = produces<StreamsStub>(branchAcceptedStubs);
0083     edPutTokenAcceptedTracks_ = produces<StreamsTrack>(branchAcceptedTracks);
0084     edPutTokenLostStubs_ = produces<StreamsStub>(branchLostStubs);
0085     edPutTokenLostTracks_ = produces<StreamsTrack>(branchLostTracks);
0086     // book ES products
0087     esGetTokenSetup_ = esConsumes<Setup, SetupRcd, Transition::BeginRun>();
0088     esGetTokenDataFormats_ = esConsumes<DataFormats, DataFormatsRcd, Transition::BeginRun>();
0089     esGetTokenChannelAssignment_ = esConsumes<ChannelAssignment, ChannelAssignmentRcd, Transition::BeginRun>();
0090   }
0091
0092   void ProducerDR::beginRun(const Run& iRun, const EventSetup& iSetup) {
0093     // helper class to store configurations
0094     setup_ = &iSetup.getData(esGetTokenSetup_);
0095     if (!setup_->configurationSupported())
0096       return;
0097     // check process history if desired
0098     if (iConfig_.getParameter<bool>("CheckHistory"))
0099       setup_->checkHistory(iRun.processHistory());
0100     // helper class to extract structured data from tt::Frames
0101     dataFormats_ = &iSetup.getData(esGetTokenDataFormats_);
0102     // helper class to assign tracks to channel
0103     channelAssignment_ = &iSetup.getData(esGetTokenChannelAssignment_);
0104   }
0105
0106   void ProducerDR::produce(Event& iEvent, const EventSetup& iSetup) {
0107     // empty DR products
0108     const int numStreamsTracks = channelAssignment_->numNodesDR() * setup_->numRegions();
0109     const int numStreamsStubs = numStreamsTracks * setup_->numLayers();
0110     StreamsStub acceptedStubs(numStreamsStubs);
0111     StreamsTrack acceptedTracks(numStreamsTracks);
0112     StreamsStub lostStubs(numStreamsStubs);
0113     StreamsTrack lostTracks(numStreamsTracks);
0114     // read in TBout Product and produce KFin product
0115     if (setup_->configurationSupported()) {
0116       Handle<StreamsStub> handleStubs;
0117       iEvent.getByToken<StreamsStub>(edGetTokenStubs_, handleStubs);
0118       const StreamsStub& stubs = *handleStubs;
0119       Handle<StreamsTrack> handleTracks;
0120       iEvent.getByToken<StreamsTrack>(edGetTokenTracks_, handleTracks);
0121       const StreamsTrack& tracks = *handleTracks;
0122       for (int region = 0; region < setup_->numRegions(); region++) {
0123         // object to remove duplicated tracks in a processing region
0124         DR dr(iConfig_, setup_, dataFormats_, channelAssignment_, region);
0125         // read in and organize input tracks and stubs
0126         dr.consume(tracks, stubs);
0127         // fill output products
0128         dr.produce(acceptedStubs, acceptedTracks, lostStubs, lostTracks);
0129       }
0130     }
0131     // store products
0132     iEvent.emplace(edPutTokenAcceptedStubs_, std::move(acceptedStubs));
0133     iEvent.emplace(edPutTokenAcceptedTracks_, std::move(acceptedTracks));
0134     iEvent.emplace(edPutTokenLostStubs_, std::move(lostStubs));
0135     iEvent.emplace(edPutTokenLostTracks_, std::move(lostTracks));
0136   }
0137
0138 }  // namespace trklet
0139
0140 DEFINE_FWK_MODULE(trklet::ProducerDR);