EclMaterialAnalysis.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
import os
from basf2 import *
from simulation import add_simulation
from reconstruction import add_reconstruction
 
# Create paths
main = create_path()
 
# Event setting and info
eventinfosetter = register_module('EventInfoSetter')
eventinfosetter.param({'evtNumList': [100000], 'runList': [1]})
main.add_module(eventinfosetter)
 
# random number for generation
import random
# intseed = random.randint(1, 10000000)
set_random_seed(123456)
 
# Create geometry
# Geometry parameter loader
# gearbox = register_module('Gearbox')
 
# Geometry builder
# geometry = register_module('Geometry')
# geometry.param('components', ['ECL'])
 
components = [  # 'MagneticField',
    # 'BeamPipe',
    'ECL',
    'PXD',
    'SVD',
    'CDC',
    # 'EKLM',
    # 'BKLM',
    # 'TOP',
    'ARICH',
]
 
# Simulation
# g4sim = register_module('FullSim')
 
# single particle generator settings
pGun = register_module('ParticleGun')
param_pGun = {
    'pdgCodes': [22],
    'nTracks': 1,
    'momentumGeneration': 'fixed',
    'momentumParams': [0.5],
    'thetaGeneration': 'uniform',  # uniformCos,
    'thetaParams': [(12.01), (31.36)],
    'phiGeneration': 'uniform',
    'phiParams': [0., 360.],
    'vertexGeneration': 'uniform',
    'xVertexParams': [0.0, 0.0],
    'yVertexParams': [0.0, 0.0],
    'zVertexParams': [0.0, 0.0],
}
 
pGun.param(param_pGun)
main.add_module(pGun)
 
bkgdir = 'bkg/'
# bkg = glob.glob(bkgdir+'*.root')
 
bkgFiles = [
    bkgdir + 'Coulomb_HER_100us.root',
    bkgdir + 'Coulomb_LER_100us.root',
    bkgdir + 'Coulomb_HER_100usECL.root',
    bkgdir + 'Coulomb_LER_100usECL.root',
    bkgdir + 'RBB_HER_100us.root',
    bkgdir + 'RBB_LER_100us.root',
    bkgdir + 'RBB_HER_100usECL.root',
    bkgdir + 'RBB_LER_100usECL.root',
    bkgdir + 'Touschek_HER_100us.root',
    bkgdir + 'Touschek_LER_100us.root',
    bkgdir + 'Touschek_HER_100usECL.root',
    bkgdir + 'Touschek_LER_100usECL.root',
]
 
# add_simulation(main)#, bkgfiles=bkgFiles)
# main.add_module(gearbox)
# main.add_module(geometry)
# main.add_module(g4sim)
 
add_simulation(main)
add_reconstruction(main)
 
# display = register_module('Display')
# main.add_module(display)
 
# eclDataAnalysis module
ecldataanalysis = register_module('ECLDataAnalysis')
ecldataanalysis.param('rootFileName', 'EclDataAnalysis_500MeV_100000_Full_FWD.root')
ecldataanalysis.param('doTracking', 0)
main.add_module(ecldataanalysis)
 
process(main)
print(statistics)