diff options
Diffstat (limited to 'sci_gateway/cpp/opencv_imfill.cpp')
| -rw-r--r-- | sci_gateway/cpp/opencv_imfill.cpp | 192 |
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 |