evtgen_sim.py

#!/usr/bin/env python3
 
 
 
import basf2 as b2
from optparse import OptionParser
from tracking import add_tracking_reconstruction
from modularAnalysis import inputMdst
import os
# --------------------------------------------------------------------
# Performs Geant4 simulation of events generated with evtgen_gen.py
# PXD, SVD, CDC and ARICH detectors are used. Reconstruction of tracks
# is also included and objects needed for ARICH reconstruction are stored
# in output root file.
# --------------------------------------------------------------------
 
parser = OptionParser()
parser.add_option('-f', '--file', dest='filename',
                  default='ARICHEvents.root')
(options, args) = parser.parse_args()
 
home = os.environ['BELLE2_LOCAL_DIR']
 
mypath = b2.create_path()
 
# Suppress messages and warnings during processing:
b2.set_log_level(b2.LogLevel.ERROR)
 
# load input ROOT file
inputMdst(home + '/B2Kpi_events.root', path=mypath)
 
# Gearbox: access to database (xml files)
mypath.add_module('Gearbox')
 
# Geometry
geometry = b2.register_module('Geometry')
geometry.param('components', [
    'MagneticField',
    'BeamPipe',
    'PXD',
    'SVD',
    'CDC',
    'ARICH'])
mypath.add_module(geometry)
 
# Simulation
mypath.add_module('FullSim')
 
# PXD digitization & clustering
mypath.add_module('PXDDigitizer')
mypath.add_module('PXDClusterizer')
 
# SVD digitization & clustering
mypath.add_module('SVDEventInfoSetter')
mypath.add_module('SVDDigitizer')
mypath.add_module('SVDClusterizer')
 
# CDC digitization
mypath.add_module('CDCDigitizer')
 
# tracking reconstruction
add_tracking_reconstruction(mypath)
 
# Track extrapolation
mypath.add_module('Ext')
 
# This creates relations between ExtHits (track points on aerogel plane, from
# extrapolated CDC tracks) and ARICHAeroHits (MC hits on aerogel plane).
# It allows to have relevant MC information
# without storing full MCParticles (which are LARGE) into output root file.
mypath.add_module('ARICHRelate')
 
# store branches needed for ARICH reconstruction in root file
output = b2.register_module('RootOutput')
output.param('outputFileName', options.filename)
output.param('branchNames', ['ARICHAeroHits', 'ARICHSimHits', 'ExtHits',
                             'ARICHAeroHitsToExtHits', 'Tracks', 'TrackFitResults'])
mypath.add_module(output)
 
# Show progress of processing
mypath.add_module('Progress')
 
# Process events
b2.process(mypath)
 
# Print call statistics
print(b2.statistics)