def draw_box(self): z1 = self.center_z - self.length / 2 z2 = z1 + self.length x1 = self.center_x - self.rad2 x2 = x1 + 2 * self.rad2 #to cm x1 *= 100 x2 *= 100 #representation as a box self.box = TFrame(z1, x1, z2, x2) self.box.SetBorderMode(0) self.box.SetFillColor(rt.kGray + 1) col = rt.kRed if self.is_electron == True: col = rt.kBlue self.box.SetLineColor(col) self.box.SetLineWidth(2) #self.box.Draw("same") #label lx = x2 + 2 align = 11 if lx < 0 and lx > -62: lx = x1 - 5 # negative lx align = 31 if self.center_z < 0 and self.center_x < 0.1: lx = x1 - 2 # negative z and x align = 31 if lx > 0 and lx < 22: lx = 22 # small positive lx self.label = TText(z2, lx, self.name) self.label.SetTextSize(0.03) self.label.SetTextAngle(90) self.label.SetTextAlign(align)
def drawTitle(text): '''Takes in the title text and draws it to the top of the current pad''' headerText = TText(.5, .95, text) headerText.SetTextAlign(22) headerText.Draw() return headerText
def rooFit106(): print ">>> setup model..." x = RooRealVar("x", "x", -3, 3) mean = RooRealVar("mean", "mean of gaussian", 1, -10, 10) sigma = RooRealVar("sigma", "width of gaussian", 1, 0.1, 10) gauss = RooGaussian("gauss", "gauss", x, mean, sigma) data = gauss.generate(RooArgSet(x), 10000) # RooDataSet gauss.fitTo(data) print ">>> plot pdf and data..." frame = x.frame(Name("frame"), Title("RooPlot with decorations"), Bins(40)) # RooPlot data.plotOn(frame) gauss.plotOn(frame) print ">>> RooGaussian::paramOn - add box with pdf parameters..." # https://root.cern.ch/doc/master/classRooAbsPdf.html#aa43b2556a1b419bad2b020ba9b808c1b # Layout(Double_t xmin, Double_t xmax, Double_t ymax) # left edge of box starts at 20% of x-axis gauss.paramOn(frame, Layout(0.55)) print ">>> RooDataSet::statOn - add box with data statistics..." # https://root.cern.ch/doc/master/classRooAbsData.html#a538d58020b296a1623323a84d2bb8acb # x size of box is from 55% to 99% of x-axis range, top of box is at 80% of y-axis range) data.statOn(frame, Layout(0.20, 0.55, 0.8)) print ">>> add text and arrow..." text = TText(2, 100, "Signal") text.SetTextSize(0.04) text.SetTextColor(kRed) frame.addObject(text) arrow = TArrow(2, 100, -1, 50, 0.01, "|>") arrow.SetLineColor(kRed) arrow.SetFillColor(kRed) arrow.SetLineWidth(3) frame.addObject(arrow) print ">>> persist frame with all decorations in ROOT file..." file = TFile("rooFit106.root", "RECREATE") frame.Write() file.Close() # To read back and plot frame with all decorations in clean root session do # [0] TFile f("rooFit106.root") # [1] xframe->Draw() print ">>> draw functions and toy data on canvas..." canvas = TCanvas("canvas", "canvas", 100, 100, 800, 600) gPad.SetLeftMargin(0.15) gPad.SetRightMargin(0.05) frame.GetYaxis().SetTitleOffset(1.6) frame.GetYaxis().SetLabelOffset(0.010) frame.GetYaxis().SetTitleSize(0.045) frame.GetYaxis().SetLabelSize(0.042) frame.GetXaxis().SetTitleSize(0.045) frame.GetXaxis().SetLabelSize(0.042) frame.Draw() canvas.SaveAs("rooFit106.png")
def timeStampCanvas(canvas): import rootpy from rootpy.plotting import Canvas from ROOT import TText import datetime # apply time stamp: text = TText() canvas.cd(0) #select canvas text.SetTextSize(0.015) text.DrawTextNDC(.7, 0.008, datetime.datetime.now().strftime("%A, %d. %B %Y %I:%M%p"))
def markPad(text, x=0.14, y=0.007, size=0.013, color=1, canvas=None, npad=-1): """ Puts a line of text on a specific canvas and pad. If none are specified, the current pad is used """ import rootpy from rootpy.plotting import Canvas from ROOT import TText t = TText() if npad >= 0 and canvas: canvas.cd(npad) #select pad (0 = whole canvas) t.SetTextSize(size) t.SetTextColor(color) t.DrawTextNDC(x, y, text)
def D1H_txt_to_root(filename, outputpath=''): from ROOT import TFile, TCanvas, TPad, TH1D, TLatex, TStyle, gStyle, TText, gPad, TPaveText from inspect import currentframe, getframeinfo #gStyle.SetOptStat(0) can = TCanvas("can", "can", 200, 10, 500, 500) fileroot = filename.replace(".txt", "") fileroot = fileroot + "_F.root" f = open(filename, "r") lineList = f.readlines() Nbin = (len(lineList)) # get number of bins Line_string = str(lineList[0]) _, bin_init, _, _ = Line_string.split() bin_init = float(bin_init) # get initial bin Line_string = str(lineList[len(lineList) - 1]) _, _, bin_final, _ = Line_string.split() bin_final = float(bin_final) # get final bin f.seek(0) # reset python read line hist = TH1D("h1f", "h1f", Nbin, bin_init, bin_final) total_e = 0 for i in range(1, Nbin + 1): Line_string = str(f.readline()) _, _, _, bin_c = Line_string.split() bin_c = float(bin_c) hist.SetBinContent(i, bin_c) total_e = total_e + bin_c total_e = int(total_e) hist.Draw() text = TText(hist.GetXaxis().GetBinCenter(2), hist.GetYaxis().GetBinCenter(1), "Recycled. Total Entry : %i" % total_e) text.SetTextFont(10) text.Draw() gPad.Update() can.Update() if (outputpath == ''): wf = TFile(fileroot, "RECREATE") print(fileroot, " root file is generated !!!") else: fileroot = outputpath + "/_processed.root" wf = TFile(fileroot, "RECREATE") print(fileroot, " root file is generated !!!") hist.Write() wf.Close() return fileroot
def draw_2d(self): #make corner points vec = [] vec.append(TVector2(self.length / 2, -self.height / 2)) vec.append(TVector2(-self.length / 2, -self.height / 2)) vec.append(TVector2(-self.length / 2, self.height / 2)) vec.append(TVector2(self.length / 2, self.height / 2)) #rotate and translate along electron beam vtrans = TVector2(-1, 0).Rotate(self.angle) vtrans.SetMagPhi(self.dist + self.length / 2, vtrans.Phi()) for i in xrange(len(vec)): vec[i] = vec[i].Rotate(self.angle) vec[i] += vtrans #last point same as the first vec.append(vec[0]) self.geom = TGraph(len(vec)) self.geom.SetLineWidth(2) self.geom.SetLineColor(rt.kYellow + 1) self.geom.SetFillColor(rt.kYellow) for i in xrange(len(vec)): self.geom.SetPoint(i, vec[i].X(), vec[i].Y() * 100) self.geom.Draw("lfsame") #label self.label = TText(vtrans.X(), (vtrans.Y() - self.height / 2 - 0.11) * 100 - 3, "Lumi detector") self.label.SetTextSize(0.03) #self.label.SetTextAngle(90) #self.label.SetTextAlign(32) self.label.SetTextAlign(23) self.label.Draw("same") #label for low Q^2 tagger self.label_tag = TText(vtrans.X() + 2, (vtrans.Y() + 0.6) * 100 - 3, "Place for low-Q2 tagger") self.label_tag.SetTextSize(0.03) #self.label.SetTextAngle(90) #self.label.SetTextAlign(32) self.label_tag.SetTextAlign(23) self.label_tag.Draw("same")
def prepare_canvas(c, channels, pad_label=None, x_label=None, y_label=None): pads = dict() # for axis labels c.Divide(1, 1, 0.03, 0.03) c.cd(1).Divide(*divide_square(len(channels))) for i, chans in enumerate(channels): ncol, nrow = divide_square(len(chans)) c.cd(1).cd(i + 1).Divide(nrow, ncol, 0., 0.) # share y axis in the sub-group ymax = 0 for j, ch in enumerate(chans): pads[ch] = c.cd(1).cd(i + 1).cd(j + 1) # labels c.cd(0) htext, vtext = TText(), TText() htext.SetTextSize(0.03) htext.SetTextAlign(22) vtext.SetTextSize(0.03) vtext.SetTextAlign(22) vtext.SetTextAngle(90) if pad_label: htext.DrawText(0.5, 0.982, pad_label) if x_label: htext.DrawText(0.5, 0.015, x_label) if y_label: vtext.DrawText(0.015, 0.5, y_label) return pads
def drawSignature(signature): """Write an unique identifier in the lower left canvas corner""" l = TText() l.SetTextAlign(11) l.SetTextSize(0.02) l.SetTextFont(82) l.DrawTextNDC(0.01, 0.01, signature)
def plotNSamples(cls, npatterns, etBins, etaBins, outname='nPatterns.pdf'): """Plot number of samples per bin""" logger = Logger.getModuleLogger("PlotNSamples") from ROOT import TCanvas, gROOT, kTRUE, kFALSE, TH2I, TText gROOT.SetBatch(kTRUE) c1 = TCanvas("plot_patterns_signal", "a", 0, 0, 800, 400) c1.Draw() shape = [len(etBins) - 1, len(etaBins) - 1] histo1 = TH2I( "text_stats", "#color[4]{Signal}/#color[2]{Background} available statistics", shape[0], 0, shape[0], shape[1], 0, shape[1]) #histo1 = TH2I("text_stats", "Signal/Background available statistics", shape[0], 0, shape[0], shape[1], 0, shape[1]) histo1.SetStats(kFALSE) histo1.Draw("TEXT") histo1.SetXTitle("E_{T}") histo1.SetYTitle("#eta") histo1.GetXaxis().SetTitleSize(0.04) histo1.GetYaxis().SetTitleSize(0.04) histo1.GetXaxis().SetLabelSize(0.04) histo1.GetYaxis().SetLabelSize(0.04) histo1.GetXaxis().SetTickSize(0) histo1.GetYaxis().SetTickSize(0) ttest = TText() ttest.SetTextAlign(22) for etBin in range(shape[0]): for etaBin in range(shape[1]): key = 'et%d_eta%d' % (etBin, etaBin) ttest.SetTextColor(4) ttest.DrawText(.5 + etBin, .75 + etaBin, 's: ' + str(npatterns['sgnPattern_' + key])) ttest.SetTextColor(2) ttest.DrawText(.5 + etBin, .25 + etaBin, 'b: ' + str(npatterns['bkgPattern_' + key])) try: histo1.GetYaxis().SetBinLabel( etaBin + 1, '#bf{%d} : %.2f->%.2f' % (etaBin, etaBins[etaBin], etaBins[etaBin + 1])) except Exception: logger.error("Couldn't retrieve eta bin %d bounderies.", etaBin) histo1.GetYaxis().SetBinLabel(etaBin + 1, str(etaBin)) try: histo1.GetXaxis().SetBinLabel( etBin + 1, '#bf{%d} : %d->%d [GeV]' % (etBin, etBins[etBin], etBins[etBin + 1])) except Exception: logger.error("Couldn't retrieve et bin %d bounderies.", etBin) histo1.GetXaxis().SetBinLabel(etBin + 1, str(etaBin)) c1.SetGrid() c1.Update() c1.SaveAs(outname)
def writeCurrentFile(rootFile): ''' Save a copy of the current script, as a TText, in the current ROOT file nothing happens if a ROOT file is not currently open! ''' import sys, os.path from ROOT import TString, TText, TFile filename = sys.argv[0] if (not os.path.exists(filename)): m_log.error("Could not open file " + filename + "!") return s = TString() t = TText() # Too fresh #with open(filename) as input: input = open(filename) if input: for line in input: s += line t.SetName(os.path.basename(filename)) t.SetText(0, 0, s.Data()) f = TFile(rootFile, 'UPDATE') t.Write() f.Close()
def draw_2d(self): #draw the exit window self.geom = TGraph(len(self.zx_pos)) self.geom.SetLineColor(rt.kGreen + 1) self.geom.SetLineWidth(4) ipoint = 0 for i in self.zx_pos: self.geom.SetPoint(ipoint, i.X(), 100 * i.Y()) ipoint += 1 self.geom.Draw("lsame") #label zpos = (self.zx_pos[0].X() + self.zx_pos[1].X()) / 2. self.label = TText(zpos, (self.zx_pos[0].Y()) * 100 - 6, "Exit window") self.label.SetTextSize(0.03) #self.label.SetTextAngle(90) #self.label.SetTextAlign(32) self.label.SetTextAlign(23)
def invert_col(self, pad): bgcol = rt.kBlack fgcol = rt.kOrange - 3 pad.SetFillColor(bgcol) pad.SetFrameLineColor(fgcol) next = TIter(pad.GetListOfPrimitives()) obj = next() while obj != None: #TText if obj.InheritsFrom(TText.Class()) == True: if obj.GetTextColor() == rt.kBlack: obj.SetTextColor(fgcol) #H1 if obj.InheritsFrom(TH1.Class()) == True: if obj.GetLineColor() == rt.kBlack: obj.SetLineColor(fgcol) obj.SetFillColor(bgcol) if obj.GetMarkerColor() == rt.kBlack: obj.SetMarkerColor(fgcol) obj.SetAxisColor(fgcol, "X") obj.SetAxisColor(fgcol, "Y") obj.SetLabelColor(fgcol, "X") obj.SetLabelColor(fgcol, "Y") obj.GetXaxis().SetTitleColor(fgcol) obj.GetYaxis().SetTitleColor(fgcol) #TFrame if obj.InheritsFrom(TFrame.Class()) == True: if obj.GetLineColor() == rt.kBlack: obj.SetLineColor(fgcol) obj.SetFillColor(bgcol) #print obj.GetName(), obj.ClassName() #move to the next item obj = next()
def plotL1AndTightL1Counters(self): hEvent = self.fileHandler.getHistogram('count/Events_Count') hAllL1 = self.fileHandler.getHistogram('count/L1Muon_Count') hAllL13x3 = self.fileHandler.getHistogram('count/L1Muon3x3_Count') hTightL1 = self.fileHandler.getHistogram('count/L1TightMuons_Count') hTightL13x3 = self.fileHandler.getHistogram( 'count/L1TightMuons3x3_Count') hL1 = self.fileHandler.getHistogram('count/energyDeposit_L1_Count') hL1Reco = self.fileHandler.getHistogram( 'count/energyDeposit_L1Reco_Count') hL1RecoHo = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoHo_Count') hL1RecoHoTight = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoHoTight_Count') hL1RecoTight = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoTight_Count') hL1RecoTightHo = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoTightHo_Count') hL1RecoHoNoThr = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoHoNoThr_Count') hL1RecoGaHoNoThr = self.fileHandler.getHistogram( 'count/energyDeposit_L1RecoGaHoNoThr_Count') histogramList = [ hEvent, hL1, hL1Reco, hL1RecoHo, hL1RecoHoTight, hL1RecoTight, hL1RecoTightHo, hL1RecoHoNoThr, hL1RecoGaHoNoThr ] names = [ 'hEvent', 'hL1', 'hL1Reco', 'hL1RecoHo', 'hL1RecoHoTight', 'hL1RecoTight', 'hL1RecoTightHo', 'hL1RecoHoNoThr', 'hL1RecoGaHoNoThr' ] nL1 = hL1.GetBinContent(2) self.commandLine.output( '###############################################') for i, h in enumerate(histogramList): self.commandLine.output('%-20s:%8d\t=> %6.2f%% +/- %5.2f%%' % (names[i], h.GetBinContent(2), calcPercent(h.GetBinContent(2), nL1), calcSigma(h.GetBinContent(2), nL1) * 100)) self.commandLine.output( '###############################################') nL1Reco = hL1Reco.GetBinContent(2) nL1RecoHo = hL1RecoHo.GetBinContent(2) nL1RecoHoTight = hL1RecoHoTight.GetBinContent(2) nL1RecoTight = hL1RecoTight.GetBinContent(2) nL1RecoTightHo = hL1RecoTightHo.GetBinContent(2) N_BINS = 4 binContents = [nL1, nL1Reco, nL1RecoHo, nL1RecoHoTight] binLabels = [ 'L1', 'L1 -> Reco', 'L1 + R -> HO', 'L1 + R + HO -> tight' ] binContentsInverted = [nL1, nL1Reco, nL1RecoTight, nL1RecoTightHo] binLabelsInverted = [ 'L1', 'L1 -> Reco', 'L1 + R -> tight', 'L1 + R + tight -> HO' ] c = TCanvas('cL1AndTightL1Count', 'L1AndTightL1Count') h = TH1D('hL1AndTightL1Count', 'L1 Cutflow', 4, -0.5, N_BINS - .5) hInverted = TH1D('hL1AndTightL1CountInverted', 'L1 Efficiency', 4, -0.5, N_BINS - .5) hInverted.SetFillStyle(3002) hInverted.SetFillColor(colorRwthOrange) hInverted.SetLineColor(colorRwthOrange) hInverted.SetLineStyle(3) hL13x3Alone = TH1D('hL1And3x3Alone', '', 1, 1.5, 2.5) hL13x3Alone.SetBinContent(1, nL1RecoHo / nL1Reco) hL13x3Alone.SetBinError(1, calcSigma(nL1RecoHo, nL1Reco)) hL13x3Alone.SetLineColor(colorRwthMagenta) hTightL13x3Alone = TH1D('hTightL1And3x3Alone', '', 1, 2.5, 3.5) hTightL13x3Alone.SetBinContent(1, nL1RecoHoTight / nL1RecoHo) hTightL13x3Alone.SetBinError(1, calcSigma(nL1RecoHoTight, nL1RecoHo)) hTightL13x3Alone.SetLineColor(colorRwthTuerkis) for i in range(2, N_BINS + 1): h.SetBinContent(i, binContents[i - 1] / binContents[1]) h.GetXaxis().SetBinLabel(i, binLabels[i - 1]) hInverted.SetBinContent( i, binContentsInverted[i - 1] / binContentsInverted[1]) hInverted.GetXaxis().SetBinLabel(i, binLabelsInverted[i - 1]) h.GetXaxis().SetBinLabel(1, 'L1') h.SetBinContent(1, 1) hInverted.GetXaxis().SetBinLabel(1, 'L1') hInverted.SetBinContent(1, 1) h.SetLineColor(colorRwthDarkBlue) h.SetStats(0) h.GetYaxis().SetTitle('rel. fraction') h.Draw() # hL13x3Alone.Draw('same e') # hTightL13x3Alone.Draw('same e') hInverted.Draw('same') hInverted.GetXaxis().Draw('same') setupAxes(h) legend = getLegend(y2=.9, x1=.55) legend.AddEntry(h, 'First match HO then use tight', 'l') # legend.AddEntry(hL13x3Alone,'3x3 matching normed to # L1 + R','le') # legend.AddEntry(hTightL13x3Alone,'Normed to # L1 + R + HO','l') legend.AddEntry(hInverted, 'Inverted order for HO and tight', 'f') legend.Draw() label = self.drawLabel() textObjects = [] #for (Int_t i=1;i<=30;i++) t.DrawText(h->GetBinCenter(i),yt,Form("%d",i%10)); for i in range(1, 4): t = TText() t.SetTextSize(0.025) t.SetTextAlign(22) t.SetTextColor(colorRwthOrange) t.DrawTextNDC(getXinNDC(hInverted.GetBinCenter(i + 1)), 0.05, binLabelsInverted[i]) # Double_t yt = - h->GetMaximum()/15.; textObjects.append(t) c.Update() self.storeCanvas(c, 'l1AndTightL1Counters') return h, c, hL13x3Alone, hTightL13x3Alone, label, legend, hInverted, textObjects
from ROOT import TCanvas, TF1, TPaveLabel, TPad, TText from ROOT import gROOT nut = TCanvas('nut', 'FirstSession', 100, 10, 700, 900) nut.Range(0, 0, 20, 24) nut.SetFillColor(10) nut.SetBorderSize(2) pl = TPaveLabel(3, 22, 17, 23.7, 'My first PyROOT interactive session', 'br') pl.SetFillColor(18) pl.Draw() t = TText(0, 0, 'a') t.SetTextFont(62) t.SetTextSize(0.025) t.SetTextAlign(12) t.DrawText( 2, 20.3, 'PyROOT provides ROOT bindings for Python, a powerful interpreter.') t.DrawText(2, 19.3, 'Blocks of lines can be entered typographically.') t.DrawText(2, 18.3, 'Previous typed lines can be recalled.') t.SetTextFont(72) t.SetTextSize(0.026) t.DrawText(3, 17, r'>>> x, y = 5, 7') t.DrawText(3, 16, r'>>> import math; x*math.sqrt(y)') t.DrawText( 3, 14, r'>>> for i in range(2,7): print "sqrt(%d) = %f" % (i,math.sqrt(i))') t.DrawText(3, 10, r'>>> import ROOT; f1 = ROOT.TF1( "f1", "sin(x)/x", 0, 10 )')
line = TLine( 2.6, 11, 2.6, 11.5 ) line.Draw() line.DrawLine( 2.6, 11.5, 7, 11.5 ) arrow = TArrow( 7, 11.5, 7, 11.1, 0.01, '|>' ) arrow.SetFillStyle( 1001 ) arrow.Draw() line.DrawLine( 7, 8.5, 7, 8.0 ) line.DrawLine( 7, 8.0, 10.6, 8 ) arrow.DrawArrow( 10.6,8, 10.6, 8.4, 0.01, '|>' ) line.DrawLine( 10.6, 11, 10.6, 11.5 ) line.DrawLine( 10.6, 11.5, 14.6, 11.5 ) arrow.DrawArrow( 14.6, 11.5, 14.6, 11.1, 0.01, '|>' ) line.DrawLine( 14.6, 8.5, 14.6, 8.0 ) line.DrawLine( 14.6, 8.0, 16, 8 ) ldot.DrawLine( 16, 8, 19, 8 ) vert = TText( 1.5, 9.75, 'File' ) vert.SetTextAlign( 21 ) vert.SetTextAngle( 90 ) vert.SetTextSize( 0.025 ) vert.Draw() vert.DrawText( 2.0, 9.75, 'Header' ) vert.DrawText( 2.9, 9.75, 'Logical Record' ) vert.DrawText( 3.2, 9.75, 'Header' ) vert.DrawText( 7.3, 9.75, 'Logical Record' ) vert.DrawText( 7.6, 9.75, 'Header' ) vert.DrawText( 10.9, 9.75, 'Logical Record' ) vert.DrawText( 11.2, 9.75, 'Header' ) vert.DrawText( 14.9, 9.75, 'Logical Record' ) vert.DrawText( 15.2, 9.75, 'Header' ) hori = TText( 4.75, 10, 'Object' ) hori.SetTextAlign( 22 )
if cpu_type == '2376': h_2_6.Fill(float(t_run)/60.) c1 = TCanvas("c1","Root Canvas",40,20,800,600) gStyle.SetOptStat(111111) h_1.SetFillColor(36) h_1.GetYaxis().SetTitleOffset(1.2) h_1.SetYTitle('Entries') h_1.SetXTitle('Time, s') c1.SetLogy() h_1.Draw() t_task = TText(0.35,0.8,'Task '+ task + " SIGNET" ) t_task.SetNDC() t_task.Draw() #print time stamp t_time = datetime.now() t_label = 'SP ' + t_time.strftime("%Y-%m-%d") t = TText(0.905,0.6,t_label) t.SetNDC() t.SetTextAlign(21) # middle, bottom t.SetTextAngle(-90) t.SetTextSize(0.017) t.Draw() c1.Modified()
def main(): cc.cd() cc.SetBorderMode(0) cc.SetFixedAspectRatio(1) cc.FeedbackMode(1) gStyle.SetOptStat(0) gStyle.SetGridStyle(1) gStyle.SetGridColor(11) hh=TH2D('hh',';X;Y',22,-10.5,11.5,22,-10.5,11.5) hi=TH2I('hi',';X;Y',22,-10.5,11.5,22,-10.5,11.5) setupHists([hh,hi]) xax,yax=hh.GetXaxis(),hh.GetYaxis() hh.Draw('COLZ') hi.Draw('TEXTSAME') gPad.SetLogz() gPad.SetGrid(1,1) gPad.SetLeftMargin(0.09) gPad.SetRightMargin(0.11) #tt2=TPaveText(0.7,0.96,0.9,0.99,'NDC') ttM=TPaveText(-3+0.05, 7-4.45, 4.0, 8-4.51) tt2=TPaveText(-3+0.05, 7-5.45, 4.0, 8-5.51) ttX=TPaveText(-2, 7-8.00, 3, 8-8.00) ttY=TPaveText(-2, 7-9.00, 3, 8-9.00) ttZ=TPaveText(-2, 6-8.80, 3, 8-9.30) ttZ.AddText("positive = beam top/right") ttime=TPaveText(-10,-12.5,10,-11.8) tchan=TPaveText(0,0,0.9,1) setupPaveTexts([tt2,ttM,ttime,tchan,ttX,ttY,ttZ]) ttM.SetTextColor(2) ttM.SetFillStyle(0) ttZ.SetFillStyle(0) tt2.SetFillStyle(0) tarrow=TText(-0.9,0.7,'Beam Right') tarrow.SetTextSizePixels(15) arrow=TArrow(-1.4,0.5,2.4,0.5,0.02,'|>') arrow.SetAngle(40) arrow.SetFillColor(1) arrow.SetLineWidth(2) tt=TText() tt.SetTextColor(1) tt.SetTextAngle(90) tt.SetTextSize(0.04) tt.DrawText(12.4,0,'kHz') tt.SetTextAngle(0) tt.SetTextColor(1) tt.DrawTextNDC(0.3,0.92,'FTC FADC SCALERS') bb=TBox() bb.SetFillStyle(1001) bb.SetFillColor(0) bb.SetLineWidth(1) bb.SetLineColor(1) bb.DrawBox(-3.47,-1.47,4.47,2.46) bb.DrawBox(-1.47,-3.47,2.49,4.47) bb.DrawBox(-2.47,-2.47,3.49,3.47) cc.cd() for xx in [ttM,tt2,ttime,arrow,tarrow,ttX,ttY,ttZ]: xx.Draw() cc2.cd() tchan.Draw('NDC') cc.cd() gPad.SetEditable(0) while True: for ch in ECAL.chans: loadPV(ch) ch=ch.vals xx,yy=ch['X'],ch['Y'] #if (ch['PVVAL']>10): # print xx,yy,ch['PVVAL'] # swap x to make it downstream view: xx=-xx #after, fix the fact x=0 / y=0 don't exists if xx<0: xx+=1 if yy<0: yy+=1 hh.SetBinContent(xax.FindBin(xx),yax.FindBin(yy),ch['PVVAL']) hi.SetBinContent(xax.FindBin(xx),yax.FindBin(yy),ch['PVVAL']) for xx in [ttime,tt2,ttM,ttX,ttY]: xx.Clear() [total,maximum,top,bottom,left,right]=calcRates(ECAL.chans) tt2.AddText('Total: %.1f MHz'%(total/1000)) ttM.AddText('Max: %.0f kHz'%(maximum)) if total>1e2: xasy = (right-left)/total yasy = (top-bottom)/total ttX.AddText('X-Asy: %+.1f%%'%(100*xasy)) ttY.AddText('Y-Asy: %+.1f%%'%(100*yasy)) else: ttX.AddText('X-Asy: N/A') ttY.AddText('Y-Asy: N/A') ttime.AddText(makeTime()) if not gPad: sys.exit() if gPad.GetEvent()==11: xy=pix2xy(gPad) ee=ECAL.findChannelXY(xy[0],xy[1]) if ee: tchan.Clear() tchan.AddText(printChannel(ee)) cc2.Modified() cc2.Update() elif gPad.GetEvent()==12: tchan.Clear() cc2.Modified() cc2.Update() cc.Modified() cc.Update() time.sleep(2)
gtexts = [] dlines = [] boxes = [] xmin = 0 for d_i, daughter in enumerate(daughters): lines = [] levels = daughter['Levels'] gamma = [] if 'Gamma' in daughter: gamma = daughter['Gamma'] ngamma = len(gamma) x_width = 2.0 * param['TextOffset'] + 2.0 * param['TextWidth'] + ( ngamma + 1) * param['width'] ttexts.append( TText(xmin + (ngamma + 1) * param['width'] / 2.0, param['TitleTextOffset'], str(daughter['Name']))) lstyles = {} lcolors = {} for level in levels: xmax = xmin + (ngamma + 1) * param['width'] lines.append( TLine(xmin, level['energy'], xmax, level['energy'])) texts.append( TText(xmin - param['TextOffset'], level['energy'] + param['TextFloat'], str(level['level']))) texts[-1].SetTextAlign(31) texts.append( TText(xmax + param['TextOffset'], level['energy'] + param['TextFloat'],
infile_T1t1t_1200_275_100 = TFile.Open(inputfile_T1t1t_1200_275_100) # Integrated luminosity in fb-1s intlumi = 19.712 # ABCD # set root styles plotTools.SetBoostStyle() tdrstyle.setTDRStyle() ######################################################## # Make comparison of gen level W pt hname = "h_MR_jet1ptg200" htitle = "" xt = "M_{R} (GeV)" legd4 = plotTools.ConstructLDict(0.35, 0.9, 0.6, 0.87) text1 = TText(0.73, 0.55, "T1t1t model") text1.SetNDC() text1.SetTextSize(0.04) hdict_bg = plotTools.ConstructHDict(infile_bg.Get(hname), name="Total background", color=rt.kCyan - 8, fillstyle=3003, linewidth=0, title=htitle, xtitle=xt) hdict_T1t1t_600_225_50 = plotTools.ConstructHDict( infile_T1t1t_600_225_50.Get(hname), name= "m_{#tilde{g}} = 600 GeV, m_{#tilde{t}_{1}} = 225 GeV, m_{#tilde{#chi}_{1}^{0}} = 50 GeV", #name="#splitline{m_{#tilde{g}} = 600 GeV, m_{#tilde{t}_{1}} = 225 GeV,}{m_{#tilde{#chi}_{1}^{0}} = 50 GeV}", color=rt.kCyan + 2,
def main(): cc.cd() cc.SetBorderMode(0) cc.SetFixedAspectRatio(1) cc.FeedbackMode(1) gStyle.SetOptStat(0) gStyle.SetGridStyle(0) gStyle.SetGridColor(18) nbinsX = 0 for ix in range(len(SECTORSPERREGION)): nbinsX += SECTORSPERREGION[ix] hh = TH2D('hh', ';Strip;Sector', STRIPSPERSECTOR, 0, STRIPSPERSECTOR, nbinsX, 0, nbinsX) setupHists([hh]) hh.Draw('COLZ') gPad.SetLogz() gPad.SetLeftMargin(0.09) gPad.SetRightMargin(0.11) tt2 = TPaveText(240, 43, 500, 45) ttime = TPaveText(100, -5.5, 412, -3) setupPaveTexts([tt2, ttime]) tt2.SetTextSize(0.03) lll = TLine() lll.SetLineColor(15) y1 = SECTORSPERREGION[0] y2 = SECTORSPERREGION[0] + SECTORSPERREGION[1] lll.DrawLine(0, y1, STRIPSPERSECTOR, y1) lll.DrawLine(0, y2, STRIPSPERSECTOR, y2) tt = TText() tt.SetTextColor(1) tt.SetTextAngle(90) tt.SetTextSize(0.04) tt.DrawText(532, 22, 'Hz') tt.SetTextSize(0.06) tt.SetTextAngle(0) tt.SetTextColor(1) tt.DrawTextNDC(0.1, 0.93, 'SVT Scalers') tt.SetTextSize(0.03) tt.DrawText(-42, 4, 'R1') tt.DrawText(-42, 16, 'R2') tt.DrawText(-42, 32, 'R3') cc.cd() for xx in [ttime, tt2]: xx.Draw() cc.cd() gPad.SetEditable(0) while True: iy = 0 for ch in SVT.chans: loadPV(ch) ch = ch.vals data = ch['PVVAL'] time2 = ch['PVTIMEVAL'] if time2 > 10: print 'More than 10 seconds since message: ' + ch['PVNAME'] for ii in range(512): data[ii] = 0 if iy < SECTORSPERREGION[0]: region = 1 sector = iy elif iy < SECTORSPERREGION[0] + SECTORSPERREGION[1]: region = 2 sector = iy - SECTORSPERREGION[0] else: region = 3 sector = iy - SECTORSPERREGION[0] - SECTORSPERREGION[1] if data == None or len(data) != STRIPSPERSECTOR: print 'Error Reading ' + ch['PVNAME'] continue for ix in range(STRIPSPERSECTOR): hh.SetBinContent(ix, iy + 1, data[ix]) iy += 1 for xx in [ttime, tt2]: xx.Clear() [r1, r2, r3] = calcRates(SVT.chans) tt2.AddText('Sums: R1 / R2 / R3 = %.2E / %.2E / %.2E Hz' % (r1, r2, r3)) ttime.AddText(makeTime()) if not gPad: sys.exit() cc.Modified() cc.Update() time.sleep(POLLPERIOD)
def setLHCbStyle(): global lhcbStyle global lhcbText global lhcbLatex lhcbStyle = TStyle("lhcbStyle", "Standard LHCb plots style") # use times new roman lhcbFont = 132 # line thickness lhcbWidth = 2 lhcbTSize = 0.06 #// use plain black on white colors lhcbStyle.SetFrameBorderMode(0) lhcbStyle.SetCanvasBorderMode(0) lhcbStyle.SetPadBorderMode(0) lhcbStyle.SetPadColor(0) lhcbStyle.SetCanvasColor(0) lhcbStyle.SetStatColor(0) lhcbStyle.SetPalette(1) lhcbStyle.SetLegendBorderSize(0) lhcbStyle.SetLegendFont(132) lhcbStyle.SetFillColor(1) lhcbStyle.SetFillStyle(1001) # set the paper & margin sizes lhcbStyle.SetPaperSize(20, 26) lhcbStyle.SetPadTopMargin(0.1) lhcbStyle.SetPadRightMargin(0.05) lhcbStyle.SetPadBottomMargin(0.16) lhcbStyle.SetPadLeftMargin(0.14) # use large fonts lhcbStyle.SetTextFont(lhcbFont) lhcbStyle.SetTextSize(lhcbTSize) # lhcbStyle.SetTextSize(0.08) lhcbStyle.SetLabelFont(lhcbFont, "x") lhcbStyle.SetLabelFont(lhcbFont, "y") lhcbStyle.SetLabelFont(lhcbFont, "z") lhcbStyle.SetLabelSize(lhcbTSize, "x") lhcbStyle.SetLabelSize(lhcbTSize, "y") lhcbStyle.SetLabelSize(lhcbTSize, "z") lhcbStyle.SetTitleFont(lhcbFont) lhcbStyle.SetTitleFont(lhcbFont, "x") lhcbStyle.SetTitleFont(lhcbFont, "y") lhcbStyle.SetTitleFont(lhcbFont, "z") lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "x") lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "y") lhcbStyle.SetTitleSize(1.2 * lhcbTSize, "z") # use bold lines and markers lhcbStyle.SetLineWidth(lhcbWidth) lhcbStyle.SetFrameLineWidth(lhcbWidth) lhcbStyle.SetHistLineWidth(lhcbWidth) lhcbStyle.SetFuncWidth(lhcbWidth) lhcbStyle.SetGridWidth(lhcbWidth) lhcbStyle.SetLineStyleString(2, "[12 12]") lhcbStyle.SetMarkerStyle(20) lhcbStyle.SetMarkerSize(1.0) # label offsets lhcbStyle.SetLabelOffset(0.010) #titles lhcbStyle.SetTitleOffset(0.95, "X") lhcbStyle.SetTitleOffset(0.95, "Y") lhcbStyle.SetTitleOffset(1.2, "Z") lhcbStyle.SetTitleFillColor(0) lhcbStyle.SetTitleStyle(0) lhcbStyle.SetTitleBorderSize(0) lhcbStyle.SetTitleFont(lhcbFont, "title") lhcbStyle.SetTitleX(0.0) lhcbStyle.SetTitleY(1.0) lhcbStyle.SetTitleW(1.0) lhcbStyle.SetTitleH(0.05) # by default, do not display histogram decorations: lhcbStyle.SetOptStat(0) #lhcbStyle.SetOptStat("emr") # show only nent -e , mean - m , rms -r #lhcbStyle.SetStatFormat("6.3g") # specified as c printf options lhcbStyle.SetOptTitle(0) lhcbStyle.SetOptFit(0) #lhcbStyle.SetOptFit(1011) # order is probability, Chi2, errors, parameters # look of the statistics box: lhcbStyle.SetStatBorderSize(0) lhcbStyle.SetStatFont(lhcbFont) lhcbStyle.SetStatFontSize(0.05) lhcbStyle.SetStatX(0.93) lhcbStyle.SetStatY(0.88) lhcbStyle.SetStatW(0.20) lhcbStyle.SetStatH(0.15) # put tick marks on top and RHS of plots lhcbStyle.SetPadTickX(1) lhcbStyle.SetPadTickY(1) # histogram divisions: only 5 in x to avoid label overlaps lhcbStyle.SetNdivisions(505, "x") lhcbStyle.SetNdivisions(505, "y") lhcbStyle.SetNdivisions(505, "z") # define style for text lhcbLabel = TText() lhcbLabel.SetTextFont(lhcbFont) lhcbLabel.SetTextColor(1) lhcbLabel.SetTextSize(0.04) lhcbLabel.SetTextAlign(12) # define style of latex text lhcbLatex = TLatex() lhcbLatex.SetTextFont(lhcbFont) lhcbLatex.SetTextColor(1) lhcbLatex.SetTextSize(0.04) lhcbLatex.SetTextAlign(12) # set this style gROOT.SetStyle("lhcbStyle") gROOT.ForceStyle() return
texts = [] gamma = [] if 'Gamma' in data: gamma = data['Gamma'] arrows = [] gtexts = [] dlines = [] for i, band in enumerate(bands): xmin = i * (param['width'] + param['space']) xmax = xmin + param['width'] xcenter = xmin + param['width'] / 2. levels = band['levels'] texts.append( TText( xcenter, plot_range['ymin'] + 0.01 * (plot_range['ymax'] - plot_range['ymin']) + param['TextFloat'], band['name'])) texts[-1].SetTextAlign(21) if 'color' in band: texts[-1].SetTextColor(band['color']) for level in levels: lines.append(TLine(xmin, level['energy'], xmax, level['energy'])) texts.append( TText(xmin, level['energy'] + param['TextFloat'], str(level['level']))) texts[-1].SetTextAlign(11) texts.append( TText(xmax, level['energy'] + param['TextFloat'], str(level['energy']))) texts[-1].SetTextAlign(31) if 'noenergy' in band:
ar1.DrawArrow(.25, .775, .39, .775, 0.015, '|>') ar1.DrawArrow(.50, .775, .59, .775, 0.015, '|>') ar1.DrawArrow(.70, .775, .79, .775, 0.015, '|>') ar1.DrawArrow(.50, .275, .59, .275, 0.015, '|>') ar1.DrawArrow(.70, .275, .79, .275, 0.015, '|>') ar1.DrawArrow(.45, .175, .54, .175, 0.015, '|>') ar1.DrawArrow(.43, .075, .54, .075, 0.015, '|>') ar1.DrawArrow(.41, -.025, .54, -.025, 0.015, '|>') ldot = TLine(.95, .92, .99, .92) ldot.SetLineStyle(3) ldot.Draw() ldot.DrawLine(.9, .775, .99, .775) ldot.DrawLine(.9, .275, .99, .275) ldot.DrawLine(.55, .05, .55, 0) ldot.DrawLine(.9, .05, .75, 0) pname = TText(.46, .21, 'fEventOffset') pname.SetTextFont(72) pname.SetTextSize(0.018) pname.Draw() pname.DrawText(.44, .11, 'fBuffer') pname.DrawText(.42, .01, 'fZipBuffer') pname.DrawText(.26, .81, 'fLeaves = TObjArray of TLeaf') pname.DrawText(.24, .40, 'fBasketEvent') pname.DrawText(.22, .31, 'fBaskets = TObjArray of TBasket') pname.DrawText(.20, 1.0, 'fBranches = TObjArray of TBranch') ntleaf = TPaveText(0.30, .42, .62, .7) ntleaf.SetTextSize(0.014) ntleaf.SetFillColor(leafcolor) ntleaf.SetTextAlign(12) ntleaf.AddText('fLen: number of fixed elements') ntleaf.AddText('fLenType: number of bytes of data type')
file.SetTextSize(0.04) file.AddText('TFile') file.AddText('Header') arrow = TArrow(6, 20.5, 17, 20.5, 0.02, '|>') arrow.SetFillStyle(1001) arrow.SetLineWidth(2) arrow.Draw() free = TPaveText(8, 20, 11, 21) free.SetFillColor(18) free.Draw() free.AddText('First:Last') free2 = TPaveText(12, 20, 15, 21) free2.SetFillColor(18) free2.Draw() free2.AddText('First:Last') tfree = TText(6.2, 21.2, 'fFree = TList of free blocks') tfree.SetTextSize(0.02) tfree.Draw() tkeys = TText(5.2, 18.2, 'fKeys = TList of Keys') tkeys.SetTextSize(0.02) tkeys.Draw() tmemory = TText(3.2, 15.2, 'fListHead = TList of Objects in memory') tmemory.SetTextSize(0.02) tmemory.Draw() arrow.DrawArrow(5, 17, 17, 17, 0.02, '|>') line = TLine(5, 19, 5, 17) line.SetLineWidth(2) line.Draw() key0 = TPaveText(7, 16, 10, 18) key0.SetTextSize(0.04)
class Calorimeter: '''class for calorimeter plotting''' #class members name = 'n' #ClassVar[TString] range_min = 0 #ClassVar[float] range_max = 0 #ClassVar[float] content = [] #ClassVar[list[float]] #of float ombox = [] #ClassVar[list[TBox]] #of TBox omtext = [] #ClassVar[list[TText]] #of TText source_foil = TLine() #ClassVar[TLine] it_label = '' #ClassVar[TText] fr_label = '' #ClassVar[TText] canvas = None #ClassVar[TCanvas] palette_index = 0 #ClassVar[int] fiber_map_lines = np.array([TLine]) ncalo = 712 #ClassVar[int] nmwall = 520 nxwall = 128 ngveto = 64 #constructor of calorimeter for calo in range(ncalo): content.append(0) range_min = range_max = -1 spacerx = 0.0125 spacery = 0.0250 mw_sizey = (1 - 4 * spacery) / (13 + 2) gv_sizey = mw_sizey xw_sizey = mw_sizey * 13. / 16. mw_sizex = (0.5 - 4 * spacerx) / (20 + 4) gv_sizex = mw_sizex * 20. / 16. xw_sizex = mw_sizex for mw0 in range(2): for mw1 in range(20): for mw2 in range(13): id = mw0 * 20 * 13 + mw1 * 13 + mw2 x1 = spacerx + 2 * xw_sizex + spacerx + 0.5 * mw0 + mw_sizex * ( mw1) if (mw0 == 1): #swap french in case of internal view x1 = spacerx + 2 * xw_sizex + spacerx + 0.5 * mw0 + mw_sizex * ( 19 - mw1) y1 = spacery + gv_sizey + spacery + mw_sizey * mw2 x2 = x1 + mw_sizex y2 = y1 + mw_sizey box = TBox(x1, y1, x2, y2) box.SetFillColor(0) box.SetLineWidth(1) ombox.append(box) text = TText( x1 + 0.33 * mw_sizex, y1 + 0.33 * mw_sizey, str(id).zfill(3) ) # "{:10.4f}".format(xid))#, TString::Form("%03d",id)) text.SetTextSize(0.02) omtext.append(text) for xw0 in range(2): for xw1 in range(2): for xw2 in range(2): for xw3 in range(16): id = 520 + xw0 * 2 * 2 * 16 + xw1 * 2 * 16 + xw2 * 16 + xw3 # x1 = 0 if xw0 == 0: if xw1 == 0: x1 = spacerx + xw_sizex * xw2 elif xw1 == 1: x1 = spacerx + 2 * xw_sizex + spacerx + 20 * mw_sizex + spacerx + ( 1 - xw2) * xw_sizex elif xw0 == 1: #wall ID if xw1 == 0: #side ID x1 = 0.5 + spacerx + 2 * xw_sizex + spacerx + 20 * mw_sizex + spacerx + ( 1 - xw2) * xw_sizex elif xw1 == 1: x1 = 0.5 + spacerx + xw_sizex * xw2 x2 = x1 + xw_sizex y1 = spacery + gv_sizey + spacery + xw_sizey * (xw3) y2 = spacery + gv_sizey + spacery + xw_sizey * (xw3 + 1) box = TBox(x1, y1, x2, y2) box.SetFillColor(0) box.SetLineWidth(1) ombox.append(box) text = TText(x1 + 0.33 * mw_sizex, y1 + 0.33 * mw_sizey, str(id).zfill(3)) text.SetTextSize(0.02) omtext.append(text) for gv0 in range(2): for gv1 in range(2): for gv2 in range(16): id = 520 + 128 + gv0 * 2 * 16 + gv1 * 16 + gv2 # x1 = 0 if gv0 == 0: #side ID 1 x1 = spacerx + 2 * xw_sizex + spacerx + gv_sizex * gv2 elif gv0 == 1: #wall ID 0 x1 = 0.5 + spacerx + 2 * xw_sizex + spacerx + gv_sizex * ( 16 - 1 - gv2) x2 = x1 + gv_sizex y1 = spacery + gv1 * (gv_sizey + spacery + 13 * mw_sizey + spacery) y2 = y1 + gv_sizey box = TBox(x1, y1, x2, y2) box.SetFillColor(0) box.SetLineWidth(1) ombox.append(box) text = TText(x1 + 0.33 * gv_sizex, y1 + 0.33 * gv_sizey, str(id).zfill(3)) text.SetTextSize(0.02) omtext.append(text) source_foil = TLine(0.5, spacery, 0.5, 1 - spacery) source_foil.SetLineWidth(2) it_label = TText( spacerx, spacery + gv_sizey + spacery + 13 * mw_sizey + spacery + 0.25 * gv_sizey, "ITALY") fr_label = TText( 0.5 + spacerx, spacery + gv_sizey + spacery + 13 * mw_sizey + spacery + 0.25 * gv_sizey, "FRANCE") it_label.SetTextSize(0.036) fr_label.SetTextSize(0.036) #fiber map line_h_IT = TLine(0.25 - 10 * mw_sizex, 2 * spacery + 9 * mw_sizey, 0.25 + 10 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_v_A1A2_IT = TLine(0.25, 2 * spacery + 9 * mw_sizey, 0.25, 2 * spacery + 14 * mw_sizey) line_v_A3A4_IT = TLine(0.25 - 4 * mw_sizex, 2 * 0.025 + mw_sizey, 0.25 - 4 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_v_A4A5_IT = TLine(0.25 + 4 * mw_sizex, 2 * 0.025 + mw_sizey, 0.25 + 4 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_h_FR = TLine(0.75 - 10 * mw_sizex, 2 * spacery + 9 * mw_sizey, 0.75 + 10 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_v_A1A2_FR = TLine(0.75, 2 * 0.025 + 9 * mw_sizey, 0.75, 2 * spacery + 14 * mw_sizey) line_v_A3A4_FR = TLine(0.75 - 4 * mw_sizex, 2 * spacery + mw_sizey, 0.75 - 4 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_v_A4A5_FR = TLine(0.75 + 4 * mw_sizex, 2 * spacery + mw_sizey, 0.75 + 4 * mw_sizex, 2 * spacery + 9 * mw_sizey) line_veto_top_IT = TLine(0.25, 3 * spacery + 14 * mw_sizey, 0.25, 3 * spacery + 15 * mw_sizey) line_veto_bottom_left_IT = TLine(0.25 - 4 * gv_sizex, spacery, 0.25 - 4 * gv_sizex, spacery + mw_sizey) line_veto_bottom_right_IT = TLine(0.25 + 4 * gv_sizex, spacery, 0.25 + 4 * gv_sizex, spacery + mw_sizey) line_veto_top_FR = TLine(0.75, 3 * spacery + 14 * mw_sizey, 0.75, 3 * spacery + 15 * mw_sizey) line_veto_bottom_left_FR = TLine(0.75 - 4 * gv_sizex, spacery, 0.75 - 4 * gv_sizex, spacery + mw_sizey) line_veto_bottom_right_FR = TLine(0.75 + 4 * gv_sizex, spacery, 0.75 + 4 * gv_sizex, spacery + mw_sizey) line_xw_left_IT = TLine(spacerx, 0.5 + 2 * xw_sizey, spacerx + 2 * xw_sizex, 0.5 + 2 * xw_sizey) line_xw_right_IT = TLine(0.5 - spacerx - 2 * xw_sizex, 0.5 + 2 * xw_sizey, 0.5 - spacerx, 0.5 + 2 * xw_sizey) line_xw_left_FR = TLine(0.5 + spacerx, 0.5 + 2 * xw_sizey, 0.5 + spacerx + 2 * xw_sizex, 0.5 + 2 * xw_sizey) line_xw_right_FR = TLine(1 - spacerx - 2 * xw_sizex, 0.5 + 2 * xw_sizey, 1 - spacerx, 0.5 + 2 * xw_sizey) fiber_map_lines = np.array([line_h_IT, line_v_A1A2_IT, line_v_A3A4_IT, line_v_A4A5_IT, \ line_h_FR, line_v_A1A2_FR, line_v_A3A4_FR, line_v_A4A5_FR, \ line_veto_top_IT, line_veto_bottom_left_IT, line_veto_bottom_right_IT, \ line_veto_top_FR, line_veto_bottom_left_FR, line_veto_bottom_right_FR, \ line_xw_right_IT, line_xw_left_IT, \ line_xw_right_FR, line_xw_left_FR]) nRGBs = 6 stops = np.array([0.00, 0.20, 0.40, 0.60, 0.80, 1.00]) red = np.array([0.25, 0.00, 0.20, 1.00, 1.00, 0.90]) green = np.array([0.25, 0.80, 1.00, 1.00, 0.80, 0.00]) blue = np.array([1.00, 1.00, 0.20, 0.00, 0.00, 0.00]) palette_index = TColor.CreateGradientColorTable(nRGBs, stops, red, green, blue, 100) def setrange(self, xmin, xmax): self.range_min = xmin self.range_max = xmax return 0 def draw(self, fiber_map=False): if (self.canvas == None): self.canvas = TCanvas(self.name, "SuperNEMO calorimeter", 1750, 500) self.canvas.SetEditable(True) self.canvas.cd() for mw0 in range(2): for mw1 in range(20): for mw2 in range(13): id = mw0 * 20 * 13 + mw1 * 13 + mw2 self.ombox[id].Draw("l") self.omtext[id].Draw() for xw0 in range(2): for xw1 in range(2): for xw2 in range(2): for xw3 in range(16): id = 520 + xw0 * 2 * 2 * 16 + xw1 * 2 * 16 + xw2 * 16 + xw3 self.ombox[id].Draw("l") self.omtext[id].Draw() for gv0 in range(2): for gv1 in range(2): for gv2 in range(16): id = 520 + 128 + gv0 * 2 * 16 + gv1 * 16 + gv2 self.ombox[id].Draw("l") self.omtext[id].Draw() self.source_foil.Draw() self.it_label.Draw() self.fr_label.Draw() if fiber_map == True: for line in self.fiber_map_lines: line.SetLineColor(kBlack) line.SetLineStyle(9) line.SetLineWidth(3) line.Draw() self.canvas.SetEditable(False) gSystem.ProcessEvents() return 0 def reset(self): for calo in range(self.ncalo): self.content[calo] = 0 # for calo=0; calo<self.ncalo; ++calo) self.ombox[calo].SetFillColor(0) self.canvas.Modified() self.canvas.Update() gSystem.ProcessEvents() return 0 def setcontent(self, pmt, value): self.content[pmt] = value return 0 def update(self): content_min = self.content[0] content_max = self.content[0] if (self.range_min == -1): for calo in range(1, self.ncalo): if (self.content[calo] < content_min): content_min = self.content[calo] if (self.content[calo] > content_max): content_max = self.content[calo] else: self.range_min = 0 content_max = self.range_max for calo in range(self.ncalo): if (self.content[calo] != 0): self.ombox[calo].SetFillColor(self.palette_index + ( int)(99 * (self.content[calo] - content_min) / (content_max - content_min))) else: self.ombox[calo].SetFillColor(0) self.canvas.Modified() self.canvas.Update() gSystem.ProcessEvents() return 0 def save_canvas(self): self.canvas.SaveAs('sncalorimeter.png') return 0
def main(): cc.cd() cc.SetBorderMode(0) cc.SetFixedAspectRatio(1) cc.FeedbackMode(1) gStyle.SetOptStat(0) gStyle.SetGridStyle(1) gStyle.SetGridColor(11) hh = TH2D('hh', ';Y;X', 2, 2, 4, 14, 2, 16) hi = TH2I('hi', ';Y;X', 2, 2, 4, 14, 2, 16) setupHists([hh, hi]) xax, yax = hh.GetXaxis(), hh.GetYaxis() hh.Draw('COLZ') hi.Draw('TEXTSAME') gPad.SetLogz() gPad.SetGrid(1, 1) gPad.SetLeftMargin(0.05) gPad.SetRightMargin(0.12) tt1 = TPaveText(0.05, 0.94, 0.22, 1.0, 'NDC') ttL = TPaveText(0.05, 0.905, 0.2, 0.93, 'NDC') ttR = TPaveText(0.75, 0.905, 0.9, 0.93, 'NDC') ttLM = TPaveText(2, 0.5, 2.5, 1) ttRM = TPaveText(3.5, 0.5, 4, 1) ttM = TPaveText(0.6, 0.94, 0.9, 1, 'NDC') ttime = TPaveText(2.8, 1.5, 3.3, 1.9) tchan = TPaveText(0, 0, 0.9, 1) setupPaveTexts([tt1, ttM, ttime, tchan, ttL, ttR, ttLM, ttRM]) ttM.SetTextColor(2) tarrow = TText(2.9, 0.25, 'Beam Left') arrow = TArrow(2.85, 0.5, 2.75, 0.5, 0.02, '|>') arrow.SetAngle(40) arrow.SetFillColor(1) arrow.SetLineWidth(2) tt = TText() tt.SetTextColor(1) tt.SetTextAngle(90) tt.SetTextSize(0.04) tt.DrawText(4.07, 9, 'kHz') tt.SetTextAngle(0) tt.SetTextColor(2) tt.DrawTextNDC(0.23, 0.905, 'SHMS PCAL FADC SCALERS') ttL.AddText('Left+') ttR.AddText('-Right') cc.cd() for xx in [ttM, tt1, ttL, ttR, arrow, tarrow, ttime, ttLM, ttRM]: xx.Draw() cc2.cd() tchan.Draw('NDC') cc.cd() xmin, xmax = xax.GetXmin(), xax.GetXmax() ymin, ymax = yax.GetXmin(), yax.GetXmax() gPad.SetEditable(0) while True: zvals = getScalars() for ii in range(len(zvals)): hh.SetBinContent(xax.FindBin(XVALS[ii] + 1), yax.FindBin(16 - YVALS[ii]), zvals[ii]) hi.SetBinContent(xax.FindBin(XVALS[ii] + 1), yax.FindBin(16 - YVALS[ii]), zvals[ii]) for xx in [ttime, ttM, tt1, ttLM, ttRM]: xx.Clear() [left, right, maximum] = calcRates(zvals) tt1.AddText('Total: %.1f MHz' % ((left + right) / 1000)) ttLM.AddText('Left total: %.1f MHz' % (left / 1000)) ttRM.AddText('Right total: %.1f MHz' % (right / 1000)) ttM.AddText('MAX SINGLE CRYSTAL = %.0f kHz' % (maximum)) ttime.AddText(makeTime()) if not gPad: sys.exit() #if gPad.GetEvent()==11: # xy=pix2xy(gPad) #ee=ECAL.findChannelXY(xy[0],xy[1]) #if ee: # tchan.Clear() #tchan.AddText(printChannel(ee)) # cc2.Modified() # cc2.Update() #elif gPad.GetEvent()==12: # tchan.Clear() # cc2.Modified() # cc2.Update() cc.Modified() cc.Update() time.sleep(1)
def main(): if True: mf=TGMainFrame(gClient.GetRoot(),1500,475) gvf=TGVerticalFrame(mf,1500,475) rec=TRootEmbeddedCanvas("ccc",gvf,1500,450) rec2=TRootEmbeddedCanvas("ccc2",gvf,1500,25) gvf.AddFrame(rec,TGLayoutHints(ROOT.kLHintsExpandX|ROOT.kLHintsTop)) gvf.AddFrame(rec2,TGLayoutHints(ROOT.kLHintsExpandX|ROOT.kLHintsBottom)) mf.AddFrame(gvf,TGLayoutHints(ROOT.kLHintsExpandX)) cc=rec.GetCanvas() cc2=rec2.GetCanvas() mf.SetEditable(0) mf.SetWindowName('HPS ECAL FADC SCALERS') mf.MapSubwindows() mf.Resize(1501,476)# resize to get proper frame placement mf.MapWindow() else: cc=TCanvas('cc','',1500,450) cc.cd() cc.SetBorderMode(0) cc.SetFixedAspectRatio(1) cc.FeedbackMode(1) gStyle.SetOptStat(0) gStyle.SetGridStyle(1) gStyle.SetGridColor(11) hh=TH2D('hh',';X;Y',46,-22,24,11,-5,6) hi=TH2I('hi',';X;Y',46,-22,24,11,-5,6) setupHists([hh,hi]) xax,yax=hh.GetXaxis(),hh.GetYaxis() hh.Draw('COLZ') hi.Draw('TEXTSAME') gPad.SetLogz() gPad.SetGrid(1,1) gPad.SetLeftMargin(0.05) tt1=TPaveText(0.1,0.9,0.3,1.0,'NDC') tt2=TPaveText(0.7,0.91,0.9,0.99,'NDC') ttT=TPaveText(-22+13+0.05,6-5,-22+22,7-5-0.05) ttB=TPaveText(-22+13+0.05,4-5+0.05,-22+22,5-5) ttM=TPaveText(-22+0+0.05,5-5+0.05,-22+13,6-5.01) ttime=TPaveText(-10,-6.5,10,-5.8) tchan=TPaveText(0,0,0.9,1) setupPaveTexts([tt1,tt2,ttT,ttB,ttM,ttime,tchan]) ttM.SetTextColor(2) bb=TBox() bb.SetFillStyle(1001) bb.SetFillColor(0) bb.SetLineWidth(1) bb.SetLineColor(1) bb.DrawBox(-9+0.05,-1,0,1.97) bb.DrawBox(-24,0,24.05,0.97) tarrow=TText(14.5,0.3,'Beam Left') arrow=TArrow(19,0.5,23,0.5,0.02,'|>') arrow.SetAngle(40) arrow.SetFillColor(1) arrow.SetLineWidth(2) tt=TText() tt.SetTextColor(1) tt.SetTextAngle(90) tt.SetTextSize(0.08) tt.DrawText(25.4,0,'kHz') tt.SetTextAngle(0) tt.SetTextColor(2) tt.DrawTextNDC(0.3,0.92,'ECAL FADC SCALERS') cc.cd() for xx in [tt2,ttT,ttB,ttM,arrow,tarrow,ttime]: xx.Draw() cc2.cd() tchan.Draw('NDC') cc.cd() ll=TLine() ll.DrawLine(xax.GetXmin(),yax.GetXmin(),xax.GetXmax(),yax.GetXmin()) ll.DrawLine(xax.GetXmin(),yax.GetXmax(),xax.GetXmax(),yax.GetXmax()) ll.DrawLine(xax.GetXmin(),yax.GetXmin(),xax.GetXmin(),0) ll.DrawLine(xax.GetXmax(),yax.GetXmin(),xax.GetXmax(),0) ll.DrawLine(xax.GetXmin(),yax.GetXmax(),xax.GetXmin(),1) ll.DrawLine(xax.GetXmax(),yax.GetXmax(),xax.GetXmax(),1) ll.DrawLine(xax.GetXmax(),0,0,0) ll.DrawLine(xax.GetXmax(),1,0,1) ll.DrawLine(xax.GetXmin(),0,-9,0) ll.DrawLine(xax.GetXmin(),1,-9,1) ll.DrawLine(-9,-1,0,-1) ll.DrawLine(-9,2,0,2) ll.DrawLine(-9,1,-9,2) ll.DrawLine(-9,-1,-9,0) ll.DrawLine(0,-1,0,0) ll.DrawLine(0,1,0,2) gPad.SetEditable(0) while True: # try: zvals=getPVS() for ii in range(len(zvals)): hh.SetBinContent(xax.FindBin(XVALS[ii]),yax.FindBin(YVALS[ii]),zvals[ii]) hi.SetBinContent(xax.FindBin(XVALS[ii]),yax.FindBin(YVALS[ii]),zvals[ii]) for xx in [ttime,tt2,ttT,ttB,ttM]: xx.Clear() [top,bottom,maximum]=calcRates(zvals) tt2.AddText('Total: %.1f MHz'%((top+bottom)/1000)) ttT.AddText('%.1f MHz'%(top/1000)) ttB.AddText('%.1f MHz'%(bottom/1000)) ttM.AddText('MAX SINGLE CRYSTAL = %.0f kHz'%(maximum)) ttime.AddText(makeTime()) if gPad.GetEvent()==11: xy=pix2xy(gPad) ee=ECAL.findChannelXY(xy[0],xy[1]) if ee: tchan.Clear() tchan.AddText(printChannel(ee)) cc2.Modified() cc2.Update() elif gPad.GetEvent()==12: tchan.Clear() cc2.Modified() cc2.Update() cc.Modified() cc.Update() time.sleep(1)
def D1H_txt_to_roothist(filename, outputpath=''): from ROOT import TFile, TCanvas, TPad, TH1D, TLatex, TStyle, gStyle, TText, gPad, TPaveText from inspect import currentframe, getframeinfo import os #gStyle.SetOptStat(0) can = TCanvas("can", "can", 200, 10, 500, 500) if (filename[0] == "/"): filename = filename else: filename = os.getcwd( ) + "/" + filename # get the path included filename loca = len(filename) for i in range(1, len(filename) + 1): # find the "/" location if (filename[-i] == "/"): loca = i - 1 break FILENAME = filename.replace(filename[:-loca], "") # this is the shorten filename # print(FILENAME, "******") fileroot = filename.replace(".txt", "_F.root") fileroot = fileroot.replace("//", "/") f = open(filename, "r") lineList = f.readlines() Nbin = (len(lineList)) # get number of bins Line_string = str(lineList[0]) _, bin_init, _, _ = Line_string.split() bin_init = float(bin_init) # get initial bin Line_string = str(lineList[len(lineList) - 1]) _, _, bin_final, _ = Line_string.split() bin_final = float(bin_final) # get final bin f.seek(0) # reset python read line hist = TH1D("hist", "hist", Nbin, bin_init, bin_final) total_e = 0 for i in range(1, Nbin + 1): Line_string = str(f.readline()) _, _, _, bin_c = Line_string.split() bin_c = float(bin_c) hist.SetBinContent(i, bin_c) total_e = total_e + bin_c total_e = int(total_e) hist.Draw() text = TText(hist.GetXaxis().GetBinCenter(2), hist.GetYaxis().GetBinCenter(1), "Recycled. Total Entry : %i" % total_e) text.SetTextFont(10) text.Draw() gPad.Update() can.Update() if (outputpath == ''): wf = TFile(fileroot, "RECREATE") print(fileroot, " root file is generated !!!") else: if (outputpath[0] == "/"): fileroot = outputpath + "/" + FILENAME.replace(".txt", "_F.root") fileroot = fileroot.replace("//", "/") elif (outputpath[0] == "~"): fileroot = outputpath.replace( "~", os.environ['HOME']) + "/" + FILENAME.replace( ".txt", "_F.root") fileroot = fileroot.replace("//", "/") else: fileroot = os.getcwd() + "/" + outputpath + "/" + FILENAME.replace( ".txt", "_F.root") fileroot = fileroot.replace("//", "/") wf = TFile(fileroot, "RECREATE") print(fileroot, " root file is generated !!!") hist.Write() wf.Close() fileroot = fileroot.replace("//", "/") # print(fileroot) return fileroot
def drawColorTable(clist=range(0, 50), nrow=None, ncol=None, cmax=10, tag="", label=False, RBG=False, newRBG=True, div=2): # https://root.cern.ch/doc/master/src_2TPad_8cxx_source.html#l01611 if not ncol: ncol = min(cmax, len(clist)) if not nrow: nrow = 1 if len(clist) <= cmax else int(ceil(len(clist) / float(cmax))) x1 = y1 = 0. x2 = y2 = 20. hs = (y2 - y1) / nrow ws = (x2 - x1) / ncol if label or RBG: width = 170 * ncol height = 80 * nrow else: width = 110 * ncol height = 80 * nrow scale = 400. / height if 400. < height: scale = sqrt(scale) canvas = TCanvas("c", "Fill Area colors", 0, 0, width, height) canvas.SetFillColor(0) canvas.Clear() canvas.Range(x1, y1, x2, y2) text = TText(0, 0, "") text.SetTextFont(61) text.SetTextSize(0.07 * scale) text.SetTextAlign(22) box = TBox() for r in range(0, nrow): ylow = y2 - hs * (r + 0.1) yup = y2 - hs * (r + 0.9) for c in range(0, ncol): i = ncol * r + c if i >= len(clist): break xlow = x1 + ws * (c + 0.1) xup = x1 + ws * (c + 0.9) color = clist[ncol * r + c] box.SetFillStyle(1001) box.SetFillColor(color) box.DrawBox(xlow, ylow, xup, yup) box.SetFillStyle(0) box.SetLineColor(1) box.DrawBox(xlow, ylow, xup, yup) if color == 1: text.SetTextColor(0) else: text.SetTextColor(1) name = "%d" % color if (isinstance(label, int) and i % div == label) or label: name = getColorString(color) if (not isinstance(RBG, bool) and isinstance(RBG, int) and i % div == RBG) or (RBG and not isinstance(label, int)): name = getRGBString(color) elif newRBG and color >= 924: name = getRGBString(color) text.DrawText(0.5 * (xlow + xup), 0.5 * (ylow + yup), name) if i >= len(clist): break canvas.SaveAs("TColorTable%s.png" % tag) canvas.SaveAs("TColorTable%s.pdf" % tag)