Homec4science
Diffusion Research Scripts (Thomas Bolton) (master) Browse

Scripts_Sharing/ISFC/Utils_DL/spams-matlab/dags/mex/mexCountConnexComponents.cpp
1c5bab09dbdamaster

Last Change
Raw File

mexCountConnexComponents.cpp
View Options

/*!
/* Software SPAMS v2.3 - Copyright 2009-2011 Julien Mairal
*
* This file is part of SPAMS.
*
* SPAMS is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* SPAMS is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with SPAMS. If not, see <http://www.gnu.org/licenses/>.
*/
#include <mex.h>
#include <mexutils.h>
#include <dag.h>
template <typename T>
inline void callFunction(mxArray* plhs[], const mxArray*prhs[]) {
if (!mexCheckType<T>(prhs[0]))
mexErrMsgTxt("type of argument 1 is not consistent");
if (!mxIsSparse(prhs[0]))
mexErrMsgTxt("argument 1 should be sparse");
const mwSize* dimsD=mxGetDimensions(prhs[0]);
int mD=static_cast<int>(dimsD[0]);
int p=static_cast<int>(dimsD[1]);
const int n = p;
double* D_v;
mwSize* D_r, *D_pB, *D_pE;
int* D_r2, *D_pB2, *D_pE2;
T* D_v2;
D_v=static_cast<double*>(mxGetPr(prhs[0]));
D_r=mxGetIr(prhs[0]);
D_pB=mxGetJc(prhs[0]);
D_pE=D_pB+1;
createCopySparse<T>(D_v2,D_r2,D_pB2,D_pE2,
D_v,D_r,D_pB,D_pE,p);
SpMatrix<T> G(D_v2,D_r2,D_pB2,D_pE2,mD,p,D_pB2[p]);
T* pr_alpha0 = reinterpret_cast<T*>(mxGetPr(prhs[1]));
const mwSize* dimsAlpha=mxGetDimensions(prhs[1]);
int nalpha=static_cast<int>(dimsAlpha[0])*static_cast<int>(dimsAlpha[1]);
if (nalpha != G.m())
mexErrMsgTxt("inconsistent vector size");
Vector<T> active(pr_alpha0,nalpha);
int num=count_cc_graph(G,active);
deleteCopySparse<T>(D_v2,D_r2,D_pB2,D_pE2,
D_v,D_r);
plhs[0]=createMatrix<T>(1,1);
T* pr_out=reinterpret_cast<T*>(mxGetPr(plhs[0]));
(*pr_out)=static_cast<T>(num);
}
void mexFunction(int nlhs, mxArray *plhs[],int nrhs, const mxArray *prhs[]) {
if (nrhs != 2)
mexErrMsgTxt("Bad number of inputs arguments");
if (nlhs != 1)
mexErrMsgTxt("Bad number of output arguments");
if (mxGetClassID(prhs[0]) == mxDOUBLE_CLASS) {
callFunction<double>(plhs,prhs);
} else {
callFunction<float>(plhs,prhs);
}
}
c4science · Help