CMS 3D CMS Logo

Functions
customiseForTICLv5_cff Namespace Reference

Functions

def customiseTICLv5FromReco (process, enableDumper=False)
 

Function Documentation

◆ customiseTICLv5FromReco()

def customiseForTICLv5_cff.customiseTICLv5FromReco (   process,
  enableDumper = False 
)

Definition at line 42 of file customiseForTICLv5_cff.py.

References RecoHGCal_EventContent_cff.customiseForTICLv5EventContent(), and ALPAKA_ACCELERATOR_NAMESPACE::caPixelDoublets.if().

42 def customiseTICLv5FromReco(process, enableDumper = False):
43  # TensorFlow ESSource
44 
45  process.TFESSource = cms.Task(process.trackdnn_source)
46 
47  process.hgcalLayerClustersTask = cms.Task(process.hgcalLayerClustersEE,
48  process.hgcalLayerClustersHSi,
49  process.hgcalLayerClustersHSci,
50  process.hgcalMergeLayerClusters)
51 
52  # Reconstruction
53 
54  process.ticlSimTracksters.computeLocalTime = cms.bool(True)
55 
56  process.ticlTrackstersCLUE3DHigh.pluginPatternRecognitionByCLUE3D.computeLocalTime = cms.bool(True)
57 
58  '''for future CLUE3D separate iterations
59  process.ticlTrackstersCLUE3DHAD.pluginPatternRecognitionByCLUE3D.computeLocalTime = cms.bool(True)
60  process.ticlTrackstersCLUE3DEM.pluginPatternRecognitionByCLUE3D.computeLocalTime = cms.bool(True)
61  '''
62 
63  process.ticlLayerTileTask = cms.Task(ticlLayerTileProducer)
64 
65  process.ticlIterationsTask = cms.Task(
66  process.ticlTrackstersCLUE3DHigh,
67  )
68 
69  process.mtdSoA = _mtdSoAProducer.clone()
70  process.mtdSoATask = cms.Task(process.mtdSoA)
71 
72  process.ticlTracksterLinks = _tracksterLinksProducer.clone()
73  process.ticlTracksterLinks = _tracksterLinksProducer.clone(
74  tracksters_collections = cms.VInputTag(
75  'ticlTrackstersCLUE3DHigh'
76  ),
77  )
78 
79  process.ticlCandidate = _ticlCandidateProducer.clone()
80  process.ticlCandidateTask = cms.Task(process.ticlCandidate)
81 
82  process.allTrackstersToSimTrackstersAssociationsByLCs = _allTrackstersToSimTrackstersAssociationsByLCs.clone()
83 
84  process.allTrackstersToSimTrackstersAssociationsByHits = _allTrackstersToSimTrackstersAssociationsByHits.clone()
85 
86  process.iterTICLTask = cms.Path(process.hgcalLayerClustersTask,
87  process.TFESSource,
88  process.ticlLayerTileTask,
89  process.mtdSoATask,
90  process.ticlIterationsTask,
91  process.ticlTracksterLinksTask,
92  process.ticlCandidateTask)
93 
94  process.particleFlowClusterHGCal.initialClusteringStep.tracksterSrc = "ticlCandidate"
95  process.globalrecoTask.remove(process.ticlTrackstersMerge)
96 
97 
98  process.mergeTICLTask = cms.Task()
99  process.pfTICL = _pfTICLProducer.clone(
100  ticlCandidateSrc = cms.InputTag('ticlCandidate'),
101  isTICLv5 = cms.bool(True)
102  )
103  process.hgcalAssociators = cms.Task(process.recHitMapProducer, process.lcAssocByEnergyScoreProducer, process.layerClusterCaloParticleAssociationProducer,
104  process.scAssocByEnergyScoreProducer, process.layerClusterSimClusterAssociationProducer,
105  # FP 07/2024 new associators:
106  process.allLayerClusterToTracksterAssociations, process.allHitToTracksterAssociations,
107  process.allTrackstersToSimTrackstersAssociationsByLCs, process.allTrackstersToSimTrackstersAssociationsByHits,
108  process.hitToSimClusterCaloParticleAssociator, process.SimClusterToCaloParticleAssociation,
109  )
110 
111  if(enableDumper):
112  process.ticlDumper = ticlDumper
113  process.TFileService = cms.Service("TFileService",
114  fileName=cms.string("histo.root")
115  )
116 
117  process.FEVTDEBUGHLToutput_step = cms.EndPath(process.ticlDumper)
118 
119  process.TICL_Validation = cms.Path(process.ticlSimTrackstersTask, process.hgcalAssociators)
120 
121 # Schedule definition
122  process.schedule = cms.Schedule(process.iterTICLTask,
123  process.TICL_Validation,
124  process.FEVTDEBUGHLToutput_step)
125  process = customiseForTICLv5EventContent(process)
126 
127  return process
128 
def customiseTICLv5FromReco(process, enableDumper=False)
if(threadIdxLocalY==0 &&threadIdxLocalX==0)