src/ShapeRecognition/ShapeRec_FeatureDetector.hxx

   1 // Copyright (C) 2007-2023  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 // 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.
  30 #ifdef HAVE_TBB
  31 #undef HAVE_TBB
  32 #include <opencv2/opencv.hpp>
  33 #define HAVE_TBB
  34 #else
  35 #include <opencv2/opencv.hpp>
  36 #endif
  37
  38 // Qt
  39 #include <QRect>
  40
  41 #ifdef WIN32
  42   #if defined GEOM_SHAPEREC_EXPORTS || defined GEOMShapeRec_EXPORTS
  43     #define GEOM_SHAPEREC_EXPORT __declspec( dllexport )
  44   #else
  45     #define GEOM_SHAPEREC_EXPORT __declspec( dllimport )
  46   #endif
  47 #else
  48    #define GEOM_SHAPEREC_EXPORT
  49 #endif
  50
  51 class GEOM_SHAPEREC_EXPORT ShapeRec_Parameters
  52 {
  53 public:
  54   ShapeRec_Parameters();
  55   virtual ~ShapeRec_Parameters();
  56
  57   int kernelSize;
  58   int findContoursMethod;
  59 };
  60
  61 class GEOM_SHAPEREC_EXPORT ShapeRec_CornersParameters : public ShapeRec_Parameters
  62 {
  63 public:
  64   ShapeRec_CornersParameters();
  65   virtual ~ShapeRec_CornersParameters();
  66
  67   double qualityLevel;
  68   double minDistance;
  69   int typeCriteria;
  70   int maxIter;
  71   double epsilon;
  72 };
  73
  74 class GEOM_SHAPEREC_EXPORT ShapeRec_CannyParameters : public ShapeRec_Parameters
  75 {
  76 public:
  77   ShapeRec_CannyParameters();
  78   virtual ~ShapeRec_CannyParameters();
  79
  80   int lowThreshold;
  81   int ratio;
  82   bool L2gradient;
  83 };
  84
  85 class GEOM_SHAPEREC_EXPORT ShapeRec_ColorFilterParameters : public ShapeRec_Parameters
  86 {
  87 public:
  88   ShapeRec_ColorFilterParameters();
  89   virtual ~ShapeRec_ColorFilterParameters();
  90
  91   int smoothSize;
  92   int* histSize;
  93   int histType;
  94   double threshold;
  95   double maxThreshold;
  96 };
  97
  98 class GEOM_SHAPEREC_EXPORT ShapeRec_FeatureDetector
  99 {
 100 public:
 101
 102   typedef std::vector<cv::Point>               CvContour;
 103   typedef std::vector<std::vector<cv::Point> > CvContoursArray;
 104
 105   ShapeRec_FeatureDetector();                                            // Constructor
 106
 107   void                    SetPath( const std::string& );                 // Sets the image path
 108   void                    SetROI( const QRect& );                        // Sets a Region Of Interest in the image
 109   CvPoint2D32f*           GetCorners()           { return corners;     };
 110   CvContoursArray         GetContours()          { return contours;    };
 111   std::vector<cv::Vec4i>  GetLines()             { return lines;       };
 112   std::vector<cv::Vec4i>  GetContoursHierarchy() { return hierarchy;   };
 113   int                     GetCornerCount()       { return cornerCount; };
 114   int                     GetImgHeight()         { return imgHeight;   };
 115   int                     GetImgWidth()          { return imgWidth;    };
 116
 117   std::string             CroppImage();
 118   void                    ComputeCorners( bool useROI = false, ShapeRec_Parameters* parameters = 0 );  // Detects the corners from the image located at imagePath
 119   bool                    ComputeLines();                                                              // Detects the lines from the image located at imagePath
 120   bool                    ComputeContours( bool useROI = false, ShapeRec_Parameters* parameters = 0 ); // Detects the contours from the image located at imagePath
 121
 122
 123 private:
 124   std::string             imagePath;
 125
 126   CvPoint2D32f*           corners;
 127   int                     cornerCount;
 128
 129   CvContoursArray         contours;
 130   std::vector<cv::Vec4i>  hierarchy;
 131   std::vector<cv::Vec4i>  lines;
 132   int                     imgHeight;
 133   int                     imgWidth;
 134   CvRect                  rect;
 135 };