dump_clusters.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
import sys
import math
from basf2 import *
 
# Some ROOT tools
import ROOT
from ROOT import Belle2
 
 
class dump_clusters(Module):
 
    """A simple module to timing of SVD clusters.
    Intended for use with the RandomizeEventTimes feature of SVDDigitizer,
    which stores (randomized) event times in EventMetaData.
    For background studies, it distinguishes signal and background clusters..
    """
 
    def __init__(self, filename='dump_clusters.txt', collection='SVDClusters'):
        """Initialize the module"""
 
        super(dump_clusters, self).__init__()
        
        self.file = open(filename, 'w')
 
        self.collection = collection
        
        self.vxdid_factors = (8192, 256, 32)
 
    def beginRun(self):
        """ Write legend for file columns """
 
        self.file.write(
            'EventNo Layer Ladder Sensor Side Position Bg Time_event' +
            'Time_cluster Charge_cluster Size_cluster Chi2_cluster\n'
        )
 
    def event(self):
        """Save information on all clusters together with event time data. """
 
        eventData = Belle2.PyStoreObj('EventMetaData')
        eventNo = eventData.getEvent()
        svd_evt_info = Belle2.PyStoreObj('SVDEventInfo')
        mode_byte = svd_evt_info.getModeByte()
        clusters = Belle2.PyStoreArray(self.collection)
 
        for cluster in clusters:
 
            bg_label = 's'
            reco_digit = cluster.getRelatedTo('SVDRecoDigits')
            triggerTime = 0.0
            if reco_digit:
                # Trigger bin from SVDModeByte
                triggerBin = ord(mode_byte.getTriggerBin())
                triggerTime = 0.25 * 31.44 * (-4 + triggerBin + 0.5)
 
            sensorID = cluster.getRawSensorID()
 
            uSide = cluster.isUCluster()
 
            # Sesnor identification
            [layer, ladder, sensor] = self.decode(sensorID)
            side_str = 'u' if uSide else 'v'
 
            s = '{event} {layer} {ladder} {sensor} {side} {strip} {bg} {etime}'.format(
                event=eventNo,
                layer=layer,
                ladder=ladder,
                sensor=sensor,
                side=side_str,
                strip=cluster.getPosition(),
                bg=bg_label,
                etime=triggerTime
            )
            # Cluster information
            s += ' {clstime} {clscharge} {clssize} {clschi}\n'.format(
                clstime=cluster.getClsTime(),
                clscharge=cluster.getCharge(),
                clssize=cluster.getSize(),
                clschi=cluster.getChi2()
            )
            self.file.write(s)
 
    def terminate(self):
        """ Close the output file."""
 
        self.file.close()
 
    def decode(self, vxdid):
        """ Utility to decode sensor IDs """
 
        result = []
        for f in self.vxdid_factors:
            result.append(vxdid // f)
            vxdid = vxdid % f
 
        return result