pix_defect.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
from ROOT import gROOT, TFile
import collections
defect_val = collections.namedtuple('defect_val',
                                    'defect, comment, recoverable')
defect_iov = collections.namedtuple('defect_iov',
                                    'defect, comment, recoverable, since, until')
 
 
 
def find_standby(nlayer, hhits, lb_max, standby_lb):
    threshold_hit = 10
    for i in range(0, nlayer):
        start = 0
        start_lb = 0
        end_lb = 0
 
        nbin = hhits[i].GetNbinsX()
 
        start_lb = 0
        for j in range(1, nbin-1):
            val = hhits[i].GetBinContent(j+1)
            if val >= threshold_hit:
                end_lb = j
                if start_lb == 0:
                    start_lb = j
 
        for j in range(1, nbin-1):
            #val_global = hglobal.GetBinContent(j+1)
            val = hhits[i].GetBinContent(j+1)
            if val < threshold_hit:
                if start == 0:
                    start = 1
                    start_lb = j
            elif start == 1:
                if j > start_lb and j < end_lb:
                    standby_lb.append([i, start_lb, j-1])
                start = 0
                start_lb = 0
        if start == 1:
            if lb_max > start_lb:
                standby_lb.append([i, start_lb, lb_max])
 
 
def find_notready(nlayer, hist, pct_low, pct_high, notready_lb, lb_max=2000):
 
    for i in range(0, nlayer):
        start = 0
        start_lb = 0
 
        nbin = hist[i].GetNbinsX()
        if i == 2:
            for j in range(1, nbin-1):
                val = hist[2].GetBinContent(
                    j+1)*38/90 + hist[3].GetBinContent(j+1)*52/90
                if val >= pct_low and val < pct_high:
                    if start == 0:
                        start = 1
                        start_lb = j
                elif start == 1:
                    notready_lb.append([i, start_lb, j-1])
                    start = 0
                    start_lb = 0
            if start == 1:
                notready_lb.append([i, start_lb, lb_max])
        elif i == 3:
            # nothing to do
            pass
        else:
            for j in range(1, nbin-1):
                val = hist[i].GetBinContent(j+1)
                if val >= pct_low and val < pct_high:
                    if start == 0:
                        start = 1
                        start_lb = j
                elif start == 1:
                    notready_lb.append([i, start_lb, j-1])
                    start = 0
                    start_lb = 0
            if start == 1:
                notready_lb.append([i, start_lb, lb_max])
 
 
def find_btagdeg(hist, pct_low, pct_high, btagdeg_lb, lb_max):
    start = 0
    start_lb = 0
 
    nbin = hist.GetNbinsX()
    for i in range(1, nbin-1):
        val = hist.GetBinContent(i+1)
        if val >= pct_low and val < pct_high:
            if start == 0:
                start = 1
                start_lb = i
        elif start == 1:
            btagdeg_lb.append([start_lb, i-1])
            start = 0
            start_lb = 0
    if start == 1:
        btagdeg_lb.append([start_lb, lb_max])
 
 
def print_def(defect_name, defect_lb):
    for i in range(0, len(defect_lb)):
        print(defect_name, "[", defect_lb[i][0], "]:", defect_lb[i][1], "-", defect_lb[i][2])
 
def print_btagdegdef(defect_name, defect_lb):
    for i in range(0, len(defect_lb)):
        print(defect_name, ": ", defect_lb[i][0], "-", defect_lb[i][1])
 
 
def assign_defect(db, defect_name, run, defect_lb):
    for i in range(0, len(defect_lb)):
        layer = defect_lb[i][0]
        if layer == 0:
            slayer = "IBL"
        elif layer == 1:
            slayer = "LAYER0"
        elif layer == 2:
            slayer = "BARREL"
        elif layer == 3:
            slayer = "BARREL"
        elif layer == 4:
            slayer = "ENDCAPA"
        elif layer == 5:
            slayer = "ENDCAPC"
 
        sdefect = "PIXEL_" + slayer + "_" + defect_name
 
        lbstart = defect_lb[i][1]
        lbend = defect_lb[i][2]
 
        start = (run << 32) + lbstart
        until = (run << 32) + lbend + 1
 
        #print(sdefect, ": LB = ", lbstart,"-",lbend,")")
        comment = "assign " + sdefect
        db.append(defect_iov(sdefect, comment, False, start, until))
 
 
 
