1b75c6efcSBarry Smith /* 2b75c6efcSBarry Smith 3b75c6efcSBarry Smith This is the equivalent of MATLAB's fwrite() only on sockets instead of 4b75c6efcSBarry Smith binary files. 5b75c6efcSBarry Smith */ 6b75c6efcSBarry Smith 7b75c6efcSBarry Smith #include <petscsys.h> 8b75c6efcSBarry Smith #include <../src/sys/classes/viewer/impls/socket/socket.h> 9b75c6efcSBarry Smith #include <mex.h> 10b75c6efcSBarry Smith 11b75c6efcSBarry Smith PetscErrorCode PetscBinaryWrite(int, const void *p, int, PetscDataType); 12b75c6efcSBarry Smith 13b75c6efcSBarry Smith #define PETSC_MEX_ERROR(a) \ 14b75c6efcSBarry Smith { \ 15b75c6efcSBarry Smith fprintf(stdout, "sread: %s \n", a); \ 16b75c6efcSBarry Smith return; \ 17b75c6efcSBarry Smith } 18*4ee01570SBarry Smith 19b75c6efcSBarry Smith PETSC_EXTERN void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) 20b75c6efcSBarry Smith { 21b75c6efcSBarry Smith int i, fd, cnt, dt; 22b75c6efcSBarry Smith PetscErrorCode ierr; 23b75c6efcSBarry Smith 24b75c6efcSBarry Smith /* check output parameters */ 25b75c6efcSBarry Smith if (nrhs != 3) PETSC_MEX_ERROR("Receive requires three input arguments."); 26b75c6efcSBarry Smith fd = (int)mxGetScalar(prhs[0]); 27b75c6efcSBarry Smith cnt = mxGetNumberOfElements(prhs[1]); 28b75c6efcSBarry Smith dt = (PetscDataType)mxGetScalar(prhs[2]); 29b75c6efcSBarry Smith 30b75c6efcSBarry Smith if (dt == PETSC_DOUBLE) { 31b75c6efcSBarry Smith ierr = PetscBinaryWrite(fd, mxGetPr(prhs[1]), cnt, (PetscDataType)dt); 32b75c6efcSBarry Smith if (ierr) PETSC_MEX_ERROR("Unable to send double items."); 33b75c6efcSBarry Smith } else if (dt == PETSC_INT) { 34b75c6efcSBarry Smith int *tmp = (int *)mxMalloc((cnt + 5) * sizeof(int)); 35b75c6efcSBarry Smith double *t = mxGetPr(prhs[1]); 36b75c6efcSBarry Smith for (i = 0; i < cnt; i++) tmp[i] = (int)t[i]; 37b75c6efcSBarry Smith ierr = PetscBinaryWrite(fd, tmp, cnt, (PetscDataType)dt); 38b75c6efcSBarry Smith if (ierr) PETSC_MEX_ERROR("Unable to send int items."); 39b75c6efcSBarry Smith mxFree(tmp); 40b75c6efcSBarry Smith } else if (dt == PETSC_CHAR) { 41b75c6efcSBarry Smith char *tmp = (char *)mxMalloc((cnt + 5) * sizeof(char)); 42b75c6efcSBarry Smith mxGetNChars(prhs[1], tmp, cnt + 1); 43b75c6efcSBarry Smith ierr = PetscBinaryWrite(fd, tmp, cnt, (PetscDataType)dt); 44b75c6efcSBarry Smith if (ierr) PETSC_MEX_ERROR("Unable to send char items."); 45b75c6efcSBarry Smith mxFree(tmp); 46b75c6efcSBarry Smith } else PETSC_MEX_ERROR("Unknown datatype."); 47b75c6efcSBarry Smith return; 48b75c6efcSBarry Smith } 49b75c6efcSBarry Smith 50b75c6efcSBarry Smith int main(int argc, char **argv) 51b75c6efcSBarry Smith { 52b75c6efcSBarry Smith return 0; 53b75c6efcSBarry Smith } 54