Belle II Software  release-05-01-25
KLMMuonGenerate.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
13 
14 """
15 <header>
16  <output>muon-KLMValidation.root</output>
17  <contact>martina.laurenza@roma3.infn.it</contact>
18  <description>Create events with 1 muon track for BKLM and EKLM validation.</description>
19 </header>
20 """
21 
22 
23 import basf2
24 import simulation as sim
25 import reconstruction as rec
26 
27 basf2.set_random_seed(981543)
28 basf2.set_log_level(basf2.LogLevel.ERROR)
29 
30 output_filename = '../muon-KLMValidation.root'
31 basf2.B2INFO('The output file name is ' + output_filename)
32 
33 main_path = basf2.create_path()
34 
35 main_path.add_module('EventInfoSetter', evtNumList=1000)
36 
37 main_path.add_module('Progress')
38 main_path.add_module('ProgressBar')
39 
40 main_path.add_module('ParticleGun',
41  pdgCodes=[-13, 13],
42  nTracks=1,
43  varyNTracks=0,
44  momentumGeneration='uniform',
45  momentumParams=[0.5, 5.0],
46  thetaGeneration='uniformCos',
47  thetaParams=[18., 155.],
48  phiGeneration='uniform',
49  phiParams=[0., 360.],
50  vertexGeneration='fixed',
51  xVertexParams=[0.0],
52  yVertexParams=[0.0],
53  zVertexParams=[0.0])
54 
55 sim.add_simulation(path=main_path)
56 rec.add_reconstruction(path=main_path)
57 
58 main_path.add_module('RootOutput',
59  outputFileName=output_filename,
60  branchNames=['MCParticles', 'KLMMuidLikelihoods', 'KLMDigits', 'BKLMHit2ds', 'EKLMHit2ds'])
61 
62 basf2.process(main_path)
63 print(basf2.statistics)
basf2.process
def process(path, max_event=0)
Definition: __init__.py:25