PXDValidationGen.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
"""<header>
<input>PXDValidationTTreeOutput.root</input>
<input>PXDValidationTTreeSimHitOutput.root</input>
<input>PXDValidationTTreeDigitOutput.root</input>
<output>PXDValidation.root</output>
<contact>Peter Kodys, peter.kodys@mff.cuni.cz</contact>
<description>This steering file run the PXD response on an input file with own generated events.</description>
</header>"""
 
from basf2 import *
from PXDValidationTTree import PXDValidationTTree
from PXDValidationTTreeSimHit import PXDValidationTTreeSimHit
from PXDValidationTTreeDigit import PXDValidationTTreeDigit
 
set_log_level(LogLevel.WARNING)
 
# Register necessary modules:
# Particle gun module
particlegun = register_module('ParticleGun')
# Create Event information
eventinfo = register_module('EventInfoSetter')
# Show progress of processing
progress = register_module('Progress')
# Load parameters
gearbox = register_module('Gearbox')
# Create geometry
geometry = register_module('Geometry')
# Run simulation
simulation = register_module('FullSim')
# PXD digitization module
pxddigi = register_module('PXDDigitizer')
# PXD clustering module
pxdclust = register_module('PXDClusterizer')
 
# ============================================================================
# Set a fixed random seed for particle generation:
set_random_seed(3038402)
 
analyze = PXDValidationTTree()
analyzeSimHit = PXDValidationTTreeSimHit()
analyzeDigit = PXDValidationTTreeDigit()
 
# Set the number of events to be generate and processed (xxxx events)
eventinfo.param({'evtNumList': [1000], 'runList': [1]})
 
# Set parameters for particlegun
particlegun.param({
    # Generate 1 track
    'nTracks': 1,
    # Always one, no fluctuations
    'varyNTracks': False,
    # Generate pi+, pi-, e+ and e-
    'pdgCodes': [211, -211, 11, -11],
    # with a normal distributed transversal momentum
    'momentumGeneration': 'normal',
    # with a center of 2 GeV and a width of 0.2 GeV
    'momentumParams': [2, 0.2],
    # a uniform distributed phi angle,
    'phiGeneration': 'uniform',
    # full circle
    'phiParams': [0, 360],
    # Generate theta angles uniform in cos theta
    'thetaGeneration': 'uniformCos',
    # between 15 and 150 degree
    'thetaParams': [15, 150],
    # normal distributed vertex generation
    'vertexGeneration': 'normal',
    # around the origin with a sigma of 0.1cm in each direction
    'xVertexParams': [0.0, 0.1],
    'yVertexParams': [0.0, 0.1],
    'zVertexParams': [0.0, 0.1],
    # all one track sharing the same vertex per event
    'independentVertices': False,
})
 
# ============================================================================
# Print the parameters of the particle gun
print_params(particlegun)
 
# Select subdetectors to be built
geometry.param('components', ['MagneticField', 'PXD', 'SVD'])
 
pxddigi.param('statisticsFilename', 'PXDValidationDiags3.root')
# pxddigi.param('ElectronicEffects', True)
# pxddigi.param('ElectronicEffects', False)
# pxddigi.param('SimpleDriftModel', False)
# pxddigi.param('PoissonSmearing', True)
# pxddigi.param('NoiseSN', 1.0)
 
# pxdclust.param('NoiseSN', 1.0)
# pxdclust.param('TanLorentz', 0.0)
 
# ============================================================================
# create processing path
main = create_path()
main.add_module(eventinfo)
main.add_module(progress)
main.add_module(particlegun)
main.add_module(gearbox)
main.add_module(geometry)
main.add_module(simulation)
main.add_module(pxddigi)
main.add_module(pxdclust)
main.add_module(analyze)
main.add_module(analyzeSimHit)
main.add_module(analyzeDigit)
 
# generate events
process(main)
 
# show call statistics
print(statistics)