MuonGenSimNoBkg.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
<header>
  <output>MuonGenSimNoBkg.root</output>
  <contact>Software team b2soft@mail.desy.de</contact>
  <cacheable/>
  <description>This steering file produces 10000 events with one Muon each
  and runs the detector simulation without mixing in background.</description>
</header>
"""
 
from basf2 import (
    set_random_seed,
    create_path,
    process,
    statistics,
    register_module,
)
from simulation import add_simulation
from beamparameters import add_beamparameters
from validation import statistics_plots, event_timing_plot
 
set_random_seed(12345)
 
main = create_path()
 
# specify number of events to be generated
eventinfosetter = register_module("EventInfoSetter")
eventinfosetter.param("evtNumList", [10000])
eventinfosetter.param("runList", [1])
eventinfosetter.param("expList", [0])
main.add_module(eventinfosetter)
 
# beam parameters
beamparameters = add_beamparameters(main, "Y4S")
 
# generate Muon events
# Particle gun
particlegun = register_module("ParticleGun")
particlegun.param("nTracks", 1)
particlegun.param("pdgCodes", [13, -13])
particlegun.param("momentumGeneration", "uniform")
particlegun.param("momentumParams", [0.1, 4])
main.add_module(particlegun)
 
# detector simulation, no background files
add_simulation(main)
 
# memory profile
main.add_module(register_module("Profile"))
 
# output
output = register_module("RootOutput")
output.param("outputFileName", "../MuonGenSimNoBkg.root")
main.add_module(output)
 
main.add_module('Progress')
process(main)
 
# Print call statistics
print(statistics)
 
statistics_plots(
    "MuonGenSimNoBkg_statistics.root",
    contact="Software team b2soft@mail.desy.de",
    job_desc="a standard simulation job with generic MuonGenSimNoBkg events",
    prefix="MuonGenSimNoBkg",
)
event_timing_plot(
    "../MuonGenSimNoBkg.root",
    "MuonGenSimNoBkg_statistics.root",
    contact="Software team b2soft@mail.desy.de",
    job_desc="a standard simulation job with Muon events",
    prefix="MuonGenSimNoBkg",
)