Rf 5 0 4_Simwstool¶

Organisation and simultaneous fits: using RooSimWSTool to construct a simultaneous pdf that is built of variations of an input pdf

Author: Wouter Verkerke
This notebook tutorial was automatically generated with ROOTBOOK-izer from the macro found in the ROOT repository on Wednesday, January 19, 2022 at 10:23 AM.

In :
%%cpp -d
#include "RooRealVar.h"
#include "RooCategory.h"
#include "RooDataSet.h"
#include "RooGaussian.h"
#include "RooPolynomial.h"
#include "RooSimultaneous.h"
#include "RooWorkspace.h"
#include "RooSimWSTool.h"
#include "RooPlot.h"
#include "TCanvas.h"
#include "TAxis.h"
#include "TFile.h"
#include "TH1.h"
In :
%%cpp -d
// This is a workaround to make sure the namespace is used inside functions
using namespace RooFit;

Create master pdf¶

Construct gauss(x,m,s)

In :
RooRealVar x("x", "x", -10, 10);
RooRealVar m("m", "m", 0, -10, 10);
RooRealVar s("s", "s", 1, -10, 10);
RooGaussian gauss("g", "g", x, m, s);
RooFit v3.60 -- Developed by Wouter Verkerke and David Kirkby
Copyright (C) 2000-2013 NIKHEF, University of California & Stanford University

