pxd_gain_validation.py

# -*- coding: utf-8 -*-
 
 
 
'''
Validation of PXD gain calibration.
'''
 
 
import basf2
from prompt import ValidationSettings
import ROOT
import sys
import subprocess
import math
from glob import glob
 
 
 
settings = ValidationSettings(name='PXD gain calibration',
                              description=__doc__,
                              download_files=['PXDPerformanceVariablesCollectorValidation.root'],
                              expert_config={})
 
 
def run_validation(job_path, input_data_path, requested_iov, expert_config):
    '''
    Run the validation.
    '''
    from conditions_db.testing_payloads import parse_testing_payloads_file
    txt_files = glob(f"{job_path}/*PXD*Gain*/outputdb/database.txt")
    cmds = ["b2pxd-db-check", "--only-hist", "--maps",
            "PXDGainMapPar"]
    for txt_file in txt_files:
        iov_entries = parse_testing_payloads_file(txt_file)
        exp = iov_entries[0].firstRun["exp"]
        run = iov_entries[0].firstRun["run"]
        cmds_add = ["--tag", txt_file,
                    "--exp", f"{exp}", "--runs", f"{run}-9999",
                    "-o", f"conditions_gain_e{exp}_r{run}.root"]
        subprocess.run(cmds + cmds_add, check=True)
 
    root_files = glob(f"{job_path}/*PXD*Gain*/*/algorithm_output/PXDPerformanceVariablesCollectorValidation.root")
    for root_file in root_files:
        cmds = ["b2pxd-val", "-i", root_file]
        subprocess.run(cmds, check=True)
 
 
if __name__ == "__main__":
    run_validation(*sys.argv[1:])