radmasses = []

for f in files:

# radmasses.append(float(f.replace("CMS_jj_","").split("_")[0])/1000.)

radmasses=[500, 550, 650, 700, 900, 1000, 1200, 1800, 2000]#,3.5,4.,4.5]

for i in range(len(radmasses)):

radmasses[i]=radmasses[i]/1000.

print radmasses

efficiencies={}

for mass in radmasses:

efficiencies[mass]=1000. # to convert from pb to fb

from ROOT import limit_t

limit_branch = limit_t()

fChain = []

for onefile in files:

print onefile

fileIN = rt.TFile.Open(onefile)

fileIN.Print()

fChain.append(copy.copy(fileIN.Get("limit;1")))

for j in range(0,len(fChain)):

chain = fChain[j]

chain.SetBranchAddress("limit", rt.AddressOf(limit_branch,'limit'))

rad = []

for j in range(0,len(fChain)):

chain = fChain[j]

thisrad = []

for i in range(0,6):

chain.GetTree().GetEntry(i)

thisrad.append(limit_branch.limit)

#print "limit = %f" %limit_branch.limit

#print thisrad

rad.append(thisrad)

# we do a plot r*MR

mg = rt.TMultiGraph()

mg.SetTitle("X -> ZZ")

c1 = rt.TCanvas("c1","A Simple Graph Example",200,10,600,600)

x = []

yobs = []

y2up = []

y1up = []

y1down = []

y2down = []

ymean = []

for i in range(0,len(fChain)):

y2up.append(rad[i][0]*efficiencies[radmasses[j]])

y1up.append(rad[i][1]*efficiencies[radmasses[j]])

ymean.append(rad[i][2]*efficiencies[radmasses[j]])

y1down.append(rad[i][3]*efficiencies[radmasses[j]])

y2down.append(rad[i][4]*efficiencies[radmasses[j]])

yobs.append(rad[i][5]*efficiencies[radmasses[j]])

grobs = rt.TGraphErrors(1)

grobs.SetMarkerStyle(rt.kFullDotLarge)

grobs.SetLineColor(rt.kRed)

grobs.SetLineWidth(3)

gr2up = rt.TGraphErrors(1)

gr2up.SetMarkerColor(0)

gr1up = rt.TGraphErrors(1)

gr1up.SetMarkerColor(0)

grmean = rt.TGraphErrors(1)

grmean.SetLineColor(1)

grmean.SetLineWidth(2)

grmean.SetLineStyle(3)

gr1down = rt.TGraphErrors(1)

gr1down.SetMarkerColor(0)

gr2down = rt.TGraphErrors(1)

gr2down.SetMarkerColor(0)

for j in range(0,len(fChain)):

grobs.SetPoint(j, radmasses[j], yobs[j])

gr2up.SetPoint(j, radmasses[j], y2up[j])

gr1up.SetPoint(j, radmasses[j], y1up[j])

grmean.SetPoint(j, radmasses[j], ymean[j])

gr1down.SetPoint(j, radmasses[j], y1down[j])

gr2down.SetPoint(j, radmasses[j], y2down[j])

#print " observed %f %f" %(radmasses[j],yobs[j])

mg.Add(gr2up)#.Draw("same")

mg.Add(gr1up)#.Draw("same")

mg.Add(grmean,"L")#.Draw("same,AC*")

mg.Add(gr1down)#.Draw("same,AC*")

mg.Add(gr2down)#.Draw("same,AC*")

if obs: mg.Add(grobs,"L")#.Draw("AC*")

c1.SetLogy(1)

mg.SetTitle("")

mg.Draw("AP")

mg.GetXaxis().SetTitle("Resonance mass (TeV)")

resonance="G_{RS}"

#resonance="G_{Bulk}"

if withAcceptance:

mg.GetYaxis().SetTitle("#sigma #times B("+resonance+" #rightarrow "+label.split("_")[0].replace("RS1","").replace("Bulk","")+") #times A (fb)")

else:

mg.GetYaxis().SetTitle("#sigma #times B("+resonance+" #rightarrow HH to b#bar{b}b#bar{b}) (fb)")

mg.GetYaxis().SetLabelFont(42)

mg.GetXaxis().SetLabelFont(42)

mg.GetYaxis().SetTitleFont(42)

mg.GetXaxis().SetTitleFont(42)

mg.GetYaxis().SetTitleSize(0.035)

mg.GetXaxis().SetTitleSize(0.035)

mg.GetXaxis().SetLabelSize(0.045)

mg.GetYaxis().SetLabelSize(0.045)

mg.GetYaxis().SetRangeUser(1.,1000)

mg.GetYaxis().SetTitleOffset(1.4)

