extmuid_createMuons.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
<header>
    <output>muon-ExtMuidValidation.root</output>
    <contact>piilonen@vt.edu</contact>
    <description>Create events with 1 muon track for ext/muid validation.</description>
</header>
"""
 
import glob
from basf2 import *
import os
from simulation import add_simulation
from reconstruction import add_reconstruction
 
set_random_seed(123460)
 
output_filename = '../muon-ExtMuidValidation.root'
 
print(output_filename)
 
path = create_path()
 
eventinfosetter = register_module('EventInfoSetter')
eventinfosetter.param('evtNumList', [1000])
path.add_module(eventinfosetter)
 
progress = register_module('Progress')
path.add_module(progress)
 
pgun = register_module('ParticleGun')
param_pgun = {
    'pdgCodes': [-13, 13],
    'nTracks': 1,
    'varyNTracks': 0,
    'momentumGeneration': 'uniform',
    'momentumParams': [0.5, 5.0],
    'thetaGeneration': 'uniformCos',
    'thetaParams': [15., 150.],
    'phiGeneration': 'uniform',
    'phiParams': [0.0, 360.0],
    'vertexGeneration': 'fixed',
    'xVertexParams': [0.0],
    'yVertexParams': [0.0],
    'zVertexParams': [0.0],
}
pgun.param(param_pgun)
path.add_module(pgun)
 
# add simulation and reconstruction modules to the path
if 'BELLE2_BACKGROUND_DIR' in os.environ:
    background_files = glob.glob(os.environ['BELLE2_BACKGROUND_DIR'] + '/*.root')
    print('Background files:')
    print(background_files)
    add_simulation(path, bkgfiles=background_files)
else:
    print('Warning: variable BELLE2_BACKGROUND_DIR is not set')
    add_simulation(path)
 
add_reconstruction(path)
 
output = register_module('RootOutput')
output.param('outputFileName', output_filename)
output.param('branchNames', ['MCParticles', 'ExtHits', 'KLMMuidLikelihoods', 'BKLMHit2ds', 'EKLMHit2ds'])
path.add_module(output)
 
process(path)
print(statistics)