event_kinematics_variables.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
a test of the EventKinematics variables using the small test file for running quickly
"""
 
import b2test_utils
from basf2 import set_random_seed, create_path, process
 
# make logging more reproducible by replacing some strings
b2test_utils.configure_logging_for_tests()
set_random_seed("1337")
testinput = [b2test_utils.require_file('analysis/tests/mdst.root')]
 
fsps = ['pi-:all', 'gamma:all']
 
testpath = create_path()
testpath.add_module('RootInput', inputFileNames=testinput)
# There is an annoying warning in ParticleLoader about zero charged tracks
# If it will be removed in the future releases, the test log should be
# updated respectively.
testpath.add_module('ParticleLoader', decayStrings=fsps)
testpath.add_module('ParticleSelector', decayString='gamma:all', cut='isFromECL')
 
# Variables created by event kinematics module
event_kinematics = [
    "missingMomentumOfEvent",
    "missingMomentumOfEvent_Px",
    "missingMomentumOfEvent_Py",
    "missingMomentumOfEvent_Pz",
    "missingMomentumOfEvent_theta",
    "missingMomentumOfEventCMS",
    "missingMomentumOfEventCMS_Px",
    "missingMomentumOfEventCMS_Py",
    "missingMomentumOfEventCMS_Pz",
    "missingMomentumOfEventCMS_theta",
    "missingEnergyOfEventCMS",
    "missingMass2OfEvent",
    "visibleEnergyOfEventCMS",
    "totalPhotonsEnergyOfEvent"
]
 
testpath.add_module('EventKinematics', particleLists=fsps)
# Print the variables to log
testpath.add_module('ParticlePrinter', listName='', fullPrint=False,
                    variables=event_kinematics)
process(testpath, 1)