def openEB(numofFiles, filename, fileList, entries, histList1, histList2):
for k in range(0, numofFiles):
if filename in fileList[k]:
rootFile = rt.TFile.Open(fileList[k])
rTree = rootFile.Get("Tree_Optim")
print "successfully cut branch from " + fileList[k]
#rootFile.Print("v")
#fills the histogram with data
if isinstance(histList1[0],list) == True: #Individual barrel crystals
if histList2 !=0:
histList1,histList2 = snf.stackTime(rTree, entries, histList1, histList2, 1, 0)
else:
histList = snf.stackTime(rTree, entries, histList, 0, 0, 0)
else: #eta baby eta
if histList2!=0:
histList1, histList2 = snf.stackTimeEta(rTree, entries, histList1,histList2)
else:
histList1 = snf.stackTimeEta(rTree,entries,histList1,0)
rootFile.Close()
def openEB(filename, fileList, runinfo, startfilepos, endfilepos, entries, histList1, histList2, transList1, transList2):
rTree = rt.TChain("Tree_Optim")
for k in range(startfilepos, endfilepos):
if filename in fileList[k]:
rTree.Add(fileList[k])
print "successfully cut branch from " + fileList[k]
#Saving run info in tuple list
runinfo = np.vstack((runinfo, [fileList[k]]))
#fills the histogram with data
if isinstance(histList1[0],list) == True: #Individual barrel crystals
if histList2 !=0:
histList1, histList2, transList1, transList2 = snf.stackTime(rTree, entries, histList1, histList2, 1, 0, transList1,transList2, 0, 0)
else:
histList1, transList1 = snf.stackTime(rTree, entries, histList1, 0, 0, 0, transList1, 0, 0, 0)
else: #eta baby eta
if histList2!=0:
histList1, histList2, transList1, transList2 = snf.stackTimeEta(rTree, entries, histList1, histList2, transList1, transList2)
else:
histList1, transList1 = snf.stackTimeEta(rTree, entries, histList1, 0, transList1, 0)
return runinfo
print "Directory changed successfully %s" % retdir
# Get list of root files in directory
fileList = os.listdir(retdir)
for k in range(0,len(fileList)):
if "2015A_EcalNtp_" in fileList[k]:
rootFile = rt.TFile.Open(fileList[k])
rTree = rootFile.Get("Tree_Optim")
print "successfully cut branch from " + fileList[k]
#rootFile.Print("v")
#fills the histogram with data
#histList = snf.stackTimeEta(rTree,histList,0)
histList1, histList2 = snf.stackTimeEta(rTree,histList1,histList2)
rootFile.Close()
#fits the histograms and saves 1D in tree
# htime = snf.fitTimeEta(histList,htime)
htime1 = snf.fitTimeEta(histList1,htime1)
htime2 = snf.fitTimeEta(histList2,htime2)
# Same procedure, going back to original working directory.
retdir = os.getcwd()
print "Current working directory %s" % retdir
os.chdir( startdir + "/result/")
retdir = os.getcwd()
print "Directory changed successfully %s" % retdir
#Progress bar for the saves
