25ROOT.ROOT.EnableImplicitMT()
28path =
"root://eospublic.cern.ch//eos/opendata/atlas/OutreachDatasets/2020-01-22"
30df[
"data"] =
ROOT.RDataFrame(
"mini", (os.path.join(path,
"GamGam/Data/data_{}.GamGam.root".
format(x))
for x
in (
"A",
"B",
"C",
"D")))
31df[
"ggH"] =
ROOT.RDataFrame(
"mini", os.path.join(path,
"GamGam/MC/mc_343981.ggH125_gamgam.GamGam.root"))
32df[
"VBF"] =
ROOT.RDataFrame(
"mini", os.path.join(path,
"GamGam/MC/mc_345041.VBFH125_gamgam.GamGam.root"))
33processes = list(df.keys())
36for p
in [
"ggH",
"VBF"]:
37 df[p] = df[p].Define(
"weight",
38 "scaleFactor_PHOTON * scaleFactor_PhotonTRIGGER * scaleFactor_PILEUP * mcWeight");
39df[
"data"] = df[
"data"].Define(
"weight",
"1.0")
44 df[p] = df[p].Filter(
"trigP")
47 df[p] = df[p].Define(
"goodphotons",
"photon_isTightID && (photon_pt > 25000) && (abs(photon_eta) < 2.37) && ((abs(photon_eta) < 1.37) || (abs(photon_eta) > 1.52))")\
48 .Filter(
"Sum(goodphotons) == 2")
51 df[p] = df[p].Filter(
"Sum(photon_ptcone30[goodphotons] / photon_pt[goodphotons] < 0.065) == 2")\
52 .Filter(
"Sum(photon_etcone20[goodphotons] / photon_pt[goodphotons] < 0.065) == 2")
55ROOT.gInterpreter.Declare(
58float ComputeInvariantMass(RVecF pt, RVecF eta, RVecF phi, RVecF e) {
59 ROOT::Math::PtEtaPhiEVector p1(pt[0], eta[0], phi[0], e[0]);
60 ROOT::Math::PtEtaPhiEVector p2(pt[1], eta[1], phi[1], e[1]);
61 return (p1 + p2).mass() / 1000.0;
69 df[p] = df[p].Define(
"m_yy",
"ComputeInvariantMass(photon_pt[goodphotons], photon_eta[goodphotons], photon_phi[goodphotons], photon_E[goodphotons])")
72 df[p] = df[p].Filter(
"photon_pt[goodphotons][0] / 1000.0 / m_yy > 0.35")\
73 .Filter(
"photon_pt[goodphotons][1] / 1000.0 / m_yy > 0.25")\
74 .Filter(
"m_yy > 105 && m_yy < 160")
77 hists[p] = df[p].Histo1D(
78 ROOT.RDF.TH1DModel(p,
"Diphoton invariant mass; m_{#gamma#gamma} [GeV];Events", 30, 105, 160),
89ggh = hists[
"ggH"].GetValue()
90vbf = hists[
"VBF"].GetValue()
91data = hists[
"data"].GetValue()
96ROOT.gROOT.SetStyle(
"ATLAS")
99c = ROOT.TCanvas(
"c",
"", 700, 750)
101upper_pad = ROOT.TPad(
"upper_pad",
"", 0, 0.35, 1, 1)
102lower_pad = ROOT.TPad(
"lower_pad",
"", 0, 0, 1, 0.35)
103for p
in [upper_pad, lower_pad]:
104 p.SetLeftMargin(0.14)
105 p.SetRightMargin(0.05)
108upper_pad.SetBottomMargin(0)
109lower_pad.SetTopMargin(0)
110lower_pad.SetBottomMargin(0.3)
116fit = ROOT.TF1(
"fit",
"([0]+[1]*x+[2]*x^2+[3]*x^3)+[4]*exp(-0.5*((x-[5])/[6])^2)", 105, 160)
117fit.FixParameter(5, 125.0)
118fit.FixParameter(4, 119.1)
119fit.FixParameter(6, 2.39)
121fit.SetLineStyle(ROOT.kSolid)
123data.Fit(
"fit",
"0",
"", 105, 160)
127data.SetMarkerStyle(20)
128data.SetMarkerSize(1.2)
130data.SetLineColor(ROOT.kBlack)
133data.GetYaxis().SetLabelSize(0.045)
134data.GetYaxis().SetTitleSize(0.05)
143bkg = ROOT.TF1(
"bkg",
"([0]+[1]*x+[2]*x^2+[3]*x^3)", 105, 160)
145 bkg.SetParameter(i, fit.GetParameter(i))
147bkg.SetLineStyle(ROOT.kDashed)
154ggh.Scale(lumi * 0.102 / 55922617.6297)
155vbf.Scale(lumi * 0.008518764 / 3441426.13711)
158higgs.Draw(
"HIST SAME")
163ratiobkg = ROOT.TH1I(
"zero",
"", 100, 105, 160)
164ratiobkg.SetLineColor(4)
165ratiobkg.SetLineStyle(ROOT.kDashed)
166ratiobkg.SetLineWidth(2)
167ratiobkg.SetMinimum(-125)
168ratiobkg.SetMaximum(250)
169ratiobkg.GetXaxis().SetLabelSize(0.08)
170ratiobkg.GetXaxis().SetTitleSize(0.12)
171ratiobkg.GetXaxis().SetTitleOffset(1.0)
172ratiobkg.GetYaxis().SetLabelSize(0.08)
173ratiobkg.GetYaxis().SetTitleSize(0.09)
174ratiobkg.GetYaxis().SetTitle(
"Data - Bkg.")
175ratiobkg.GetYaxis().CenterTitle()
176ratiobkg.GetYaxis().SetTitleOffset(0.7)
177ratiobkg.GetYaxis().SetNdivisions(503,
False)
178ratiobkg.GetYaxis().ChangeLabel(-1, -1, 0)
179ratiobkg.GetXaxis().SetTitle(
"m_{#gamma#gamma} [GeV]")
182ratiosig = ROOT.TH1F(
"ratiosig",
"ratiosig", 5500, 105, 160)
184ratiosig.SetLineColor(2)
185ratiosig.SetLineStyle(ROOT.kSolid)
186ratiosig.SetLineWidth(2)
190ratiodata = data.Clone()
191ratiodata.Add(bkg, -1)
192for i
in range(1, data.GetNbinsX()):
193 ratiodata.SetBinError(i, data.GetBinError(i))
194ratiodata.Draw(
"E SAME")
198legend = ROOT.TLegend(0.55, 0.55, 0.89, 0.85)
199legend.SetTextFont(42)
200legend.SetFillStyle(0)
201legend.SetBorderSize(0)
202legend.SetTextSize(0.05)
203legend.SetTextAlign(32)
204legend.AddEntry(data,
"Data" ,
"lep")
205legend.AddEntry(bkg,
"Background",
"l")
206legend.AddEntry(fit,
"Signal + Bkg.",
"l")
207legend.AddEntry(higgs,
"Signal",
"l")
214text.SetTextSize(0.05)
215text.DrawLatex(0.18, 0.84,
"ATLAS")
217text.DrawLatex(0.18 + 0.13, 0.84,
"Open Data")
218text.SetTextSize(0.04)
219text.DrawLatex(0.18, 0.78,
"#sqrt{s} = 13 TeV, 10 fb^{-1}")
222c.SaveAs(
"df104_HiggsToTwoPhotons.png")
223print(
"Saved figure to df104_HiggsToTwoPhotons.png")
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h Atom_t Int_t ULong_t ULong_t unsigned char prop_list Atom_t Atom_t Atom_t Time_t format
ROOT's RDataFrame offers a modern, high-level interface for analysis of data stored in TTree ,...
unsigned int RunGraphs(std::vector< RResultHandle > handles)
Trigger the event loop of multiple RDataFrames concurrently.
A struct which stores the parameters of a TH1D.