Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-02-14 14:31:41

0001 #ifndef TrackingToolsPatternToolsTrackCollectionTokens_H
0002 #define TrackingToolsPatternToolsTrackCollectionTokens_H
0003 
0004 #include "FWCore/Framework/interface/ConsumesCollector.h"
0005 #include "FWCore/Utilities/interface/InputTag.h"
0006 
0007 #include "DataFormats/TrackReco/interface/TrackFwd.h"
0008 #include "TrackingTools/PatternTools/interface/Trajectory.h"
0009 #include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
0010 
0011 #include "FWCore/Framework/interface/Event.h"
0012 
0013 struct TrackCollectionTokens {
0014   TrackCollectionTokens(edm::InputTag const& tag, edm::ConsumesCollector&& iC)
0015       : hTrackToken_(iC.consumes<reco::TrackCollection>(tag)),
0016         hTrajToken_(iC.mayConsume<std::vector<Trajectory>>(tag)),
0017         hIndToken_(iC.mayConsume<std::vector<int>>(tag)),
0018         hTTAssToken_(iC.mayConsume<TrajTrackAssociationCollection>(tag)) {}
0019 
0020   /// source collection label
0021   edm::EDGetTokenT<reco::TrackCollection> hTrackToken_;
0022   edm::EDGetTokenT<std::vector<Trajectory>> hTrajToken_;
0023   edm::EDGetTokenT<std::vector<int>> hIndToken_;
0024   edm::EDGetTokenT<TrajTrackAssociationCollection> hTTAssToken_;
0025 
0026   reco::TrackCollection const& tracks(edm::Event& evt) const {
0027     edm::Handle<reco::TrackCollection> h;
0028     evt.getByToken(hTrackToken_, h);
0029     return *h;
0030   }
0031 
0032   std::vector<Trajectory> const& trajectories(edm::Event& evt) const {
0033     edm::Handle<std::vector<Trajectory>> h;
0034     evt.getByToken(hTrajToken_, h);
0035     return *h;
0036   }
0037 
0038   std::vector<int> const& indicesInput(edm::Event& evt) const {
0039     edm::Handle<std::vector<int>> h;
0040     evt.getByToken(hIndToken_, h);
0041     return *h;
0042   }
0043 };
0044 
0045 #endif