Logo ROOT  
Reference Guide
RooNDKeysPdf.h
Go to the documentation of this file.
1/*****************************************************************************
2 * Project: RooFit *
3 * Package: RooFitModels *
4 * File: $Id: RooNDKeysPdf.h 44368 2012-05-30 15:38:44Z axel $
5 * Authors: *
6 * Max Baak, CERN, mbaak@cern.ch *
7 * *
8 * Copyright (c) 2000-2005, Regents of the University of California *
9 * and Stanford University. All rights reserved. *
10 * *
11 * Redistribution and use in source and binary forms, *
12 * with or without modification, are permitted according to the terms *
13 * listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
14 *****************************************************************************/
15#ifndef ROO_NDKEYS_PDF
16#define ROO_NDKEYS_PDF
17
18#include "RooAbsPdf.h"
19#include "RooRealProxy.h"
20#include "RooSetProxy.h"
21#include "RooDataSet.h"
22#include "RooListProxy.h"
23#include "TH1.h"
24#include "TAxis.h"
25#include "TVectorD.h"
26#include "TMatrixD.h"
27#include "TMatrixDSym.h"
28#include <map>
29#include <vector>
30#include <string>
31
32class RooRealVar;
33class RooArgList;
34class RooArgSet;
36
37#ifndef __CINT__
38class VecVecDouble : public std::vector<std::vector<Double_t> > { } ;
39class VecTVecDouble : public std::vector<TVectorD> { } ;
40typedef std::pair<Int_t, VecVecDouble::iterator > iiPair;
41typedef std::vector< iiPair > iiVec;
42typedef std::pair<Int_t, VecTVecDouble::iterator > itPair;
43typedef std::vector< itPair > itVec;
44#else
45class itPair ;
46#endif
47
48class RooNDKeysPdf : public RooAbsPdf {
49
50public:
51
56
57 RooNDKeysPdf() = default;
58
59 RooNDKeysPdf(const char *name, const char *title, const RooArgList &varList, const RooDataSet &data,
60 TString options = "ma", Double_t rho = 1, Double_t nSigma = 3, Bool_t rotate = kTRUE,
61 Bool_t sortInput = kTRUE);
62
63 RooNDKeysPdf(const char *name, const char *title, const RooArgList &varList, const TH1 &hist, TString options = "ma",
64 Double_t rho = 1, Double_t nSigma = 3, Bool_t rotate = kTRUE, Bool_t sortInput = kTRUE);
65
66 RooNDKeysPdf(const char *name, const char *title, const RooArgList &varList, const RooDataSet &data,
67 const TVectorD &rho, TString options = "ma", Double_t nSigma = 3, Bool_t rotate = kTRUE,
68 Bool_t sortInput = kTRUE);
69
70 RooNDKeysPdf(const char *name, const char *title, const RooArgList &varList, const RooDataSet &data,
71 const RooArgList &rhoList, TString options = "ma", Double_t nSigma = 3, Bool_t rotate = kTRUE,
72 Bool_t sortInput = kTRUE);
73
74 RooNDKeysPdf(const char *name, const char *title, const RooArgList &varList, const TH1 &hist,
75 const RooArgList &rhoList, TString options = "ma", Double_t nSigma = 3, Bool_t rotate = kTRUE,
76 Bool_t sortInput = kTRUE);
77
78 RooNDKeysPdf(const char *name, const char *title, RooAbsReal &x, const RooDataSet &data, Mirror mirror = NoMirror,
79 Double_t rho = 1, Double_t nSigma = 3, Bool_t rotate = kTRUE, Bool_t sortInput = kTRUE);
80
81 RooNDKeysPdf(const char *name, const char *title, RooAbsReal &x, RooAbsReal &y, const RooDataSet &data,
82 TString options = "ma", Double_t rho = 1.0, Double_t nSigma = 3, Bool_t rotate = kTRUE,
83 Bool_t sortInput = kTRUE);
84
85 RooNDKeysPdf(const RooNDKeysPdf& other, const char* name=0);
86 virtual ~RooNDKeysPdf();
87
88 virtual TObject* clone(const char* newname) const { return new RooNDKeysPdf(*this,newname); }
89
90 Int_t getAnalyticalIntegral(RooArgSet& allVars, RooArgSet& analVars, const char* rangeName=0) const ;
91 Double_t analyticalIntegral(Int_t code, const char* rangeName=0) const ;
92
93 inline void fixShape(Bool_t fix) {
95 _fixedShape=fix;
96 }
97
98 TMatrixD getWeights(const int &k) const;
99
100 struct BoxInfo {
105 std::vector<Double_t> xVarLo, xVarHi;
106 std::vector<Double_t> xVarLoM3s, xVarLoP3s, xVarHiM3s, xVarHiP3s;
107 std::map<Int_t,Bool_t> bpsIdcs;
108 std::vector<Int_t> sIdcs;
109 std::vector<Int_t> bIdcs;
110 std::vector<Int_t> bmsIdcs;
111 } ;
112
113protected:
114
117
118 Double_t evaluate() const;
119
120 void createPdf(Bool_t firstCall = kTRUE);
121 void setOptions();
122 void initialize();
123 void loadDataSet(Bool_t firstCall);
124 void mirrorDataSet();
125 void loadWeightSet();
126 void calculateShell(BoxInfo *bi) const;
127 void calculatePreNorm(BoxInfo *bi) const;
128 void sortDataIndices(BoxInfo *bi = 0);
129 void calculateBandWidth();
130 Double_t gauss(std::vector<Double_t> &x, std::vector<std::vector<Double_t>> &weights) const;
131 void loopRange(std::vector<Double_t> &x, std::map<Int_t, Bool_t> &ibMap) const;
132 void boxInfoInit(BoxInfo *bi, const char *rangeName, Int_t code) const;
133 RooDataSet *createDatasetFromHist(const RooArgList &varList, const TH1 &hist) const;
134 void updateRho() const;
135 void checkInitWeights() const {
136 if (_weights == &_weights0 || _weights == &_weights1)
137 return;
138 const_cast<RooNDKeysPdf*>(this)->calculateBandWidth();
139 }
140
141 std::unique_ptr<RooDataSet> _ownedData{nullptr};
142 const RooDataSet* _data; //! do not persist
146
149 Bool_t _debug{false}; //!
150 Bool_t _verbose{false}; //!
151
158
159 // cached info on variable
160 std::vector<std::vector<Double_t> > _dataPts;
161 std::vector<TVectorD> _dataPtsR;
162 std::vector<std::vector<Double_t> > _weights0; // Plain weights
163 std::vector<std::vector<Double_t> > _weights1; // Weights for adaptive kernels
164 std::vector<std::vector<Double_t> >* _weights{nullptr}; //! Weights to be used. Points either to _weights0 or _weights1
165
166 std::vector<itVec> _sortTVIdcs; //!
167
168 std::vector<std::string> _varName;
169 mutable std::vector<Double_t> _rho;
171 mutable std::vector<Double_t> _x; // Cache for x values
172 std::vector<Double_t> _x0, _x1, _x2;
173 std::vector<Double_t> _mean, _sigma;
174 std::vector<Double_t> _xDatLo, _xDatHi;
175 std::vector<Double_t> _xDatLo3s, _xDatHi3s;
176
180 std::vector<Double_t> _xVarLo, _xVarHi;
181 std::vector<Double_t> _xVarLoM3s, _xVarLoP3s, _xVarHiM3s, _xVarHiP3s;
182 std::map<Int_t,Bool_t> _bpsIdcs;
183 std::map<Int_t, Bool_t> _ibNoSort;
184 std::vector<Int_t> _sIdcs;
185 std::vector<Int_t> _bIdcs;
186 std::vector<Int_t> _bmsIdcs;
187
188 // Data for analytical integrals:
189 mutable std::map<std::pair<std::string,int>,BoxInfo*> _rangeBoxInfo ;
191
192 std::vector<Int_t> _idx;
195 std::map<Int_t,Double_t> _wMap;
196
199 TMatrixD* _rotMat{nullptr};
200 TVectorD* _sigmaR{nullptr};
201 TVectorD* _dx{nullptr};
203
207
209
210 ClassDef(RooNDKeysPdf, 1) // General N-dimensional non-parametric kernel estimation p.d.f
211};
212
213#endif
std::pair< Int_t, VecVecDouble::iterator > iiPair
Definition: RooNDKeysPdf.h:40
std::vector< itPair > itVec
Definition: RooNDKeysPdf.h:43
std::vector< iiPair > iiVec
Definition: RooNDKeysPdf.h:41
std::pair< Int_t, VecTVecDouble::iterator > itPair
Definition: RooNDKeysPdf.h:42
int Int_t
Definition: RtypesCore.h:45
const Bool_t kFALSE
Definition: RtypesCore.h:101
bool Bool_t
Definition: RtypesCore.h:63
double Double_t
Definition: RtypesCore.h:59
const Bool_t kTRUE
Definition: RtypesCore.h:100
#define ClassDef(name, id)
Definition: Rtypes.h:325
char name[80]
Definition: TGX11.cxx:110
RooAbsReal is the common abstract base class for objects that represent a real value and implements f...
Definition: RooAbsReal.h:61
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:35
RooChangeTracker is a meta object that tracks value changes in a given set of RooAbsArgs by registeri...
RooDataSet is a container class to hold unbinned data.
Definition: RooDataSet.h:36
RooListProxy is the concrete proxy for RooArgList objects.
Definition: RooListProxy.h:24
Generic N-dimensional implementation of a kernel estimation p.d.f.
Definition: RooNDKeysPdf.h:48
BoxInfo _fullBoxInfo
Definition: RooNDKeysPdf.h:190
RooListProxy _rhoList
Definition: RooNDKeysPdf.h:116
Double_t _nSigma
Definition: RooNDKeysPdf.h:145
TVectorD * _dx
Definition: RooNDKeysPdf.h:201
std::unique_ptr< RooDataSet > _ownedData
Definition: RooNDKeysPdf.h:141
void loadWeightSet()
Double_t _minWeight
Definition: RooNDKeysPdf.h:193
std::vector< Int_t > _sIdcs
Definition: RooNDKeysPdf.h:184
void initialize()
initialization
std::vector< Double_t > _xVarLo
Definition: RooNDKeysPdf.h:180
void calculatePreNorm(BoxInfo *bi) const
bi->nEventsBMSW=0.
Double_t analyticalIntegral(Int_t code, const char *rangeName=0) const
Implements the actual analytical integral(s) advertised by getAnalyticalIntegral.
TVectorD * _sigmaR
Definition: RooNDKeysPdf.h:200
std::vector< Double_t > _xDatLo3s
Definition: RooNDKeysPdf.h:175
Double_t _nEventsBW
Definition: RooNDKeysPdf.h:178
std::vector< Double_t > _x0
Definition: RooNDKeysPdf.h:172
virtual ~RooNDKeysPdf()
std::vector< Double_t > _xVarLoP3s
Definition: RooNDKeysPdf.h:181
Bool_t _fixedShape
Definition: RooNDKeysPdf.h:147
Bool_t _sortInput
Definition: RooNDKeysPdf.h:205
Bool_t _verbose
Definition: RooNDKeysPdf.h:150
std::vector< Double_t > _xDatHi3s
Definition: RooNDKeysPdf.h:175
const RooDataSet * _data
Definition: RooNDKeysPdf.h:142
std::vector< Int_t > _bIdcs
Definition: RooNDKeysPdf.h:185
std::vector< TVectorD > _dataPtsR
Definition: RooNDKeysPdf.h:161
void loopRange(std::vector< Double_t > &x, std::map< Int_t, Bool_t > &ibMap) const
determine closest points to x, to loop over in evaluate()
void fixShape(Bool_t fix)
Definition: RooNDKeysPdf.h:93
std::vector< Double_t > _xVarHi
Definition: RooNDKeysPdf.h:180
void createPdf(Bool_t firstCall=kTRUE)
evaluation order of constructor.
Double_t _nEventsBMSW
Definition: RooNDKeysPdf.h:179
void updateRho() const
Double_t _nEventsW
Definition: RooNDKeysPdf.h:155
void calculateShell(BoxInfo *bi) const
determine points in +/- nSigma shell around the box determined by the variable ranges.
void calculateBandWidth()
std::vector< Double_t > _x1
Definition: RooNDKeysPdf.h:172
Bool_t _netFluxZ
Definition: RooNDKeysPdf.h:177
Double_t _maxWeight
Definition: RooNDKeysPdf.h:194
RooArgSet _dataVars
Definition: RooNDKeysPdf.h:170
std::vector< Double_t > _mean
Definition: RooNDKeysPdf.h:173
Bool_t _rotate
Definition: RooNDKeysPdf.h:204
RooListProxy _varList
Definition: RooNDKeysPdf.h:115
TMatrixDSym * _corrMat
Definition: RooNDKeysPdf.h:198
std::vector< Int_t > _bmsIdcs
Definition: RooNDKeysPdf.h:186
std::vector< Double_t > _rho
Definition: RooNDKeysPdf.h:169
std::vector< Double_t > _xDatHi
Definition: RooNDKeysPdf.h:174
std::vector< Double_t > _x
Definition: RooNDKeysPdf.h:171
std::vector< itVec > _sortTVIdcs
Weights to be used. Points either to _weights0 or _weights1.
Definition: RooNDKeysPdf.h:166
void boxInfoInit(BoxInfo *bi, const char *rangeName, Int_t code) const
std::map< std::pair< std::string, int >, BoxInfo * > _rangeBoxInfo
Definition: RooNDKeysPdf.h:189
std::vector< Double_t > _xVarLoM3s
Definition: RooNDKeysPdf.h:181
Double_t _d
Definition: RooNDKeysPdf.h:156
Double_t _sigmaAvgR
Definition: RooNDKeysPdf.h:202
Double_t _widthFactor
Definition: RooNDKeysPdf.h:144
std::vector< Int_t > _idx
Definition: RooNDKeysPdf.h:192
std::vector< Double_t > _xDatLo
Definition: RooNDKeysPdf.h:174
void sortDataIndices(BoxInfo *bi=0)
sort entries, as needed for loopRange()
Int_t getAnalyticalIntegral(RooArgSet &allVars, RooArgSet &analVars, const char *rangeName=0) const
Interface function getAnalyticalIntergral advertises the analytical integrals that are supported.
void loadDataSet(Bool_t firstCall)
copy the dataset and calculate some useful variables
TString _options
do not persist
Definition: RooNDKeysPdf.h:143
std::vector< std::vector< Double_t > > * _weights
Definition: RooNDKeysPdf.h:164
TMatrixD getWeights(const int &k) const
Return evaluated weights.
std::vector< Double_t > _xVarHiM3s
Definition: RooNDKeysPdf.h:181
Double_t evaluate() const
Evaluate this PDF / function / constant. Needs to be overridden by all derived classes.
std::map< Int_t, Bool_t > _ibNoSort
Definition: RooNDKeysPdf.h:183
std::vector< std::vector< Double_t > > _weights0
Definition: RooNDKeysPdf.h:162
void mirrorDataSet()
determine mirror dataset.
virtual TObject * clone(const char *newname) const
Definition: RooNDKeysPdf.h:88
void setOptions()
set the configuration
std::vector< std::vector< Double_t > > _dataPts
Definition: RooNDKeysPdf.h:160
RooDataSet * createDatasetFromHist(const RooArgList &varList, const TH1 &hist) const
RooNDKeysPdf()=default
std::vector< Double_t > _sigma
Definition: RooNDKeysPdf.h:173
std::vector< std::string > _varName
Definition: RooNDKeysPdf.h:168
Double_t gauss(std::vector< Double_t > &x, std::vector< std::vector< Double_t > > &weights) const
loop over all closest point to x, as determined by loopRange()
void checkInitWeights() const
Definition: RooNDKeysPdf.h:135
Double_t _n
Definition: RooNDKeysPdf.h:157
std::map< Int_t, Double_t > _wMap
Definition: RooNDKeysPdf.h:195
std::vector< std::vector< Double_t > > _weights1
Definition: RooNDKeysPdf.h:163
TMatrixD * _rotMat
Definition: RooNDKeysPdf.h:199
Bool_t _mirror
Definition: RooNDKeysPdf.h:148
std::vector< Double_t > _x2
Definition: RooNDKeysPdf.h:172
TMatrixDSym * _covMat
Definition: RooNDKeysPdf.h:197
std::vector< Double_t > _xVarHiP3s
Definition: RooNDKeysPdf.h:181
std::map< Int_t, Bool_t > _bpsIdcs
Definition: RooNDKeysPdf.h:182
RooChangeTracker * _tracker
Definition: RooNDKeysPdf.h:208
RooRealVar represents a variable that can be changed from the outside.
Definition: RooRealVar.h:39
TH1 is the base class of all histogram classes in ROOT.
Definition: TH1.h:58
Mother of all ROOT objects.
Definition: TObject.h:37
Basic string class.
Definition: TString.h:136
Double_t y[n]
Definition: legend1.C:17
Double_t x[n]
Definition: legend1.C:17
std::vector< Double_t > xVarHiP3s
Definition: RooNDKeysPdf.h:106
std::vector< Double_t > xVarHi
Definition: RooNDKeysPdf.h:105
std::map< Int_t, Bool_t > bpsIdcs
Definition: RooNDKeysPdf.h:107
std::vector< Int_t > bIdcs
Definition: RooNDKeysPdf.h:109
std::vector< Double_t > xVarLoM3s
Definition: RooNDKeysPdf.h:106
std::vector< Double_t > xVarHiM3s
Definition: RooNDKeysPdf.h:106
std::vector< Double_t > xVarLoP3s
Definition: RooNDKeysPdf.h:106
std::vector< Double_t > xVarLo
Definition: RooNDKeysPdf.h:105
std::vector< Int_t > sIdcs
Definition: RooNDKeysPdf.h:108
std::vector< Int_t > bmsIdcs
Definition: RooNDKeysPdf.h:110