-
Notifications
You must be signed in to change notification settings - Fork 4.3k
/
Copy pathHLTHgcalTiclPFClusteringForEgammaUnseededSequence_cfi.py
154 lines (142 loc) · 9.3 KB
/
HLTHgcalTiclPFClusteringForEgammaUnseededSequence_cfi.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
149
150
151
152
153
154
import FWCore.ParameterSet.Config as cms
from ..modules.hltFilteredLayerClustersCLUE3DHigh_cfi import *
from ..modules.hltHgcalDigis_cfi import *
from ..modules.hltHgcalLayerClustersEE_cfi import *
from ..modules.hltHgcalLayerClustersHSci_cfi import *
from ..modules.hltHgcalLayerClustersHSi_cfi import *
from ..modules.hltHgcalMergeLayerClusters_cfi import *
from ..modules.hltHGCalRecHit_cfi import *
from ..modules.hltHGCalUncalibRecHit_cfi import *
from ..modules.hltParticleFlowClusterHGCalFromTICLUnseeded_cfi import *
from ..modules.hltParticleFlowRecHitHGC_cfi import *
from ..modules.hltParticleFlowSuperClusterHGCalFromTICLUnseeded_cfi import *
from ..modules.hltTiclLayerTileProducer_cfi import *
from ..modules.hltTiclSeedingGlobal_cfi import *
from ..modules.hltTiclTrackstersCLUE3DHigh_cfi import *
from ..modules.hltHgcalSoARecHitsProducer_cfi import *
from ..modules.hltHgcalSoARecHitsLayerClustersProducer_cfi import *
from ..modules.hltHgcalSoALayerClustersProducer_cfi import *
from ..modules.hltHgcalLayerClustersFromSoAProducer_cfi import *
from ..modules.hltTiclTracksterLinksUnseeded_cfi import *
_HgcalLocalRecoUnseededSequence = cms.Sequence(hltHgcalDigis+hltHGCalUncalibRecHit+hltHGCalRecHit+hltParticleFlowRecHitHGC+hltHgcalLayerClustersEE+hltHgcalLayerClustersHSci+hltHgcalLayerClustersHSi+hltHgcalMergeLayerClusters)
_HgcalTICLPatternRecognitionUnseededSequence = cms.Sequence(hltFilteredLayerClustersCLUE3DHigh+hltTiclSeedingGlobal+hltTiclLayerTileProducer+hltTiclTrackstersCLUE3DHigh)
_SuperclusteringUnseededSequence = cms.Sequence(hltParticleFlowClusterHGCalFromTICLUnseeded+hltParticleFlowSuperClusterHGCalFromTICLUnseeded)
# The baseline sequence
HLTHgcalTiclPFClusteringForEgammaUnseededSequence = cms.Sequence(_HgcalLocalRecoUnseededSequence + _HgcalTICLPatternRecognitionUnseededSequence + _SuperclusteringUnseededSequence)
# Alpaka
from Configuration.ProcessModifiers.alpaka_cff import alpaka
alpaka.toReplaceWith(_HgcalLocalRecoUnseededSequence,
cms.Sequence(
hltHgcalDigis
+ hltHGCalUncalibRecHit
+ hltHGCalRecHit+hltParticleFlowRecHitHGC
+ hltHgcalSoARecHitsProducer
+ hltHgcalSoARecHitsLayerClustersProducer
+ hltHgcalSoALayerClustersProducer
+ hltHgCalLayerClustersFromSoAProducer
+ hltHgcalLayerClustersHSci
+ hltHgcalLayerClustersHSi
+ hltHgcalMergeLayerClusters
)
)
alpaka.toModify(hltHgcalMergeLayerClusters,
layerClustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer"),
time_layerclustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer", "timeLayerCluster"))
# Use EGammaSuperClusterProducer at HLT in ticl v5
hltTiclTracksterLinksSuperclusteringDNNUnseeded = hltTiclTracksterLinksUnseeded.clone(
linkingPSet = cms.PSet(
type=cms.string("SuperClusteringDNN"),
algo_verbosity=cms.int32(0),
onnxModelPath = cms.FileInPath("RecoHGCal/TICL/data/superclustering/supercls_v2p1.onnx"),
nnWorkingPoint=cms.double(0.3),
),
tracksters_collections = [cms.InputTag("hltTiclTrackstersCLUE3DHigh")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
)
hltTiclTracksterLinksSuperclusteringMustacheUnseeded = hltTiclTracksterLinksUnseeded.clone(
linkingPSet = cms.PSet(
type=cms.string("SuperClusteringMustache"),
algo_verbosity=cms.int32(0)
),
tracksters_collections = [cms.InputTag("hltTiclTrackstersCLUE3DHigh")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
)
from RecoHGCal.TICL.ticlEGammaSuperClusterProducer_cfi import ticlEGammaSuperClusterProducer
hltTiclEGammaSuperClusterProducerUnseeded = ticlEGammaSuperClusterProducer.clone()
from Configuration.ProcessModifiers.ticl_superclustering_dnn_cff import ticl_superclustering_dnn
ticl_superclustering_dnn.toReplaceWith(_SuperclusteringUnseededSequence,
cms.Sequence(
hltTiclTracksterLinksSuperclusteringDNNUnseeded
+ hltTiclEGammaSuperClusterProducerUnseeded
)
)
ticl_superclustering_dnn.toModify(hltTiclEGammaSuperClusterProducerUnseeded,
ticlSuperClusters=cms.InputTag("hltTiclTracksterLinksSuperclusteringDNNUnseeded"),
ticlTrackstersEM=cms.InputTag("hltTiclTrackstersCLUE3DHigh"),
layerClusters=cms.InputTag("hltHgcalMergeLayerClusters"))
# Ticl mustache
from Configuration.ProcessModifiers.ticl_superclustering_mustache_ticl_cff import ticl_superclustering_mustache_ticl
ticl_superclustering_mustache_ticl.toReplaceWith(_SuperclusteringUnseededSequence,
cms.Sequence(
hltTiclTracksterLinksSuperclusteringMustacheUnseeded
+ hltTiclEGammaSuperClusterProducerUnseeded
)
)
ticl_superclustering_mustache_ticl.toModify(hltTiclEGammaSuperClusterProducerUnseeded,
ticlSuperClusters=cms.InputTag("hltTiclTracksterLinksSuperclusteringMustacheUnseeded"),
ticlTrackstersEM=cms.InputTag("hltTiclTrackstersCLUE3DHigh"),
layerClusters=cms.InputTag("hltHgcalMergeLayerClusters"))
# Alpaka & DNN
(alpaka & ticl_superclustering_dnn).toReplaceWith(_HgcalLocalRecoUnseededSequence,
cms.Sequence(
hltHgcalDigis
+ hltHGCalUncalibRecHit
+ hltHGCalRecHit+hltParticleFlowRecHitHGC
+ hltHgcalSoARecHitsProducer
+ hltHgcalSoARecHitsLayerClustersProducer
+ hltHgcalSoALayerClustersProducer
+ hltHgCalLayerClustersFromSoAProducer
+ hltHgcalLayerClustersHSci
+ hltHgcalLayerClustersHSi
+ hltHgcalMergeLayerClusters
)
)
(alpaka & ticl_superclustering_dnn).toModify(hltHgcalMergeLayerClusters,
layerClustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer"),
time_layerclustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer", "timeLayerCluster"))
(alpaka & ticl_superclustering_dnn).toReplaceWith(_SuperclusteringUnseededSequence,
cms.Sequence(
hltTiclTracksterLinksSuperclusteringDNNUnseeded
+ hltTiclEGammaSuperClusterProducerUnseeded
)
)
(alpaka & ticl_superclustering_dnn).toModify(hltTiclEGammaSuperClusterProducerUnseeded,
ticlSuperClusters=cms.InputTag("hltTiclTracksterLinksSuperclusteringDNNUnseeded"),
ticlTrackstersEM=cms.InputTag("hltTiclTrackstersCLUE3DHigh"),
layerClusters=cms.InputTag("hltHgcalMergeLayerClusters"))
# Alpaka & ticl mustache
(alpaka & ticl_superclustering_mustache_ticl).toReplaceWith(_HgcalLocalRecoUnseededSequence,
cms.Sequence(
hltHgcalDigis
+ hltHGCalUncalibRecHit
+ hltHGCalRecHit+hltParticleFlowRecHitHGC
+ hltHgcalSoARecHitsProducer
+ hltHgcalSoARecHitsLayerClustersProducer
+ hltHgcalSoALayerClustersProducer
+ hltHgCalLayerClustersFromSoAProducer
+ hltHgcalLayerClustersHSci
+ hltHgcalLayerClustersHSi
+ hltHgcalMergeLayerClusters
)
)
(alpaka & ticl_superclustering_mustache_ticl).toModify(hltHgcalMergeLayerClusters,
layerClustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer"),
time_layerclustersEE = cms.InputTag("hltHgCalLayerClustersFromSoAProducer", "timeLayerCluster"))
(alpaka & ticl_superclustering_mustache_ticl).toReplaceWith(_SuperclusteringUnseededSequence,
cms.Sequence(
hltTiclTracksterLinksSuperclusteringMustacheUnseeded
+ hltTiclEGammaSuperClusterProducerUnseeded
)
)
(alpaka & ticl_superclustering_mustache_ticl).toModify(hltTiclEGammaSuperClusterProducerUnseeded,
ticlSuperClusters=cms.InputTag("hltTiclTracksterLinksSuperclusteringMustacheUnseeded"),
ticlTrackstersEM=cms.InputTag("hltTiclTrackstersCLUE3DHigh"),
layerClusters=cms.InputTag("hltHgcalMergeLayerClusters"))