ParallelSurfExtractor.h

Go to the documentation of this file.

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

#ifndef SurfExtractor_H
#define SurfExtractor_H

#include <vector>
#include <string>


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

#include "../../Workers/Puma2/ImageMask.h"
#include "../../Workers/Puma2/GrayLevelImage8.h"

#include "KeyPoint.h"
#include "SurfExtractorBase.h"

#include "../../Workers/ParallelSurf/KeyPointDetector.h"
#include "threadpool.hpp"



class ParallelSurfExtractor: public SurfExtractorBase
{
  public:

   ParallelSurfExtractor( int numThreads=0 );

   ParallelSurfExtractor( const ParallelSurfExtractor& other );

    virtual ~ParallelSurfExtractor();

    ParallelSurfExtractor& operator=( const ParallelSurfExtractor& other );

    virtual void setImage( const puma2::GrayLevelImage8 &pumaImage );

    virtual void setImage( const puma2::ColorImageRGB8 &pumaImage );

    virtual void getKeyPoints ( std::vector< KeyPoint >& keyPoints );
    
    virtual std::string getName();

  private:

    parallelsurf::Image* m_IntegralImage;
    
    boost::threadpool::pool *m_ThreadPool;

    // define a Keypoint insertor
    class KeyPointVectInsertor : public parallelsurf::KeyPointInsertor
    {
      public:

        KeyPointVectInsertor ( std::vector<parallelsurf::KeyPoint>& keyPoints ) : m_KeyPoints ( keyPoints ) {};

        inline virtual void operator() ( const parallelsurf::KeyPoint &panoKeyPoint )
        {
//           std::cout << ".";
          m_KeyPoints.push_back ( panoKeyPoint );
        }

      private:

        std::vector<parallelsurf::KeyPoint>& m_KeyPoints;
    };
};


#endif