Belle II Software  release-05-01-25
test_sim_reco.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 import os
5 from basf2 import *
6 from tracking import *
7 from simulation import add_simulation
8 from reconstruction import add_reconstruction
9 from ROOT import Belle2
10 
11 numEvents = 10000
12 
13 set_random_seed(1)
14 
15 eventinfosetter = register_module('EventInfoSetter')
16 eventinfosetter.param('expList', [0])
17 eventinfosetter.param('runList', [1])
18 eventinfosetter.param('evtNumList', [numEvents])
19 
20 eventinfoprinter = register_module('EventInfoPrinter')
21 
22 evtgeninput = register_module('EvtGenInput')
23 evtgeninput.logging.log_level = LogLevel.INFO
24 
25 # Create paths
26 main = create_path()
27 
28 # Add modules to paths
29 main.add_module(eventinfosetter)
30 main.add_module(eventinfoprinter)
31 main.add_module(evtgeninput)
32 add_simulation(main)
33 add_reconstruction(main)
34 
35 fil = register_module('SVDShaperDigitsFromTracks')
36 fil.param('outputINArrayName', 'SVDShaperDigitsFromTracks')
37 main.add_module(fil)
38 
39 input_branches = [
40  'SVDShaperDigitsFromTracks',
41  'EventT0',
42  'SVDShaperDigits'
43 ]
44 
45 main.add_module("RootOutput", branchNames=input_branches, outputFileName="RootOutput_CoGCalibration_10k.root")
46 
47 print_path(main)
48 
49 # Process events
50 process(main)
51 
52 print(statistics)