L1TNtuples/python/customiseL1Ntuple.py

0001 import FWCore.ParameterSet.Config as cms
0002 import os
0003
0004 ##############################################################################
0005 # customisations for L1 ntuple generation
0006 #
0007 # Add new customisations to this file!
0008 #
0009 # Example usage :
0010 #   cmsDriver.py testNtuple -s NONE --customise=L1Trigger/L1TNtuples/customiseL1Ntuple.customiseL1NtupleAOD --conditions=auto:run2_mc_50ns --filein='/store/relval/CMSSW_7_5_0_pre1/RelValProdTTbar_13/AODSIM/MCRUN2_74_V7-v1/00000/48159643-5EE3-E411-818F-0025905A48F0.root' -n 100
0011 #
0012 ##############################################################################
0013
0014 def L1NtupleTFileOut(process):
0015
0016     process.TFileService = cms.Service(
0017         "TFileService",
0018         fileName = cms.string('L1Ntuple.root')
0019     )
0020
0021     return process
0022
0023 from L1Trigger.L1TNtuples.customiseL1CustomReco import *
0024
0025
0026 def L1NtupleAOD(process):
0027
0028     L1NtupleTFileOut(process)
0029     L1NtupleCustomReco(process)
0030
0031     process.load('L1Trigger.L1TNtuples.L1NtupleAOD_cff')
0032     process.l1ntupleaod = cms.Path(
0033         process.L1NtupleAOD
0034     )
0035
0036     process.schedule.append(process.l1ntupleaod)
0037
0038     return process
0039
0040 def L1NtupleAODCalo(process):
0041
0042     L1NtupleTFileOut(process)
0043     L1NtupleCustomReco(process)
0044
0045     process.load('L1Trigger.L1TNtuples.L1NtupleAODCalo_cff')
0046     process.l1ntupleaodcalo = cms.Path(
0047         process.L1NtupleAODCalo
0048     )
0049
0050     process.schedule.append(process.l1ntupleaodcalo)
0051
0052     return process
0053
0054
0055 def L1NtupleAOD_MC(process):
0056
0057     L1NtupleAOD(process)
0058
0059     process.l1JetRecoTree.jecToken = cms.untracked.InputTag("ak4PFCHSL1FastL2L3Corrector")
0060
0061     return process
0062
0063
0064
0065 def L1NtupleRAW(process):
0066
0067     L1NtupleTFileOut(process)
0068
0069     process.load('L1Trigger.L1TNtuples.L1NtupleRAW_cff')
0070     process.l1ntupleraw = cms.Path(
0071         process.L1NtupleRAW
0072     )
0073
0074     process.schedule.append(process.l1ntupleraw)
0075
0076     # for 5 BX of candidates in L1Extra
0077     if "gctDigis" in process.producers:
0078         process.gctDigis.numberOfGctSamplesToUnpack = cms.uint32(5)
0079
0080     if "l1extraParticles" in process.producers:
0081         process.l1extraParticles.centralBxOnly = cms.bool(False)
0082
0083     return process
0084
0085 def L1NtupleNanoDST(process):
0086
0087     L1NtupleTFileOut(process)
0088
0089     process.load('L1Trigger.L1TNtuples.L1NtupleNanoDST_cff')
0090     process.l1ntuplenano = cms.Path(
0091         process.L1NtupleNANO
0092     )
0093
0094     process.schedule.append(process.l1ntuplenano)
0095
0096     # unpack uGT/test crate from the dedicated FED selectors post this update: https://its.cern.ch/jira/browse/CMSHLT-3172
0097     process.gtStage2Digis.InputLabel = cms.InputTag("hltFEDSelectorL1")
0098     process.gtTestcrateStage2Digis.InputLabel = cms.InputTag("hltFEDSelectorL1uGTTest")
0099
0100     return process
0101
0102 def L1NtupleNANO(process):
0103
0104     L1NtupleTFileOut(process)
0105
0106     process.load('L1Trigger.L1TNtuples.L1NtupleNANO_cff')
0107     process.l1ntuplenano = cms.Path(
0108         process.L1NtupleNANO
0109     )
0110
0111     process.schedule.append(process.l1ntuplenano)
0112
0113     return process
0114
0115 def L1NtupleRAWCalo(process):
0116
0117     L1NtupleTFileOut(process)
0118
0119     process.load('L1Trigger.L1TNtuples.L1NtupleRAWCalo_cff')
0120     process.l1ntuplerawcalo = cms.Path(
0121         process.L1NtupleRAWCalo
0122     )
0123
0124     process.schedule.append(process.l1ntuplerawcalo)
0125
0126     return process
0127
0128
0129 def L1NtupleEMU(process):
0130
0131     L1NtupleTFileOut(process)
0132
0133     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0134     process.l1ntupleemu = cms.Path(
0135         process.L1NtupleEMU
0136     )
0137
0138     process.schedule.append(process.l1ntupleemu)
0139
0140     return process
0141
0142 def L1NtupleEMUCalo(process):
0143
0144     L1NtupleTFileOut(process)
0145
0146     process.load('L1Trigger.L1TNtuples.L1NtupleEMUCalo_cff')
0147     process.l1ntupleemucalo = cms.Path(
0148         process.L1NtupleEMUCalo
0149     )
0150
0151     process.schedule.append(process.l1ntupleemucalo)
0152
0153     return process
0154
0155
0156 def L1NtupleEMULegacy(process):
0157
0158     L1NtupleTFileOut(process)
0159
0160     process.load('L1Trigger.L1TNtuples.L1NtupleEMULegacy_cff')
0161     process.l1ntupleemulegacy = cms.Path(
0162         process.L1NtupleEMULegacy
0163     )
0164
0165     process.schedule.append(process.l1ntupleemulegacy)
0166
0167     return process
0168
0169
0170 def L1NtupleGEN(process):
0171
0172     L1NtupleTFileOut(process)
0173
0174     process.load('L1Trigger.L1TNtuples.L1NtupleGEN_cff')
0175     process.l1ntuplegen = cms.Path(
0176         process.L1NtupleGEN
0177     )
0178
0179     process.schedule.append(process.l1ntuplegen)
0180
0181     return process
0182
0183
0184 def L1NtupleRAWEMU(process):
0185
0186     L1NtupleRAW(process)
0187     L1NtupleEMU(process)
0188
0189     return process
0190
0191 def L1NtupleRAWEMUCalo(process):
0192
0193     L1NtupleRAWCalo(process)
0194     L1NtupleEMUCalo(process)
0195
0196     return process
0197
0198 def L1NtupleNANOEMU(process):
0199
0200     L1NtupleNANO(process)
0201     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0202
0203     process.L1NtupleGTEMU = cms.Sequence( process.l1uGTEmuTree )
0204     process.l1ntuplegtemu = cms.Path(
0205         process.L1NtupleGTEMU
0206     )
0207     process.schedule.append(process.l1ntuplegtemu)
0208
0209     return process
0210
0211 def L1NtupleRAWEMULegacy(process):
0212
0213     L1NtupleRAW(process)
0214     L1NtupleEMU(process)
0215     L1NtupleEMULegacy(process)
0216
0217     return process
0218
0219
0220 def L1NtupleAODRAW(process):
0221
0222     L1NtupleRAW(process)
0223     L1NtupleAOD(process)
0224
0225     return process
0226
0227 def L1NtupleAODRAWCalo(process):
0228
0229     L1NtupleRAWCalo(process)
0230     L1NtupleAODCalo(process)
0231
0232     return process
0233
0234 def L1NtupleAODRAWEMULegacy(process):
0235
0236     L1NtupleRAW(process)
0237     L1NtupleEMU(process)
0238     L1NtupleEMULegacy(process)
0239     L1NtupleAOD(process)
0240
0241     return process
0242
0243
0244 def L1NtupleAODRAWEMU(process):
0245
0246     L1NtupleRAW(process)
0247     L1NtupleEMU(process)
0248     L1NtupleAOD(process)
0249
0250     return process
0251
0252 def L1NtupleAODRAWEMUCalo(process):
0253
0254     L1NtupleRAWCalo(process)
0255     L1NtupleEMUCalo(process)
0256     L1NtupleAODCalo(process)
0257
0258     return process
0259
0260 def L1NtupleAODEMU(process):
0261
0262     L1NtupleEMU(process)
0263     L1NtupleAOD(process)
0264
0265     return process
0266
0267 def L1NtupleAODEMUCalo(process):
0268
0269     L1NtupleEMUCalo(process)
0270     L1NtupleAODCalo(process)
0271
0272     return process
0273
0274
0275 def L1NtupleAODEMU_MC(process):
0276
0277     L1NtupleEMU(process)
0278     L1NtupleAOD_MC(process)
0279
0280     return process
0281
0282
0283 def L1NtupleRAWEMUGEN_MC(process):
0284
0285     L1NtupleRAW(process)
0286     L1NtupleEMU(process)
0287     L1NtupleGEN(process)
0288
0289     return process
0290
0291 def L1NtupleAODEMUGEN_MC(process):
0292
0293     L1NtupleEMU(process)
0294     L1NtupleAOD_MC(process)
0295     L1NtupleGEN(process)
0296
0297     return process
0298
0299 def L1NtupleAODRAWEMUGEN_MC(process):
0300
0301     L1NtupleRAW(process)
0302     L1NtupleEMU(process)
0303     L1NtupleAOD_MC(process)
0304     L1NtupleGEN(process)
0305
0306     return process
0307
0308 def L1NtupleEMUNoEventTree(process):
0309
0310     L1NtupleTFileOut(process)
0311
0312     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0313     process.L1NtupleEMU = cms.Sequence( process.l1CaloTowerEmuTree+process.l1UpgradeEmuTree+process.l1UpgradeTfMuonEmuTree )
0314     process.l1ntuplesim = cms.Path(
0315         process.L1NtupleEMU
0316     )
0317     process.schedule.append(process.l1ntuplesim)
0318
0319     return process
0320
0321 def PrefireVetoFilter(process):
0322
0323     process.load('EventFilter.L1TRawToDigi.triggerRulePrefireVetoFilter_cfi')
0324
0325     if hasattr(process, 'l1ntupleraw'):
0326         process.l1ntupleraw.insert(0,process.triggerRulePrefireVetoFilter)
0327     if hasattr(process, 'l1ntupleemu'):
0328         process.l1ntupleemu.insert(0,process.triggerRulePrefireVetoFilter)
0329     if hasattr(process, 'l1ntupleaod'):
0330         process.l1ntupleaod.insert(0,process.triggerRulePrefireVetoFilter)
0331     if hasattr(process, 'l1ntuplerawcalo'):
0332         process.l1ntuplerawcalo.insert(0,process.triggerRulePrefireVetoFilter)
0333     if hasattr(process, 'l1ntupleemucalo'):
0334         process.l1ntupleemucalo.insert(0,process.triggerRulePrefireVetoFilter)
0335     if hasattr(process, 'l1ntupleaodcalo'):
0336         process.l1ntupleaodcalo.insert(0,process.triggerRulePrefireVetoFilter)
0337
0338     return process