Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
rf102_dataimport.C File Reference

Detailed Description

View in nbviewer Open in SWAN
Basic functionality: importing data from ROOT TTrees and THx histograms.

#include "RooRealVar.h"
#include "RooDataSet.h"
#include "RooDataHist.h"
#include "RooGaussian.h"
#include "TCanvas.h"
#include "RooPlot.h"
#include "TTree.h"
#include "TH1D.h"
#include "TRandom.h"
using namespace RooFit;
TH1 *makeTH1();
TTree *makeTTree();
{
// ---------------------------------------------------
// I m p o r t i n g R O O T h i s t o g r a m s
// ===================================================
// I m p o r t T H 1 i n t o a R o o D a t a H i s t
// ---------------------------------------------------------
// Create a ROOT TH1 histogram
TH1 *hh = makeTH1();
// Declare observable x
RooRealVar x("x", "x", -10, 10);
// Create a binned dataset that imports contents of TH1 and associates its contents to observable 'x'
RooDataHist dh("dh", "dh", x, Import(*hh));
// P l o t a n d f i t a R o o D a t a H i s t
// ---------------------------------------------------
// Make plot of binned dataset showing Poisson error bars (RooFit default)
RooPlot *frame = x.frame(Title("Imported TH1 with Poisson error bars"));
dh.plotOn(frame);
// Fit a Gaussian pdf to the data
RooRealVar mean("mean", "mean", 0, -10, 10);
RooRealVar sigma("sigma", "sigma", 3, 0.1, 10);
RooGaussian gauss("gauss", "gauss", x, mean, sigma);
gauss.fitTo(dh, PrintLevel(-1));
gauss.plotOn(frame);
// P l o t a n d f i t a R o o D a t a H i s t w i t h i n t e r n a l e r r o r s
// ---------------------------------------------------------------------------------------------
// If histogram has custom error (i.e. its contents is does not originate from a Poisson process
// but e.g. is a sum of weighted events) you can data with symmetric 'sum-of-weights' error instead
// (same error bars as shown by ROOT)
RooPlot *frame2 = x.frame(Title("Imported TH1 with internal errors"));
dh.plotOn(frame2, DataError(RooAbsData::SumW2));
gauss.plotOn(frame2);
// Please note that error bars shown (Poisson or SumW2) are for visualization only, the are NOT used
// in a maximum likelihood fit
//
// A (binned) ML fit will ALWAYS assume the Poisson error interpretation of data (the mathematical definition
// of likelihood does not take any external definition of errors). Data with non-unit weights can only be correctly
// fitted with a chi^2 fit (see rf602_chi2fit.C)
// -----------------------------------------
// I m p o r t i n g R O O T T T r e e s
// =========================================
// I m p o r t T T r e e i n t o a R o o D a t a S e t
// -----------------------------------------------------------
TTree *tree = makeTTree();
// Define 2nd observable y
RooRealVar y("y", "y", -10, 10);
// Construct unbinned dataset importing tree branches x and y matching between branches and RooRealVars
// is done by name of the branch/RRV
//
// Note that ONLY entries for which x,y have values within their allowed ranges as defined in
// RooRealVar x and y are imported. Since the y values in the import tree are in the range [-15,15]
// and RRV y defines a range [-10,10] this means that the RooDataSet below will have less entries than the TTree
// 'tree'
RooDataSet ds("ds", "ds", RooArgSet(x, y), Import(*tree));
// U s e a s c i i i m p o r t / e x p o r t f o r d a t a s e t s
// ------------------------------------------------------------------------------------
{
// Write data to output stream
std::ofstream outstream("rf102_testData.txt");
// Optionally, adjust the stream here (e.g. std::setprecision)
ds.write(outstream);
outstream.close();
}
// Read data from input stream. The variables of the dataset need to be supplied
// to the RooDataSet::read() function.
std::cout << "\n-----------------------\nReading data from ASCII\n";
RooDataSet *dataReadBack =
RooDataSet::read("rf102_testData.txt",
RooArgList(x, y), // variables to be read. If the file has more fields, these are ignored.
"D"); // Prints if a RooFit message stream listens for debug messages. Use Q for quiet.
dataReadBack->Print("V");
std::cout << "\nOriginal data, line 20:\n";
ds.get(20)->Print("V");
std::cout << "\nRead-back data, line 20:\n";
dataReadBack->get(20)->Print("V");
// P l o t d a t a s e t s w i t h m u l t i p l e b i n n i n g c h o i c e s
// ------------------------------------------------------------------------------------
// Print number of events in dataset
ds.Print();
// Print unbinned dataset with default frame binning (100 bins)
RooPlot *frame3 = y.frame(Title("Unbinned data shown in default frame binning"));
ds.plotOn(frame3);
// Print unbinned dataset with custom binning choice (20 bins)
RooPlot *frame4 = y.frame(Title("Unbinned data shown with custom binning"));
ds.plotOn(frame4, Binning(20));
RooPlot *frame5 = y.frame(Title("Unbinned data read back from ASCII file"));
ds.plotOn(frame5, Binning(20));
dataReadBack->plotOn(frame5, Binning(20), MarkerColor(kRed), MarkerStyle(5));
// Draw all frames on a canvas
TCanvas *c = new TCanvas("rf102_dataimport", "rf102_dataimport", 1000, 800);
c->Divide(3, 2);
c->cd(1);
gPad->SetLeftMargin(0.15);
frame->GetYaxis()->SetTitleOffset(1.4);
frame->Draw();
c->cd(2);
gPad->SetLeftMargin(0.15);
frame2->GetYaxis()->SetTitleOffset(1.4);
frame2->Draw();
c->cd(4);
gPad->SetLeftMargin(0.15);
frame3->GetYaxis()->SetTitleOffset(1.4);
frame3->Draw();
c->cd(5);
gPad->SetLeftMargin(0.15);
frame4->GetYaxis()->SetTitleOffset(1.4);
frame4->Draw();
c->cd(6);
gPad->SetLeftMargin(0.15);
frame4->GetYaxis()->SetTitleOffset(1.4);
frame5->Draw();
}
// Create ROOT TH1 filled with a Gaussian distribution
TH1 *makeTH1()
{
TH1D *hh = new TH1D("hh", "hh", 25, -10, 10);
for (int i = 0; i < 100; i++) {
hh->Fill(gRandom->Gaus(0, 3));
}
return hh;
}
// Create ROOT TTree filled with a Gaussian distribution in x and a uniform distribution in y
TTree *makeTTree()
{
TTree *tree = new TTree("tree", "tree");
double *px = new double;
double *py = new double;
tree->Branch("x", px, "x/D");
tree->Branch("y", py, "y/D");
for (int i = 0; i < 100; i++) {
*px = gRandom->Gaus(0, 3);
*py = gRandom->Uniform() * 30 - 15;
tree->Fill();
}
return tree;
}
#define c(i)
Definition RSha256.hxx:101
@ kRed
Definition Rtypes.h:66
R__EXTERN TRandom * gRandom
Definition TRandom.h:62
#define gPad
void Print(Option_t *options=nullptr) const override
This method must be overridden when a class wants to print itself.
virtual RooPlot * plotOn(RooPlot *frame, const RooCmdArg &arg1={}, const RooCmdArg &arg2={}, const RooCmdArg &arg3={}, const RooCmdArg &arg4={}, const RooCmdArg &arg5={}, const RooCmdArg &arg6={}, const RooCmdArg &arg7={}, const RooCmdArg &arg8={}) const
void Print(Option_t *options=nullptr) const override
This method must be overridden when a class wants to print itself.
Definition RooAbsData.h:225
RooArgList is a container object that can hold multiple RooAbsArg objects.
Definition RooArgList.h:22
RooArgSet is a container object that can hold multiple RooAbsArg objects.
Definition RooArgSet.h:24
Container class to hold N-dimensional binned data.
Definition RooDataHist.h:40
Container class to hold unbinned data.
Definition RooDataSet.h:33
const RooArgSet * get(Int_t index) const override
Return RooArgSet with coordinates of event 'index'.
static RooDataSet * read(const char *filename, const RooArgList &variables, const char *opts="", const char *commonPath="", const char *indexCatName=nullptr)
Read data from a text file and create a dataset from it.
Plain Gaussian p.d.f.
Definition RooGaussian.h:24
Plot frame and a container for graphics objects within that frame.
Definition RooPlot.h:45
static RooPlot * frame(const RooAbsRealLValue &var, double xmin, double xmax, Int_t nBins)
Create a new frame for a given variable in x.
Definition RooPlot.cxx:225
TAxis * GetYaxis() const
Definition RooPlot.cxx:1264
void Draw(Option_t *options=nullptr) override
Draw this plot and all of the elements it contains.
Definition RooPlot.cxx:637
Variable that can be changed from the outside.
Definition RooRealVar.h:37
virtual void SetTitleOffset(Float_t offset=1)
Set distance between the axis and the axis title.
Definition TAttAxis.cxx:298
The Canvas class.
Definition TCanvas.h:23
1-D histogram with a double per channel (see TH1 documentation)
Definition TH1.h:670
TH1 is the base class of all histogram classes in ROOT.
Definition TH1.h:59
virtual Int_t Fill(Double_t x)
Increment bin with abscissa X by 1.
Definition TH1.cxx:3344
virtual Double_t Gaus(Double_t mean=0, Double_t sigma=1)
Samples a random number from the standard Normal (Gaussian) Distribution with the given mean and sigm...
Definition TRandom.cxx:275
virtual Double_t Uniform(Double_t x1=1)
Returns a uniform deviate on the interval (0, x1).
Definition TRandom.cxx:682
A TTree represents a columnar dataset.
Definition TTree.h:79
const Double_t sigma
Double_t y[n]
Definition legend1.C:17
Double_t x[n]
Definition legend1.C:17
The namespace RooFit contains mostly switches that change the behaviour of functions of PDFs (or othe...
Definition JSONIO.h:26
[#1] INFO:Fitting -- RooAbsPdf::fitTo(gauss_over_gauss_Int[x]) fixing normalization set for coefficient determination to observables in data
[#1] INFO:Fitting -- using CPU computation library compiled with -mavx2
[#1] INFO:Fitting -- RooAddition::defaultErrorLevel(nll_gauss_over_gauss_Int[x]_dh) Summation contains a RooNLLVar, using its error level
[#1] INFO:Minimization -- RooAbsMinimizerFcn::setOptimizeConst: activating const optimization
[#1] INFO:Minimization -- RooAbsMinimizerFcn::setOptimizeConst: deactivating const optimization
[#1] INFO:DataHandling -- RooTreeDataStore::loadValues(ds) Skipping event #0 because y cannot accommodate the value 14.424
[#1] INFO:DataHandling -- RooTreeDataStore::loadValues(ds) Skipping event #3 because y cannot accommodate the value -12.0022
[#1] INFO:DataHandling -- RooTreeDataStore::loadValues(ds) Skipping event #5 because y cannot accommodate the value 13.8261
[#1] INFO:DataHandling -- RooTreeDataStore::loadValues(ds) Skipping event #6 because y cannot accommodate the value -14.9925
[#1] INFO:DataHandling -- RooTreeDataStore::loadValues(ds) Skipping ...
[#0] WARNING:DataHandling -- RooTreeDataStore::loadValues(ds) Ignored 36 out-of-range events
-----------------------
Reading data from ASCII
[#1] INFO:DataHandling -- RooDataSet::read: reading file rf102_testData.txt
[#1] INFO:DataHandling -- RooDataSet::read: read 64 events (ignored 0 out of range events)
DataStore dataset (rf102_testData.txt)
Contains 64 entries
Observables:
1) x = 0.0174204 L(-10 - 10) "x"
2) y = 9.46654 L(-10 - 10) "y"
3) blindState = Normal(idx = 0)
"Blinding State"
Original data, line 20:
1) RooRealVar:: x = -0.79919
2) RooRealVar:: y = 0.0106407
Read-back data, line 20:
1) RooRealVar:: x = -0.79919
2) RooRealVar:: y = 0.0106407
3) RooCategory:: blindState = Normal(idx = 0)
RooDataSet::ds[x,y] = 64 entries
Date
July 2008
Author
Wouter Verkerke

Definition in file rf102_dataimport.C.