Belle II Software development
TRGValidationGen.py
1#!/usr/bin/env python3
2
3
10
11"""
12<header>
13<output>TRGValidationGen.root</output>
14<contact>Yun-Tsung Lai, ytlai@post.kek.jp</contact>
15<description>This steering file generates 5000 e+/e- mu+/mu- particle guns to validate the trg package.</description>
16</header>
17"""
18
19import basf2 as b2
20from simulation import add_simulation
21import reconstruction as re
22import mdst
23
24b2.set_random_seed(1337)
25
26main = b2.create_path()
27eventinfosetter = b2.register_module('EventInfoSetter')
28eventinfosetter.param({'evtNumList': [5000], 'runList': [1]})
29main.add_module(eventinfosetter)
30
31particlegun = b2.register_module('ParticleGun')
32particlegun.param('pdgCodes', [11, -11, 13, -13])
33particlegun.param('nTracks', 1)
34particlegun.param('momentumGeneration', 'uniformPt')
35particlegun.param('momentumParams', [0.5, 3.0])
36particlegun.param('thetaParams', [35, 127])
37particlegun.param('phiGeneration', 'uniform')
38particlegun.param('phiParams', [0, 360])
39particlegun.param('vertexGeneration', 'uniform')
40particlegun.param('xVertexParams', [0, 0])
41particlegun.param('yVertexParams', [0, 0])
42particlegun.param('zVertexParams', [-20.0, 20.0])
43main.add_module(particlegun)
44
45
46# trigger simulation is included in latest basf2
47add_simulation(main)
48
49
50re.add_reconstruction(main)
51
52# 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
69main.add_module('Progress')
70# main
71b2.process(main)
72# ===<END>
add_mdst_output(path, mc=True, filename='mdst.root', additionalBranches=[], dataDescription=None)
Definition mdst.py:38