rf504_simwstool.py File Reference

Detailed Description

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

 
import ROOT
 
 
# Create master pdf
# ---------------------------------
 
# Construct gauss(x,m,s)
x = ROOT.RooRealVar("x", "x", -10, 10)
m = ROOT.RooRealVar("m", "m", 0, -10, 10)
s = ROOT.RooRealVar("s", "s", 1, -10, 10)
gauss = ROOT.RooGaussian("g", "g", x, m, s)
 
# Construct poly(x,p0)
p0 = ROOT.RooRealVar("p0", "p0", 0.01, 0.0, 1.0)
poly = ROOT.RooPolynomial("p", "p", x, [p0])
 
# model = f*gauss(x) + (1-f)*poly(x)
f = ROOT.RooRealVar("f", "f", 0.5, 0.0, 1.0)
model = ROOT.RooAddPdf("model", "model", [gauss, poly], [f])
 
# Create category observables for splitting
# ----------------------------------------------------------------------------------
 
# Define two categories that can be used for splitting
c = ROOT.RooCategory("c", "c")
c.defineType("run1")
c.defineType("run2")
 
d = ROOT.RooCategory("d", "d")
d.defineType("foo")
d.defineType("bar")
 
# Set up SimWSTool
# -----------------------------
 
# Import ingredients in a workspace
w = ROOT.RooWorkspace("w", "w")
w.Import([model, c, d])
 
# Make Sim builder tool
sct = ROOT.RooSimWSTool(w)
 
# Build a simultaneous model with one split
# ---------------------------------------------------------------------------------
 
# Construct a simultaneous pdf with the following form
#
# model_run1(x) = f*gauss_run1(x,m_run1,s) + (1-f)*poly
# model_run2(x) = f*gauss_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
model_sim = sct.build("model_sim", "model", SplitParam=("m", "c"))
 
# Print tree structure of model
model_sim.Print("t")
 
# Adjust model_sim parameters in workspace
w.var("m_run1").setVal(-3)
w.var("m_run2").setVal(+3)
 
# Print contents of workspace
w.Print("v")
 
# Build a simultaneous model with product split
# -----------------------------------------------------------------------------------------
 
# Build another simultaneous pdf using a composite split in states c X d
model_sim2 = sct.build("model_sim2", "model", SplitParam=("p0", "c,d"))
 
# Print tree structure of self model
model_sim2.Print("t")

[#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.
[#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
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
0x55f929344800 RooSimultaneous::model_sim = 1 [Auto,Dirty] 
  0x55f92902a830/V- RooCategory::c = run2(idx = 1)
 
  0x55f92938f190/V- RooAddPdf::model_run1 = 1/1 [Auto,Clean] 
    0x55f92934a880/V- RooGaussian::g_run1 = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f92327de70/V- RooRealVar::m_run1 = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f9290fb5e0/V- RooPolynomial::p = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f92327c090/V- RooRealVar::p0 = 0.01
  0x55f9293a9450/V- RooAddPdf::model_run2 = 1/1 [Auto,Clean] 
    0x55f9293a4310/V- RooGaussian::g_run2 = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f9293a6870/V- RooRealVar::m_run2 = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f9290fb5e0/V- RooPolynomial::p = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f92327c090/V- RooRealVar::p0 = 0.01
 
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/1
RooAddPdf::model_run1[ f * g_run1 + [%] * p ] = 0.505554/1
RooAddPdf::model_run2[ f * g_run2 + [%] * p ] = 0.505554/1
RooSimultaneous::model_sim[ indexCat=c run1=model_run1 run2=model_run2 ] = 0.505554
RooPolynomial::p[ x=x coefList=(p0) ] = 1
 
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;bar}
[#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;bar}
[#1] INFO:ObjectHandling -- RooSimWSTool::executeBuild: Customizing prototype pdf model for mode {run2;foo}
[#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;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_{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;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooPolynomial::p_{run2;bar}
[#1] INFO:ObjectHandling -- RooWorkspace::import(w) importing RooRealVar::p0_{run2;bar}
[#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}
0x55f9294b85c0 RooSimultaneous::model_sim2 = 1 [Auto,Dirty] 
  0x55f92937ea00/V- RooSuperCategory::model_sim2_index = {run2;foo}(idx = 1)
 [Auto,Clean] 
    0x55f9294a66c0/VS RooMultiCategory::model_sim2_index_internalMultiCat = {run2;foo}(idx = 1)
 [Auto,Clean] 
      0x55f92902a830/VS RooCategory::c = run2(idx = 1)
 
      0x55f927f453b0/VS RooCategory::d = foo(idx = 0)
 
    0x55f9294a66c0/VS RooMultiCategory::model_sim2_index_internalMultiCat = {run2;foo}(idx = 1)
 [Auto,Clean] 
      0x55f92902a830/VS RooCategory::c = run2(idx = 1)
 
      0x55f927f453b0/VS RooCategory::d = foo(idx = 0)
 
  0x55f9294a6af0/V- RooAddPdf::model_{run1;bar} = 1/1 [Auto,Clean] 
    0x55f92902bd10/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f928af0650/V- RooRealVar::m = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f9294a18c0/V- RooPolynomial::p_{run1;bar} = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f9294a3df0/V- RooRealVar::p0_{run1;bar} = 0.01
  0x55f9294a4470/V- RooAddPdf::model_{run1;foo} = 1/1 [Auto,Clean] 
    0x55f92902bd10/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f928af0650/V- RooRealVar::m = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f9294d8510/V- RooPolynomial::p_{run1;foo} = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f9294daaa0/V- RooRealVar::p0_{run1;foo} = 0.01
  0x55f9294db0b0/V- RooAddPdf::model_{run2;bar} = 1/1 [Auto,Clean] 
    0x55f92902bd10/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f928af0650/V- RooRealVar::m = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f9294bcba0/V- RooPolynomial::p_{run2;bar} = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f9294bf130/V- RooRealVar::p0_{run2;bar} = 0.01
  0x55f9294bf7b0/V- RooAddPdf::model_{run2;foo} = 1/1 [Auto,Clean] 
    0x55f92902bd10/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f928af0650/V- RooRealVar::m = 0
      0x55f927382930/V- RooRealVar::s = 1
    0x55f923278580/V- RooRealVar::f = 0.5
    0x55f928a27a80/V- RooPolynomial::p_{run2;foo} = 1 [Auto,Dirty] 
      0x55f9291f96a0/V- RooRealVar::x = 0
      0x55f928a2a010/V- RooRealVar::p0_{run2;foo} = 0.01

Date

February 2018

Authors

Clemens Lange, Wouter Verkerke (C++ version)

Definition in file rf504_simwstool.py.