ATLAS Offline Software
Functions | Variables
python.FPGATrackSimDataPrepConfig Namespace Reference

Functions

def nameWithRegionSuffix (flags, basename)
 
def getRegionNumber (nPiOver16Min, minAbsEta, isPositiveEta, verbosePrint=True)
 
def getBaseName (flags)
 
def getPhiRange (flags)
 
def getEtaRange (flags)
 
def getEtaSideBits (flags)
 
def FPGATrackSimRawLogicCfg (flags, name="FPGATrackSimRawLogicTool")
 
def FPGATrackSimSpacePointsToolCfg (flags)
 
def prepareFlagsForFPGATrackSimDataPrepAlg (flags)
 
def FPGATrackSimDataPrepOutputCfg (flags)
 
def FPGAConversionAlgCfg (inputFlags, name='FPGAConversionAlg', stage='', **kwargs)
 
def FPGAClusterConverterCfg (flags)
 
def FPGAActsTrkConverterCfg (flags)
 
def WriteToAOD (flags, stage='', finalTrackParticles='')
 
def FPGATrackSimEventSelectionSvcCfg (flags, name="FPGATrackSimEventSelectionSvc")
 
def FPGATrackSimEventSelectionToolCfg (flags, name="FPGATrackSimEventSelectionTool")
 
def FPGATrackSimMappingCfg (flags, name="FPGATrackSimMappingSvc")
 
def FPGATrackSimReadInputCfg (flags)
 
def FPGATrackSimReadInput2Cfg (flags)
 
def FPGATrackSimHitFilteringToolCfg (flags)
 
def FPGATrackSimLorentzAngleToolCfg (flags)
 
def FPGATrackSimDataPrepAlgCfg (inputFlags)
 
def FPGATrackSimDataPrepConnectToFastTracking (flagsIn, FinalTracks="F100-", **kwargs)
 
def FPGATrackSimRegionFlagCfg (flags)
 
def FPGATrackSimClusteringCfg (flags)
 
def FPGATrackSimDataPrepFlagCfg (flags)
 
def FixITkMainPassFlags (flags)
 
def FPGATrackSimDataPrepSetup (flags, runReco=True)
 
def runDataPrepChain ()
 

Variables

 log
 

Function Documentation

◆ FixITkMainPassFlags()

def python.FPGATrackSimDataPrepConfig.FixITkMainPassFlags (   flags)

Definition at line 637 of file FPGATrackSimDataPrepConfig.py.

637 def FixITkMainPassFlags(flags):
638  flags.Tracking.ITkMainPass.doAthenaToActsSpacePoint=True
639  flags.Tracking.ITkMainPass.doAthenaToActsCluster=True
640  from ActsConfig.ActsCIFlags import actsLegacyWorkflowFlags
642  flags.Acts.doRotCorrection = False
643  return flags
644 
645 

◆ FPGAActsTrkConverterCfg()

def python.FPGATrackSimDataPrepConfig.FPGAActsTrkConverterCfg (   flags)

Definition at line 232 of file FPGATrackSimDataPrepConfig.py.

232 def FPGAActsTrkConverterCfg(flags):
233  result=ComponentAccumulator()
234  FPGAActsTrkConverter = CompFactory.FPGAActsTrkConverter()
235  result.setPrivateTools(FPGAActsTrkConverter)
236 
237  return result
238 
239 

◆ FPGAClusterConverterCfg()

def python.FPGATrackSimDataPrepConfig.FPGAClusterConverterCfg (   flags)

Definition at line 216 of file FPGATrackSimDataPrepConfig.py.

216 def FPGAClusterConverterCfg(flags):
217  result=ComponentAccumulator()
218  FPGAClusterConverter = CompFactory.FPGAClusterConverter()
219  FPGAClusterConverter.doLorentzShift=True
220 
221  from SiLorentzAngleTool.ITkPixelLorentzAngleConfig import ITkPixelLorentzAngleToolCfg
222  FPGAClusterConverter.LorentzAngleToolPixel = result.popToolsAndMerge(ITkPixelLorentzAngleToolCfg(flags))
223 
224  from SiLorentzAngleTool.ITkStripLorentzAngleConfig import ITkStripLorentzAngleToolCfg
225  FPGAClusterConverter.LorentzAngleToolStrip = result.popToolsAndMerge(ITkStripLorentzAngleToolCfg(flags))
226 
227  result.setPrivateTools(FPGAClusterConverter)
228 
229  return result
230 
231 

◆ FPGAConversionAlgCfg()

def python.FPGATrackSimDataPrepConfig.FPGAConversionAlgCfg (   inputFlags,
  name = 'FPGAConversionAlg',
  stage = '',
**  kwargs 
)

Definition at line 182 of file FPGATrackSimDataPrepConfig.py.

182 def FPGAConversionAlgCfg(inputFlags, name = 'FPGAConversionAlg', stage = '', **kwargs):
183 
184  flags = prepareFlagsForFPGATrackSimDataPrepAlg(inputFlags)
185 
186  result=ComponentAccumulator()
187  from StripGeoModelXml.ITkStripGeoModelConfig import ITkStripReadoutGeometryCfg
188  result.merge(ITkStripReadoutGeometryCfg(flags))
189 
190  kwargs.setdefault("FPGATrackSimClusterKey", "FPGAClusters_1st")
191  kwargs.setdefault("FPGATrackSimHitKey", "FPGAHits%s" %(stage))
192  kwargs.setdefault("FPGATrackSimHitInRoadsKey", "FPGAHitsInRoads")
193  kwargs.setdefault("FPGATrackSimRoadKey", "FPGARoads")
194  kwargs.setdefault("FPGATrackSimTrackKey", "FPGATracks")
195  kwargs.setdefault("xAODPixelClusterFromFPGAClusterKey", "xAODPixelClustersFromFPGACluster")
196  kwargs.setdefault("xAODStripClusterFromFPGAClusterKey", "xAODStripClustersFromFPGACluster")
197  kwargs.setdefault("xAODStripSpacePointFromFPGAKey", "xAODStripSpacePoints%sFromFPGA" %(stage))
198  kwargs.setdefault("xAODPixelSpacePointFromFPGAKey", "xAODPixelSpacePoints%sFromFPGA" %(stage))
199  kwargs.setdefault("xAODPixelClusterFromFPGAHitKey", "xAODPixelClusters%sFromFPGAHit" %(stage))
200  kwargs.setdefault("xAODStripClusterFromFPGAHitKey", "xAODStripClusters%sFromFPGAHit" %(stage))
201  kwargs.setdefault("ActsProtoTrackFromFPGARoadKey", "ActsProtoTracks%sFromFPGARoad" %(stage))
202  kwargs.setdefault("ActsProtoTrackFromFPGATrackKey", "ActsProtoTracks%sFromFPGATrack" %(stage))
203  kwargs.setdefault("doHits", False)
204  kwargs.setdefault("doClusters", True)
205  kwargs.setdefault("doActsTrk", False)
206  kwargs.setdefault("useRoads", False)
207  kwargs.setdefault("ClusterConverter", result.popToolsAndMerge(FPGAClusterConverterCfg(flags)))
208  kwargs.setdefault("ActsTrkConverter", result.popToolsAndMerge(FPGAActsTrkConverterCfg(flags)))
209 
210  ConvertionAlgorithm = CompFactory.FPGAConversionAlgorithm(name, **kwargs)
211  ConvertionAlgorithm.ClusterConverter.doLorentzShift=False # this should not be turned on unless needed for validation purposes
212 
213  result.addEventAlgo(ConvertionAlgorithm)
214  return result
215 

◆ FPGATrackSimClusteringCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimClusteringCfg (   flags)

Definition at line 557 of file FPGATrackSimDataPrepConfig.py.

