16 from basf2
import set_log_level, register_module, process, LogLevel, \
17 set_random_seed, print_params, create_path, statistics
18 from simulation
import add_simulation
19 from reconstruction
import add_reconstruction
22 set_log_level(LogLevel.WARNING)
25 particlegun = register_module(
'ParticleGun')
41 particlegun.param(
'pdgCodes', [-11, 11])
46 particlegun.param(
'nTracks', 10)
59 particlegun.param(
'varyNTracks',
False)
102 particlegun.param(
'momentumGeneration',
'uniform')
103 particlegun.param(
'momentumParams', [0.05, 3])
126 particlegun.param(
'thetaGeneration',
'uniform')
127 particlegun.param(
'thetaParams', [17, 150])
154 particlegun.param(
'phiGeneration',
'uniform')
155 particlegun.param(
'phiParams', [0, 360])
165 particlegun.param(
'vertexGeneration',
'fixed')
166 particlegun.param(
'xVertexParams', [0])
167 particlegun.param(
'yVertexParams', [0])
168 particlegun.param(
'zVertexParams', [0])
190 particlegun.param(
'independentVertices',
False)
194 print_params(particlegun)
200 eventinfosetter = register_module(
'EventInfoSetter')
202 progress = register_module(
'Progress')
204 gearbox = register_module(
'Gearbox')
206 geometry = register_module(
'Geometry')
208 output = register_module(
'RootOutput')
212 eventinfosetter.param({
'evtNumList': [100],
'runList': [1]})
215 output.param(
'outputFileName',
'ParticleGunOutput.root')
219 main.add_module(eventinfosetter)
220 main.add_module(progress)
221 main.add_module(particlegun)
223 main.add_module(output)