Belle II Software  release-05-01-25
collector.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 import os
5 import sys
6 
7 from basf2 import *
8 # from simulation import register_simulation
9 # from reconstruction import register_reconstruction
10 
11 set_log_level(LogLevel.ERROR)
12 
13 argvs = sys.argv
14 argc = len(argvs)
15 
16 print(argvs[1])
17 print(argc)
18 
19 # detecor components to be reconstructed
20 components = [
21  'MagneticField',
22  'BeamPipe',
23  'PXD',
24  'SVD',
25  'CDC',
26  'TOP',
27  'ARICH',
28  'BKLM',
29  'ECL',
30 ]
31 
32 # Register modules to declare objects
33 # register_simulation(components)
34 # register_reconstruction(components)
35 
36 # Register RoI related modules for object decoding
37 roiGen = register_module('ROIGenerator')
38 roiPayloadAssembler = register_module('ROIPayloadAssembler')
39 
40 # create a main path
41 main = create_path()
42 
43 # Load Geometry module
44 gearbox = register_module('Gearbox')
45 geometry = register_module('Geometry')
46 
47 # Add input module
48 # input = register_module("SeqRootInput")
49 # input.param ( "inputFileName", "/fcdisk1-1/data/sim/sim-evtgen.sroot")
50 # main.add_module(input)
51 
52 # Add output module
53 # output= register_module("SeqRootOutput")
54 # output.param ( "outputFileName", "/dev/null" )
55 # main.add_module(output)
56 
57 # Add Rbuf2Ds
58 rbuf2ds = register_module("Rbuf2Ds")
59 # rbuf2ds = register_module("FastRbuf2Ds")
60 rbuf2ds.param("RingBufferName", argvs[1])
61 # rbuf2ds.param("NumThreads", 4 )
62 main.add_module(rbuf2ds)
63 
64 # Add Progress
65 progress = register_module("Progress")
66 progress.param('maxN', 4)
67 main.add_module(progress)
68 
69 # Add Elapsed Time
70 elapsed = register_module("ElapsedTime")
71 elapsed.param("EventInterval", 20000)
72 main.add_module(elapsed)
73 
74 # Add Ds2Rbuf
75 # ds2rbuf = register_module("Ds2Rbuf")
76 # ds2rbuf.param("RingBufferName", argvs[2])
77 # main.add_module(ds2rbuf)
78 
79 # Add Ds2Raw
80 ds2rbuf = register_module("Ds2Raw")
81 ds2rbuf.param("RingBufferName", argvs[2])
82 main.add_module(ds2rbuf)
83 
84 # Test seqrootoutput
85 # output = register_module("SeqRootOutput" )
86 # output.param ( "outputFileName", "/dev/null" )
87 # main.add_module(output)
88 
89 # Run
90 process(main)