dump_digits.py

#!/usr/bin/env python3
 
 
 
import basf2 as b2
 
# Some ROOT tools
from ROOT import Belle2
 
 
class dump_digits(b2.Module):
 
    """A simple module to check SVD digit fits."""
 
    def __init__(self, filename='dumped_digits.txt'):
        """Initialize the module"""
 
        super().__init__()
        
        self.outfile = open(filename, 'w')
        
        self.vxdid_factors = (8192, 256, 32)
        # Get a handle on the GeoCache
 
        self.geoCache = Belle2.VXD.GeoCache.getInstance()
        # Record filenames
 
        self.noise_cal = Belle2.SVDNoiseCalibrations()
 
        self.pulse_cal = Belle2.SVDPulseShapeCalibrations()
 
    def beginRun(self):
        """ Tasks at the start of a run """
        # Write header to output file
        self.outfile.write(
            'EventNo Layer Ladder Sensor Side StripNo TimeTrigger ' +
            'GoodStrip Gain Noise Width TimeShift ' +
            'Sample0 Sample1 Sample2 Sample3 Sample4 Sample5 Charge TimeFit Chi2\n')
 
    def event(self):
        """Cycle through RecoDigit/ShaperDigit pairs and dump the corresponding data"""
 
        evt_info = Belle2.PyStoreObj('EventMetaData')
        svd_evt_info = Belle2.PyStoreObj('SVDEventInfo')
        event_number = evt_info.getEvent()
        mode_byte = svd_evt_info.getModeByte()
        reco_digits = Belle2.PyStoreArray('SVDRecoDigits')
 
        for reco_digit in reco_digits:
 
            shaper_digit = reco_digit.getRelatedTo('SVDShaperDigits')
 
            s = ''
 
            # Sesnor/side/strip identification
            [layer, ladder, sensor] = self.decode(reco_digit.getRawSensorID())
            s += f'{event_number} {layer} {ladder} {sensor} {"u" if reco_digit.isUStrip() else "v"} {reco_digit.getCellID()} '
            sensorID = reco_digit.getSensorID()
            stripNo = reco_digit.getCellID()
            # Trigger bin from SVDModeByte
            triggerBin = ord(mode_byte.getTriggerBin())
            triggerTime = 0.25 * 31.44 * (-4 + triggerBin + 0.5)
            s += f'{triggerTime:.3f} '
            # Calibrations
            stripNoise = self.noise_cal.getNoise(sensorID, reco_digit.isUStrip(), stripNo)
            stripGain = 22500 / self.pulse_cal.getADCFromCharge(sensorID, reco_digit.isUStrip(), stripNo, 22500)
 
            stripT0 = self.pulse_cal.getPeakTime(sensorID, reco_digit.isUStrip(), stripNo)
            stripWidth = self.pulse_cal.getWidth(sensorID, reco_digit.isUStrip(), stripNo)
            s += f'{"y"} {stripGain:.3f} {stripNoise:.3f} {stripWidth} {stripT0:.3f} '
 
            # Digit information
            samples = shaper_digit.getSamples()
            for iSample in range(6):
                s += f'{samples[iSample]} '
 
            # Data from time fit
            s += f'{reco_digit.getCharge():.3f} {reco_digit.getTime():.3f} {reco_digit.getChi2Ndf():.3f}'
            s += '\n'
            self.outfile.write(s)
 
    def terminate(self):
        """ Close the output file."""
 
        self.outfile.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
 
    def three_test(self, digit, threshold):
        ''' 3-samples digit test '''
 
        counter = 0
        for sample in digit.getSamples():
            if counter == 3:
                continue
            elif sample >= threshold:
                counter += 1
            else:
                counter = 0
        return (counter >= 3)