|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
import FWCore.ParameterSet.Config as cms
from PhysicsTools.NanoAOD.nano_cff import nanoMetadata
l1tPh2NanoTask = cms.Task(nanoMetadata)
l1tPh2NanoSequence = cms.Sequence(l1tPh2NanoTask)
### P2GT objects
from DPGAnalysis.Phase2L1TNanoAOD.l1tPh2GTtables_cff import *
def addPh2GTObjects(process):
process.l1tPh2NanoTask.add(p2GTL1TablesTask)
return process
### Main Ph2L1 objects
from DPGAnalysis.Phase2L1TNanoAOD.l1tPh2Nanotables_cff import *
def addPh2L1Objects(process):
process.l1tPh2NanoTask.add(p2L1TablesTask)
return process
#### GENERATOR INFO
## based on https://github.com/cms-sw/cmssw/blob/master/PhysicsTools/NanoAOD/python/nanogen_cff.py#L2-L36
from PhysicsTools.NanoAOD.genparticles_cff import * ## for GenParts
from PhysicsTools.NanoAOD.jetMC_cff import * ## for GenJets
from PhysicsTools.NanoAOD.met_cff import metMCTable ## for GenMET
from PhysicsTools.NanoAOD.globals_cff import puTable ## for PU
from PhysicsTools.NanoAOD.taus_cff import * ## for Gen taus
def addGenObjects(process):
## add more GenVariables
# from L1Ntuple Gen: https://github.com/artlbv/cmssw/blob/94a5ec13b8ce76afb8ea4f157bb92fb547fadee2/L1Trigger/L1TNtuples/plugins/L1GenTreeProducer.cc#L203
genParticleTable.variables.vertX = Var("vertex.x", float, "vertex X")
genParticleTable.variables.vertY = Var("vertex.y", float, "vertex Y")
genParticleTable.variables.vertZ = Var("vertex.z", float, "vertex Z")
genParticleTable.variables.lXY = Var("sqrt(vertex().x() * vertex().x() + vertex().y() * vertex().y())", float, "lXY")
genParticleTable.variables.dXY = Var("-vertex().x() * sin(phi()) + vertex().y() * cos(phi())", float, "dXY")
## add pruned gen particles a la Mini
if False:
## Gen all
# genParticleTable.src = "genParticles" # see
## Mini default, see https://github.com/cms-sw/cmssw/blob/master/PhysicsTools/PatAlgos/python/slimming/prunedGenParticles_cfi.py
# genParticleTable.src = "prunedGenParticles"
## Nano default, see https://github.com/cms-sw/cmssw/blob/master/PhysicsTools/NanoAOD/python/genparticles_cff.py#L8
# genParticleTable.src = "finalGenParticles"
process.prunedGenParticleTable = genParticleTable.clone()
process.prunedGenParticleTable.src = "prunedGenParticles"
process.prunedGenParticleTable.name = "prunedGenPart"
process.l1tPh2NanoTask.add(process.prunedGenParticleTable)
# lower genVisTau pt threshold
process.genVisTauTable.cut = "pt > 1"
# lower AK8 gen jet threshold
process.genJetAK8Table.cut = "pt > 10"
process.l1tPh2NanoTask.add(
puTable, metMCTable,
genParticleTask, genParticleTablesTask,
genTauTask,
)
# add all GenJets: AK4 and AK8
process.l1tPh2NanoTask.add(genJetTable,patJetPartonsNano,genJetFlavourTable)
process.l1tPh2NanoTask.add(genJetAK8Table,genJetAK8FlavourAssociation,genJetAK8FlavourTable)
return process
def addFullPh2L1Nano(process):
addGenObjects(process)
addPh2L1Objects(process)
addPh2GTObjects(process)
return process
|