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 L1NtupleMINI(process):
0103
0104     L1NtupleTFileOut(process)
0105
0106     process.load('L1Trigger.L1TNtuples.L1NtupleMINI_cff')
0107     process.l1ntuplenano = cms.Path(
0108         process.L1NtupleMINI
0109     )
0110     process.schedule.append(process.l1ntuplenano)
0111
0112     return process
0113
0114
0115 def L1NtupleNANO(process):
0116
0117     L1NtupleTFileOut(process)
0118
0119     process.load('L1Trigger.L1TNtuples.L1NtupleNANO_cff')
0120     process.l1ntuplenano = cms.Path(
0121         process.L1NtupleNANO
0122     )
0123
0124     process.schedule.append(process.l1ntuplenano)
0125
0126     return process
0127
0128 def L1NtupleRAWCalo(process):
0129
0130     L1NtupleTFileOut(process)
0131
0132     process.load('L1Trigger.L1TNtuples.L1NtupleRAWCalo_cff')
0133     process.l1ntuplerawcalo = cms.Path(
0134         process.L1NtupleRAWCalo
0135     )
0136
0137     process.schedule.append(process.l1ntuplerawcalo)
0138
0139     return process
0140
0141
0142 def L1NtupleEMU(process):
0143
0144     L1NtupleTFileOut(process)
0145
0146     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0147     process.l1ntupleemu = cms.Path(
0148         process.L1NtupleEMU
0149     )
0150
0151     process.schedule.append(process.l1ntupleemu)
0152
0153     return process
0154
0155 def L1NtupleEMUCalo(process):
0156
0157     L1NtupleTFileOut(process)
0158
0159     process.load('L1Trigger.L1TNtuples.L1NtupleEMUCalo_cff')
0160     process.l1ntupleemucalo = cms.Path(
0161         process.L1NtupleEMUCalo
0162     )
0163
0164     process.schedule.append(process.l1ntupleemucalo)
0165
0166     return process
0167
0168
0169 def L1NtupleEMULegacy(process):
0170
0171     L1NtupleTFileOut(process)
0172
0173     process.load('L1Trigger.L1TNtuples.L1NtupleEMULegacy_cff')
0174     process.l1ntupleemulegacy = cms.Path(
0175         process.L1NtupleEMULegacy
0176     )
0177
0178     process.schedule.append(process.l1ntupleemulegacy)
0179
0180     return process
0181
0182
0183 def L1NtupleGEN(process):
0184
0185     L1NtupleTFileOut(process)
0186
0187     process.load('L1Trigger.L1TNtuples.L1NtupleGEN_cff')
0188     process.l1ntuplegen = cms.Path(
0189         process.L1NtupleGEN
0190     )
0191
0192     process.schedule.append(process.l1ntuplegen)
0193
0194     return process
0195
0196
0197 def L1NtupleRAWEMU(process):
0198
0199     L1NtupleRAW(process)
0200     L1NtupleEMU(process)
0201
0202     return process
0203
0204 def L1NtupleRAWEMUCalo(process):
0205
0206     L1NtupleRAWCalo(process)
0207     L1NtupleEMUCalo(process)
0208
0209     return process
0210
0211 def L1NtupleNANOEMU(process):
0212
0213     L1NtupleNANO(process)
0214     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0215
0216     process.L1NtupleGTEMU = cms.Sequence( process.l1uGTEmuTree )
0217     process.l1ntuplegtemu = cms.Path(
0218         process.L1NtupleGTEMU
0219     )
0220     process.schedule.append(process.l1ntuplegtemu)
0221
0222     return process
0223
0224 def L1NtupleRAWEMULegacy(process):
0225
0226     L1NtupleRAW(process)
0227     L1NtupleEMU(process)
0228     L1NtupleEMULegacy(process)
0229
0230     return process
0231
0232
0233 def L1NtupleAODRAW(process):
0234
0235     L1NtupleRAW(process)
0236     L1NtupleAOD(process)
0237
0238     return process
0239
0240 def L1NtupleAODRAWCalo(process):
0241
0242     L1NtupleRAWCalo(process)
0243     L1NtupleAODCalo(process)
0244
0245     return process
0246
0247 def L1NtupleAODRAWEMULegacy(process):
0248
0249     L1NtupleRAW(process)
0250     L1NtupleEMU(process)
0251     L1NtupleEMULegacy(process)
0252     L1NtupleAOD(process)
0253
0254     return process
0255
0256
0257 def L1NtupleAODRAWEMU(process):
0258
0259     L1NtupleRAW(process)
0260     L1NtupleEMU(process)
0261     L1NtupleAOD(process)
0262
0263     return process
0264
0265 def L1NtupleAODRAWEMUCalo(process):
0266
0267     L1NtupleRAWCalo(process)
0268     L1NtupleEMUCalo(process)
0269     L1NtupleAODCalo(process)
0270
0271     return process
0272
0273 def L1NtupleAODEMU(process):
0274
0275     L1NtupleEMU(process)
0276     L1NtupleAOD(process)
0277
0278     return process
0279
0280 def L1NtupleAODEMUCalo(process):
0281
0282     L1NtupleEMUCalo(process)
0283     L1NtupleAODCalo(process)
0284
0285     return process
0286
0287
0288 def L1NtupleAODEMU_MC(process):
0289
0290     L1NtupleEMU(process)
0291     L1NtupleAOD_MC(process)
0292
0293     return process
0294
0295
0296 def L1NtupleRAWEMUGEN_MC(process):
0297
0298     L1NtupleRAW(process)
0299     L1NtupleEMU(process)
0300     L1NtupleGEN(process)
0301
0302     return process
0303
0304 def L1NtupleAODEMUGEN_MC(process):
0305
0306     L1NtupleEMU(process)
0307     L1NtupleAOD_MC(process)
0308     L1NtupleGEN(process)
0309
0310     return process
0311
0312 def L1NtupleAODRAWEMUGEN_MC(process):
0313
0314     L1NtupleRAW(process)
0315     L1NtupleEMU(process)
0316     L1NtupleAOD_MC(process)
0317     L1NtupleGEN(process)
0318
0319     return process
0320
0321 def L1NtupleEMUNoEventTree(process):
0322
0323     L1NtupleTFileOut(process)
0324
0325     process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
0326     process.L1NtupleEMU = cms.Sequence( process.l1CaloTowerEmuTree+process.l1UpgradeEmuTree+process.l1UpgradeTfMuonEmuTree )
0327     process.l1ntuplesim = cms.Path(
0328         process.L1NtupleEMU
0329     )
0330     process.schedule.append(process.l1ntuplesim)
0331
0332     return process
0333
0334 def PrefireVetoFilter(process):
0335
0336     process.load('EventFilter.L1TRawToDigi.triggerRulePrefireVetoFilter_cfi')
0337
0338     if hasattr(process, 'l1ntupleraw'):
0339         process.l1ntupleraw.insert(0,process.triggerRulePrefireVetoFilter)
0340     if hasattr(process, 'l1ntupleemu'):
0341         process.l1ntupleemu.insert(0,process.triggerRulePrefireVetoFilter)
0342     if hasattr(process, 'l1ntupleaod'):
0343         process.l1ntupleaod.insert(0,process.triggerRulePrefireVetoFilter)
0344     if hasattr(process, 'l1ntuplerawcalo'):
0345         process.l1ntuplerawcalo.insert(0,process.triggerRulePrefireVetoFilter)
0346     if hasattr(process, 'l1ntupleemucalo'):
0347         process.l1ntupleemucalo.insert(0,process.triggerRulePrefireVetoFilter)
0348     if hasattr(process, 'l1ntupleaodcalo'):
0349         process.l1ntupleaodcalo.insert(0,process.triggerRulePrefireVetoFilter)
0350
0351     return process