B2A910-ApplyKsSelector.py

#!/usr/bin/env python3
 

 

 
import basf2 as b2
import modularAnalysis as ma
import variables.collections as vc
import stdV0s as stdV0s
import ksSelector as ksSelector
 
output_file = 'output.root'
 
# create path
my_path = b2.create_path()
 
# load input ROOT file
ma.inputMdst(filename=b2.find_file('B02JpsiKs_Jpsi2mumu_Ks2pipi.root', 'examples', False),
             path=my_path)
 
# load V0s and copy it
stdV0s.stdKshorts(path=my_path)
 
b2.conditions.prepend_globaltag(ma.getAnalysisGlobaltag())
 
# apply ksSelector to the K_S0 particle list
# list type can be 'all', 'standard', 'tight', or 'loose'
# If 'all', no cut is applied by KsSelector, and mva output become available via extraInfo.
# if other list type, cut is applied on the original particle list.
# For any other payload except MC16 payload, please use customized cut.
ksSelector.ksSelector(particleListName='K_S0:merged',
                      output_label_name='custom',  # use 'standard', 'tight', or 'loose' for recommend cut of latest payload
                      extraInfoName_V0Selector='V0_mva',
                      extraInfoName_LambdaVeto='Lam_mva',
                      useCustomThreshold=True,  # if you want to use customized threshold, set it true
                      threshold_V0Selector=0.91,  # set customized v0 cut
                      threshold_LambdaVeto=0.19,  # set customized lambda_veto cut
                      identifier_Ks="Ks_LGBM_V0Selector",  # use Ks_LGBM_V0Selector_MC16 for MC 16
                      identifier_vLambda="Ks_LGBM_LambdaVeto",  # use Ks_LGBM_LambdaVeto_MC16 for MC 16
                      path=my_path)
 
 
ma.matchMCTruth(list_name='K_S0:custom', path=my_path)
 
# set variables
vars = vc.kinematics + vc.mc_kinematics + vc.mc_truth
vars += ['extraInfo(V0_mva)']
vars += ['extraInfo(Lam_mva)']
 
# output
ma.variablesToNtuple('K_S0:custom',
                     variables=vars,
                     filename=output_file,
                     treename='tree',
                     path=my_path)
 
b2.process(my_path)