idamnjanovic@70: /**************************************************************************
idamnjanovic@70:  *
idamnjanovic@70:  * File name: sprow.c
idamnjanovic@70:  *
idamnjanovic@70:  * Ron Rubinstein
idamnjanovic@70:  * Computer Science Department
idamnjanovic@70:  * Technion, Haifa 32000 Israel
idamnjanovic@70:  * ronrubin@cs
idamnjanovic@70:  *
idamnjanovic@70:  * Last Updated: 24.8.2009
idamnjanovic@70:  *
idamnjanovic@70:  *************************************************************************/
idamnjanovic@70: 
idamnjanovic@70: 
idamnjanovic@70: #include "mex.h"
idamnjanovic@70: #include "mexutils.h"
idamnjanovic@70: 
idamnjanovic@70: 
idamnjanovic@70: /* Input Arguments */
idamnjanovic@70: 
idamnjanovic@70: #define	A_IN	prhs[0]
idamnjanovic@70: #define J_IN  prhs[1]
idamnjanovic@70: 
idamnjanovic@70: 
idamnjanovic@70: /* Output Arguments */
idamnjanovic@70: 
idamnjanovic@70: #define	X_OUT	  plhs[0]
idamnjanovic@70: #define	ID_OUT	plhs[1]
idamnjanovic@70: 
idamnjanovic@70: 
idamnjanovic@70: void mexFunction(int nlhs, mxArray *plhs[], 
idamnjanovic@70: 		             int nrhs, const mxArray*prhs[])
idamnjanovic@70:      
idamnjanovic@70: { 
idamnjanovic@70:   double *pr, *x, *id, rowid;
idamnjanovic@70:   mwIndex *ir, *jc;
idamnjanovic@70:   mwSize m, n;
idamnjanovic@70:   mwIndex i, j, k, l, rowlen;
idamnjanovic@70:   
idamnjanovic@70:   if (nrhs != 2) {
idamnjanovic@70:     mexErrMsgTxt("GETSPROW requires two input arguments.");
idamnjanovic@70:   } else if (nlhs > 2) {
idamnjanovic@70:     mexErrMsgTxt("Too many output arguments.");
idamnjanovic@70:   }
idamnjanovic@70:   
idamnjanovic@70:   checkmatrix(A_IN, "GETSPROW", "A");
idamnjanovic@70:   checksparse(A_IN, "GETSPROW", "A");
idamnjanovic@70:   checkscalar(J_IN, "GETSPROW", "J");
idamnjanovic@70:   
idamnjanovic@70:   m = mxGetM(A_IN);
idamnjanovic@70:   n = mxGetN(A_IN);
idamnjanovic@70:   
idamnjanovic@70:   rowid = mxGetScalar(J_IN);
idamnjanovic@70:   if (rowid < 0) {
idamnjanovic@70:     mexErrMsgTxt("Invalid row index.");
idamnjanovic@70:   }
idamnjanovic@70:   j = (mwIndex)(rowid + 1e-2);
idamnjanovic@70:   if (j<1 || j>m) {
idamnjanovic@70:     mexErrMsgTxt("Row index is out of range.");
idamnjanovic@70:   }
idamnjanovic@70:   j--;
idamnjanovic@70:   
idamnjanovic@70:   pr = mxGetPr(A_IN);
idamnjanovic@70:   ir = mxGetIr(A_IN);
idamnjanovic@70:   jc = mxGetJc(A_IN);
idamnjanovic@70:   
idamnjanovic@70:   /* Determine length of row */
idamnjanovic@70:   rowlen = 0;
idamnjanovic@70:   for (i=0; i<jc[n]; ++i) {
idamnjanovic@70:     rowlen += (ir[i]==j);
idamnjanovic@70:   }
idamnjanovic@70:   
idamnjanovic@70:   /* Allocate output parameters */
idamnjanovic@70:   X_OUT = mxCreateDoubleMatrix(1, rowlen, mxREAL);
idamnjanovic@70:   ID_OUT = mxCreateDoubleMatrix(1, rowlen, mxREAL);
idamnjanovic@70:   
idamnjanovic@70:   x = mxGetPr(X_OUT);
idamnjanovic@70:   id = mxGetPr(ID_OUT);
idamnjanovic@70:   
idamnjanovic@70:   /* Compute j-th row */
idamnjanovic@70:   k=0;
idamnjanovic@70:   for (l=1; l<=n; ++l) {
idamnjanovic@70:     i = jc[l-1];
idamnjanovic@70:     while (i<jc[l] && ir[i]<j) {
idamnjanovic@70:       i++;
idamnjanovic@70:     }
idamnjanovic@70:     if (i<jc[l] && ir[i]==j) {
idamnjanovic@70:       x[k] = pr[i];
idamnjanovic@70:       id[k] = l;
idamnjanovic@70:       k++;
idamnjanovic@70:     }
idamnjanovic@70:   }
idamnjanovic@70:   
idamnjanovic@70: }