diff --git a/Examples/Scripts/Python/full_chain_test.py b/Examples/Scripts/Python/full_chain_test.py index 595dbd43a4e..1c839f31b3a 100755 --- a/Examples/Scripts/Python/full_chain_test.py +++ b/Examples/Scripts/Python/full_chain_test.py @@ -81,7 +81,12 @@ def parse_args(): "--output-csv", action="count", default=0, - help="Use CSV output instead of ROOT. Specify -cc to output both.", + help="Use CSV output instead of ROOT. Specify -cc to output all formats (ROOT, CSV, and obj).", + ) + parser.add_argument( + "--output-obj", + action="store_true", + help="Enable obj output", ) parser.add_argument( "-n", @@ -272,6 +277,7 @@ def full_chain(args): outputDirCsv = outputDir if args.output_csv != 0 else None outputDirLessCsv = outputDirLess if args.output_csv != 0 else None outputDirMoreCsv = outputDirMore if args.output_csv != 0 else None + outputDirObj = outputDirLess if args.output_obj else outputDir if args.output_csv == 2 else None # fmt: off if args.generic_detector: @@ -388,8 +394,7 @@ def full_chain(args): "LongStripEndcapReadout", ], outputParticlesGenerator="particles_input", - outputParticlesInitial="particles_initial", - outputParticlesFinal="particles_final", + outputParticlesSimulation="particles_simulated", outputSimHits="simhits", graphvizOutput="graphviz", dd4hepDetector=detector, @@ -479,6 +484,7 @@ def full_chain(args): postSelectParticles=postSelectParticles, outputDirRoot=outputDirRoot, outputDirCsv=outputDirCsv, + outputDirObj=outputDirObj, ) else: if s.config.numThreads != 1: @@ -504,6 +510,7 @@ def full_chain(args): killAfterTime=25 * u.ns, outputDirRoot=outputDirRoot, outputDirCsv=outputDirCsv, + outputDirObj=outputDirObj, ) addDigitization(