Merge pull request cms-sw#11 from mandrenguyen/bAggTagInfos_132X

add back tag infos
mandrenguyen · Dec 5, 2024 · 625351b · 625351b
2 parents 50605df + cb8f937
commit 625351b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/HeavyIonsAnalysis/Configuration/test/aggregate_MC.py b/HeavyIonsAnalysis/Configuration/test/aggregate_MC.py
@@ -224,3 +224,13 @@
 process.pphfCoincFilter2Th4 = cms.Path(process.phfCoincFilter2Th4)
 process.pAna = cms.EndPath(process.skimanalysis)
 
+
+process.patJetsAK2PFUnsubJets.addBTagInfo = True
+process.patJetsAK2PFUnsubJets.addTagInfos = True
+process.patJetsAK2PFUnsubJets.tagInfoSources = cms.VInputTag(["pfInclusiveSecondaryVertexFinderTagInfos","pfImpactParameterTagInfos"])
+process.akCs2PFJetAnalyzer.doTracks = cms.untracked.bool(True)
+process.akCs2PFJetAnalyzer.ipTagInfoLabel = cms.untracked.string('pfImpactParameter')
+process.akCs2PFJetAnalyzer.svTagInfoLabel = cms.untracked.string('pfInclusiveSecondaryVertexFinder')
+
+process.aggregatedPFCands.ipTagInfoLabel = "pfImpactParameter"
+process.aggregatedPFCands.svTagInfoLabel = "pfInclusiveSecondaryVertexFinder"
diff --git a/HeavyIonsAnalysis/JetAnalysis/python/deepNtupleSettings_cff.py b/HeavyIonsAnalysis/JetAnalysis/python/deepNtupleSettings_cff.py
@@ -90,7 +90,18 @@ def candidateBtaggingMiniAOD(process, isMC = True, jetPtMin = 15, jetCorrLevels
         primaryVertices = "offlineSlimmedPrimaryVertices"
     )
     process.inclusiveCandidateSecondaryVertices = inclusiveCandidateSecondaryVertices.clone()
-    process.svTask = cms.Task(process.inclusiveCandidateVertexFinder, process.candidateVertexMerger, process.candidateVertexArbitrator, process.inclusiveCandidateSecondaryVertices)
+
+    from RecoBTag.ImpactParameter.pfImpactParameterTagInfos_cfi import pfImpactParameterTagInfos
+    from RecoBTag.SecondaryVertex.pfInclusiveSecondaryVertexFinderTagInfos_cfi import pfInclusiveSecondaryVertexFinderTagInfos
+    process.pfImpactParameterTagInfos = pfImpactParameterTagInfos.clone(
+        jets = "ak"+labelR+"PFUnsubJets",
+        candidates = 'packedPFCandidates',
+        primaryVertex = "offlineSlimmedPrimaryVertices",
+        maxDeltaR = jetR
+    )
+    process.pfInclusiveSecondaryVertexFinderTagInfos = pfInclusiveSecondaryVertexFinderTagInfos.clone()
+
+    process.svTask = cms.Task(process.inclusiveCandidateVertexFinder, process.candidateVertexMerger, process.candidateVertexArbitrator, process.inclusiveCandidateSecondaryVertices,process.pfImpactParameterTagInfos, process.pfInclusiveSecondaryVertexFinderTagInfos)
     svSource = cms.InputTag("inclusiveCandidateSecondaryVertices")
 
     # Create unsubtracted reco jets