[#0] WARNING:InputArguments -- The parameter 's' with range [-10, 10] of the RooGaussian 'g' exceeds the safe range of (0, inf). Advise to limit its range.

Construct poly(x,p0)

In :
RooRealVar p0("p0", "p0", 0.01, 0., 1.);
RooPolynomial poly("p", "p", x, p0);

Construct model = fgauss(x) + (1-f)poly(x)

In :
RooRealVar f("f", "f", 0.5, 0., 1.);
RooAddPdf model("model", "model", RooArgSet(gauss, poly), f);

Create category observables for splitting¶

Define two categories that can be used for splitting

In :
RooCategory c("c", "c");
c.defineType("run1");
c.defineType("run2");

RooCategory d("d", "d");
d.defineType("foo");
d.defineType("bar");

Setup simwstool¶

Import ingredients in a workspace

In :
RooWorkspace w("w", "w");
w.import(RooArgSet(model, c, d));
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooGaussian::g
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::x
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::m
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::s
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::f
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooCategory::c
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooCategory::d

Make sim builder tool

In :
RooSimWSTool sct(w);

Build a simultaneous model with one split¶

Construct a simultaneous pdf with the following form

model_run1(x) = fgauss_run1(x,m_run1,s) + (1-f)poly model_run2(x) = fgauss_run2(x,m_run2,s) + (1-f)poly simpdf(x,c) = model_run1(x) if c=="run1" = model_run2(x) if c=="run2"

Returned pdf is owned by the workspace

In :
RooSimultaneous *model_sim = sct.build("model_sim", "model", SplitParam("m", "c"));
Splitrule for p.d.f model with state list
parameter m is split with constraint in categories (c)
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: list of prototype pdfs (model)
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: list of splitting categories (c)
[#1] INFO:ObjectHandling -- RooSimPdfBuilder::executeBuild: processing prototype pdf model
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: configured customizers for all prototype pdfs
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode run1
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode run2
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooSimultaneous::model_sim
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_run1
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooGaussian::g_run1
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::m_run1
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_run2
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooGaussian::g_run2
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::m_run2

Print tree structure of model

In :
model_sim->Print("t");
0x7f4fede86cf0 RooSimultaneous::model_sim = 1 [Auto,Dirty]
0x7f4fee049760/V- RooGaussian::g_run1 = 1 [Auto,Dirty]
0x7f4fee04a9e0/V- RooRealVar::m_run1 = 0
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcefd50/V- RooRealVar::s = 1
0x7f4fedd4c6a0/V- RooRealVar::f = 0.5
0x7f4fede0df60/V- RooPolynomial::p = 1 [Auto,Dirty]
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedd39300/V- RooRealVar::p0 = 0.01
0x7f4fee04c710/V- RooGaussian::g_run2 = 1 [Auto,Dirty]
0x7f4fee04d490/V- RooRealVar::m_run2 = 0
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcefd50/V- RooRealVar::s = 1
0x7f4fedd4c6a0/V- RooRealVar::f = 0.5
0x7f4fede0df60/V- RooPolynomial::p = 1 [Auto,Dirty]
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedd39300/V- RooRealVar::p0 = 0.01
0x7f4fedcff2a0/V- RooCategory::c = run2(idx = 1)

In :
w.var("m_run1")->setVal(-3);
w.var("m_run2")->setVal(+3);

Print contents of workspace

In :
w.Print("v");
RooWorkspace(w) w contents

variables
---------
(c,d,f,m,m_run1,m_run2,p0,s,x)

p.d.f.s
-------
RooGaussian::g[ x=x mean=m sigma=s ] = 1
RooGaussian::g_run1[ x=x mean=m_run1 sigma=s ] = 0.011109
RooGaussian::g_run2[ x=x mean=m_run2 sigma=s ] = 0.011109
RooAddPdf::model[ f * g + [%] * p ] = 1
RooAddPdf::model_run1[ f * g_run1 + [%] * p ] = 0.505554
RooAddPdf::model_run2[ f * g_run2 + [%] * p ] = 0.505554
RooSimultaneous::model_sim[ indexCat=c run1=model_run1 run2=model_run2 ] = 0.505554
RooPolynomial::p[ x=x coefList=(p0) ] = 1

Build a simultaneous model with product split¶

Build another simultaneous pdf using a composite split in states c x d

In :
RooSimultaneous *model_sim2 = sct.build("model_sim2", "model", SplitParam("p0", "c,d"));
Splitrule for p.d.f model with state list
parameter p0 is split with constraint in categories (c,d)
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: list of prototype pdfs (model)
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: list of splitting categories (c,d)
[#1] INFO:ObjectHandling -- RooSimPdfBuilder::executeBuild: processing prototype pdf model
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooMultiCategory::c,d
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: configured customizers for all prototype pdfs
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode {run1;foo}
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode {run2;foo}
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode {run1;bar}
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode {run2;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooSimultaneous::model_sim2
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooSuperCategory::model_sim2_index
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooMultiCategory::model_sim2_index_internalMultiCat
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_{run1;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p_{run1;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0_{run1;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_{run2;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p_{run2;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0_{run2;foo}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_{run1;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p_{run1;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0_{run1;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooAddPdf::model_{run2;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p_{run2;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0_{run2;bar}

Print tree structure of this model

In :
model_sim2->Print("t");
0x7f4fee080960 RooSimultaneous::model_sim2 = 1 [Auto,Dirty]
0x7f4fecc15350/V- RooSuperCategory::model_sim2_index = {run2;bar}(idx = 3)
[Auto,Clean]
0x7f4fee089e50/VS RooMultiCategory::model_sim2_index_internalMultiCat = {run2;bar}(idx = 3)
[Auto,Clean]
0x7f4fedcff2a0/VS RooCategory::c = run2(idx = 1)

0x7f4fed8a4290/VS RooCategory::d = bar(idx = 1)

0x7f4fee065100/V- RooPolynomial::p_{run1;foo} = 1 [Auto,Dirty]
0x7f4fee077dc0/V- RooRealVar::p0_{run1;foo} = 0.01
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcf2820/V- RooGaussian::g = 1 [Auto,Dirty]
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcdb490/V- RooRealVar::m = 0
0x7f4fedcefd50/V- RooRealVar::s = 1
0x7f4fedd4c6a0/V- RooRealVar::f = 0.5
0x7f4fee0cf130/V- RooPolynomial::p_{run2;foo} = 1 [Auto,Dirty]
0x7f4fec7c2a40/V- RooRealVar::p0_{run2;foo} = 0.01
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcf2820/V- RooGaussian::g = 1 [Auto,Dirty]
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcdb490/V- RooRealVar::m = 0
0x7f4fedcefd50/V- RooRealVar::s = 1
0x7f4fedd4c6a0/V- RooRealVar::f = 0.5
0x7f4fedd2f4b0/V- RooPolynomial::p_{run1;bar} = 1 [Auto,Dirty]
0x7f4fee06f7d0/V- RooRealVar::p0_{run1;bar} = 0.01
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcf2820/V- RooGaussian::g = 1 [Auto,Dirty]
0x7f4fed7d3650/V- RooRealVar::x = 0
0x7f4fedcdb490/V- RooRealVar::m = 0
0x7f4fedcefd50/V- RooRealVar::s = 1
0x7f4fedd4c6a0/V- RooRealVar::f = 0.5