MCTruth/interface/TrackerMuonHitExtractor.h

0001 //
0002 // modified & integrated by Giovanni Abbiendi
0003 // from code by Arun Luthra:
0004 // UserCode/luthra/MuonTrackSelector/src/MuonTrackSelector.cc
0005 //
0006 #ifndef MCTruth_TrackerMuonHitExtractor_h
0007 #define MCTruth_TrackerMuonHitExtractor_h
0008
0009 #include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
0010 #include "DataFormats/DTRecHit/interface/DTRecSegment4DCollection.h"
0011 #include "DataFormats/MuonReco/interface/Muon.h"
0012 #include "DataFormats/MuonReco/interface/MuonFwd.h"
0013 #include "FWCore/Framework/interface/ConsumesCollector.h"
0014 #include "FWCore/Framework/interface/Event.h"
0015 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0016 #include <memory>
0017
0018 class TrackerMuonHitExtractor {
0019 public:
0020   explicit TrackerMuonHitExtractor(const edm::ParameterSet &, edm::ConsumesCollector &&ic);
0021   ~TrackerMuonHitExtractor() = default;
0022
0023   void init(const edm::Event &);
0024   std::vector<const TrackingRecHit *> getMuonHits(const reco::Muon &mu) const;
0025
0026 private:
0027   const edm::EDGetTokenT<DTRecSegment4DCollection> inputDTRecSegment4DToken_;
0028   const edm::EDGetTokenT<CSCSegmentCollection> inputCSCSegmentToken_;
0029 };
0030
0031 #endif