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('NNBatchSize', 1)
68 cf.addFlag('windowRScaling', 1.0)
69 cf.addFlag('windowPhiScaling', 1.0)
70 cf.addFlag('windowZScaling', 1.0)
71 cf.addFlag('lowPtvalueR', -1)
72 cf.addFlag('lowPtvaluePhi', -1)
73 cf.addFlag('lowPtvalueZ', -1)
74 cf.addFlag('lowPtWindowRScaling', 1.0)
75 cf.addFlag('lowPtWindowPhiScaling', 1.0)
76 cf.addFlag('lowPtWindowZScaling', 1.0)
77 cf.addFlag('missedHitRScaling', -1.0)
78 cf.addFlag('missedHitPhiScaling', -1.0)
79 cf.addFlag('missedHitZScaling', -1.0)
80 cf.addFlag('maxBranches', [2])
81 cf.addFlag('hitThreshold', 2)
82 cf.addFlag('varyingThreshold', True)
83 cf.addFlag('varyingHitThresholds', [])
84 cf.addFlag('MinSpacePointsPerSeed',3)
85 cf.addFlag('MaxSpacePointsPerSeed',3)
86
87 cf.addFlag('runF150hw', False)
88
89
90 def __httHough1DFlags():
91 """Additional function delays import"""
92 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createHough1dFPGATrackSimConfigFlags
93 return createHough1dFPGATrackSimConfigFlags()
94 cf.addFlagsCategory("Hough1D", __httHough1DFlags, prefix=True )
95
96 def __httHoughFlags():
97 """Additional function delays import"""
98 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createHoughFPGATrackSimConfigFlags
99 return createHoughFPGATrackSimConfigFlags()
100 cf.addFlagsCategory("Hough", __httHoughFlags, prefix=True )
101
102 def __httGenScanFlags():
103 """Additional function delays import"""
104 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createGenScanFPGATrackSimConfigFlags
105 return createGenScanFPGATrackSimConfigFlags()
106 cf.addFlagsCategory("GenScan", __httGenScanFlags, prefix=True )
107
108 def __httSecondStageFlags():
109 """Additional function delays import"""
110 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createSecondStageFPGATrackSimConfigFlags
111 return createSecondStageFPGATrackSimConfigFlags()
112 cf.addFlagsCategory("SecondStage", __httSecondStageFlags, prefix=True )
113
114 def __httSecondStageFlags():
115 """Additional function delays import"""
116 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createSecondStageFPGATrackSimConfigFlags
117 return createSecondStageFPGATrackSimConfigFlags()
118 cf.addFlagsCategory("SecondStage", __httSecondStageFlags, prefix=True )
119
120 def __httDev21_02_15Flags():
121 """Additional function delays import"""
122 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createDev21_02_15_FPGATrackSimConfigFlags
123 return createDev21_02_15_FPGATrackSimConfigFlags()
124 cf.addFlagsCategory("Dev21_02_15", __httDev21_02_15Flags, prefix=True )
125
126 def __GNNFlags():
127 """Additional function delays import"""
128 from FPGATrackSimConfTools.FPGATrackSimConfigFlags import createGNNFPGATrackSimConfigFlags
129 return createGNNFPGATrackSimConfigFlags()
130 cf.addFlagsCategory("GNN", __GNNFlags, prefix=True )
131
132 # EDM conversion
133 cf.addFlag('doEDMConversion', False)
134 cf.addFlag('convertSPs', False)
135 cf.addFlag('convertUnmappedHits', False)
136 cf.addFlag('writeToAOD', False)
137 cf.addFlag('writeClustersToAOD', False)
138 cf.addFlag('writeOfflPRDInfo', False)
139 cf.addFlag('broadErrors', False)
140
141 # Monitoring
142 cf.addFlag('writeAdditionalOutputData', True)
143 cf.addFlag('regionToWriteDPTree', -1)
144 cf.addFlag('writeOutputEventLimit', -1)
145 cf.addFlag('writeRegion', -1)
146 cf.addFlag('readOfflineObjects', True)
147
148 # ACTS Tracking
149 cf.addFlag('runCKF',False)
150 cf.addFlag('useFPGATruthTrackMatching',False)
151 return cf
152
153
154def createBasicFPGATrackSimConfigFlags():
155 from AthenaConfiguration.AthConfigFlags import AthConfigFlags
156 cf = AthConfigFlags()
157
158 # common
159 cf.addFlag('name', '')
160
161 # input
162 cf.addFlag('firstInputToolN', 1)
163 cf.addFlag('realHitsOverlay', False)
164 cf.addFlag('hitFiltering', False)
165 cf.addFlag('spacePointFiltering', False)
166 cf.addFlag('writeTestOutput', True)
167
168 # road finding selection
169 cf.addFlag('houghRootoutput1st', False)
170 cf.addFlag('houghRootoutput2nd', False)
171 cf.addFlag('hough', True)
172 cf.addFlag('hough1D', False)
173 cf.addFlag('houghType', 'Original')
174 cf.addFlag('roadMerge', False)
175 cf.addFlag('genScan', False)
176 cf.addFlag('NumOfHitPerGrouping', 5)
177
178 cf.addFlag('etaPatternFilter', False)
179 cf.addFlag('phiRoadFilter', False)
180 cf.addFlag('GNN', False)
181
182
183 # NN filtering 1st stage
184 cf.addFlag('trackNNAnalysis', False)
185 # NN filtering 2nd stage
186 cf.addFlag('trackNNAnalysis2nd', False)
187
188
189 # overlap removal
190 cf.addFlag('doFastOR', False)
191
192 #pass lowest chi2 track
193 cf.addFlag('passLowestChi2TrackOnly', False)
194
195 # hough
196 cf.addFlag('xVar', 'phi')
197 cf.addFlag('yVar', 'q/pt')
198 cf.addFlag('qptMin', -1.0)
199 cf.addFlag('qptMax', 1.0)
200 cf.addFlag('d0Slices', [])
201 cf.addFlag('slicing', True)
202 cf.addFlag('localMaxWindowSize', 0)
203 cf.addFlag('roadSliceOR', False)
204 cf.addFlag('fieldCorrection', True)
205 cf.addFlag('addAllHits',False)
206 cf.addFlag('detectorZoneWindows', False)
207 cf.addFlag('phiMin', 0.0)
208 cf.addFlag('phiMax', 0.0)
209 cf.addFlag('xBins', 216)
210 cf.addFlag('yBins', 216)
211 cf.addFlag('xBufferBins', 6)
212 cf.addFlag('yBufferBins', 2)
213 cf.addFlag('threshold', [8])
214 cf.addFlag('IdealGeoRoads', True)
215 cf.addFlag('convolution', [])
216 cf.addFlag('convSizeX', 0)
217 cf.addFlag('convSizeY', 0)
218 cf.addFlag('hitExtendX', [])
219
220 cf.addFlag('doMajority', 1)
221 cf.addFlag('doTracking', False)
222 cf.addFlag('outputHitTxt', False)
223
224 cf.addFlag('requirements', '')
225 # only for 2DHT Flex
226 cf.addFlag('r_max',2047)
227 cf.addFlag('phi_coord_max',65535)
228 cf.addFlag('phi_range', 6.399609375)
229 cf.addFlag('r_max_mm',1137.5)
230 cf.addFlag('bitwise_qApt_conv',204)
231 cf.addFlag('bitwise_phi0_conv',1)
232 cf.addFlag('phi0_sectors',1)
233 cf.addFlag('qApt_sectors',7)
234 cf.addFlag('pipes_qApt',8)
235 cf.addFlag('pipes_phi0',1)
236
237 # performance monitoring
238 cf.addFlag('barcodeFracMatch', 0.5)
239
240 # track Fitter
241 cf.addFlag('chi2DoFRecoveryMin', 40)
242 cf.addFlag('chi2DoFRecoveryMax', 20000)
243 cf.addFlag('nHitsNoRecovery', -1)
244 cf.addFlag('guessHits', True)
245 cf.addFlag('doMissingHitsChecks', False)
246 cf.addFlag('idealCoordFitType', 2)
247 cf.addFlag('doDeltaGPhis', False)
248 cf.addFlag('chi2cut', 2.25)
249 cf.addFlag('useVaryingChi2Cut', False)
250 cf.addFlag('fitFromRoad', True)
251
252 # second stage fitting
253 cf.addFlag('secondStage', False)
254 cf.addFlag('secondChi2Cut', 36)
255
256 # fast monitoring
257 cf.addFlag('fastMon', False)
258 cf.addFlag('canExtendHistRanges', False)
259 cf.addFlag('dataClass', 2)
260
261 # lrt settings
262 cf.addFlag('lrt', False)
263 cf.addFlag('lrtUseBasicHitFilter', False)
264 cf.addFlag('lrtUseMlHitFilter', False)
265 cf.addFlag('lrtUseStraightTrackHT', False)
266 cf.addFlag('lrtUseDoubletHT', False)
267 cf.addFlag('lrtDoubletD0Range', 120.0)
268 cf.addFlag('lrtDoubletD0Bins', 216)
269 cf.addFlag('lrtDoubletQptRange', 0.02)
270 cf.addFlag('lrtDoubletQptBins', 216)
271 cf.addFlag('lrtMonPhiRange', (0.2, 0.5))
272 cf.addFlag('lrtMonD0Range', (-100,100))
273 cf.addFlag('lrtMonZ0Range', (-300,300))
274 cf.addFlag('sampleType', '')
275
276 return cf
277
278
279def createHough1dFPGATrackSimConfigFlags():
280 cf = createBasicFPGATrackSimConfigFlags()
281
282 cf.name = 'hough_1d'
283
284 cf.hough1D = True
285
286 cf.phiMin = 0.0
287 cf.phiMax = 0.8
288 cf.xBins = 200
289 cf.threshold = [7]
290 cf.hitExtendX = [1] * 9
291
292 cf.addFlag('phiRangeCut', True)
293 cf.addFlag('splitpt', 1)
294 cf.addFlag('phifilterwindow', 0.005)
295
296 cf.outputHitTxt = ""
297
298 return cf
299
300
301def createHoughFPGATrackSimConfigFlags():
302 cf = createBasicFPGATrackSimConfigFlags()
303
304 cf.name = 'hough'
305 cf.hough = True
306
307
308 cf.phiMin = 0.3
309 cf.phiMax = 0.5
310 cf.xBins = 216
311 cf.yBins = 216
312 cf.xBufferBins = 6
313 cf.yBufferBins = 2
314 cf.addFlag('combineLayers', [])
315 cf.addFlag('scale', [])
316 cf.hitExtendX = [2,1,0,0,0,0,0,0,0]
317
318 cf.addFlag('lrtSkipHitFiltering', False)
319 cf.addFlag('lrtPtmin', 5)
320 cf.addFlag('allowHighTruthBarcode', False)
321 cf.addFlag('mLRTpdgID', 0)
322 cf.addFlag('lrtStraighttrackXVar', 'phi')
323 cf.addFlag('lrtStraighttrackYVar', 'd0')
324 cf.addFlag('lrtStraighttrackPhiMin', 0.3)
325 cf.addFlag('lrtStraighttrackPhiMax', 0.5)
326 cf.addFlag('lrtStraighttrackD0Min', -300.0)
327 cf.addFlag('lrtStraighttrackD0Max', 300.0)
328 cf.addFlag('lrtStraighttrackXBins', 216)
329 cf.addFlag('lrtStraighttrackYBins', 216)
330 cf.addFlag('lrtStraighttrackXBufferBins', 6)
331 cf.addFlag('lrtStraighttrackYBufferBins', 2)
332 cf.addFlag('lrtStraighttrackSlicing', True)
333 cf.addFlag('lrtStraighttrackThreshold', [1])
334 cf.addFlag('lrtStraighttrackConvolution', [])
335 cf.addFlag('lrtStraighttrackCombineLayers', [0,1,2,3,4,5,6,7])
336 cf.addFlag('lrtStraighttrackScale', [1])
337 cf.addFlag('lrtStraighttrackConvSizeX', 0)
338 cf.addFlag('lrtStraighttrackConvSizeY', 0)
339 cf.addFlag('lrtStraighttrackHitExtendX', [])
340 cf.addFlag('lrtStraighttrackStereo', False)
341 cf.addFlag('lrtStraighttrackLocalMaxWindowSize', 0)
342
343 return cf
344
345
346def createDev21_02_15_FPGATrackSimConfigFlags():
347 cf = createBasicFPGATrackSimConfigFlags()
348
349 cf.name = 'dev_21-02-15'
350 cf.phiMin = 0.3
351 cf.phiMax = 0.5
352 cf.xBins = 64
353 cf.yBins = 216
354 cf.xBufferBins = 2
355 cf.yBufferBins = 2
356 cf.threshold = [70]
357 cf.convolution = [1, 10, 1]
358 cf.addFlag('combineLayers', [])
359 cf.addFlag('scale', [])
360 cf.convSizeX = 3
361 cf.convSizeY = 1
362
363 cf.doMajority = 0
364 cf.doTracking = True
365
366 return cf
367
368def createGenScanFPGATrackSimConfigFlags():
369 cf = createBasicFPGATrackSimConfigFlags()
370
371 cf.name = 'genScan'
372 cf.addFlag('genScanCuts','FPGATrackSimGenScanCuts')
373 cf.addFlag('reverse','True')
374 cf.addFlag('binFilter','IncrementalBuild')
375 cf.addFlag('layerStudy',False)
376 cf.addFlag('layerMapFile','')
377 cf.addFlag('useLayerRadiiFile',False)
378 cf.addFlag('noCuts',False)
379
380 cf.addFlag('filterInBin', False)
381 cf.addFlag('phiChi2Weight', 1.0)
382 cf.addFlag('etaChi2Weight', 1.0)
383
384 # These are used by the layer study, to avoid the need for Cuts_step0.
385 # They are only used in the layer study if initialLayerStudy is set to True
386 cf.addFlag('initialLayerStudy', False)
387 cf.addFlag('layerStudyCutFile', "")
388 cf.addFlag('rin', 30)
389 cf.addFlag('rout', 300)
390 cf.addFlag('parBins', [20, 20, 5, 20, 3])
391 cf.addFlag('parMin', [-1000, -1000, 0.0, 0.0, -10])
392 cf.addFlag('parMax', [ 1000, 1000, 1.0, 1.0, 10])
393 cf.addFlag('parSet', "PhiSlicedKeyLyrPars")
394 cf.addFlag('keepHitsStrategy', -1)
395 return cf
396
397def createSecondStageFPGATrackSimConfigFlags():
398 cf = createBasicFPGATrackSimConfigFlags()
399 cf.name = 'SecondStage'
400 cf.addFlag('CutFile', "Cuts_step1_2ndStage")
401 cf.addFlag('LayerMapFile', "")
402 return cf
403
404class graphTool(FlagEnum):
405 ModuleMap = 'ModuleMap'
406 MetricLearning = 'MetricLearning'
407
408class moduleMapType(FlagEnum):
409 doublet = 'doublet'
410
411class moduleMapFunc(FlagEnum):
412 minmax = 'minmax'
413
414class roadMakerTool(FlagEnum):
415 ConnectedComponents = 'ConnectedComponents'
416 JunctionAwareCC = 'JunctionAwareCC'
417
418def createGNNFPGATrackSimConfigFlags():
419 cf = createBasicFPGATrackSimConfigFlags()
420
421 cf.name = 'GNN'
422 cf.addFlag("graphTool", graphTool.ModuleMap, type=graphTool)
423 cf.addFlag("moduleMapType", moduleMapType.doublet, type=moduleMapType)
424 cf.addFlag("moduleMapFunc", moduleMapFunc.minmax, type=moduleMapFunc)
425 cf.addFlag("moduleMapTol",0.0000000001) # 1e-10
426 cf.addFlag("moduleMapPath",'')
427 cf.addFlag("metricLearningR",0.1)
428 cf.addFlag("metricLearningMaxN", 512)
429 cf.addFlag("MLModelPath",'')
430 cf.addFlag("GNNModelPath",'')
431 cf.addFlag("roadMakerTool", roadMakerTool.ConnectedComponents, type=roadMakerTool)
432 cf.addFlag("edgeScoreCut",0.8)
433 cf.addFlag("doGNNRootOutput",False)
434 cf.addFlag("doGNNTracking",False)
435 cf.addFlag("doGNNPixelSeeding",False)
436 cf.addFlag("nInputsGNN",13)
437
438 return cf
439
440
443
444if __name__ == "__main__":
445
446 flags = createFPGATrackSimConfigFlags()
447 flags.loadAllDynamicFlags()
448 flags.initAll()
449 flags.dump()
450 assert flags.Hough.firstInputToolN == 1 , "default firstInputToolN is wrong"
451 assert flags.Hough.barcodeFracMatch == 0.5, "default barcodeFracMatch is wrong"
452 assert flags.Hough.fastMon is False , "default fastMon is wrong"
453 assert flags.Hough.lrtMonZ0Range == (-300,300), "default lrtMonZ0Rang is wrong"
454
455 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