Added PromptLeptonTagging decorations into HIGG4D1

ATLHLEPTON-360

In order to allow non-overlapping Hmumu and ttHmumu selections on different derivations and also for improving background rejection prompt lepton tagging info is needed in  HIGG4D1.


Former-commit-id: 7c5d593abc9e2086f1f037ca2ef8b154a9de5523

PhysicsAnalysis/DerivationFramework/DerivationFrameworkHiggs/python/HIGG4DxAugmentation.py

@@ -14,6 +14,9 @@ DFisMC = (globalflags.DataSource()=='geant4')
 ## add LHE3 weights metadata
 from DerivationFrameworkCore.LHE3WeightMetadata import *
 
+## jet tag nonprompt lepton
+import JetTagNonPromptLepton.JetTagNonPromptLeptonConfig as JetTagConfig
+
 def setup(HIGG4DxName, ToolSvc):
 
     augmentationTools=[]
@@ -129,3 +132,14 @@ def addVRJetsAndBTagging(HIGG4DxName, sequence):
     # alias for VR
     BTaggingFlags.CalibrationChannelAliases += ["AntiKtVR30Rmax4Rmin02Track->AntiKtVR30Rmax4Rmin02Track,AntiKt4EMTopo"]
     
+#====================================================================
+# JetTagNonPromptLepton decorations. Called from HIGG4D1.py
+#====================================================================
+def addJetTagNonPromptLepton(HIGG4DxName, sequence):
+    
+    # Build AntiKt4PV0TrackJets and run b-tagging
+    JetTagConfig.ConfigureAntiKt4PV0TrackJets(sequence, HIGG4DxName)
+    
+    # Add BDT decoration algs
+    sequence += JetTagConfig.GetDecoratePromptLeptonAlgs()
+    sequence += JetTagConfig.GetDecoratePromptTauAlgs()

PhysicsAnalysis/DerivationFramework/DerivationFrameworkHiggs/python/HIGG4DxSlimming.py

@@ -120,6 +120,12 @@ def setup(HIGG4DxName, HIGG4DxStream, HIGG4DxSlimmingHelper):
     if HIGG4DxName == 'HIGG4D6':
         HIGG4DxSlimmingHelper.ExtraVariables += ExtraContentJets
 
+    if HIGG4DxName == 'HIGG4D1':
+        from HIGG4DxAugmentation import JetTagConfig
+        HIGG4DxSlimmingHelper.ExtraVariables += JetTagConfig.GetExtraPromptVariablesForDxAOD()
+        HIGG4DxSlimmingHelper.ExtraVariables += JetTagConfig.GetExtraPromptTauVariablesForDxAOD()
+    
+
     #extra containers
     if HIGG4DxName in ['HIGG4D2', 'HIGG4D3', 'HIGG4D4', 'HIGG4D5', 'HIGG4D6']:
         HIGG4DxSlimmingHelper.AllVariables += ["LVL1JetRoIs"]

PhysicsAnalysis/DerivationFramework/DerivationFrameworkHiggs/share/HIGG4D1.py

@@ -79,8 +79,8 @@ HIGG4D1Sequence += CfgMgr.DerivationFramework__CommonAugmentation("HIGG4DxCommon
 #skimming
 HIGG4D1Sequence += CfgMgr.DerivationFramework__DerivationKernel(DAOD_StreamID+"SkimmingKernel", SkimmingTools = skimmingTools)
 
-# fat/trimmed jet building (after skimming)
-# not in this derivation
+# NonPrompt Lepton Tagger
+DerivationFrameworkHiggs.HIGG4DxAugmentation.addJetTagNonPromptLepton(DAOD_StreamID, HIGG4D1Sequence)
 
 #thinning
 HIGG4D1Sequence += CfgMgr.DerivationFramework__DerivationKernel(DAOD_StreamID+"Kernel", ThinningTools = thinningTools)