Logo ROOT  
Reference Guide
rf405_realtocatfuncs.C
Go to the documentation of this file.
1 /// \file
2 /// \ingroup tutorial_roofit
3 /// \notebook -js
4 ///
5 /// Data and categories: demonstration of real-->discrete mapping functions
6 ///
7 /// \macro_image
8 /// \macro_output
9 /// \macro_code
10 ///
11 /// \date 07/2008
12 /// \author Wouter Verkerke
13 
14 #include "RooRealVar.h"
15 #include "RooDataSet.h"
16 #include "RooGaussian.h"
17 #include "RooConstVar.h"
18 #include "RooCategory.h"
19 #include "RooThresholdCategory.h"
20 #include "RooBinningCategory.h"
21 #include "Roo1DTable.h"
22 #include "RooArgusBG.h"
23 #include "TCanvas.h"
24 #include "TAxis.h"
25 #include "RooPlot.h"
26 using namespace RooFit;
27 
29 {
30 
31  // D e f i n e p d f i n x , s a m p l e d a t a s e t i n x
32  // ------------------------------------------------------------------------
33 
34  // Define a dummy PDF in x
35  RooRealVar x("x", "x", 0, 10);
36  RooArgusBG a("a", "argus(x)", x, RooConst(10), RooConst(-1));
37 
38  // Generate a dummy dataset
39  RooDataSet *data = a.generate(x, 10000);
40 
41  // C r e a t e a t h r e s h o l d r e a l - > c a t f u n c t i o n
42  // --------------------------------------------------------------------------
43 
44  // A RooThresholdCategory is a category function that maps regions in a real-valued
45  // input observable observables to state names. At construction time a 'default'
46  // state name must be specified to which all values of x are mapped that are not
47  // otherwise assigned
48  RooThresholdCategory xRegion("xRegion", "region of x", x, "Background");
49 
50  // Specify thresholds and state assignments one-by-one.
51  // Each statement specifies that all values _below_ the given value
52  // (and above any lower specified threshold) are mapped to the
53  // category state with the given name
54  //
55  // Background | SideBand | Signal | SideBand | Background
56  // 4.23 5.23 8.23 9.23
57  xRegion.addThreshold(4.23, "Background");
58  xRegion.addThreshold(5.23, "SideBand");
59  xRegion.addThreshold(8.23, "Signal");
60  xRegion.addThreshold(9.23, "SideBand");
61 
62  // U s e t h r e s h o l d f u n c t i o n t o p l o t d a t a r e g i o n s
63  // -------------------------------------------------------------------------------------
64 
65  // Add values of threshold function to dataset so that it can be used as observable
66  data->addColumn(xRegion);
67 
68  // Make plot of data in x
69  RooPlot *xframe = x.frame(Title("Demo of threshold and binning mapping functions"));
70  data->plotOn(xframe);
71 
72  // Use calculated category to select sideband data
73  data->plotOn(xframe, Cut("xRegion==xRegion::SideBand"), MarkerColor(kRed), LineColor(kRed));
74 
75  // C r e a t e a b i n n i n g r e a l - > c a t f u n c t i o n
76  // ----------------------------------------------------------------------
77 
78  // A RooBinningCategory is a category function that maps bins of a (named) binning definition
79  // in a real-valued input observable observables to state names. The state names are automatically
80  // constructed from the variable name, the binning name and the bin number. If no binning name
81  // is specified the default binning is mapped
82 
83  x.setBins(10, "coarse");
84  RooBinningCategory xBins("xBins", "coarse bins in x", x, "coarse");
85 
86  // U s e b i n n i n g f u n c t i o n f o r t a b u l a t i o n a n d p l o t t i n g
87  // -----------------------------------------------------------------------------------------------
88 
89  // Print table of xBins state multiplicity. Note that xBins does not need to be an observable in data
90  // it can be a function of observables in data as well
91  Roo1DTable *xbtable = data->table(xBins);
92  xbtable->Print("v");
93 
94  // Add values of xBins function to dataset so that it can be used as observable
95  RooCategory *xb = (RooCategory *)data->addColumn(xBins);
96 
97  // Define range "alt" as including bins 1,3,5,7,9
98  xb->setRange("alt", "x_coarse_bin1,x_coarse_bin3,x_coarse_bin5,x_coarse_bin7,x_coarse_bin9");
99 
100  // Construct subset of data matching range "alt" but only for the first 5000 events and plot it on the frame
101  RooDataSet *dataSel = (RooDataSet *)data->reduce(CutRange("alt"), EventRange(0, 5000));
102  dataSel->plotOn(xframe, MarkerColor(kGreen), LineColor(kGreen));
103 
104  new TCanvas("rf405_realtocatfuncs", "rf405_realtocatfuncs", 600, 600);
105  xframe->SetMinimum(0.01);
106  gPad->SetLeftMargin(0.15);
107  xframe->GetYaxis()->SetTitleOffset(1.4);
108  xframe->Draw();
109 }
RooPlot::Draw
virtual void Draw(Option_t *options=0)
Draw this plot and all of the elements it contains.
Definition: RooPlot.cxx:691
RooArgusBG.h
RooCategory::setRange
void setRange(const char *rangeName, const char *stateNameList)
Definition: RooCategory.cxx:330
RooBinningCategory.h
Roo1DTable
Definition: Roo1DTable.h:23
kGreen
@ kGreen
Definition: Rtypes.h:66
RooFit::EventRange
RooCmdArg EventRange(Int_t nStart, Int_t nStop)
Definition: RooGlobalFunc.cxx:180
RooGaussian.h
Roo1DTable::Print
virtual void Print(Option_t *options=0) const
Print TNamed name and title.
Definition: Roo1DTable.h:64
RooThresholdCategory.h
x
Double_t x[n]
Definition: legend1.C:17
TCanvas.h
RooDataSet.h
RooFit::CutRange
RooCmdArg CutRange(const char *rangeName)
Definition: RooGlobalFunc.cxx:88
rf405_realtocatfuncs
Definition: rf405_realtocatfuncs.py:1
RooAbsData::table
virtual Roo1DTable * table(const RooArgSet &catSet, const char *cuts="", const char *opts="") const
Construct table for product of categories in catSet.
Definition: RooAbsData.cxx:749
RooFit::Cut
RooCmdArg Cut(const char *cutSpec)
Definition: RooGlobalFunc.cxx:80
RooFit
Definition: RooCFunction1Binding.h:29
RooAbsData::plotOn
virtual RooPlot * plotOn(RooPlot *frame, const RooCmdArg &arg1=RooCmdArg::none(), const RooCmdArg &arg2=RooCmdArg::none(), const RooCmdArg &arg3=RooCmdArg::none(), const RooCmdArg &arg4=RooCmdArg::none(), const RooCmdArg &arg5=RooCmdArg::none(), const RooCmdArg &arg6=RooCmdArg::none(), const RooCmdArg &arg7=RooCmdArg::none(), const RooCmdArg &arg8=RooCmdArg::none()) const
Definition: RooAbsData.cxx:547
a
auto * a
Definition: textangle.C:12
RooPlot.h
RooPlot::GetYaxis
TAxis * GetYaxis() const
Definition: RooPlot.cxx:1256
RooPlot
Definition: RooPlot.h:44
RooCategory.h
RooBinningCategory
Definition: RooBinningCategory.h:23
RooRealVar.h
kRed
@ kRed
Definition: Rtypes.h:66
RooConstVar.h
RooFit::LineColor
RooCmdArg LineColor(Color_t color)
Definition: RooGlobalFunc.cxx:56
TCanvas
Definition: TCanvas.h:23
Roo1DTable.h
RooCategory
Definition: RooCategory.h:27
RooAbsData::reduce
RooAbsData * reduce(const RooCmdArg &arg1, const RooCmdArg &arg2=RooCmdArg(), const RooCmdArg &arg3=RooCmdArg(), const RooCmdArg &arg4=RooCmdArg(), const RooCmdArg &arg5=RooCmdArg(), const RooCmdArg &arg6=RooCmdArg(), const RooCmdArg &arg7=RooCmdArg(), const RooCmdArg &arg8=RooCmdArg())
Create a reduced copy of this dataset.
Definition: RooAbsData.cxx:382
TAxis.h
RooArgusBG
Definition: RooArgusBG.h:25
RooPlot::SetMinimum
virtual void SetMinimum(Double_t minimum=-1111)
Set minimum value of Y axis.
Definition: RooPlot.cxx:1091
gPad
#define gPad
Definition: TVirtualPad.h:287
RooDataSet
Definition: RooDataSet.h:33
RooFit::MarkerColor
RooCmdArg MarkerColor(Color_t color)
Definition: RooGlobalFunc.cxx:87
TAttAxis::SetTitleOffset
virtual void SetTitleOffset(Float_t offset=1)
Set distance between the axis and the axis title.
Definition: TAttAxis.cxx:293
RooRealVar
Definition: RooRealVar.h:35
RooFit::Title
RooCmdArg Title(const char *name)
Definition: RooGlobalFunc.cxx:173
RooThresholdCategory
Definition: RooThresholdCategory.h:26
RooDataSet::addColumn
virtual RooAbsArg * addColumn(RooAbsArg &var, Bool_t adjustRange=kTRUE)
Add a column with the values of the given (function) argument to this dataset.
Definition: RooDataSet.cxx:1383
RooFit::RooConst
RooConstVar & RooConst(Double_t val)
Definition: RooGlobalFunc.cxx:341