SVDCoGTimeCalibrationImporterTBIndependent.py

# !/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
 
 
 
import basf2 as b2
import svd
import os
import sys
from svd.CoGCalibration_utils_tbindependent import SVDCoGTimeCalibrationImporterModule
from basf2 import conditions as b2conditions
 
localdb = sys.argv[1]
# filename = sys.argv[2]
# branches = ['SVDShaperDigits', 'SVDRecoDigits', 'SVDClusters', 'EventT0', 'Tracks', 'RecoTracks']
branches = ['SVDShaperDigits', 'SVDShaperDigitsFromTracks', 'EventT0']
 
trk_outputFile = "TrackFilterControlNtuples_" + localdb + ".root"
nSVD = 6
nCDC = 1
pVal = 0.0  # 0001
 
 
inputFileList = [
    "/group/belle2/dataprod/Data/release-03-02-02/DB00000635/proc00000009/\
e0008/4S/r01309/skim/hlt_bhabha/cdst/sub00/cdst.physics.0008.01309.HLT*"
]
'''
if filename.rfind(".root") != -1:
    inputFileList.append(filename)
else:
    with open(filename, 'r') as f:
        inputFileList = [line.strip() for line in f]
'''
 
b2conditions.prepend_globaltag("online")
b2conditions.prepend_globaltag("data_reprocessing_prompt")
if '_CHECK' not in localdb:
    b2conditions.prepend_globaltag('svd_NOCoGCorrections')
    if os.path.isdir(localdb):
        print('WARNING, you are using CoG corrections in ' + localdb + ' exiting')
        sys.exit(1)
    else:
        b2conditions.testing_payloads = [str(localdb)]
else:
    localdb = localdb.split('_')
    if not os.path.isdir(localdb[0]):
        print('WARNING, the localDB ' + localdb[0] + ' containing the corrections is missing, exiting.')
        sys.exit(1)
    else:
        b2conditions.testing_payloads = [str(localdb)]
    localdb = sys.argv[1]
    if os.path.isdir(localdb):
        print('WARNING, you are using CoG corrections in ' + localdb + ' exiting')
        sys.exit(1)
    else:
        b2conditions.testing_payloads = [str(localdb)]
 
 
main = b2.create_path()
 
rootinput = b2.register_module('RootInput')
rootinput.param('inputFileNames', inputFileList)
rootinput.param('branchNames', branches)
main.add_module(rootinput)
 
main.add_module("Gearbox")
main.add_module("Geometry", useDB=True)
 
# Track selection - NOT YET
trkFlt = b2.register_module('TrackFilter')
trkFlt.param('outputFileName', trk_outputFile)
trkFlt.param('outputINArrayName', 'SelectedTracks')
trkFlt.param('outputOUTArrayName', 'ExcludedTracks')
trkFlt.param('min_NumHitSVD', nSVD)
trkFlt.param('min_NumHitCDC', nCDC)
trkFlt.param('min_Pvalue', pVal)
# trkFlt.logging.log_level = LogLevel.DEBUG
# main.add_module(trkFlt)
 
# fil = register_module('SVDShaperDigitsFromTracks')
# fil.param('outputINArrayName', 'SVDShaperDigitsFromTracks')
# main.add_module(fil)
 
svd.add_svd_reconstruction(main)
 
for moda in main.modules():
    if moda.name() == 'SVDCoGTimeEstimator':
        moda.param("ShaperDigits", 'SVDShaperDigitsFromTracks')
        moda.param("RecoDigits", 'SVDRecoDigitsFromTracks')
    if moda.name() == 'SVDSimpleClusterizer':
        moda.param("Clusters", 'SVDClustersFromTracks')
        moda.param("RecoDigits", 'SVDRecoDigitsFromTracks')
    if moda.name() == 'SVDSpacePointCreator':
        moda.param("SVDClusters", 'SVDClustersFromTracks')
 
calib = SVDCoGTimeCalibrationImporterModule()
calib.set_localdb(localdb)
calib.notApplyCorrectForCDCLatency(False)  # False = apply correction, True = not apply correction
main.add_module(calib)
 
# Show progress of processing
progress = b2.register_module('ProgressBar')
main.add_module(progress)
 
b2.print_path(main)
 
# Process events
b2.process(main)
 
print(b2.statistics)