testEntireBKLMChain.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
import basf2
import simulation as sim
 
bklm_dataObjects = ['BKLMDigits',
                    'BKLMDigitOutOfRanges',
                    'KLMDigitEventInfos',
                    'BKLMHit1ds',
                    'BKLMHit2ds',
                    'BKLMTracks']
use_KKMC = False
 
# Create the main path
main = basf2.create_path()
 
# Set EventInfoSetter and add a progress bar
main.add_module('EventInfoSetter',
                expList=0,
                runList=0,
                evtNumList=1000)
main.add_module('Progress')
main.add_module('ProgressBar')
 
if use_KKMC:  # Use KKMC to generate generic mu+mu- events
    main.add_module('KKGenInput',
                    tauinputFile=Belle2.FileSystem.findFile('data/generators/kkmc/mu.input.dat'),
                    KKdefaultFile=Belle2.FileSystem.findFile('data/generators/kkmc/KK2f_defaults.dat'),
                    taudecaytableFile='',
                    kkmcoutputfilename='output.kkmc.dat')
else:  # Use ParticleGun to generate 4GeV mu+ and mu-
    main.add_module('ParticleGun',
                    nTracks=1,
                    pdgCodes=[13, -13],
                    momentumGeneration='fixed',
                    momentumParams=[4],
                    thetaGeneration='uniform',
                    thetaParams=[37, 130])
 
# Add simulation
main.add_module('Gearbox')
main.add_module('Geometry',
                components=['KLM'],
                useDB=False)
sim.add_simulation(path=main,
                   components=['KLM'])
 
# Pack and unpack data
main.add_module('KLMPacker')
main.add_module('KLMUnpacker',
                WriteDigitRaws=True)
 
# Add the digit analyzer module (optional)
main.add_module('BKLMDigitAnalyzer',
                outputRootName='bklmHitmap')
 
# Add the reconstruction
main.add_module('KLMReconstructor')
 
# Add the self-tracking
main.add_module('BKLMTracking',
                StudyEffiMode=True,
                outputName='bklmEfficiency.root')
 
# Save the dataobjects in a .root output
main.add_module('RootOutput',
                outputFileName='bklm_dataObjects.root',
                branchNames=bklm_dataObjects,
                branchNamesPersistent='FileMetaData')
 
# Process the path
basf2.process(main)
print(basf2.statistics)