5 from ROOT
import Belle2
7 from monitor_module
import Monitor
17 if not input_files.empty()
and input_files.front().endswith(
".sroot"):
18 root_input = register_module(
'SeqRootInput')
20 root_input = register_module(
'RootInput')
23 main.add_module(root_input)
26 main.add_module(
'Progress')
31 cdc_unpacker = register_module(
'CDCUnpacker')
32 cdc_unpacker.param(
'enableStoreCDCRawHit',
True)
33 main.add_module(cdc_unpacker)
35 unpacker = register_module(
'CDCTriggerUnpacker')
36 unpacker.logging.log_level = LogLevel.DEBUG
38 unpacker.logging.debug_level = 10
39 unpacker.logging.set_info(LogLevel.DEBUG, LogInfo.LEVEL | LogInfo.MESSAGE)
41 unpacker.param(
'headerSize', 3)
43 unpacker.param(
'unpackTracker2D',
True)
45 unpacker.param(
'decode2DFinderInput',
True)
47 unpacker.param(
'2DNodeId', [
53 main.add_module(unpacker)
57 empty_path = create_path()
58 unpacker.if_false(empty_path)
60 main.add_module(
'Gearbox')
61 main.add_module(
'Geometry', components=[
'BeamPipe',
63 'MagneticFieldConstant4LimitedRCDC'])
64 cdcdigitizer = register_module(
'CDCDigitizer')
67 param_cdcdigi = {
'Fraction': 1,
71 cdcdigitizer.param(param_cdcdigi)
72 cdcdigitizer.param(
'AddInWirePropagationDelay',
True)
73 cdcdigitizer.param(
'AddTimeOfFlight',
True)
74 cdcdigitizer.param(
'UseSimpleDigitization',
True)
75 main.add_module(cdcdigitizer)
76 main.add_module(
'CDCTriggerTSF',
79 TSHitCollectionName=
'TSimSegmentHits')
87 output_name = input_files.front().split(
'/')[-1]
88 output_name = output_name[:output_name.rfind(
'.')] +
'.unpacked.root'
89 main.add_module(
'RootOutput',
90 outputFileName=output_name,
91 excludeBranchNames=[
'RawCDCs',