select_events.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
# Demonstrates selecting events to save in the event display:
# * create 'conditionspath' with e.g. RootOutput
# * use if_true() on the display module to activate it
#
# Once started, a widget will be shown to toggle the return value,
# if the box is checked when advancing to the next event, the event
# will be saved in saved_from_display.root.
 
from basf2 import *
from simulation import add_simulation
from reconstruction import add_reconstruction
 
# register necessary modules
eventinfosetter = register_module('EventInfoSetter')
eventinfosetter.param('evtNumList', [5000])
 
main = Path()
 
# add modules to paths
main.add_module(eventinfosetter)
 
main.add_module('EvtGenInput')
 
# (no ECL because of unsupported solids)
components = [
    'MagneticField',
    'BeamPipe',
    'PXD',
    'SVD',
    'CDC',
    'TOP',
    'ARICH',
    'EKLM',
    'BKLM',
]
add_simulation(main, components)
 
# add_reconstruction(main, components)
 
display = main.add_module('Display')
conditionpath = Path()
conditionpath.add_module('RootOutput', outputFileName='saved_from_display.root')
display.if_true(conditionpath)
 
process(main)
print(statistics)