KLMDQM2Test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
import modularAnalysis as ma
import rawdata as raw
import reconstruction as re
from daqdqm.commondqm import add_common_dqm
import basf2 as b2
 
# Set the log level to show only error and fatal messages
# set_log_level(LogLevel.ERROR)
b2.set_log_level(b2.LogLevel.INFO)
 
path = "/group/belle2/dataprod/Data/Raw/e0018/r01553/sub00/"
files = path+"physics.0018.01553.HLT8.f0000*.root"
 
# Create main path
main = b2.create_path()
 
# useful tags for release-06
b2.conditions.globaltags = ['online',
                            'dp_recon_release6_patch',
                            'ecl_release06_forOnline']
 
main.add_module("RootInput", inputFileNames=files)
main.add_module("HistoManager", histoFileName="KLMDQMHistograms.root")
main.add_module('Gearbox')
main.add_module('Geometry')
raw.add_unpackers(main)
re.add_reconstruction(main, add_muid_hits=True)
 
software_trigger = "software_trigger_cut&skim&accept_mumutight"
cut_string = f'SoftwareTriggerResult({software_trigger})>0'
 
add_common_dqm(main, dqm_environment='HLT', dqm_mode='dont_care')
ma.fillParticleList('mu+:all', cut="", path=main)
ma.cutAndCopyList("mu+:cut", "mu+:all", cut_string, path=main)
 
collector = b2.register_module('KLMStripEfficiencyCollector')
collector.param('Debug', True)
collector.param('DebugFileName', 'KLMStripEffCollector_debugFile.root')
collector.param('MuonListName', 'mu+:cut')  # to be consistent with KLMDQM2
main.add_module(collector)
 
klm2 = b2.register_module('KLMDQM2')
main.add_module(klm2)
 
main.add_module('Progress')
b2.print_path(main)
 
 
# Process all events
b2.process(main, max_event=1000)
print(b2.statistics)