Scripts/Python/reader_and_seeding.py

0001 import sys
0002 from pathlib import Path
0003
0004 import acts.examples
0005 import acts
0006
0007 from acts.examples import (
0008     CsvSpacePointReader,
0009     TrackParamsEstimationAlgorithm,
0010 )
0011 from acts.examples.reconstruction import (
0012     addStandardSeeding,
0013 )
0014
0015 from acts.examples.itk import itkSeedingAlgConfig, InputSpacePointsType
0016
0017 s = acts.examples.Sequencer(events=1, numThreads=1, outputDir="output")
0018
0019 # loggingLevel = acts.logging.INFO
0020 loggingLevel = acts.logging.DEBUG
0021
0022 s.addReader(
0023     acts.examples.root.RootAthenaDumpReader(
0024         level=loggingLevel,
0025         treename="GNN4ITk",
0026         inputfile="Dump_GNN4Itk.root",
0027         onlySpacePoints=True,
0028         outputPixelSpacePoints="pixel_spacepoints",
0029         outputStripSpacePoints="strip_spacepoints",
0030         outputSpacePoints="spacepoints",
0031     )
0032 )
0033
0034
0035 # run pixel seeding
0036 seeding_pixel = addStandardSeeding(
0037     s,
0038     "pixel_spacepoints",
0039     *acts.examples.itk.itkSeedingAlgConfig(
0040         InputSpacePointsType.PixelSpacePoints, highOccupancyConfig=True
0041     ),
0042     logLevel=loggingLevel,
0043     outputSeeds="pixel_seeds",
0044 )
0045
0046
0047 # run strip seeding
0048 seeding_strip = addStandardSeeding(
0049     s,
0050     "strip_spacepoints",
0051     *acts.examples.itk.itkSeedingAlgConfig(InputSpacePointsType.StripSpacePoints),
0052     logLevel=acts.logging.DEBUG,
0053     outputSeeds="strip_seeds",
0054 )
0055
0056
0057 s.run()