inclusiveDstarReconstruction.py

# Doxygen should skip this script
# @cond
 
"""
An example script showing how to reconstruct D* mesons inclusively from slow
pions.
"""
 
import basf2
import modularAnalysis as mA
import stdPi0s
 
from ROOT import Belle2
 
 
main_path = basf2.create_path()
 
mA.inputMdstList(
    environmentType="default",
    filelist=[Belle2.FileSystem.findFile("analysis/tests/mdst.root")],
    path=main_path,
)
 
list_tree_tuples = list()
 
# MC Truth
mA.fillParticleListFromMC("pi-:from_mc", cut="[dr < 2] and [abs(dz) < 4]", addDaughters=True, path=main_path)
mA.addInclusiveDstarReconstruction("pi-:from_mc", "D*-:Dstcharged_slowPicharged_MC", "[useCMSFrame(p) < 0.2]", main_path)
 
mA.fillParticleListFromMC("pi0:from_mc", "", addDaughters=True, path=main_path)
mA.addInclusiveDstarReconstruction("pi0:from_mc", "D*-:Dstcharged_slowPizero_MC", "[useCMSFrame(p) < 0.2]", main_path)
mA.addInclusiveDstarReconstruction("pi0:from_mc", "D*0:Dstzero_slowPizero_MC", "[useCMSFrame(p) < 0.2]", main_path)
 
list_tree_tuples.append(("D*-:Dstcharged_slowPicharged_MC", "Dstcharged_slowPicharged_MC"))
list_tree_tuples.append(("D*-:Dstcharged_slowPizero_MC", "Dstcharged_slowPizero_MC"))
list_tree_tuples.append(("D*0:Dstzero_slowPizero_MC", "Dstzero_slowPizero_MC"))
 
# Reco Level
mA.fillParticleList("pi-:slow", cut="[dr < 2] and [abs(dz) < 4]", path=main_path)
mA.matchMCTruth("pi-:slow", path=main_path)
mA.addInclusiveDstarReconstruction("pi-:slow", "D*-:Dstcharged_slowPicharged", "[useCMSFrame(p) < 0.2]", main_path)
mA.matchMCTruth("D*-:Dstcharged_slowPicharged", main_path)
 
stdPi0s.stdPi0s("eff60_Jan2020", main_path)
mA.matchMCTruth("pi0:eff60_Jan2020", main_path)
mA.addInclusiveDstarReconstruction("pi0:eff60_Jan2020", "D*-:Dstcharged_slowPizero", "[useCMSFrame(p) < 0.2]", main_path)
mA.matchMCTruth("D*-:Dstcharged_slowPizero", main_path)
mA.addInclusiveDstarReconstruction("pi0:eff60_Jan2020", "D*0:Dstzero_slowPizero", "[useCMSFrame(p) < 0.2]", main_path)
mA.matchMCTruth("D*0:Dstzero_slowPizero", main_path)
 
list_tree_tuples.append(("D*-:Dstcharged_slowPicharged", "Dstcharged_slowPicharged"))
list_tree_tuples.append(("D*-:Dstcharged_slowPizero", "Dstcharged_slowPizero"))
list_tree_tuples.append(("D*0:Dstzero_slowPizero", "Dstzero_slowPizero"))
 
dstar_variables = [
    "E",
    "p",
    "pt",
    "useCMSFrame(p)",
    "useCMSFrame(E)",
    "matchedMC(p)",
    "matchedMC(pt)",
    "matchedMC(E)",
    "matchedMC(useCMSFrame(p))",
    "matchedMC(PDG)",
    "px",
    "py",
    "pz",
    "M",
    "InvM",
    "daughter(0, isSignal)",
    "daughter(0, E)",
    "daughter(0, p)",
    "daughter(0, useCMSFrame(p))",
    "daughter(0, matchedMC(p))",
    "daughter(0, matchedMC(E))",
    "daughter(0, matchedMC(useCMSFrame(p)))",
    "daughter(0, matchedMC(PDG))",
    "daughter(0, daughter(0, matchedMC(PDG)))",
    "daughter(0, daughter(1, matchedMC(PDG)))",
    "daughter(0, mcMother(PDG))",
    "daughter(0, mcMother(p))",
    "daughter(0, mcMother(useCMSFrame(p)))",
    "daughter(0, mcMother(E))",
    "daughter(0, mcMother(mcDaughter(0, PDG)))",
    "daughter(0, mcMother(mcDaughter(0, p)))",
    "daughter(0, mcMother(mcDaughter(0, E)))",
    "daughter(0, mcMother(mcDaughter(0, useCMSFrame(p))))",
    "daughter(0, mcMother(mcDaughter(1, PDG)))",
    "daughter(0, mcMother(mcDaughter(1, p)))",
    "daughter(0, mcMother(mcDaughter(1, E)))",
    "daughter(0, mcMother(mcDaughter(1, useCMSFrame(p))))",
    "daughter(0, mcMother(mcMother(mcDaughter(0, PDG))))",
    "daughter(0, mcMother(mcMother(mcDaughter(1, PDG))))",
]
 
for list_tree_tuple in list_tree_tuples:
    mA.variablesToNtuple(
        decayString=list_tree_tuple[0],
        variables=dstar_variables,
        treename=list_tree_tuple[1],
        path=main_path
    )
 
basf2.process(main_path)
print(basf2.statistics)
 
# @endcond