rf504_simwstool.C File Reference

Detailed Description

'ORGANIZATION AND SIMULTANEOUS FITS' RooFit tutorial macro #504

Using RooSimWSTool to construct a simultaneous p.d.f that is built of variations of an input p.d.f

 
 
␛[1mRooFit v3.60 -- Developed by Wouter Verkerke and David Kirkby␛[0m 
                Copyright (C) 2000-2013 NIKHEF, University of California & Stanford University
                All rights reserved, please read http://roofit.sourceforge.net/license.txt
 
[#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
0x5565c24dd100 RooSimultaneous::model_sim = 1 [Auto,Dirty] 
  0x5565c24ff020/V- RooAddPdf::model_run1 = 1 [Auto,Dirty] 
    0x5565c1e59250/V- RooGaussian::g_run1 = 1 [Auto,Dirty] 
      0x5565c1e5ad30/V- RooRealVar::m_run1 = 0
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5
    0x5565c22ee630/V- RooPolynomial::p = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c22c1790/V- RooRealVar::p0 = 0.01
  0x5565c1e5b720/V- RooAddPdf::model_run2 = 1 [Auto,Dirty] 
    0x5565c23f4550/V- RooGaussian::g_run2 = 1 [Auto,Dirty] 
      0x5565c23f52d0/V- RooRealVar::m_run2 = 0
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5
    0x5565c22ee630/V- RooPolynomial::p = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c22c1790/V- RooRealVar::p0 = 0.01
  0x5565c22c25e0/V- RooCategory::c = run2(idx = 1)
 
 
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
 
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 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}
0x5565c27468a0 RooSimultaneous::model_sim2 = 1 [Auto,Dirty] 
  0x5565c2747f90/V- RooSuperCategory::model_sim2_index = {run2;bar}(idx = 3)
 [Auto,Clean] 
    0x5565c22c25e0/VS RooCategory::c = run2(idx = 1)
 
    0x5565c22ebd70/VS RooCategory::d = bar(idx = 1)
 
  0x5565c274a080/V- RooAddPdf::model_{run1;foo} = 1 [Auto,Dirty] 
    0x5565c274f3b0/V- RooPolynomial::p_{run1;foo} = 1 [Auto,Dirty] 
      0x5565c27505e0/V- RooRealVar::p0_{run1;foo} = 0.01
      0x5565c23a71b0/V- RooRealVar::x = 0
    0x5565c23dbd80/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c21c5630/V- RooRealVar::m = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5
  0x5565c2751020/V- RooAddPdf::model_{run2;foo} = 1 [Auto,Dirty] 
    0x5565c2752d70/V- RooPolynomial::p_{run2;foo} = 1 [Auto,Dirty] 
      0x5565c2753fd0/V- RooRealVar::p0_{run2;foo} = 0.01
      0x5565c23a71b0/V- RooRealVar::x = 0
    0x5565c23dbd80/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c21c5630/V- RooRealVar::m = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5
  0x5565c2754a10/V- RooAddPdf::model_{run1;bar} = 1 [Auto,Dirty] 
    0x5565c2756760/V- RooPolynomial::p_{run1;bar} = 1 [Auto,Dirty] 
      0x5565c27579e0/V- RooRealVar::p0_{run1;bar} = 0.01
      0x5565c23a71b0/V- RooRealVar::x = 0
    0x5565c23dbd80/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c21c5630/V- RooRealVar::m = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5
  0x5565c2758420/V- RooAddPdf::model_{run2;bar} = 1 [Auto,Dirty] 
    0x5565c275a150/V- RooPolynomial::p_{run2;bar} = 1 [Auto,Dirty] 
      0x5565c275b3d0/V- RooRealVar::p0_{run2;bar} = 0.01
      0x5565c23a71b0/V- RooRealVar::x = 0
    0x5565c23dbd80/V- RooGaussian::g = 1 [Auto,Dirty] 
      0x5565c23a71b0/V- RooRealVar::x = 0
      0x5565c21c5630/V- RooRealVar::m = 0
      0x5565c1e49e20/V- RooRealVar::s = 1
    0x5565c234a4d0/V- RooRealVar::f = 0.5

 
 
#include "RooRealVar.h"
#include "RooCategory.h"
#include "RooDataSet.h"
#include "RooGaussian.h"
#include "RooConstVar.h"
#include "RooPolynomial.h"
#include "RooSimultaneous.h"
#include "RooAddPdf.h"
#include "RooWorkspace.h"
#include "RooSimWSTool.h"
#include "RooPlot.h"
#include "TCanvas.h"
#include "TAxis.h"
#include "TFile.h"
#include "TH1.h"
using namespace RooFit ;
 
 
void rf504_simwstool()
{
   // C r e a t e   m a s t e r   p d f 
   // ---------------------------------
   
   // Construct gauss(x,m,s)
   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) ;
 
   // Construct poly(x,p0)
   RooRealVar p0("p0","p0",0.01,0.,1.) ;
   RooPolynomial poly("p","p",x,p0) ;      
 
   // Construct model = f*gauss(x) + (1-f)*poly(x)
   RooRealVar f("f","f",0.5,0.,1.) ;
   RooAddPdf model("model","model",RooArgSet(gauss,poly),f) ;
 
 
 
   // C r e a t e   c a t e g o r y   o b s e r v a b l e s   f o r   s p l i t t i n g
   // ----------------------------------------------------------------------------------
 
   // Define two categories that can be used for splitting
   RooCategory c("c","c") ;
   c.defineType("run1") ;
   c.defineType("run2") ;
 
   RooCategory d("d","d") ;
   d.defineType("foo") ;
   d.defineType("bar") ;
 
 
 
   // S e t u p   S i m W S T o o l 
   // -----------------------------
 
   // Import ingredients in a workspace
   RooWorkspace w("w","w") ;
   w.import(RooArgSet(model,c,d)) ;
 
   // Make Sim builder tool
   RooSimWSTool sct(w) ;
 
 
   // B u i l d   a   s i m u l t a n e o u s   m o d e l   w i t h   o n e   s p l i t
   // ---------------------------------------------------------------------------------
 
   // Construct a simultaneous p.d.f 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 p.d.f is owned by the workspace
   RooSimultaneous* 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") ;
 
 
 
   // B u i l d   a   s i m u l t a n e o u s   m o d e l   w i t h   p r o d u c t   s p l i t
   // -----------------------------------------------------------------------------------------
 
   // Build another simultaneous p.d.f using a composite split in states c X d
   RooSimultaneous* model_sim2 = sct.build("model_sim2","model",SplitParam("p0","c,d")) ;
 
   // Print tree structure of this model
   model_sim2->Print("t") ;
 
}

Author

07/2008 - Wouter Verkerke

Definition in file rf504_simwstool.C.