-
Notifications
You must be signed in to change notification settings - Fork 2
/
produce_ntuple_multialgos_reduced_genmatch_v9_cfg.py
149 lines (130 loc) · 7.07 KB
/
produce_ntuple_multialgos_reduced_genmatch_v9_cfg.py
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
import FWCore.ParameterSet.Config as cms
from Configuration.StandardSequences.Eras import eras
# For old samples use the digi converter
process = cms.Process('DIGI',eras.Phase2C4)
# import of standard configurations
process.load('Configuration.StandardSequences.Services_cff')
process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
process.load('FWCore.MessageService.MessageLogger_cfi')
process.load('Configuration.EventContent.EventContent_cff')
process.load('SimGeneral.MixingModule.mixNoPU_cfi')
process.load('Configuration.Geometry.GeometryExtended2023D35Reco_cff')
process.load('Configuration.Geometry.GeometryExtended2023D35_cff')
process.load('Configuration.StandardSequences.MagneticField_cff')
process.load('Configuration.StandardSequences.Generator_cff')
process.load('IOMC.EventVertexGenerators.VtxSmearedHLLHC14TeV_cfi')
process.load('GeneratorInterface.Core.genFilterSummary_cff')
process.load('Configuration.StandardSequences.SimIdeal_cff')
process.load('Configuration.StandardSequences.Digi_cff')
process.load('Configuration.StandardSequences.SimL1Emulator_cff')
process.load('Configuration.StandardSequences.DigiToRaw_cff')
process.load('Configuration.StandardSequences.EndOfProcess_cff')
process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
process.maxEvents = cms.untracked.PSet(
input = cms.untracked.int32(5)
)
# Input source
process.source = cms.Source("PoolSource",
fileNames = cms.untracked.vstring('/store/mc/PhaseIIMTDTDRAutumn18DR/SingleE_FlatPt-2to100/FEVT/PU200_103X_upgrade2023_realistic_v2-v1/70000/008A0FE7-A01F-CB4B-B101-903A2A27BE2E.root'),
inputCommands=cms.untracked.vstring(
'keep *',
'drop l1tEMTFHit2016Extras_simEmtfDigis_CSC_HLT',
'drop l1tEMTFHit2016Extras_simEmtfDigis_RPC_HLT',
'drop l1tEMTFHit2016s_simEmtfDigis__HLT',
'drop l1tEMTFTrack2016Extras_simEmtfDigis__HLT',
'drop l1tEMTFTrack2016s_simEmtfDigis__HLT',
)
)
process.options = cms.untracked.PSet(
)
# Output definition
process.TFileService = cms.Service(
"TFileService",
fileName = cms.string("ntuple.root")
)
# Other statements
from Configuration.AlCa.GlobalTag import GlobalTag
process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic', '')
# load HGCAL TPG simulation
process.load('L1Trigger.L1THGCal.hgcalTriggerPrimitives_cff')
process.load('L1Trigger.L1THGCalUtilities.HGC3DClusterGenMatchSelector_cff')
process.load('L1Trigger.L1THGCalUtilities.hgcalTriggerNtuples_cff')
process.ntuple_multiclusters.FillLayerInfo = cms.bool(True)
from L1Trigger.L1THGCalUtilities.hgcalTriggerChains import HGCalTriggerChains
import L1Trigger.L1THGCalUtilities.vfe as vfe
import L1Trigger.L1THGCalUtilities.concentrator as concentrator
import L1Trigger.L1THGCalUtilities.clustering2d as clustering2d
import L1Trigger.L1THGCalUtilities.clustering3d as clustering3d
import L1Trigger.L1THGCalUtilities.selectors as selectors
import L1Trigger.L1THGCalUtilities.customNtuples as ntuple
chains = HGCalTriggerChains()
# Register algorithms
## VFE
chains.register_vfe("Floatingpoint8", lambda p : vfe.create_compression(p, 4, 4, True))
chains.register_vfe("Floatingpoint7", lambda p : vfe.create_compression(p, 4, 3, True))
## ECON
chains.register_concentrator("Supertriggercell", concentrator.create_supertriggercell)
chains.register_concentrator("Threshold", concentrator.create_threshold)
chains.register_concentrator("Bestchoice", concentrator.create_bestchoice)
chains.register_concentrator("Supertriggercell4444Fixed", lambda p,i : concentrator.create_supertriggercell(p,i,stcSize=cms.vuint32(4,8,8,8),fixedDataSizePerHGCROC=True ))
chains.register_concentrator("Equalshare", lambda p,i : concentrator.create_supertriggercell(p,i,stcSize=cms.vuint32(4,8,8,8),fixedDataSizePerHGCROC=True,type_energy_division='equalShare'))
chains.register_concentrator("Onebit", concentrator.create_onebitfraction)
## BE1
chains.register_backend1("Ref2d", clustering2d.create_constrainedtopological)
chains.register_backend1("Dummy", clustering2d.create_dummy)
## BE2
chains.register_backend2("Ref3d", clustering3d.create_distance)
### Varies seed threshold
chains.register_backend2("Histomaxth0", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
seed_threshold=0.))
chains.register_backend2("Histomaxth10", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
seed_threshold=10.))
chains.register_backend2("Histomaxth20", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
seed_threshold=20.))
### Varies clustering distance
from L1Trigger.L1THGCal.customClustering import dr_layerbylayer
dr_layerbylayer_25 = [dr*1.25 for dr in dr_layerbylayer]
dr_layerbylayer_50 = [dr*1.5 for dr in dr_layerbylayer]
dr_layerbylayer_100 = [dr*2. for dr in dr_layerbylayer]
chains.register_backend2("Histomaxdr25", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
distances=dr_layerbylayer_25,
seed_threshold=10.))
chains.register_backend2("Histomaxdr50", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
distances=dr_layerbylayer_50,
seed_threshold=10.))
chains.register_backend2("Histomaxdr100", lambda p,i : clustering3d.create_histoMax_variableDr(p,i,
distances=dr_layerbylayer_100,
seed_threshold=10.))
# Register selector
chains.register_selector("Genmatch", selectors.create_genmatch)
# Register ntuples
# Store gen info only in the reference ntuple
ntuple_list_ref = ['event', 'gen', 'multiclusters']
ntuple_list = ['event', 'multiclusters']
chains.register_ntuple("Genclustersntuple", lambda p,i : ntuple.create_ntuple(p,i, ntuple_list_ref))
chains.register_ntuple("Clustersntuple", lambda p,i : ntuple.create_ntuple(p,i, ntuple_list))
# Register trigger chains
## Reference chain
chains.register_chain('Floatingpoint8', "Threshold", 'Ref2d', 'Ref3d', 'Genmatch', 'Genclustersntuple')
## Test impact of 7 bits FP
chains.register_chain('Floatingpoint7', "Threshold", 'Dummy', 'Histomaxth10', 'Genmatch', 'Clustersntuple')
concentrator_algos = ['Supertriggercell', 'Threshold', 'Bestchoice', 'Supertriggercell4444Fixed', 'Equalshare', 'Onebit']
backend_algos = ['Histomaxth0', 'Histomaxth10', 'Histomaxth20',
'Histomaxdr25', 'Histomaxdr50', 'Histomaxdr100']
## Make cross product fo ECON and BE algos
import itertools
for cc,be in itertools.product(concentrator_algos,backend_algos):
chains.register_chain('Floatingpoint8', cc, 'Dummy', be, 'Genmatch', 'Clustersntuple')
process = chains.create_sequences(process)
# Remove towers from sequence
process.hgcalTriggerPrimitives.remove(process.hgcalTowerMap)
process.hgcalTriggerPrimitives.remove(process.hgcalTower)
process.hgcl1tpg_step = cms.Path(process.hgcalTriggerPrimitives)
process.selector_step = cms.Path(process.hgcalTriggerSelector)
process.ntuple_step = cms.Path(process.hgcalTriggerNtuples)
# Schedule definition
process.schedule = cms.Schedule(process.hgcl1tpg_step, process.selector_step, process.ntuple_step)
# Add early deletion of temporary data products to reduce peak memory need
from Configuration.StandardSequences.earlyDeleteSettings_cff import customiseEarlyDelete
process = customiseEarlyDelete(process)
# End adding early deletion