Added back inverse beta switch

cms-sw · Apr 14, 2020 · 9347528 · 9347528
1 parent 366abb2
commit 9347528
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 9 deletions.
diff --git a/PhysicsTools/PatAlgos/plugins/PATMuonProducer.cc b/PhysicsTools/PatAlgos/plugins/PATMuonProducer.cc
@@ -109,8 +109,11 @@ PATMuonProducer::PATMuonProducer(const edm::ParameterSet& iConfig, PATMuonHeavyO
   embedTpfmsMuon_ = iConfig.getParameter<bool>("embedTpfmsMuon");
   embedDytMuon_ = iConfig.getParameter<bool>("embedDytMuon");
   // embedding of inverse beta variable information
-  muonTimeExtraToken_ =
-      consumes<edm::ValueMap<reco::MuonTimeExtra>>(iConfig.getParameter<edm::InputTag>("sourceMuonTimeExtra"));
+  addInverseBeta_ = iConfig.getParameter<bool>("addInverseBeta");
+  if (addInverseBeta_) {
+    muonTimeExtraToken_ =
+        consumes<edm::ValueMap<reco::MuonTimeExtra>>(iConfig.getParameter<edm::InputTag>("sourceMuonTimeExtra"));
+  }
   // Monte Carlo matching
   addGenMatch_ = iConfig.getParameter<bool>("addGenMatch");
   if (addGenMatch_) {
@@ -519,8 +522,10 @@ void PATMuonProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
     }
 
     edm::Handle<edm::ValueMap<reco::MuonTimeExtra>> muonsTimeExtra;
-    // get MuonTimerExtra value map
-    iEvent.getByToken(muonTimeExtraToken_, muonsTimeExtra);
+    if (addInverseBeta_) {
+      // get MuonTimerExtra value map
+      iEvent.getByToken(muonTimeExtraToken_, muonsTimeExtra);
+    }
 
     for (edm::View<reco::Muon>::const_iterator itMuon = muons->begin(); itMuon != muons->end(); ++itMuon) {
       // construct the Muon from the ref -> save ref to original object
@@ -612,7 +617,9 @@ void PATMuonProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
           }
         }
       }
-      aMuon.readTimeExtra((*muonsTimeExtra)[muonRef]);
+      if (addInverseBeta_) {
+        aMuon.readTimeExtra((*muonsTimeExtra)[muonRef]);
+      }
       // MC info
       aMuon.initSimInfo();
       if (simInfoIsAvailalbe) {

diff --git a/PhysicsTools/PatAlgos/plugins/PATMuonProducer.h b/PhysicsTools/PatAlgos/plugins/PATMuonProducer.h
@@ -173,7 +173,9 @@ namespace pat {
     bool embedTpfmsMuon_;
     /// embed track from DYT muon fit into the muon
     bool embedDytMuon_;
-    /// input tag for reading timing information
+    /// add combined inverse beta measurement into the muon
+    bool addInverseBeta_;
+    /// input tag for reading inverse beta
     edm::EDGetTokenT<edm::ValueMap<reco::MuonTimeExtra>> muonTimeExtraToken_;
     /// add generator match information
     bool addGenMatch_;

diff --git a/PhysicsTools/PatAlgos/python/producersLayer1/muonProducer_cfi.py b/PhysicsTools/PatAlgos/python/producersLayer1/muonProducer_cfi.py
@@ -71,10 +71,9 @@
         #        deltaR = cms.double(0.3)
         #    )),
     ),
-
-    # Where to read and store from the timing information
+    # Read and store combined inverse beta
+    addInverseBeta    = cms.bool(True),  
     sourceMuonTimeExtra = cms.InputTag("muons","combined"), #Use combined info, not only csc or dt
-
     # mc matching
     addGenMatch   = cms.bool(True),
     embedGenMatch = cms.bool(True),