histList[eta] = rt.TH1F(histname,histtitle,1000,-30,30)
a.openEB(nof,rootfilename,fileList, p.numberofEntries, histList, 0)
# Same procedure, going back to directory where results are printed
retdir = os.getcwd()
print "Current working directory %s" % retdir
os.chdir(p.resultPathLocal + '/' + p.folderName + '/')
folder = 'ctEB'+ datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
os.system('mkdir ' + folder)
os.chdir(os.getcwd() + '/' + folder +'/')
retdir = os.getcwd()
print "Directory changed successfully %s" % retdir
#fits the histograms and saves 1D in tree
if p.splitPhotons == True:
htime1, fitdata1 = snf.fitTimeEta(histList1,htime1)
htime2, fitdata2 = snf.fitTimeEta(histList2,htime2)
#saving all 1D histograms in tree
a.saveEB(p.runNumber,dataList1,dataList2,histList1,histList2,htime1,htime2,fitdata1,fitdata2)
#saving all data into a numpy file for analyzing later
np.save("A_etaTimeResponseEB1_0T.npy", dataList1)
np.save("A_etaTimeResponseEB2_0T.npy", dataList2)
#Tacks on histogram to canvas frame and ouputs on canvas
a.printPrettyPictureEB(p.runNumber,htime1,htime2)
else:
htime, fitdata = snf.fitTimeEta(histList,htime)
a.saveEB(p.runNumber,dataList,0,histList,0,htime,0,fitdata,0)
np.save("etaTimeResponseEBAll_0T.npy", dataList)
f.close()
newscript = filedata.replace("p.runNumber", str(p.runNumber))
f = open('fast_restack.py','w')
f.write(newscript)
f.close()
#returns to the result folder made
os.chdir(retdir)
#saving run info to a numpy file for reference later
np.save(p.runNumber+"EtaRunInfoEB.npy", runinfo)
#fits the histograms and saves 1D in tree
if p.splitPhotons == True:
htime1, hlaser1, fitdata1, seedmap1 = snf.fitTimeEta(histList1, transList1, htime1, hlaser1, p.minStat, p.minNormal, p.includeSeedMap, p.manualHitCounterCut, "p1_", p.graphs2printEB)
htime2, hlaser2, fitdata2, seedmap2 = snf.fitTimeEta(histList2, transList2, htime2, hlaser2, p.minStat, p.minNormal, p.includeSeedMap, p.manualHitCounterCut, "p2_", p.graphs2printEB)
#saving all 1D histograms in tree
a.saveEB(p.runNumber,dataList1,dataList2,histList1,histList2,transList1,transList2,htime1,htime2,hlaser1,hlaser2,fitdata1,fitdata2,seedmap1,seedmap2)
#Tacks on histogram to canvas frame and ouputs on canvas
a.printPrettyPictureEB(p.runNumber,htime1,htime2,hlaser1,hlaser2,seedmap1,seedmap2)
else:
htime, hlaser, fitdata, seedmap = snf.fitTimeEta(histList, transList, htime, hlaser, p.minStat, p.minNormal, p.includeSeedMap, p.manualHitCounterCut, "c_", p.graphs2printEB)
a.saveEB(p.runNumber,dataList,0,histList,0,transList,0,htime,0,hlaser,0,fitdata,0,seedmap,0)
a.printPrettyPictureEB(p.runNumber,htime,0,hlaser,0,seedmap,0)
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
# pbar = progressbar("saving data", 171).start()
pbar = progressbar("saving data", 342).start()
#saving all 1D histograms in tree
# f = rt.TFile("clustertimeEBAll_"+str(int(time.time()))+".root","new")