557 def FPGATrackSimClusteringCfg(flags): # to be used in the Reco_tf configuration
559  if not flags.Trigger.FPGATrackSim.runOnPreProducedHeaderFiles:
560  acc.merge(FPGATrackSimDataPrepAlgCfg(flags))
561  acc.merge(FPGAConversionAlgCfg(flags, name = 'FPGAConversionAlg', stage = '_1st',**{
562  'FPGATrackSimSPKey': 'FPGAClusters_1st',
563  'xAODPixelClusterFromFPGAClusterKey': 'FPGAPixelClusters',
564  'xAODStripClusterFromFPGAClusterKey': 'FPGAStripClusters',
565  'xAODPixelSpacePointFromFPGAKey': 'ITkPixelSpacePoints',
566  'doActsTrk': False,
567  'useRoads': False,
568  'doSP': flags.Trigger.FPGATrackSim.convertSPs,
569  }))
570  else:
571  from FPGATrackSimConfTools.FPGATrackSimMergeOutputsConfig import FPGATrackSimMergeOutputsAlgCfg
572  acc.merge(FPGATrackSimMergeOutputsAlgCfg(flags))
573  acc.merge(FPGAConversionAlgCfg(flags, name = 'FPGAConversionAlg', stage = '_1st',**{
574  'FPGATrackSimHitKey': 'FPGAHits',
575  'xAODPixelClusterFromFPGAHitKey': 'FPGAPixelClusters',
576  'xAODStripClusterFromFPGAHitKey': 'FPGAStripClusters',
577  'xAODPixelSpacePointFromFPGAKey': 'ITkPixelSpacePoints',
578  'doActsTrk': True,
579  'useRoads': False,
580  'doClusters': False,
581  'doHits': True,
582  'doSP': flags.Trigger.FPGATrackSim.convertSPs,
583  }))
584 
585  from FPGAClusterSorting.FPGAClusterSortingConfig import FPGAClusterSortingAlgCfg
586  ClusterSorting = FPGAClusterSortingAlgCfg(flags,**{'xAODPixelClusterContainer': 'FPGAPixelClusters',
587  'xAODStripClusterContainer': 'FPGAStripClusters',
588  'sortedxAODPixelClusterContainer': 'ITkPixelClusters',
589  'sortedxAODStripClusterContainer': 'ITkStripClusters'})
590 
591  acc.merge(ClusterSorting)
592 
593  from PixelConditionsAlgorithms.ITkPixelConditionsConfig import ITkPixelDetectorElementStatusAlgCfg
594  acc.merge(ITkPixelDetectorElementStatusAlgCfg(flags))
595 
596  from SCT_ConditionsAlgorithms.ITkStripConditionsAlgorithmsConfig import ITkStripDetectorElementStatusAlgCfg
597  acc.merge(ITkStripDetectorElementStatusAlgCfg(flags))
598 
599  if flags.Acts.EDM.PersistifyClusters or flags.Acts.EDM.PersistifySpacePoints:
600  toAOD = []
601 
602  pixel_cluster_shortlist = ['-pixelClusterLink']
603  strip_cluster_shortlist = ['-sctClusterLink']
604 
605  pixel_cluster_variables = '.'.join(pixel_cluster_shortlist)
606  strip_cluster_variables = '.'.join(strip_cluster_shortlist)
607 
608  toAOD += ['xAOD::PixelClusterContainer#ITkPixelClusters',
609  'xAOD::PixelClusterAuxContainer#ITkPixelClustersAux.' + pixel_cluster_variables,
610  'xAOD::StripClusterContainer#ITkStripClusters',
611  'xAOD::StripClusterAuxContainer#ITkStripClustersAux.' + strip_cluster_variables]
612  from OutputStreamAthenaPool.OutputStreamConfig import addToAOD
613  acc.merge(addToAOD(flags, toAOD))
614 
615 
616  from FPGATrackSimReporting.FPGATrackSimReportingConfig import FPGATrackSimReportingCfg
617  acc.merge(FPGATrackSimReportingCfg(flags,
618  perEventReports = False, # set to True if per-event information is needed for debugging (e.g. cluster, tracks). Otherwise it produces a lot of output
619  **{'xAODPixelClusterContainers' : ['ITkPixelClusters'],
620  'xAODStripClusterContainers' : ['ITkStripClusters'],
621  'FPGAActsTracks' : [f'{flags.Tracking.ActiveConfig.extension}Tracks',f'SiSPTracksSeedSegments{flags.Tracking.ActiveConfig.extension}PixelTracks'],
622  'isDataPrep': True} ))
623 
624  return acc
625 

◆ FPGATrackSimDataPrepAlgCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepAlgCfg (   inputFlags)

Definition at line 374 of file FPGATrackSimDataPrepConfig.py.

374 def FPGATrackSimDataPrepAlgCfg(inputFlags):
375 
376  flags = prepareFlagsForFPGATrackSimDataPrepAlg(inputFlags)
377 
378  result=ComponentAccumulator()
379 
380 
381  theFPGATrackSimDataPrepAlg=CompFactory.FPGATrackSimDataPrepAlg()
382  theFPGATrackSimDataPrepAlg.HitFiltering = flags.Trigger.FPGATrackSim.ActiveConfig.hitFiltering
383  theFPGATrackSimDataPrepAlg.writeOutputData = flags.Trigger.FPGATrackSim.writeAdditionalOutputData
384  writePreClusterBranch=flags.Trigger.FPGATrackSim.writeAdditionalOutputData and (flags.Trigger.FPGATrackSim.regionToWriteDPTree < 0)
385  theFPGATrackSimDataPrepAlg.writePreClusterBranch=writePreClusterBranch
386 
387  theFPGATrackSimDataPrepAlg.Clustering = flags.Trigger.FPGATrackSim.clustering
388  theFPGATrackSimDataPrepAlg.doEvtSel= False if flags.Trigger.FPGATrackSim.pipeline.startswith('F-1') or flags.Trigger.FPGATrackSim.sampleType == 'skipTruth' else True
389  theFPGATrackSimDataPrepAlg.useInternalTruthTracks = flags.Trigger.FPGATrackSim.useFPGATruthTrackMatching
390  theFPGATrackSimDataPrepAlg.recordHits = not flags.Trigger.FPGATrackSim.pipeline.startswith('F-1')
391 
392 
393  theFPGATrackSimDataPrepAlg.RawToLogicalHitsTools = []
394  for region in flags.Trigger.FPGATrackSim.regionList:
395  flagsForEachRegion = inputFlags.clone()
396  flagsForEachRegion = flagsForEachRegion.cloneAndReplace("Trigger.FPGATrackSim.ActiveConfig", "Trigger.FPGATrackSim." + inputFlags.Trigger.FPGATrackSim.algoTag,keepOriginal=True)
397  flagsForEachRegion.Trigger.FPGATrackSim.region = region
398  flagsForEachRegion.lock()
399 
400  theFPGATrackSimDataPrepAlg.RawToLogicalHitsTools.append(result.getPrimaryAndMerge(FPGATrackSimRawLogicCfg(flagsForEachRegion)))
401  theFPGATrackSimDataPrepAlg.eventSelectors.append(result.getPrimaryAndMerge(FPGATrackSimEventSelectionToolCfg(flagsForEachRegion)))
402 
403  if flags.Trigger.FPGATrackSim.wrapperFileName and flags.Trigger.FPGATrackSim.wrapperFileName is not None:
404  theFPGATrackSimDataPrepAlg.InputTool = result.getPrimaryAndMerge(FPGATrackSimReadInputCfg(flags))
405  if flags.Trigger.FPGATrackSim.wrapperFileName2 and flags.Trigger.FPGATrackSim.wrapperFileName2 is not None:
406  theFPGATrackSimDataPrepAlg.InputTool2 = result.getPrimaryAndMerge(FPGATrackSimReadInput2Cfg(flags))
407  theFPGATrackSimDataPrepAlg.SecondInputToolN = flags.Trigger.FPGATrackSim.secondInputToolN
408  theFPGATrackSimDataPrepAlg.SGInputTool = ""
409  else:
410  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometryToolCfg
411  result.getPrimaryAndMerge(ActsTrackingGeometryToolCfg(flags))
412  theFPGATrackSimDataPrepAlg.InputTool = ""
413  theFPGATrackSimDataPrepAlg.InputTool2 = ""
414  from FPGATrackSimSGInput.FPGATrackSimSGInputConfig import FPGATrackSimSGInputToolCfg
415  theFPGATrackSimDataPrepAlg.SGInputTool = result.getPrimaryAndMerge(FPGATrackSimSGInputToolCfg(flags))
416  theFPGATrackSimDataPrepAlg.SGInputTool.ReadOfflineClusters=False
417  theFPGATrackSimDataPrepAlg.SGInputTool.ReadOfflineTracks=False
418 
419  theFPGATrackSimDataPrepAlg.HitFilteringTool = result.getPrimaryAndMerge(FPGATrackSimHitFilteringToolCfg(flags))
420 
421  theFPGATrackSimDataPrepAlg.ClusteringTool = CompFactory.FPGATrackSimClusteringTool()
422  if not flags.Trigger.FPGATrackSim.wrapperFileName:
423  theFPGATrackSimDataPrepAlg.ClusteringTool.LorentzAngleTool = result.getPrimaryAndMerge(FPGATrackSimLorentzAngleToolCfg(flags))
424  theFPGATrackSimDataPrepAlg.ClusteringTool.LorentzAngleShift = flags.Trigger.FPGATrackSim.LorentzAngleShift
425  else:
426  theFPGATrackSimDataPrepAlg.ClusteringTool.LorentzAngleShift = -1 # should be set to False by default in the tool class but just in case...
427 
428 
429 
430  theFPGATrackSimDataPrepAlg.OutputTool = result.getPrimaryAndMerge(FPGATrackSimDataPrepOutputCfg(flags))
431 
432  from FPGATrackSimAlgorithms.FPGATrackSimAlgorithmConfig import FPGATrackSimDataPrepMonitoringCfg
433  theFPGATrackSimDataPrepAlg.MonTool = result.getPrimaryAndMerge(FPGATrackSimDataPrepMonitoringCfg(flags))
434 
435  result.addEventAlgo(theFPGATrackSimDataPrepAlg)
436 
437  return result
438 
439 

◆ FPGATrackSimDataPrepConnectToFastTracking()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepConnectToFastTracking (   flagsIn,
  FinalTracks = "F100-",
**  kwargs 
)

Definition at line 442 of file FPGATrackSimDataPrepConfig.py.

