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
0018 s = acts.examples.Sequencer(events=1, numThreads=1, outputDir="output")
0019
0020 # loggingLevel = acts.logging.INFO
0021 loggingLevel = acts.logging.DEBUG
0022
0023 s.addReader(
0024     acts.examples.RootAthenaDumpReader(
0025         level=loggingLevel,
0026         treename="GNN4ITk",
0027         inputfile="Dump_GNN4Itk.root",
0028         onlySpacepoints=True,
0029         outputPixelSpacePoints="pixel_spacepoints",
0030         outputStripSpacePoints="strip_spacepoints",
0031         outputSpacePoints="spacepoints",
0032     )
0033 )
0034
0035
0036 # run pixel seeding
0037 seeding_pixel = addStandardSeeding(
0038     s,
0039     "pixel_spacepoints",
0040     *acts.examples.itk.itkSeedingAlgConfig(
0041         InputSpacePointsType.PixelSpacePoints, highOccupancyConfig=True
0042     ),
0043     logLevel=loggingLevel,
0044     outputSeeds="pixel_seeds"
0045 )
0046
0047
0048 # run strip seeding
0049 seeding_strip = addStandardSeeding(
0050     s,
0051     "strip_spacepoints",
0052     *acts.examples.itk.itkSeedingAlgConfig(InputSpacePointsType.StripSpacePoints),
0053     logLevel=acts.logging.DEBUG,
0054     outputSeeds="strip_seeds"
0055 )
0056
0057
0058 s.run()