lambda x: abs(x.pdgId) == 13,
lambda x: x.TMOneStationTightMuonId,
lambda x: x.trackerLayers > 5,
lambda x: x.pixelLayers > 0,
lambda x: abs(x.dxy) < 0.3 and abs(x.dz) < 20.,
])))
#--- Lepton MVA in friend tree
from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend
MODULES.append(('LepMVAFriendTTH', lambda: LepMVAFriend(
(
os.environ["CMSSW_BASE"] +
"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml",
os.environ["CMSSW_BASE"] +
"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml",
),
training="forMoriond16",
label="TTHMoriond16")))
MODULES.append(('LepMVAFriendSUSY', lambda: LepMVAFriend(
(
os.environ["CMSSW_BASE"] +
"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml",
os.environ["CMSSW_BASE"] +
"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml",
),
training="forMoriond16",
label="SUSY")))
MODULES.append(('LepMVAFriendJetLessIVF', lambda: LepMVAFriend((
os.environ["CMSSW_BASE"] +
#from CMGTools.TTHAnalysis.tools.finalMVA_susy_2lss import FinalMVA_SUSY_2LSS
#MODULES.append( ('finalMVA_susy_2lss', FinalMVA_SUSY_2LSS()) )
#from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco_MC
#MODULES.append( ('ttreco_mc', TTEventReco_MC()) )
#from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco
#MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"", "BestBySum4NoTJJb":"_bySum4"})) )
#MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"","ByGuessLL2B":"_byLL"})) )
#from CMGTools.TTHAnalysis.tools.MuonMVAFriend import MuonMVAFriend
#MODULES.append( ('MuMVAId', MuonMVAFriend("BPH", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XBPH_BDTG.weights.xml", label="BPH")) )
#MODULES.append( ('MuMVAId', MuonMVAFriend("BPHCalo", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XBPHCalo_BDTG.weights.xml", label="BPHCalo")) )
#MODULES.append( ('MuMVAId', MuonMVAFriend("Full", "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_muMVAId_v1/train70XFull_BDTG.weights.xml", label="Full")) )
from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend
MODULES.append((
'LepMVAFriend',
LepMVAFriend((
"/afs/cern.ch/user/g/gpetrucc/w/TREES_72X_171214/0_lepMVA_v1/%s_BDTG.weights.xml",
"/afs/cern.ch/user/g/gpetrucc/w/TREES_72X_171214/0_lepMVA_v1/%s_BDTG.weights.xml"
))))
#MODULES.append( ('LepMVAFriend', LepMVAFriend(("/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_lepMVA_v1/SV_%s_BDTG.weights.xml",
# "/afs/cern.ch/work/g/gpetrucc/TREES_70X_240914/0_lepMVA_v1/SV_%s_BDTG.weights.xml",),
# training="muMVAId_SV", label="SV")) )
#from CMGTools.TTHAnalysis.tools.eventVars_MT2 import EventVarsMT2
#MODULES.append( ('MT2', EventVarsMT2()) )
class VariableProducer(Module):
def __init__(self, name, booker, modules):
Module.__init__(self, name, booker)
self._modules = modules
#--- TTH event variables
from CMGTools.TTHAnalysis.tools.eventVars_2lss import EventVars2LSS
MODULES.append( ('ttH2lss', lambda : EventVars2LSS()) )
from CMGTools.TTHAnalysis.tools.kinMVA_2D_2lss_3l import KinMVA_2D_2lss_3l
MODULES.append( ('kinMVA_2D_2lss_3l', lambda : KinMVA_2D_2lss_3l(os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/kinMVA/tth/%s_BDTG.weights.xml")) )
from CMGTools.TTHAnalysis.tools.BDT2_HadTop import BDT2_HadTop
MODULES.append( ('BDT2_HadTop', lambda : BDT2_HadTop(os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/kinMVA/tth/TMVAClassification_BDTG.weights_BDT2.xml")) )
#--- Lepton MVA in friend tree
from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend
MODULES.append( ('LepMVAFriendTTH', lambda: LepMVAFriend((os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml",
os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/tth/%s_BDTG.weights.xml",),
training="forMoriond16", label="TTHMoriond16")) )
MODULES.append( ('LepMVAFriendSUSY', lambda: LepMVAFriend((os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml",
os.environ["CMSSW_BASE"]+"/src/CMGTools/TTHAnalysis/data/leptonMVA/susy/%s_BDTG.weights.xml",),
training="forMoriond16", label="TTZMoriond16")) )
class VariableProducer(Module):
def __init__(self,name,booker,modules):
Module.__init__(self,name,booker)
self._modules = [ (n,m() if type(m) == types.FunctionType else m) for (n,m) in modules ]
def beginJob(self):
self.t = PyTree(self.book("TTree","t","t"))
self.branches = {}
for name,mod in self._modules:
print name
print mod.listBranches()
#MODULES.append( ('3l_mva', FinalMVA_3L()) )
#from CMGTools.TTHAnalysis.tools.bbvars import bbVars
#MODULES.append( ('bbvars', bbVars()) )
#from CMGTools.TTHAnalysis.tools.finalMVA_2lss_2 import FinalMVA_2LSS_2
#MODULES.append( ('finalMVA_2lss_2', FinalMVA_2LSS_2()) )
#from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco_MC
#MODULES.append( ('ttreco_mc', TTEventReco_MC()) )
#from CMGTools.TTHAnalysis.tools.ttbarEventReco_2lss import TTEventReco
#MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"", "BestBySum4NoTJJb":"_bySum4"})) )
#MODULES.append( ('ttreco', TTEventReco(sortersToUse={"BestGuess":"","ByGuessLL2B":"_byLL"})) )
from CMGTools.TTHAnalysis.tools.LepMVAFriend import LepMVAFriend
MODULES.append((
'LepMVAFriend',
LepMVAFriend(
"/afs/cern.ch/user/b/botta/CMGToolsGit/newRecipe/CMSSW_5_3_19/src/CMGTools/TTHAnalysis/macros/leptons"
)))
class VariableProducer(Module):
def __init__(self, name, booker, modules):
Module.__init__(self, name, booker)
self._modules = modules
def beginJob(self):
self.t = PyTree(self.book("TTree", "t", "t"))
self.branches = {}
for name, mod in self._modules:
for B in mod.listBranches():
# don't add the same branch twice
if B in self.branches: