B2A406-Rave-DecayStringVertexFit.py

#!/usr/bin/env python3
 
 
 
 
 
import basf2 as b2
from modularAnalysis import inputMdst
from modularAnalysis import reconstructDecay
from modularAnalysis import matchMCTruth
from vertex import raveFit
from stdCharged import stdPi, stdK
from modularAnalysis import variablesToNtuple
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 "pi+:all" ParticleList (and c.c.)
stdPi('all', path=my_path)
# creates "pi+:loose" ParticleList (and c.c.)
stdPi('loose', path=my_path)
# creates "K+:loose" ParticleList (and c.c.)
stdK('loose', path=my_path)
 
# reconstruct D0 -> K- pi+ decay
# keep only candidates with 1.8 < M(Kpi) < 1.9 GeV
reconstructDecay('D0:kpi -> K-:loose pi+:loose', '1.8 < M < 1.9', path=my_path)
reconstructDecay('D0:st -> K-:loose pi+:loose', '1.8 < M < 1.9', path=my_path)
reconstructDecay('D0:du -> K-:loose pi+:loose', '1.8 < M < 1.9', path=my_path)
 
# perform D0 vertex fit
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D0:kpi', 0.0, fit_type='massvertex', path=my_path)
 
# perform D0 single track fit (production vertex)
# D0 vertex and covariance matrix must be defined
raveFit('D0:st', 0.0, path=my_path)
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D0:st', 0.0, decay_string='^D0 -> K- pi+', constraint='ipprofile', path=my_path)
 
# perform D0 vertex fit updating daughters
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D0:du', 0.0, daughtersUpdate=True, path=my_path)
 
# reconstruct 3 times the D*+ -> D0 pi+ decay
# keep only candidates with Q = M(D0pi) - M(D0) - M(pi) < 20 MeV
# and D* CMS momentum > 2.5 GeV
reconstructDecay('D*+:1 -> D0:kpi pi+:all',
                 '0.0 <= Q < 0.02 and 2.5 < useCMSFrame(p) < 5.5', path=my_path)
reconstructDecay('D*+:2 -> D0:kpi pi+:all',
                 '0.0 <= Q < 0.02 and 2.5 < useCMSFrame(p) < 5.5', path=my_path)
reconstructDecay('D*+:3 -> D0:kpi pi+:all',
                 '0.0 <= Q < 0.02 and 2.5 < useCMSFrame(p) < 5.5', path=my_path)
 
# perform MC matching (MC truth association)
matchMCTruth('D*+:1', path=my_path)
matchMCTruth('D*+:2', path=my_path)
matchMCTruth('D*+:3', path=my_path)
 
# perform D*+ kinematic vertex fit using the D0 and the pi+
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D*+:1', 0.0, path=my_path)
 
# perform D*+ kinematic beam spot constrained vertex fit using the D0 and the pi+
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D*+:2', 0.0, constraint='ipprofile', path=my_path)
 
# perform D*+ kinematic beam spot constrained vertex fit using only the pi+
# keep candidates only passing C.L. value of the fit > 0.0 (no cut)
raveFit('D*+:3', 0.0, decay_string='D*+ -> D0 ^pi+', constraint='ipprofile', path=my_path)
 
# Select variables that we want to store to ntuple
dstar_vars = vc.inv_mass + vc.mc_truth + \
    vc.mc_flight_info + vc.flight_info + vc.vertex
 
fs_hadron_vars = vu.create_aliases_for_selected(
    vc.pid + vc.track + vc.mc_truth,
    'D*+ -> [D0 -> ^K- ^pi+] ^pi+')
 
d0_vars = vu.create_aliases_for_selected(
    vc.inv_mass + vc.mc_truth + vc.vertex,
    'D*+ -> ^D0 pi+', 'D0')
 
dstt = vc.kinematics + vc.vertex + vc.mc_vertex + vc.flight_info + \
    vu.create_aliases_for_selected(
        vc.kinematics,
        '^D0 -> ^K- ^pi+')
 
dstu = vc.kinematics + vu.create_aliases_for_selected(
    vc.kinematics,
    '^D0 -> ^K- ^pi+')
 
# Saving variables to ntuple
output_file = 'B2A406-Rave-DecayStringVertexFit.root'
variablesToNtuple('D*+:1', dstar_vars + d0_vars + fs_hadron_vars,
                  filename=output_file, treename='dsttree1', path=my_path)
variablesToNtuple('D*+:2', dstar_vars + d0_vars + fs_hadron_vars,
                  filename=output_file, treename='dsttree2', path=my_path)
variablesToNtuple('D*+:3', dstar_vars + d0_vars + fs_hadron_vars,
                  filename=output_file, treename='dsttree3', path=my_path)
variablesToNtuple('D0:st', dstt,
                  filename=output_file, treename='d0tree1', path=my_path)
variablesToNtuple('D0:du', dstu,
                  filename=output_file, treename='d0tree2', path=my_path)
 
 
# Process the events
b2.process(my_path)
 
# print out the summary
print(b2.statistics)