cr_example.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
from basf2 import *
from reconstruction import add_cosmics_reconstruction
from cdc.cr import *
 
# use the CDC CR tag
use_central_database("cdc_cr_test1", LogLevel.WARNING)
 
# create path
main = create_path()
 
# specify number of events to be generated
rootinput = register_module('RootInput')
input = '/ghi/fs01/belle2/bdata/group/detector/CDC/unpacked/exp00/cr.cdc.0000.001733.root'
rootinput.param('inputFileName', input)
main.add_module(rootinput)
 
# this sample is for CDC cosmics only
components = ['CDC']
 
# get the run number
run_number = getRunNumber(input)
 
# Set the peiod of data taking.
data_period = getDataPeriod(run_number)
 
# create a working directory for this sample
if os.path.exists(data_period) is False:
    os.mkdir(data_period)
 
# gearbox & geometry needs to be registered
globalPhiRotation = getPhiRotation()
main.add_module('Gearbox',
                override=[("/DetectorComponent[@name='CDC']//GlobalPhiRotation", str(globalPhiRotation), "deg")
                          ],
                fileName='/geometry/Beast2_phase2.xml')
main.add_module('Geometry', components=components)
 
# Set CDC CR parameters.
set_cdc_cr_parameters(data_period)
 
# reconstruction
add_cosmics_reconstruction(main, components=components)
 
# full output
main.add_module('RootOutput', outputFileName='cr_output.root')
 
# display a progress bar while running
main.add_module('ProgressBar')
 
# process events and print call statistics
process(main)
print(statistics)