442 def FPGATrackSimDataPrepConnectToFastTracking(flagsIn,FinalTracks="F100-", **kwargs):
443 
444  flags = flagsIn.clone()
445 
446  # configure FastTracking based on C-100 flags
447  from ActsConfig.ActsCIFlags import actsWorkflowFlags
448  actsWorkflowFlags(flags)
449 
450  flags.Tracking.ActiveConfig.extension=FinalTracks
451  flags.lock()
452 
453  flags = flags.cloneAndReplace("Tracking.ActiveConfig", "Tracking.ITkMainPass") # TODO: Check if it's really necessary
454  prefix=flags.Tracking.ActiveConfig.extension # prefix for the name of final tracks (this is what IDTPM reads)
455 
456  result = ComponentAccumulator()
457 
458 
459 
460  from ActsConfig.ActsUtilities import extractChildKwargs
461 
462 
465  kwargs.setdefault('PixelSeedingAlg.InputSpacePoints',['ITkPixelSpacePoints'])
466  kwargs.setdefault('StripSeedingAlg.InputSpacePoints',['ITkStripSpacePoints']) # possibly will never be used but in case it's needed, the strip SP conversion should be enabled for this container to be available in F100 (off by default)
467 
468  # -- Track Finding args --
469  kwargs.setdefault('TrackFindingAlg.UncalibratedMeasurementContainerKeys',['ITkPixelClusters','ITkStripClusters'])
470 
471  # -- Truth Matching args --
472  kwargs.setdefault('PixelClusterToTruthAssociationAlg.Measurements','ITkPixelClusters')
473  kwargs.setdefault('StripClusterToTruthAssociationAlg.Measurements','ITkStripClusters')
474 
475 
477  from ActsConfig.ActsSeedingConfig import ActsSeedingCfg
478  result.merge(ActsSeedingCfg(flags, **kwargs))
479 
480  # ACTS Track Finding
481  from ActsConfig.ActsTrackFindingConfig import ActsTrackFindingCfg,ActsAmbiguityResolutionCfg
482  result.merge(ActsTrackFindingCfg(flags,**extractChildKwargs(prefix='TrackFindingAlg.', **kwargs)))
483 
484  # if ambiguity is enabled for FastTracking run here as well
485  if flags.Acts.doAmbiguityResolution:
486  result.merge(ActsAmbiguityResolutionCfg(flags,**extractChildKwargs(prefix='AmbiguityResolutionAlg.',**kwargs)))
487 
488  # modify the tracks' name (not the final one) accordingly in case ambiguity resolution runs
489  acts_tracks=f"{prefix}Tracks" if not flags.Acts.doAmbiguityResolution else f"{prefix}ResolvedTracks"
490 
491 
493  if(flags.ITk.doTruth):
494  from ActsConfig.ActsTruthConfig import ActsTruthParticleHitCountAlgCfg, ActsPixelClusterToTruthAssociationAlgCfg,ActsStripClusterToTruthAssociationAlgCfg
495  result.merge(ActsPixelClusterToTruthAssociationAlgCfg(flags,
496  name=f"{prefix}PixelClusterToTruthAssociationAlg",
497  InputTruthParticleLinks="xAODTruthLinks",
498  AssociationMapOut=f"{prefix}ITkPixelClustersToTruthParticles",
499  Measurements=kwargs.get('PixelClusterToTruthAssociationAlg.Measurements')))
500 
501  result.merge(ActsStripClusterToTruthAssociationAlgCfg(flags,
502  name=f"{prefix}StripClusterToTruthAssociationAlg",
503  InputTruthParticleLinks="xAODTruthLinks",
504  AssociationMapOut=f"{prefix}ITkStripClustersToTruthParticles",
505  Measurements=kwargs.get('StripClusterToTruthAssociationAlg.Measurements')))
506 
507  result.merge(ActsTruthParticleHitCountAlgCfg(flags,
508  name=f"{prefix}TruthParticleHitCountAlg",
509  PixelClustersToTruthAssociationMap=f"{prefix}ITkPixelClustersToTruthParticles",
510  StripClustersToTruthAssociationMap=f"{prefix}ITkStripClustersToTruthParticles",
511  TruthParticleHitCountsOut=f"{prefix}TruthParticleHitCounts"))
512 
513  from ActsConfig.ActsTruthConfig import ActsTrackToTruthAssociationAlgCfg, ActsTrackFindingValidationAlgCfg
514  result.merge(ActsTrackToTruthAssociationAlgCfg(flags,
515  name=f"{prefix}TrackToTruthAssociationAlg",
516  PixelClustersToTruthAssociationMap=f"{prefix}ITkPixelClustersToTruthParticles",
517  StripClustersToTruthAssociationMap=f"{prefix}ITkStripClustersToTruthParticles",
518  ACTSTracksLocation=acts_tracks,
519  AssociationMapOut=f"{acts_tracks}ToTruthParticleAssociation"))
520 
521  result.merge(ActsTrackFindingValidationAlgCfg(flags,
522  name=f"{prefix}TrackFindingValidationAlg",
523  TrackToTruthAssociationMap=f"{acts_tracks}ToTruthParticleAssociation",
524  TruthParticleHitCounts=f"{prefix}TruthParticleHitCounts"
525  ))
526 
528  from ActsConfig.ActsTrackFindingConfig import ActsTrackToTrackParticleCnvAlgCfg
529  result.merge(ActsTrackToTrackParticleCnvAlgCfg(flags, name=f"{prefix}TrackToTrackParticleCnvAlg",
530  ACTSTracksLocation=[acts_tracks],
531  TrackParticlesOutKey=f"{prefix}TrackParticles"))
532  if(flags.ITk.doTruth):
533  from ActsConfig.ActsTruthConfig import ActsTrackParticleTruthDecorationAlgCfg
534  result.merge(ActsTrackParticleTruthDecorationAlgCfg(flags, name=f"{prefix}TrackParticleTruthDecorationAlg",
535  TrackToTruthAssociationMaps=[f"{acts_tracks}ToTruthParticleAssociation"],
536  TrackParticleContainerName=f"{FinalTracks}TrackParticles",
537  TruthParticleHitCounts=f"{prefix}TruthParticleHitCounts",
538  ComputeTrackRecoEfficiency=True))
539 
540 
541  if flags.Trigger.FPGATrackSim.writeOfflPRDInfo:
542  from InDetConfig.InDetPrepRawDataToxAODConfig import ITkActsPrepDataToxAODCfg
543  result.merge( ITkActsPrepDataToxAODCfg( flags,
544  PixelMeasurementContainer = "ITkPixelMeasurements_offl",
545  StripMeasurementContainer = "ITkStripMeasurements_offl" ) )
546 
547  return result
548 

◆ FPGATrackSimDataPrepFlagCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepFlagCfg (   flags)

Definition at line 626 of file FPGATrackSimDataPrepConfig.py.

626 def FPGATrackSimDataPrepFlagCfg(flags): # to be used in the Reco_tf configuration
627  flags.Scheduler.ShowDataDeps=True
628  flags.Scheduler.CheckDependencies=True
629 
630  flags.Trigger.FPGATrackSim.readOfflineObjects=False
631  flags.Trigger.FPGATrackSim.doMultiTruth=False
632 
633  flags = FPGATrackSimRegionFlagCfg(flags)
634 
635  return flags
636 

◆ FPGATrackSimDataPrepOutputCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepOutputCfg (   flags)

Definition at line 163 of file FPGATrackSimDataPrepConfig.py.

164  result=ComponentAccumulator()
165 
166  FPGATrackSimWriteOutput = CompFactory.FPGATrackSimOutputHeaderTool("FPGATrackSimWriteOutputDataPrep")
167  FPGATrackSimWriteOutput.InFileName = ["test.root"]
168  FPGATrackSimWriteOutput.OutputTreeName = "FPGATrackSimDataPrepTree"
169  writeThis=flags.Trigger.FPGATrackSim.writeAdditionalOutputData and ((flags.Trigger.FPGATrackSim.regionToWriteDPTree in flags.Trigger.FPGATrackSim.regionList) or (flags.Trigger.FPGATrackSim.regionToWriteDPTree < 0))
170 
171  if not writeThis:
172  FPGATrackSimWriteOutput.EventLimit = 0
173  else:
174  FPGATrackSimWriteOutput.EventLimit = flags.Trigger.FPGATrackSim.writeOutputEventLimit
175  # RECREATE means that that this tool opens the file.
176  # HEADER would mean that something else (e.g. THistSvc) opens it and we just add the object.
177  FPGATrackSimWriteOutput.RWstatus = "HEADER"
178  FPGATrackSimWriteOutput.THistSvc = CompFactory.THistSvc()
179  result.addPublicTool(FPGATrackSimWriteOutput, primary=True)
180  return result
181 

◆ FPGATrackSimDataPrepSetup()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepSetup (   flags,
  runReco = True 
)

Definition at line 646 of file FPGATrackSimDataPrepConfig.py.

