skim_utils.py

# !/usr/bin/env python
# -*- coding: utf-8 -*-
 
 
 
 
 
import basf2
import ROOT
from ROOT import Belle2
from basf2 import *
 
 
class skimOutRNDTrgModule(basf2.Module):
    """
    returns True if the event is NOT a random triggered event
    """
 
    def event(self):
        """
        check RawFTSW to get the event type
        """
        
        rawFTSW = Belle2.PyStoreArray('RawFTSWs')
 
        if not rawFTSW.isValid():
            B2WARNING('No RawFTSW available - event ignored')
            self.return_value(0)
 
            return
 
        # unknown meaning of this number
        unknownInt = 0
        if rawFTSW[0].GetTRGType(unknownInt) != Belle2.TRGSummary.TTYP_RAND:
            self.return_value(1)
        else:
            self.return_value(0)
 
 
class skimSVDBurstEventsModule(basf2.Module):
    """
    returns True if the event is a Burst event (number of strips > max number of strips)
    use set_nMaxStrips(nMaxStrips) to set the max number of strips iof a non-burst event, default is nMaxStrips=5000
    """
 
    def __init__(self):
        """constructor"""
 
        super().__init__()
 
        self.nMaxStrips = 5000
        '''max number of strips for NON-BURST events'''
 
    def set_nMaxStrips(self, user_nMaxStrips):
        """set the max strips, otherwise 5000"""
 
        self.nMaxStrips = user_nMaxStrips
 
    def event(self):
        '''event'''
 
        strips = Belle2.PyStoreArray('SVDShaperDigits')
 
        if not strips.isValid():
            B2WARNING('No SVDShaperDigits - event ignored')
            self.return_value(0)
 
            return
 
        if strips.getEntries() > self.nMaxStrips:
            self.return_value(1)
        else:
            self.return_value(0)
 
 
class skim6SampleEventsPyModule(basf2.Module):
    """
    returns True if the event is acquired with 6 samples
    """
 
    def __init__(self):
        """constructor"""
 
        super().__init__()
 
    def event(self):
        '''event'''
 
        # take SVDEventInfo or SVDvVentInfoSim
        eventInfo = Belle2.PyStoreObj('SVDEventInfo')
 
        if not eventInfo.isValid():
            eventInfo = Belle2.PyStoreObjPtr('SVDEventInfoSim')
 
        if not eventInfo.isValid():
            B2WARNING('No SVDEventInfo/SVDEventInfoSim - event ignored')
            self.return_value(0)
 
            return
 
        # check if we acquired 6-sample strips in this event
        if eventInfo.getNSamples() == 6:
            self.return_value(1)
        else:
            self.return_value(0)
 
 
"""
class skimSVDBurstEventsDataSizeModule(basf2.Module):
 
    returns True if the event is a Burst event (svd data size above limit))
    use set_MaxDataSize(maxDataSize) to set the max data sieze of a non-burst event, default is maxDataSize=50kB
 
 
    def __init__(self):
 
        super().__init__()
        self.maxDataSize = 50000
 
    def set_maxDataSize(self,user_maxDataSize):
        "set the max strips, otherwise 5000"
 
        self.maxDataSize = user_maxDataSize
 
    def event(self):
 
        raw = Belle2.PyStoreArray('RawSVDs')
 
        if not raw.isValid():
            B2WARNING('No RawSVDs - event ignored')
            self.return_value(0)
 
            return
 
        StoreArray<RawSVD> rawsvd;
        int nsvd = rawsvd.getEntries();
        int svdsize = 0;
        for (int i = 0; i < nsvd; i++) { // Loop over COPPERs
        int nbytes = rawsvd[i]->GetBlockNwords(0) * sizeof(unsigned int);
        svdsize += nbytes;
        }
 
        svdSize = 0
 
        for aRaw in raw:
            svdSize+=aRaw.GetBlockNWords(0)*sizeof(unsigned int)
        if raw.getEntries() > self.maxDataSize:
            self.return_value(1)
        else:
            self.return_value(0)
"""
 
 
class skimSVDTriggerBinEventsModule(basf2.Module):
    """
    returns True if TriggerBin of the event is the selected one
    use set_tb(tb) to set the value of the selected TriggerBin (0,1,2,3)
    """
 
    def __init__(self):
        """constructor"""
 
        super().__init__()
        self.tb = 99  # no selection of TB
        ''' trigger bin to select'''
 
    def set_tb(self, user_tb):
        '''set the trigger bin, otherwise 99, i.e. no selection'''
 
        self.tb = user_tb
 
    def event(self):
        '''event'''
 
        if int(self.tb) == 99:
            self.return_value(1)
 
        eventInfo = Belle2.PyStoreObj('SVDEventInfo')
 
        if not eventInfo.isValid():
            B2ERROR('No SVDEventInfo - event ignored')
            self.return_value(0)
 
            return
 
        if ord(eventInfo.getModeByte().getTriggerBin()) == int(self.tb):
            self.return_value(1)
        else:
            self.return_value(0)