SVDHotStripsCalibrationsCAF.py

 
 
 
 
import basf2 as b2
 
import os
import sys
import multiprocessing
 
from caf.framework import CAF, Calibration, CentralDatabase
from caf import backends
from caf import strategies
 
import rawdata as raw
# import vertex as vx
 
b2.set_log_level(b2.LogLevel.INFO)
 
input_branches = [
    'RawSVDs'
]
 
# set_log_level(LogLevel.WARNING)
 
 
def SVDHotStripsCalibrations(files, tags):
    # avoid top level ROOT imports
    from ROOT import Belle2  # noqa: make Belle2 namespace available
    from ROOT.Belle2 import SVDHotStripsCalibrationsAlgorithm
 
    # Set-up re-processing path
    path = b2.create_path()
 
    # logging.log_level = LogLevel.WARNING
 
    path.add_module('Progress')
    # Remove all non-raw data to run the full reco again
    path.add_module('RootInput', branchNames=input_branches)
 
    path.add_module("Gearbox")
    path.add_module("Geometry", useDB=True)
 
    # fil = register_module('SVDShaperDigitsFromTracks')
    # fil.param('outputINArrayName', 'SVDShaperDigitsFromTracks')
    # main.add_module(fil)
 
    # Not needed for di-muon skim cdst or mdst, but needed to re-run reconstruction
    # with possibly changed global tags
    raw.add_unpackers(path, components=['SVD'])
 
    collector = b2.register_module('SVDOccupancyCalibrationsCollector')
    collector.param("SVDShaperDigitsName", "SVDShaperDigits")
    algorithm = SVDHotStripsCalibrationsAlgorithm("SVDHotStripsCAF")
 
    calibration = Calibration('SVDHotStrips',
                              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__":
    # use by default raw data from cdst of exp8, run1309 (shaperDigits need to be unpacked, not available in cdst format)
    input_files = [os.path.abspath(file) for file in [
            "/group/belle2/dataprod/Data/Raw/e0008/r01309/sub00/physics.0008.01309.HLT5.f00098.root"]]
    #       "/group/belle2/dataprod/Data/Raw/e0008/r01309/sub00/physics.0008.01309.HLT5*"]]
# "/group/belle2/dataprod/Data/release-03-02-02/DB00000635/proc00000009/\
# e0008/4S/r01309/skim/hlt_bhabha/cdst/sub00/cdst.physics.0008.01309.HLT*"]]
 
    if not len(input_files):
        print("You have to specify some input file(s)\n"
              "using the standard basf2 command line option - i")
        print("See: basf2 -h")
        sys.exit(1)
 
    svdOccupCAF = SVDHotStripsCalibrations(input_files,
                                           ['giulia_CDCEDepToADCConversions_rel4_patch',
                                            'data_reprocessing_prompt_rel4_patch'])
    # beamspot.max_iterations = 0
 
    cal_fw = CAF()
    cal_fw.add_calibration(svdOccupCAF)
    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()