File indexing completed on 2024-11-25 02:29:56
0001
0002 from RecoHGCal.TICL.iterativeTICL_cff import *
0003 from RecoLocalCalo.HGCalRecProducers.hgcalLayerClusters_cff import hgcalLayerClustersEE, hgcalLayerClustersHSi, hgcalLayerClustersHSci
0004 from RecoLocalCalo.HGCalRecProducers.hgcalMergeLayerClusters_cfi import hgcalMergeLayerClusters
0005 from RecoHGCal.TICL.ticlDumper_cff import ticlDumper
0006
0007 from Validation.HGCalValidation.HGCalValidator_cff import *
0008 from RecoLocalCalo.HGCalRecProducers.recHitMapProducer_cfi import recHitMapProducer
0009
0010
0011 from RecoTracker.IterativeTracking.iterativeTk_cff import trackdnn_source
0012
0013
0014 from RecoHGCal.Configuration.RecoHGCal_EventContent_cff import customiseHGCalOnlyEventContent
0015 from SimCalorimetry.HGCalAssociatorProducers.simTracksterAssociatorByEnergyScore_cfi import simTracksterAssociatorByEnergyScore as simTsAssocByEnergyScoreProducer
0016 from SimCalorimetry.HGCalAssociatorProducers.TSToSimTSAssociation_cfi import allTrackstersToSimTrackstersAssociationsByLCs
0017 from SimCalorimetry.HGCalAssociatorProducers.TSToSimTSAssociationByHits_cfi import allTrackstersToSimTrackstersAssociationsByHits
0018 from SimCalorimetry.HGCalAssociatorProducers.SimClusterToCaloParticleAssociation_cfi import SimClusterToCaloParticleAssociation
0019
0020
0021 def customiseTICLFromReco(process):
0022
0023 process.TFESSource = cms.Task(process.trackdnn_source)
0024
0025 process.hgcalLayerClustersTask = cms.Task(process.hgcalLayerClustersEE,
0026 process.hgcalLayerClustersHSi,
0027 process.hgcalLayerClustersHSci,
0028 process.hgcalMergeLayerClusters)
0029
0030
0031 process.TICL = cms.Path(process.hgcalLayerClustersTask,
0032 process.TFESSource,
0033 process.ticlLayerTileTask,
0034 process.ticlIterationsTask,
0035 process.ticlTracksterMergeTask)
0036
0037 process.TICL_ValidationProducers = cms.Task(process.recHitMapProducer,
0038 process.lcAssocByEnergyScoreProducer,
0039 process.layerClusterCaloParticleAssociationProducer,
0040 process.scAssocByEnergyScoreProducer,
0041 process.layerClusterSimClusterAssociationProducer,
0042 process.simTsAssocByEnergyScoreProducer,
0043 process.simTracksterHitLCAssociatorByEnergyScoreProducer,
0044 process.allTrackstersToSimTrackstersAssociationsByLCs,
0045 process.allTrackstersToSimTrackstersAssociationsByHits,
0046 process.SimClusterToCaloParticleAssociation,
0047 )
0048
0049 process.TICL_Validator = cms.Task(process.hgcalValidator)
0050 process.TICL_Validation = cms.Path(process.TICL_ValidationProducers,
0051 process.TICL_Validator
0052 )
0053
0054 process.FEVTDEBUGHLToutput_step = cms.EndPath(process.FEVTDEBUGHLToutput)
0055 process.DQMoutput_step = cms.EndPath(process.DQMoutput)
0056
0057
0058 process.schedule = cms.Schedule(process.TICL,
0059 process.TICL_Validation,
0060 process.FEVTDEBUGHLToutput_step,
0061 process.DQMoutput_step)
0062
0063 process = customiseHGCalOnlyEventContent(process)
0064
0065 return process
0066
0067
0068 def customiseTICLForDumper(process, histoName="histo.root"):
0069
0070 process.ticlDumper = ticlDumper.clone()
0071
0072 process.TFileService = cms.Service("TFileService",
0073 fileName=cms.string(histoName)
0074 )
0075 process.FEVTDEBUGHLToutput_step = cms.EndPath(
0076 process.FEVTDEBUGHLToutput + process.ticlDumper)
0077 return process