ATLAS Offline Software
Loading...
Searching...
No Matches
FPGATrackSimConfigFlags.py
Go to the documentation of this file.
1#Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
2import AthenaCommon.Constants
3from AthenaConfiguration.Enums import FlagEnum
4
5def createFPGATrackSimConfigFlags():
6 from AthenaConfiguration.AthConfigFlags import AthConfigFlags
7 cf = AthConfigFlags()
8 cf.addFlag('pipeline', '')
9 cf.addFlag('algoTag', 'Hough')
10 cf.addFlag('wrapperFileName', [])
11 cf.addFlag('wrapperFileName2', [])
12 cf.addFlag('runOnPreProducedHeaderFiles', False)
13 cf.addFlag('secondInputToolN', 0)
14 cf.addFlag('mapsDir', '__MUST_BE_SET__')
15 cf.addFlag('loadRegionMap', True)
16 cf.addFlag('loadRadii', True)
17 cf.addFlag('wrapperMetaData', "Default Meta Data")
18 cf.addFlag('sampleType', 'singleMuons')
19 cf.addFlag('doMultiTruth', True)
20 cf.addFlag('SetTruthParametersForTracks', -1)
21 cf.addFlag('FPGATrackSimTestFiles', '')
22 cf.addFlag('FPGATrackSimMatrixFileRegEx', [])
23 cf.addFlag('FPGATrackSimMaxnMatrixInputFiles', -1)
24 cf.addFlag('outputMergedFPGATrackSimMatrixFile', 'combined_matrix.root')
25 cf.addFlag('FPGATrackSimNBanks', 1)
26 cf.addFlag('FPGATrackSimallBanks', False)
27 cf.addFlag('region', 34)
28 cf.addFlag('regionList',"") # can take ranges, e.g. 30-52, exceptions e.g. !34-36 and regex e.g. 2*0 as comma separated list
29 cf.addFlag('d0min', -2.0)
30 cf.addFlag('d0max', +2.0)
31 cf.addFlag('z0min', -150.0)
32 cf.addFlag('z0max', +150.0)
33 cf.addFlag('qOverPtmin', -0.001)
34 cf.addFlag('qOverPtmax', +0.001)
35 cf.addFlag('oldRegionDefs', False)
36 cf.addFlag('phiShift', 0.0)
37 cf.addFlag('minSpacePlusPixel', 3)
38 cf.addFlag('dropUnpairedIfSP', False)
39 cf.addFlag('baseName', '')
40 cf.addFlag('CheckGood2ndStage', True)
41 cf.addFlag('Is2ndStage', False)
42 cf.addFlag('UseHitScaleFactor', False)
43 cf.addFlag('missHitsConsts', False)
44 cf.addFlag('tracking', False)
45 cf.addFlag('secondTracking', True)
46 cf.addFlag('layerStudyStage', 0)
47 cf.addFlag('doOverlapRemoval', True)
48 cf.addFlag('doOverlapRemovalBetweenRegions', True)
49 cf.addFlag('clustering', 1)
50 cf.addFlag('LorentzAngleShift', 2)
51 cf.addFlag('bankDir', '')
52 cf.addFlag('slicesFile', 'eventSelectionSlices/v1.0/slices.txt')
53 cf.addFlag('spacePoints', True)
54 cf.addFlag('outputMonitorFile',"monitoring.root")
55 cf.addFlag('connectToToITkTracking',True)
56 cf.addFlag('loglevel', AthenaCommon.Constants.INFO)
57 cf.addFlag('msgLimit',-1)
58 cf.addFlag('singleTrackSample', True)
59 cf.addFlag('FakeNNonnxFile1st', 'NN/1stStage/v0.11/Class_5Hits_V007')
60 cf.addFlag('FakeNNonnxFile2nd', 'NN/2ndStage/v0.11/Class_13Hits_V007')
61 cf.addFlag('ExtensionNNVolonnxFile', 'NN/2ndStage/v0.11/NNPathfinderVol_4Hits_V011')
62 cf.addFlag('ExtensionNNHitonnxFile', 'NN/2ndStage/v0.11/NNPathfinderHit_4Hits_V011')
63 cf.addFlag('ParamNNonnxFile1st', 'NN/1stStage/v0.11/Param_5Hits_V004')
64 cf.addFlag('ParamNNonnxFile2nd', 'NN/2ndStage/v0.11/Param_13Hits_V007')
65 cf.addFlag('doNNPathFinder', False)
66 cf.addFlag('NNCartesianCoordinates', False)
67 cf.addFlag('windowRScaling', 1.0)
68 cf.addFlag('windowPhiScaling', 1.0)
69 cf.addFlag('windowZScaling', 1.0)
70 cf.addFlag('lowPtvalueR', -1)
71 cf.addFlag('lowPtvaluePhi', -1)
72 cf.addFlag('lowPtvalueZ', -1)
73 cf.addFlag('lowPtWindowRScaling', 1.0)
74 cf.addFlag('lowPtWindowPhiScaling', 1.0)
75 cf.addFlag('lowPtWindowZScaling', 1.0)
76 cf.addFlag('missedHitRScaling', -1.0)
77 cf.addFlag('missedHitPhiScaling', -1.0)
78 cf.addFlag('missedHitZScaling', -1.0)
79 cf.addFlag('maxBranches', [2])
80 cf.addFlag('hitThreshold', 2)
81 cf.addFlag('varyingThreshold', True)
82 cf.addFlag('varyingHitThresholds', [])
83 cf.addFlag('MinSpacePointsPerSeed',3)
84 cf.addFlag('MaxSpacePointsPerSeed',3)
85
86 cf.addFlag('runF150hw', False)
87
88
89 def __httHough1DFlags():
90 """Additional function delays import"""
91 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createHough1dFPGATrackSimConfigFlags
92 return createHough1dFPGATrackSimConfigFlags()
93 cf.addFlagsCategory("Hough1D", __httHough1DFlags, prefix=True )
94
95 def __httHoughFlags():
96 """Additional function delays import"""
97 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createHoughFPGATrackSimConfigFlags
98 return createHoughFPGATrackSimConfigFlags()
99 cf.addFlagsCategory("Hough", __httHoughFlags, prefix=True )
100
101 def __httGenScanFlags():
102 """Additional function delays import"""
103 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createGenScanFPGATrackSimConfigFlags
104 return createGenScanFPGATrackSimConfigFlags()
105 cf.addFlagsCategory("GenScan", __httGenScanFlags, prefix=True )
106
107 def __httSecondStageFlags():
108 """Additional function delays import"""
109 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createSecondStageFPGATrackSimConfigFlags
110 return createSecondStageFPGATrackSimConfigFlags()
111 cf.addFlagsCategory("SecondStage", __httSecondStageFlags, prefix=True )
112
113 def __httSecondStageFlags():
114 """Additional function delays import"""
115 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createSecondStageFPGATrackSimConfigFlags
116 return createSecondStageFPGATrackSimConfigFlags()
117 cf.addFlagsCategory("SecondStage", __httSecondStageFlags, prefix=True )
118
119 def __httDev21_02_15Flags():
120 """Additional function delays import"""
121 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createDev21_02_15_FPGATrackSimConfigFlags
122 return createDev21_02_15_FPGATrackSimConfigFlags()
123 cf.addFlagsCategory("Dev21_02_15", __httDev21_02_15Flags, prefix=True )
124
125 def __GNNFlags():
126 """Additional function delays import"""
127 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createGNNFPGATrackSimConfigFlags
128 return createGNNFPGATrackSimConfigFlags()
129 cf.addFlagsCategory("GNN", __GNNFlags, prefix=True )
130
131 # EDM conversion
132 cf.addFlag('doEDMConversion', False)
133 cf.addFlag('convertSPs', False)
134 cf.addFlag('convertUnmappedHits', False)
135 cf.addFlag('writeToAOD', False)
136 cf.addFlag('writeClustersToAOD', False)
137 cf.addFlag('writeOfflPRDInfo', False)
138 cf.addFlag('broadErrors', False)
139
140 # Monitoring
141 cf.addFlag('writeAdditionalOutputData', True)
142 cf.addFlag('regionToWriteDPTree', -1)
143 cf.addFlag('writeOutputEventLimit', -1)
144 cf.addFlag('writeRegion', -1)
145 cf.addFlag('readOfflineObjects', True)
146
147 # ACTS Tracking
148 cf.addFlag('runCKF',False)
149 cf.addFlag('useFPGATruthTrackMatching',False)
150 return cf
151
152
153def createBasicFPGATrackSimConfigFlags():
154 from AthenaConfiguration.AthConfigFlags import AthConfigFlags
155 cf = AthConfigFlags()
156
157 # common
158 cf.addFlag('name', '')
159
160 # input
161 cf.addFlag('firstInputToolN', 1)
162 cf.addFlag('realHitsOverlay', False)
163 cf.addFlag('hitFiltering', False)
164 cf.addFlag('spacePointFiltering', False)
165 cf.addFlag('writeTestOutput', True)
166
167 # road finding selection
168 cf.addFlag('houghRootoutput1st', False)
169 cf.addFlag('houghRootoutput2nd', False)
170 cf.addFlag('hough', True)
171 cf.addFlag('hough1D', False)
172 cf.addFlag('houghType', 'Original')
173 cf.addFlag('roadMerge', False)
174 cf.addFlag('genScan', False)
175 cf.addFlag('NumOfHitPerGrouping', 5)
176
177 cf.addFlag('etaPatternFilter', False)
178 cf.addFlag('phiRoadFilter', False)
179 cf.addFlag('GNN', False)
180
181
182 # NN filtering 1st stage
183 cf.addFlag('trackNNAnalysis', False)
184 # NN filtering 2nd stage
185 cf.addFlag('trackNNAnalysis2nd', False)
186
187
188 # overlap removal
189 cf.addFlag('doFastOR', False)
190
191 #pass lowest chi2 track
192 cf.addFlag('passLowestChi2TrackOnly', False)
193
194 # hough
195 cf.addFlag('xVar', 'phi')
196 cf.addFlag('yVar', 'q/pt')
197 cf.addFlag('qptMin', -1.0)
198 cf.addFlag('qptMax', 1.0)
199 cf.addFlag('d0Slices', [])
200 cf.addFlag('slicing', True)
201 cf.addFlag('localMaxWindowSize', 0)
202 cf.addFlag('roadSliceOR', False)
203 cf.addFlag('fieldCorrection', True)
204 cf.addFlag('addAllHits',False)
205 cf.addFlag('detectorZoneWindows', False)
206 cf.addFlag('phiMin', 0.0)
207 cf.addFlag('phiMax', 0.0)
208 cf.addFlag('xBins', 216)
209 cf.addFlag('yBins', 216)
210 cf.addFlag('xBufferBins', 6)
211 cf.addFlag('yBufferBins', 2)
212 cf.addFlag('threshold', [8])
213 cf.addFlag('IdealGeoRoads', True)
214 cf.addFlag('convolution', [])
215 cf.addFlag('convSizeX', 0)
216 cf.addFlag('convSizeY', 0)
217 cf.addFlag('hitExtendX', [])
218
219 cf.addFlag('doMajority', 1)
220 cf.addFlag('doTracking', False)
221 cf.addFlag('outputHitTxt', False)
222
223 cf.addFlag('requirements', '')
224 # only for 2DHT Flex
225 cf.addFlag('r_max',2047)
226 cf.addFlag('phi_coord_max',65535)
227 cf.addFlag('phi_range', 6.399609375)
228 cf.addFlag('r_max_mm',1137.5)
229 cf.addFlag('bitwise_qApt_conv',204)
230 cf.addFlag('bitwise_phi0_conv',1)
231 cf.addFlag('phi0_sectors',1)
232 cf.addFlag('qApt_sectors',7)
233 cf.addFlag('pipes_qApt',8)
234 cf.addFlag('pipes_phi0',1)
235
236 # performance monitoring
237 cf.addFlag('barcodeFracMatch', 0.5)
238
239 # track Fitter
240 cf.addFlag('chi2DoFRecoveryMin', 40)
241 cf.addFlag('chi2DoFRecoveryMax', 20000)
242 cf.addFlag('nHitsNoRecovery', -1)
243 cf.addFlag('guessHits', True)
244 cf.addFlag('doMissingHitsChecks', False)
245 cf.addFlag('idealCoordFitType', 2)
246 cf.addFlag('doDeltaGPhis', False)
247 cf.addFlag('chi2cut', 2.25)
248 cf.addFlag('useVaryingChi2Cut', False)
249 cf.addFlag('fitFromRoad', True)
250
251 # second stage fitting
252 cf.addFlag('secondStage', False)
253 cf.addFlag('secondChi2Cut', 36)
254
255 # fast monitoring
256 cf.addFlag('fastMon', False)
257 cf.addFlag('canExtendHistRanges', False)
258 cf.addFlag('dataClass', 2)
259
260 # lrt settings
261 cf.addFlag('lrt', False)
262 cf.addFlag('lrtUseBasicHitFilter', False)
263 cf.addFlag('lrtUseMlHitFilter', False)
264 cf.addFlag('lrtUseStraightTrackHT', False)
265 cf.addFlag('lrtUseDoubletHT', False)
266 cf.addFlag('lrtDoubletD0Range', 120.0)
267 cf.addFlag('lrtDoubletD0Bins', 216)
268 cf.addFlag('lrtDoubletQptRange', 0.02)
269 cf.addFlag('lrtDoubletQptBins', 216)
270 cf.addFlag('lrtMonPhiRange', (0.2, 0.5))
271 cf.addFlag('lrtMonD0Range', (-100,100))
272 cf.addFlag('lrtMonZ0Range', (-300,300))
273 cf.addFlag('sampleType', '')
274
275 return cf
276
277
278def createHough1dFPGATrackSimConfigFlags():
279 cf = createBasicFPGATrackSimConfigFlags()
280
281 cf.name = 'hough_1d'
282
283 cf.hough1D = True
284
285 cf.phiMin = 0.0
286 cf.phiMax = 0.8
287 cf.xBins = 200
288 cf.threshold = [7]
289 cf.hitExtendX = [1] * 9
290
291 cf.addFlag('phiRangeCut', True)
292 cf.addFlag('splitpt', 1)
293 cf.addFlag('phifilterwindow', 0.005)
294
295 cf.outputHitTxt = ""
296
297 return cf
298
299
300def createHoughFPGATrackSimConfigFlags():
301 cf = createBasicFPGATrackSimConfigFlags()
302
303 cf.name = 'hough'
304 cf.hough = True
305
306
307 cf.phiMin = 0.3
308 cf.phiMax = 0.5
309 cf.xBins = 216
310 cf.yBins = 216
311 cf.xBufferBins = 6
312 cf.yBufferBins = 2
313 cf.addFlag('combineLayers', [])
314 cf.addFlag('scale', [])
315 cf.hitExtendX = [2,1,0,0,0,0,0,0,0]
316
317 cf.addFlag('lrtSkipHitFiltering', False)
318 cf.addFlag('lrtPtmin', 5)
319 cf.addFlag('allowHighTruthBarcode', False)
320 cf.addFlag('mLRTpdgID', 0)
321 cf.addFlag('lrtStraighttrackXVar', 'phi')
322 cf.addFlag('lrtStraighttrackYVar', 'd0')
323 cf.addFlag('lrtStraighttrackPhiMin', 0.3)
324 cf.addFlag('lrtStraighttrackPhiMax', 0.5)
325 cf.addFlag('lrtStraighttrackD0Min', -300.0)
326 cf.addFlag('lrtStraighttrackD0Max', 300.0)
327 cf.addFlag('lrtStraighttrackXBins', 216)
328 cf.addFlag('lrtStraighttrackYBins', 216)
329 cf.addFlag('lrtStraighttrackXBufferBins', 6)
330 cf.addFlag('lrtStraighttrackYBufferBins', 2)
331 cf.addFlag('lrtStraighttrackSlicing', True)
332 cf.addFlag('lrtStraighttrackThreshold', [1])
333 cf.addFlag('lrtStraighttrackConvolution', [])
334 cf.addFlag('lrtStraighttrackCombineLayers', [0,1,2,3,4,5,6,7])
335 cf.addFlag('lrtStraighttrackScale', [1])
336 cf.addFlag('lrtStraighttrackConvSizeX', 0)
337 cf.addFlag('lrtStraighttrackConvSizeY', 0)
338 cf.addFlag('lrtStraighttrackHitExtendX', [])
339 cf.addFlag('lrtStraighttrackStereo', False)
340 cf.addFlag('lrtStraighttrackLocalMaxWindowSize', 0)
341
342 return cf
343
344
345def createDev21_02_15_FPGATrackSimConfigFlags():
346 cf = createBasicFPGATrackSimConfigFlags()
347
348 cf.name = 'dev_21-02-15'
349 cf.phiMin = 0.3
350 cf.phiMax = 0.5
351 cf.xBins = 64
352 cf.yBins = 216
353 cf.xBufferBins = 2
354 cf.yBufferBins = 2
355 cf.threshold = [70]
356 cf.convolution = [1, 10, 1]
357 cf.addFlag('combineLayers', [])
358 cf.addFlag('scale', [])
359 cf.convSizeX = 3
360 cf.convSizeY = 1
361
362 cf.doMajority = 0
363 cf.doTracking = True
364
365 return cf
366
367def createGenScanFPGATrackSimConfigFlags():
368 cf = createBasicFPGATrackSimConfigFlags()
369
370 cf.name = 'genScan'
371 cf.addFlag('genScanCuts','FPGATrackSimGenScanCuts')
372 cf.addFlag('reverse','True')
373 cf.addFlag('binFilter','IncrementalBuild')
374 cf.addFlag('layerStudy',False)
375 cf.addFlag('layerMapFile','')
376 cf.addFlag('useLayerRadiiFile',False)
377 cf.addFlag('noCuts',False)
378
379 cf.addFlag('filterInBin', False)
380 cf.addFlag('phiChi2Weight', 1.0)
381 cf.addFlag('etaChi2Weight', 1.0)
382
383 # These are used by the layer study, to avoid the need for Cuts_step0.
384 # They are only used in the layer study if initialLayerStudy is set to True
385 cf.addFlag('initialLayerStudy', False)
386 cf.addFlag('layerStudyCutFile', "")
387 cf.addFlag('rin', 30)
388 cf.addFlag('rout', 300)
389 cf.addFlag('parBins', [20, 20, 5, 20, 3])
390 cf.addFlag('parMin', [-1000, -1000, 0.0, 0.0, -10])
391 cf.addFlag('parMax', [ 1000, 1000, 1.0, 1.0, 10])
392 cf.addFlag('parSet', "PhiSlicedKeyLyrPars")
393 cf.addFlag('keepHitsStrategy', -1)
394 return cf
395
396def createSecondStageFPGATrackSimConfigFlags():
397 cf = createBasicFPGATrackSimConfigFlags()
398 cf.name = 'SecondStage'
399 cf.addFlag('CutFile', "Cuts_step1_2ndStage")
400 cf.addFlag('LayerMapFile', "")
401 return cf
402
403class graphTool(FlagEnum):
404 ModuleMap = 'ModuleMap'
405 MetricLearning = 'MetricLearning'
406
407class moduleMapType(FlagEnum):
408 doublet = 'doublet'
409
410class moduleMapFunc(FlagEnum):
411 minmax = 'minmax'
412
413class roadMakerTool(FlagEnum):
414 ConnectedComponents = 'ConnectedComponents'
415 JunctionAwareCC = 'JunctionAwareCC'
416
417def createGNNFPGATrackSimConfigFlags():
418 cf = createBasicFPGATrackSimConfigFlags()
419
420 cf.name = 'GNN'
421 cf.addFlag("graphTool", graphTool.ModuleMap, type=graphTool)
422 cf.addFlag("moduleMapType", moduleMapType.doublet, type=moduleMapType)
423 cf.addFlag("moduleMapFunc", moduleMapFunc.minmax, type=moduleMapFunc)
424 cf.addFlag("moduleMapTol",0.0000000001) # 1e-10
425 cf.addFlag("moduleMapPath",'')
426 cf.addFlag("metricLearningR",0.1)
427 cf.addFlag("metricLearningMaxN", 512)
428 cf.addFlag("MLModelPath",'')
429 cf.addFlag("GNNModelPath",'')
430 cf.addFlag("roadMakerTool", roadMakerTool.ConnectedComponents, type=roadMakerTool)
431 cf.addFlag("edgeScoreCut",0.8)
432 cf.addFlag("doGNNRootOutput",False)
433 cf.addFlag("doGNNTracking",False)
434 cf.addFlag("doGNNPixelSeeding",False)
435 cf.addFlag("nInputsGNN",13)
436
437 return cf
438
439
442
443if __name__ == "__main__":
444
445 flags = createFPGATrackSimConfigFlags()
446 flags.loadAllDynamicFlags()
447 flags.initAll()
448 flags.dump()
449 assert flags.Hough.firstInputToolN == 1 , "default firstInputToolN is wrong"
450 assert flags.Hough.barcodeFracMatch == 0.5, "default barcodeFracMatch is wrong"
451 assert flags.Hough.fastMon is False , "default fastMon is wrong"
452 assert flags.Hough.lrtMonZ0Range == (-300,300), "default lrtMonZ0Rang is wrong"
453
454 print( "allok" )
print
void print(char *figname, TCanvas *c1)
Definition TRTCalib_StrawStatusPlots.cxx:26
Generated on for ATLAS Offline Software by doxygen 1.14.0