phase2_sim_reco_tracking.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
import os
from basf2 import *
from tracking import *
from svd import *
from rawdata import *
from ROOT import Belle2
from reconstruction import add_reconstruction
import os.path
import sys
 
print('***')
print('*** Used steering script:')
with open(sys.argv[0], 'r') as fin:
    print(fin.read(), end="")
print('*** end of the script.')
print('***')
 
fileIN = sys.argv[1]
fileOUT = sys.argv[2]
 
main = create_path()
main.add_module('RootInput', inputFileName=fileIN)
 
gearbox = register_module('Gearbox')
geomfile = '/geometry/Beast2_phase2.xml'
if geomfile != 'None':
    gearbox.param('fileName', geomfile)
 
main.add_module(gearbox)
geometry = register_module('Geometry')
geometry.param('useDB', False)
geometry.param('components', ['SVD'])
main.add_module(geometry)
 
# SVD reconstruction
main.add_module('SVDCoGTimeEstimator')
main.add_module('SVDSimpleClusterizer', Clusters="SVDClusters")
 
# add tracking recontruction
add_tracking_reconstruction(main, components=['SVD'])
# remove cut on CoG in SpacePointCreator
for module in main.modules():
    if module.name() == 'SVDSpacePointCreator':
        module.param("MinClusterTime", -999)
 
main.add_module('VXDDedxPID', enableDebugOutput=True)
 
main.add_module(
    'RootOutput',
    outputFileName=fileOUT,
    branchNames=[
        "SVDShaperDigits",
        "SVDClusters",
        "SVDRecoDigits",
        "Tracks",
        "RecoTracks",
        "TracksToRecoTracks",
        "TrackFitResults",
        "MCParticles",
        "SVDTrueHits",
        "TracksToVXDDedxLikelihoods",
        "TracksToVXDDedxTracks",
        "VXDDedxLikelihoods",
        "VXDDedxTracks"])
print_path(main)
main.add_module('ProgressBar')
# Process events
process(main)
print(statistics)