pol_locator_hough.h

Go to the documentation of this file.

// $Header: /cvs/mimas2/include/pol_locator_hough.h,v 1.1.1.1 2005/08/09 15:37:45 engmb Exp $
#ifndef POL_LOCATOR_HOUGH_H
#define POL_LOCATOR_HOUGH_H

#include "mimasconfig.h"
#include <boost/numeric/ublas/vector.hpp>
#include <cmath>
#include "image.h"
#include "angle.h"
#include "pol_locator.h"

#define MAXVAL  0x7fff

namespace mimas {
class pol_model;
class pol_match;

class pol_angle
{
    public:
    int16_t x,y;  

    pol_angle(int a)
    {
  x=y=(int16_t) a;
    }

    pol_angle(double v)
    {
  x=y=(int16_t) v;
    }

    pol_angle()
    {
  x=y=(int16_t) 0;
    }

    operator  int()
    {
  return (int)(x+y);
    }

    operator  double()
    {
  return (double)(x+y);
    }


    int mag()
    {
  double  dx=(double)x,dy=(double)y;
  return (int)sqrt(dx*dx+dy*dy);
    }

    int magSquared()
    {
  int ix=(int)x;
  int iy=(int)y;
  return (ix*ix+iy*iy);
    }
};


class pol_locator_hough : public pol_locator, public image<pol_angle>
{
    protected:
  image<double> *bell_curve;
  double      thres;      
  double      ox,oy;

    public:
    typedef boost::numeric::ublas::vector< double > Vector;
  pol_locator_hough():
   bell_curve(0), thres(0), ox(0), oy(0)
  {}
  pol_locator_hough(double t):
   bell_curve(0), thres(t), ox(0), oy(0)
  {}

  ~pol_locator_hough()
  {
      if (bell_curve) delete bell_curve;
  }

protected:
  int calcGaussBoxSize( double sigma, double accuracy );
  void  setLocation( int x, int y, double ca, double sa );

  void  initFastGauss( double sigma, double precision );

  void  addGaussPointFast( const Vector &posn, double w, double angle );

public:
  /*
   * display the hough transform, i.e. for debugging
   *
   */
  // virtual void display( void );

  virtual void      init( int numdims, ...);
  virtual void      addEntry( double weight, ... );
  virtual pol_match_ptr_list getMatches();
  void        setOffset( double dx, double dy )
          { ox=dx; oy=dy; }

};


}

#endif

---