File indexing completed on 2024-08-09 23:47:45
0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014 #ifndef L1_TRACK_TRIGGER_TRACK_ASSOCIATOR_H
0015 #define L1_TRACK_TRIGGER_TRACK_ASSOCIATOR_H
0016
0017 #include "FWCore/Framework/interface/Frameworkfwd.h"
0018 #include "FWCore/Framework/interface/stream/EDProducer.h"
0019 #include "FWCore/Framework/interface/Event.h"
0020 #include "FWCore/Framework/interface/EventSetup.h"
0021 #include "FWCore/Framework/interface/ESHandle.h"
0022 #include "FWCore/Framework/interface/MakerMacros.h"
0023 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0024
0025 #include "DataFormats/L1TrackTrigger/interface/TTTrack.h"
0026
0027 #include "L1Trigger/TrackTrigger/interface/classNameFinder.h"
0028 #include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
0029 #include "SimDataFormats/Associations/interface/TTStubAssociationMap.h"
0030 #include "SimDataFormats/Associations/interface/TTTrackAssociationMap.h"
0031 #include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
0032 #include "DataFormats/Common/interface/DetSetVectorNew.h"
0033
0034 #include "TTStubAssociator.h"
0035 #include "TTClusterAssociator.h"
0036
0037 #include <memory>
0038 #include <map>
0039 #include <vector>
0040
0041 template <typename T>
0042 class TTTrackAssociator : public edm::stream::EDProducer<> {
0043
0044
0045 public:
0046
0047 explicit TTTrackAssociator(const edm::ParameterSet& iConfig);
0048
0049
0050 ~TTTrackAssociator() override;
0051
0052 private:
0053
0054 std::vector<edm::InputTag> ttTracksInputTags_;
0055
0056 std::vector<edm::EDGetTokenT<std::vector<TTTrack<T> > > > ttTracksTokens_;
0057 edm::EDGetTokenT<TTStubAssociationMap<T> > ttStubTruthToken_;
0058 edm::EDGetTokenT<TTClusterAssociationMap<T> > ttClusterTruthToken_;
0059
0060 bool TTTrackAllowOneFalse2SStub;
0061
0062
0063 void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;
0064
0065 };
0066
0067
0068
0069
0070
0071
0072
0073
0074
0075 template <typename T>
0076 TTTrackAssociator<T>::TTTrackAssociator(const edm::ParameterSet& iConfig) {
0077 ttTracksInputTags_ = iConfig.getParameter<std::vector<edm::InputTag> >("TTTracks");
0078 ttClusterTruthToken_ = consumes<TTClusterAssociationMap<T> >(iConfig.getParameter<edm::InputTag>("TTClusterTruth"));
0079 ttStubTruthToken_ = consumes<TTStubAssociationMap<T> >(iConfig.getParameter<edm::InputTag>("TTStubTruth"));
0080 TTTrackAllowOneFalse2SStub = iConfig.getParameter<bool>("TTTrackAllowOneFalse2SStub");
0081 if (TTTrackAllowOneFalse2SStub) {
0082 edm::LogInfo("TTTrackAssociator< ") << "Allow track if no more than one 2S stub doesn't match truth.";
0083 } else {
0084 edm::LogInfo("TTTrackAssociator< ") << "All 2S stubs must match truth.";
0085 }
0086
0087 for (const auto& iTag : ttTracksInputTags_) {
0088 ttTracksTokens_.push_back(consumes<std::vector<TTTrack<T> > >(iTag));
0089
0090 produces<TTTrackAssociationMap<T> >(iTag.instance());
0091 }
0092
0093 edm::LogInfo("TTStubAssociator< ") << "TTTrackAssociator< " << templateNameFinder<T>() << " > loaded.";
0094 }
0095
0096
0097 template <typename T>
0098 TTTrackAssociator<T>::~TTTrackAssociator() {}
0099
0100
0101 template <>
0102 void TTTrackAssociator<Ref_Phase2TrackerDigi_>::produce(edm::Event& iEvent, const edm::EventSetup& iSetup);
0103
0104 #endif