1 // Copyright (C) 2007-2020 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, or (at your option) any later version.
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
23 // File : ShapeRec_FeatureDetector.h
24 // Author : Renaud NEDELEC, Open CASCADE S.A.S.
27 // RNV: Workaround: HAVE_TBB definition from OCCT conflicts with OPENCV, undef it.
28 // This workaround will be removed after correction of the
29 // https://tracker.dev.opencascade.org/view.php?id=28457 issue.
34 #include <opencv2/imgproc/imgproc.hpp>
35 #include <opencv2/highgui/highgui.hpp>
40 #include <opencv2/imgproc/imgproc.hpp>
41 #include <opencv2/highgui/highgui.hpp>
48 #if defined GEOM_SHAPEREC_EXPORTS || defined GEOMShapeRec_EXPORTS
49 #define GEOM_SHAPEREC_EXPORT __declspec( dllexport )
51 #define GEOM_SHAPEREC_EXPORT __declspec( dllimport )
54 #define GEOM_SHAPEREC_EXPORT
57 class GEOM_SHAPEREC_EXPORT ShapeRec_Parameters
60 ShapeRec_Parameters();
61 virtual ~ShapeRec_Parameters();
64 int findContoursMethod;
67 class GEOM_SHAPEREC_EXPORT ShapeRec_CornersParameters : public ShapeRec_Parameters
70 ShapeRec_CornersParameters();
71 virtual ~ShapeRec_CornersParameters();
80 class GEOM_SHAPEREC_EXPORT ShapeRec_CannyParameters : public ShapeRec_Parameters
83 ShapeRec_CannyParameters();
84 virtual ~ShapeRec_CannyParameters();
91 class GEOM_SHAPEREC_EXPORT ShapeRec_ColorFilterParameters : public ShapeRec_Parameters
94 ShapeRec_ColorFilterParameters();
95 virtual ~ShapeRec_ColorFilterParameters();
104 class GEOM_SHAPEREC_EXPORT ShapeRec_FeatureDetector
108 typedef std::vector<cv::Point> CvContour;
109 typedef std::vector<std::vector<cv::Point> > CvContoursArray;
111 ShapeRec_FeatureDetector(); // Constructor
113 void SetPath( const std::string& ); // Sets the image path
114 void SetROI( const QRect& ); // Sets a Region Of Interest in the image
115 CvPoint2D32f* GetCorners() { return corners; };
116 CvContoursArray GetContours() { return contours; };
117 std::vector<cv::Vec4i> GetLines() { return lines; };
118 std::vector<cv::Vec4i> GetContoursHierarchy() { return hierarchy; };
119 int GetCornerCount() { return cornerCount; };
120 int GetImgHeight() { return imgHeight; };
121 int GetImgWidth() { return imgWidth; };
123 std::string CroppImage();
124 void ComputeCorners( bool useROI = false, ShapeRec_Parameters* parameters = 0 ); // Detects the corners from the image located at imagePath
125 bool ComputeLines(); // Detects the lines from the image located at imagePath
126 bool ComputeContours( bool useROI = false, ShapeRec_Parameters* parameters = 0 ); // Detects the contours from the image located at imagePath
130 std::string imagePath;
132 CvPoint2D32f* corners;
135 CvContoursArray contours;
136 std::vector<cv::Vec4i> hierarchy;
137 std::vector<cv::Vec4i> lines;