B2A420-OrcaKinFit.py

#!/usr/bin/env python3
 
 
 
 
 
import basf2 as b2
from modularAnalysis import inputMdst
from modularAnalysis import fillParticleList
from modularAnalysis import reconstructDecay
from modularAnalysis import matchMCTruth
from kinfit import fitKinematic4C
from modularAnalysis import variablesToNtuple
import variables.collections as vc
import variables.utils as vu
 
# create path
mypath = b2.create_path()
 
b2.set_log_level(b2.LogLevel.WARNING)
 
# Input file(s).
inputMdst(environmentType='default',
          filename=b2.find_file('B2A424-SimulateMuonPairs.root', 'examples', False),
          path=mypath)
 
# use standard final state particle lists for muons
fillParticleList('mu-:z0', 'chiProb > 0.001 and p > 1.0', path=mypath)
fillParticleList('mu-:z0fit', 'chiProb > 0.001 and p > 1.0', path=mypath)
 
# reconstruct Z -> mu+ mu-
reconstructDecay('Z0:mm_rec -> mu+:z0 mu-:z0', '9.0 < M < 11.0', path=mypath)
reconstructDecay('Z0:mm_kinfit -> mu+:z0fit mu-:z0fit', '9.0 < M < 11.0', path=mypath)
 
# MC truth matching
matchMCTruth('Z0:mm_rec', path=mypath)
matchMCTruth('Z0:mm_kinfit', path=mypath)
 
# kinematic 4C hard fit
fitKinematic4C('Z0:mm_kinfit', path=mypath)
 
# Select variables that we want to store to ntuple
muvars = vc.kinematics + vc.mc_truth + vc.mc_kinematics + vc.momentum_uncertainty
z0vars = vc.inv_mass + vc.kinematics + vc.mc_kinematics + vc.mc_truth + \
    vu.create_aliases_for_selected(muvars, 'Z0 -> ^mu+ ^mu-')
 
z0uvars = z0vars + \
    vu.create_aliases(['OrcaKinFitProb', 'OrcaKinFitChi2', 'OrcaKinFitErrorCode'], 'extraInfo({variable})', "")
 
 
# Saving variables to ntuple
output_file = 'B2A420-OrcaKinFit.root'
variablesToNtuple('Z0:mm_rec', z0vars,
                  filename=output_file, treename='Z0_mm_rec', path=mypath)
variablesToNtuple('Z0:mm_kinfit', z0uvars,
                  filename=output_file, treename='Z0_mm_kinfit', path=mypath)
 
 
# Process the events
b2.process(mypath)
 
# print out the summary
print(b2.statistics)