cosmicsExtrapolationReconstruction.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
"""
<header>
  <contact>software-tracking@belle2.org</contact>
  <input>CosmicsSimNoBkg.root</input>
  <output>CosmicsExtrapolation.root</output>
  <description>Validation of cosmic track extrapolation.</description>
</header>
"""
 
import basf2
from reconstruction import add_cosmics_reconstruction
 
basf2.set_random_seed(12345)
 
main = basf2.create_path()
 
# Input.
main.add_module('RootInput', inputFileName='../CosmicsSimNoBkg.root')
 
# Extrapolation. Tracks are not merged (otherwise, many backward-extrapolated
# tracks are removed).
main.add_module('Gearbox')
main.add_module('Geometry')
add_cosmics_reconstruction(main, merge_tracks=False)
 
# Output.
output = basf2.register_module('RootOutput')
output.param('outputFileName', '../CosmicsExtrapolation.root')
output.param('branchNames', ['ExtHits', 'BKLMHit2ds', 'EKLMHit2ds'])
main.add_module(output)
 
# Process the path.
basf2.process(main, max_event=1000)