Example #1
0
one = RooConstVar("one", "one", 1)
mean_gexp = RooRealVar("mean_gexp", "mean_gexp", -0.0529544, -10, 10)
sigma_gexp = RooRealVar("sigma_gexp", "sigma_gexp", 10, 0.1, 50)
rlife1 = RooRealVar("rlife1", "rlife1", 3.2416, 0.1, 10)
gexp1 = RooGExpModel("gexp1", "gexp1", t_diff_st, mean_gexp, sigma_gexp, rlife1, one, one, one)

# 2nd GExp
rlife2 = RooRealVar("rlife2", "rlife2", 6.07734, 0.1, 10)
gexp2 = RooGExpModel("gexp2", "gexp2", t_diff_st, mean_gexp, sigma_gexp, rlife2, one, one, one, False, RooGExpModel.Flipped)

frac_gexp2 = RooRealVar("frac_gexp2", "frac_gexp2", 0.184357, 0.01, 0.99)
gexps = RooAddPdf("gexps", "gexps", RooArgList(gexp2, gexp1), RooArgList(frac_gexp2))

frac_gexps = RooRealVar("frac_gexps", "frac_gexps", 0.0118392, 0.001, 0.99)
model = RooAddPdf("model", "model", RooArgList(gexps, gaussians), RooArgList(frac_gexps))
model.setParameterizeIntegral(RooArgSet(st))


from P2VV import RooFitDecorators
from ROOT import RooWorkspace
from P2VV.RooFitWrappers import RooObject
obj = RooObject(workspace = 'w')
w = obj.ws()
w.put(model)
w.put(sdata)

from ROOT import TPaveText
if options.momentum_calib:
    year_label = TPaveText(0.71, 0.32, 0.89, 0.45, "NDC")
else:
    year_label = TPaveText(0.71, 0.72, 0.89, 0.85, "NDC")