LocalFit.h

Go to the documentation of this file.

#ifndef ANP_LOCALFIT_H
#define ANP_LOCALFIT_H

// $Id: LocalFit.h,v 1.1 2007/11/30 16:03:57 rustem Exp $
//
// Implementation of robust locally weighted regression algorithm:
// Cleveland, W.S., 1979, Journal of the American Statistical Association, 74, 829
//

// C++
#include <list>
#include <map>
#include <string>
#include <vector>

// ROOT
#include "Rtypes.h"

// Local
#include "PhysicsNtuple/Handle.h"

// Local
#include "FitPoint.h"

class Registry;

namespace Anp
{
   class LocalFit: virtual public Base
   {
   public:
      
      LocalFit();
      virtual ~LocalFit();
      
      bool Add(int index, double x_, double y_);
      
      bool RunFit();
      
      void Clear();
      
      void Config(const Registry &reg);
      
      const std::vector<Anp::FitPoint>& GetFit() const;
      
      double FitY(double x) const;

      void SetWindow(int value);
      void SetIter(int value);
      void SetDegree(int value);

      unsigned int Window() const;
      unsigned int Iter() const;
      unsigned int Degree() const;
      
      void Print(std::ostream &os = std::cout) const;

   private:
      
      typedef std::list<std::pair<unsigned int,double> > List;
      
   private:
      
      bool RunLFit(FitPoint &point, const std::vector<FitPoint> &fvec) const;
      bool RunQFit(FitPoint &point, const std::vector<FitPoint> &fvec) const;
      
      bool SortCleanCheck(std::vector<Anp::FitPoint> &fvec) const;
      
      const List GetList(unsigned int pos) const;
      
      double BFunction(double value) const;
      double WFunction(double value) const;
      
   private:

      Bool_t fDebug;
      Bool_t fValid;
      
      UInt_t fDegree;
      UInt_t fIter;
      UInt_t fWindow;
      
      std::vector<Anp::FitPoint> fFitVec;
   };
   
   //
   // inlined member functions
   //
   inline const std::vector<Anp::FitPoint>& LocalFit::GetFit() const
   {
      return fFitVec;
   }
   inline unsigned int LocalFit::Window() const
   {
      return fWindow;
   }
   inline unsigned int LocalFit::Iter() const
   {
      return fIter;
   }
   inline unsigned int LocalFit::Degree() const
   {
      return fDegree;
   }   
}

#endif

---