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 8 of file customiseForTICLv5_cff.py.

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

8 def customiseTICLv5FromReco(process, enableDumper = False):
9  # TensorFlow ESSource
10  process.TFESSource = cms.Task(process.trackdnn_source)
11 
12  # Reconstruction
13  process.hgcalLayerClustersTask = cms.Task(process.hgcalLayerClustersEE,
14  process.hgcalLayerClustersHSi,
15  process.hgcalLayerClustersHSci,
16  process.hgcalMergeLayerClusters)
17 
18  process.ticlIterationsTask = cms.Task(
19  process.ticlCLUE3DHighStepTask,
20  process.ticlTracksterLinksTask,
21  process.ticlPassthroughStepTask
22  )
23 
24  process.mergeTICLTask = cms.Task()
25 
26  process.iterTICLTask = cms.Path(process.hgcalLayerClustersTask,
27  process.TFESSource,
28  process.ticlLayerTileTask,
29  process.mtdSoATask,
30  process.mergeTICLTask,
31  process.ticlIterationsTask,
32  process.ticlCandidateTask,
33  process.ticlPFTask)
34 
35  process.tracksterSimTracksterAssociationLinkingbyCLUE3DHigh = _tracksterSimTracksterAssociationLinkingbyCLUE3D.clone(
36  label_tst = cms.InputTag("ticlTrackstersCLUE3DHigh")
37  )
38  process.tracksterSimTracksterAssociationPRbyCLUE3DHigh = _tracksterSimTracksterAssociationPRbyCLUE3D.clone(
39  label_tst = cms.InputTag("ticlTrackstersCLUE3DHigh")
40  )
41 
42  '''for future CLUE3D separate iterations, merge collections and compute scores
43  process.tracksterSimTracksterAssociationLinkingbyCLUE3DEM = _tracksterSimTracksterAssociationLinkingbyCLUE3D.clone(
44  label_tst = cms.InputTag("ticlTrackstersCLUE3DEM")
45  )
46  process.tracksterSimTracksterAssociationPRbyCLUE3DEM = _tracksterSimTracksterAssociationPRbyCLUE3D.clone(
47  label_tst = cms.InputTag("ticlTrackstersCLUE3DEM")
48  )
49  process.tracksterSimTracksterAssociationLinkingbyCLUE3DHAD = _tracksterSimTracksterAssociationLinkingbyCLUE3D.clone(
50  label_tst = cms.InputTag("ticlTrackstersCLUE3DHAD")
51  )
52  process.tracksterSimTracksterAssociationPRbyCLUE3DHAD = _tracksterSimTracksterAssociationPRbyCLUE3D.clone(
53  label_tst = cms.InputTag("ticlTrackstersCLUE3DHAD")
54  )
55 
56  process.mergedTrackstersProducer = _mergedTrackstersProducer.clone()
57  process.tracksterSimTracksterAssociationLinkingbyCLUE3D = _tracksterSimTracksterAssociationLinkingbyCLUE3D.clone(
58  label_tst = cms.InputTag("mergedTrackstersProducer")
59  )
60  process.tracksterSimTracksterAssociationPRbyCLUE3D = _tracksterSimTracksterAssociationPRbyCLUE3D.clone(
61  label_tst = cms.InputTag("mergedTrackstersProducer")
62  )
63  '''
64 
65  process.hgcalAssociators = cms.Task(process.recHitMapProducer, process.lcAssocByEnergyScoreProducer, process.layerClusterCaloParticleAssociationProducer,
66  process.scAssocByEnergyScoreProducer, process.layerClusterSimClusterAssociationProducer,
67  process.lcSimTSAssocByEnergyScoreProducer, process.layerClusterSimTracksterAssociationProducer,
68  process.simTsAssocByEnergyScoreProducer, process.simTracksterHitLCAssociatorByEnergyScoreProducer,
69  process.tracksterSimTracksterAssociationLinking, process.tracksterSimTracksterAssociationPR,
70  process.tracksterSimTracksterAssociationLinkingbyCLUE3DHigh, process.tracksterSimTracksterAssociationPRbyCLUE3DHigh,
71  process.tracksterSimTracksterAssociationLinkingPU, process.tracksterSimTracksterAssociationPRPU
72  )
73 
74  '''for future CLUE3D separate iterations, merge collections and compute scores
75  process.tracksterSimTracksterAssociationLinkingbyCLUE3D, process.tracksterSimTracksterAssociationPRbyCLUE3D,
76  process.tracksterSimTracksterAssociationLinkingbyCLUE3DEM, process.tracksterSimTracksterAssociationPRbyCLUE3DEM,
77  process.tracksterSimTracksterAssociationLinkingbyCLUE3DHAD, process.tracksterSimTracksterAssociationPRbyCLUE3DHAD,
78  '''
79 
80  if(enableDumper):
81  process.ticlDumper = ticlDumper.clone(
82  saveLCs=True,
83  saveCLUE3DTracksters=True,
84  saveTrackstersMerged=True,
85  saveSimTrackstersSC=True,
86  saveSimTrackstersCP=True,
87  saveTICLCandidate=True,
88  saveSimTICLCandidate=True,
89  saveTracks=True,
90  saveAssociations=True,
91  trackstersclue3d = cms.InputTag('ticlTrackstersCLUE3DHigh'),
92  ticlcandidates = cms.InputTag("ticlCandidate"),
93  trackstersmerged = cms.InputTag("ticlCandidate"),
94  trackstersInCand = cms.InputTag("ticlCandidate")
95  )
96  process.TFileService = cms.Service("TFileService",
97  fileName=cms.string("histo.root")
98  )
99 
100  process.FEVTDEBUGHLToutput_step = cms.EndPath(process.ticlDumper)
101 
102  process.TICL_Validator = cms.Task(process.hgcalValidator)
103  process.TICL_Validation = cms.Path(process.ticlSimTrackstersTask, process.hgcalAssociators, process.TICL_Validator)
104 
105  # Schedule definition
106  process.schedule = cms.Schedule(process.iterTICLTask,
107  process.TICL_Validation,
108  process.FEVTDEBUGHLToutput_step)
109  process = customiseForTICLv5EventContent(process)
110 
111  return process
112 
def customiseTICLv5FromReco(process, enableDumper=False)
if(threadIdxLocalY==0 &&threadIdxLocalX==0)