summaryrefslogtreecommitdiff
path: root/sci_gateway/cpp/opencv_regionfill.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'sci_gateway/cpp/opencv_regionfill.cpp')
-rw-r--r--sci_gateway/cpp/opencv_regionfill.cpp180
1 files changed, 180 insertions, 0 deletions
diff --git a/sci_gateway/cpp/opencv_regionfill.cpp b/sci_gateway/cpp/opencv_regionfill.cpp
new file mode 100644
index 0000000..85d89b3
--- /dev/null
+++ b/sci_gateway/cpp/opencv_regionfill.cpp
@@ -0,0 +1,180 @@
+/********************************************************
+Author: Vinay Bhat
+********************************************************
+Usage: return_image = regionfill(input_image, column_list, row_list)
+ return_image = regionfill(input_image, mask)
+Example:
+ im = regionfill(image, [0 100 100 0], [0 0 100 100])
+ im = regionfill(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_regionfill(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, mask, mask_orig, fin_image;
+ retrieveImage(image, 1);
+
+
+ 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_orig = Mat::zeros(image.size(), image.type());
+ cvtColor(mask_orig, mask, CV_BGR2GRAY);
+
+ // 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;
+
+ }
+/* ==================================================================== */
+}