SVDReconstructionWithNN.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
from basf2 import *
from svd.dump_digits import dump_digits
from svd.dump_clusters import dump_clusters
 
# show warnings during processing
set_log_level(LogLevel.WARNING)
 
# Set database
"""
For phase 3 related studies, use this tag
"""
# use_local_database('localDB_NN_toImport/database_NN_toImport.txt','localDB_NN_toImport')
use_central_database("development")
 
# Register modules
 
# Particle gun module
particlegun = register_module('ParticleGun')
# Create Event information
eventinfosetter = 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')
# Add the PXD digitizer, too, to avoid problems with empty events.
PXDDIGI = register_module('PXDDigitizer')
# SVD digitization module
SVDDIGI = register_module('SVDDigitizer')
SVDDIGI.param('StartSampling', -31.44)
SVDDIGI.param('GenerateShaperDigits', True)
# SVDDIGI.param('signalsList', 'SVDSignalsList.csv')
# SVDDIGI.param('RandomizeEventTimes', True)
# SVDDIGI.param('TimeFrameLow', -180)
# SVDDIGI.param('TimeFrameHigh', 150)
# SVD signal reconstructor
SVDSIGR = register_module('SVDNNShapeReconstructor')
# Write RecoDigits so that we can check them
SVDSIGR.param('WriteRecoDigits', True)
# RecoDigit clusterizer
SVDCLUST1 = register_module('SVDNNClusterizer')
# ShaperDigit clusterizer
SVDCLUST2 = register_module("SVDClusterizerDirect")
# Save output of simulation
output = register_module('RootOutput')
 
# ============================================================================
# Set a fixed random seed for particle generation:
set_random_seed(1028307)
 
# ============================================================================
# Setting the list of particle codes (PDG codes) for the generated particles
particlegun.param('pdgCodes', [-11, 11])
 
# ============================================================================
# Setting the number of tracks to be generated per event:
particlegun.param('nTracks', 1)
 
# Set the number of events to be processed (100 events)
eventinfosetter.param({'evtNumList': [1000], 'expList': [0], 'runList': [0]})
 
# Set output filename
output.param('outputFileName', 'SVDTestOutput.root')
 
# Select subdetectors to be built
# geometry.param('Components', ['PXD','SVD'])
geometry.param('components', ['MagneticField', 'PXD', 'SVD'])
 
# ============================================================================
# Do the simulation
 
main = create_path()
main.add_module(eventinfosetter)
main.add_module(progress)
main.add_module(gearbox)
main.add_module(geometry)
main.add_module(particlegun)
main.add_module(simulation)
main.add_module(PXDDIGI)
main.add_module(SVDDIGI)
main.add_module(SVDSIGR)
main.add_module(dump_digits())
main.add_module(SVDCLUST2)
main.add_module(dump_clusters())
main.add_module(output)
 
# Process events
process(main)
 
# Print call statistics
print(statistics)