src/ShapeRecognition/ShapeRec_FeatureDetector.hxx

   1 // Copyright (C) 2007-2011  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.
  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 // File   : ShapeRec_FeatureDetector.h
  23 // Author : Renaud NEDELEC, Open CASCADE S.A.S.
  24
  25 // OpenCV includes
  26 #include <cv.h>
  27 #include <highgui.h>
  28 #include "opencv2/imgproc/imgproc.hpp"
  29 #include "opencv2/highgui/highgui.hpp"
  30
  31 // Qt
  32 #include <QRect>
  33
  34 enum              // Method used for contour detection
  35 {
  36   CANNY,
  37   COLORFILTER,
  38   RIDGE_DETECTOR
  39 };
  40
  41 class ShapeRec_FeatureDetector
  42 {
  43 public:
  44
  45   typedef std::vector<cv::Point>               CvContour;
  46   typedef std::vector<std::vector<cv::Point> > CvContoursArray;
  47
  48   ShapeRec_FeatureDetector();                                            // Constructor
  49
  50   void                    SetPath( const std::string& );                 // Sets the image path
  51   void                    SetROI( const QRect& );                        // Sets a Region Of Interest in the image
  52   CvPoint2D32f*           GetCorners()           { return corners;     };
  53   CvContoursArray         GetContours()          { return contours;    };
  54   std::vector<cv::Vec4i>  GetLines()             { return lines;       };
  55   std::vector<cv::Vec4i>  GetContoursHierarchy() { return hierarchy;   };
  56   int                     GetCornerCount()       { return cornerCount; };
  57   int                     GetImgHeight()         { return imgHeight;   };
  58   int                     GetImgWidth()          { return imgWidth;    };
  59
  60   std::string             CroppImage();
  61   void                    ComputeCorners();                              // Detects the corners from the image located at imagePath
  62   bool                    ComputeLines();                                // Detects the lines from the image located at imagePath
  63   bool                    ComputeContours( int method );                 // Detects the contours from the image located at imagePath
  64
  65
  66 private:
  67   std::string             imagePath;
  68
  69   CvPoint2D32f*           corners;
  70   int                     cornerCount;
  71
  72   CvContoursArray         contours;
  73   std::vector<cv::Vec4i>  hierarchy;
  74   std::vector<cv::Vec4i>  lines;
  75   int                     imgHeight;
  76   int                     imgWidth;
  77   CvRect                  rect;
  78
  79   void                    _detectAndRetrieveContours( cv::Mat binaryImg );
  80   cv::Mat                 _colorFiltering();
  81 };