GEMCSCSegmentProducer.cc

Go to the documentation of this file.

#include <RecoLocalMuon/GEMCSCSegment/plugins/GEMCSCSegmentProducer.h>
#include <RecoLocalMuon/GEMCSCSegment/plugins/GEMCSCSegmentBuilder.h>

#include <DataFormats/Common/interface/Handle.h>
#include <FWCore/Framework/interface/ESHandle.h>
#include <FWCore/MessageLogger/interface/MessageLogger.h>

#include <Geometry/Records/interface/MuonGeometryRecord.h>

#include <DataFormats/GEMRecHit/interface/GEMRecHitCollection.h>
#include <DataFormats/GEMRecHit/interface/GEMCSCSegmentCollection.h>
#include <DataFormats/GEMRecHit/interface/GEMCSCSegment.h>

#include <DataFormats/CSCRecHit/interface/CSCRecHit2DCollection.h>
#include <DataFormats/CSCRecHit/interface/CSCSegmentCollection.h>
#include <DataFormats/CSCRecHit/interface/CSCSegment.h>

GEMCSCSegmentProducer::GEMCSCSegmentProducer(const edm::ParameterSet& pas) : iev(0) {
  csc_token = consumes<CSCSegmentCollection>(pas.getParameter<edm::InputTag>("inputObjectsCSC"));
  gem_token = consumes<GEMRecHitCollection>(pas.getParameter<edm::InputTag>("inputObjectsGEM"));
  segmentBuilder_ = new GEMCSCSegmentBuilder(pas);  // pass on the parameterset

  // register what this produces
  produces<GEMCSCSegmentCollection>();
}

GEMCSCSegmentProducer::~GEMCSCSegmentProducer() {
  LogDebug("GEMCSCSegment") << "deleting GEMCSCSegmentBuilder after " << iev << " events w/ gem and csc data.";
  delete segmentBuilder_;
}

void GEMCSCSegmentProducer::produce(edm::Event& ev, const edm::EventSetup& setup) {
  LogDebug("GEMCSCSegment") << "start producing segments for " << ++iev << "th event w/ gem and csc data";

  // find the geometry (& conditions?) for this event & cache it in the builder
  edm::ESHandle<CSCGeometry> cscg;
  setup.get<MuonGeometryRecord>().get(cscg);
  const CSCGeometry* cgeom = &*cscg;

  edm::ESHandle<GEMGeometry> gemg;
  setup.get<MuonGeometryRecord>().get(gemg);
  const GEMGeometry* ggeom = &*gemg;

  // cache the geometry in the builder
  segmentBuilder_->setGeometry(ggeom, cgeom);

  // fill the map with matches between GEM and CSC chambers
  segmentBuilder_->LinkGEMRollsToCSCChamberIndex(ggeom, cgeom);

  // get the collection of CSCSegment and GEMRecHits
  edm::Handle<CSCSegmentCollection> cscSegment;
  ev.getByToken(csc_token, cscSegment);
  edm::Handle<GEMRecHitCollection> gemRecHits;
  ev.getByToken(gem_token, gemRecHits);

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

  // pass the empty collection of GEMCSC Segments and fill it
  segmentBuilder_->build(gemRecHits.product(), cscSegment.product(), *oc);  //@@ FILL oc

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