PXDDataBaseImporter.py

#!/usr/bin/env python3
 

 
 

 
 
import basf2 as b2
from ROOT import Belle2
import ROOT
 
# set some random seed
b2.set_random_seed(10346)
 
 
def importEmptyHotPixelMask(expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports an empty payloads for the PXD
    hot pixel masking.
 
    This function works for every Belle 2 geometry
    """
 
    # Create an empty hot pixel mask
    HotPixelMask = Belle2.PXDMaskedPixelPar()
 
    # Set the iov for the mask
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDMaskedPixelPar', HotPixelMask, iov)
 
 
def importEmptyDeadPixelMask(expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports an empty payloads for the PXD
    dead pixel masking.
 
    This function works for every Belle 2 geometry
    """
 
    # Create an empty mask
    DeadPixelMask = Belle2.PXDDeadPixelPar()
 
    # Set the iov for the mask
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDDeadPixelPar', DeadPixelMask, iov)
 
 
def importEmptyOccupancyInfo(expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports an empty payloads for the PXD
    pixel masking.
 
    This function works for every Belle 2 geometry
    """
 
    # Create a dummy occupancy info
    OccupancyInfo = Belle2.PXDOccupancyInfoPar()
 
    # Set the iov for the mask
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDOccupancyInfoPar', OccupancyInfo, iov)
 
 
def importEmptyClusterChargeMap(expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports an empty payloads for the PXD
    gain calibration.
 
    This function works for every Belle 2 geometry
    """
 
    # Create a dummy payload
    ChargeMap = Belle2.PXDClusterChargeMapPar()
 
    # Set the iov for the mask
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDClusterChargeMapPar', ChargeMap, iov)
 
 
def importEmptyGainMap(expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports an empty payloads for the PXD
    gain calibration.
 
    This function works for every Belle 2 geometry
    """
 
    # Create a dummy payload
    GainMap = Belle2.PXDGainMapPar()
 
    # Set the iov for the mask
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDGainMapPar', GainMap, iov)
 
 
def importRandomPixelMaskPhase2(HotPixelFraction=0.001, expLow=0, runLow=0, expHigh=-1, runHigh=-1):
    """
    This function imports payloads for the PXD
    pixel masking with random hot pixels
 
    This function works for Phase 2
    """
 
    # Create an empty mask
    pixelMask = Belle2.PXDMaskedPixelPar()
 
    # Create a list of sensors
    sensorIDList = [
        Belle2.VxdID("1.1.1"), Belle2.VxdID("1.1.2"),
        Belle2.VxdID("1.2.1"), Belle2.VxdID("1.2.2"),
        Belle2.VxdID("1.3.1"), Belle2.VxdID("1.3.2"),
        Belle2.VxdID("1.4.1"), Belle2.VxdID("1.4.2"),
        Belle2.VxdID("1.5.1"), Belle2.VxdID("1.5.2"),
        Belle2.VxdID("1.6.1"), Belle2.VxdID("1.6.2"),
        Belle2.VxdID("1.7.1"), Belle2.VxdID("1.7.2"),
        Belle2.VxdID("1.8.1"), Belle2.VxdID("1.8.2"),
        Belle2.VxdID("2.4.1"), Belle2.VxdID("2.4.2"),
        Belle2.VxdID("2.5.1"), Belle2.VxdID("2.5.2")]
 
    # Mean number of hot pixels per sensor
    AverageHotPixels = HotPixelFraction * 250 * 768
 
    # ... and mask some random pixels
    for sensorID in sensorIDList:
        # Randomized number of hot pixels
        nHotPixels = ROOT.gRandom.Poisson(AverageHotPixels)
 
        for i in range(nHotPixels):
            uid = ROOT.gRandom.Integer(250)
            vid = ROOT.gRandom.Integer(768)
            print(f"mask pixel uid={uid}, vid={vid}, id={uid * 768 + vid}")
            pixelMask.maskSinglePixel(sensorID.getID(), uid * 768 + vid)
 
    # Print the final mask before committing to db
    for sensorMask in pixelMask.getMaskedPixelMap():
        sensorID = Belle2.VxdID(sensorMask.first)
        print(f"Mask for sensor {sensorID} contains {sensorMask.second.size()} pixels")
 
    iov = Belle2.IntervalOfValidity(expLow, runLow, expHigh, runHigh)
    Belle2.Database.Instance().storeData('PXDMaskedPixelPar', pixelMask, iov)
 
 
if __name__ == "__main__":
 
    importEmptyHotPixelMask()
    importEmptyDeadPixelMask()
    importEmptyOccupancyInfo()
    importEmptyGainMap()
    importEmptyClusterChargeMap()