checkRpcDigits Namespace Reference

Classes
class	RpcDigit
	Representation of a single RPC digit. More...
class	RpcSDO

Functions
def	readDigitTree (in_file, treeName="HitValidTree")
def	readSDOTree (in_file, treeName="HitValidTree")

Variables
	parser = ArgumentParser(prog="checkRpcDigits", description="Script comparing the Rpc digitization validation n-tuples")
	help
	default
	type
	args = parser.parse_args()
def	refDigits = readDigitTree(args.refFile)
def	testDigits = readDigitTree(args.testFile)
def	refSDOs = readSDOTree(args.refFile)
def	testSDOs = readSDOTree(args.testFile)
	outFile = open(args.outFile, "w")
bool	allGood = True
	Loop over the SDOs & Digits. More...
def	testContent = testDigits[event]
string	errorStr
	digit

Function Documentation

readDigitTree()

def checkRpcDigits.readDigitTree	(		in_file,
			treeName = "HitValidTree"
	)

Definition at line 77 of file checkRpcDigits.py.

def readDigitTree(in_file, treeName = "HitValidTree"):
 
    t_file = TFile.Open(in_file, "READ")
    if not t_file:
        print(f"Failed to open {in_file}")
        exit(1)
    tree = t_file.Get(treeName) 
    if not tree:
        print("Failed to read tree {treeName} from {file}".format(treeName=treeName, file=t_file.GetName()))
        exit(1)
    allDigits = {}
    for entry in range(tree.GetEntries()):
        tree.GetEntry(entry)
        evt= tree.eventNumber
        digits = []
        for digit in range(len(tree.Digits_RPC_stationIndex)):
            digits.append(RpcDigit(stationIndex = ord(tree.Digits_RPC_stationIndex[digit]),
                                   stationEta =  ord(tree.Digits_RPC_stationEta[digit]),
                                   stationPhi = ord(tree.Digits_RPC_stationPhi[digit]),
                                   doubletR = ord(tree.Digits_RPC_doubletR[digit]),
                                   doubletPhi= ord(tree.Digits_RPC_doubletPhi[digit]),
                                   gasGap = ord(tree.Digits_RPC_gasGap[digit]),
                                   measuresPhi=tree.Digits_RPC_measuresPhi[digit],
                                   strip = ord(tree.Digits_RPC_strip[digit]),
                                   globPosX= tree.Digits_RPC_globalPosX[digit],
                                   globPosY= tree.Digits_RPC_globalPosY[digit],
                                   globPosZ= tree.Digits_RPC_globalPosZ[digit],
                                   
                                   locPosX = tree.Digits_RPC_localPosX[digit],
                                   locPosY = tree.Digits_RPC_localPosY[digit],
                                   time = tree.Digits_RPC_time[digit]
                                   ))
            if digits[-1].identify().stationName() == "BIS": 
                digits.pop()
        if len (digits): allDigits[evt] = digits
    return allDigits
 

readSDOTree()

def checkRpcDigits.readSDOTree	(		in_file,
			treeName = "HitValidTree"
	)

Definition at line 114 of file checkRpcDigits.py.

def readSDOTree(in_file, treeName = "HitValidTree"):
    allSDOS = {}    
    t_file = TFile.Open(in_file, "READ")
    if not t_file:
        print(f"Failed to open {in_file}")
        exit(1)
    tree = t_file.Get(treeName) 
    if not tree:
        print("Failed to read tree {treeName} from {file}".format(treeName=treeName, file=t_file.GetName()))
        exit(1)
    allSDOS = {}
    for entry in range(tree.GetEntries()):
        tree.GetEntry(entry)
        evt= tree.eventNumber
 
        SDOs = []
        for sdo in range(len(tree.SDO_RPC_stationIndex)):
            SDOs.append(RpcSDO(stationIndex = ord(tree.SDO_RPC_stationIndex[sdo]),
                               stationEta =  ord(tree.SDO_RPC_stationEta[sdo]),
                               stationPhi = ord(tree.SDO_RPC_stationPhi[sdo]),
                               doubletR = ord(tree.SDO_RPC_doubletR[sdo]),
                               doubletPhi = ord(tree.SDO_RPC_doubletPhi[sdo]),
                               gasGap = ord(tree.SDO_RPC_gasGap[sdo]),
                               measuresPhi=tree.SDO_RPC_measuresPhi[sdo],
                               strip = ord(tree.SDO_RPC_strip[sdo]),
                               globPosX = tree.SDO_RPC_globalPosX[sdo],
                               globPosY = tree.SDO_RPC_globalPosY[sdo],
                               globPosZ = tree.SDO_RPC_globalPosZ[sdo],
                               
                               locPosX = tree.SDO_RPC_localPosX[sdo],
                               locPosY = tree.SDO_RPC_localPosY[sdo],
                               #time = tree.SDO_RPC_global_time[sdo],
                               barcode=tree.SDO_RPC_barcode[sdo]
                               ))
 
        if len(SDOs): 
            allSDOS[evt] = SDOs
 
    return allSDOS
 

Variable Documentation

allGood

bool checkRpcDigits.allGood = True

Loop over the SDOs & Digits.

Definition at line 171 of file checkRpcDigits.py.

args

checkRpcDigits.args = parser.parse_args()

Definition at line 160 of file checkRpcDigits.py.

default

checkRpcDigits.default

Definition at line 157 of file checkRpcDigits.py.

digit

checkRpcDigits.digit

Definition at line 186 of file checkRpcDigits.py.

errorStr

string checkRpcDigits.errorStr

Initial value:

= "In Event {event}, {digit} cannot be found in the test sample. Potential candidates: \n".format(digit = str(refObj),
                                                                                                                          event = event)

Definition at line 182 of file checkRpcDigits.py.

help

checkRpcDigits.help

Definition at line 157 of file checkRpcDigits.py.

outFile

checkRpcDigits.outFile = open(args.outFile, "w")

Definition at line 168 of file checkRpcDigits.py.

parser

checkRpcDigits.parser = ArgumentParser(prog="checkRpcDigits", description="Script comparing the Rpc digitization validation n-tuples")

Definition at line 156 of file checkRpcDigits.py.

refDigits

def checkRpcDigits.refDigits = readDigitTree(args.refFile)

Definition at line 162 of file checkRpcDigits.py.

refSDOs

def checkRpcDigits.refSDOs = readSDOTree(args.refFile)

Definition at line 165 of file checkRpcDigits.py.

testContent

def checkRpcDigits.testContent = testDigits[event]

Definition at line 176 of file checkRpcDigits.py.

testDigits

def checkRpcDigits.testDigits = readDigitTree(args.testFile)

Definition at line 163 of file checkRpcDigits.py.

testSDOs

def checkRpcDigits.testSDOs = readSDOTree(args.testFile)

Definition at line 166 of file checkRpcDigits.py.

type

checkRpcDigits.type

Definition at line 157 of file checkRpcDigits.py.