DumpDigits.py

#!/usr/bin/env python3
 

 
import basf2 as b2
 
# Some ROOT tools
from ROOT import Belle2
 
 
class DumpDigits(b2.Module):
 
    """A simple module to dump PXD digits."""
 
    def __init__(self):
        """Initialize the module"""
 
        super().__init__()
        
        self.dumpfile = 'PXDDigitsDump.txt'
        
        self.vxdid_factors = (8192, 256, 32)
 
    def beginRun(self):
        """ Write legend for file columns """
        with open(self.dumpfile, 'w') as dumpfile:
            dumpfile.write('vxd.id layer ladder sensor digit.u digit.v digit.charge ')
 
    def event(self):
        """Find clusters with a truehit and print some stats."""
 
        digits = Belle2.PyStoreArray('PXDDigits')
        # nDigits = digits.getEntries()
        # Start with clusters and use the relation to get the corresponding
        # digits and truehits.
        with open(self.dumpfile, 'a') as dumpfile:
            s = ''
            for digit in digits:
                # Sensor identification
                sensorID = digit.getSensorID().getID()
                [layer, ladder, sensor] = self.decode(sensorID)
                s += f'{sensorID} {layer:3d} {ladder:3d} {sensor:3d} {digit.getUCellID():6d} ' + \
                    f'{digit.getVCellID():6d} {digit.getCharge():8.1f}'
                s += '\n'
            dumpfile.write(s)
 
    def terminate(self):
        """ Nothing."""
 
    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