runUVTimeDifference.py

#!/usr/bin/env python3
 
 
 
 
 
import glob
import sys
 
import basf2 as b2
from basf2 import conditions as b2conditions
from rawdata import add_unpackers
from simulation import add_simulation
from svd import add_svd_reconstruction
 
MCrun = int(sys.argv[1])
data = False  # True
unpack = True
if not data:
    unpack = False
 
# only relevant for data:
run = 5900  # 3239 #5751
fileout = 'hitTimeSelection_exp10run'+str(run)+'.root'
 
# RAW
files = ['/group/belle2/dataprod/Data/Raw/e0010/r0'+str(run)+'/sub00/physics.0010.0'+str(run)+'.HLT*.root']
# 5751
# 5900
# RAW CT skims
# files = ['/gpfs/fs02/belle2/group/detector/SVD/']
# run = 3239
 
# old-format cDST
# files=["/group/belle2/dataprod/Data/release-04-00-02/DB00000523/Unofficial/e0010/4S/r04295/skim/hlt_hadron/cdst/sub00/cdst.physics.0010.04295.HLT1*.root","/group/belle2/dataprod/Data/release-04-00-02/DB00000523/Unofficial/e0010/4S/r04295/skim/hlt_bhabha/cdst/sub00/cdst.physics.0010.04295.HLT1.*.root","/group/belle2/dataprod/Data/release-04-00-02/DB00000523/Unofficial/e0010/4S/r04295/skim/hlt_mumu_2trk/cdst/sub00/cdst.physics.0010.04295.HLT1.*.root"]
 
bkg = glob.glob('/group/belle2/BGFile/OfficialBKG/early_phase3/prerelease-04-00-00a/overlay/phase31/BGx1/set0/*.root')
 
# needed for some temporary issues with BKLMDisplacement payload
if data:
    b2conditions.override_globaltags()
    b2conditions.globaltags = ['svd_svdHitTimeSelection_test', 'klm_alignment_testing', 'online']
else:
    b2conditions.globaltags = ['svd_svdHitTimeSelection_test']
 
# svd_svd_svdHitTimeSelection_test CONTENT:
# exp 0, run 0 maxTimeDiff = 100
# exp 0, run 1 maxTimeDiff = 10
# exp 0, run 2 maxTimeDiff = 15
# exp 0, run 3 maxTimeDiff = 20
# exp 0, run 4 maxTimeDiff = 25
# exp 0, run 5 maxTimeDiff = 30
# exp 0, run 6 maxTimeDiff = 35
# exp 0, run 7 maxTimeDiff = 40
# exp 0, run 8 maxTimeDiff = 45
# exp 0, run 9 maxTimeDiff = 50
 
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param('expList', [0])
eventinfosetter.param('runList', [MCrun])
 
evtgeninput = b2.register_module('EvtGenInput')
evtgeninput.logging.log_level = b2.LogLevel.INFO
 
# main main
main = b2.create_path()
 
# read input rootfile
# -> can be overwritten with the -i option
if data:
    main.add_module("RootInput", inputFileNames=files)
else:
    main.add_module(eventinfosetter)
    main.add_module(evtgeninput)
 
# nee to know SVD geometry to create histograms
main.add_module('Gearbox')
main.add_module('Geometry')
 
# if using RAW data you need to unpack them
if data and unpack:
    add_unpackers(main, components=['SVD'])
 
if not data:
    main.add_module('FullSim')
    add_simulation(main, bkgfiles=None, usePXDDataReduction=False, forceSetPXDDataReduction=True)
#    add_simulation(main, bkgfiles=bkg, usePXDDataReduction=False, forceSetPXDDataReduction=True)
 
if not data:
    fileout = 'SVDHitTimeSelection_MCnoBKG_run'+str(MCrun)+'.root'
 
add_svd_reconstruction(main)
 
main.add_module('RootOutput', outputFileName=fileout, branchNames=['SVDSpacePoints'])
# Show progress
main.add_module('Progress')
 
b2.print_path(main)
 
# Process events
b2.process(main)
 
print(b2.statistics)