CVF.zip_curvature_curvature flow_flow

#include "mex.h" #include "mat.h" #include "matrix.h" mxArray* MapaDistancies(mxArray* Im) { mxArray* Res; int ndim; int *dims; double *pIm; double *pRes; int nCol, nRow; int i, j; double max; ndim = mxGetNumberOfDimensions(Im); dims = mxGetDimensions(Im); if (ndim >= 2) Res = mxCreateDoubleMatrix(dims[0], dims[1], mxREAL); else return NULL; pIm = mxGetPr(Im); pRes = (double *) mxGetData(Res); nCol = dims[1]; nRow = dims[0]; /* Calcul del mapa de distancies */ /* Inicialitzacio */ for (i=0; i<nRow; i++) { for (j=0; j<nCol; j++) { if (pIm[j*nRow + i]) pRes[j*nRow + i] = 0; else pRes[j*nRow + i] = mxGetInf(); } } /* UP -> DOWN */ for (i=1; i<nRow; i++) { for (j=0; j<nCol; j++) { pRes[j*nRow + i] = min(pRes[j*nRow + i - 1] + 2, pRes[j*nRow + i]); } } /* DOWN -> UP */ for (i=nRow-2; i>=0; i--) { for (j=0; j<nCol; j++) { pRes[j*nRow + i] = min(pRes[j*nRow + i + 1] + 2,pRes[j*nRow + i]); } } /* LEFT -> RIGHT */ for (j=1; j<nCol; j++) { for (i=0; i<nRow; i++) { pRes[j*nRow + i] = min(pRes[(j - 1)*nRow + i] + 2, pRes[j*nRow + i]); } } /* RIGHT -> LEFT */ for (j=nCol-2; j>=0; j--) { for (i=0; i<nRow; i++) { pRes[j*nRow + i] = min(pRes[(j + 1)*nRow + i] + 2, pRes[j*nRow + i]); } } /* DIAGONAL 1 / */ for (j=1; j<nCol; j++) { for (i=1; i<nRow; i++) { pRes[j*nRow + i] = min(pRes[(j - 1)*nRow + i - 1] + 3, pRes[j*nRow + i]); } } /* DIAGONAL 2 / */ for (j=nCol-2; j>=0; j--) { for (i=nRow-2; i>=0; i--) { pRes[j*nRow + i] = min(pRes[(j + 1)*nRow + i + 1] + 3, pRes[j*nRow + i]); } } /* DIAGONAL 3 / */ for (j=1; j<nCol; j++) { for (i=nRow-2; i>=0; i--) { pRes[j*nRow + i] = min(pRes[(j - 1)*nRow + i + 1] + 3, pRes[j*nRow + i]); } } /* DIAGONAL 4 / */ for (j=nCol-2; j>=0; j--) { for (i=1; i<nRow; i++) { pRes[j*nRow + i] = min(pRes[(j + 1)*nRow + i - 1] + 3, pRes[j*nRow + i]); } } /* Normalitzacio del mapa de distancies max = 1; for (i=0; i<nCol*nRow; i++) { if (max < pRes[i]) max = pRes[i]; } for (i=0; i<nCol*nRow; i++) { pRes[i] /= max; } */ return Res; } void mexFunction(int nlhs,mxArray *plhs[],int nrhs,const mxArray *prhs[]) { char buffer[80]; /* Comprova els parametres */ if (nrhs!=1) { mexErrMsgTxt("Numero de parametres incorrecte"); } if (!mxIsDouble(prhs[0])) { mexErrMsgTxt("Format d'imatge incorrecte. Ha de ser una imatge DOUBLE."); } /* Crida a la funcio de proces */ plhs[0] = MapaDistancies(prhs[0]); nlhs = 1; }