rf511_wsfactory_basic.C

Go to the documentation of this file.

/// \file
/// \ingroup tutorial_roofit
/// \notebook -nodraw
///  'ORGANIZATION AND SIMULTANEOUS FITS' RooFit tutorial macro #511
///
///   Basic use of the 'object factory' associated with a workspace
///   to rapidly build p.d.f.s functions and their parameter components
///
/// \macro_output
/// \macro_code
/// \author 04/2009 - Wouter Verkerke



#include "RooRealVar.h"
#include "RooDataSet.h"
#include "RooGaussian.h"
#include "RooConstVar.h"
#include "RooChebychev.h"
#include "RooAddPdf.h"
#include "RooWorkspace.h"
#include "RooPlot.h"
#include "TCanvas.h"
#include "TAxis.h"
using namespace RooFit ;


void rf511_wsfactory_basic(Bool_t compact=kFALSE)
{
   RooWorkspace* w = new RooWorkspace("w") ;

   // C r e a t i n g   a n d   a d d i n g   b a s i c  p . d . f . s
   // ----------------------------------------------------------------


   // Remake example p.d.f. of tutorial rs502_wspacewrite.C:
   //
   // Basic p.d.f. construction: ClassName::ObjectName(constructor arguments)
   // Variable construction    : VarName[x,xlo,xhi], VarName[xlo,xhi], VarName[x]
   // P.d.f. addition          : SUM::ObjectName(coef1*pdf1,...coefM*pdfM,pdfN)
   //

   if (!compact) {

      // Use object factory to build p.d.f. of tutorial rs502_wspacewrite
      w->factory("Gaussian::sig1(x[-10,10],mean[5,0,10],0.5)") ;
      w->factory("Gaussian::sig2(x,mean,1)") ;
      w->factory("Chebychev::bkg(x,{a0[0.5,0.,1],a1[0.2,0.,1.]})") ;
      w->factory("SUM::sig(sig1frac[0.8,0.,1.]*sig1,sig2)") ;
      w->factory("SUM::model(bkgfrac[0.5,0.,1.]*bkg,sig)") ;

   } else {

      // Use object factory to build p.d.f. of tutorial rs502_wspacewrite but
      //  - Contracted to a single line recursive expression,
      //  - Omitting explicit names for components that are not referred to explicitly later

      w->factory("SUM::model(bkgfrac[0.5,0.,1.]*Chebychev::bkg(x[-10,10],{a0[0.5,0.,1],a1[0.2,0.,1.]}),"
                                                "SUM(sig1frac[0.8,0.,1.]*Gaussian(x,mean[5,0,10],0.5), Gaussian(x,mean,1)))") ;
   }


   // A d v a n c e d   p . d . f .  c o n s t r u c t o r   a r g u m e n t s
   // ----------------------------------------------------------------
   //
   // P.d.f. constructor arguments may by any type of RooAbsArg, but also
   //
   // Double_t --> converted to RooConst(...)
   // {a,b,c} --> converted to RooArgSet() or RooArgList() depending on required ctor arg
   // dataset name --> converted to RooAbsData reference for any dataset residing in the workspace
   // enum --> Any enum label that belongs to an enum defined in the (base) class


   // Make a dummy dataset p.d.f. 'model' and import it in the workspace
   RooDataSet* data = w->pdf("model")->generate(*w->var("x"),1000) ;
   w->import(*data,Rename("data")) ;

   // Construct a KEYS p.d.f. passing a dataset name and an enum type defining the
   // mirroring strategy
   w->factory("KeysPdf::k(x,data,NoMirror,0.2)") ;


   // Print workspace contents
   w->Print() ;

   // Make workspace visible on command line
   gDirectory->Add(w) ;

}