646 def FPGATrackSimDataPrepSetup(flags,runReco=True):
647  acc = ComponentAccumulator()
648  if flags.Trigger.FPGATrackSim.wrapperFileName and flags.Trigger.FPGATrackSim.wrapperFileName is not None:
649  return acc
650 
651  from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
652  acc.merge(PoolReadCfg(flags))
653 
654  if not runReco:
655  from xAODEventInfoCnv.xAODEventInfoCnvConfig import EventInfoCnvAlgCfg
656  acc.merge(EventInfoCnvAlgCfg(flags))
657  else:
658  if flags.Input.isMC:
659  from xAODTruthCnv.xAODTruthCnvConfig import GEN_AOD2xAODCfg
660  acc.merge(GEN_AOD2xAODCfg(flags))
661 
662  from JetRecConfig.JetRecoSteering import addTruthPileupJetsToOutputCfg # TO DO: check if this is indeed necessary for pileup samples
663  acc.merge(addTruthPileupJetsToOutputCfg(flags))
664 
665  if flags.Detector.EnableCalo:
666  from CaloRec.CaloRecoConfig import CaloRecoCfg
667  acc.merge(CaloRecoCfg(flags))
668 
669  if not flags.Reco.EnableTrackOverlay:
670 
671  from InDetConfig.TrackRecoConfig import InDetTrackRecoCfg
672  acc.merge(InDetTrackRecoCfg(flags))
673  from InDetConfig.InDetPrepRawDataFormationConfig import ITkXAODToInDetClusterConversionCfg
674  acc.merge(ITkXAODToInDetClusterConversionCfg(flags))
675  from InDetConfig.InDetPrepRawDataToxAODConfig import TruthParticleIndexDecoratorAlgCfg
676  acc.merge( TruthParticleIndexDecoratorAlgCfg(flags) )
677 
678  return acc
679 
680 

◆ FPGATrackSimEventSelectionSvcCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimEventSelectionSvcCfg (   flags,
  name = "FPGATrackSimEventSelectionSvc" 
)

Definition at line 274 of file FPGATrackSimDataPrepConfig.py.

274 def FPGATrackSimEventSelectionSvcCfg(flags,name="FPGATrackSimEventSelectionSvc"):
275 
276  result=ComponentAccumulator()
277  eventSelector = CompFactory.FPGATrackSimEventSelectionSvc(nameWithRegionSuffix(flags,name))
278  eventSelector.regions = flags.Trigger.FPGATrackSim.slicesFile
279  eventSelector.regionID = flags.Trigger.FPGATrackSim.region
280  eventSelector.sampleType = flags.Trigger.FPGATrackSim.sampleType
281  eventSelector.skipRegionCheck = flags.Trigger.FPGATrackSim.pipeline.startswith('F-1') # if set to True, it will essentially run for the whole detector
282  eventSelector.withPU = False
283  eventSelector.oldRegionDefs = flags.Trigger.FPGATrackSim.oldRegionDefs
284 
285 
286  eventSelector.mind0 = flags.Trigger.FPGATrackSim.d0min
287  eventSelector.maxd0 = flags.Trigger.FPGATrackSim.d0max
288  eventSelector.minz0 = flags.Trigger.FPGATrackSim.z0min
289  eventSelector.maxz0 = flags.Trigger.FPGATrackSim.z0max
290  eventSelector.minqOverPt = flags.Trigger.FPGATrackSim.qOverPtmin
291  eventSelector.maxqOverPt = flags.Trigger.FPGATrackSim.qOverPtmax
292 
293  result.addService(eventSelector, create=True, primary=True)
294  return result
295 

◆ FPGATrackSimEventSelectionToolCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimEventSelectionToolCfg (   flags,
  name = "FPGATrackSimEventSelectionTool" 
)

Definition at line 296 of file FPGATrackSimDataPrepConfig.py.

296 def FPGATrackSimEventSelectionToolCfg(flags,name="FPGATrackSimEventSelectionTool"):
297  result=ComponentAccumulator()
298  eventSelectionTool=CompFactory.FPGATrackSim.FPGATrackSimEventSelectionTool(nameWithRegionSuffix(flags,name))
299  eventSelectionTool.evtSelectionService = result.getPrimaryAndMerge(FPGATrackSimEventSelectionSvcCfg(flags))
300 
301  result.addPublicTool(eventSelectionTool, primary=True)
302  return result
303 

◆ FPGATrackSimHitFilteringToolCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimHitFilteringToolCfg (   flags)

Definition at line 344 of file FPGATrackSimDataPrepConfig.py.

345  result=ComponentAccumulator()
346  HitFilteringTool = CompFactory.FPGATrackSimHitFilteringTool()
347  HitFilteringTool.barrelStubDphiCut = 3.0
348  HitFilteringTool.doRandomRemoval = False
349  HitFilteringTool.doStubs = False
350  HitFilteringTool.endcapStubDphiCut = 1.5
351  HitFilteringTool.pixelClusRmFrac = 0
352  HitFilteringTool.pixelHitRmFrac = 0
353  HitFilteringTool.stripClusRmFrac = 0
354  HitFilteringTool.stripHitRmFrac = 0
355  HitFilteringTool.useNstrips = False
356  result.addPublicTool(HitFilteringTool, primary=True)
357  return result
358 

◆ FPGATrackSimLorentzAngleToolCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimLorentzAngleToolCfg (   flags)

Definition at line 359 of file FPGATrackSimDataPrepConfig.py.

360  result=ComponentAccumulator()
361  LorentzAngleTool = CompFactory.FPGATrackSim.LorentzAngleTool()
362  LorentzAngleTool.UseAthenaLorentzAngleTools=False
363  LorentzAngleTool.shiftGlobalPosition=True
364 
365  from SiLorentzAngleTool.ITkPixelLorentzAngleConfig import ITkPixelLorentzAngleToolCfg
366  LorentzAngleTool.LorentzAngleToolPixel = result.popToolsAndMerge(ITkPixelLorentzAngleToolCfg(flags))
367 
368  from SiLorentzAngleTool.ITkStripLorentzAngleConfig import ITkStripLorentzAngleToolCfg
369  LorentzAngleTool.LorentzAngleToolStrip = result.popToolsAndMerge(ITkStripLorentzAngleToolCfg(flags))
370 
371  result.addPublicTool(LorentzAngleTool, primary=True)
372  return result
373 

◆ FPGATrackSimMappingCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimMappingCfg (   flags,
  name = "FPGATrackSimMappingSvc" 
)

Definition at line 304 of file FPGATrackSimDataPrepConfig.py.

304 def FPGATrackSimMappingCfg(flags,name="FPGATrackSimMappingSvc"):
305  result=ComponentAccumulator()
306  mappingSvc = CompFactory.FPGATrackSimMappingSvc(nameWithRegionSuffix(flags,name))
307  mappingSvc.regionID = flags.Trigger.FPGATrackSim.region
308  mappingSvc.mappingType = "FILE"
309  mappingSvc.rmap = flags.Trigger.FPGATrackSim.mapsDir+"/"+getBaseName(flags)+".rmap" # we need more configurability here i.e. file choice should depend on some flag
310  mappingSvc.subrmap = flags.Trigger.FPGATrackSim.mapsDir+"/"+getBaseName(flags)+".rmap" # At this point this is the same as the region map, so let's not have two files
311  mappingSvc.pmap = flags.Trigger.FPGATrackSim.mapsDir+"/"+getBaseName(flags)+".pmap"
312  mappingSvc.modulemap = flags.Trigger.FPGATrackSim.mapsDir+"/moduleidmap"
313  mappingSvc.radiiFile = flags.Trigger.FPGATrackSim.mapsDir + "/"+getBaseName(flags)+"_radii.txt"
314  mappingSvc.radiiFile2nd = flags.Trigger.FPGATrackSim.mapsDir + "/"+getBaseName(flags)+"_radii_2nd.txt"
315  mappingSvc.loadRadii = (not flags.Trigger.FPGATrackSim.ActiveConfig.GNN) and flags.Trigger.FPGATrackSim.loadRadii
316  mappingSvc.loadRegionMap = flags.Trigger.FPGATrackSim.loadRegionMap
317  mappingSvc.DoGNNTrack = flags.Trigger.FPGATrackSim.GNN.doGNNTracking
318 
319  mappingSvc.FakeNNonnx1st = flags.Trigger.FPGATrackSim.FakeNNonnxFile1st
320  mappingSvc.FakeNNonnx2nd = flags.Trigger.FPGATrackSim.FakeNNonnxFile2nd
321  mappingSvc.ParamNNonnx1st = flags.Trigger.FPGATrackSim.ParamNNonnxFile1st
322  mappingSvc.ParamNNonnx2nd = flags.Trigger.FPGATrackSim.ParamNNonnxFile2nd
323  mappingSvc.ExtensionNNVolonnx = flags.Trigger.FPGATrackSim.ExtensionNNVolonnxFile
324  mappingSvc.ExtensionNNHitonnx = flags.Trigger.FPGATrackSim.ExtensionNNHitonnxFile
325  mappingSvc.GNNModuleMap = flags.Trigger.FPGATrackSim.GNN.moduleMapPath
326  mappingSvc.layerOverride = []
327  result.addService(mappingSvc, create=True, primary=True)
328  return result
329 
330 

◆ FPGATrackSimRawLogicCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimRawLogicCfg (   flags,
  name = "FPGATrackSimRawLogicTool" 
)

Definition at line 135 of file FPGATrackSimDataPrepConfig.py.

