FitPoint.h

Go to the documentation of this file.

#ifndef ANP_FITPOINT_H
#define ANP_FITPOINT_H

// $Id: FitPoint.h,v 1.1 2007/11/30 16:03:57 rustem Exp $
//
// Point in local fit routine. Local regression is done with
// linear or quadratic polynomial:
// fDegree = 1: y = a x + b
// fDegree = 2: y = a x^2 + b x + c

// C++
#include <iostream>
#include <map>

// ROOT
#include "Rtypes.h"

namespace Anp
{   
   class FitPoint
   {
   public:
      
      FitPoint();
      FitPoint(int index, double x, double y);
      ~FitPoint();     
      
      int Index() const;
      
      double X() const;
      double Y() const;
      
      double FitY() const;
      double FitY(double x) const;
      
      bool Pass() const;
      
      short Degree() const;
      
      double A() const;
      double B() const;
      double C() const;
      
      double Weight() const;
      
      double Residual() const;
      double Derivative(double x) const;
      
      void Print(std::ostream &os = std::cout) const;

   private:
      
      friend class LocalFit;

      void SetWeight(double weight);
      bool SetWeight(unsigned int position, double weight);

      void SetL(double a, double b, bool pass);
      void SetQ(double a, double b, double c, bool pass);
      
      const std::map<unsigned int, double>& GetWeightMap() const;

   private:

      Int_t fIndex;      // global point index, set by user
      
      Double_t fX;       // x position
      Double_t fY;       // y position
      
      Double_t fA;       // a coefficient in linear or quadratic polynomial
      Double_t fB;       // b coefficient in linear or quadratic polynomial
      Double_t fC;       // c coefficient in quadratic polynomial
      
      Bool_t fPass;      // this point passed local regression
      
      Short_t fDegree;   // polynomial degree: =1 for linear and =2 for quadratic
      
      Double_t fWeight;  // global weight of this point

      std::map<unsigned int, double> fMap;  
   };
   
   
   bool operator==(const FitPoint &lhs, const FitPoint &rhs);
   bool operator <(const FitPoint &lhs, const FitPoint &rhs);
   bool operator <(double value, const FitPoint &fit);
   bool operator <(const FitPoint &rhs, double value);
   
   bool operator==(int index, const FitPoint &fit);
   bool operator==(const FitPoint &fit, int index);
   
   //
   // Inlined member functions
   //
   inline const std::map<unsigned int, double>& FitPoint::GetWeightMap() const
   {
      return fMap;
   }
   inline int FitPoint::Index() const
   {
      return fIndex;
   }
   inline double FitPoint::X() const
   {
      return fX;
   }
   inline double FitPoint::Y() const
   {
      return fY;
   }
   inline bool FitPoint::Pass() const
   {
      return fPass;
   }
   inline short FitPoint::Degree() const
   {
      return fDegree;
   }
   inline double FitPoint::A() const
   {
      return fA;
   }      
   inline double FitPoint::B() const
   {
      return fB;
   }
   inline double FitPoint::C() const
   {
      return fC;
   }
   inline double FitPoint::Weight() const
   {
      return fWeight;
   }
}

#endif

---