src/ShapeRecognition/ShapeRec_FeatureDetector.hxx

   1 // Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 // This library is free software; you can redistribute it and/or
   7 // modify it under the terms of the GNU Lesser General Public
   8 // License as published by the Free Software Foundation; either
   9 // version 2.1 of the License, or (at your option) any later version.
  10 //
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 // Lesser General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU Lesser General Public
  17 // License along with this library; if not, write to the Free Software
  18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22
  23 // File   : ShapeRec_FeatureDetector.h
  24 // Author : Renaud NEDELEC, Open CASCADE S.A.S.
  25
  26 // OpenCV includes
  27 #include <cv.h>
  28 #include <highgui.h>
  29 #include <opencv2/imgproc/imgproc.hpp>
  30 #include <opencv2/highgui/highgui.hpp>
  31
  32 // Qt
  33 #include <QRect>
  34
  35 class ShapeRec_Parameters
  36 {
  37 public:
  38   ShapeRec_Parameters();
  39   virtual ~ShapeRec_Parameters();
  40
  41   int kernelSize;
  42   int findContoursMethod;
  43 };
  44
  45 class ShapeRec_CornersParameters : public ShapeRec_Parameters
  46 {
  47 public:
  48   ShapeRec_CornersParameters();
  49   virtual ~ShapeRec_CornersParameters();
  50
  51   double qualityLevel;
  52   double minDistance;
  53   int typeCriteria;
  54   int maxIter;
  55   double epsilon;
  56 };
  57
  58 class ShapeRec_CannyParameters : public ShapeRec_Parameters
  59 {
  60 public:
  61   ShapeRec_CannyParameters();
  62   virtual ~ShapeRec_CannyParameters();
  63
  64   int lowThreshold;
  65   int ratio;
  66   bool L2gradient;
  67 };
  68
  69 class ShapeRec_ColorFilterParameters : public ShapeRec_Parameters
  70 {
  71 public:
  72   ShapeRec_ColorFilterParameters();
  73   virtual ~ShapeRec_ColorFilterParameters();
  74
  75   int smoothSize;
  76   int* histSize;
  77   int histType;
  78   double threshold;
  79   double maxThreshold;
  80 };
  81
  82 class ShapeRec_FeatureDetector
  83 {
  84 public:
  85
  86   typedef std::vector<cv::Point>               CvContour;
  87   typedef std::vector<std::vector<cv::Point> > CvContoursArray;
  88
  89   ShapeRec_FeatureDetector();                                            // Constructor
  90
  91   void                    SetPath( const std::string& );                 // Sets the image path
  92   void                    SetROI( const QRect& );                        // Sets a Region Of Interest in the image
  93   CvPoint2D32f*           GetCorners()           { return corners;     };
  94   CvContoursArray         GetContours()          { return contours;    };
  95   std::vector<cv::Vec4i>  GetLines()             { return lines;       };
  96   std::vector<cv::Vec4i>  GetContoursHierarchy() { return hierarchy;   };
  97   int                     GetCornerCount()       { return cornerCount; };
  98   int                     GetImgHeight()         { return imgHeight;   };
  99   int                     GetImgWidth()          { return imgWidth;    };
 100
 101   std::string             CroppImage();
 102   void                    ComputeCorners( bool useROI = false, ShapeRec_Parameters* parameters = 0 );  // Detects the corners from the image located at imagePath
 103   bool                    ComputeLines();                                                              // Detects the lines from the image located at imagePath
 104   bool                    ComputeContours( bool useROI = false, ShapeRec_Parameters* parameters = 0 ); // Detects the contours from the image located at imagePath
 105
 106
 107 private:
 108   std::string             imagePath;
 109
 110   CvPoint2D32f*           corners;
 111   int                     cornerCount;
 112
 113   CvContoursArray         contours;
 114   std::vector<cv::Vec4i>  hierarchy;
 115   std::vector<cv::Vec4i>  lines;
 116   int                     imgHeight;
 117   int                     imgWidth;
 118   CvRect                  rect;
 119 };