caf_ecl_WaveformTemplates.py

# -*- coding: utf-8 -*-
 
 
 
"""ECL waveform template calibration"""
 
from prompt import CalibrationSettings, INPUT_DATA_FILTERS
 
# --------------------------------------------------------------
# ..Tell the automated script some required details
settings = CalibrationSettings(
    name="ecl_WaveformTemplateShapeCalibration",
    expert_username="longos",
    description=__doc__,
    input_data_formats=["cdst"],
    input_data_names=["gamma_gamma_calib"],
    input_data_filters={
        "gamma_gamma_calib": [
            INPUT_DATA_FILTERS["Data Tag"]["gamma_gamma_calib"],
            INPUT_DATA_FILTERS["Data Quality Tag"]["Good Or Recoverable"],
            INPUT_DATA_FILTERS["Run Type"]["physics"],
            INPUT_DATA_FILTERS["Magnet"]["On"]]},
    depends_on=[],
    expert_config={
        "C2_MinEnergyThreshold": 2.0,
        "nFilesCollector": 50
    })
 
 
# --------------------------------------------------------------
# ..The calibration functions
 
def get_calibrations(input_data, **kwargs):
    import basf2
    from ROOT import Belle2
    from caf.utils import IoV
    from caf.framework import Calibration
 
    # --------------------------------------------------------------
    # ..gamma gamma
 
    # ..Input data
    file_to_iov_gamma_gamma = input_data["gamma_gamma_calib"]
    input_files = list(file_to_iov_gamma_gamma.keys())
 
    # ..Algorithm
    algo_C1 = Belle2.ECL.eclWaveformTemplateCalibrationC1Algorithm()
 
    expert_config = kwargs.get("expert_config")
    C2_MinEnergyThreshold = expert_config["C2_MinEnergyThreshold"]
    nFilesCollector = expert_config["nFilesCollector"]
 
    # ..The calibration
    collector_C1 = basf2.register_module("eclWaveformTemplateCalibrationC1Collector")
    collector_C1.param('MinEnergyThreshold', C2_MinEnergyThreshold)
 
    cal_ecl_Wave_C1 = Calibration(
        "ecl_Wave_C1",
        collector=collector_C1,
        algorithms=[algo_C1],
        input_files=input_files,
        max_files_per_collector_job=nFilesCollector)
 
    # ..Add prepare_cdst_analysis to pre_collector_path
    gamma_gamma_pre_path = basf2.create_path()
    gamma_gamma_pre_path.add_module("RootInput", inputFileNames="", branchNames=["EventMetaData", "RawECLs"])
    eclunpacker = basf2.register_module('ECLUnpacker')
    gamma_gamma_pre_path.add_module(eclunpacker)
 
    cal_ecl_Wave_C1.pre_collector_path = gamma_gamma_pre_path
 
    calibrations_C2 = []
    algos_C2 = []
    algos_C3 = []
    collectors_C2 = []
 
    batchsize = 100
    nbatches = 88
 
    # keep option to run in parallel
    for i in range(0, nbatches):
 
        lowLimit = (batchsize*i)+1
 
        highLimit = (batchsize*(i+1))
 
        if (highLimit > 8736):
            highLimit = 8736
 
        print("lowLimit,highLimit", lowLimit, highLimit)
 
        # ..Algorithm
        algos_C2.append(Belle2.ECL.eclWaveformTemplateCalibrationC2Algorithm())
        algos_C2[-1].setFirstCellID(lowLimit)
        algos_C2[-1].setLastCellID(highLimit)
 
        algos_C3.append(Belle2.ECL.eclWaveformTemplateCalibrationC3Algorithm())
        algos_C3[-1].setFirstCellID(lowLimit)
        algos_C3[-1].setLastCellID(highLimit)
 
        collectors_C2.append(basf2.register_module("eclWaveformTemplateCalibrationC2Collector"))
        collectors_C2[-1].pre_collector_path = gamma_gamma_pre_path
        collectors_C2[-1].param('MinCellID', lowLimit)
        collectors_C2[-1].param('MaxCellID', highLimit)
        collectors_C2[-1].param('MinEnergyThreshold', C2_MinEnergyThreshold)
 
        # ..The calibration
        calibrations_C2.append(Calibration("ecl_Wave_C2_"+str(lowLimit)+"_"+str(highLimit),
                                           collector=collectors_C2[-1],
                                           algorithms=[algos_C2[-1],
                                                       algos_C3[-1]],
                                           input_files=input_files,
                                           max_files_per_collector_job=nFilesCollector))
        calibrations_C2[-1].pre_collector_path = gamma_gamma_pre_path
        calibrations_C2[-1].depends_on(cal_ecl_Wave_C1)
 
    # ..Algorithm
    algo_C4 = Belle2.ECL.eclWaveformTemplateCalibrationC4Algorithm()
    algo_C4.setFirstCellID(1)
    algo_C4.setLastCellID(8736)
 
    # ..The calibration
    cal_ecl_Wave_C4 = Calibration("ecl_Wave_C4",
                                  collector="DummyCollector",
                                  algorithms=[algo_C4],
                                  input_files=input_files[:1],
                                  )
 
    cal_ecl_Wave_C4.depends_on(cal_ecl_Wave_C1)
    for cal in calibrations_C2:
        cal_ecl_Wave_C4.depends_on(cal)
 
    # --------------------------------------------------------------
    # ..Force the output iovs to be open
    requested_iov = kwargs.get("requested_iov", None)
    output_iov = IoV(requested_iov.exp_low, requested_iov.run_low, -1, -1)
    for algorithm in cal_ecl_Wave_C1.algorithms:
        algorithm.params = {"apply_iov": output_iov}
    for C2 in calibrations_C2:
        for algorithm in C2.algorithms:
            algorithm.params = {"apply_iov": output_iov}
    for algorithm in cal_ecl_Wave_C4.algorithms:
        algorithm.params = {"apply_iov": output_iov}
 
    # --------------------------------------------------------------
    # ..Return the calibrations
    calList = [cal_ecl_Wave_C1] + calibrations_C2 + [cal_ecl_Wave_C4]
    return calList