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