B2A401-KFit-MassFit.py

#!/usr/bin/env python3
 
 
 
import basf2 as b2
from modularAnalysis import inputMdst
from modularAnalysis import reconstructDecay
from modularAnalysis import matchMCTruth
from vertex import kFit
from modularAnalysis import variablesToNtuple
from stdPi0s import stdPi0s
import variables.collections as vc
import variables.utils as vu
 
# create path
my_path = b2.create_path()
 
# load input ROOT file
inputMdst(environmentType='default',
          filename=b2.find_file('B02pi0D0_D2kpi_B2Dstarpi_Dstar2Dpi_D2kpi.root', 'examples', False),
          path=my_path)
 
# use standard final state particle lists
#
# creates "pi0:eff40_Jan2020Fit" ParticleList
stdPi0s('eff40_Jan2020Fit', path=my_path)
 
# reconstruct D0 -> pi0 pi0 decay
# keep only candidates with 1.7 < M(pi0pi0) < 2.0 GeV
reconstructDecay('D0:pi0pi0 -> pi0:eff40_Jan2020Fit pi0:eff40_Jan2020Fit', '1.7 < M < 2.0', path=my_path)
 
# perform mass fit using KFit
# Reject the candidates with failed fit.
kFit('D0:pi0pi0', 0.0, 'mass', path=my_path)
 
# reconstruct B0 -> D0 pi0 decay
# keep only candidates with Mbc > 5.24 GeV
# and -1 < Delta E < 1 GeV
reconstructDecay('B0:all -> D0:pi0pi0 pi0:eff40_Jan2020Fit', '5.24 < Mbc < 5.29 and abs(deltaE) < 1.0', path=my_path)
 
# perform MC matching (MC truth association)
matchMCTruth('B0:all', path=my_path)
 
# Select variables that we want to store to ntuple
B0_vars = vc.inv_mass + vc.mc_truth + \
    vu.create_aliases_for_selected(
        vc.inv_mass + vc.mc_truth,
        'B0 -> ^D0 ^pi0') + \
    vu.create_aliases_for_selected(
        vc.cluster, 'B0 -> D0 [pi0 -> ^gamma ^gamma]')
 
pi0_vars = vc.mc_truth + vc.kinematics + \
    ['extraInfo(BDT)', 'decayAngle(0)'] + \
    vu.create_aliases_for_selected(
        vc.cluster + vc.kinematics, 'pi0 -> ^gamma ^gamma')
 
# Saving variables to ntuple
output_file = 'B2A401-KFit-MassFit.root'
variablesToNtuple('B0:all', B0_vars,
                  filename=output_file, treename='b0', path=my_path)
variablesToNtuple('pi0:eff40_Jan2020Fit', pi0_vars,
                  filename=output_file, treename='pi0', path=my_path)
 
# Process the events
b2.process(my_path)
 
# print out the summary
print(b2.statistics)