source: issm/trunk/src/mex/Scotch/Scotch.cpp@ 6412

Last change on this file since 6412 was 6412, checked in by Mathieu Morlighem, 14 years ago

moved ISSMERROR to _error_, ISSMASSERT to _assert_ and ISSMPRINTF to _printf_
File size: 7.1 KB
Line 
1/*\file Scotch.c
2 *\brief: Scotch partitioner mex module
3 */
4
5#ifdef HAVE_CONFIG_H
6#include "config.h"
7#else
8#error "Cannot compile with HAVE_CONFIG_H symbol! run configure first!"
9#endif
10
11
12#include "./Scotch.h"
13
14/******************************/
15/* */
16/* This is the main function. */
17/* */
18/******************************/
19
20void mexFunction( int nlhs,
21 mxArray *plhs[],
22 int nrhs,
23 const mxArray *prhs[] )
24{
25
26#ifndef _HAVE_SCOTCH_ //only works if scotch library has been compiled in.
27 _error_(" Scotch not available! Cannot carry out Scotch partitioning!");
28 #else
29
30 int argcm;
31 char **argvm=NULL;
32 int nvert =0,nedge2=0,napar =0;
33 mwIndex *ir=NULL,*jc=NULL;
34 int *adjir=NULL,*adjjc=NULL;
35 double *vld=NULL,*vwd=NULL,*ewd=NULL,*apd=NULL;
36 int *vli=NULL,*vwi=NULL,*ewi=NULL,*api=NULL;
37 char *archtyp=NULL;
38 int (*maptabi)[2]=NULL;
39 double* maptabd=NULL;
40 int i,j,k,imi=0,imo=0,isi=0,ierr;
41
42 /* Check for proper number of arguments */
43
44 if (nrhs == 0 && nlhs == 0) {
45 GmapUsage();
46 return;
47 }
48 else if (nrhs < 6 || nlhs > 1) {
49 GmapUsage();
50 mexErrMsgTxt(" ");
51 }
52
53/* load matlab argument list and convert to integer (note that converting here
54 and in the x-layer is inefficient, but it makes the x-layer more general) */
55
56 argvm = (char **) calloc(nrhs,sizeof(char *));
57
58 if (!(mxIsNumeric(prhs[imi]) &&
59 (mxGetM(prhs[imi]) == 1 && mxGetN(prhs[imi]) == 1))) {
60 argvm[isi] = (char *) calloc(4+1,sizeof(char));
61 strcpy(argvm[isi],"gmap");
62 mexPrintf("%s -- Using \"%s\" entry point.\n",
63 __FUNCT__,argvm[isi]);
64 isi++;
65 }
66 else {
67 argvm[isi] = (char *) calloc(5+1,sizeof(char));
68 strcpy(argvm[isi],"gpart");
69 mexPrintf("%s -- Using \"%s\" entry point.\n",
70 __FUNCT__,argvm[isi]);
71 isi++;
72
73 argvm[isi] = (char *) calloc(17,sizeof(char));
74 sprintf(argvm[isi],"%d",(int)mxGetScalar(prhs[imi]));
75 mexPrintf("%s -- Number of parts is %s.\n",
76 __FUNCT__,argvm[isi]);
77 isi++;
78 imi++;
79 }
80
81 if (!mxIsNumeric(prhs[imi]) || (!mxIsEmpty(prhs[imi]) && !mxIsSparse(prhs[imi]))) {
82 mexPrintf("%s -- Adjacency matrix must be numeric and sparse.\n",__FUNCT__);
83 mexErrMsgTxt(" ");
84 }
85 else {
86 nvert =mxGetM(prhs[imi]);
87 nedge2=mxGetNzmax(prhs[imi]);
88 if (mxGetNzmax(prhs[imi])) {
89 ir =mxGetIr(prhs[imi]);
90 adjir = (int *) malloc(mxGetNzmax(prhs[imi])*sizeof(int));
91 for (i=0; i<mxGetNzmax(prhs[imi]); i++)
92 adjir[i]=(int)ir[i];
93 }
94 if (mxGetN(prhs[imi])) {
95 jc =mxGetJc(prhs[imi]);
96 adjjc = (int *) malloc((mxGetN(prhs[imi])+1)*sizeof(int));
97 for (i=0; i<(mxGetN(prhs[imi])+1); i++)
98 adjjc[i]=(int)jc[i];
99 }
100 mexPrintf("%s -- Adjacency matrix is of size %d by %d with %d non-zeroes.\n",
101 __FUNCT__,mxGetM(prhs[imi]),mxGetN(prhs[imi]),mxGetNzmax(prhs[imi]));
102 }
103 imi++;
104
105 if (!mxIsNumeric(prhs[imi])) {
106 mexPrintf("%s -- Vertex label vector must be numeric.\n",__FUNCT__);
107 mexErrMsgTxt(" ");
108 }
109 else {
110 if (mxGetM(prhs[imi])*mxGetN(prhs[imi])) {
111 vld=mxGetPr(prhs[imi]);
112 vli = (int *) malloc(mxGetM(prhs[imi])*mxGetN(prhs[imi])*sizeof(int));
113 for (i=0; i<mxGetM(prhs[imi])*mxGetN(prhs[imi]); i++)
114 vli[i]=(int)vld[i];
115 }
116 mexPrintf("%s -- Vertex label vector is of size %d by %d.\n",
117 __FUNCT__,mxGetM(prhs[imi]),mxGetN(prhs[imi]));
118 }
119 imi++;
120
121 if (!mxIsNumeric(prhs[imi])) {
122 mexPrintf("%s -- Vertex weight vector must be numeric.\n",__FUNCT__);
123 mexErrMsgTxt(" ");
124 }
125 else {
126 if (mxGetM(prhs[imi])*mxGetN(prhs[imi])) {
127 vwd=mxGetPr(prhs[imi]);
128 vwi = (int *) malloc(mxGetM(prhs[imi])*mxGetN(prhs[imi])*sizeof(int));
129 for (i=0; i<mxGetM(prhs[imi])*mxGetN(prhs[imi]); i++)
130 vwi[i]=(int)vwd[i];
131 }
132 mexPrintf("%s -- Vertex weight vector is of size %d by %d.\n",
133 __FUNCT__,mxGetM(prhs[imi]),mxGetN(prhs[imi]));
134 }
135 imi++;
136
137 if (!mxIsNumeric(prhs[imi]) || (!mxIsEmpty(prhs[imi]) && !mxIsSparse(prhs[imi]))) {
138 mexPrintf("%s -- Edge weight matrix must be numeric and sparse.\n",__FUNCT__);
139 mexErrMsgTxt(" ");
140 }
141 else {
142 if (mxGetM(prhs[imi])) {
143 ewd=mxGetPr(prhs[imi]);
144 ewi = (int *) malloc(mxGetM(prhs[imi])*sizeof(int));
145 for (i=0; i<mxGetNzmax(prhs[imi]); i++)
146 ewi[i]=(int)ewd[i];
147 }
148 mexPrintf("%s -- Edge weight matrix is of size %d by %d with %d non-zeroes.\n",
149 __FUNCT__,mxGetM(prhs[imi]),mxGetN(prhs[imi]),mxGetNzmax(prhs[imi]));
150 }
151 imi++;
152
153 if (!((strlen (argvm[0]) >= 5) &&
154 (strncmp (argvm[0] + strlen (argvm[0]) - 5, "gpart", 5) == 0))) {
155 if (!mxIsChar(prhs[imi])) {
156 mexPrintf("%s -- Architecture type must be character.\n",__FUNCT__);
157 mexErrMsgTxt(" ");
158 }
159 else {
160 if (mxGetM(prhs[imi])*mxGetN(prhs[imi])) {
161 archtyp = (char *) calloc(mxGetM(prhs[imi])*mxGetN(prhs[imi])+1,sizeof(char));
162 mxGetString(prhs[imi],archtyp,mxGetM(prhs[imi])*mxGetN(prhs[imi])+1);
163 }
164 mexPrintf("%s -- Architecture type is \"%s\".\n",
165 __FUNCT__,archtyp);
166 }
167 imi++;
168
169 if (!mxIsNumeric(prhs[imi])) {
170 mexPrintf("%s -- Architecture parameter vector must be numeric.\n",__FUNCT__);
171 mexErrMsgTxt(" ");
172 }
173 else {
174 napar =mxGetM(prhs[imi])*mxGetN(prhs[imi]);
175 if (mxGetM(prhs[imi])*mxGetN(prhs[imi])) {
176 apd=mxGetPr(prhs[imi]);
177 api = (int *) malloc(mxGetM(prhs[imi])*mxGetN(prhs[imi])*sizeof(int));
178 for (i=0; i<mxGetM(prhs[imi])*mxGetN(prhs[imi]); i++)
179 api[i]=(int)apd[i];
180 }
181 mexPrintf("%s -- Architecture parameter vector is of size %d by %d.\n",
182 __FUNCT__,mxGetM(prhs[imi]),mxGetN(prhs[imi]));
183 }
184 imi++;
185 }
186
187 while (imi < nrhs) {
188 if (!mxIsChar(prhs[imi])) {
189 mexPrintf("%s -- prhs[%d] must be character.\n",__FUNCT__,imi);
190 mexErrMsgTxt(" ");
191 }
192 else {
193 argvm[isi] = (char *) calloc(mxGetM(prhs[imi])*mxGetN(prhs[imi])+1,sizeof(char));
194 mxGetString(prhs[imi],argvm[isi],mxGetM(prhs[imi])*mxGetN(prhs[imi])+1);
195 }
196 isi++;
197 imi++;
198 }
199 argcm=isi;
200 mexPrintf("argcm=%d\n",argcm);
201 for (i=0; i<argcm; i++)
202 mexPrintf("argvm[%d]=\"%s\"\n",i,argvm[i]);
203
204 /* Do the actual computations in a subroutine */
205
206 mexPrintf("Gmapx:\n");
207 ierr=gmapx(&maptabi,
208 argcm,
209 argvm,
210 nvert,
211 nedge2,
212 adjir,
213 adjjc,
214 vli,
215 vwi,
216 ewi,
217 archtyp,
218 napar,
219 api);
220 mexPrintf("%s -- Error %d from Gmapx.\n",__FUNCT__,ierr);
221
222/* for (i=0; i<nvert; i++)
223 mexPrintf("maptabi[%d][0]=%d, maptabi[%d][1]=%d\n",
224 i,maptabi[i][0],i,maptabi[i][1]); */
225
226 /* Create matrices for the return arguments */
227
228 if (maptabi) {
229 plhs[imo]=mxCreateDoubleMatrix(nvert, 2, mxREAL);
230 maptabd = mxGetPr(plhs[imo]);
231 k=0;
232 for (j=0; j<2; j++)
233 for (i=0; i<nvert; i++)
234 maptabd[k++]=(double)maptabi[i][j];
235 //free(maptabi);
236 }
237 else {
238 plhs[imo]=mxCreateDoubleMatrix(0, 2, mxREAL);
239 }
240 imo++;
241
242 /*if (argvm)
243 for (i=argcm-1; i>=0; i--)
244 free(argvm[i]);
245 if (api) free(api);
246 if (archtyp) free(archtyp);
247 if (ewi) free(ewi);
248 if (vwi) free(vwi);
249 if (vli) free(vli);
250 if (adjjc) free(adjjc);
251 if (adjir) free(adjir);*/
252
253 return;
254#endif //#ifndef _HAVE_SCOTCH_
255}
256
257void GmapUsage( void )
258{
259
260 mexPrintf("\n");
261 mexPrintf("Usage: [maptab]=Scotch(adjmat,vertlb,vertwt,edgewt,archtyp,archpar,\n");
262 mexPrintf(" Scotch-specific parameters);\n");
263 mexPrintf("\n");
264
265 return;
266}
267
Note: See TracBrowser for help on using the repository browser.