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