17 x = ROOT.RooRealVar(
"x",
"x", -5, 5)
18 y = ROOT.RooRealVar(
"y",
"y", -5, 5)
21 a0 = ROOT.RooRealVar(
"a0",
"a0", -3.5, -5, 5)
22 a1 = ROOT.RooRealVar(
"a1",
"a1", -1.5, -1, 1)
23 sigma = ROOT.RooRealVar(
"sigma",
"width of gaussian", 1.5)
26 fy = ROOT.RooFormulaVar(
"fy",
"a0-a1*sqrt(10*abs(y))",
27 ROOT.RooArgList(y, a0, a1))
30 model = ROOT.RooGaussian(
31 "model",
"Gaussian with shifting mean", x, fy, sigma)
34 data = model.generate(ROOT.RooArgSet(x, y), 10000)
43 hh_data = ROOT.RooAbsData.createHistogram(data,
"x,y", x, ROOT.RooFit.Binning(
44 20), ROOT.RooFit.YVar(y, ROOT.RooFit.Binning(20)))
48 hh_pdf = model.createHistogram(
"x,y", 50, 50)
49 hh_pdf.SetLineColor(ROOT.kBlue)
55 z = ROOT.RooRealVar(
"z",
"z", -5, 5)
57 gz = ROOT.RooGaussian(
58 "gz",
"gz", z, ROOT.RooFit.RooConst(0), ROOT.RooFit.RooConst(2))
59 model3 = ROOT.RooProdPdf(
"model3",
"model3", ROOT.RooArgList(model, gz))
61 data3 = model3.generate(ROOT.RooArgSet(x, y, z), 10000)
68 hh_data3 = ROOT.RooAbsData.createHistogram(
69 data3,
"hh_data3", x, ROOT.RooFit.Binning(8), ROOT.RooFit.YVar(
70 y, ROOT.RooFit.Binning(8)), ROOT.RooFit.ZVar(
71 z, ROOT.RooFit.Binning(8)))
74 hh_pdf3 = model3.createHistogram(
75 "hh_model3", x, ROOT.RooFit.Binning(20), ROOT.RooFit.YVar(
76 y, ROOT.RooFit.Binning(20)), ROOT.RooFit.ZVar(
77 z, ROOT.RooFit.Binning(20)))
78 hh_pdf3.SetFillColor(ROOT.kBlue)
80 c1 = ROOT.TCanvas(
"rf309_2dimplot",
"rf309_2dimplot", 800, 800)
83 ROOT.gPad.SetLeftMargin(0.15)
84 hh_data.GetZaxis().SetTitleOffset(1.4)
87 ROOT.gPad.SetLeftMargin(0.20)
88 hh_pdf.GetZaxis().SetTitleOffset(2.5)
91 ROOT.gPad.SetLeftMargin(0.15)
92 hh_data.GetZaxis().SetTitleOffset(1.4)
95 ROOT.gPad.SetLeftMargin(0.15)
96 hh_pdf.GetZaxis().SetTitleOffset(2.5)
98 c1.SaveAs(
"rf309_2dimplot.png")
100 c2 = ROOT.TCanvas(
"rf309_3dimplot",
"rf309_3dimplot", 800, 400)
103 ROOT.gPad.SetLeftMargin(0.15)
104 hh_data3.GetZaxis().SetTitleOffset(1.4)
105 hh_data3.Draw(
"lego")
107 ROOT.gPad.SetLeftMargin(0.15)
108 hh_pdf3.GetZaxis().SetTitleOffset(1.4)
110 c2.SaveAs(
"rf309_3dimplot.png")