Belle II Software  release-05-01-25
clean_bg_file.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 # remove empty events from beam background data files.
5 
6 import os
7 import random
8 from basf2 import *
9 from ROOT import Belle2
10 
11 
12 class SVDTrigger(Module):
13 
14  """Returns 1 if current event contains at least one SVDSimHit or one
15  SVD-related BeamBackHit, 0 otherwise"""
16 
17  def initialize(self):
18  """reimplementation of Module::initialize()."""
19 
20  def event(self):
21  """reimplementation of Module::event()."""
22 
23  self.return_value(0)
24  simhits = Belle2.PyStoreArray('SVDSimHits')
25  bghits = Belle2.PyStoreArray('BeamBackHits')
26  if len(simhits) > 0 or len(bghits) > 0:
27  self.return_value(1)
28 
29 
30 main = create_path()
31 
32 input = register_module('RootInput')
33 main.add_module(input)
34 
35 
37 trigger = SVDTrigger()
38 main.add_module(trigger)
39 
40 # if SVDTrigger returns 0, we'll jump into an empty path
41 # (skipping output)
42 emptypath = create_path()
43 trigger.if_false(emptypath)
44 
45 
46 output = register_module('RootOutput')
47 main.add_module(output)
48 
49 main.add_module(register_module('ProgressBar'))
50 
51 process(main)
52 
53 print(statistics)
clean_bg_file.SVDTrigger
Definition: clean_bg_file.py:12
clean_bg_file.SVDTrigger.initialize
def initialize(self)
Definition: clean_bg_file.py:17
Belle2::PyStoreArray
a (simplified) python wrapper for StoreArray.
Definition: PyStoreArray.h:58
clean_bg_file.SVDTrigger.event
def event(self)
Definition: clean_bg_file.py:20