SimpleMCTracking.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
 
 
import basf2 as b2
 
b2.set_log_level(b2.LogLevel.INFO)
# set_log_level(LogLevel.DEBUG)
b2.set_random_seed(1028307)
 
# register necessary modules
eventinfosetter = b2.register_module('EventInfoSetter')
 
# generate one event
eventinfosetter.param('expList', [0])
eventinfosetter.param('runList', [1])
eventinfosetter.param('evtNumList', [1])
eventinfoprinter = b2.register_module('EventInfoPrinter')
 
# create geometry
gearbox = b2.register_module('Gearbox')
geometry = b2.register_module('Geometry')
 
# simulate CDC only; B-field outside CDC = 0
geometry.param('components', ['MagneticFieldConstant4LimitedRCDC', 'CDC'])
 
# particle gun to shoot particles in the detector
pGun = b2.register_module('ParticleGun')
 
# choose the particles you want to simulate
param_pGun = {
    'pdgCodes': [211],
    'nTracks': 1,
    'varyNTracks': 0,
    'momentumGeneration': 'uniform',
    'momentumParams': [1.0, 1.0],
    'thetaGeneration': 'uniform',
    #    'thetaGeneration': 'uniformCos',
    'thetaParams': [90., 90.],
    'phiGeneration': 'uniform',
    'phiParams': [0., 360.],
    'xVertexParams': [0.0, 0.0],
    'yVertexParams': [0.0, 0.0],
    'zVertexParams': [0.0, 0.0],
}
pGun.param(param_pGun)
 
# simulation
g4sim = b2.register_module('FullSim')
# this is needed for the MCTrackFinder to work correctly <- obsolete ?
g4sim.param('StoreAllSecondaries', True)
# g4sim.param('SecondariesEnergyCut', 0.0)
g4sim.param('UICommandsAtIdle', ['/control/execute interactions.mac'])
# suppress delta-rays
g4sim.param('ProductionCut', 1000000.)
 
# digitizer
cdcDigitizer = b2.register_module('CDCDigitizer')
 
# find MCTracks
# mctrackfinder = register_module('MCTrackFinder')
# mctrackfinder = register_module('TrackFinderMCTruth')
mctrackfinder = b2.register_module('TrackFinderMCTruthRecoTracks')
 
# select which detectors you would like to use
param_mctrackfinder = {  # select which particles to use: primary particles
    'UseCDCHits': 1,
    'UseSVDHits': 0,
    'UsePXDHits': 0,
    #    'UseClusters': False,
    'WhichParticles': ['primary'],
    #    'WhichParticles': ['CDC'],
}
mctrackfinder.param(param_mctrackfinder)
 
# setupgf
setupgf = b2.register_module('SetupGenfitExtrapolation')
# param_setupgf = {}
# setupgf.param(param_setupgf)
 
# fitting
cdcfitting = b2.register_module('DAFRecoFitter')
# param_cdcfitting = {}
# cdcfitting.param(param_cdcfitting)
 
# build track
buildtrack = b2.register_module("TrackCreator")
 
# output
output = b2.register_module('RootOutput')
output.param('outputFileName', 'SimpleMCTrackingOutput.root')
 
# create path
main = b2.create_path()
 
# add modules to path
main.add_module(eventinfosetter)
main.add_module(eventinfoprinter)
main.add_module(gearbox)
main.add_module(geometry)
 
main.add_module(pGun)
main.add_module(g4sim)
main.add_module(cdcDigitizer)
 
main.add_module(mctrackfinder)
main.add_module(setupgf)
main.add_module(cdcfitting)
main.add_module(buildtrack)
 
main.add_module(output)
 
# Process events
b2.process(main)
print(b2.statistics)