135 def FPGATrackSimRawLogicCfg(flags,name="FPGATrackSimRawLogicTool"):
136  result=ComponentAccumulator()
137  FPGATrackSimRawLogic = CompFactory.FPGATrackSimRawToLogicalHitsTool(nameWithRegionSuffix(flags,name))
138  FPGATrackSimRawLogic.SaveOptional = 2
139  if (flags.Trigger.FPGATrackSim.ActiveConfig.sampleType == 'skipTruth'):
140  FPGATrackSimRawLogic.SaveOptional = 1
141  FPGATrackSimRawLogic.TowersToMap = [0] # TODO TODO why is this hardcoded?
142  FPGATrackSimRawLogic.FPGATrackSimEventSelectionSvc = result.getPrimaryAndMerge(FPGATrackSimEventSelectionSvcCfg(flags))
143  FPGATrackSimRawLogic.FPGATrackSimMappingSvc = result.getPrimaryAndMerge(FPGATrackSimMappingCfg(flags))
144  result.addPublicTool(FPGATrackSimRawLogic,primary=True)
145  return result
146 

◆ FPGATrackSimReadInput2Cfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimReadInput2Cfg (   flags)

Definition at line 338 of file FPGATrackSimDataPrepConfig.py.

338 def FPGATrackSimReadInput2Cfg(flags):
339  result=ComponentAccumulator()
340  InputTool2 = CompFactory.FPGATrackSimReadRawRandomHitsTool(name="FPGATrackSimReadInput2", InFileName = flags.Trigger.FPGATrackSim.wrapperFileName2)
341  result.addPublicTool(InputTool2, primary=True)
342  return result
343 

◆ FPGATrackSimReadInputCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimReadInputCfg (   flags)

Definition at line 331 of file FPGATrackSimDataPrepConfig.py.

331 def FPGATrackSimReadInputCfg(flags):
332  result=ComponentAccumulator()
333  InputTool = CompFactory.FPGATrackSimInputHeaderTool(name="FPGATrackSimReadInput",
334  InFileName = flags.Trigger.FPGATrackSim.wrapperFileName)
335  result.addPublicTool(InputTool, primary=True)
336  return result
337 

◆ FPGATrackSimRegionFlagCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimRegionFlagCfg (   flags)

Definition at line 549 of file FPGATrackSimDataPrepConfig.py.

549 def FPGATrackSimRegionFlagCfg(flags):
550  if flags.Trigger.FPGATrackSim.regionList == "": # in case of empty list just use the region set to flags.Trigger.FPGATrackSim.region
551  flags.Trigger.FPGATrackSim.regionList = [flags.Trigger.FPGATrackSim.region]
552  else: # otherwise use the regionList (this overrides the region flag)
553  from FPGATrackSimConfTools.FPGATrackSimHelperFunctions import convertRegionsExpressionToArray
554  flags.Trigger.FPGATrackSim.regionList = convertRegionsExpressionToArray(flags.Trigger.FPGATrackSim.regionList)
555  return flags
556 

◆ FPGATrackSimSpacePointsToolCfg()

def python.FPGATrackSimDataPrepConfig.FPGATrackSimSpacePointsToolCfg (   flags)

Definition at line 147 of file FPGATrackSimDataPrepConfig.py.

148  result=ComponentAccumulator()
149  SpacePointTool = CompFactory.FPGATrackSimSpacePointsTool()
150  SpacePointTool.Filtering = flags.Trigger.FPGATrackSim.ActiveConfig.spacePointFiltering
151  SpacePointTool.FilteringClosePoints = False
152  SpacePointTool.PhiWindow = 0.004
153  SpacePointTool.Duplication = True
154  result.addPublicTool(SpacePointTool, primary=True)
155  return result
156 
157 

◆ getBaseName()

def python.FPGATrackSimDataPrepConfig.getBaseName (   flags)

Definition at line 43 of file FPGATrackSimDataPrepConfig.py.

43 def getBaseName(flags):
44  if (flags.Trigger.FPGATrackSim.oldRegionDefs):
45  if (not (flags.Trigger.FPGATrackSim.baseName == '')):
46  return flags.Trigger.FPGATrackSim.baseName
47  elif (flags.Trigger.FPGATrackSim.region == 0):
48  return 'eta0103phi0305'
49  elif (flags.Trigger.FPGATrackSim.region == 1):
50  return 'eta0709phi0305'
51  elif (flags.Trigger.FPGATrackSim.region == 2):
52  return 'eta1214phi0305'
53  elif (flags.Trigger.FPGATrackSim.region == 3):
54  return 'eta2022phi0305'
55  elif (flags.Trigger.FPGATrackSim.region == 4):
56  return 'eta3234phi0305'
57  elif (flags.Trigger.FPGATrackSim.region == 5):
58  return 'eta0103phi1113'
59  elif (flags.Trigger.FPGATrackSim.region == 6):
60  return 'eta0103phi1921'
61  elif (flags.Trigger.FPGATrackSim.region == 7):
62  return 'eta0103phi3436'
63  else:
64  return 'default'
65  else:
66  if (flags.Trigger.FPGATrackSim.region >= 1280 or flags.Trigger.FPGATrackSim.region < 0): return 'default'
67  else:
68  return "region"+str(flags.Trigger.FPGATrackSim.region)
69 

◆ getEtaRange()

def python.FPGATrackSimDataPrepConfig.getEtaRange (   flags)

Definition at line 99 of file FPGATrackSimDataPrepConfig.py.

99 def getEtaRange(flags):
100  if (flags.Trigger.FPGATrackSim.oldRegionDefs):
101  if (not (flags.Trigger.FPGATrackSim.baseName == '')):
102  return [0.1,0.3]
103  elif (flags.Trigger.FPGATrackSim.region == 0):
104  return [0.1,0.3]
105  elif (flags.Trigger.FPGATrackSim.region == 1):
106  return [0.7,0.9]
107  elif (flags.Trigger.FPGATrackSim.region == 2):
108  return [1.2,1.4]
109  elif (flags.Trigger.FPGATrackSim.region == 3):
110  return [2.0,2.2]
111  elif (flags.Trigger.FPGATrackSim.region == 4):
112  return [3.2,3.4]
113  elif (flags.Trigger.FPGATrackSim.region == 5):
114  return [0.1,0.3]
115  elif (flags.Trigger.FPGATrackSim.region == 6):
116  return [0.1,0.3]
117  elif (flags.Trigger.FPGATrackSim.region == 7):
118  return [0.1,0.3]
119  else:
120  return [0.3,0.5]
121  else:
122  if (flags.Trigger.FPGATrackSim.region >= 1280 or flags.Trigger.FPGATrackSim.region < 0): return [0.2,0.4]
123  else:
124  binSize = 0.2
125  side = (flags.Trigger.FPGATrackSim.region >> 5) & 0x1
126  etaBin = (flags.Trigger.FPGATrackSim.region >> 6) & 0x1f
127  if (side): return [binSize*etaBin,binSize*(etaBin+1)]
128  else: return [-binSize*(etaBin+1),-binSize*etaBin]
129 
130 

◆ getEtaSideBits()

def python.FPGATrackSimDataPrepConfig.getEtaSideBits (   flags)

Definition at line 131 of file FPGATrackSimDataPrepConfig.py.

131 def getEtaSideBits(flags):
132  # this includes both the eta and side bits
133  return flags.Trigger.FPGATrackSim.region >> 5
134 

◆ getPhiRange()

def python.FPGATrackSimDataPrepConfig.getPhiRange (   flags)

Definition at line 70 of file FPGATrackSimDataPrepConfig.py.

70 def getPhiRange(flags):
71  if (flags.Trigger.FPGATrackSim.oldRegionDefs):
72  if (not (flags.Trigger.FPGATrackSim.baseName == '')):
73  return [0.3,0.5]
74  elif (flags.Trigger.FPGATrackSim.region == 0):
75  return [0.3,0.5]
76  elif (flags.Trigger.FPGATrackSim.region == 1):
77  return [0.3,0.5]
78  elif (flags.Trigger.FPGATrackSim.region == 2):
79  return [0.3,0.5]
80  elif (flags.Trigger.FPGATrackSim.region == 3):
81  return [0.3,0.5]
82  elif (flags.Trigger.FPGATrackSim.region == 4):
83  return [0.3,0.5]
84  elif (flags.Trigger.FPGATrackSim.region == 5):
85  return [1.1,1.3]
86  elif (flags.Trigger.FPGATrackSim.region == 6):
87  return [1.9,2.1]
88  elif (flags.Trigger.FPGATrackSim.region == 7):
89  return [3.4,3.6]
90  else:
91  return [0.3,0.5]
92  else:
93  binSize = pi/16
94  phiBin=flags.Trigger.FPGATrackSim.region & 0x1f
95  if (flags.Trigger.FPGATrackSim.region >= 1280 or flags.Trigger.FPGATrackSim.region < 0): return [binSize*2,binSize*3]
96  else:
97  return [binSize*phiBin,binSize*(phiBin+1)]
98 

◆ getRegionNumber()

def python.FPGATrackSimDataPrepConfig.getRegionNumber (   nPiOver16Min,
  minAbsEta,
  isPositiveEta,
  verbosePrint = True 
)

Definition at line 12 of file FPGATrackSimDataPrepConfig.py.

