grabber.hpp

/* $Id$ */

#ifndef _GRABBER_HPP
#define _GRABBER_HPP

#include "kernel/jafarException.hpp"
#include "jmath/jblas.hpp"
#include "image/Image.hpp"
#include <map>
#include <opencv/cxcore.h>
#if CV_MAJOR_VERSION*10 + CV_MINOR_VERSION > 21
#include <opencv2/imgproc/imgproc.hpp>
#else
#include <cv.hpp>
#include <cvtypes.h>
#endif

namespace jafar {
  
  namespace cutter {    
    
    class Grabber {
    public:
      typedef enum surface_type {DARK, LIGHT};
    private:
      jafar::image::Image m_image;
      jafar::image::Image m_result;
      std::multimap<unsigned int, int> boxes;
      cv::Rect largest;
      cv::Rect plausible;
      void findContoursOfInterest();
      void grab();
      surface_type m_surface;
    public:
      Grabber(const image::Image& image, surface_type surface=LIGHT);
      Grabber(const std::string& imageFile, surface_type surface=LIGHT);
      void run();
      const image::Image& source() const;
      const image::Image& result() const;
      void writeImages(const std::string& srcFileName="source.png",
           const std::string& objFileName="result.png");
      const cv::Rect& backgroundBorders() const;
      const cv::Rect& objectBorders() const;
    };

  }//namespace cutter

}//namespace jafar

#endif