Scripts/Python/truth_jet_test.py

0001 #!/usr/bin/env python3
0002 import acts
0003 import acts.examples
0004 from acts.examples.simulation import addPythia8, addTruthJetAlg, TruthJetConfig
0005
0006
0007 outputDir = "./truth_jet_test_output"
0008 u = acts.UnitConstants
0009 rnd = acts.examples.RandomNumbers(seed=42)
0010
0011 s = acts.examples.Sequencer(events=10, outputDir=outputDir)
0012
0013 addPythia8(
0014     s,
0015     hardProcess=["Top:qqbar2ttbar=on"],
0016     npileup=50,
0017     vtxGen=acts.examples.GaussianVertexGenerator(
0018         mean=acts.Vector4(0, 0, 0, 0),
0019         stddev=acts.Vector4(0.0125 * u.mm, 0.0125 * u.mm, 55.5 * u.mm, 5.0 * u.ns),
0020     ),
0021     rnd=rnd,
0022     outputDirRoot=outputDir,
0023     outputDirCsv=outputDir,
0024 )
0025
0026 addTruthJetAlg(
0027     s,
0028     TruthJetConfig(
0029         inputTruthParticles="particles_generated",
0030         outputJets="output_jets",
0031         jetPtMin=20 * u.GeV,
0032     ),
0033     loglevel=acts.logging.INFO,
0034 )
0035
0036 s.run()