elif channel=='di-mu':
process.schedule = cms.Schedule(
process.mvaMetInputPath,
process.diMuPath,
process.outpath
)
else:
raise ValueError('unrecognized channel')
### Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
if channel=='tau-mu' or 'all' in channel:
addTauMuOutput(process, debugEventContent, addPreSel=False, oneFile=(channel=='all'))
if channel=='tau-ele' or 'all' in channel:
addTauEleOutput(process, debugEventContent, addPreSel=False, oneFile=(channel=='all'))
if channel=='mu-ele' or 'all' in channel:
addMuEleOutput(process, debugEventContent, addPreSel=False, oneFile=(channel=='all'))
if channel=='di-mu' or 'all' in channel:
addDiMuOutput(process, debugEventContent, addPreSel=False, oneFile=(channel=='all'))
if channel=='di-tau' or 'all' in channel:
addDiTauOutput(process, debugEventContent, addPreSel=False, oneFile=(channel=='all'))
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.options = cms.untracked.PSet(wantSummary = cms.untracked.bool(False))
if newSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
def createProcess(runOnMC=True, channel='di-mu', runSVFit=False,
p4TransferFunctionFile='CMGTools/SVfitStandalone/data/svFitVisMassAndPtResolutionPDF.root', # Christians's default. If not touched, it would default to this anyways
integrateOverP4=False):
'''Set up CMSSW process to run MVA MET and SVFit.
Args:
runOnMC (bool): run on MC (access to gen-level products) or data
channel (string): choose from 'tau-mu' 'di-tau' 'tau-ele' 'mu-ele'
'all-separate' 'all'
runSVFit (bool): enables the svfit mass reconstruction used for the
H->tau tau analysis.
'''
sep_line = '-'*70
process = cms.Process("H2TAUTAU")
process.maxEvents = cms.untracked.PSet(input=cms.untracked.int32(-1))
numberOfFilesToProcess = -1
numberOfFilesToProcess = 1
debugEventContent = False
# increase to 1000 before running on the batch, to reduce size of log files
# on your account
reportInterval = 100
print sep_line
print 'channel', channel
print 'runSVFit', runSVFit
# Input & JSON -------------------------------------------------
# dataset_user = '******'
# dataset_name = '/VBF_HToTauTau_M-125_13TeV-powheg-pythia6/Spring14dr-PU20bx25_POSTLS170_V5-v1/AODSIM/SS14/'
# dataset_files = 'miniAOD-prod_PAT_.*root'
if runOnMC:
from CMGTools.H2TauTau.proto.samples.fall15.higgs_susy import HiggsSUSYGG160 as ggh160
process.source = cms.Source(
"PoolSource",
noEventSort = cms.untracked.bool(True),
duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
fileNames = cms.untracked.vstring(ggh160.files)
)
else:
# from CMGTools.RootTools.samples.samples_13TeV_DATA2015 import SingleMuon_Run2015D_Promptv4
process.source = cms.Source(
"PoolSource",
noEventSort = cms.untracked.bool(True),
duplicateCheckMode = cms.untracked.string("noDuplicateCheck"),
fileNames = cms.untracked.vstring('root://eoscms.cern.ch//eos/cms/store/data/Run2015D/SingleMuon/MINIAOD/PromptReco-v4/000/258/159/00000/6CA1C627-246C-E511-8A6A-02163E014147.root')
)
process.source.inputCommands = cms.untracked.vstring(
'keep *'
)
process.options = cms.untracked.PSet(
allowUnscheduled=cms.untracked.bool(True)
)
process.genEvtWeightsCounter = cms.EDProducer(
'GenEvtWeightCounter',
verbose = cms.untracked.bool(False)
)
if numberOfFilesToProcess > 0:
process.source.fileNames = process.source.fileNames[:numberOfFilesToProcess]
print 'Run on MC?', runOnMC, process.source.fileNames[0]
if not runOnMC:
print 'Running on data, setting up JSON file'
json = setupJSON(process)
# load the channel paths -------------------------------------------
process.load('CMGTools.H2TauTau.h2TauTau_cff')
# JAN: recoil correction disabled for now; reactivate if necessary
# setting up the recoil correction according to the input file
# recoilEnabled = False
# setupRecoilCorrection( process, runOnMC,
# enable=recoilEnabled, is53X=isNewerThan('CMSSW_5_2_X'))
# isEmbedded = setupEmbedding(process, channel)
addAK4 = True
# Adding jet collection
process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_condDBv2_cff')
process.GlobalTag.globaltag = '76X_mcRun2_asymptotic_v12'
if not runOnMC:
process.GlobalTag.globaltag = '76X_dataRun2_v15'
process.load('JetMETCorrections.Configuration.DefaultJEC_cff')
process.load('JetMETCorrections.Configuration.JetCorrectors_cff')
# process.load('PhysicsTools.PatAlgos.slimming.unpackedTracksAndVertices_cfi')
# process.load("Configuration.StandardSequences.Geometry_cff")
# process.load("Configuration.StandardSequences.MagneticField_38T_cff")
# process.load('TrackingTools.TransientTrack.TransientTrackBuilder_cfi')
# process.load('RecoBTag.Configuration.RecoBTag_cff')
if addAK4:
addAK4Jets(process)
process.tauMuSequence.insert(4, process.jetSequenceAK4)
process.tauEleSequence.insert(4, process.jetSequenceAK4)
process.muEleSequence.insert(4, process.jetSequenceAK4)
process.diMuSequence.insert(2, process.jetSequenceAK4)
process.diTauSequence.insert(2, process.jetSequenceAK4)
# if '25ns' in process.source.fileNames[0] or 'mcRun2_asymptotic_v2' in process.source.fileNames[0]:
print 'Using 25 ns MVA MET training'
for mvaMETCfg in [process.mvaMETTauMu, process.mvaMETTauEle, process.mvaMETDiMu, process.mvaMETDiTau, process.mvaMETMuEle]:
mvaMETCfg.inputFileNames = cms.PSet(
U = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru_7_4_X_miniAOD_25NS_July2015.root'),
DPhi = cms.FileInPath('RecoMET/METPUSubtraction/data/gbrphi_7_4_X_miniAOD_25NS_July2015.root'),
CovU1 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru1cov_7_4_X_miniAOD_25NS_July2015.root'),
CovU2 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru2cov_7_4_X_miniAOD_25NS_July2015.root')
)
# OUTPUT definition ----------------------------------------------------------
process.outpath = cms.EndPath()
# JAN: In 2015, we should finally make sure that we apply the correction to all
# generator-matched taus, regardless of the process
# 2012: don't apply Tau ES corrections for data (but do for embedded) or
# processes not containing real taus
# signalTauProcess = (process.source.fileNames[0].find('HToTauTau') != -1) or (process.source.fileNames[0].find('DY') != -1) or isEmbedded
if channel == 'all' or channel == 'all-separate':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.tauMuPath,
process.tauElePath,
process.muElePath,
process.diTauPath,
process.outpath
)
elif channel == 'tau-mu':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.tauMuPath,
process.outpath
)
elif channel == 'tau-ele':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.tauElePath,
process.outpath
)
elif channel == 'di-tau':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.diTauPath,
process.outpath
)
elif channel == 'mu-ele':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.muElePath,
process.outpath
)
elif channel == 'di-mu':
process.schedule = cms.Schedule(
# process.mvaMetInputPath,
process.diMuPath,
process.outpath
)
else:
raise ValueError('unrecognized channel')
if runOnMC:
process.genEvtWeightsCounterPath = cms.Path(process.genEvtWeightsCounter)
process.schedule.insert(0, process.genEvtWeightsCounterPath)
# Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
oneFile = (channel == 'all')
if channel == 'tau-mu' or 'all' in channel:
addTauMuOutput(process, debugEventContent, addPreSel=False, oneFile=oneFile)
if channel == 'tau-ele' or 'all' in channel:
addTauEleOutput(process, debugEventContent, addPreSel=False, oneFile=oneFile)
if channel == 'mu-ele' or 'all' in channel:
addMuEleOutput(process, debugEventContent, addPreSel=False, oneFile=oneFile)
if channel == 'di-mu' or 'all' in channel:
addDiMuOutput(process, debugEventContent, addPreSel=False, oneFile=oneFile)
if channel == 'di-tau' or 'all' in channel:
addDiTauOutput(process, debugEventContent, addPreSel=False, oneFile=oneFile)
if not runOnMC:
process.tauMuSequence.remove(process.cmgTauMuCor)
process.cmgTauMuTauPtSel.src = 'cmgTauMu'
process.tauEleSequence.remove(process.cmgTauEleCor)
process.cmgTauEleTauPtSel.src = 'cmgTauEle'
process.diTauSequence.remove(process.cmgDiTauCor)
process.cmgDiTauTauPtSel.src = 'cmgDiTau'
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.options = cms.untracked.PSet(wantSummary=cms.untracked.bool(False))
if runSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
process.cmgDiTauCorSVFitPreSel.SVFitVersion = 2
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 2
process.cmgDiMuCorSVFitPreSel.SVFitVersion = 2
else:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 0
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 0
process.cmgDiTauCorSVFitPreSel.SVFitVersion = 0
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 0
process.cmgDiMuCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgTauMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
process.cmgTauEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
process.cmgDiTauCorSVFitPreSel.integrateOverP4 = integrateOverP4
process.cmgMuEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgTauMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
process.cmgTauEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
process.cmgDiTauCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
process.cmgMuEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
print sep_line
print 'INPUT:'
print sep_line
print process.source.fileNames
print
if not runOnMC:
print 'json:', json
print
print sep_line
print 'PROCESSING'
print sep_line
print 'runOnMC:', runOnMC
print
return process
def createProcess(
runOnMC=True,
channel='tau-mu',
runSVFit=False,
p4TransferFunctionFile='CMGTools/SVfitStandalone/data/svFitVisMassAndPtResolutionPDF.root', # Christians's default. If not touched, it would default to this anyways
integrateOverP4=False,
scaleTau=0.):
'''Set up CMSSW process to run MVA MET and SVFit.
Args:
runOnMC (bool): run on MC (access to gen-level products) or data
channel (string): choose from 'tau-mu' 'di-tau' 'tau-ele' 'mu-ele'
'all-separate' 'all'
runSVFit (bool): enables the svfit mass reconstruction used for the
H->tau tau analysis.
'''
sep_line = '-' * 70
process = cms.Process("H2TAUTAU")
# Adding jet collection
process.load(
'Configuration.StandardSequences.FrontierConditions_GlobalTag_condDBv2_cff'
)
process.GlobalTag.globaltag = '76X_mcRun2_asymptotic_RunIIFall15DR76_v1'
if not runOnMC:
process.GlobalTag.globaltag = '76X_dataRun2_16Dec2015_v0'
process.load('Configuration.StandardSequences.GeometryRecoDB_cff')
process.load('Configuration.StandardSequences.MagneticField_38T_cff')
runMVAMET(process, jetCollectionPF="patJetsReapplyJEC")
from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cff import patJetCorrFactorsUpdated
process.patJetCorrFactorsReapplyJEC = patJetCorrFactorsUpdated.clone(
src=cms.InputTag("slimmedJets"),
levels=['L1FastJet', 'L2Relative', 'L3Absolute'],
payload='AK4PFchs'
) # Make sure to choose the appropriate levels and payload here!
if not runOnMC:
process.patJetCorrFactorsReapplyJEC.levels += ['L2L3Residual']
from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cff import patJetsUpdated
process.patJetsReapplyJEC = patJetsUpdated.clone(
jetSource=cms.InputTag("slimmedJets"),
jetCorrFactorsSource=cms.VInputTag(
cms.InputTag("patJetCorrFactorsReapplyJEC")))
process.maxEvents = cms.untracked.PSet(input=cms.untracked.int32(-1))
numberOfFilesToProcess = -1
numberOfFilesToProcess = 1
debugEventContent = False
# increase to 1000 before running on the batch, to reduce size of log files
# on your account
reportInterval = 100
print sep_line
print 'channel', channel
print 'runSVFit', runSVFit
# Input & JSON -------------------------------------------------
# dataset_user = '******'
# dataset_name = '/VBF_HToTauTau_M-125_13TeV-powheg-pythia6/Spring14dr-PU20bx25_POSTLS170_V5-v1/AODSIM/SS14/'
# dataset_files = 'miniAOD-prod_PAT_.*root'
if runOnMC:
from CMGTools.H2TauTau.proto.samples.fall15.higgs_susy import HiggsSUSYGG160 as ggh160
process.source = cms.Source(
"PoolSource",
noEventSort=cms.untracked.bool(True),
duplicateCheckMode=cms.untracked.string("noDuplicateCheck"),
fileNames=cms.untracked.vstring(ggh160.files))
else:
# from CMGTools.RootTools.samples.samples_13TeV_DATA2015 import SingleMuon_Run2015D_Promptv4
process.source = cms.Source(
"PoolSource",
noEventSort=cms.untracked.bool(True),
duplicateCheckMode=cms.untracked.string("noDuplicateCheck"),
fileNames=cms.untracked.vstring(
'root://eoscms.cern.ch//eos/cms/store/data/Run2015D/SingleMuon/MINIAOD/16Dec2015-v1/10000/FEA1FD2B-B5A8-E511-85F7-0025907B5048.root'
) # mu-tau
# fileNames = cms.untracked.vstring('root://eoscms.cern.ch//eos/cms/store/data/Run2015D/Tau/MINIAOD/16Dec2015-v1/00000/F8B6DB5A-69B0-E511-96D4-20CF305B0590.root') # tau-tau
)
if runOnMC:
process.genEvtWeightsCounter = cms.EDProducer(
'GenEvtWeightCounter', verbose=cms.untracked.bool(False))
if numberOfFilesToProcess > 0:
process.source.fileNames = process.source.fileNames[:
numberOfFilesToProcess]
print 'Run on MC?', runOnMC, process.source.fileNames[0]
if not runOnMC:
print 'Running on data, setting up JSON file'
json = setupJSON(process)
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.options = cms.untracked.PSet(wantSummary=cms.untracked.bool(True))
process.options = cms.untracked.PSet(
allowUnscheduled=cms.untracked.bool(True))
pickEvents = False
if pickEvents:
process.pickEvents = cms.EDFilter(
"PickEvents",
# the original format to input run/event -based selection is described in :
# DPGAnalysis/Skims/data/listrunev
# and kept as default, for historical reasons
RunEventList=cms.untracked.string(
"CMGTools/H2TauTau/data/eventList.txt"),
# run/lumiSection @json -based input of selection can be toggled (but not used in THIS example)
IsRunLsBased=cms.bool(False),
# json is not used in this example -> list of LS left empty
LuminositySectionsBlockRange=cms.untracked.VLuminosityBlockRange(
()))
# process.load('JetMETCorrections.Configuration.DefaultJEC_cff')
# process.load('JetMETCorrections.Configuration.JetCorrectors_cff')
# # if '25ns' in process.source.fileNames[0] or 'mcRun2_asymptotic_v2' in process.source.fileNames[0]:
# print 'Using 25 ns MVA MET training'
# for mvaMETCfg in [process.mvaMETTauMu, process.mvaMETTauEle, process.mvaMETDiMu, process.mvaMETDiTau, process.mvaMETMuEle]:
# mvaMETCfg.inputFileNames = cms.PSet(
# U = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru_7_4_X_miniAOD_25NS_July2015.root'),
# DPhi = cms.FileInPath('RecoMET/METPUSubtraction/data/gbrphi_7_4_X_miniAOD_25NS_July2015.root'),
# CovU1 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru1cov_7_4_X_miniAOD_25NS_July2015.root'),
# CovU2 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru2cov_7_4_X_miniAOD_25NS_July2015.root')
# )
# load the channel paths -------------------------------------------
process.MVAMET.requireOS = cms.bool(False)
if channel == 'tau-mu':
process.load('CMGTools.H2TauTau.objects.tauMuObjectsMVAMET_cff')
if pickEvents:
process.tauMuPath.insert(0, process.pickEvents)
process.mvaMETTauMu = process.MVAMET.clone()
process.mvaMETTauMu.srcLeptons = cms.VInputTag(
"tauPreSelectionTauMu", "muonPreSelectionTauMu")
process.mvaMETTauMu.MVAMETLabel = cms.string('mvaMETTauMu')
process.cmgTauMu.metCollection = cms.InputTag('mvaMETTauMu',
'mvaMETTauMu')
if not runSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgTauMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgTauMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
if scaleTau:
process.cmgTauMuCor.nSigma = scaleTau
elif channel == 'tau-ele':
process.load('CMGTools.H2TauTau.objects.tauEleObjectsMVAMET_cff')
process.mvaMETTauEle = process.MVAMET.clone()
process.mvaMETTauEle.srcLeptons = cms.VInputTag(
"tauPreSelectionTauEle", "electronPreSelectionTauEle")
process.mvaMETTauEle.MVAMETLabel = cms.string('mvaMETTauEle')
process.cmgTauEle.metCollection = cms.InputTag('mvaMETTauEle',
'mvaMETTauEle')
if not runSVFit:
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgTauEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgTauEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
elif channel == 'mu-ele':
process.load('CMGTools.H2TauTau.objects.muEleObjectsMVAMET_cff')
process.mvaMETMuEle = process.MVAMET.clone()
process.mvaMETMuEle.srcLeptons = cms.VInputTag(
"muonPreSelectionMuEle", "electronPreSelectionMuEle")
process.mvaMETMuEle.MVAMETLabel = cms.string('mvaMETMuEle')
process.cmgMuEle.metCollection = cms.InputTag('mvaMETMuEle',
'mvaMETMuEle')
if not runSVFit:
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 0
else:
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 2
# elif channel == 'mu-ele':
# process.MVAMET.srcLeptons = cms.VInputTag("electronPreSelectionMuEle", "muonPreSelectionMuEle")
# # process.muEleSequence.insert(4, process.MVAMET)
elif channel == 'di-tau':
process.load('CMGTools.H2TauTau.objects.diTauObjectsMVAMET_cff')
process.mvaMETDiTau = process.MVAMET.clone()
process.mvaMETDiTau.srcLeptons = cms.VInputTag("tauPreSelectionDiTau",
"tauPreSelectionDiTau")
process.mvaMETDiTau.MVAMETLabel = cms.string('mvaMETDiTau')
process.cmgDiTau.metCollection = cms.InputTag('mvaMETDiTau',
'mvaMETDiTau')
if not runSVFit:
process.cmgDiTauCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgDiTauCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgDiTauCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# elif channel == 'di-mu':
# process.MVAMET.srcLeptons = cms.VInputTag("muonPreSelectionDiMu", "muonPreSelectionDiMu")
# # process.diMuSequence.insert(2, process.MVAMET)
# OUTPUT definition ----------------------------------------------------------
process.outpath = cms.EndPath()
if runOnMC:
# pass
process.genEvtWeightsCounterPath = cms.Path(
process.genEvtWeightsCounter)
# process.schedule.insert(0, process.genEvtWeightsCounterPath)
# Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
oneFile = (channel == 'all')
if channel == 'tau-mu' or 'all' in channel:
addTauMuOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'tau-ele' or 'all' in channel:
addTauEleOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'mu-ele' or 'all' in channel:
addMuEleOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'di-mu' or 'all' in channel:
addDiMuOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'di-tau' or 'all' in channel:
addDiTauOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if not runOnMC:
if channel == 'tau-mu' or 'all' in channel:
process.tauMuSequence.remove(process.cmgTauMuCor)
process.cmgTauMuTauPtSel.src = 'cmgTauMu'
if channel == 'tau-ele' or 'all' in channel:
process.tauEleSequence.remove(process.cmgTauEleCor)
process.cmgTauEleTauPtSel.src = 'cmgTauEle'
if channel == 'di-tau' or 'all' in channel:
process.diTauSequence.remove(process.cmgDiTauCor)
process.cmgDiTauTauPtSel.src = 'cmgDiTau'
# if runSVFit:
# process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
# process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
# process.cmgDiTauCorSVFitPreSel.SVFitVersion = 2
# process.cmgMuEleCorSVFitPreSel.SVFitVersion = 2
# process.cmgDiMuCorSVFitPreSel.SVFitVersion = 2
# else:
# process.cmgTauMuCorSVFitPreSel.SVFitVersion = 0
# process.cmgTauEleCorSVFitPreSel.SVFitVersion = 0
# process.cmgDiTauCorSVFitPreSel.SVFitVersion = 0
# process.cmgMuEleCorSVFitPreSel.SVFitVersion = 0
# process.cmgDiMuCorSVFitPreSel.SVFitVersion = 0
# if integrateOverP4:
# process.cmgTauMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgTauEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgDiTauCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgMuEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
# if p4TransferFunctionFile:
# process.cmgTauMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgTauEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgDiTauCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgMuEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
print sep_line
print 'INPUT:'
print sep_line
print process.source.fileNames
print
if not runOnMC:
print 'json:', json
print
print sep_line
print 'PROCESSING'
print sep_line
print 'runOnMC:', runOnMC
print
# from FWCore.ParameterSet.Utilities import convertToUnscheduled
# convertToUnscheduled(process)
return process
if runOnMC:
process.genEvtWeightsCounterPath = cms.Path(process.genEvtWeightsCounter)
process.schedule.insert(0, process.genEvtWeightsCounterPath)
if addPuppi:
process.schedule.insert(-1, process.puppiPath)
# Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
if channel == 'tau-mu' or 'all' in channel:
addTauMuOutput(process, debugEventContent, addPreSel=False, oneFile=(channel == 'all'))
if channel == 'tau-ele' or 'all' in channel:
addTauEleOutput(process, debugEventContent, addPreSel=False, oneFile=(channel == 'all'))
if channel == 'mu-ele' or 'all' in channel:
addMuEleOutput(process, debugEventContent, addPreSel=False, oneFile=(channel == 'all'))
if channel == 'di-mu' or 'all' in channel:
addDiMuOutput(process, debugEventContent, addPreSel=False, oneFile=(channel == 'all'))
if channel == 'di-tau' or 'all' in channel:
addDiTauOutput(process, debugEventContent, addPreSel=False, oneFile=(channel == 'all'))
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.options = cms.untracked.PSet(wantSummary=cms.untracked.bool(False))
if runSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
def createProcess(
runOnMC=True,
channel='tau-mu',
runSVFit=False,
runMVAETmiss=False,
# Christians's default. If not touched, it would default to this anyways
p4TransferFunctionFile='CMGTools/SVfitStandalone/data/svFitVisMassAndPtResolutionPDF.root',
integrateOverP4=False,
scaleTau=0.,
recorrectJets=True,
verbose=False):
'''Set up CMSSW process to run MVA MET and SVFit.
Args:
runOnMC (bool): run on MC (access to gen-level products) or data
channel (string): choose from 'tau-mu' 'di-tau' 'tau-ele' 'mu-ele'
'all-separate' 'all'
runSVFit (bool): enables the svfit mass reconstruction used for the
H->tau tau analysis.
'''
sep_line = '-' * 70
process = cms.Process("H2TAUTAU")
addNewTauID(process)
addMETFilters(process)
if recorrectJets:
# Adding jet collection
process.load(
'Configuration.StandardSequences.FrontierConditions_GlobalTag_condDBv2_cff'
)
# Global tags from https://twiki.cern.ch/twiki/bin/view/CMS/JECDataMC 07 Feb 2017
process.GlobalTag.globaltag = '94X_mc2017_realistic_v14'
if not runOnMC:
process.GlobalTag.globaltag = '94X_mc2017_realistic_v14'
process.load('Configuration.StandardSequences.GeometryRecoDB_cff')
process.load('Configuration.StandardSequences.MagneticField_38T_cff')
from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cff import updatedPatJetCorrFactors
process.patJetCorrFactorsReapplyJEC = updatedPatJetCorrFactors.clone(
src=cms.InputTag("slimmedJets"),
levels=['L1FastJet', 'L2Relative', 'L3Absolute'],
payload='AK4PFchs'
) # Make sure to choose the appropriate levels and payload here!
if not runOnMC:
process.patJetCorrFactorsReapplyJEC.levels += ['L2L3Residual']
from PhysicsTools.PatAlgos.producersLayer1.jetUpdater_cff import updatedPatJets
process.patJetsReapplyJEC = updatedPatJets.clone(
jetSource=cms.InputTag("slimmedJets"),
jetCorrFactorsSource=cms.VInputTag(
cms.InputTag("patJetCorrFactorsReapplyJEC")))
if runMVAETmiss:
if recorrectJets:
runMVAMET(process, jetCollectionPF="patJetsReapplyJEC")
else:
runMVAMET(process)
# We always need this
runMetCorAndUncFromMiniAOD(process, isData=not runOnMC)
process.selectedVerticesForPFMEtCorrType0.src = cms.InputTag(
"offlineSlimmedPrimaryVertices")
# loadLocalSqlite(process, 'Spring16_25nsV3_DATA.db') #os.environ['CMSSW_BASE'] + '/src/CMGTools/RootTools/data/jec/'
process.maxEvents = cms.untracked.PSet(input=cms.untracked.int32(-1))
numberOfFilesToProcess = -1
numberOfFilesToProcess = 1
debugEventContent = False
# increase to 1000 before running on the batch, to reduce size of log files
# on your account
reportInterval = 100
print sep_line
print 'channel', channel
print 'runSVFit', runSVFit
# Input & JSON -------------------------------------------------
# dataset_user = '******'
# dataset_name = '/VBF_HToTauTau_M-125_13TeV-powheg-pythia6/Spring14dr-PU20bx25_POSTLS170_V5-v1/AODSIM/SS14/'
# dataset_files = 'miniAOD-prod_PAT_.*root'
if runOnMC:
from CMGTools.H2TauTau.proto.samples.summer16.higgs_susy import HiggsSUSYGG160 as ggh160
process.source = cms.Source(
"PoolSource",
noEventSort=cms.untracked.bool(True),
duplicateCheckMode=cms.untracked.string("noDuplicateCheck"),
fileNames=cms.untracked.vstring(ggh160.files))
else:
# from CMGTools.RootTools.samples.samples_13TeV_DATA2015 import SingleMuon_Run2015D_Promptv4
from CMGTools.H2TauTau.proto.samples.summer16.htt_common import data_single_muon
process.source = cms.Source(
"PoolSource",
noEventSort=cms.untracked.bool(True),
duplicateCheckMode=cms.untracked.string("noDuplicateCheck"),
fileNames=cms.untracked.vstring(
data_single_muon[1].files) # mu-tau
)
if runOnMC:
process.genEvtWeightsCounter = cms.EDProducer(
'GenEvtWeightCounter', verbose=cms.untracked.bool(False))
if numberOfFilesToProcess > 0:
process.source.fileNames = process.source.fileNames[:
numberOfFilesToProcess]
print 'Run on MC?', runOnMC #, process.source.fileNames[0]
# if not runOnMC:
# from CMGTools.H2TauTau.proto.samples.spring16.htt_common import json
# # print 'Running on data, setting up JSON file'
# # json = setupJSON(process)
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.MessageLogger.suppressWarning = cms.untracked.vstring('cmgDiTau')
process.options = cms.untracked.PSet(wantSummary=cms.untracked.bool(True))
process.options = cms.untracked.PSet(
allowUnscheduled=cms.untracked.bool(True))
pickEvents = False
if pickEvents:
process.pickEvents = cms.EDFilter(
"PickEvents",
# the original format to input run/event -based selection is described in :
# DPGAnalysis/Skims/data/listrunev
# and kept as default, for historical reasons
RunEventList=cms.untracked.string(
"CMGTools/H2TauTau/data/eventList.txt"),
# run/lumiSection @json -based input of selection can be toggled (but not used in THIS example)
IsRunLsBased=cms.bool(False),
# json is not used in this example -> list of LS left empty
LuminositySectionsBlockRange=cms.untracked.VLuminosityBlockRange(
()))
# process.load('JetMETCorrections.Configuration.DefaultJEC_cff')
# process.load('JetMETCorrections.Configuration.JetCorrectors_cff')
# # if '25ns' in process.source.fileNames[0] or 'mcRun2_asymptotic_v2' in process.source.fileNames[0]:
# print 'Using 25 ns MVA MET training'
# for mvaMETCfg in [process.mvaMETTauMu, process.mvaMETTauEle, process.mvaMETDiMu, process.mvaMETDiTau, process.mvaMETMuEle]:
# mvaMETCfg.inputFileNames = cms.PSet(
# U = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru_7_4_X_miniAOD_25NS_July2015.root'),
# DPhi = cms.FileInPath('RecoMET/METPUSubtraction/data/gbrphi_7_4_X_miniAOD_25NS_July2015.root'),
# CovU1 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru1cov_7_4_X_miniAOD_25NS_July2015.root'),
# CovU2 = cms.FileInPath('RecoMET/METPUSubtraction/data/gbru2cov_7_4_X_miniAOD_25NS_July2015.root')
# )
# load the channel paths -------------------------------------------
if runMVAETmiss:
process.MVAMET.requireOS = cms.bool(False)
process.MVAMET.srcMETs = cms.VInputTag(
cms.InputTag("slimmedMETs", "", "PAT" if runOnMC else "RECO"),
cms.InputTag("patpfMET"), cms.InputTag("patpfMETT1"),
cms.InputTag("patpfTrackMET"), cms.InputTag("patpfNoPUMET"),
cms.InputTag("patpfPUCorrectedMET"), cms.InputTag("patpfPUMET"),
cms.InputTag("slimmedMETsPuppi", "", "PAT" if runOnMC else "RECO"))
if channel == 'tau-mu':
process.load('CMGTools.H2TauTau.objects.tauMuObjectsMVAMET_cff')
if pickEvents:
process.tauMuPath.insert(0, process.pickEvents)
if runMVAETmiss:
process.mvaMETTauMu = process.MVAMET.clone()
process.mvaMETTauMu.srcLeptons = cms.VInputTag(
"tauPreSelectionTauMu", "muonPreSelectionTauMu")
process.mvaMETTauMu.MVAMETLabel = cms.string('mvaMETTauMu')
process.cmgTauMu.metCollection = cms.InputTag(
'mvaMETTauMu', 'mvaMETTauMu')
else:
process.cmgTauMu.metCollection = cms.InputTag('slimmedMETs')
if not runSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgTauMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgTauMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
if scaleTau:
process.cmgTauMuCor.nSigma = scaleTau
elif channel == 'tau-ele':
process.load('CMGTools.H2TauTau.objects.tauEleObjectsMVAMET_cff')
if runMVAETmiss:
process.mvaMETTauEle = process.MVAMET.clone()
process.mvaMETTauEle.srcLeptons = cms.VInputTag(
"tauPreSelectionTauEle", "electronPreSelectionTauEle")
process.mvaMETTauEle.MVAMETLabel = cms.string('mvaMETTauEle')
process.cmgTauEle.metCollection = cms.InputTag(
'mvaMETTauEle', 'mvaMETTauEle')
else:
process.cmgTauEle.metCollection = cms.InputTag('slimmedMETs')
if not runSVFit:
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgTauEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgTauEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
elif channel == 'mu-ele':
process.load('CMGTools.H2TauTau.objects.muEleObjectsMVAMET_cff')
if runMVAETmiss:
process.mvaMETMuEle = process.MVAMET.clone()
process.mvaMETMuEle.srcLeptons = cms.VInputTag(
"muonPreSelectionMuEle", "electronPreSelectionMuEle")
process.mvaMETMuEle.MVAMETLabel = cms.string('mvaMETMuEle')
process.cmgMuEle.metCollection = cms.InputTag(
'mvaMETMuEle', 'mvaMETMuEle')
else:
process.cmgMuEle.metCollection = cms.InputTag('slimmedMETs')
if not runSVFit:
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 0
else:
process.cmgMuEleCorSVFitPreSel.SVFitVersion = 2
# elif channel == 'mu-ele':
# process.MVAMET.srcLeptons = cms.VInputTag("electronPreSelectionMuEle", "muonPreSelectionMuEle")
# # process.muEleSequence.insert(4, process.MVAMET)
elif channel == 'di-tau':
process.load('CMGTools.H2TauTau.objects.diTauObjectsMVAMET_cff')
if runMVAETmiss:
process.mvaMETDiTau = process.MVAMET.clone()
process.mvaMETDiTau.srcLeptons = cms.VInputTag(
"tauPreSelectionDiTau", "tauPreSelectionDiTau")
process.mvaMETDiTau.MVAMETLabel = cms.string('mvaMETDiTau')
process.cmgDiTau.metCollection = cms.InputTag(
'mvaMETDiTau', 'mvaMETDiTau')
else:
process.cmgDiTau.metCollection = cms.InputTag('slimmedMETs')
if not runSVFit:
process.cmgDiTauCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgDiTauCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgDiTauCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
elif channel == 'di-mu':
process.load('CMGTools.H2TauTau.objects.diMuObjectsMVAMET_cff')
if runMVAETmiss:
process.mvaMETDiMu = process.MVAMET.clone()
process.mvaMETDiMu.srcLeptons = cms.VInputTag(
"muonPreSelectionDiMu", "muonPreSelectionDiMu")
process.mvaMETDiMu.MVAMETLabel = cms.string('mvaMETDiMu')
process.cmgDiMu.metCollection = cms.InputTag(
'mvaMETDiMu', 'mvaMETDiMu')
else:
process.cmgDiMu.metCollection = cms.InputTag('slimmedMETs')
if not runSVFit:
process.cmgDiMuCorSVFitPreSel.SVFitVersion = 0
if integrateOverP4:
process.cmgDiMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
if p4TransferFunctionFile:
process.cmgDiMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# OUTPUT definition ----------------------------------------------------------
process.outpath = cms.EndPath()
if runOnMC:
# pass
process.genEvtWeightsCounterPath = cms.Path(
process.genEvtWeightsCounter)
# process.schedule.insert(0, process.genEvtWeightsCounterPath)
# Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
oneFile = (channel == 'all')
if channel == 'tau-mu' or 'all' in channel:
addTauMuOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'tau-ele' or 'all' in channel:
addTauEleOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'mu-ele' or 'all' in channel:
addMuEleOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'di-mu' or 'all' in channel:
addDiMuOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if channel == 'di-tau' or 'all' in channel:
addDiTauOutput(process,
debugEventContent,
addPreSel=False,
oneFile=oneFile)
if not runOnMC:
if channel == 'tau-mu' or 'all' in channel:
process.tauMuSequence.remove(process.cmgTauMuCor)
process.cmgTauMuTauPtSel.src = 'cmgTauMu'
if channel == 'tau-ele' or 'all' in channel:
process.tauEleSequence.remove(process.cmgTauEleCor)
process.cmgTauEleTauPtSel.src = 'cmgTauEle'
if channel == 'di-tau' or 'all' in channel:
process.diTauSequence.remove(process.cmgDiTauCor)
process.cmgDiTauTauPtSel.src = 'cmgDiTau'
# if runSVFit:
# process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
# process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
# process.cmgDiTauCorSVFitPreSel.SVFitVersion = 2
# process.cmgMuEleCorSVFitPreSel.SVFitVersion = 2
# process.cmgDiMuCorSVFitPreSel.SVFitVersion = 2
# else:
# process.cmgTauMuCorSVFitPreSel.SVFitVersion = 0
# process.cmgTauEleCorSVFitPreSel.SVFitVersion = 0
# process.cmgDiTauCorSVFitPreSel.SVFitVersion = 0
# process.cmgMuEleCorSVFitPreSel.SVFitVersion = 0
# process.cmgDiMuCorSVFitPreSel.SVFitVersion = 0
# if integrateOverP4:
# process.cmgTauMuCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgTauEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgDiTauCorSVFitPreSel.integrateOverP4 = integrateOverP4
# process.cmgMuEleCorSVFitPreSel.integrateOverP4 = integrateOverP4
# if p4TransferFunctionFile:
# process.cmgTauMuCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgTauEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgDiTauCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
# process.cmgMuEleCorSVFitPreSel.p4TransferFunctionFile = p4TransferFunctionFile
if verbose:
print sep_line
print 'INPUT:'
print sep_line
print process.source.fileNames
print
# if not runOnMC:
# print 'json:', json
print
print sep_line
print 'PROCESSING'
print sep_line
print 'runOnMC:', runOnMC
print
return process
elif channel == 'tau-ele':
process.schedule = cms.Schedule(process.mvaMetInputPath,
process.tauElePath, process.outpath)
elif channel == 'di-tau':
process.schedule = cms.Schedule(process.mvaMetInputPath, process.diTauPath,
process.outpath)
else:
raise ValueError('unrecognized channel')
### Enable printouts like this:
# process.cmgTauMuCorSVFitPreSel.verbose = True
if channel == 'tau-mu' or channel == 'all':
addTauMuOutput(process, debugEventContent, addPreSel=False)
if channel == 'tau-ele' or channel == 'all':
addTauEleOutput(process, debugEventContent, addPreSel=False)
if channel == 'mu-ele' or channel == 'all':
addMuEleOutput(process, debugEventContent, addPreSel=False)
if channel == 'di-tau' or channel == 'all':
addDiTauOutput(process, debugEventContent, addPreSel=False)
# Message logger setup.
process.load("FWCore.MessageLogger.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = reportInterval
process.options = cms.untracked.PSet(wantSummary=cms.untracked.bool(False))
if newSVFit:
process.cmgTauMuCorSVFitPreSel.SVFitVersion = 2
process.cmgTauEleCorSVFitPreSel.SVFitVersion = 2
process.cmgDiTauCorSVFitPreSel.SVFitVersion = 2
else: