import basf2 as b2
import modularAnalysis as ma

main = b2.Path()

ma.inputMdst('/home/belle2/fmeier/BelleIISummerSchool2022/MC15ri_b_ccbar.mdst.root', path=main)

ma.fillParticleList("K+:all", "", path=main)
ma.fillParticleList("pi+:all", "", path=main)

ma.reconstructDecay("D0 -> K-:all pi+:all", "", path=main)

ma.matchMCTruth("D0", path=main)

ma.variablesToNtuple("D0", variables=['InvM', 'isSignal'], path=main)

b2.process(main)