diff options
Diffstat (limited to 'sci_gateway1/cpp/opencv_BlockMatcher.cpp')
| -rw-r--r-- | sci_gateway1/cpp/opencv_BlockMatcher.cpp | 606 |
1 files changed, 606 insertions, 0 deletions
diff --git a/sci_gateway1/cpp/opencv_BlockMatcher.cpp b/sci_gateway1/cpp/opencv_BlockMatcher.cpp new file mode 100644 index 0000000..654e5c8 --- /dev/null +++ b/sci_gateway1/cpp/opencv_BlockMatcher.cpp @@ -0,0 +1,606 @@ +/*************************************************** +Author : Tanmay Chaudhari, Shashank Shekhar + ***************************************************/ + +#include <bits/stdc++.h> +#include "opencv2/core/core.hpp" +#include "opencv2/features2d/features2d.hpp" +#include "opencv2/nonfree/features2d.hpp" +#include "opencv2/highgui/highgui.hpp" +#include "opencv2/imgproc/imgproc.hpp" + +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" + //#include "../common.cpp" + vector<pair<long double, pair<int, int > > > madv, msev; + vector<pair<pair<int, int>, pair<int, int > > > v; + + int opencv_BlockMatcher(char *fname, unsigned long fname_len) + { + //Error management variable + SciErr sciErr; + + //Variable declaration + int maxm, itr; + int iRows = 0; + int iCols = 0; + int nbInputArguments = 0; + int *piLen = NULL; + int *piAddr = NULL; + bool *providedArgs = NULL; + char *currentArg = NULL; + char **pstData = NULL; + double *blocksize = NULL; + double *maximumDisplacement = NULL; + double *overlap = NULL; + double *horizontal = NULL; + double *vertical = NULL; + double *magnitude = NULL; + string s,t; + string matchCriteria; + string outputValue; + Mat img, img1, img2; + + //Check input output arguments + CheckInputArgument(pvApiCtx, 2, 12); + CheckOutputArgument(pvApiCtx, 1, 1); + + nbInputArguments = *getNbInputArgument(pvApiCtx); + + if((nbInputArguments-2)%2!=0) + { + Scierror(999, "Incorrect number of arguments provided. Please check the documentation for more information.\n"); + return 0; + } + + providedArgs = (bool*)malloc(sizeof(bool) * 5); + for(int i=0;i<5;i++) + providedArgs[i] = 0; + + //Read image + //retrieveImage(img, 1); + //cvtColor(img, img1, CV_BGR2GRAY); + //retrieveImage(img, 2); + //cvtColor(img, img2, CV_BGR2GRAY); + + //Reading first image + sciErr = getVarAddressFromPosition(pvApiCtx, 1, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + // Extracting name of next argument takes three calls to getMatrixOfString + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + piLen = (int*) malloc(sizeof(int) * iRows * iCols); + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + pstData = (char**) malloc(sizeof(char*) * iRows * iCols); + for(int iterPstData = 0; iterPstData < iRows * iCols; iterPstData++) + { + pstData[iterPstData] = (char*) malloc(sizeof(char) * piLen[iterPstData] + 1); + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + img1 = imread(pstData[0], CV_LOAD_IMAGE_GRAYSCALE); + + //Reading second image + sciErr = getVarAddressFromPosition(pvApiCtx, 2, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + // Extracting name of next argument takes three calls to getMatrixOfString + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + piLen = (int*) malloc(sizeof(int) * iRows * iCols); + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + pstData = (char**) malloc(sizeof(char*) * iRows * iCols); + for(int iterPstData = 0; iterPstData < iRows * iCols; iterPstData++) + { + pstData[iterPstData] = (char*) malloc(sizeof(char) * piLen[iterPstData] + 1); + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + img2 = imread(pstData[0], CV_LOAD_IMAGE_GRAYSCALE); + + for(int iter=3;iter<=nbInputArguments;iter++) + { + // Getting address of next argument + sciErr = getVarAddressFromPosition(pvApiCtx, iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + // Extracting name of next argument takes three calls to getMatrixOfString + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + piLen = (int*) malloc(sizeof(int) * iRows * iCols); + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + pstData = (char**) malloc(sizeof(char*) * iRows * iCols); + for(int iterPstData = 0; iterPstData < iRows * iCols; iterPstData++) + { + pstData[iterPstData] = (char*) malloc(sizeof(char) * piLen[iterPstData] + 1); + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + currentArg = pstData[0]; + // providedArgs[] makes sure that no optional argument is provided more than once + + if(strcmp(currentArg,"BlockSize")==0) + { + if(!providedArgs[0]) + { + sciErr = getVarAddressFromPosition(pvApiCtx, ++iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(!isDoubleType(pvApiCtx, piAddr) || isVarComplex(pvApiCtx, piAddr)) + { + Scierror(999, "%s: Wrong type for input argument #%d: A real matrix expected.\n", fname, iter); + return 0; + } + sciErr = getMatrixOfDouble(pvApiCtx, piAddr, &iRows, &iCols, &blocksize); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(iRows!=1 || iCols!=2) + { + Scierror(999, "Incorrect dimension of matrix for argument BlockSize.\n"); + return 0; + } + if(blocksize[0]<=0 || blocksize[1]<=0) + { + Scierror(999, "Positive inputs for BlockSize needed.\n"); + return 0; + } + providedArgs[0]=1; + } + else if(providedArgs[0]) + { + Scierror(999, "Please provide optional arguments only once.\n"); + return 0; + } + } + else if(strcmp(currentArg, "Overlap")==0) + { + if(!providedArgs[1]) + { + sciErr = getVarAddressFromPosition(pvApiCtx, ++iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(!isDoubleType(pvApiCtx, piAddr) || isVarComplex(pvApiCtx, piAddr)) + { + Scierror(999, "%s: Wrong type for input argument #%d: A real matrix expected.\n", fname, iter); + return 0; + } + sciErr = getMatrixOfDouble(pvApiCtx, piAddr, &iRows, &iCols, &overlap); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(iRows!=1 || iCols!=2) + { + Scierror(999, "Incorrect dimension of matrix for argument Overlap.\n"); + return 0; + } + if(overlap[0]<0 || overlap[1]<0) + { + Scierror(999, "Non-negative inputs for Overlap needed\n"); + return 0; + } + providedArgs[1]=1; + + } + else if(providedArgs[1]) + { + Scierror(999, "Please provide optional arguments only once.\n"); + return 0; + } + } + else if(strcmp(currentArg, "MaximumDisplacement")==0) + { + if(!providedArgs[2]) + { + sciErr = getVarAddressFromPosition(pvApiCtx, ++iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(!isDoubleType(pvApiCtx, piAddr) || isVarComplex(pvApiCtx, piAddr)) + { + Scierror(999, "%s: Wrong type for input argument #%d: A real matrix expected.\n", fname, iter); + return 0; + } + sciErr = getMatrixOfDouble(pvApiCtx, piAddr, &iRows, &iCols, &maximumDisplacement); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + if(iRows!=1 || iCols!=2) + { + Scierror(999, "Incorrect dimension of matrix for argument MaximumDisplacement.\n"); + return 0; + } + if(maximumDisplacement[0]<0 || maximumDisplacement[1]<0) + { + Scierror(999, "Non-negative inputs for Overlap needed\n"); + return 0; + } + providedArgs[2]=1; + + } + else if(providedArgs[2]) + { + Scierror(999, "Please provide optional arguments only once.\n"); + return 0; + } + } + else if(strcmp(currentArg, "MatchCriteria")==0) + { + if(!providedArgs[3]) + { + sciErr = getVarAddressFromPosition(pvApiCtx, ++iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + piLen = (int*) malloc(sizeof(int) * iRows * iCols); + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + pstData = (char**) malloc(sizeof(char*) * iRows * iCols); + for(int iterPstData = 0; iterPstData < iRows * iCols; iterPstData++) + { + pstData[iterPstData] = (char*) malloc(sizeof(char) * piLen[iterPstData] + 1); + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + if(strcmp(pstData[0],"Mean square error")==0) + matchCriteria = "Mean Square Error"; + else if(strcmp(pstData[0], "Mean absolute difference")==0) + matchCriteria = " "; + else + { + Scierror(999, "Error: Please provide proper value for \"%s\". Permissible values are Mean square error or Mean absolute difference.\n", currentArg); + return 0; + } + providedArgs[3]=1; + } + else if(providedArgs[3]) + { + Scierror(999, "Please provide optional arguments only once.\n"); + return 0; + } + } + else if(strcmp(currentArg, "OutputValue")==0) + { + if(!providedArgs[4]) + { + sciErr = getVarAddressFromPosition(pvApiCtx, ++iter, &piAddr); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, NULL, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + piLen = (int*) malloc(sizeof(int) * iRows * iCols); + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, NULL); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + pstData = (char**) malloc(sizeof(char*) * iRows * iCols); + for(int iterPstData = 0; iterPstData < iRows * iCols; iterPstData++) + { + pstData[iterPstData] = (char*) malloc(sizeof(char) * piLen[iterPstData] + 1); + } + + sciErr = getMatrixOfString(pvApiCtx, piAddr, &iRows, &iCols, piLen, pstData); + if (sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + if(strcmp(pstData[0],"Magnitude-squared")==0) + outputValue = "Magnitude-squared"; + else if(strcmp(pstData[0], "Horizontal and vertical components")==0) + outputValue = "Horizontal and vertical components"; + else + { + Scierror(999, "Error: Please provide proper value for \"%s\". Permissible values are Magnitude-squared or Horizontal and vertical components.\n", currentArg); + return 0; + } + providedArgs[4]=1; + } + else if(providedArgs[4]) + { + Scierror(999, "Please provide optional arguments only once.\n"); + return 0; + } + } + } + + //Initialisaton + if(!providedArgs[0]) + { + blocksize = (double*)malloc(sizeof(double) * 2); + blocksize[0] = 17; + blocksize[1] = 17; + } + + if(!providedArgs[1]) + { + overlap = (double*)malloc(sizeof(double) * 2); + overlap[0] = 0; + overlap[1] = 0; + } + + if(!providedArgs[2]) + { + maximumDisplacement = (double*)malloc(sizeof(double) * 2); + maximumDisplacement[0] = 7; + maximumDisplacement[1] = 7; + } + + if(!providedArgs[3]) + { + matchCriteria = "Mean Square Error"; + } + + if(!providedArgs[4]) + { + outputValue = "Magnitude-squared"; + } + + if(outputValue == "Magnitude-squared") + { + magnitude = (double*)malloc(sizeof(double) * int(img1.rows) * int(img1.cols)); + } + else + { + horizontal = (double*)malloc(sizeof(double) * int(img1.rows) * int(img1.cols)); + vertical = (double*)malloc(sizeof(double) * int(img1.rows) * int(img1.cols)); + } + + iRows = iCols = 0; + for(int i=0;i<int(img1.rows);i+=int(blocksize[0])-int(overlap[0])) + { + if(i+int(blocksize[0])>=int(img1.rows)) + break; + for(int j=0;j<int(img1.cols);j+=int(blocksize[1])-int(overlap[1])) + { + if(j+int(blocksize[1])>=int(img1.cols)) + break; + madv.clear(); + msev.clear(); + for(int k=-int(maximumDisplacement[0]);k<=int(maximumDisplacement[0]);k++) + { + if((i+k>=int(img2.rows)) || (i+k+int(blocksize[0])>=int(img2.rows))) + break; + if(i+k<0) + continue; + for(int l=-int(maximumDisplacement[1]);l<=int(maximumDisplacement[1]);l++) + { + if((j+l>=int(img2.cols)) || (j+l+int(blocksize[1])>=int(img2.cols))) + break; + if(j+l<0) + continue; + + long double mad = 0; + long double mse = 0; + + for(int it1=0;it1<blocksize[0];it1++) + { + for(int it2=0;it2<blocksize[1];it2++) + { + long double cur1=img1.at<int>(i+it1,j+it2); + long double cur2=img2.at<int>(i+k+it1,j+l+it2); + long double error=cur1-cur2; + mad+=abs(error); + mad/=blocksize[0]*blocksize[1]; + mse+=error*error; + mse/=blocksize[0]*blocksize[1]; + } + } + + madv.push_back(make_pair(mad,make_pair(i+k,j+l))); + msev.push_back(make_pair(mse,make_pair(i+k,j+l))); + } + } + if(outputValue == "Magnitude-squared") + { + sort(msev.begin(),msev.end()); + if(msev.empty()) + v.push_back(make_pair(make_pair(i,j),make_pair(i+int(blocksize[0]),j+int(blocksize[1])))); + else + v.push_back(make_pair(make_pair(i,j),msev[0].second)); + } + else + { + sort(madv.begin(),madv.end()); + if(madv.empty()) + v.push_back(make_pair(make_pair(i,j),make_pair(i+int(blocksize[0]),j+int(blocksize[1])))); + else + v.push_back(make_pair(make_pair(i,j),madv[0].second)); + } + iCols++; + } + iRows++; + } + + if(outputValue == "Magnitude-squared") + { + //Creating list to store magnitude matrix in it + magnitude = (double*)malloc(sizeof(double) * int (v.size())); + for(int i=0;i<v.size();i++) + { + double mag = (v[i].second.first-v[i].first.first)*(v[i].second.first-v[i].first.first) + (v[i].second.second-v[i].first.second)*(v[i].second.second-v[i].first.second); + magnitude[i]=mag; + } + + sciErr = createList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, 1, &piAddr); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + sciErr = createMatrixOfDoubleInList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, piAddr, 1, iRows, (iCols/iRows), magnitude); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + } + else + { + //Creating list to store horizontal and vertical matrix in it + horizontal = (double*)malloc(sizeof(double) * int(v.size())); + vertical = (double*)malloc(sizeof(double) * int(v.size())); + for(int i=0;i<v.size();i++) + { + horizontal[i] = v[i].second.first-v[i].first.first; + vertical[i] = v[i].second.second-v[i].first.second; + } + + sciErr = createList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, 2, &piAddr); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + sciErr = createMatrixOfDoubleInList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, piAddr, 1, iRows, (iCols/iRows), horizontal); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + + sciErr = createMatrixOfDoubleInList(pvApiCtx, nbInputArgument(pvApiCtx) + 1, piAddr, 2, iRows, (iCols/iRows), vertical); + if(sciErr.iErr) + { + printError(&sciErr, 0); + return 0; + } + } + + //return output Arguments + AssignOutputVariable(pvApiCtx, 1) = nbInputArgument(pvApiCtx) + 1; + ReturnArguments(pvApiCtx); + madv.clear(); + msev.clear(); + v.clear(); + return 0; + } + /* ==================================================================== */ +} |