12 def getRegionNumber(nPiOver16Min, minAbsEta, isPositiveEta, verbosePrint=True):
13  binSizePhi = pi/16
14  binSizeEta = 0.2
15  region = nPiOver16Min
16  if (nPiOver16Min >= 32 or nPiOver16Min < 0):
17  print("You are asking for nPiOver16 = ",nPiOver16Min," but we only have 32 phi bins! Returning -1")
18  return -1
19  if (minAbsEta > 3.8 or minAbsEta < 0):
20  print("You are asking for minAbsEta = ",minAbsEta," but we only have 20 abs eta bins! Returning -1")
21  return -1
22  if not (isPositiveEta == 0 or isPositiveEta == 1):
23  print("You are asking for isPositiveEta = ",isPositiveEta, " but this should only be 0 or 1! Returning -1")
24  return -1
25 
26  etanum = round(minAbsEta / binSizeEta)
27  mineta = binSizeEta*etanum
28  maxeta = binSizeEta*(etanum+1)
29  if (not isPositiveEta):
30  mineta = -binSizeEta*(etanum+1)
31  maxeta = -binSizeEta*etanum
32 
33  region = region | ((isPositiveEta & 0x1) << 5)
34  region = region | (etanum << 6)
35  if (verbosePrint):
36  print("I am returning the region number for minphi =",nPiOver16Min, "*pi/16, which is equal to",binSizePhi*nPiOver16Min)
37  print(" and maxphi =",nPiOver16Min+1, "*pi/16, which is equal to",binSizePhi*(nPiOver16Min+1))
38  print(" and mineta =",mineta, "and max eta =", maxeta)
39 
40  return region
41 
42 

◆ nameWithRegionSuffix()

def python.FPGATrackSimDataPrepConfig.nameWithRegionSuffix (   flags,
  basename 
)

Definition at line 8 of file FPGATrackSimDataPrepConfig.py.

8 def nameWithRegionSuffix(flags, basename):
9  return f"{basename}_reg{flags.Trigger.FPGATrackSim.region}"
10 
11 

◆ prepareFlagsForFPGATrackSimDataPrepAlg()

def python.FPGATrackSimDataPrepConfig.prepareFlagsForFPGATrackSimDataPrepAlg (   flags)

Definition at line 158 of file FPGATrackSimDataPrepConfig.py.

159  newFlags = flags.cloneAndReplace("Trigger.FPGATrackSim.ActiveConfig", "Trigger.FPGATrackSim." + flags.Trigger.FPGATrackSim.algoTag)
160  return newFlags
161 
162 

◆ runDataPrepChain()

def python.FPGATrackSimDataPrepConfig.runDataPrepChain ( )

Definition at line 681 of file FPGATrackSimDataPrepConfig.py.

681 def runDataPrepChain():
682  from AthenaConfiguration.AllConfigFlags import initConfigFlags
683  from AthenaConfiguration.MainServicesConfig import MainServicesCfg
684 
685  FinalDataPrepTrackChainxAODTracksKeyPrefix="FPGA"
686 
687  flags = initConfigFlags()
688  from InDetConfig.ConfigurationHelpers import OnlyTrackingPreInclude
690 
691 
693  flags = FixITkMainPassFlags(flags)
694 
695 
696  flags.Concurrency.NumThreads=1
697  flags.Concurrency.NumConcurrentEvents=1
698  flags.Concurrency.NumProcs=0
699  flags.Scheduler.ShowDataDeps=False
700  flags.Scheduler.CheckDependencies=True
701  flags.Debug.DumpEvtStore=False # Set to Truth to enable Event Store printouts
702  # flags.Exec.DebugStage="exec" # useful option to debug the execution of the job - we want it commented out for production
703  flags.fillFromArgs()
704  if isinstance(flags.Trigger.FPGATrackSim.wrapperFileName, str):
705  log.info("wrapperFile is string, converting to list")
706  flags.Trigger.FPGATrackSim.wrapperFileName = [flags.Trigger.FPGATrackSim.wrapperFileName]
707  flags.Input.Files = lambda f: [f.Trigger.FPGATrackSim.wrapperFileName]
708 
709  flags.Tracking.writeExtendedSi_PRDInfo = not flags.Trigger.FPGATrackSim.writeOfflPRDInfo # Don't write ITkPixel/StripMeasurements if writeOfflPRDInfo = True
710  # In this case, ITkPixel/StripMeasurements_offl written based on F100 clustering
711 
712  flags = FPGATrackSimRegionFlagCfg(flags)
713 
714  flags.lock()
715  flags = flags.cloneAndReplace("Tracking.ActiveConfig", "Tracking.ITkMainPass", keepOriginal=True)
716 
717  acc=MainServicesCfg(flags)
718  if flags.Trigger.FPGATrackSim.writeAdditionalOutputData:
719  acc.addService(CompFactory.THistSvc(Output = [f"EXPERT DATAFILE='{flags.Trigger.FPGATrackSim.outputMonitorFile}', OPT='RECREATE'"]))
720  acc.addService(CompFactory.THistSvc(Output = ["FPGATRACKSIMOUTPUT DATAFILE='dataprep.root', OPT='RECREATE'"]))
721 
722 
723  acc.merge(FPGATrackSimDataPrepSetup(flags))
724 
725  # Use the imported configuration function for the data prep algorithm.
726  acc.merge(FPGATrackSimDataPrepAlgCfg(flags))
727 
728  if flags.Trigger.FPGATrackSim.doEDMConversion:
729  acc.merge(FPGAConversionAlgCfg(flags, name = 'FPGAConversionAlg', stage = '_1st', doActsTrk=False, doSP = False))
730 
731  # convert Pixel Clusters to SPs
732  from ActsConfig.ActsSpacePointFormationConfig import ActsPixelSpacePointFormationAlgCfg
733  acc.merge(ActsPixelSpacePointFormationAlgCfg(flags,name="FPGAActsPixelSpacePointFormationAlg",
734  **{'PixelClusters':"xAODPixelClustersFromFPGACluster",
735  'PixelSpacePoints':"xAODPixelSpacePointsFromFPGA"}))
736 
737  if flags.Trigger.FPGATrackSim.connectToToITkTracking:
738  if flags.Trigger.FPGATrackSim.writeAdditionalOutputData:
739  # Run ACTS Fast Tracking on offline objects (starting from seeding)
740  acc.merge(FPGATrackSimDataPrepConnectToFastTracking(flags, FinalTracks="ActsFast"))
741 
742  # Run ACTS Fast Tracking for FPGA clusters (starting from seeding)
743  acc.merge(FPGATrackSimDataPrepConnectToFastTracking(flags, FinalTracks=FinalDataPrepTrackChainxAODTracksKeyPrefix,
744  **{'PixelSeedingAlg.InputSpacePoints' : ['xAODPixelSpacePointsFromFPGA'],
745  'StripSeedingAlg.InputSpacePoints' : [''],
746  'TrackFindingAlg.UncalibratedMeasurementContainerKeys' : ["xAODPixelClustersFromFPGACluster","xAODStripClustersFromFPGACluster"],
747  'PixelClusterToTruthAssociationAlg.Measurements' : 'xAODPixelClustersFromFPGACluster',
748  'StripClusterToTruthAssociationAlg.Measurements' : 'xAODStripClustersFromFPGACluster'}))
749 
750 
751  if flags.Trigger.FPGATrackSim.writeToAOD:
752  acc.merge(WriteToAOD(flags,
753  stage = '_1st',
754  finalTrackParticles=f"{FinalDataPrepTrackChainxAODTracksKeyPrefix}TrackParticles"))
755 
756  # Printout for various FPGA-related objects
757  from FPGATrackSimReporting.FPGATrackSimReportingConfig import FPGATrackSimReportingCfg
758  acc.merge(FPGATrackSimReportingCfg(flags,stage="_1st",
759  perEventReports = (flags.Trigger.FPGATrackSim.sampleType != 'skipTruth'),
760  isDataPrep=True))
761 
762  # cluster monitoring
763  if flags.Trigger.FPGATrackSim.writeAdditionalOutputData:
764  from EFTrackingFPGAOutputValidation.FPGAOutputValidationConfig import FPGAOutputValidationCfg
765  acc.merge(FPGAOutputValidationCfg(flags, **{'pixelKeys' : ["xAODPixelClustersFromFPGACluster","ITkPixelClusters"],
766  'stripKeys':["xAODStripClustersFromFPGACluster","ITkStripClusters"],
767  'doDiffHistograms':True,
768  'matchByID' : True}))
769 
770  acc.store(open('AnalysisConfig.pkl','wb'))
771 
772  statusCode = acc.run(flags.Exec.MaxEvents)
773  assert statusCode.isSuccess() is True, "Application execution did not succeed"
774 
775 

◆ WriteToAOD()

def python.FPGATrackSimDataPrepConfig.WriteToAOD (   flags,
  stage = '',
  finalTrackParticles = '' 
)

Definition at line 240 of file FPGATrackSimDataPrepConfig.py.

