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")