B2A501-BremsCorrection.py

#!/usr/bin/env python3
 

 

 
import basf2 as b2
import modularAnalysis as ma
 
# create path
my_path = b2.create_path()
 
# load input ROOT file
ma.inputMdst(filename=b2.find_file('JPsi2ee_e2egamma.root', 'examples', False),
             path=my_path)
 
 
# fill particleLists
ma.fillParticleList(decayString='e+:uncorrected',
                    cut='electronID > 0.2 and dr < 2 and abs(dz) < 4',
                    path=my_path)
ma.fillParticleList(decayString='gamma:bremsinput',
                    cut='E < 1.0',
                    path=my_path)
 
# MC matching
ma.matchMCTruth(list_name='e+:uncorrected', path=my_path)
ma.matchMCTruth(list_name='gamma:bremsinput', path=my_path)
 
# correction of Bremsstrahlung
# A new lepton is generated, with the old electron and, if found, a gamma as daughters.
# Note: the photon selection and the module parameters used here are kept simple on purpose.
# For the current recommendation on the photon selection and the module parameters, use the
# b2help-recommendation tool (or https://belle2.pages.desy.de/performance/recommendations/).
ma.correctBrems(outputList='e+:corrected',
                inputList='e+:uncorrected',
                gammaList='gamma:bremsinput',
                path=my_path)
ma.matchMCTruth(list_name='e+:corrected',
                path=my_path)
 
# uncorrected
ma.reconstructDecay(decayString='J/psi:uncorrected -> e+:uncorrected e-:uncorrected',
                    cut='',
                    path=my_path)
ma.reconstructDecay(decayString='J/psi:corrected -> e+:corrected e-:corrected',
                    cut='',
                    path=my_path)
 
# MC matching
ma.matchMCTruth(list_name='J/psi:uncorrected', path=my_path)
ma.matchMCTruth(list_name='J/psi:corrected', path=my_path)
 
# get all MC particles
ma.fillParticleListFromMC(decayString='J/psi:MC', cut="", path=my_path)
 
# write out ntuples
 
var0 = ['p',
        'px',
        'py',
        'pz',
        'x',
        'y',
        'z',
        'electronID',
        'PDG',
        'mcPDG',
        'E',
        'mcPX',
        'mcPY',
        'mcPZ',
        'mcE',
        'pxErr',
        'pyErr',
        'pzErr',
        'pErr',
        'isSignal',
        'mcErrors',
        'extraInfo(bremsCorrected)'
        ]
var1 = ['M',
        'p',
        'E',
        'x', 'y', 'z',
        'isSignal',
        'mcErrors',
        'daughter(0, p)',
        'daughter(1, p)',
        'daughter(0, extraInfo(bremsCorrected))',
        'daughter(1, extraInfo(bremsCorrected))'
        ]
 
ma.variablesToNtuple(decayString='e+:uncorrected',
                     variables=var0,
                     treename='e_uncorrected',
                     path=my_path)
ma.variablesToNtuple(decayString='e+:corrected',
                     variables=var0,
                     treename='e_corrected',
                     path=my_path)
ma.variablesToNtuple(decayString='J/psi:uncorrected',
                     variables=var1,
                     treename='Jpsi_uncorrected',
                     path=my_path)
ma.variablesToNtuple(decayString='J/psi:corrected',
                     variables=var1,
                     treename='Jpsi_corrected',
                     path=my_path)
ma.variablesToNtuple(decayString='J/psi:MC',
                     variables=var1,
                     treename='Jpsi_MC',
                     path=my_path)
 
# process the events
b2.process(my_path, calculateStatistics=True)