Scripts/Python/pythia8.py

0001 #!/usr/bin/env python3
0002 from pathlib import Path
0003
0004 import acts
0005 import acts.examples
0006 from acts.examples.simulation import addPythia8
0007
0008 u = acts.UnitConstants
0009
0010
0011 def runPythia8(
0012     outputDir,
0013     outputRoot: bool = True,
0014     outputCsv: bool = True,
0015     s: acts.examples.Sequencer = None,
0016     vtxGen=None,
0017 ):
0018     # Preliminaries
0019     rnd = acts.examples.RandomNumbers()
0020     outputDir = Path(outputDir)
0021
0022     # Sequencer
0023     s = s or acts.examples.Sequencer(
0024         events=10, numThreads=-1, logLevel=acts.logging.INFO
0025     )
0026
0027     addPythia8(
0028         s,
0029         rnd=rnd,
0030         outputDirCsv=outputDir / "csv" if outputCsv else None,
0031         outputDirRoot=outputDir if outputRoot else None,
0032         vtxGen=vtxGen,
0033     )
0034
0035     return s
0036
0037
0038 if "__main__" == __name__:
0039     runPythia8(Path.cwd()).run()