TICL/python/superclustering_cff.py

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")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
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")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
0028 )
0029
0030 ### Superclustering : 3 options : DNN, Mustache-TICL (from tracksters), Mustache-PF (converting tracksters to PFClusters, default for ticl_v4, enable with modifier for v5)
0031 ticlSuperclusteringTask = cms.Task()
0032
0033 # DNN
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 # Mustache-TICL
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 # Mustache-PF
0048 # (no changes to make)