Belle II Software  release-05-01-25
pxdDataReductionClusters.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 simulation import add_roiFinder
9 from svd import add_svd_reconstruction
10 from tracking import add_tracking_for_PXDDataReduction_simulation
11 from ROOT import Belle2
12 
13 numEvents = 100
14 
15 # first register the modules
16 
17 set_random_seed(1)
18 
19 eventinfosetter = register_module('EventInfoSetter')
20 eventinfosetter.param('expList', [0])
21 eventinfosetter.param('runList', [1])
22 eventinfosetter.param('evtNumList', [numEvents])
23 
24 eventinfoprinter = register_module('EventInfoPrinter')
25 
26 evtgeninput = register_module('EvtGenInput')
27 evtgeninput.logging.log_level = LogLevel.INFO
28 
29 pxdROIFinder = register_module('PXDROIFinder')
30 pxdROIFinder.logging.log_level = LogLevel.DEBUG
31 # pxdROIFinder.logging.debug_level = 2
32 param_pxdROIFinder = {
33  'recoTrackListName': 'RecoTracks',
34  'PXDInterceptListName': 'PXDIntercepts',
35  'ROIListName': 'ROIs',
36  'tolerancePhi': 0.15,
37  'toleranceZ': 0.5,
38  # optimized performance
39  # 'sigmaSystU': 0.1,
40  # 'sigmaSystV': 0.1,
41  # 'numSigmaTotU': 10,
42  # 'numSigmaTotV': 10,
43  # 'maxWidthU': 2,
44  # 'maxWidthV': 6,
45  # official simulation
46  'sigmaSystU': 0.02,
47  'sigmaSystV': 0.02,
48  'numSigmaTotU': 10,
49  'numSigmaTotV': 10,
50  'maxWidthU': 0.5,
51  'maxWidthV': 0.5,
52 }
53 pxdROIFinder.param(param_pxdROIFinder)
54 
55 # PXD digitization module
56 PXDDIGI = register_module('PXDDigitizer')
57 # PXD clusterizer
58 PXDCLUST = register_module('PXDClusterizer')
59 
60 pxdClusterFilter = register_module('PXDclusterFilter')
61 pxdClusterFilter.param({'ROIidsName': 'ROIs', 'CreateOutside': True, 'overrideDB': True, 'enableFiltering': True})
62 
63 # Create paths
64 main = create_path()
65 
66 # Add modules to paths
67 main.add_module(eventinfosetter)
68 main.add_module(eventinfoprinter)
69 main.add_module(evtgeninput)
70 add_simulation(main, components=['PXD', 'SVD', 'CDC'], forceSetPXDDataReduction=True, usePXDDataReduction=False)
71 add_tracking_reconstruction(main, ['SVD', 'CDC'])
72 main.add_module(pxdROIFinder)
73 main.add_module(PXDDIGI)
74 main.add_module(PXDCLUST)
75 main.add_module(pxdClusterFilter)
76 # display = register_module("Display")
77 # main.add_module(display)
78 
79 main.add_module('RootOutput')
80 
81 # Process events
82 process(main)
83 
84 print(statistics)