RecoHGCal/Configuration/python/RecoHGCal_EventContent_cff.py - master - githubmirror/cmssw

RecoHGCal/Configuration/python/RecoHGCal_EventContent_cff.py

38 строк2 KB

Felice Pantaleo

pyTICL: generate the TICL Event Content from the assembled graph

29 июн 2026, 23:40

29 июн 2026, 23:4016847af

import FWCore.ParameterSet.Config as cms

# The TICL Event Content is owned and generated by pyTICL: the modules the
# pyTICL graph assembles define what is kept, so the reconstruction keep-
# statements stay in sync with the configuration with no hand-edited list.
# See RecoTICL/Configuration/python/{event_content.py,ticlEventContent_cff.py}.
from RecoTICL.Configuration.ticlEventContent_cff import (
    TICL_AOD, TICL_RECO, TICL_FEVT, TICL_FEVTHLT)


def customiseHGCalOnlyEventContent(process):
    def cleanOutputAndSet(outputModule, ticl_outputCommands):
        outputModule.outputCommands = ['drop *_*_*_*']
        outputModule.outputCommands.extend(ticl_outputCommands)
        outputModule.outputCommands.extend(['keep *_HGCalRecHit_*_*',
                                            'keep *_hgcalMergeLayerClusters_*_*',
                                            'keep CaloParticles_mix_*_*',
                                            'keep SimClusters_mix_*_*',
                                            'keep recoTracks_generalTracks_*_*',
                                            'keep recoTrackExtras_generalTracks_*_*',
                                            'keep SimTracks_g4SimHits_*_*',
                                            'keep SimVertexs_g4SimHits_*_*',
                                            'keep *_layerClusterSimClusterAssociationProducer_*_*',
                                            'keep *_layerClusterCaloParticleAssociationProducer_*_*',
                                            'keep *_randomEngineStateProducer_*_*',
                                            'keep *_layerClusterSimTracksterAssociationProducer_*_*',
                                            'keep *_SimClusterToCaloParticleAssociation*_*_*',
                                            'keep *_simClusterToCaloParticleAssociator*_*_*',
                                            'keep *_SimTau*_*_*',
                                            'keep *_allTrackstersToSimTrackstersAssociations*_*_*'
                                            ])

    if hasattr(process, 'FEVTDEBUGEventContent'):
        cleanOutputAndSet(process.FEVTDEBUGEventContent, TICL_FEVT.outputCommands)
    if hasattr(process, 'FEVTDEBUGHLToutput'):
        cleanOutputAndSet(process.FEVTDEBUGHLToutput, TICL_FEVTHLT.outputCommands)

    return process