mg.GetYaxis().CenterTitle(True)

mg.GetXaxis().SetTitleOffset(1.1)

mg.GetXaxis().CenterTitle(True)

mg.GetXaxis().SetNdivisions(508)

if "qW" in label.split("_")[0] or "qZ" in label.split("_")[0]:

mg.GetXaxis().SetLimits(0.5,2.4)

else:

mg.GetXaxis().SetLimits(0.5,2.4)

# histo to shade

n=len(fChain)

grgreen = rt.TGraph(2*n)

for i in range(0,n):

grgreen.SetPoint(i,radmasses[i],y2up[i])

grgreen.SetPoint(n+i,radmasses[n-i-1],y2down[n-i-1])

grgreen.SetFillColor(rt.kGreen)

grgreen.Draw("f")

gryellow = rt.TGraph(2*n)

for i in range(0,n):

gryellow.SetPoint(i,radmasses[i],y1up[i])

gryellow.SetPoint(n+i,radmasses[n-i-1],y1down[n-i-1])

gryellow.SetFillColor(rt.kYellow)

gryellow.Draw("f,same")

grmean.Draw("L")

if obs: grobs.Draw("L")

gtheory = rt.TGraphErrors(1)

gtheory.SetLineColor(rt.kBlue)

gtheory.SetLineWidth(4)

#ftheory=open("signal_cross_section_RS1Graviton.txt")

ftheory=open("bulk_graviton_exo15002.txt")

ij=0

glogtheory = rt.TGraphErrors(1)

for lines in ftheory.readlines():

for line in lines.split("\r"):

split=line.split(":")

print(split[1][0:])

gtheory.SetPoint(ij, float(split[0][-4:])/1000., float(split[1])*0.57*0.57*1000.)

glogtheory.SetPoint(ij, float(split[0][-4:])/1000., log(float(split[1])*0.57*0.57*1000.))

ij+=1

mg.Add(gtheory,"L")

gtheory.Draw("L")

#ltheory="G_{RS1} #rightarrow HH (k/#bar{M}_{Pl}=0.1)"

ltheory ="BulkGrav k=0.5"

# crossing=0

# for mass in range(int(radmasses[0]*1000.),int(radmasses[-1]*1000.)):

# if exp(glogtheory.Eval(mass/1000.))>grmean.Eval(mass/1000.) and crossing>=0:

# print label,"exp crossing",mass

# crossing=-1

# if exp(glogtheory.Eval(mass/1000.))<grmean.Eval(mass/1000.) and crossing<=0:

# print label,"exp crossing",mass

# crossing=1

# crossing=0

# for mass in range(int(radmasses[0]*1000.),int(radmasses[-1]*1000.)):

# if exp(glogtheory.Eval(mass/1000.))>grobs.Eval(mass/1000.) and crossing>=0:

# print label,"obs crossing",mass

# crossing=-1

# if exp(glogtheory.Eval(mass/1000.))<grobs.Eval(mass/1000.) and crossing<=0:

#print label,"obs crossing",mass

#crossing=1

if "WW" in label.split("_")[0] or "ZZ" in label.split("_")[0]:

leg = rt.TLegend(0.53,0.65,0.95,0.89)

leg2 = rt.TLegend(0.33,0.55,0.95,0.89)

else:

leg = rt.TLegend(0.59,0.65,0.95,0.89)

leg2 = rt.TLegend(0.49,0.55,0.95,0.89)

leg.SetFillColor(rt.kWhite)

leg.SetFillStyle(0)

leg.SetTextSize(0.04)

leg.SetTextFont(42)

leg.SetBorderSize(0)

leg2.SetFillColor(rt.kWhite)

leg2.SetFillStyle(0)

leg2.SetTextSize(0.04)

leg2.SetBorderSize(0)

if obs: leg.AddEntry(grobs, "Observed", "L")

leg.AddEntry(gryellow, "Expected (68%)", "f")

leg.AddEntry(grgreen, "Expected (95%)", "f")

leg.AddEntry(gtheory, ltheory, "L")

if obs: leg2.AddEntry(grobs, " ", "")

#leg2.AddEntry(grmean, " ", "L")

#leg2.AddEntry(grmean, " ", "L")

#leg2.AddEntry(gtheory, " ", "")

leg.Draw()

#leg2.Draw("same")

CMS_lumi.CMS_lumi(c1, iPeriod, iPos)

c1.cd()

c1.Update()

if withAcceptance:

c1.SaveAs("brazilianFlag_acc_%s_13TeV.root" %label)

c1.SaveAs("brazilianFlag_acc_%s_13TeV.pdf" %label)

else:

c1.SaveAs("brazilianFlag_%s_13TeV.root" %label)