Belle II Software  release-05-01-25
collector_multi.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 modules for object definitions
37 gearbox = register_module('Gearbox')
38 SVDUnpack = register_module('SVDUnpacker')
39 SVDClust = register_module('SVDClusterizer')
40 vxdtf = register_module('VXDTF')
41 SVD_DQM = register_module('SVDDQM')
42 vxdtf_dqm = register_module('VXDTFDQM')
43 trackfitter = register_module('GenFitter')
44 roiprod = register_module('PXDROIFinder')
45 roipayload = register_module('ROIPayloadAssembler')
46 
47 # create a main path
48 main = create_path()
49 
50 # Load Geometry module
51 gearbox = register_module('Gearbox')
52 geometry = register_module('Geometry')
53 
54 # Add input module
55 # input = register_module("SeqRootInput")
56 # input.param ( "inputFileName", "/fcdisk1-1/data/sim/sim-evtgen.sroot")
57 # main.add_module(input)
58 
59 # Add output module
60 # output= register_module("SeqRootOutput")
61 # output.param ( "outputFileName", "/dev/null" )
62 # main.add_module(output)
63 
64 # Add Rbuf2Ds
65 rbuf2ds = register_module("Rbuf2Ds")
66 rbuf2ds.param("RingBufferName", argvs[1])
67 main.add_module(rbuf2ds)
68 
69 # HLT Tagger
70 tagger = register_module("HLTTagger")
71 main.add_module(tagger)
72 
73 # Add Progress
74 progress = register_module("Progress")
75 main.add_module(progress)
76 
77 # Add Elapsed Time
78 elapsed = register_module("ElapsedTime")
79 elapsed.param("EventInterval", 1000)
80 main.add_module(elapsed)
81 
82 # Add Ds2Rbuf
83 # ds2rbuf = register_module("Ds2Rbuf")
84 # ds2rbuf.param("RingBufferName", argvs[2])
85 # main.add_module(ds2rbuf)
86 
87 # Add Ds2Raw
88 ds2rbuf = register_module("Ds2Raw")
89 ds2rbuf.param("RingBufferName", argvs[2])
90 main.add_module(ds2rbuf)
91 
92 # Test seqrootoutput
93 # output = register_module("SeqRootOutput" )
94 # output.param ( "outputFileName", "/dev/null" )
95 # output.param ( "outputFileName", "/data1/data/TestOutput.sroot" )
96 # main.add_module(output)
97 
98 # Run
99 process(main)