evtgen_rec.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
import basf2 as b2
from optparse import OptionParser
import os
# --------------------------------------------------------------------
# Example of using ARICH reconstruction
# needs reconstructed tracks (Tracks), extrapolated to ARICH (ExtHits)
# (generated by evtgen_sim.py)
# Produces root file with flat ntuple from ARICHNtuple module
# --------------------------------------------------------------------
 
parser = OptionParser()
parser.add_option('-f', '--file', dest='filename',
                  default='ARICHRec.root')
(options, args) = parser.parse_args()
 
home = os.environ['BELLE2_LOCAL_DIR']
 
# Suppress messages and warnings during processing:
b2.set_log_level(b2.LogLevel.ERROR)
 
# Create path
main = b2.create_path()
 
# input root file
input_module = b2.register_module('RootInput')
input_module.param('inputFileName', home + "/ARICHEvents.root")
main.add_module(input_module)
 
# Histogram manager immediately after master module
histo = b2.register_module('HistoManager')
histo.param('histoFileName', 'DQMhistograms.root')  # File to save histograms
main.add_module(histo)
 
# Gearbox: access to database (xml files)
gearbox = b2.register_module('Gearbox')
main.add_module(gearbox)
 
# Geometry
geometry = b2.register_module('Geometry')
geometry.param('components', ['ARICH'])
main.add_module(geometry)
 
# ARICH digitizer
arichDigi = b2.register_module('ARICHDigitizer')
main.add_module(arichDigi)
 
# convert ARICHDigits to ARICHHits
arichHits = b2.register_module('ARICHFillHits')
main.add_module(arichHits)
 
# ARICH reconstruction
# calculate PID likelihoods for all tracks
arichreco = b2.register_module('ARICHReconstructor')
main.add_module(arichreco)
 
# ARICH Ntuple
# create flat ntuple for performance analysis
arichNtuple = b2.register_module('ARICHNtuple')
arichNtuple.param('outputFile', options.filename)
main.add_module(arichNtuple)
 
# ARICH DQM
# create DQM occupancy plots
arichdqm = b2.register_module('ARICHDQM')
main.add_module(arichdqm)
 
# Uncomment to store DataStore content to root file
# output = register_module('RootOutput')
# output.param('outputFileName', 'DataStore.root')
# main.add_module(output)
 
# Show progress of processing
progress = b2.register_module('Progress')
main.add_module(progress)
 
# Process events
b2.process(main)
 
# Print call statistics
print(b2.statistics)