RandomDisplacement.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
# Generates random EKLM displacement data.
 
import basf2
from ROOT.Belle2 import KLMDatabaseImporter, BKLMAlignment, EKLMAlignment, \
                        EKLMSegmentAlignment, KLMAlignmentData, \
                        KLMChannelIndex, KLMDisplacementGenerator
 
# Create main path
main = basf2.create_path()
basf2.set_log_level(basf2.LogLevel.INFO)
 
# EventInfoSetter
main.add_module('EventInfoSetter')
 
# Gearbox
main.add_module('Gearbox')
 
# Process the main path
basf2.process(main)
 
dbImporter = KLMDatabaseImporter()
 
bklmAlignment = BKLMAlignment()
eklmAlignment = EKLMAlignment()
eklmSegmentAlignment = EKLMSegmentAlignment()
 
# Random displacement for EKLM.
displacementGenerator = KLMDisplacementGenerator()
displacementGenerator.generateRandomDisplacement(
    eklmAlignment, eklmSegmentAlignment, True, False)
displacementGenerator.saveDisplacement(
    eklmAlignment, eklmSegmentAlignment, 'EKLMDisplacement.root')
 
# Zero displacement for BKLM.
alignmentData = KLMAlignmentData(0, 0, 0, 0, 0, 0)
index = KLMChannelIndex(KLMChannelIndex.c_IndexLevelLayer)
index2 = KLMChannelIndex(KLMChannelIndex.c_IndexLevelLayer)
index = index2.beginBKLM()
 
while (index != index2.endBKLM()):
    module = index.getKLMModuleNumber()
    bklmAlignment.setModuleAlignment(module, alignmentData)
    index.increment()
 
# Import the payloads.
dbImporter.setIOV(0, 0, -1, -1)
dbImporter.importAlignment(
    bklmAlignment, eklmAlignment, eklmSegmentAlignment, True)