a01_trackingEfficiency_createData.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
 
 
import sys
import basf2 as b2
from beamparameters import add_beamparameters
 
b2.set_random_seed(1509)
 
release = sys.argv[1]
 
print(release)
 
output_filename = './' + release + '/TV_data_' + release + '.root'
 
print(output_filename)
 
path = b2.create_path()
 
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param('expList', [0])
eventinfosetter.param('runList', [1])
eventinfosetter.param('evtNumList', [10000])
 
progress = b2.register_module('Progress')
 
evtgeninput = b2.register_module('EvtGenInput')
evtgeninput.logging.log_level = b2.LogLevel.WARNING
 
beamparameters = add_beamparameters(path, "Y4S")
 
root_output = b2.register_module('RootOutput')
root_output.param('outputFileName', output_filename)
 
path.add_module(eventinfosetter)
path.add_module(progress)
path.add_module(beamparameters)
path.add_module(evtgeninput)
path.add_module(root_output)
 
b2.process(path)
print(b2.statistics)