doc/v636/rf309__ndimplot_8C.html

#include "RooRealVar.h"

#include "RooDataSet.h"

#include "RooGaussian.h"

#include "RooProdPdf.h"

#include "TCanvas.h"

#include "TAxis.h"

#include "TH1.h"

#include "RooPlot.h"

using namespace RooFit;


void rf309_ndimplot()

{


   // C r e a t e   2 D   m o d e l   a n d   d a t a s e t

   // -----------------------------------------------------


   // Create observables

   RooRealVar x("x", "x", -5, 5);

   RooRealVar y("y", "y", -5, 5);


   // Create parameters

   RooRealVar a0("a0", "a0", -3.5, -5, 5);

   RooRealVar a1("a1", "a1", -1.5, -1, 1);

   RooRealVar sigma("sigma", "width of gaussian", 1.5);


   // Create interpreted function f(y) = a0 - a1*sqrt(10*abs(y))

   RooFormulaVar fy("fy", "a0-a1*sqrt(10*abs(y))", RooArgSet(y, a0, a1));


   // Create gauss(x,f(y),s)

   RooGaussian model("model", "Gaussian with shifting mean", x, fy, sigma);


   // Sample dataset from gauss(x,y)

   std::unique_ptr<RooDataSet> data{model.generate({x, y}, 10000)};


   // M a k e   2 D   p l o t s   o f   d a t a   a n d   m o d e l

   // -------------------------------------------------------------


   // Create and fill ROOT 2D histogram (20x20 bins) with contents of dataset

   TH1 *hh_data = data->createHistogram("x,y", Binning(20), Binning(20));


   // Create and fill ROOT 2D histogram (50x50 bins) with sampling of pdf

   // TH2D* hh_pdf = model.createHistogram("hh_model",x,Binning(50),YVar(y,Binning(50))) ;

   TH1 *hh_pdf = model.createHistogram("x,y", 50, 50);

   hh_pdf->SetLineColor(kBlue);


   // C r e a t e   3 D   m o d e l   a n d   d a t a s e t

   // -----------------------------------------------------


   // Create observables

   RooRealVar z("z", "z", -5, 5);


   RooGaussian gz("gz", "gz", z, 0.0, 2.0);

   RooProdPdf model3("model3", "model3", RooArgSet(model, gz));


   std::unique_ptr<RooDataSet> data3{model3.generate({x, y, z}, 10000)};


   // M a k e   3 D   p l o t s   o f   d a t a   a n d   m o d e l

   // -------------------------------------------------------------


   // Create and fill ROOT 2D histogram (8x8x8 bins) with contents of dataset

   TH1 *hh_data3 = data3->createHistogram("hh_data3", x, Binning(8), YVar(y, Binning(8)), ZVar(z, Binning(8)));


   // Create and fill ROOT 2D histogram (20x20x20 bins) with sampling of pdf

   TH1 *hh_pdf3 = model3.createHistogram("hh_model3", x, Binning(20), YVar(y, Binning(20)), ZVar(z, Binning(20)));

   hh_pdf3->SetFillColor(kBlue);


   TCanvas *c1 = new TCanvas("rf309_2dimplot", "rf309_2dimplot", 800, 800);

   c1->Divide(2, 2);

   c1->cd(1);

   gPad->SetLeftMargin(0.15);

   hh_data->GetZaxis()->SetTitleOffset(1.4);

   hh_data->Draw("lego");

   c1->cd(2);

   gPad->SetLeftMargin(0.20);

   hh_pdf->GetZaxis()->SetTitleOffset(2.5);

   hh_pdf->Draw("surf");

   c1->cd(3);

   gPad->SetLeftMargin(0.15);

   hh_data->GetZaxis()->SetTitleOffset(1.4);

   hh_data->Draw("box");

   c1->cd(4);

   gPad->SetLeftMargin(0.15);

   hh_pdf->GetZaxis()->SetTitleOffset(2.5);

   hh_pdf->Draw("cont3");


   TCanvas *c2 = new TCanvas("rf309_3dimplot", "rf309_3dimplot", 800, 400);

   c2->Divide(2);

   c2->cd(1);

   gPad->SetLeftMargin(0.15);

   hh_data3->GetZaxis()->SetTitleOffset(1.4);

   hh_data3->Draw("lego");

   c2->cd(2);

   gPad->SetLeftMargin(0.15);

   hh_pdf3->GetZaxis()->SetTitleOffset(1.4);

   hh_pdf3->Draw("iso");

}

RooDataSet.h

RooGaussian.h

RooPlot.h

RooProdPdf.h

RooRealVar.h

kBlue
@ kBlue
Definition Rtypes.h:66

TAxis.h

TCanvas.h

TRangeDynCast
ROOT::Detail::TRangeCast< T, true > TRangeDynCast
TRangeDynCast is an adapter class that allows the typed iteration through a TCollection.
Definition TCollection.h:358

data
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void data
Definition TGWin32VirtualXProxy.cxx:104

TH1.h

gPad
#define gPad
Definition TVirtualPad.h:308

ROOT::Detail::TRangeCast
Definition TCollection.h:311

RooArgSet
RooArgSet is a container object that can hold multiple RooAbsArg objects.
Definition RooArgSet.h:24

RooFormulaVar
A RooFormulaVar is a generic implementation of a real-valued object, which takes a RooArgList of serv...
Definition RooFormulaVar.h:30

RooGaussian
Plain Gaussian p.d.f.
Definition RooGaussian.h:24

RooProdPdf
Efficient implementation of a product of PDFs of the form.
Definition RooProdPdf.h:36

RooRealVar
Variable that can be changed from the outside.
Definition RooRealVar.h:37

TCanvas
The Canvas class.
Definition TCanvas.h:23

TH1
TH1 is the base class of all histogram classes in ROOT.
Definition TH1.h:108

RooFit::ZVar
RooCmdArg ZVar(const RooAbsRealLValue &var, const RooCmdArg &arg={})
Definition RooGlobalFunc.cxx:830

RooFit::YVar
RooCmdArg YVar(const RooAbsRealLValue &var, const RooCmdArg &arg={})
Definition RooGlobalFunc.cxx:826

RooFit::Binning
RooCmdArg Binning(const RooAbsBinning &binning)
Definition RooGlobalFunc.cxx:308

sigma
const Double_t sigma
Definition h1analysisProxy.h:11

y
Double_t y[n]
Definition legend1.C:17

c1
return c1
Definition legend1.C:41

x
Double_t x[n]
Definition legend1.C:17

c2
return c2
Definition legend2.C:14

RooFit
The namespace RooFit contains mostly switches that change the behaviour of functions of PDFs (or othe...
Definition CodegenImpl.h:65

rf309_ndimplot
Definition rf309_ndimplot.py:1