exampleOfRetentionCheck.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Example showing the use of the RetentionCheck module
 
To run the example, simply do
 
    basf2 exampleOfRetentionCheck.py -n 100
"""
 
__authors__ = "Cyrille Praz"
 
import basf2 as b2
import modularAnalysis as ma
from skimExpertFunctions import get_test_file
from skimRetentionCheck import RetentionCheck, pathWithRetentionCheck
from skim.fei import runFEIforB0SL, B0SL
 
# Create a basf2 path.
path = b2.create_path()
 
# Input a test mdst file.
ma.inputMdstList('default', get_test_file("MC12_mixedBGx1"), path=path)
 
# Run pre-selection cuts and FEI.
runFEIforB0SL(path)
# Apply *arbitrary* cuts on the B tag list.
ma.applyCuts('B0:semileptonic', 'sigProb>0.0001', path=path)
ma.applyCuts('B0:semileptonic', '-4.0<cosThetaBY<3.0', path=path)
ma.applyCuts('B0:semileptonic', 'dmID<8', path=path)
ma.applyCuts('B0:semileptonic', '[[dmID<4 and d1_p_CMSframe>1.0] or [dmID>=4 and d2_p_CMSframe>1.0]]', path=path)
 
# Add the retention tracker.
path = pathWithRetentionCheck(['B0:semileptonic'], path)
 
# Process the basf2 path.
b2.process(path)
 
# Print out the retention tracker results and create an associated plot.
RetentionCheck.print_results()
RetentionCheck.plot_retention('B0:semileptonic', save_as='retention_rate.pdf')