Wrapped OpenALPR library in "alpr" namespace. Resolves issue #60.

src/openalpr/textdetection/textcontours.cpp

@@ -22,117 +22,119 @@
 using namespace std;
 using namespace cv;
 
-
-TextContours::TextContours() {
-  
-
-}
-
-
-TextContours::TextContours(cv::Mat threshold) {
-  
-  load(threshold);
-}
-
-
-TextContours::~TextContours() {
-}
-
-void TextContours::load(cv::Mat threshold) {
-
-  Mat tempThreshold(threshold.size(), CV_8U);
-  threshold.copyTo(tempThreshold);
-  findContours(tempThreshold,
-               contours, // a vector of contours
-               hierarchy,
-               CV_RETR_TREE, // retrieve all contours
-               CV_CHAIN_APPROX_SIMPLE ); // all pixels of each contours
-
-  for (uint i = 0; i < contours.size(); i++)
-    goodIndices.push_back(true);
-  
-  this->width = threshold.cols;
-  this->height = threshold.rows;
-}
-
-
-uint TextContours::size() {
-  return contours.size();
-}
-
-
-
-int TextContours::getGoodIndicesCount()
+namespace alpr
 {
-  int count = 0;
-  for (uint i = 0; i < goodIndices.size(); i++)
-  {
-    if (goodIndices[i])
-      count++;
+
+  TextContours::TextContours() {
+
+
   }
 
-  return count;
-}
 
+  TextContours::TextContours(cv::Mat threshold) {
 
-std::vector<bool> TextContours::getIndicesCopy()
-{
-  vector<bool> copyArray;
-  for (uint i = 0; i < goodIndices.size(); i++)
-  {
-    bool val = goodIndices[i];
-    copyArray.push_back(goodIndices[i]);
+    load(threshold);
   }
-  
-  return copyArray;
-}
 
-void TextContours::setIndices(std::vector<bool> newIndices)
-{
-  if (newIndices.size() == goodIndices.size())
-  {
-    for (uint i = 0; i < newIndices.size(); i++)
-      goodIndices[i] = newIndices[i];
+
+  TextContours::~TextContours() {
   }
-  else
-  {
-    assert("Invalid set operation on indices");
+
+  void TextContours::load(cv::Mat threshold) {
+
+    Mat tempThreshold(threshold.size(), CV_8U);
+    threshold.copyTo(tempThreshold);
+    findContours(tempThreshold,
+                 contours, // a vector of contours
+                 hierarchy,
+                 CV_RETR_TREE, // retrieve all contours
+                 CV_CHAIN_APPROX_SIMPLE ); // all pixels of each contours
+
+    for (uint i = 0; i < contours.size(); i++)
+      goodIndices.push_back(true);
+
+    this->width = threshold.cols;
+    this->height = threshold.rows;
   }
-}
 
-Mat TextContours::drawDebugImage() {
 
-    Mat img_contours = Mat::zeros(Size(width, height), CV_8U);
-    
-    return drawDebugImage(img_contours);
-}
+  uint TextContours::size() {
+    return contours.size();
+  }
 
-Mat TextContours::drawDebugImage(Mat baseImage) {
-    Mat img_contours(baseImage.size(), CV_8U);
-    baseImage.copyTo(img_contours);
-    
-    cvtColor(img_contours, img_contours, CV_GRAY2RGB);
 
-    vector<vector<Point> > allowedContours;
-    for (uint i = 0; i < this->contours.size(); i++)
+
+  int TextContours::getGoodIndicesCount()
+  {
+    int count = 0;
+    for (uint i = 0; i < goodIndices.size(); i++)
     {
-      if (this->goodIndices[i])
-        allowedContours.push_back(this->contours[i]);
+      if (goodIndices[i])
+        count++;
     }
 
-    drawContours(img_contours, this->contours,
-                 -1, // draw all contours
-                 cv::Scalar(255,0,0), // in blue
-                 1); // with a thickness of 1
+    return count;
+  }
 
-    drawContours(img_contours, allowedContours,
-                 -1, // draw all contours
-                 cv::Scalar(0,255,0), // in green
-                 1); // with a thickness of 1
 
-    
-    return img_contours;
+  std::vector<bool> TextContours::getIndicesCopy()
+  {
+    vector<bool> copyArray;
+    for (uint i = 0; i < goodIndices.size(); i++)
+    {
+      bool val = goodIndices[i];
+      copyArray.push_back(goodIndices[i]);
+    }
+
+    return copyArray;
+  }
+
+  void TextContours::setIndices(std::vector<bool> newIndices)
+  {
+    if (newIndices.size() == goodIndices.size())
+    {
+      for (uint i = 0; i < newIndices.size(); i++)
+        goodIndices[i] = newIndices[i];
+    }
+    else
+    {
+      assert("Invalid set operation on indices");
+    }
+  }
+
+  Mat TextContours::drawDebugImage() {
+
+      Mat img_contours = Mat::zeros(Size(width, height), CV_8U);
+
+      return drawDebugImage(img_contours);
+  }
+
+  Mat TextContours::drawDebugImage(Mat baseImage) {
+      Mat img_contours(baseImage.size(), CV_8U);
+      baseImage.copyTo(img_contours);
+
+      cvtColor(img_contours, img_contours, CV_GRAY2RGB);
+
+      vector<vector<Point> > allowedContours;
+      for (uint i = 0; i < this->contours.size(); i++)
+      {
+        if (this->goodIndices[i])
+          allowedContours.push_back(this->contours[i]);
+      }
+
+      drawContours(img_contours, this->contours,
+                   -1, // draw all contours
+                   cv::Scalar(255,0,0), // in blue
+                   1); // with a thickness of 1
+
+      drawContours(img_contours, allowedContours,
+                   -1, // draw all contours
+                   cv::Scalar(0,255,0), // in green
+                   1); // with a thickness of 1
+
+
+      return img_contours;
+  }
+
 }
 
-
-