Belle II Software  release-08-00-04
TRGValidationGen.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
11 
12 """
13 <header>
14 <output>TRGValidationGen.root</output>
15 <contact>Yun-Tsung Lai, ytlai@post.kek.jp</contact>
16 <description>This steering file generates 1000 e+/e- mu+/mu- particle guns to validate the trg package.</description>
17 </header>
18 """
19 
20 import basf2 as b2
21 from simulation import add_simulation
22 import reconstruction as re
23 import mdst
24 
25 
26 main = b2.create_path()
27 eventinfosetter = b2.register_module('EventInfoSetter')
28 eventinfosetter.param({'evtNumList': [5000], 'runList': [1]})
29 main.add_module(eventinfosetter)
30 
31 particlegun = b2.register_module('ParticleGun')
32 particlegun.param('pdgCodes', [11, -11, 13, -13])
33 particlegun.param('nTracks', 1)
34 particlegun.param('momentumGeneration', 'uniformPt')
35 particlegun.param('momentumParams', [1.0, 3.0])
36 particlegun.param('thetaParams', [35, 127])
37 particlegun.param('phiGeneration', 'uniform')
38 particlegun.param('phiParams', [0, 360])
39 particlegun.param('vertexGeneration', 'uniform')
40 particlegun.param('xVertexParams', [0, 0])
41 particlegun.param('yVertexParams', [0, 0])
42 particlegun.param('zVertexParams', [-20.0, 20.0])
43 main.add_module(particlegun)
44 
45 
46 # trigger simulation is included in latest basf2
47 add_simulation(main)
48 
49 
50 re.add_reconstruction(main)
51 
52 # output
53 mdst.add_mdst_output(
54  path=main,
55  mc=True,
56  filename='../TRGValidationGen.root',
57  additionalBranches=[
58  "TRGKLMHits",
59  "TRGKLMTracks",
60  "KLMTrgSummary",
61  "TRGCDC2DFinderTracks",
62  "TRGCDC3DFitterTracks",
63  "TRGCDCNeuroTracks",
64  "TRGECLClusters",
65  "TRGSummary",
66  "MCInitialParticles"])
67 
68 
69 main.add_module('Progress')
70 # main
71 b2.process(main)
72 print(b2.statistics)
73 # ===<END>
mdst.add_mdst_output
def add_mdst_output(path, mc=True, filename='mdst.root', additionalBranches=[], dataDescription=None)
Definition: mdst.py:37