Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2024-11-25 02:29:56

0001 # Reconstruction
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 # Validation
0007 from Validation.HGCalValidation.HGCalValidator_cff import *
0008 from RecoLocalCalo.HGCalRecProducers.recHitMapProducer_cfi import recHitMapProducer
0009 
0010 # Load DNN ESSource
0011 from RecoTracker.IterativeTracking.iterativeTk_cff import trackdnn_source
0012 
0013 # Automatic addition of the customisation function from RecoHGCal.Configuration.RecoHGCal_EventContent_cff
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     # TensorFlow ESSource
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 # Reconstruction
0031     process.TICL = cms.Path(process.hgcalLayerClustersTask,
0032                             process.TFESSource,
0033                             process.ticlLayerTileTask,
0034                             process.ticlIterationsTask,
0035                             process.ticlTracksterMergeTask)
0036 # Validation
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 # Path and EndPath definitions
0054     process.FEVTDEBUGHLToutput_step = cms.EndPath(process.FEVTDEBUGHLToutput)
0055     process.DQMoutput_step = cms.EndPath(process.DQMoutput)
0056 
0057 # Schedule definition
0058     process.schedule = cms.Schedule(process.TICL,
0059                                     process.TICL_Validation,
0060                                     process.FEVTDEBUGHLToutput_step,
0061                                     process.DQMoutput_step)
0062 # call to customisation function customiseHGCalOnlyEventContent imported from RecoHGCal.Configuration.RecoHGCal_EventContent_cff
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