Complete migration of TrackClassifier to esConsumes()

cms-sw · Aug 17, 2021 · 973fb0f · 973fb0f
1 parent f24ac41
commit 973fb0f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/SimTracker/TrackHistory/interface/TrackClassifier.h b/SimTracker/TrackHistory/interface/TrackClassifier.h
@@ -74,6 +74,7 @@ class TrackClassifier : public TrackCategories {
   edm::Handle<edm::HepMCProduct> mcInformation_;
 
   edm::ESHandle<ParticleDataTable> particleDataTable_;
+  edm::ESGetToken<ParticleDataTable, PDTRecord> particleDataTableToken_;
 
   edm::ESHandle<TransientTrackBuilder> transientTrackBuilder_;
   edm::ESGetToken<TransientTrackBuilder, TransientTrackRecord> transientTrackBuilderToken_;

diff --git a/SimTracker/TrackHistory/src/TrackClassifier.cc b/SimTracker/TrackHistory/src/TrackClassifier.cc
@@ -19,6 +19,7 @@ TrackClassifier::TrackClassifier(edm::ParameterSet const &config, edm::ConsumesC
       tracer_(config, std::move(collector)),
       quality_(config, collector),
       magneticFieldToken_(collector.esConsumes<MagneticField, IdealMagneticFieldRecord>()),
+      particleDataTableToken_(collector.esConsumes()),
       transientTrackBuilderToken_(collector.esConsumes<TransientTrackBuilder, TransientTrackRecord>()),
       tTopoHandToken_(collector.esConsumes<TrackerTopology, TrackerTopologyRcd>()) {
   collector.consumes<edm::HepMCProduct>(hepMCLabel_);
@@ -58,7 +59,7 @@ void TrackClassifier::newEvent(edm::Event const &event, edm::EventSetup const &s
   magneticField_ = setup.getHandle(magneticFieldToken_);
 
   // Get the partivle data table
-  setup.getData(particleDataTable_);
+  particleDataTable_ = setup.getHandle(particleDataTableToken_);
 
   // get the beam spot
   event.getByLabel(beamSpotLabel_, beamSpot_);