customiseL1Ntuple.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms
import os

##############################################################################
# customisations for L1 ntuple generation
#
# Add new customisations to this file!
#
# Example usage :
#   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
#
##############################################################################

def L1NtupleTFileOut(process):

    process.TFileService = cms.Service(
        "TFileService",
        fileName = cms.string('L1Ntuple.root')
    )

    return process

from L1Trigger.L1TNtuples.customiseL1CustomReco import *
        

def L1NtupleAOD(process):
    
    L1NtupleTFileOut(process)
    L1NtupleCustomReco(process)

    process.load('L1Trigger.L1TNtuples.L1NtupleAOD_cff')
    process.l1ntupleaod = cms.Path(
        process.L1NtupleAOD
    )

    process.schedule.append(process.l1ntupleaod)

    return process



def L1NtupleRAW(process):

    L1NtupleTFileOut(process)

    process.load('L1Trigger.L1TNtuples.L1NtupleRAW_cff')
    process.l1ntupleraw = cms.Path(
        process.L1NtupleRAW
    )

    process.schedule.append(process.l1ntupleraw)

    # for 5 BX of candidates in L1Extra
    if "gctDigis" in process.producers:
        process.gctDigis.numberOfGctSamplesToUnpack = cms.uint32(5)

    if "l1extraParticles" in process.producers:
        process.l1extraParticles.centralBxOnly = cms.bool(False)

    return process



def L1NtupleEMU(process):

    L1NtupleTFileOut(process)

    process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
    process.l1ntuplesim = cms.Path(
        process.L1NtupleEMU
    )

    process.schedule.append(process.l1ntuplesim)

    return process


def L1NtupleRAWEMU(process):

    L1NtupleRAW(process)
    L1NtupleEMU(process)

    return process


def L1NtupleAODRAW(process):

    L1NtupleRAW(process)
    L1NtupleAOD(process)

    return process


def L1NtupleAODRAWEMU(process):

    L1NtupleRAW(process)
    L1NtupleEMU(process)
    L1NtupleAOD(process)

    return process

def L1NtupleAODEMU(process):

    L1NtupleEMU(process)
    L1NtupleAOD(process)

    return process

def L1NtupleEMUNoEventTree(process):

    L1NtupleTFileOut(process)

    process.load('L1Trigger.L1TNtuples.L1NtupleEMU_cff')
    process.L1NtupleEMU = cms.Sequence( process.l1CaloTowerEmuTree+process.l1UpgradeEmuTree )
    process.l1ntuplesim = cms.Path(
        process.L1NtupleEMU
    )
    process.schedule.append(process.l1ntuplesim)

    return process


def L1NtupleMC(process):

    # add gen tree
    process.load('L1Trigger.L1TNtuples.L1NtupleMC_cff')

    if hasattr(process, 'l1JetRecoTree'):
        process.l1JetRecoTree.jecToken = cms.untracked.InputTag("ak4PFCHSL1FastL2L3Corrector")

    return process