First CommitHEADmaster

1 files changed, 180 insertions, 0 deletions

diff --git a/sci_gateway/cpp/opencv_roifill.cpp b/sci_gateway/cpp/opencv_roifill.cpp
new file mode 100644
index 0000000..919be08
--- /dev/null
+++ b/sci_gateway/cpp/opencv_roifill.cpp
@@ -0,0 +1,180 @@
+/********************************************************
+Author: Vinay Bhat
+********************************************************
+Usage: return_image = roifill(input_image, column_list, row_list)
+       return_image = roifill(input_image, mask)
+Example:
+  im = roifill(image, [0 100 100 0], [0 0 100 100])
+  im = roifill(image, mask)
+********************************************************/
+
+#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"
+
+  int opencv_roifill(char *fname, unsigned long fname_len)
+  {
+
+    SciErr sciErr;
+    int intErr = 0;
+    int iRowsR=0,iColsR=0,iColsC=0,iRowsC=0;
+    int *piAddr = NULL;
+    int *piAddr2  = NULL;
+    double *pstDataR = NULL;
+    double *pstDataC = NULL;
+    int i, number_of_points;
+    int lineType = 8;
+
+    //checking input argument
+    CheckInputArgument(pvApiCtx, 2, 3);
+    CheckOutputArgument(pvApiCtx, 1, 1);
+
+    // Get the input image from the Scilab environment
+    Mat image, image_orig, mask, fin_image, mask_orig;
+    retrieveImage(image_orig, 1);
+    cvtColor(image_orig, image, CV_BGR2GRAY);
+
+    if (nbInputArgument(pvApiCtx) == 2)
+    {   
+        retrieveImage(mask_orig, 2);
+
+        for (int i = 0; i < mask_orig.cols; i++)
+        {
+          for (int j = 0; j < mask_orig.rows; j++)
+          {
+            unsigned char val = mask_orig.at<uchar>(i,j);
+            if (!(val == 0 || val == 1 || val == 255))
+            {
+              sciprint("Please enter binary mask (second argument).");
+              return 0;
+            } 
+          }
+        }
+
+        if (mask_orig.type() != CV_8UC1)
+        {
+          cvtColor(mask_orig, mask, CV_BGR2GRAY);
+        }
+        else
+        {
+          mask = mask_orig.clone();
+        }
+        
+    }
+    else
+    {
+        // Get the address of 2nd argument, the column list
+        sciErr = getVarAddressFromPosition(pvApiCtx, 2, &piAddr);
+        if (sciErr.iErr)
+        {
+            printError(&sciErr, 0);
+            return 0;
+        }
+
+        // Get the address of the 3rd agument, the row list
+        sciErr = getVarAddressFromPosition(pvApiCtx, 3, &piAddr2);
+        if (sciErr.iErr)
+        {
+            printError(&sciErr, 0);
+            return 0;
+        }
+
+        // Get the column list in the form of a matrix 
+        // No. of columns = No. of elements in the list
+        // No. of rows = 1
+        sciErr = getMatrixOfDouble(pvApiCtx, piAddr, &iRowsC, &iColsC, &pstDataC);
+        if(sciErr.iErr)
+        {
+            printError(&sciErr, 0);
+            return 0;
+        }
+
+        if (iRowsC != 1)
+        {
+            sciprint("Please enter a list of column coordinates.\n");
+            return 0;
+        }
+        
+        // Get the row list in the form of a matrix
+        // No. of columns = No. of elements in the list
+        // No. of rows = 1
+        sciErr = getMatrixOfDouble(pvApiCtx, piAddr2, &iRowsR, &iColsR, &pstDataR);
+        if(sciErr.iErr)
+        {
+            printError(&sciErr, 0);
+            return 0;
+        }
+
+        if (iRowsR != 1)
+        {
+            sciprint("Please enter a list of row coordinates.\n");
+            return 0;
+        }
+
+        if (iColsC != iColsR)
+        {
+            sciprint("Please ensure number of elements in both row and column lists are equal.\n");
+            return 0;
+        }
+
+        // Number of points is number of columns
+        number_of_points = iColsC;
+
+        // Create the n points which define
+        // the polygon
+        Point points[1][number_of_points];
+        
+        for (i = 0; i < number_of_points; i++)
+        {
+            if (pstDataR[i] < 0 || pstDataC[i] < 0)
+            {
+                sciprint("Coordinates cannot be negative.\n");
+                return 0;
+            }
+            else
+                points[0][i] = Point(pstDataR[i], pstDataC[i]);
+        }
+
+        const Point* ppt[1] = { points[0] };
+
+        int npt[] = { number_of_points };
+
+        // Create a new, black, blank image same size as of input
+        mask = Mat::zeros(image.size(), image.type());
+
+        // Call the fillPoly OpenCV function
+        // Fill the blank image in the polygon specified
+        // by the points
+        fillPoly(mask, ppt, npt, 1, Scalar(255, 255, 255), lineType);
+    }
+    
+    inpaint(image, mask, fin_image, 3, INPAINT_TELEA);
+
+    string tempstring = type2str(fin_image.type());
+    char *checker;
+    checker = (char *)malloc(tempstring.size() + 1);
+    memcpy(checker, tempstring.c_str(), tempstring.size() + 1);
+    returnImage(checker, fin_image, 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;
+
+  }
+/* ==================================================================== */
+}