fix warning (thanks Federico)
[u/mrichter/AliRoot.git] / STAT / TKDPDF.h
1 #ifndef ROOT_TKDPDF
2 #define ROOT_TKDPDF
3
4 #ifndef ROOT_TKDInterpolatorBase
5 #include "TKDInterpolatorBase.h"
6 #endif
7
8 #ifndef ROOT_TKDTree
9 #include "TKDTree.h"
10 #endif
11
12 // Non parametric interpolation class based on local polinomial regression.
13 // The class can be used to approximate PDF together with TKDTree or for
14 // general regression when the data points are given.
15
16 template <typename Index, typename Value> class TKDTree;
17 typedef class TKDTree<Int_t, Float_t> TKDTreeIF;
18 class TTree;
19 class TLinearFitter;
20 class TKDPDF : public TKDTreeIF, public TKDInterpolatorBase
21 {
22 public:
23         TKDPDF();
24         TKDPDF(TTree *t, const Char_t *var, const Char_t *cut = 0, UInt_t bsize = 100, Long64_t nentries = 1000000000, Long64_t firstentry = 0);
25         TKDPDF(Int_t npoints, Int_t ndim, UInt_t bsize, Float_t **data);
26         ~TKDPDF();
27
28         inline Bool_t     GetDataPoint(Int_t n, Float_t *p) const;
29         inline Int_t      GetNodeIndex(const Float_t *p);
30                                         void       DrawNode(Int_t tnode, UInt_t ax1=0, UInt_t ax2=1);
31
32 private:
33         TKDPDF(const TKDPDF &);
34         TKDPDF& operator=(const TKDPDF &);
35                 void       Build(Int_t ndim = 0);
36
37                                         
38         ClassDef(TKDPDF, 1)   // data interpolator based on KD tree
39 };
40
41
42 //__________________________________________________________________
43 Bool_t  TKDPDF::GetDataPoint(Int_t n, Float_t *p) const
44 {
45         if(n < 0 || n >= fNPoints) return kFALSE;
46         if(!fData) return kFALSE;
47                 
48         for(int i=0; i<fNDim; i++) p[i] = fData[i][n];
49         return kTRUE;
50 }
51
52 //__________________________________________________________________
53 Int_t   TKDPDF::GetNodeIndex(const Float_t *p)
54 {
55         return FindNode(p) - fNNodes;
56 }
57
58 #endif
59