Add PromptLeptonImprovedVeto to stdm3 for 21 2

Add a flag to save only the BDT outputs of the LeptonTaggers to the leptons.

For the PromptLeptonVeto and the PromptLeptonIso, they are obsolete taggers. To save more space will only save those BDTs' outputs by default.

Add the new tagger PromptLeptonImprovedVeto to the STDM3 derivation.

@fuhe and @rustem will be responsible for this merge request.

PhysicsAnalysis/AnalysisCommon/LeptonTaggers/python/LeptonTaggersConfig.py

@@ -66,10 +66,23 @@ def GetDecoratePromptTauAlgs():
     return algs
 
 #------------------------------------------------------------------------------
-def GetExtraPromptVariablesForDxAOD(name='', addSpectators=False):
+def GetExtraPromptVariablesForDxAOD(name='', addSpectators=False, onlyBDT=True):
 
     prompt_lep_vars = []
 
+    #
+    # Decorate lepton only with the BDT outputs when the onlyBDT flag is true.
+    #
+    if onlyBDT:
+        if name == "" or name == "Electrons":
+            prompt_lep_vars += ["Electrons.PromptLeptonVeto.PromptLeptonIso."]
+
+        if name == "" or name == "Muons":
+            prompt_lep_vars += ["Muons.PromptLeptonVeto.PromptLeptonIso."]
+
+        return prompt_lep_vars
+ 
+ 
     prompt_vars  = "PromptLeptonVeto.PromptLeptonIso."
     prompt_vars += "PromptLeptonInput_TrackJetNTrack.PromptLeptonInput_sv1_jf_ntrkv."
     prompt_vars += "PromptLeptonInput_ip2.PromptLeptonInput_ip3."
@@ -104,10 +117,23 @@ def GetExtraPromptVariablesForDxAOD(name='', addSpectators=False):
 
 
 #------------------------------------------------------------------------------
-def GetExtraImprovedPromptVariablesForDxAOD(name=''):
+def GetExtraImprovedPromptVariablesForDxAOD(name='', onlyBDT=False):
 
     prompt_lep_vars = []
 
+    #
+    # Decorate lepton only with the BDT outputs when the onlyBDT flag is true.
+    #
+    if onlyBDT:
+        if name == "" or name == "Electrons":
+            prompt_lep_vars += ["Electrons.PromptLeptonImprovedVetoBARR.PromptLeptonImprovedVetoECAP."]
+
+        if name == "" or name == "Muons":
+            prompt_lep_vars += ["Muons.PromptLeptonImprovedVeto."]
+
+        return prompt_lep_vars
+ 
+
     prompt_vars  = "PromptLeptonImprovedInput_MVAXBin."
     prompt_vars += "PromptLeptonImprovedInput_PtFrac.PromptLeptonImprovedInput_DRlj."
     prompt_vars += "PromptLeptonImprovedInput_topoetcone30rel.PromptLeptonImprovedInput_ptvarcone30rel."