First CommitHEADmaster

1 files changed, 192 insertions, 0 deletions

diff --git a/sci_gateway/cpp/opencv_imfill.cpp b/sci_gateway/cpp/opencv_imfill.cpp
new file mode 100644
index 0000000..7e0c79d
--- /dev/null
+++ b/sci_gateway/cpp/opencv_imfill.cpp
@@ -0,0 +1,192 @@
+/********************************************************
+Author: Vinay Bhat
+********************************************************
+Usage: return_image = imfill(image)
+       return_image = imfill(image, 'holes')
+       return_image = imfill(image, points)
+Example:
+  im = imfill(image, [1 1; 10 10; 11 12])
+********************************************************/
+
+#include <numeric>
+#include "opencv2/core/core.hpp"
+#include "opencv2/highgui/highgui.hpp"
+#include "opencv2/opencv.hpp"
+#include <iostream>
+using namespace cv;
+using namespace std;
+extern "C"
+{
+  #include "api_scilab.h"
+  #include "Scierror.h"
+  #include "BOOL.h"
+  #include <localization.h>
+  #include "sciprint.h"
+  #include "../common.h"
+
+  void imfill_imreconstruct(Mat, Mat, Mat&);
+  
+  int opencv_imfill(char *fname, unsigned long fname_len)
+  {
+
+    SciErr sciErr;
+    int intErr = 0;
+    int *piAddr = NULL;
+    int i, j;
+    double maxVal;
+    int iRet;
+    int iRows, iCols;
+    double *pstData = NULL;
+
+    //checking input argument
+    CheckInputArgument(pvApiCtx, 1, 2);
+    CheckOutputArgument(pvApiCtx, 1, 1);
+
+    // Get the input image from the Scilab environment
+    Mat image, dst;
+    retrieveImage(image, 1);
+
+    if (nbInputArgument(pvApiCtx) == 2)
+    {
+      sciErr = getVarAddressFromPosition(pvApiCtx, 2, &piAddr);
+      if (sciErr.iErr)
+      {
+        printError(&sciErr, 0);
+        return 0;
+      }
+
+      if (image.type() != CV_8UC1)
+      { 
+        Mat temp;
+        temp = image.clone();
+        cvtColor(temp, image, CV_BGR2GRAY);
+      }
+
+      if(isStringType(pvApiCtx, piAddr))
+      {
+        char* pstData = NULL;
+
+        iRet = getAllocatedSingleString(pvApiCtx, piAddr, &pstData);
+        
+        if(iRet)
+        {
+          freeAllocatedSingleString(pstData);
+          return iRet;
+        }
+
+        if (strcmp(pstData, "holes") == 0)
+        {
+          freeAllocatedSingleString(pstData);
+          // Floodfill from point (0, 0)
+          Mat im_floodfill = image.clone();
+          floodFill(im_floodfill, Point(0,0), Scalar(255));
+           
+          // Invert floodfilled image
+          Mat im_floodfill_inv;
+          bitwise_not(im_floodfill, im_floodfill_inv);
+           
+          // Combine the two images to get the foreground.
+          Mat dst2 = (image | im_floodfill_inv);
+          erode(dst2, dst, Mat());
+
+        }
+        else
+        {
+          sciprint("Option not identified. Maybe you meant 'holes'.");
+          return 0;
+        }
+      }
+      else
+      {
+
+        // Get the points list in the form of a matrix 
+        // No. of columns = 2
+        // No. of rows = No. of points
+        sciErr = getMatrixOfDouble(pvApiCtx, piAddr, &iRows, &iCols, &pstData);
+        if(sciErr.iErr)
+        {
+            printError(&sciErr, 0);
+            return 0;
+        }
+
+        if (iCols != 2)
+        {
+          sciprint("Please enter a valid column and row index.");
+          return 0;
+        }
+
+        for (i = 0; i < iRows*iCols; i += iCols)
+        {
+            if (pstData[i] < 0 || pstData[i+1] < 0)
+            {
+                sciprint("Coordinates cannot be negative.\n");
+                return 0;
+            }
+            else
+                floodFill(image, Point(pstData[i], pstData[i+1]), Scalar(255));
+        }
+
+        dst = image.clone();
+      }
+   
+    }
+    else
+    {
+      Mat gray_image;
+      cvtColor(image, gray_image, CV_BGR2GRAY);
+
+      minMaxLoc(gray_image, NULL, &maxVal, NULL, NULL);
+
+      Mat marker = Mat(gray_image.size(), gray_image.type());
+
+      marker = Scalar::all(maxVal);
+
+      for (i = 0; (i == 0 || i == marker.cols-1); i += (marker.cols - 1))
+      {
+        for (j = 0; j < marker.rows; j++)
+        {
+          marker.at<uchar>(i,j) = gray_image.at<uchar>(i,j);
+        } 
+      }
+
+      for (i = 1; i < marker.cols-1; i++)
+      {
+        for (j = 0; (j == 0 || j == marker.rows-1); j += (marker.rows - 1))
+        {
+          marker.at<uchar>(i,j) = gray_image.at<uchar>(i,j);
+        }
+      }
+
+      imfill_imreconstruct(gray_image, marker, dst);
+    }
+    
+    
+    
+    string tempstring = type2str(dst.type());
+    char *checker;
+    checker = (char *)malloc(tempstring.size() + 1);
+    memcpy(checker, tempstring.c_str(), tempstring.size() + 1);
+    returnImage(checker, dst, 1);
+    free(checker);
+
+    //Assigning the list as the Output Variable
+    AssignOutputVariable(pvApiCtx, 1) = nbInputArgument(pvApiCtx) + 1;
+    //Returning the Output Variables as arguments to the Scilab environment
+    ReturnArguments(pvApiCtx);
+    return 0;
+
+  }
+  void imfill_imreconstruct(Mat g, Mat f, Mat& dest)
+  {
+    Mat m0, m1, m;
+    m1 = f;
+    do {
+      m0 = m1.clone();
+      erode(m0, m, Mat());
+      max(g, m, m1);
+    } while(countNonZero(m1 != m0) != 0);
+    dest = m1.clone();
+  }  
+
+/* ==================================================================== */
+}
\ No newline at end of file