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