STAT/TKDInterpolatorBase.h

   1 #ifndef ROOT_TKDInterpolatorBase
   2 #define ROOT_TKDInterpolatorBase
   3
   4 #ifndef ROOT_Rtypes
   5 #include "Rtypes.h"
   6 #endif
   7
   8 ///////////////////////////////////////////////////////////////
   9 //
  10 // Base non parametric interpolation algorithm.
  11 // The class implements local polynomial regression (LOWESS).
  12 // The user will work with daughter classes which implements
  13 // particular data configurations.
  14 //
  15 ///////////////////////////////////////////////////////////////
  16
  17 template <typename Value> class TVectorT;
  18 typedef struct TVectorT<Double_t> TVectorD;
  19 template <typename Value> class TMatrixT;
  20 typedef class TMatrixT<Double_t> TMatrixD;
  21 template <typename Index, typename Value> class TKDTree;
  22 typedef class TKDTree<Int_t, Float_t> TKDTreeIF;
  23 class TLinearFitter;
  24 class TClonesArray;
  25 class TKDNodeInfo;
  26 class TKDInterpolatorBase
  27 {
  28 public:
  29   TKDInterpolatorBase(Int_t size = 0);
  30   virtual ~TKDInterpolatorBase();
  31
  32   Double_t   Eval(const Double_t *point, Double_t &result, Double_t &error, Bool_t force = kFALSE);
  33   virtual Int_t GetNodeIndex(const Float_t *p) = 0;
  34   Float_t    GetAlpha() const {return fAlpha;}
  35   Bool_t     GetCOGPoint(Int_t node, Float_t *&coord, Float_t &val, Float_t &error) const;
  36   Bool_t     GetInterpolationMethod() const {return fStatus&1;}
  37   TKDNodeInfo* GetNodeInfo(Int_t inode) const;
  38   Int_t      GetNTNodes() const {return fNTNodes;}
  39   void       GetStatus();
  40   Bool_t     GetStore() const {return fStatus&2;}
  41   Bool_t     GetWeights() const {return fStatus&4;}
  42
  43   void       DrawBins(UInt_t ax1 = 0, UInt_t ax2 = 1, Float_t ax1min=-1., Float_t ax1max=1., Float_t ax2min=-1., Float_t ax2max=1.);
  44   void       SetAlpha(Float_t a){if(a>0.) fAlpha = a;}
  45   void       SetInterpolationMethod(Bool_t on = kTRUE);
  46   void       SetStore(Bool_t on = kTRUE);
  47   void       SetWeights(Bool_t on = kTRUE);
  48
  49 protected:
  50   virtual void      Build(Int_t nnodes);
  51
  52 private:
  53   TKDInterpolatorBase(const TKDInterpolatorBase &);
  54   TKDInterpolatorBase& operator=(const TKDInterpolatorBase &);
  55
  56 protected:
  57   Int_t         fNSize;       // data dimension
  58   Int_t         fNTNodes;     //Number of evaluation data points
  59   TClonesArray  *fTNodes;     //interpolation nodes
  60
  61 //private:
  62   UChar_t       fStatus;      // status of the interpolator
  63   UChar_t       fLambda;      // number of parameters in polynom
  64   Short_t                   fDepth;       //! depth of the KD Tree structure used
  65   Float_t       fAlpha;       // alpha parameter
  66   Float_t       **fRefPoints; //! temporary storage of COG data
  67   Double_t          *fBuffer;     //! working space [2*fLambda]
  68   TKDTree<Int_t, Float_t> *fKDhelper;      //! kNN finder
  69   TLinearFitter *fFitter;     //! linear fitter
  70
  71   ClassDef(TKDInterpolatorBase, 1)   // data interpolator based on KD tree
  72 };
  73
  74
  75 #endif
  76