RecoHGCal/TICL/python/superclustering_cff.py - master - githubmirror/cmssw

RecoHGCal/TICL/python/superclustering_cff.py

35 строк2 KB

Felice Pantaleo

Use TICL superclustering producers as the default HLT egamma path

21 апр 2026, 17:44

21 апр 2026, 17:444a2de9f

import FWCore.ParameterSet.Config as cms

from RecoHGCal.TICL.tracksterLinksProducer_cfi import tracksterLinksProducer as _tracksterLinksProducer
from RecoHGCal.TICL.ticlEGammaSuperClusterProducer_cfi import ticlEGammaSuperClusterProducer

from Configuration.ProcessModifiers.ticl_superclustering_mustache_pf_cff import ticl_superclustering_mustache_pf
from Configuration.ProcessModifiers.ticl_superclustering_mustache_ticl_cff import ticl_superclustering_mustache_ticl

ticlTracksterLinksSuperclusteringDNN = _tracksterLinksProducer.clone(
    linkingPSet = cms.PSet(
        type=cms.string("SuperClusteringDNN"),
        algo_verbosity=cms.int32(0),
        onnxModelPath = cms.string("RecoHGCal/TICL/data/superclustering/supercls_v3.onnx"),
        nnWorkingPoint=cms.double(0.57247),
    ),
    tracksters_collections = [cms.InputTag("ticlTrackstersCLUE3DHigh")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
)

ticlTracksterLinksSuperclusteringMustache = _tracksterLinksProducer.clone(
    linkingPSet = cms.PSet(
        type=cms.string("SuperClusteringMustache"),
        algo_verbosity=cms.int32(0)
    ),
    tracksters_collections = [cms.InputTag("ticlTrackstersCLUE3DHigh")], # to be changed to ticlTrackstersCLUE3DEM once separate CLUE3D iterations are introduced
)

### Superclustering : 3 options : DNN, Mustache-TICL (from tracksters)
ticlSuperclusteringTask = cms.Task(ticlTracksterLinksSuperclusteringDNN, ticlEGammaSuperClusterProducer)

# Mustache-TICL
ticl_superclustering_mustache_ticl.toModify(ticlEGammaSuperClusterProducer,
                                            ticlSuperClusters=cms.InputTag("ticlTracksterLinksSuperclusteringMustache"),
                                            enableRegression=cms.bool(False))
_mustache_ticl_task = cms.Task(ticlTracksterLinksSuperclusteringMustache, ticlEGammaSuperClusterProducer)
ticl_superclustering_mustache_ticl.toReplaceWith(ticlSuperclusteringTask, _mustache_ticl_task)