00001 // 00002 // modified & integrated by Giovanni Abbiendi 00003 // from code by Arun Luthra: UserCode/luthra/MuonTrackSelector/src/MuonTrackSelector.cc 00004 // 00005 #ifndef MCTruth_TrackerMuonHitExtractor_h 00006 #define MCTruth_TrackerMuonHitExtractor_h 00007 00008 #include <memory> 00009 #include "FWCore/Framework/interface/Event.h" 00010 #include "FWCore/ParameterSet/interface/ParameterSet.h" 00011 #include "DataFormats/MuonReco/interface/MuonFwd.h" 00012 #include "DataFormats/MuonReco/interface/Muon.h" 00013 #include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h" 00014 #include "DataFormats/DTRecHit/interface/DTRecSegment4DCollection.h" 00015 00016 class TrackerMuonHitExtractor { 00017 public: 00018 explicit TrackerMuonHitExtractor(const edm::ParameterSet&); 00019 ~TrackerMuonHitExtractor(); 00020 00021 void init(const edm::Event&, const edm::EventSetup&); 00022 std::vector<const TrackingRecHit *> getMuonHits(const reco::Muon &mu) const ; 00023 private: 00024 edm::Handle<DTRecSegment4DCollection> dtSegmentCollectionH_; 00025 edm::Handle<CSCSegmentCollection> cscSegmentCollectionH_; 00026 00027 edm::InputTag inputDTRecSegment4DCollection_; 00028 edm::InputTag inputCSCSegmentCollection_; 00029 }; 00030 00031 #endif