testVXDTF2.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
 
from basf2 import *
import argparse
# Import custom module chain for VXDTF2
from setup_modules import setup_VXDTF2
 
 
# ---------------------------------------------------------------------------------------
# Argument parser for input of trained Sector Map.
arg_parser = argparse.ArgumentParser(description='VXDTF2 application:\
                                     Applies trained SecMap on provided data and returns RecoTrackCands.\n\
                                     Usage: basf2 testVXDTF2.py -i <inputFileName> -o <outputFileName> -- --secmap <secmapFile>')
 
arg_parser.add_argument('--secmap', '-s', type=str,
                        help='Inclusion of the root file containing the trained SecMap for the application of the VXDTF2.')
 
arguments = arg_parser.parse_args(sys.argv[1:])
secmap_name = arguments.secmap
 
 
# ---------------------------------------------------------------------------------------
# Settings
usePXD = False
useDisplay = False
 
performFit = False
generateTimeSeedAfterFit = False
 
# Logging and Debug Levels
set_log_level(LogLevel.ERROR)
log_to_file('logVXDTF2Execution.log', append=False)
 
 
# ---------------------------------------------------------------------------------------
path = create_path()
 
# Input
rootInput = register_module('RootInput')
path.add_module(rootInput)
 
# Event Info Module
eventinfoprinter = register_module('EventInfoPrinter')
path.add_module(eventinfoprinter)
 
# Gearbox
gearbox = register_module('Gearbox')
path.add_module(gearbox)
 
# Geometry
geometry = register_module('Geometry')
geometry.param('components', ['BeamPipe',
                              'MagneticFieldConstant4LimitedRSVD',
                              'PXD',
                              'SVD'])
path.add_module(geometry)
 
# Event counter
eventCounter = register_module('EventCounter')
path.add_module(eventCounter)
 
 
# VXDTF2: Including actual VXDTF2 Modul Chain
setup_VXDTF2(path=path,
             use_pxd=usePXD,
             secmap_name=secmap_name,
             overlap_filter='hopfield',
             quality_estimator='circleFit')
 
 
if performFit:
    # This is required for RecoFitter
    genFitExtrapolation = register_module('SetupGenfitExtrapolation')
    path.add_module(genFitExtrapolation)
 
    if not generateTimeSeedAfterFit:
        timeSeed = register_module('IPTrackTimeEstimator')
        timeSeed.param('useFittedInformation', False)
        path.add_module(timeSeed)
 
    fitter = register_module('DAFRecoFitter')
    path.add_module(fitter)
 
    if generateTimeSeedAfterFit:
        timeSeedAfterFit = register_module('IPTrackTimeEstimator')
        timeSeedAfterFit.param('useFittedInformation', True)
        path.add_module(timeSeedAfterFit)
 
 
output = register_module('RootOutput')
path.add_module(output)
 
 
if useDisplay:
    display = register_module('Display')
    display.param('showAllPrimaries', True)
    path.add_module(display)
 
 
process(path)
print(statistics)