run_eclMuMuE_collector.py

#!/usr/bin/env python3
 
 
 
# Usage: basf2 run_eclMuMuE_collector.py
# option: basf2 run_eclMuMuE_collector.py OutputFileName.root
 
# Run just the collector part of the eclMuMuE calibration, which calibrates the single crystal
# energy response using muon pairs.
# Input file should be a muon pair skim (or MC) and include ECLDigits and tracks.
# Also requires ECLCalDigits if code is being used to find expected deposited energies.
# Output histograms are written to specified output file.
# run_eclMuMuE_algorithm.py is then used to perform calibration using these histograms, or to simply copy
# them to an output file.
 
import sys
import basf2 as b2
 
 
main = b2.create_path()
DR2 = '/ghi/fs01/belle2/bdata/users/karim/MC/DR2/release-00-09-02/mdst_calib/*.root'
main.add_module('RootInput', inputFileNames=[DR2])
 
narg = len(sys.argv)
outputName = "eclMuMuECollectorOutput.root"
if(narg >= 2):
    outputName = sys.argv[1]
main.add_module("HistoManager", histoFileName=outputName)
 
# Genfit and track extrapolation
gearbox = b2.register_module('Gearbox')
main.add_module(gearbox)
geometry = b2.register_module('Geometry')
main.add_module(geometry)
main.add_module("SetupGenfitExtrapolation")
ext = b2.register_module('Ext')
 
# extrapolate using muon hypothesis only
pdgcodes = [13]
ext.param('pdgCodes', pdgcodes)
 
main.add_module(ext)
 
eclMuMuE = b2.register_module('eclMuMuECollector')
eclMuMuE.param('minPairMass', 9.0)
eclMuMuE.param('minTrackLength', 30.)
eclMuMuE.param('MaxNeighbourE', 0.010)
eclMuMuE.param('thetaLabMinDeg', 17.)
eclMuMuE.param('thetaLabMaxDeg', 150.)
# Can fill histograms with MC eclCalDigits to calculate true deposited energy
eclMuMuE.param('measureTrueEnergy', False)
eclMuMuE.param('requireL1', True)
main.add_module(eclMuMuE)
 
main.add_module('Progress')
 
b2.set_log_level(b2.LogLevel.INFO)
 
# It is possible to force the job to use the specified global tag.
# Default localdb is the subdirectory of current working directory, but can be overwritten
b2.reset_database()
b2.use_database_chain()
b2.use_central_database("development")
b2.use_local_database("localdb/database.txt")
 
b2.process(main)
 
print(b2.statistics)