run_eclHadronTimeCalibrationValidation_collector.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
#
# This script executes the hadron timing calibration collector,
# which reads in the hadron events to perform the hadron
# validation, where the final validation fits is performed
# by the algorithm.
# This script is run directly with basf2.
#
# There are two ways you can use this script:
# 1. Provide parameters from command line:
#   basf2 run_eclHadronTimeCalibrationValidation_collector.py -i "/path/to/input/files/*.root" -o collector_output.root
#
# 2. Set parameters directly in steering file.
#   Change INPUT_LIST and OUTPUT variables.
#   (Multiple files can be easily added with glob.glob("/path/to/your/files/*.root"))
#   And then call
#   basf2 run_eclHadronTimeCalibrationValidation_collector.py
 
 
import basf2 as b2
from ROOT import Belle2
import glob
import sys
import tracking
import rawdata
import reconstruction
 
from basf2 import conditions as b2conditions
from reconstruction import prepare_user_cdst_analysis
 
 
env = Belle2.Environment.Instance()
 
 
 
# == List of input files
# NOTE: It is going to be sorted (alphabetic and length sorting, files with
#       shortest names are first)
INPUT_LIST = []
# = Processed data
INPUT_LIST += glob.glob("oneTestFile/*.root")
 
# == Output file
OUTPUT = "eclHadronTimeCalibrationValidationCollector.root"
 
 
 
# Override input if "-i file.root" argument was sent to basf2.
input_arg = env.getInputFilesOverride()
if len(input_arg) > 0:
    INPUT_LIST = [str(x) for x in input_arg]
# Sort list of input files.
INPUT_LIST.sort(key=lambda item: (len(item), item))
 
# Override output if "-o file.root" argument was sent to basf2.
output_arg = env.getOutputFileOverride()
if len(output_arg) > 0:
    OUTPUT = output_arg
 
 
 
 
# Events with abs(time_ECL-time_CDC) > TIME_ABS_MAX are excluded
TIME_ABS_MAX = 70
 
# If true, output file will contain TTree "tree" with detailed
# event information.
SAVE_TREE = True
 
 
 
components = ['CDC', 'ECL']
 
# == Create path
main = b2.create_path()
 
add_unpackers = False
 
# == SeqRoot/Root input
if INPUT_LIST[0].endswith('sroot'):
    main.add_module('SeqRootInput', inputFileNames=INPUT_LIST)
    add_unpackers = True
else:
    main.add_module('RootInput', inputFileNames=INPUT_LIST)
 
main.add_module("HistoManager", histoFileName=OUTPUT)
 
if 'Raw' in INPUT_LIST[0]:
    add_unpackers = True
 
 
main.add_module('Gearbox')
 
if add_unpackers:
    rawdata.add_unpackers(main, components=components)
 
    # = Get Tracks, RecoTracks, ECLClusters, add relations between them.
    tracking.add_tracking_reconstruction(main, components=components)
    reconstruction.add_ext_module(main, components)
    reconstruction.add_ecl_modules(main, components)
    reconstruction.add_ecl_track_matcher_module(main, components)
 
prepare_user_cdst_analysis(main)  # for new 2020 cdst format
 
# == Generate time calibration matrix from ECLDigit
ECLtimeCalibValidationCollectorInfo = main.add_module('eclHadronTimeCalibrationValidationCollector', timeAbsMax=TIME_ABS_MAX,
                                                      saveTree=SAVE_TREE)
 
ECLtimeCalibValidationCollectorInfo.set_log_level(b2.LogLevel.INFO)     # OR: b2.LogLevel.DEBUG
ECLtimeCalibValidationCollectorInfo.set_debug_level(36)
 
 
# == Show progress
main.add_module('Progress')
 
b2.set_log_level(b2.LogLevel.INFO)
b2.set_debug_level(100)
 
# == Configure database
# reset_database()
# use_database_chain()
 
b2conditions.reset()
b2conditions.override_globaltags()
 
b2.B2INFO("Adding Local Database {} to head of chain of local databases.")
b2conditions.prepend_testing_payloads("localdb/database.txt")
b2.B2INFO("Using Global Tag {}")
b2conditions.prepend_globaltag("dp_recon_release6_patch")
b2conditions.prepend_globaltag("Reco_master_patch_rel5")
b2conditions.prepend_globaltag("AIRFLOW_online_snapshot_20210719-124441")
b2conditions.prepend_globaltag("data_reprocessing_prompt")
 
 
# == Process events
# b2.process(main, max_event=350000)    # reasonable stats for one crate
# b2.process(main, max_event=600000)    # reasonable stats for crystal calibs for proc10
# b2.process(main, max_event=3000)      # reasonable stats and speed for a quick test
# b2.process(main, max_event=30)        # fast test
b2.process(main)                      # process all events
 
print(b2.statistics)