srcnativelibs.Vision.finder.h Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2014, Sikuli.org, sikulix.com
 * Released under the MIT License.
 *
 */
#ifndef _FINDER_H_
#define _FINDER_H_

#include "pyramid-template-matcher.h"

#define DEFAULT_PYRAMID_MIM_TARGET_DIMENSION 12
#define DEFAULT_FIND_ALL_MAX_RETURN 100
#define PYRAMID_MIM_TARGET_DIMENSION_ALL 50
#define REMATCH_THRESHOLD 0.9
#define CENTER_REMATCH_THRESHOLD 0.99
#define BORDER_MARGIN 0.2


class BaseFinder{
   
public:
   
   BaseFinder(IplImage* screen_image);
   BaseFinder(Mat source);
   BaseFinder(const char* source_image_filename);
   virtual ~BaseFinder();
   
   void setROI(int x, int y, int w, int h);
   
   int get_screen_height() const { return source.rows;};
   int get_screen_width()  const {return source.cols;};
   
   
   void find();
   
 //  virtual void find(const char* str, double min_similarity) = 0;   
   
   virtual bool hasNext() = 0;
   virtual FindResult next() = 0;
   
   
protected:
   
   Rect roi;
   Mat source;
   Mat roiSource;
   
   
   double min_similarity;
};

class TextFinder : public BaseFinder {
   
public:
   TextFinder(Mat source);
   static void train(Mat& trainingImage);
   
   void find(const char* word, double min_similarity);   
   void find(vector words, double min_similarity);

   void find_all(const char* word, double min_similarity);   
   void find_all(vector words, double min_similarity);

   
   bool hasNext();
   FindResult next();
   
   static vector recognize(const Mat& inputImage);
   
private:
   vector matches;
   vector::iterator matches_iterator;
   
   void test_find(const Mat& inputImage, const vector& testWords);
};

class TemplateFinder : public BaseFinder{
   float PyramidMinTargetDimension;
   
public:
   
   TemplateFinder(Mat source);
   TemplateFinder(IplImage* source);
   TemplateFinder(const char* source_image_filename);
   ~TemplateFinder();
   
   
   void find(Mat target, double min_similarity);
   void find(IplImage* target, double min_similarity);
   void find(const char *target_image_filename, double min_similarity);
   
   
   void find_all(Mat target, double min_similarity);
   void find_all(IplImage*  target, double min_similarity);
   void find_all(const char *target_image_filename, double min_similarity);  
   
   bool hasNext();
   FindResult next();
   
private:
   
   void init();
   void create_matcher(const MatchingData& data, int level, float ratio);
   PyramidTemplateMatcher* matcher;
   
   FindResult current_match;
   int current_rank;   
   
   // buffer matches and return top score
   void add_matches_to_buffer(int num_matches_to_add);
   float top_score_in_buffer();   
   
   vector buffered_matches;   
};

class Finder {
public:
   
   
   Finder(Mat mat);   
   Finder(IplImage* source);
   Finder(const char* source);
   ~Finder();

   void setROI(int x, int y, int w, int h);
   
   void find(IplImage* target, double min_similarity);
   void find(const char *target, double min_similarity);
   
   void find_all(IplImage*  target, double min_similarity);
   void find_all(const char *target, double min_similarity);  
   
   bool hasNext();
   FindResult next();
   
private:
   
   Mat _source;
   BaseFinder* _finder;
   Rect _roi;
};

//class FaceFinder : public BaseFinder {
//   
//public:
//   
//   FaceFinder(const char* screen_image_filename);
//   ~FaceFinder();
//   
//   void find();
//   bool hasNext();
//   FindResult next();
//   
//private:
//   
//   CvMemStorage* storage;
//   
//   static CvHaarClassifierCascade* cascade;
//   
//   CvSeq* faces;
//   int face_i;
//   
//};


class ChangeFinder : public BaseFinder {
   
public:
   
   ChangeFinder(const IplImage* screen_image);
   ChangeFinder(const Mat screen_image);
   ChangeFinder(const char* screen_image_filename);
   
   ~ChangeFinder();
   
   void find(IplImage* new_img);
   void find(Mat new_img);
   void find(const char* new_screen_image_filename);
   
   bool hasNext();
   FindResult next(); 
   
private:
   
   bool is_identical;
   
   IplImage *prev_img;
   CvSeq* c;
   CvMemStorage* storage;
   
};




#endif // _FINDER_H_