a01234_trackingEfficiency_oneshot.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
 
 
import sys
import glob
import basf2 as b2
from generators import add_evtgen_generator
from simulation import add_simulation
from tracking import add_tracking_reconstruction
import modularAnalysis as ma
 
b2.set_random_seed(1509)
 
particleGun = False
 
release = sys.argv[1]
 
# roi = {noROI, vxdtf1, vxdtf2}
# bkg = {noBkg, stdBKG, std2GBKG}
# vxdtf = {vxdtf1, vxdtf2}
 
roi = sys.argv[2]
bkg = sys.argv[3]
vxdtf = sys.argv[4]
 
release = 'merged'
 
print('Performance evaluated on: ')
print('simulation: ' + roi + ' ' + bkg)
print('reconstruction: ' + vxdtf)
print()
 
root_file_name_TRK = './' + release + '/oneshotTV_TRK_analysis_' + roi + '_' + bkg + '_' + vxdtf + '_' + release + '.root'
root_file_name_V0 = './' + release + '/oneshotTV_V0_analysis_' + roi + '_' + bkg + '_' + vxdtf + '_' + release + '.root'
 
bkgFiles = None
usePXDDataReduction = True
mcTrackFinding = False
 
if roi == 'noROI':
    usePXDDataReduction = False
 
if bkg == 'stdBKG':
    bkgFiles = glob.glob('/sw/belle2/bkg/*.root')
if bkg == 'std2GsmallBKG':
    bkgFiles = glob.glob('/sw/belle2/bkg/*.root')
    bkgFiles = bkgFiles + glob.glob('/sw/belle2/bkg.twoPhoton/*usual.root')
if bkg == 'std2GBKG':
    bkgFiles = glob.glob('/sw/belle2/bkg/*.root')
    bkgFiles = bkgFiles + glob.glob('/sw/belle2/bkg.twoPhoton/*.root')
 
print(bkgFiles)
 
path = b2.create_path()
 
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param('expList', [0])
eventinfosetter.param('runList', [1])
 
progress = b2.register_module('Progress')
 
v0matcher = b2.register_module('MCV0Matcher')
v0matcher.param('V0ColName', 'V0ValidationVertexs')
v0matcher.logging.log_level = b2.LogLevel.INFO
 
create_plots_TRK = b2.register_module('TrackingPerformanceEvaluation')
create_plots_TRK.param('outputFileName', root_file_name_TRK)
create_plots_TRK.logging.log_level = b2.LogLevel.INFO
 
 
create_plots_V0 = b2.register_module('V0findingPerformanceEvaluation')
create_plots_V0.param('outputFileName', root_file_name_V0)
create_plots_V0.logging.log_level = b2.LogLevel.INFO
 
if particleGun:
    particleGunModule = b2.register_module('ParticleGun')
    particleGunModule.param({
        'pdgCodes': [211, -211],
        'nTracks': 1,
        'varyNTracks': False,
        'momentumGeneration': 'uniformpt',
        'momentumParams': [0.5, 1.],
        'thetaGeneration': 'uniform',
        'thetaParams': [60., 60.],
    })
    path.add_module('EventInfoSetter')
    path.add_module(particleGunModule)
else:
    ma.setupEventInfo(100, path)
    add_evtgen_generator(path, 'signal', None)
 
path.add_module(progress)
 
add_simulation(path, None, bkgFiles, None, 1.0, usePXDDataReduction)
 
add_tracking_reconstruction(
    path,
    components=None,
    pruneTracks=False,
    mcTrackFinding=mcTrackFinding,
    skipGeometryAdding=False
)
 
 
modList = path.modules()
for modItem in modList:
    if modItem.name() == 'V0Finder':
        modItem.param('Validation', True)
path.add_module(v0matcher)
 
path.add_module(create_plots_TRK)
path.add_module(create_plots_V0)
 
b2.process(path)
 
print(b2.statistics)