Belle II Software  release-05-02-19
executionTime.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
15 
16 import sys
17 import os
18 from basf2 import *
19 from svd.executionTime_utils import SVDExtraEventStatisticsModule
20 import simulation as simu
21 import tracking as trk
22 
23 import glob
24 
25 tag = "test"
26 if len(sys.argv) == 2:
27  tag = sys.argv[1]
28 
29 # EXAMPLE OF EXECUTION TIME MEASUREMENT
30 main = create_path()
31 
32 set_random_seed(1)
33 
34 # set the exp/run event informations
35 eventinfosetter = register_module('EventInfoSetter')
36 eventinfosetter.param('expList', [0])
37 eventinfosetter.param('runList', [1])
38 eventinfosetter.param('evtNumList', [10])
39 main.add_module(eventinfosetter)
40 
41 # generate signal
42 main.add_module('EvtGenInput')
43 
44 # add default simulation
45 bkgDir = '/group/belle2/BGFile/OfficialBKG/early_phase3/prerelease-04-00-00a/overlay/phase31/BGx1/set0/*.root'
46 bg = glob.glob(bkgDir)
47 if len(bg) == 0:
48  B2ERROR('No files found in ', bkgDir)
49  sys.exit()
50 simu.add_simulation(main, bkgfiles=bg, usePXDDataReduction=False, forceSetPXDDataReduction=True)
51 
52 # add default tracking reconstruction
53 trk.add_tracking_reconstruction(main)
54 
55 # add offline ZS for ExecutionTime module
56 main.add_module(
57  'SVDZeroSuppressionEmulator',
58  SNthreshold=5,
59  ShaperDigits='SVDShaperDigits',
60  ShaperDigitsIN='SVDShaperDigitsZS5',
61  FADCmode=True)
62 
63 main.add_module(SVDExtraEventStatisticsModule("SVDExecutionTime_"+str(tag)+".root"))
64 
65 main.add_module('Progress')
66 
67 process(main)
68 
69 print(statistics)
svd.executionTime_utils
Definition: executionTime_utils.py:1