File indexing completed on 2024-10-19 04:58:39
0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoHGCal.TICL.tracksterLinksProducer_cfi import tracksterLinksProducer as _tracksterLinksProducer
0004 from RecoHGCal.TICL.ticlEGammaSuperClusterProducer_cfi import ticlEGammaSuperClusterProducer
0005 from RecoEcal.EgammaClusterProducers.particleFlowSuperClusteringSequence_cff import particleFlowSuperClusterHGCal
0006
0007 from Configuration.ProcessModifiers.ticl_v5_cff import ticl_v5
0008 from Configuration.ProcessModifiers.ticl_superclustering_dnn_cff import ticl_superclustering_dnn
0009 from Configuration.ProcessModifiers.ticl_superclustering_mustache_pf_cff import ticl_superclustering_mustache_pf
0010 from Configuration.ProcessModifiers.ticl_superclustering_mustache_ticl_cff import ticl_superclustering_mustache_ticl
0011
0012 ticlTracksterLinksSuperclusteringDNN = _tracksterLinksProducer.clone(
0013 linkingPSet = cms.PSet(
0014 type=cms.string("SuperClusteringDNN"),
0015 algo_verbosity=cms.int32(0),
0016 onnxModelPath = cms.FileInPath("RecoHGCal/TICL/data/superclustering/supercls_v2p1.onnx"),
0017 nnWorkingPoint=cms.double(0.3),
0018 ),
0019 tracksters_collections = [cms.InputTag("ticlTrackstersCLUE3DHigh")],
0020 )
0021
0022 ticlTracksterLinksSuperclusteringMustache = _tracksterLinksProducer.clone(
0023 linkingPSet = cms.PSet(
0024 type=cms.string("SuperClusteringMustache"),
0025 algo_verbosity=cms.int32(0)
0026 ),
0027 tracksters_collections = [cms.InputTag("ticlTrackstersCLUE3DHigh")],
0028 )
0029
0030
0031 ticlSuperclusteringTask = cms.Task()
0032
0033
0034 _dnn_task = cms.Task(ticlTracksterLinksSuperclusteringDNN)
0035 ticl_superclustering_dnn.toReplaceWith(ticlSuperclusteringTask, _dnn_task)
0036 ticl_superclustering_dnn.toModify(ticlEGammaSuperClusterProducer, ticlSuperClusters=cms.InputTag("ticlTracksterLinksSuperclusteringDNN"))
0037 ticl_superclustering_dnn.toReplaceWith(particleFlowSuperClusterHGCal, ticlEGammaSuperClusterProducer)
0038
0039
0040 _mustache_ticl_task = cms.Task(ticlTracksterLinksSuperclusteringMustache)
0041 ticl_superclustering_mustache_ticl.toReplaceWith(ticlSuperclusteringTask, _mustache_ticl_task)
0042 ticl_superclustering_mustache_ticl.toModify(ticlEGammaSuperClusterProducer,
0043 ticlSuperClusters=cms.InputTag("ticlTracksterLinksSuperclusteringMustache"),
0044 enableRegression=cms.bool(False))
0045 ticl_superclustering_mustache_ticl.toReplaceWith(particleFlowSuperClusterHGCal, ticlEGammaSuperClusterProducer)
0046
0047
0048