240 def WriteToAOD(flags, stage = '',finalTrackParticles = ''): # store xAOD containers in AOD file
241  result = ComponentAccumulator()
242  from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
243  from OutputStreamAthenaPool.OutputStreamConfig import outputStreamName
244  from AthenaConfiguration.Enums import MetadataCategory
245 
246  result.merge( SetupMetaDataForStreamCfg( flags,"AOD",
247  createMetadata=[
248  MetadataCategory.ByteStreamMetaData,
249  MetadataCategory.LumiBlockMetaData,
250  MetadataCategory.TruthMetaData,
251  MetadataCategory.IOVMetaData,],)
252  )
253  log.info("AOD ItemList: %s", result.getEventAlgo(outputStreamName("AOD")).ItemList)
254  log.info("AOD MetadataItemList: %s", result.getEventAlgo(outputStreamName("AOD")).MetadataItemList)
255  log.info("---------- Configured AOD writing")
256 
257  from OutputStreamAthenaPool.OutputStreamConfig import addToAOD
258  toAOD = [f"xAOD::TrackParticleContainer#{finalTrackParticles}",f"xAOD::TrackParticleAuxContainer#{finalTrackParticles}Aux."]
259  if flags.Trigger.FPGATrackSim.writeClustersToAOD:
260  toAOD += ["xAOD::PixelClusterContainer#xAODPixelClustersFromFPGACluster","xAOD::PixelClusterAuxContainer#xAODPixelClustersFromFPGAClusterAux.",
261  "xAOD::StripClusterContainer#xAODStripClustersFromFPGACluster","xAOD::StripClusterAuxContainer#xAODStripClustersFromFPGAClusterAux."]
262  if flags.Trigger.FPGATrackSim.writeOfflPRDInfo:
263  toAOD += ['xAOD::TrackMeasurementValidationContainer#ITkPixelMeasurements_offl','xAOD::TrackMeasurementValidationAuxContainer#ITkPixelMeasurements_offlAux.',
264  'xAOD::TrackMeasurementValidationContainer#ITkStripMeasurements_offl','xAOD::TrackMeasurementValidationAuxContainer#ITkStripMeasurements_offlAux.']
265  if flags.Trigger.FPGATrackSim.writeAdditionalOutputData:
266  toAOD += [f"xAOD::SpacePointContainer#xAODPixelSpacePoints{stage}FromFPGA",f"xAOD::SpacePointAuxContainer#xAODPixelSpacePoints{stage}FromFPGAAux.-measurements",
267  f"xAOD::SpacePointContainer#xAODStripSpacePoints{stage}FromFPGA",f"xAOD::SpacePointAuxContainer#xAODStripSpacePoints{stage}FromFPGAAux.-measurements.-sctSpacePointLink"]
268 
269  result.merge(addToAOD(flags, toAOD))
270 
271  return result
272 
273 

Variable Documentation

◆ log

python.FPGATrackSimDataPrepConfig.log

Definition at line 440 of file FPGATrackSimDataPrepConfig.py.

