File indexing completed on 2023-03-17 11:19:26
0001
0002
0003
0004
0005
0006
0007 #include "FWCore/Framework/interface/Frameworkfwd.h"
0008 #include "FWCore/Framework/interface/stream/EDProducer.h"
0009 #include "FWCore/Framework/interface/Event.h"
0010 #include "FWCore/Framework/interface/ESHandle.h"
0011 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0012 #include "FWCore/Utilities/interface/InputTag.h"
0013 #include "FWCore/Utilities/interface/ESGetToken.h"
0014 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0015
0016 #include "DataFormats/Common/interface/Handle.h"
0017 #include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
0018 #include "DataFormats/GEMRecHit/interface/GEMSegmentCollection.h"
0019 #include "DataFormats/GEMRecHit/interface/GEMSegment.h"
0020
0021 #include "RecoLocalMuon/GEMSegment/plugins/GEMSegmentBuilder.h"
0022 #include "Geometry/Records/interface/MuonGeometryRecord.h"
0023
0024 class GEMSegmentProducer : public edm::stream::EDProducer<> {
0025 public:
0026
0027 explicit GEMSegmentProducer(const edm::ParameterSet&);
0028
0029 ~GEMSegmentProducer() override {}
0030
0031 void produce(edm::Event&, const edm::EventSetup&) override;
0032
0033 private:
0034 int iev;
0035 edm::EDGetTokenT<GEMRecHitCollection> theGEMRecHitToken;
0036 std::unique_ptr<GEMSegmentBuilder> segmentBuilder_;
0037 edm::ESGetToken<GEMGeometry, MuonGeometryRecord> gemGeomToken_;
0038 };
0039
0040 GEMSegmentProducer::GEMSegmentProducer(const edm::ParameterSet& ps) : iev(0) {
0041 theGEMRecHitToken = consumes<GEMRecHitCollection>(ps.getParameter<edm::InputTag>("gemRecHitLabel"));
0042 segmentBuilder_ = std::make_unique<GEMSegmentBuilder>(ps);
0043 gemGeomToken_ = esConsumes<GEMGeometry, MuonGeometryRecord>();
0044
0045 produces<GEMSegmentCollection>();
0046 }
0047
0048 void GEMSegmentProducer::produce(edm::Event& ev, const edm::EventSetup& setup) {
0049 LogDebug("GEMSegmentProducer") << "start producing segments for " << ++iev << "th event with GEM data";
0050
0051
0052 edm::ESHandle<GEMGeometry> gemg = setup.getHandle(gemGeomToken_);
0053 const GEMGeometry* mgeom = &*gemg;
0054 segmentBuilder_->setGeometry(mgeom);
0055
0056
0057 edm::Handle<GEMRecHitCollection> gemRecHits;
0058 ev.getByToken(theGEMRecHitToken, gemRecHits);
0059
0060
0061 auto oc = std::make_unique<GEMSegmentCollection>();
0062
0063
0064 segmentBuilder_->build(gemRecHits.product(), *oc);
0065
0066
0067 ev.put(std::move(oc));
0068 }
0069
0070 #include "FWCore/Framework/interface/MakerMacros.h"
0071 DEFINE_FWK_MODULE(GEMSegmentProducer);