HoughAccumulator.h

Go to the documentation of this file.

/*******************************************************************************
 *  HoughAccumulator.h
 *
 *  (C) 2007 AG Aktives Sehen <agas@uni-koblenz.de>
 *           Universitaet Koblenz-Landau
 *******************************************************************************/

#ifndef HoughAccumulator_H
#define HoughAccumulator_H

#include "../KeyPointExtraction/KeyPointMatch.h"
#include "Workers/Math/Point2D.h"

#include <vector>
#include <list>
#include <sstream>
#include <opencv2/opencv.hpp>

class HoughAccumulator
{
  public:

    HoughAccumulator();

    ~HoughAccumulator();

    bool incrAccumulatorValue(int scaleIndex, int orientationIndex, int xIndex, int yIndex, KeyPointMatch match);

    bool getAccumulatorValue(int scaleIndex, int orientationIndex, int xIndex, int yIndex, unsigned int& value);

    void resetAccumulator();

    std::vector< std::list< KeyPointMatch > > getClusteredMatches();

    std::vector< std::list< KeyPointMatch > > getMaximumMatches();

    void getImage( cv::Mat& target );

    float getVariance();

    std::string getLog(){return m_Log.str();}

  private:

    //Sort KeyPointMatch-List in descending order
    struct compareMatchList
    {
      bool operator()(const std::list< KeyPointMatch>& a, const std::list< KeyPointMatch>& b )
      {
        return a.size() > b.size();
      }
    };

    unsigned int getIndex(int scaleIndex, int orientationIndex, int xIndex, int yIndex);
    bool verifyAccumulatorIndex(int scale, int orientation, int xLocation, int yLocation);
    unsigned int getMaxAccumulatorValue();

    int m_ScaleBins;
    int m_OrientationBins;
    int m_XLocationBins;
    int m_YLocationBins;

    std::list< KeyPointMatch >* m_AccumulatorArray;

    unsigned int m_AccumulatorSize;

    std::ostringstream m_Log;
};

#endif