Scripts/Python/particle_gun.py

0001 #!/usr/bin/env python3
0002 from pathlib import Path
0003
0004 import acts
0005 from acts.examples import Sequencer
0006 from acts.examples.simulation import addParticleGun, EtaConfig, ParticleConfig
0007
0008
0009 def runParticleGun(outputDir, s=None):
0010     s = s or Sequencer(events=10, numThreads=-1)
0011     s.config.logLevel = acts.logging.INFO
0012     outputDir = Path(outputDir)
0013     addParticleGun(
0014         s,
0015         EtaConfig(-4.0, 4.0),
0016         ParticleConfig(2),
0017         outputDirCsv=outputDir / "csv",
0018         outputDirRoot=outputDir,
0019         printParticles=True,
0020     )
0021     return s
0022
0023
0024 if "__main__" == __name__:
0025     runParticleGun(Path.cwd()).run()