1 // Copyright (C) 2007-2011 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 // File : ShapeRec_FeatureDetector.h
23 // Author : Renaud NEDELEC, Open CASCADE S.A.S.
28 #include "opencv2/imgproc/imgproc.hpp"
29 #include "opencv2/highgui/highgui.hpp"
35 enum // Method used for contour detection
42 class ShapeRec_FeatureDetector
46 typedef std::vector<cv::Point> CvContour;
47 typedef std::vector<std::vector<cv::Point> > CvContoursArray;
49 ShapeRec_FeatureDetector( const QString& ); // Constructor
51 void ComputeCorners(); // Detects the corners from the image located at imagePath
52 bool ComputeLines(); // Detects the lines from the image located at imagePath
53 bool ComputeContours( int method ); // Detects the contours from the image located at imagePath
55 void SetROI( const QRect& ); // Sets a Region Of Interest in the image
56 CvPoint2D32f* GetCorners() { return corners; };
57 CvContoursArray GetContours() { return contours; };
58 std::vector<cv::Vec4i> GetLines() { return lines; };
59 std::vector<cv::Vec4i> GetContoursHierarchy() { return hierarchy; };
60 int GetCornerCount() { return cornerCount; };
61 int GetImgHeight() { return imgHeight; };
62 int GetImgWidth() { return imgWidth; };
66 std::string imagePath;
68 CvPoint2D32f* corners;
71 CvContoursArray contours;
72 std::vector<cv::Vec4i> hierarchy;
73 std::vector<cv::Vec4i> lines;
78 void _detectAndRetrieveContours( cv::Mat binaryImg );
79 cv::Mat _colorFiltering();