test0_Dz2BodyHadronic.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Descriptor: [D*+ -> [D0 -> K- pi+] pi+ ]
 
"""
<header>
  <output>./WG6_DstToD0pi_D0ToKpi.dst.root</output>
  <contact>lilongke@ihep.ac.cn</contact>
</header>
"""
 
import basf2 as b2
from simulation import add_simulation
from reconstruction import add_reconstruction, add_mdst_output
from ROOT import Belle2
 
b2.set_random_seed(12345)
 
# background (collision) files
# bg = glob.glob('./BG/[A-Z]*.root')
 
# create path
myMain = b2.create_path()
 
# specify number of events to be generated
eventinfosetter = b2.register_module('EventInfoSetter')
eventinfosetter.param('evtNumList', [1000])
eventinfosetter.param('runList', [0])
eventinfosetter.param('expList', [0])
myMain.add_module(eventinfosetter)
 
# Generator
# to run the framework the used modules need to be registered
evtgeninput = b2.register_module('EvtGenInput')
evtgeninput.param('ParentParticle', 'vpho')
evtgeninput.param('userDECFile', Belle2.FileSystem.findFile('skim/validation/WG6_validation.dec'))
evtgeninput.param('InclusiveParticle', 'D*+')
evtgeninput.param('InclusiveType', 2)
myMain.add_module(evtgeninput)
 
# detector simulation
# add_simulation(myMain, bkgfiles=bg)
add_simulation(myMain)
 
# reconstruction
add_reconstruction(myMain)
 
 
# Finally add mdst output
output_filename = "../WG6_DstToD0pi_D0ToKpi.dst.root"
add_mdst_output(myMain, filename=output_filename)
 
# process events and print call statistics
b2.process(myMain)
print(b2.statistics)