STARDetector.hpp

/* $Id$ */

#ifndef _FDETECT_STAR_DETECTOR_HPP_
#define _FDETECT_STAR_DETECTOR_HPP_

#include <vector>

#include <cv.h>
#include <cv.hpp>

#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/text_iarchive.hpp>

#include "fdetect/Detector.hpp"
#include "fdetect/FloatDescriptor.hpp"
#include "fdetect/RealDescriptor.hpp"

namespace jafar {
  namespace fdetect {
    class STARDetector : public fdetect::Detector {
    private:
      cv::StarDetector cv_star;
      std::vector<cv::KeyPoint> cv_keypoints;
      std::vector<float> cv_descriptors;
      friend class boost::serialization::access;
      template<class Archive>
      void serialize(Archive & ar, const unsigned int /* file_version */){
        ar & cv_star.maxSize; 
        ar & cv_star.responseThreshold;
        ar & cv_star.lineThresholdProjected; 
        ar & cv_star.lineThresholdBinarized;
        ar & cv_star.suppressNonmaxSize;
      }

    public:
      STARDetector(int _maxSize, 
                   int _responseThreshold,
                   int _lineThresholdProjected,
                   int _lineThresholdBinarized,
                   int _suppressNonmaxSize) : 
        cv_star(_maxSize, _responseThreshold,
                _lineThresholdProjected, _lineThresholdBinarized,
                _suppressNonmaxSize) {};
      
      STARDetector() : cv_star() {};

      virtual fdetect::DetectionResult detectIn(jafar::image::Image const& image);

      void setMaxSize(int _maxSize); 

      void setResponseThreshold(int _responseThreshold);

      void setLineThresholdProjected(int _lineThresholdProjected);

      void setLineThresholdBinarized(int _lineThresholdBinarized);

      void setSuppressNonmaxSize(int _suppressNonmaxSize);

      int maxSize() const;

      int responseThreshold() const;

      int lineThresholdProjected() const;

      int lineThresholdBinarized() const;

      int suppressNonmaxSize() const;

    };
  }
#ifndef SWIG
  namespace fdetect_v2 {
    template< typename InterestFeatureT >
    class STARDetectorT : public fdetect_v2::Detector< InterestFeatureT > {
    public:
      typedef InterestFeatureT InterestFeatureType;
      typedef fdetect_v2::DetectionResult<InterestFeatureT> DetectionResult;
      STARDetectorT(int _maxSize, 
                   int _responseThreshold,
                   int _lineThresholdProjected,
                   int _lineThresholdBinarized,
                   int _suppressNonmaxSize) : 
        cv_star(_maxSize, _responseThreshold,
                _lineThresholdProjected, _lineThresholdBinarized,
                _suppressNonmaxSize) {}      
      STARDetectorT() : cv_star() {};
      virtual DetectionResult detectIn(jafar::image::Image const& image) 
      {
        return detectIn(image, true);
      }
      DetectionResult detectIn(jafar::image::Image const& image, 
                               bool with_descriptors);
      void computeDescriptors(jafar::image::Image const& image, 
                              DetectionResult& dr); 

      void setMaxSize(int _maxSize); 

      void setResponseThreshold(int _responseThreshold);

      void setLineThresholdProjected(int _lineThresholdProjected);

      void setLineThresholdBinarized(int _lineThresholdBinarized);

      void setSuppressNonmaxSize(int _suppressNonmaxSize);

      int maxSize() const;

      int responseThreshold() const;

      int lineThresholdProjected() const;

      int lineThresholdBinarized() const;

      int suppressNonmaxSize() const;

    private:
      cv::StarDetector cv_star;
      friend class boost::serialization::access;
      template<class Archive>
      void serialize(Archive & ar, const unsigned int /* file_version */){
        ar & cv_star.maxSize; 
        ar & cv_star.responseThreshold;
        ar & cv_star.lineThresholdProjected; 
        ar & cv_star.lineThresholdBinarized;
        ar & cv_star.suppressNonmaxSize;
      }
    };
    typedef STARDetectorT<fdetect_v2::InterestFeature<fdetect_v2::FloatDescriptor> > STARDetector;
  }
#endif
}
#include "fdetect/STARDetector.hxx"
#endif