0_dedxPID_ParticleGun.py

#!/usr/bin/env python3
 

 
"""
<header>
  <input>PartGunChargedStableGenSim.root</input>
  <output>ParticleGunSimRec_dedx.root</output>
  <description>Generates dE/dx debug data (DedxTracks) for testing</description>
  <contact>renu2@andrew.cmu.edu</contact>
</header>
"""
 
import basf2
from reconstruction import add_reconstruction
 
basf2.set_random_seed(1337)
 
main = basf2.create_path()
 
# Read input.
inputFileName = "../PartGunChargedStableGenSim.root"
main.add_module("RootInput", inputFileName=inputFileName)
 
# Load parameters.
main.add_module("Gearbox")
# Create geometry.
main.add_module("Geometry")
 
# Reconstruct events.
add_reconstruction(main)
 
output = basf2.register_module('RootOutput')
output.param('outputFileName', '../ParticleGunSimRec_dedx.root')
# let's keep this small
output.param('branchNames', ['CDCDedxLikelihoods', 'CDCDedxTracks', 'VXDDedxLikelihoods', 'VXDDedxTracks', 'EventMetaData'])
main.add_module(output)
 
main.add_module("ProgressBar")
 
basf2.process(main)