getBrightness.py

Go to the documentation of this file.

#!/usr/bin/env python
import ROOT
import argparse


if __name__=='__main__':

    parser = argparse.ArgumentParser(description='Process fiber brightness files')
    parser.add_argument('-fd', '--far-detector', action='store_true')
    parser.add_argument('-nd', '--near-detector', action='store_true')
    parser.add_argument('-p', '--plane-number', type=int, required=True)
    parser.add_argument('-c', '--cell-number', type=int, required=True)
    args = parser.parse_args()

    if args.far_detector and args.near_detector:
        print "Can only set --fd or --nd!"
        exit(1)

    if not args.far_detector and not args.near_detector:
        print "Must set either --fd or --nd!"
        exit(1)

    if args.far_detector:
        fInName = "fdBrightness.root"
    if args.near_detector:
        fInName = "ndBrightness.root"

    plane = args.plane_number
    cell = args.cell_number
    infile = ROOT.TFile(fInName, "READ")

    bmap = infile.Get("BrightnessByBin")
    bvalue = infile.Get("BrightnessValue")

    ibin = bmap.GetBinContent(plane+1, cell+1)

    br = bvalue.GetBinContent(int(ibin))

    print "Brightness level for plane {}, cell {} is {} in bin {}".format(plane, cell, br, ibin)

    for i in range(bvalue.GetNbinsX()+2):
        print "Brightness Bin: {} value {}".format(i, bvalue.GetBinContent(i))