caf_simplest_iov.py

 
 
import basf2 as b2
 
import os
import sys
 
from ROOT.Belle2 import TestCalibrationAlgorithm
from caf.framework import Calibration, CAF
from caf.utils import IoV
 
b2.set_log_level(b2.LogLevel.INFO)
# add time stamp to all INFO messages
# currentInfo = logging.get_info(LogLevel.INFO)
# logging.set_info(LogLevel.INFO, currentInfo | LogInfo.TIMESTAMP)
 
 
def main(argv):
    if len(argv) == 1:
        data_dir = argv[0]
    else:
        print("Usage: basf2 CAF_simplest.py <data directory>")
        sys.exit(1)
 
    
    input_files_test = []
    input_files_test.append(os.path.join(os.path.abspath(data_dir), '*.root'))
 
    
    alg_test = TestCalibrationAlgorithm()  # Getting a calibration algorithm instance
 
    # Create a single calibration from a collector module name + algorithm + input files
    cal_test = Calibration(name="TestCalibration", collector="CaTest", algorithms=alg_test, input_files=input_files_test)
 
    
    cal_fw = CAF()
    cal_fw.add_calibration(cal_test)
 
    # Let's only calibrate a subset of the data
    iov_to_calibrate = IoV(exp_low=0, run_low=2, exp_high=0, run_high=3)
    cal_fw.run(iov=iov_to_calibrate)
    print("End of CAF processing.")
 
 
if __name__ == "__main__":
    main(sys.argv[1:])