GEMSegmentProducer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMSegmentCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMSegment.h"

#include "RecoLocalMuon/GEMSegment/plugins/GEMSegmentBuilder.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"

class GEMSegmentProducer : public edm::stream::EDProducer<> {
public:
  explicit GEMSegmentProducer(const edm::ParameterSet&);
  ~GEMSegmentProducer() override {}
  void produce(edm::Event&, const edm::EventSetup&) override;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  int iev;  // events through
  edm::EDGetTokenT<GEMRecHitCollection> theGEMRecHitToken;
  std::unique_ptr<GEMSegmentBuilder> segmentBuilder_;
  edm::ESGetToken<GEMGeometry, MuonGeometryRecord> gemGeomToken_;
};

GEMSegmentProducer::GEMSegmentProducer(const edm::ParameterSet& ps) : iev(0) {
  theGEMRecHitToken = consumes<GEMRecHitCollection>(ps.getParameter<edm::InputTag>("gemRecHitLabel"));
  segmentBuilder_ = std::make_unique<GEMSegmentBuilder>(ps);  // pass on the Parameter Set
  gemGeomToken_ = esConsumes<GEMGeometry, MuonGeometryRecord>();
  // register what this produces
  produces<GEMSegmentCollection>();
}

void GEMSegmentProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("gemRecHitLabel", edm::InputTag("gemRecHits"));
  GEMSegmentBuilder::fillDescription(desc);
  descriptions.add("gemSegments", desc);
}

void GEMSegmentProducer::produce(edm::Event& ev, const edm::EventSetup& setup) {
  LogDebug("GEMSegmentProducer") << "start producing segments for " << ++iev << "th event with GEM data";

  // find the geometry (& conditions?) for this event & cache it in the builder
  edm::ESHandle<GEMGeometry> gemg = setup.getHandle(gemGeomToken_);
  const GEMGeometry* mgeom = &*gemg;
  segmentBuilder_->setGeometry(mgeom);

  // get the collection of GEMRecHit
  edm::Handle<GEMRecHitCollection> gemRecHits;
  ev.getByToken(theGEMRecHitToken, gemRecHits);

  // create empty collection of Segments
  auto oc = std::make_unique<GEMSegmentCollection>();

  // fill the collection
  segmentBuilder_->build(gemRecHits.product(), *oc);  //@@ FILL oc

  // put collection in event
  ev.put(std::move(oc));
}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(GEMSegmentProducer);