AthenaPoolExample_WriteCond.outputStreamName
string outputStreamName
Definition: AthenaPoolExample_WriteCond.py:21
python.FPGATrackSimDataPrepConfig.WriteToAOD
def WriteToAOD(flags, stage='', finalTrackParticles='')
Definition: FPGATrackSimDataPrepConfig.py:240
python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepAlgCfg
def FPGATrackSimDataPrepAlgCfg(inputFlags)
Definition: FPGATrackSimDataPrepConfig.py:374
python.CaloRecoConfig.CaloRecoCfg
def CaloRecoCfg(flags, clustersname=None)
Definition: CaloRecoConfig.py:9
python.ITkStripLorentzAngleConfig.ITkStripLorentzAngleToolCfg
def ITkStripLorentzAngleToolCfg(flags, name="ITkStripLorentzAngleTool", **kwargs)
Definition: ITkStripLorentzAngleConfig.py:10
ActsTrackFindingConfig.ActsTrackFindingCfg
ComponentAccumulator ActsTrackFindingCfg(flags, **kwargs)
Definition: ActsTrackFindingConfig.py:252
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
ActsUtilities.extractChildKwargs
dict extractChildKwargs(*str prefix, **dict kwargs)
Definition: ActsUtilities.py:3
ActsCIFlags.actsWorkflowFlags
None actsWorkflowFlags(flags)
Definition: ActsCIFlags.py:11
python.FPGATrackSimDataPrepConfig.getEtaRange
def getEtaRange(flags)
Definition: FPGATrackSimDataPrepConfig.py:99
python.FPGATrackSimDataPrepConfig.FPGAConversionAlgCfg
def FPGAConversionAlgCfg(inputFlags, name='FPGAConversionAlg', stage='', **kwargs)
Definition: FPGATrackSimDataPrepConfig.py:182
ITkPixelConditionsConfig.ITkPixelDetectorElementStatusAlgCfg
def ITkPixelDetectorElementStatusAlgCfg(flags, name="ITkPixelDetectorElementStatusAlg", **kwargs)
Definition: ITkPixelConditionsConfig.py:178
python.FPGATrackSimDataPrepConfig.FPGATrackSimReadInput2Cfg
def FPGATrackSimReadInput2Cfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:338
xAODTruthCnvConfig.GEN_AOD2xAODCfg
def GEN_AOD2xAODCfg(flags, name="GEN_AOD2xAOD", **kwargs)
Definition: xAODTruthCnvConfig.py:22
MuonGM::round
float round(const float toRound, const unsigned int decimals)
Definition: Mdt.cxx:27
ITkStripGeoModelConfig.ITkStripReadoutGeometryCfg
def ITkStripReadoutGeometryCfg(flags)
Definition: ITkStripGeoModelConfig.py:43
FPGATrackSimAlgorithmConfig.FPGATrackSimDataPrepMonitoringCfg
def FPGATrackSimDataPrepMonitoringCfg(flags)
original monitor histograms
Definition: FPGATrackSimAlgorithmConfig.py:247
ActsTruthConfig.ActsTruthParticleHitCountAlgCfg
ComponentAccumulator ActsTruthParticleHitCountAlgCfg(flags, str name='ActsTruthParticleHitCountAlg', **dict kwargs)
Definition: ActsTruthConfig.py:86
ActsTruthConfig.ActsTrackToTruthAssociationAlgCfg
ComponentAccumulator ActsTrackToTruthAssociationAlgCfg(flags, str name='ActsTracksToTruthAssociationAlg', **dict kwargs)
Definition: ActsTruthConfig.py:65
FPGAClusterSortingConfig.FPGAClusterSortingAlgCfg
def FPGAClusterSortingAlgCfg(flags, name="FPGAClusterSortingAlg", **kwargs)
Definition: FPGAClusterSortingConfig.py:6
python.FPGATrackSimMergeOutputsConfig.FPGATrackSimMergeOutputsAlgCfg
def FPGATrackSimMergeOutputsAlgCfg(flags, **kwargs)
Definition: FPGATrackSimMergeOutputsConfig.py:9
FPGATrackSimHelperFunctions.convertRegionsExpressionToArray
def convertRegionsExpressionToArray(expression, min_value=0, max_value=1279)
Definition: FPGATrackSimHelperFunctions.py:6
python.FPGATrackSimDataPrepConfig.FPGATrackSimRegionFlagCfg
def FPGATrackSimRegionFlagCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:549
python.InDetPrepRawDataToxAODConfig.TruthParticleIndexDecoratorAlgCfg
def TruthParticleIndexDecoratorAlgCfg(flags, name='TruthParticleIndexDecoratorAlg', **kwargs)
Definition: InDetPrepRawDataToxAODConfig.py:6
python.FPGATrackSimDataPrepConfig.FPGATrackSimSpacePointsToolCfg
def FPGATrackSimSpacePointsToolCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:147
python.FPGATrackSimDataPrepConfig.FPGAClusterConverterCfg
def FPGAClusterConverterCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:216
ActsTruthConfig.ActsPixelClusterToTruthAssociationAlgCfg
ComponentAccumulator ActsPixelClusterToTruthAssociationAlgCfg(flags, str name='ActsPixelClusterToTruthAssociationAlg', **dict kwargs)
Definition: ActsTruthConfig.py:20
python.ITkStripConditionsAlgorithmsConfig.ITkStripDetectorElementStatusAlgCfg
def ITkStripDetectorElementStatusAlgCfg(flags, name="ITkStripDetectorElementStatusAlg", **kwargs)
Definition: ITkStripConditionsAlgorithmsConfig.py:91
python.FPGATrackSimDataPrepConfig.getEtaSideBits
def getEtaSideBits(flags)
Definition: FPGATrackSimDataPrepConfig.py:131
python.ITkPixelLorentzAngleConfig.ITkPixelLorentzAngleToolCfg
def ITkPixelLorentzAngleToolCfg(flags, name="ITkPixelLorentzAngleTool", **kwargs)
Definition: ITkPixelLorentzAngleConfig.py:14
ActsTruthConfig.ActsTrackFindingValidationAlgCfg
ComponentAccumulator ActsTrackFindingValidationAlgCfg(flags, str name='ActsTracksValidationAlg', **dict kwargs)
Definition: ActsTruthConfig.py:162
python.FPGATrackSimDataPrepConfig.getBaseName
def getBaseName(flags)
Definition: FPGATrackSimDataPrepConfig.py:43
python.ConfigurationHelpers.OnlyTrackingPreInclude
def OnlyTrackingPreInclude(flags)
Definition: InnerDetector/InDetConfig/python/ConfigurationHelpers.py:6
python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepFlagCfg
def FPGATrackSimDataPrepFlagCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:626
python.FPGATrackSimDataPrepConfig.FPGATrackSimRawLogicCfg
def FPGATrackSimRawLogicCfg(flags, name="FPGATrackSimRawLogicTool")
Definition: FPGATrackSimDataPrepConfig.py:135
python.MainServicesConfig.MainServicesCfg
def MainServicesCfg(flags, LoopMgr='AthenaEventLoopMgr')
Definition: MainServicesConfig.py:310
python.FPGATrackSimDataPrepConfig.FPGATrackSimClusteringCfg
def FPGATrackSimClusteringCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:557
ActsGeometryConfig.ActsTrackingGeometryToolCfg
ComponentAccumulator ActsTrackingGeometryToolCfg(flags, str name="ActsTrackingGeometryTool")
Definition: ActsGeometryConfig.py:144
python.InDetPrepRawDataFormationConfig.ITkXAODToInDetClusterConversionCfg
def ITkXAODToInDetClusterConversionCfg(flags, name="ITkXAODToInDetClusterConversion", **kwargs)
Definition: InDetPrepRawDataFormationConfig.py:66
ActsTrackFindingConfig.ActsAmbiguityResolutionCfg
ComponentAccumulator ActsAmbiguityResolutionCfg(flags, **kwargs)
Definition: ActsTrackFindingConfig.py:409
print
void print(char *figname, TCanvas *c1)
Definition: TRTCalib_StrawStatusPlots.cxx:26
python.FPGATrackSimDataPrepConfig.FPGATrackSimHitFilteringToolCfg
def FPGATrackSimHitFilteringToolCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:344
python.FPGATrackSimDataPrepConfig.FixITkMainPassFlags
def FixITkMainPassFlags(flags)
Definition: FPGATrackSimDataPrepConfig.py:637
TCS::join
std::string join(const std::vector< std::string > &v, const char c=',')
Definition: Trigger/TrigT1/L1Topo/L1TopoCommon/Root/StringUtils.cxx:10
python.InDetPrepRawDataToxAODConfig.ITkActsPrepDataToxAODCfg
ComponentAccumulator ITkActsPrepDataToxAODCfg(flags, *str PixelClusterContainer="ITkPixelClusters", str StripClusterContainer="ITkStripClusters", str PixelMeasurementContainer="ITkPixelMeasurements", str StripMeasurementContainer="ITkStripMeasurements", list[str] TrackParticles=None)
Definition: InDetPrepRawDataToxAODConfig.py:46
python.JetRecoSteering.addTruthPileupJetsToOutputCfg
def addTruthPileupJetsToOutputCfg(flags, toAOD=True, toESD=True)
Definition: JetRecoSteering.py:7
python.FPGATrackSimDataPrepConfig.getRegionNumber
def getRegionNumber(nPiOver16Min, minAbsEta, isPositiveEta, verbosePrint=True)
Definition: FPGATrackSimDataPrepConfig.py:12
python.FPGATrackSimDataPrepConfig.FPGATrackSimReadInputCfg
def FPGATrackSimReadInputCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:331
python.FPGATrackSimDataPrepConfig.FPGATrackSimEventSelectionSvcCfg
def FPGATrackSimEventSelectionSvcCfg(flags, name="FPGATrackSimEventSelectionSvc")
Definition: FPGATrackSimDataPrepConfig.py:274
python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepConnectToFastTracking
def FPGATrackSimDataPrepConnectToFastTracking(flagsIn, FinalTracks="F100-", **kwargs)
Definition: FPGATrackSimDataPrepConfig.py:442
ActsSpacePointFormationConfig.ActsPixelSpacePointFormationAlgCfg
ComponentAccumulator ActsPixelSpacePointFormationAlgCfg(flags, str name="ActsPixelSpacePointFormationAlg", *bool useCache=False, **dict kwargs)
Definition: ActsSpacePointFormationConfig.py:144
FPGATrackSimSGInputConfig.FPGATrackSimSGInputToolCfg
def FPGATrackSimSGInputToolCfg(flags, **kwargs)
Definition: FPGATrackSimSGInputConfig.py:8
python.TrackRecoConfig.InDetTrackRecoCfg
def InDetTrackRecoCfg(flags)
Main ID tracking config #####################.
Definition: TrackRecoConfig.py:804
Trk::open
@ open
Definition: BinningType.h:40
python.AllConfigFlags.initConfigFlags
def initConfigFlags()
Definition: AllConfigFlags.py:19
python.FPGATrackSimDataPrepConfig.FPGATrackSimMappingCfg
def FPGATrackSimMappingCfg(flags, name="FPGATrackSimMappingSvc")
Definition: FPGATrackSimDataPrepConfig.py:304
ActsSeedingConfig.ActsSeedingCfg
ComponentAccumulator ActsSeedingCfg(flags, **kwargs)
Definition: ActsSeedingConfig.py:455
python.OutputStreamConfig.addToAOD
def addToAOD(flags, itemOrList, **kwargs)
Definition: OutputStreamConfig.py:202
if
if(febId1==febId2)
Definition: LArRodBlockPhysicsV0.cxx:567
ActsCIFlags.actsLegacyWorkflowFlags
None actsLegacyWorkflowFlags(flags)
Definition: ActsCIFlags.py:6
python.FPGATrackSimDataPrepConfig.runDataPrepChain
def runDataPrepChain()
Definition: FPGATrackSimDataPrepConfig.py:681
str
Definition: BTagTrackIpAccessor.cxx:11
DataPrepConfig.FPGATrackSimReportingCfg
def FPGATrackSimReportingCfg(flags, name='FPGATrackSimReportingAlg', **kwargs)
Definition: DataPrepConfig.py:6
python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepOutputCfg
def FPGATrackSimDataPrepOutputCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:163
python.FPGATrackSimDataPrepConfig.FPGAActsTrkConverterCfg
def FPGAActsTrkConverterCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:232
python.xAODEventInfoCnvConfig.EventInfoCnvAlgCfg
def EventInfoCnvAlgCfg(flags, name="EventInfoCnvAlg", inputKey="McEventInfo", outputKey="EventInfo", disableBeamSpot=False, **kwargs)
Definition: xAODEventInfoCnvConfig.py:11
python.FPGATrackSimDataPrepConfig.getPhiRange
def getPhiRange(flags)
Definition: FPGATrackSimDataPrepConfig.py:70
InfileMetaDataConfig.SetupMetaDataForStreamCfg
def SetupMetaDataForStreamCfg(flags, streamName="", AcceptAlgs=None, createMetadata=None, propagateMetadataFromInput=True, *args, **kwargs)
Definition: InfileMetaDataConfig.py:222
python.PoolReadConfig.PoolReadCfg
def PoolReadCfg(flags)
Definition: PoolReadConfig.py:71
python.FPGATrackSimDataPrepConfig.prepareFlagsForFPGATrackSimDataPrepAlg
def prepareFlagsForFPGATrackSimDataPrepAlg(flags)
Definition: FPGATrackSimDataPrepConfig.py:158
python.FPGATrackSimDataPrepConfig.FPGATrackSimEventSelectionToolCfg
def FPGATrackSimEventSelectionToolCfg(flags, name="FPGATrackSimEventSelectionTool")
Definition: FPGATrackSimDataPrepConfig.py:296
FPGAOutputValidationConfig.FPGAOutputValidationCfg
def FPGAOutputValidationCfg(flags, **kwargs)
Definition: FPGAOutputValidationConfig.py:3
ActsTruthConfig.ActsTrackParticleTruthDecorationAlgCfg
ComponentAccumulator ActsTrackParticleTruthDecorationAlgCfg(flags, str name='ActsTrackParticleTruthDecorationAlg', **dict kwargs)
Definition: ActsTruthConfig.py:137
ActsTruthConfig.ActsStripClusterToTruthAssociationAlgCfg
ComponentAccumulator ActsStripClusterToTruthAssociationAlgCfg(flags, str name='ActsStripClusterToTruthAssociationAlg', **dict kwargs)
Definition: ActsTruthConfig.py:35
python.FPGATrackSimDataPrepConfig.FPGATrackSimDataPrepSetup
def FPGATrackSimDataPrepSetup(flags, runReco=True)
Definition: FPGATrackSimDataPrepConfig.py:646
python.FPGATrackSimDataPrepConfig.nameWithRegionSuffix
def nameWithRegionSuffix(flags, basename)
Definition: FPGATrackSimDataPrepConfig.py:8
ActsTrackFindingConfig.ActsTrackToTrackParticleCnvAlgCfg
ComponentAccumulator ActsTrackToTrackParticleCnvAlgCfg(flags, str name="ActsTrackToTrackParticleCnvAlg", **kwargs)
Definition: ActsTrackFindingConfig.py:461
python.FPGATrackSimDataPrepConfig.FPGATrackSimLorentzAngleToolCfg
def FPGATrackSimLorentzAngleToolCfg(flags)
Definition: FPGATrackSimDataPrepConfig.py:359