Belle II Software  release-05-01-25
TREPSGenerationOnlypipi.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
13 
14 import basf2
15 from basf2 import set_log_level, LogLevel, process, statistics
16 from ROOT import Belle2
17 from beamparameters import add_beamparameters
18 from generators import add_treps_generator
19 
20 set_log_level(LogLevel.DEBUG)
21 
22 # main path
23 main = basf2.create_path()
24 
25 # event info setter
26 main.add_module("EventInfoSetter", expList=0, runList=1, evtNumList=100000)
27 main.add_module("Progress")
28 
29 # beam parameters
30 beamparameters = add_beamparameters(main, "Y4S")
31 
32 # add TREPS generator
33 add_treps_generator(main, "e+e-pi+pi-", useDiscreteAndSortedW=False)
34 
35 main.add_module("RootOutput", outputFileName="utrepsbpipi_100k.root")
36 main.add_module("PrintMCParticles", logLevel=basf2.LogLevel.INFO, onlyPrimaries=False)
37 
38 # generate events
39 process(main)
40 
41 # show call statistics
42 print(statistics)