EvtReconstruction.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
<header>
  <input>EvtGenSim.root</input>
  <output>EvtRec.root,EvtRec_mdst.root</output>
  <cacheable/>
  <contact>Software team b2soft@mail.desy.de</contact>
  <description>
    This steering file runs the standard reconstruction on an input file with
    generic BBbar events.
  </description>
</header>
"""
 
from basf2 import set_random_seed, create_path, process, statistics
from reconstruction import add_reconstruction
from mdst import add_mdst_output
from validation import statistics_plots, event_timing_plot
 
set_random_seed(12345)
 
main = create_path()
 
# read file of simulated events
main.add_module("RootInput", inputFileName="../EvtGenSim.root")
 
# geometry parameter database
main.add_module("Gearbox")
 
# detector geometry
main.add_module("Geometry")
 
# reconstruction
add_reconstruction(main)
 
# memory profile
main.add_module("Profile")
 
# output
main.add_module("RootOutput", outputFileName="../EvtRec.root")
add_mdst_output(main, True, "../EvtRec_mdst.root")
 
main.add_module('Progress')
process(main)
 
# Print call statistics
print(statistics)
 
statistics_plots(
    "EvtRec_statistics.root",
    contact="Software team b2soft@mail.desy.de",
    job_desc="a standard reconstruction job with generic EvtGen events",
    prefix="EvtRec",
)
event_timing_plot(
    "../EvtRec.root",
    "EvtRec_statistics.root",
    contact="Software team b2soft@mail.desy.de",
    job_desc="a standard reconstruction job with generic EvtGen events",
    prefix="EvtRec",
)