def assign_lowstat(db, run, comment):
    sdefect = "PIXEL_LOWSTAT"
    lbstart = 1
    lbend = 2000
    start = (run << 32) + lbstart
    until = (run << 32) + lbend + 1
    #print(sdefect, ": LB = ", lbstart,"-",lbend,")")
    db.append(defect_iov(sdefect, comment, False, start, until))
 
 
def assign_btagdegdef(db, defect_name, run, defect_lb):
    for i in range(0, len(defect_lb)):
        sdefect = "PIXEL_" + defect_name
        lbstart = defect_lb[i][0]
        lbend = defect_lb[i][1]
 
        start = (run << 32) + lbstart
        until = (run << 32) + lbend + 1
 
        comment = "assign " + sdefect
        db.append(defect_iov(sdefect, comment, False, start, until))
 
 
 
def execute(run, sfile, lb_max):
    db = []
    gROOT.Reset()
    file = TFile(sfile)
 
    shits = []
    shits.append("InnerDetector/Pixel/IBL/Hits/AvgOccActivePerLumi_IBL2D")
    shits.append("InnerDetector/Pixel/BLayer/Hits/AvgOccActivePerLumi_BLayer")
    shits.append("InnerDetector/Pixel/Layer1/Hits/AvgOccActivePerLumi_Layer1")
    shits.append("InnerDetector/Pixel/Layer2/Hits/AvgOccActivePerLumi_Layer2")
    shits.append("InnerDetector/Pixel/ECA/Hits/AvgOccActivePerLumi_ECA")
    shits.append("InnerDetector/Pixel/ECC/Hits/AvgOccActivePerLumi_ECC")
 
    sbtagdeg = "InnerDetector/Pixel/PixelExpert/BTagDegEstimation/TotalDegradationPerLumi"
 
    sclus = "Global/Luminosity/AnyTrigger/nClustersAll_vs_LB"
 
    nlayer = 6
    hhits = []
    fexist_hhits = True
    for i in range(0, nlayer):
        if not file.Get(shits[i]):
            fexist_hhits = False
            continue
 
        hhits.append(file.Get(shits[i]))
 
    standby_lb = []
    if fexist_hhits is True:
        find_standby(nlayer, hhits, lb_max, standby_lb)
 
    #print_def("standby", standby_lb)
    #user = "atlpixdq"
    #assign_defect(db, "STANDBY", run, standby_lb)
 
    fexist_hbtagdeg = True
    hbtagdeg = file.Get(sbtagdeg)
    if not hbtagdeg:
        fexist_hbtagdeg = False
    btagdegestim_tolerable_lb = []
    btagdegestim_intolerable_lb = []
 
    if fexist_hhits is True and fexist_hbtagdeg is True: 
        find_btagdeg(hbtagdeg, 0.05, 0.25, btagdegestim_tolerable_lb, lb_max) 
        find_btagdeg(hbtagdeg, 0.25, 1., btagdegestim_intolerable_lb, lb_max)
 
    if 0:
        print_btagdegdef("tolerable", btagdegestim_tolerable_lb)
        print_btagdegdef("intolerable", btagdegestim_intolerable_lb)
 
    if 1:
        assign_btagdegdef(db, "PERFORMANCE_TOLERABLE", run, btagdegestim_tolerable_lb)
        assign_btagdegdef(db, "PERFORMANCE_INTOLERABLE", run, btagdegestim_intolerable_lb)
 
    fexist_hclus = True
    hclus = file.Get(sclus)
    if not hclus:
        fexist_hclus = False
    if fexist_hclus is True:
        nevent = hclus.GetEntries()
        if nevent < 100000:
            assign_lowstat(db, run, "assign PIXEL_LOWSTAT")
 
    return db