Belle II Software  release-08-01-10
trackingInputValidationBkg.py
1 #!/usr/bin/env python3
2 
3 
10 
11 """
12 <header>
13  <contact>software-tracking@belle2.org</contact>
14  <input>EvtGenSim.root</input>
15  <output>TrackingInputValidationBkg.root</output>
16  <description>
17  This module checks the number of input hits for tracking.
18  </description>
19 </header>
20 """
21 
22 from tracking.validation.plot import ValidationPlot
23 from ROOT import Belle2
24 import collections
25 import ROOT
26 import basf2
27 from svd import add_svd_reconstruction
28 from pxd import add_pxd_reconstruction
29 NAME = 'Tracking Input Validation' # not used?
30 CONTACT = 'software-tracking@belle2.org'
31 INPUT_FILE = '../EvtGenSim.root'
32 OUTPUT_FILE = 'TrackingInputValidationBkg.root'
33 N_EVENTS = 1000
34 
35 ACTIVE = True
36 
37 
38 def run():
39  """
40  Check the number of hits that are the input for tracking.
41  """
42  class TrackingInputValidation(basf2.Module):
43  """
44  Module to collect information about the number of
45  * PXDDigits
46  * PXDClusters
47  * PXDSpacePoints
48  * SVDShaperDigits
49  * SVDClusters
50  * SVDSpacePoints
51  * CDCHits
52  """
53 
54  def __init__(
55  self,
56  name='',
57  contact='',
58  output_file_name=None,
59  ):
60  """Constructor"""
61 
62  super().__init__()
63 
64  self.validation_name = NAME
65 
66  self.contact = CONTACT
67 
68  self.output_file_name = OUTPUT_FILE
69 
70  def initialize(self):
71  """Receive signal at the start of event processing"""
72 
73 
74  self.PXDDigits = Belle2.PyStoreArray('PXDDigits')
75 
76  self.PXDClusters = Belle2.PyStoreArray('PXDClusters')
77 
78  self.PXDSpacePoints = Belle2.PyStoreArray('PXDSpacePoints')
79 
80 
81  self.SVDShaperDigits = Belle2.PyStoreArray('SVDShaperDigits')
82 
83  self.SVDClusters = Belle2.PyStoreArray('SVDClusters')
84 
85  self.SVDSpacePoints = Belle2.PyStoreArray('SVDSpacePoints')
86 
87 
88  self.CDCHits = Belle2.PyStoreArray('CDCHits')
89 
90 
91  self.nPXDDigits = collections.deque()
92 
93  self.nPXDClusters = collections.deque()
94 
95  self.nPXDSpacePoints = collections.deque()
96 
97 
98  self.nSVDShaperDigits = collections.deque()
99 
100  self.nSVDClusters = collections.deque()
101 
102  self.nSVDSpacePoints = collections.deque()
103 
104 
105  self.nCDCHits = collections.deque()
106 
107  def event(self):
108  '''Event function'''
109 
110  self.nPXDDigits.append(self.PXDDigits.getEntries())
111  self.nPXDClusters.append(self.PXDClusters.getEntries())
112  self.nPXDSpacePoints.append(self.PXDSpacePoints.getEntries())
113  self.nSVDShaperDigits.append(self.SVDShaperDigits.getEntries())
114  self.nSVDClusters.append(self.SVDClusters.getEntries())
115  self.nSVDSpacePoints.append(self.SVDSpacePoints.getEntries())
116  self.nCDCHits.append(self.CDCHits.getEntries())
117 
118  def terminate(self):
119  """Receive signal at the end of event processing"""
120 
121  ''' Saving'''
122  output_tfile = ROOT.TFile(self.output_file_name, 'recreate')
123 
124  h_nPXDDigits = ValidationPlot('h_nPXDDigits')
125  h_nPXDDigits.hist(self.nPXDDigits, bins=100, lower_bound=0, upper_bound=2000)
126  h_nPXDDigits.contact = self.contact
127  h_nPXDDigits.check = 'Average of 500 +/- 100 is expected.'
128  h_nPXDDigits.description = 'Number of PXDDigits after ROI filtering'
129  h_nPXDDigits.title = 'Number of selected PXDDigits per event'
130  h_nPXDDigits.xlabel = 'Number of PXDDigits'
131  h_nPXDDigits.ylabel = ''
132  h_nPXDDigits.write(output_tfile)
133 
134  h_nPXDClusters = ValidationPlot('h_nPXDClusters')
135  h_nPXDClusters.hist(self.nPXDClusters, bins=100, lower_bound=0, upper_bound=500)
136  h_nPXDClusters.contact = self.contact
137  h_nPXDClusters.check = 'Average of 150 +/- 30 is expected'
138  h_nPXDClusters.description = 'Number of PXDClusters after ROI filtering.'
139  h_nPXDClusters.title = 'Number of PXDClusters per event'
140  h_nPXDClusters.xlabel = 'Number of PXDClusters'
141  h_nPXDClusters.ylabel = ''
142  h_nPXDClusters.write(output_tfile)
143 
144  h_nPXDSpacePoints = ValidationPlot('h_nPXDSpacePoints')
145  h_nPXDSpacePoints.hist(self.nPXDSpacePoints, bins=100, lower_bound=0, upper_bound=500)
146  h_nPXDSpacePoints.contact = self.contact
147  h_nPXDSpacePoints.check = 'Average of 150 +/- 30 is expected'
148  h_nPXDSpacePoints.description = 'Number of PXDSpacePoints after ROI filtering. \
149  Should be the same as the number of PXDClusters.'
150  h_nPXDSpacePoints.title = 'Number of PXDSpacePoints per event'
151  h_nPXDSpacePoints.xlabel = 'Number of PXDSpacePoints'
152  h_nPXDSpacePoints.ylabel = ''
153  h_nPXDSpacePoints.write(output_tfile)
154 
155  h_nSVDShaperDigits = ValidationPlot('h_nSVDShaperDigits')
156  h_nSVDShaperDigits.hist(self.nSVDShaperDigits, bins=100, lower_bound=0, upper_bound=10000)
157  h_nSVDShaperDigits.contact = self.contact
158  h_nSVDShaperDigits.check = 'First peak at about 5500 +/- 200 is expected.'
159  h_nSVDShaperDigits.description = 'Number of SVDDigits'
160  h_nSVDShaperDigits.title = 'Number of SVDDigits per event'
161  h_nSVDShaperDigits.xlabel = 'Number of SVDDigits'
162  h_nSVDShaperDigits.ylabel = ''
163  h_nSVDShaperDigits.write(output_tfile)
164 
165  h_nSVDClusters = ValidationPlot('h_nSVDClusters')
166  h_nSVDClusters.hist(self.nSVDClusters, bins=100, lower_bound=0, upper_bound=2000)
167  h_nSVDClusters.contact = self.contact
168  h_nSVDClusters.check = 'Average of 1100 +/- 100 is expected.'
169  h_nSVDClusters.description = 'Number of SVDClusters'
170  h_nSVDClusters.title = 'Number of SVDClusters per event'
171  h_nSVDClusters.xlabel = 'Number of SVDClusters'
172  h_nSVDClusters.ylabel = ''
173  h_nSVDClusters.write(output_tfile)
174 
175  h_nSVDSpacePoints = ValidationPlot('h_nSVDSpacePoints')
176  h_nSVDSpacePoints.hist(self.nSVDSpacePoints, bins=100, lower_bound=0, upper_bound=2000)
177  h_nSVDSpacePoints.contact = self.contact
178  h_nSVDSpacePoints.check = 'Average of 200 +/- 50 is expected.'
179  h_nSVDSpacePoints.description = 'Number of SVDSpacePoints'
180  h_nSVDSpacePoints.title = 'Number of SVDSpacePoints per event'
181  h_nSVDSpacePoints.xlabel = 'Number of SVDSpacePoints'
182  h_nSVDSpacePoints.ylabel = ''
183  h_nSVDSpacePoints.write(output_tfile)
184 
185  h_nCDCHits = ValidationPlot('h_nCDCHits')
186  h_nCDCHits.hist(self.nCDCHits, bins=100, lower_bound=0, upper_bound=6000)
187  h_nCDCHits.contact = self.contact
188  h_nCDCHits.check = 'Average of 3500 +/- 200 is expected.'
189  h_nCDCHits.description = 'Number of CDCHits'
190  h_nCDCHits.title = 'Number of CDCHits per event'
191  h_nCDCHits.xlabel = 'Number of CDCHits'
192  h_nCDCHits.ylabel = ''
193  h_nCDCHits.write(output_tfile)
194 
195  output_tfile.Close()
196 
197  """ Steering """
198  basf2.set_random_seed(1509)
199 
200  path = basf2.create_path()
201 
202  path.add_module('RootInput', inputFileName=INPUT_FILE)
203  path.add_module('Gearbox')
204  path.add_module('Geometry')
205  add_svd_reconstruction(path, isROIsimulation=False)
206  add_pxd_reconstruction(path)
207 
208  path.add_module(TrackingInputValidation())
209 
210  path.add_module('Progress')
211 
212  print(path)
213  basf2.process(path)
214  print(basf2.statistics)
215 
216 
217 if __name__ == '__main__':
218  if ACTIVE:
219  run()
220  else:
221  print("This validation deactivated and thus basf2 is not executed.\n"
222  "If you want to run this validation, please set the 'ACTIVE' flag above to 'True'.\n"
223  "Exiting.")
Belle2::PyStoreArray
A (simplified) python wrapper for StoreArray.
Definition: PyStoreArray.h:72