FLANNMatcher.h

Go to the documentation of this file.

/*******************************************************************************
 *  FLANNMatcher.h
 *
 *  (C) 2007 AG Aktives Sehen <agas@uni-koblenz.de>
 *           Universitaet Koblenz-Landau
 *
 *  Additional information:
 *  $Id: $
 *******************************************************************************/

#ifndef FLANNMatcher_H
#define FLANNMatcher_H

#include <vector>
#include <deque>
#include <sstream>
#include <list>

#include <flann/flann.h>

#include "Workers/Math/Math.h"
#include "Workers/Math/Box2D.h"

#include "Workers/KeyPointExtraction/KeyPoint.h"
#include "Workers/KeyPointExtraction/KeyPointMatch.h"

class FLANNMatcher
{
  public:

    FLANNMatcher();

    FLANNMatcher( const FLANNMatcher& other );
    FLANNMatcher& operator=( const FLANNMatcher& other );


    ~FLANNMatcher();
   
    void createIndex( std::vector< KeyPoint >* keyPoints );

    void match( std::vector< KeyPoint >* keyPoints, float maxDistRatio=0.7 );

    std::list< KeyPointMatch > getMatches() { return m_Matches; }

    bool hasIndex(){return m_hasIndex;}

    int getNumMatches() { return m_Matches.size(); }

//    int getNumIndexedKeypoints(){return m_Matches.size();}

    FLANNParameters& getFlannParameters(){
       return m_flannParams;
     }

    std::string getLog();

  private:

    void eliminateMultipleMatches( );
    void clearFLANNMembers();
    void fillFlannDataWithDescriptors(const std::vector<KeyPoint>* features, float* flannDataPtr);

    //stores a list of matches
    typedef std::list< KeyPointMatch > MatchList;

    //iterator for accessing and deleting elements of the match list
    typedef std::list<KeyPointMatch>::iterator MatchElem;

    MatchList m_Matches;

    std::ostringstream m_Log;
   
    FLANN_INDEX m_flannIndex;
    FLANNParameters m_flannParams;
    bool m_hasIndex;
    unsigned int m_descriptorLength;

    float* m_FlannModelData;

};

#endif