SVDCrossTalkCalibrationsCAF.py

 
 
 
 
import basf2 as b2
 
import os
import multiprocessing
 
from ROOT import Belle2
from ROOT.Belle2 import SVDCrossTalkCalibrationsAlgorithm
 
from caf.framework import CAF, Calibration, CentralDatabase
from caf import backends
from caf import strategies
 
import rawdata as raw
 
 
b2.set_log_level(b2.LogLevel.INFO)
input_branches = [
    'RawSVDs'
]
 
 
def SVDCrossTalkCalibrations(files, tags):
 
    # Set-up re-processing path
    path = b2.create_path()
 
    path.add_module('Progress')
    path.add_module('RootInput', branchNames=input_branches)
 
    path.add_module("Gearbox")
    path.add_module("Geometry")
 
    raw.add_unpackers(path, components=['SVD'])
 
    collector = b2.register_module('SVDCrossTalkCalibrationsCollector')
    algorithm = SVDCrossTalkCalibrationsAlgorithm("SVDCrossTalkCAF")
 
    calibration = Calibration('SVDCrossTalk',
                              collector=collector,
                              algorithms=algorithm,
                              input_files=files,
                              pre_collector_path=path,
                              database_chain=[CentralDatabase(tag) for tag in tags],
                              output_patterns=None,
                              max_files_per_collector_job=-1,
                              backend_args=None
                              )
 
    calibration.strategies = strategies.SequentialRunByRun
 
    return calibration
 
 
if __name__ == "__main__":
    #    input_files = [os.path.abspath(file) for file in [
    #            "/group/belle2/dataprod/Data/Raw/e0008/r01309/sub00/physics.0008.01309.HLT5.f00098.root"]]
    input_files = [os.path.abspath(file) for file in Belle2.Environment.Instance()
                   .getInputFilesOverride()]
 
    print(" ")
    print("INPUT FILES")
    print(" ")
    print(input_files)
    print(" ")
 
    svdCrossTalkCAF = SVDCrossTalkCalibrations(input_files,
                                               ['data_reprocessing_prompt',
                                                'svd_basic', 'online'])
 
    cal_fw = CAF()
    cal_fw.add_calibration(svdCrossTalkCAF)
    cal_fw.backend = backends.LSF()
 
    # Try to guess if we are at KEKCC and change the backend to Local if not
    if multiprocessing.cpu_count() < 10:
        cal_fw.backend = backends.Local(8)
 
    cal_fw.run()