-
Notifications
You must be signed in to change notification settings - Fork 18
/
rf107_plotstyles.py
103 lines (80 loc) · 3.16 KB
/
rf107_plotstyles.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#####################################
#
# 'BASIC FUNCTIONALITY' ROOT.RooFit tutorial macro #107
#
# Demonstration of various plotting styles of data, functions
# in a ROOT.RooPlot
#
# 07/2008 - Wouter Verkerke
#
# /
import ROOT
def rf107_plotstyles():
# S e t u p m o d e l
# ---------------------
# Create observables
x = ROOT.RooRealVar("x", "x", -10, 10)
# Create Gaussian
sigma = ROOT.RooRealVar("sigma", "sigma", 3, 0.1, 10)
mean = ROOT.RooRealVar("mean", "mean", -3, -10, 10)
gauss = ROOT.RooGaussian("gauss", "gauss", x, mean, sigma)
# Generate a sample of 100 events with sigma=3
data = gauss.generate(ROOT.RooArgSet(x), 100)
# Fit pdf to data
gauss.fitTo(data)
# M a k e p l o t f r a m e s
# -------------------------------
# Make four plot frames to demonstrate various plotting features
frame1 = x.frame(ROOT.RooFit.Name("xframe"), ROOT.RooFit.Title(
"Red Curve / SumW2 Histo errors"), ROOT.RooFit.Bins(20))
frame2 = x.frame(ROOT.RooFit.Name("xframe"), ROOT.RooFit.Title(
"Dashed Curve / No XError bars"), ROOT.RooFit.Bins(20))
frame3 = x.frame(ROOT.RooFit.Name("xframe"), ROOT.RooFit.Title(
"Filled Curve / Blue Histo"), ROOT.RooFit.Bins(20))
frame4 = x.frame(ROOT.RooFit.Name("xframe"), ROOT.RooFit.Title(
"Partial Range / Filled Bar chart"), ROOT.RooFit.Bins(20))
# D a t a p l o t t i n g s t y l e s
# ---------------------------------------
# Use sqrt(sum(weights^2)) error instead of Poisson errors
data.plotOn(frame1, ROOT.RooFit.DataError(ROOT.RooAbsData.SumW2))
# Remove horizontal error bars
data.plotOn(frame2, ROOT.RooFit.XErrorSize(0))
# Blue markers and error bors
data.plotOn(frame3, ROOT.RooFit.MarkerColor(
ROOT.kBlue), ROOT.RooFit.LineColor(ROOT.kBlue))
# Filled bar chart
data.plotOn(frame4, ROOT.RooFit.DrawOption("B"), ROOT.RooFit.DataError(
ROOT.RooAbsData.None), ROOT.RooFit.XErrorSize(0), ROOT.RooFit.FillColor(ROOT.kGray))
# F u n c t i o n p l o t t i n g s t y l e s
# -----------------------------------------------
# Change line color to red
gauss.plotOn(frame1, ROOT.RooFit.LineColor(ROOT.kRed))
# Change line style to dashed
gauss.plotOn(frame2, ROOT.RooFit.LineStyle(ROOT.kDashed))
# Filled shapes in green color
gauss.plotOn(frame3, ROOT.RooFit.DrawOption("F"),
ROOT.RooFit.FillColor(ROOT.kOrange), ROOT.RooFit.MoveToBack())
#
gauss.plotOn(frame4, ROOT.RooFit.Range(-8, 3),
ROOT.RooFit.LineColor(ROOT.kMagenta))
c = ROOT.TCanvas("rf107_plotstyles", "rf107_plotstyles", 800, 800)
c.Divide(2, 2)
c.cd(1)
ROOT.gPad.SetLeftMargin(0.15)
frame1.GetYaxis().SetTitleOffset(1.6)
frame1.Draw()
c.cd(2)
ROOT.gPad.SetLeftMargin(0.15)
frame2.GetYaxis().SetTitleOffset(1.6)
frame2.Draw()
c.cd(3)
ROOT.gPad.SetLeftMargin(0.15)
frame3.GetYaxis().SetTitleOffset(1.6)
frame3.Draw()
c.cd(4)
ROOT.gPad.SetLeftMargin(0.15)
frame4.GetYaxis().SetTitleOffset(1.6)
frame4.Draw()
c.SaveAs("rf107_plotstyles.png")
if __name__ == "__main__":
rf107_plotstyles()