match.hpp

/* $Id$ */
#ifndef IMDESC_MATCH_HPP
#define IMDESC_MATCH_HPP

#include <vector>
#include "image/JfrImage.hpp"
#include "imdesc/feature.hpp"

namespace jafar {
  namespace imdesc {
    
    struct oneMatch {
      KBFeature one;
      KBFeature two;
      double score;
      
      bool operator<(const oneMatch& m_) const {
  return (score < m_.score);
      }
      
      bool operator>(const oneMatch& m_) const {
  return (score > m_.score);
      }
    }; // struct oneMatch
    
    class Match {
    public:
      Match(double thd_ = 0.8) : thd(thd_) {};
      
      inline void setThreshold(double thd_) {
  thd = thd_;
      };
      
      inline int getNbMatches() {
  return matches.size();
      };
      
      inline oneMatch& get(int i) {
  return *(matches.begin()+i);
      };
      
      void compute(const std::vector<KBFeature>& f1_,
       const jafar::image::JfrImage& src1_,
       const std::vector<KBFeature>& f2_,
       const jafar::image::JfrImage& src2_);
      
      jafar::image::JfrImage* drawMatches(const jafar::image::JfrImage& src1_,
            const jafar::image::JfrImage& src2_);
      
    private:
      double thd;
      std::vector<oneMatch> matches;
    };
    
  } // namespace imdesc
} // namespace jafar

#endif // IMDESC_MATCH_HPP