doc/master/hf001__example_8py.html

import ROOT


"""

Create a HistFactory measurement from python

"""


InputFile = "./data/example.root"

if ROOT.gSystem.AccessPathName(InputFile):

    ROOT.Info("example.py", InputFile + " does not exist")

    ROOT.gROOT.ProcessLine(".! prepareHistFactory .")

    if ROOT.gSystem.AccessPathName(InputFile):

        ROOT.Info("example.py", InputFile + " still does not exist. \n EXIT")

        exit()


# Create the measurement

meas = ROOT.RooStats.HistFactory.Measurement("meas", "meas")


meas.SetOutputFilePrefix("./results/example_UsingPy")

meas.SetPOI("SigXsecOverSM")

meas.AddConstantParam("Lumi")

meas.AddConstantParam("alpha_syst1")


meas.SetLumi(1.0)

meas.SetLumiRelErr(0.10)


# Create a channel


chan = ROOT.RooStats.HistFactory.Channel("channel1")

chan.SetData("data", InputFile)

chan.SetStatErrorConfig(0.05, "Poisson")


# Now, create some samples


# Create the signal sample

signal = ROOT.RooStats.HistFactory.Sample("signal", "signal", InputFile)

signal.AddOverallSys("syst1", 0.95, 1.05)

signal.AddNormFactor("SigXsecOverSM", 1, 0, 3)

chan.AddSample(signal)


# Background 1

background1 = ROOT.RooStats.HistFactory.Sample("background1", "background1", InputFile)

background1.ActivateStatError("background1_statUncert", InputFile)

background1.AddOverallSys("syst2", 0.95, 1.05)

chan.AddSample(background1)


# Background 1

background2 = ROOT.RooStats.HistFactory.Sample("background2", "background2", InputFile)

background2.ActivateStatError()

background2.AddOverallSys("syst3", 0.95, 1.05)

chan.AddSample(background2)


# Done with this channel

# Add it to the measurement:


meas.AddChannel(chan)


# Collect the histograms from their files,

# print some output,

meas.CollectHistograms()

meas.PrintTree()


# One can print XML code to an output directory:

# meas.PrintXML("xmlFromPyCode", meas.GetOutputFilePrefix())


# Now, do the measurement

ws = ROOT.RooStats.HistFactory.MakeModelAndMeasurementFast(meas)


# Retrieve the ModelConfig

modelConfig = ws["ModelConfig"]


# Extract the PDF and global observables

pdf = modelConfig.GetPdf()


# Perform the fit using Minos to get the correct asymmetric uncertainties

result = pdf.fitTo(

    ws["obsData"], Save=True, PrintLevel=-1, GlobalObservables=modelConfig.GetGlobalObservables(), Minos=True

)


# Getting list of Parameters of Interest and getting first from them

poi = modelConfig.GetParametersOfInterest().first()


nll = pdf.createNLL(ws["obsData"])

profile = nll.createProfile(poi)


# frame for future plot

frame = poi.frame()


frame.SetTitle("")

frame.GetYaxis().SetTitle("-log likelihood")

frame.GetXaxis().SetTitle(poi.GetTitle())


profileLikelihoodCanvas = ROOT.TCanvas("combined", "", 800, 600)


xmin = poi.getMin()

xmax = poi.getMax()


line = ROOT.TLine(xmin, 0.5, xmax, 0.5)

line.SetLineColor(ROOT.kGreen)

line90 = ROOT.TLine(xmin, 2.71 / 2, xmax, 2.71 / 2)

line90.SetLineColor(ROOT.kGreen)

line95 = ROOT.TLine(xmin, 3.84 / 2, xmax, 3.84 / 2)

line95.SetLineColor(ROOT.kGreen)


frame.addObject(line)

frame.addObject(line90)

frame.addObject(line95)


nll.plotOn(frame, ShiftToZero=True, LineColor="r", LineStyle="--")

profile.plotOn(frame)


frame.SetMinimum(0)

frame.SetMaximum(2)


frame.Draw()


# Print fit results to console in verbose mode to see asymmetric uncertainties

result.Print("v")