idamnjanovic@60: /**************************************************************************
idamnjanovic@60:  *
idamnjanovic@60:  * File name: addtocols.c
idamnjanovic@60:  *
idamnjanovic@60:  * Ron Rubinstein
idamnjanovic@60:  * Computer Science Department
idamnjanovic@60:  * Technion, Haifa 32000 Israel
idamnjanovic@60:  * ronrubin@cs
idamnjanovic@60:  *
idamnjanovic@60:  * Last Updated: 19.4.2009
idamnjanovic@60:  *
idamnjanovic@60:  *************************************************************************/
idamnjanovic@60: 
idamnjanovic@60: 
idamnjanovic@60: #include "mex.h"
idamnjanovic@60: 
idamnjanovic@60: 
idamnjanovic@60: /* Input Arguments */
idamnjanovic@60: 
idamnjanovic@60: #define	X_IN	prhs[0]
idamnjanovic@60: #define V_IN  prhs[1]
idamnjanovic@60: 
idamnjanovic@60: 
idamnjanovic@60: /* Output Arguments */
idamnjanovic@60: 
idamnjanovic@60: #define	Y_OUT	plhs[0]
idamnjanovic@60: 
idamnjanovic@60: 
idamnjanovic@60: void mexFunction(int nlhs, mxArray *plhs[], 
idamnjanovic@60: 		             int nrhs, const mxArray*prhs[])
idamnjanovic@60:      
idamnjanovic@60: { 
idamnjanovic@60:     double *x, *y, *v, *xend;  
idamnjanovic@60:     mwSize m,n,m1,n1;
idamnjanovic@60:     mwIndex counter; 
idamnjanovic@60:     
idamnjanovic@60:     
idamnjanovic@60:     /* Check for proper number of arguments */
idamnjanovic@60:     
idamnjanovic@60:     if (nrhs != 2) {
idamnjanovic@60:       mexErrMsgTxt("Two input arguments required."); 
idamnjanovic@60:     } else if (nlhs > 1) {
idamnjanovic@60:       mexErrMsgTxt("Too many output arguments."); 
idamnjanovic@60:     } 
idamnjanovic@60:     
idamnjanovic@60:     
idamnjanovic@60:     /* Check the the input dimensions */ 
idamnjanovic@60:     
idamnjanovic@60:     m = mxGetM(X_IN);
idamnjanovic@60:     n = mxGetN(X_IN);
idamnjanovic@60:     if (!mxIsDouble(X_IN) || mxIsComplex(X_IN) || mxGetNumberOfDimensions(X_IN)>2) {
idamnjanovic@60:       mexErrMsgTxt("ADDTOCOLS requires that X be a double matrix.");
idamnjanovic@60:     }
idamnjanovic@60:     m1 = mxGetM(V_IN);
idamnjanovic@60:     n1 = mxGetN(V_IN);
idamnjanovic@60:     if (!mxIsDouble(V_IN) || mxIsComplex(V_IN) || (m1!=1 && n1!=1)) {
idamnjanovic@60:       mexErrMsgTxt("ADDTOCOLS requires that V be a double vector.");
idamnjanovic@60:     }
idamnjanovic@60:     if ((m1==1 && n1!=n) || (n1==1 && m1!=n)) {
idamnjanovic@60:       mexErrMsgTxt("Error in ADDTOCOLS: dimensions of V and X must agree.");
idamnjanovic@60:     }
idamnjanovic@60:     
idamnjanovic@60:     
idamnjanovic@60:     /* Create a matrix for the return argument */ 
idamnjanovic@60:     Y_OUT = mxCreateDoubleMatrix(m, n, mxREAL);
idamnjanovic@60:          
idamnjanovic@60:     
idamnjanovic@60:     /* Assign pointers to the various parameters */ 
idamnjanovic@60:     x = mxGetPr(X_IN); 
idamnjanovic@60:     v = mxGetPr(V_IN);
idamnjanovic@60:     y = mxGetPr(Y_OUT);
idamnjanovic@60:             
idamnjanovic@60:     
idamnjanovic@60:     /* Do the actual computation */
idamnjanovic@60:     
idamnjanovic@60:     xend = x+(m*n);
idamnjanovic@60:     counter = 0;
idamnjanovic@60:     while (x<xend) {
idamnjanovic@60:       (*y) = (*x) + (*v);
idamnjanovic@60:       y++; x++; counter++;
idamnjanovic@60:       if (counter==m) {v++; counter=0;}
idamnjanovic@60:     }
idamnjanovic@60:     
idamnjanovic@60:     return;
idamnjanovic@60: }