SVDValidationSteering.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
 
 
"""
<header>
  <input>EvtGenSimRec.root</input>
  <output>SVDValidationTTree.root</output>
  <output>SVDValidationTTreeStrip.root</output>
  <output>SVDValidationTTreeSpacePoint.root</output>
  <output>SVDValidationTTreeSimhit.root</output>
  <output>SVDValidationTTreeRecoTrack.root</output>
  <output>SVDValidationTTreeRecoDigit.root</output>
  <output>SVDValidationTTreeTrueHit.root</output>
  <output>SVDValidationTTreeCluster.root</output>
  <contact>SVD Software Group, svd-software@belle2.org</contact>
  <description>This is the SVD validation steering file.</description>
</header>
"""
 
import basf2 as b2
# from basf2 import conditions as b2c
# Individual validation packages
from SVDValidationTTreeStrip import SVDValidationTTreeStrip
from SVDValidationTTreeSimhit import SVDValidationTTreeSimhit
from SVDValidationTTreeSpacePoint import SVDValidationTTreeSpacePoint
from SVDValidationTTreeRecoTrack import SVDValidationTTreeRecoTrack
from SVDValidationTTreeRecoDigit import SVDValidationTTreeRecoDigit
from SVDValidationTTreeTrueHit import SVDValidationTTreeTrueHit
from SVDValidationTTreeCluster import SVDValidationTTreeCluster
 
b2.set_random_seed(12345)
 
# b2c.prepend_globaltag("svd_onlySVDinGeoConfiguration")
 
main = b2.create_path()
 
# Using 1000 EvtGen events already simulated and reconstructed
input = b2.register_module('RootInput')
input.param('inputFileName', '../EvtGenSimRec.root')
main.add_module(input)
 
# Gearbox and Geometry modules need to be registered anyway
 
# Load parameters
gearbox = b2.register_module('Gearbox')
main.add_module(gearbox)
 
# Create geometry
main.add_module("Geometry")
 
# Show progress of processing
progress = b2.register_module('Progress')
main.add_module(progress)
 
# SVD validation modules
svdvalidationstrip = SVDValidationTTreeStrip()
main.add_module(svdvalidationstrip)
svdvalidationspacepoint = SVDValidationTTreeSpacePoint()
main.add_module(svdvalidationspacepoint)
svdvalidationsimhit = SVDValidationTTreeSimhit()
main.add_module(svdvalidationsimhit)
svdvalidationtrack = SVDValidationTTreeRecoTrack()
main.add_module(svdvalidationtrack)
svdvalidationdigit = SVDValidationTTreeRecoDigit()
main.add_module(svdvalidationdigit)
svdvalidationtruehit = SVDValidationTTreeTrueHit()
main.add_module(svdvalidationtruehit)
svdvalidationcluster = SVDValidationTTreeCluster()
main.add_module(svdvalidationcluster)
 
 
main.add_module('Progress')
b2.process(main)
 
# Print call statistics
print(b2.statistics)