1e5c89e4eSSatish Balay 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay This file contains simple binary read/write routines. 4e5c89e4eSSatish Balay */ 5e5c89e4eSSatish Balay 6c6db04a5SJed Brown #include <petscsys.h> 7e5c89e4eSSatish Balay #include <errno.h> 8e5c89e4eSSatish Balay #include <fcntl.h> 9e5c89e4eSSatish Balay #if defined(PETSC_HAVE_UNISTD_H) 10e5c89e4eSSatish Balay #include <unistd.h> 11e5c89e4eSSatish Balay #endif 12e5c89e4eSSatish Balay #if defined (PETSC_HAVE_IO_H) 13e5c89e4eSSatish Balay #include <io.h> 14e5c89e4eSSatish Balay #endif 15c6db04a5SJed Brown #include <petscbt.h> 16e5c89e4eSSatish Balay 176de02169SBarry Smith /* --------------------------------------------------------- */ 18e5c89e4eSSatish Balay #undef __FUNCT__ 196de02169SBarry Smith #define __FUNCT__ "PetscByteSwapEnum" 20e5c89e4eSSatish Balay /* 216de02169SBarry Smith PetscByteSwapEnum - Swap bytes in a PETSc Enum 22e5c89e4eSSatish Balay 23e5c89e4eSSatish Balay */ 247087cfbeSBarry Smith PetscErrorCode PetscByteSwapEnum(PetscEnum *buff,PetscInt n) 25e5c89e4eSSatish Balay { 266de02169SBarry Smith PetscInt i,j; 270b20345dSBarry Smith PetscEnum tmp = ENUM_DUMMY; 28e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 29e5c89e4eSSatish Balay 30e5c89e4eSSatish Balay PetscFunctionBegin; 31e5c89e4eSSatish Balay for (j=0; j<n; j++) { 32e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 330b20345dSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) { 346de02169SBarry Smith ptr2[i] = ptr1[sizeof(PetscEnum)-1-i]; 35e5c89e4eSSatish Balay } 36e0890e22SSatish Balay for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) { 37e0890e22SSatish Balay ptr1[i] = ptr2[i]; 38e0890e22SSatish Balay } 39e5c89e4eSSatish Balay } 40e5c89e4eSSatish Balay PetscFunctionReturn(0); 41e5c89e4eSSatish Balay } 426de02169SBarry Smith 436de02169SBarry Smith #undef __FUNCT__ 44acfcf0e5SJed Brown #define __FUNCT__ "PetscByteSwapBool" 456de02169SBarry Smith /* 46acfcf0e5SJed Brown PetscByteSwapBool - Swap bytes in a PETSc Bool 476de02169SBarry Smith 486de02169SBarry Smith */ 497087cfbeSBarry Smith PetscErrorCode PetscByteSwapBool(PetscBool *buff,PetscInt n) 506de02169SBarry Smith { 516de02169SBarry Smith PetscInt i,j; 52ace3abfcSBarry Smith PetscBool tmp = PETSC_FALSE; 53e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 546de02169SBarry Smith 556de02169SBarry Smith PetscFunctionBegin; 566de02169SBarry Smith for (j=0; j<n; j++) { 576de02169SBarry Smith ptr1 = (char*)(buff + j); 58ace3abfcSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscBool); i++) { 59ace3abfcSBarry Smith ptr2[i] = ptr1[sizeof(PetscBool)-1-i]; 606de02169SBarry Smith } 61ace3abfcSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscBool); i++) { 62e0890e22SSatish Balay ptr1[i] = ptr2[i]; 63e0890e22SSatish Balay } 646de02169SBarry Smith } 656de02169SBarry Smith PetscFunctionReturn(0); 666de02169SBarry Smith } 676de02169SBarry Smith 68e5c89e4eSSatish Balay #undef __FUNCT__ 69bd1d2e58SBarry Smith #define __FUNCT__ "PetscByteSwapInt" 70bd1d2e58SBarry Smith /* 716de02169SBarry Smith PetscByteSwapInt - Swap bytes in a PETSc integer (which may be 32 or 64 bits) 72bd1d2e58SBarry Smith 73bd1d2e58SBarry Smith */ 747087cfbeSBarry Smith PetscErrorCode PetscByteSwapInt(PetscInt *buff,PetscInt n) 75bd1d2e58SBarry Smith { 76bd1d2e58SBarry Smith PetscInt i,j,tmp = 0; 77e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 78bd1d2e58SBarry Smith 79bd1d2e58SBarry Smith PetscFunctionBegin; 80bd1d2e58SBarry Smith for (j=0; j<n; j++) { 81bd1d2e58SBarry Smith ptr1 = (char*)(buff + j); 820b20345dSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscInt); i++) { 83bd1d2e58SBarry Smith ptr2[i] = ptr1[sizeof(PetscInt)-1-i]; 84bd1d2e58SBarry Smith } 85e0890e22SSatish Balay for (i=0; i<(PetscInt)sizeof(PetscInt); i++) { 86e0890e22SSatish Balay ptr1[i] = ptr2[i]; 87e0890e22SSatish Balay } 88bd1d2e58SBarry Smith } 89bd1d2e58SBarry Smith PetscFunctionReturn(0); 90bd1d2e58SBarry Smith } 91bd1d2e58SBarry Smith /* --------------------------------------------------------- */ 92bd1d2e58SBarry Smith #undef __FUNCT__ 93e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort" 94e5c89e4eSSatish Balay /* 95e5c89e4eSSatish Balay PetscByteSwapShort - Swap bytes in a short 96e5c89e4eSSatish Balay */ 977087cfbeSBarry Smith PetscErrorCode PetscByteSwapShort(short *buff,PetscInt n) 98e5c89e4eSSatish Balay { 99e5c89e4eSSatish Balay PetscInt i,j; 100e5c89e4eSSatish Balay short tmp; 101e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 102e5c89e4eSSatish Balay 103e5c89e4eSSatish Balay PetscFunctionBegin; 104e5c89e4eSSatish Balay for (j=0; j<n; j++) { 105e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 1060b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(short); i++) { 107f348e1baSSatish Balay ptr2[i] = ptr1[sizeof(short)-1-i]; 108e5c89e4eSSatish Balay } 109e0890e22SSatish Balay for (i=0; i<(PetscInt) sizeof(short); i++) { 110e0890e22SSatish Balay ptr1[i] = ptr2[i]; 111e0890e22SSatish Balay } 112e5c89e4eSSatish Balay } 113e5c89e4eSSatish Balay PetscFunctionReturn(0); 114e5c89e4eSSatish Balay } 115e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 116e5c89e4eSSatish Balay #undef __FUNCT__ 117e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar" 118e5c89e4eSSatish Balay /* 119e5c89e4eSSatish Balay PetscByteSwapScalar - Swap bytes in a double 120e5c89e4eSSatish Balay Complex is dealt with as if array of double twice as long. 121e5c89e4eSSatish Balay */ 1227087cfbeSBarry Smith PetscErrorCode PetscByteSwapScalar(PetscScalar *buff,PetscInt n) 123e5c89e4eSSatish Balay { 124e5c89e4eSSatish Balay PetscInt i,j; 125e5c89e4eSSatish Balay PetscReal tmp,*buff1 = (PetscReal*)buff; 126e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 127e5c89e4eSSatish Balay 128e5c89e4eSSatish Balay PetscFunctionBegin; 129e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 130e5c89e4eSSatish Balay n *= 2; 131e5c89e4eSSatish Balay #endif 132e5c89e4eSSatish Balay for (j=0; j<n; j++) { 133e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 1340b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(PetscReal); i++) { 135e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(PetscReal)-1-i]; 136e5c89e4eSSatish Balay } 137e0890e22SSatish Balay for (i=0; i<(PetscInt) sizeof(PetscReal); i++) { 138e0890e22SSatish Balay ptr1[i] = ptr2[i]; 139e0890e22SSatish Balay } 140e5c89e4eSSatish Balay } 141e5c89e4eSSatish Balay PetscFunctionReturn(0); 142e5c89e4eSSatish Balay } 143e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 144e5c89e4eSSatish Balay #undef __FUNCT__ 145e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapDouble" 146e5c89e4eSSatish Balay /* 147e5c89e4eSSatish Balay PetscByteSwapDouble - Swap bytes in a double 148e5c89e4eSSatish Balay */ 1497087cfbeSBarry Smith PetscErrorCode PetscByteSwapDouble(double *buff,PetscInt n) 150e5c89e4eSSatish Balay { 151e5c89e4eSSatish Balay PetscInt i,j; 152e5c89e4eSSatish Balay double tmp,*buff1 = (double*)buff; 153e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 154e5c89e4eSSatish Balay 155e5c89e4eSSatish Balay PetscFunctionBegin; 156e5c89e4eSSatish Balay for (j=0; j<n; j++) { 157e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 1580b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(double); i++) { 159e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(double)-1-i]; 160e5c89e4eSSatish Balay } 161e0890e22SSatish Balay for (i=0; i<(PetscInt) sizeof(double); i++) { 162e0890e22SSatish Balay ptr1[i] = ptr2[i]; 163e0890e22SSatish Balay } 164e5c89e4eSSatish Balay } 165e5c89e4eSSatish Balay PetscFunctionReturn(0); 166e5c89e4eSSatish Balay } 167e39fd77fSBarry Smith 168e39fd77fSBarry Smith #undef __FUNCT__ 169e95bf02fSSatish Balay #define __FUNCT__ "PetscByteSwapFloat" 170e95bf02fSSatish Balay /* 171e95bf02fSSatish Balay PetscByteSwapFloat - Swap bytes in a float 172e95bf02fSSatish Balay */ 173e95bf02fSSatish Balay PetscErrorCode PetscByteSwapFloat(float *buff,PetscInt n) 174e95bf02fSSatish Balay { 175e95bf02fSSatish Balay PetscInt i,j; 176e95bf02fSSatish Balay float tmp,*buff1 = (float*)buff; 177e95bf02fSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 178e95bf02fSSatish Balay 179e95bf02fSSatish Balay PetscFunctionBegin; 180e95bf02fSSatish Balay for (j=0; j<n; j++) { 181e95bf02fSSatish Balay ptr1 = (char*)(buff1 + j); 182e95bf02fSSatish Balay for (i=0; i<(PetscInt) sizeof(float); i++) { 183e95bf02fSSatish Balay ptr2[i] = ptr1[sizeof(float)-1-i]; 184e95bf02fSSatish Balay } 185e95bf02fSSatish Balay for (i=0; i<(PetscInt) sizeof(float); i++) { 186e95bf02fSSatish Balay ptr1[i] = ptr2[i]; 187e95bf02fSSatish Balay } 188e95bf02fSSatish Balay } 189e95bf02fSSatish Balay PetscFunctionReturn(0); 190e95bf02fSSatish Balay } 191e95bf02fSSatish Balay 192e95bf02fSSatish Balay #undef __FUNCT__ 193e39fd77fSBarry Smith #define __FUNCT__ "PetscByteSwap" 194e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count) 195e39fd77fSBarry Smith { 196e39fd77fSBarry Smith PetscErrorCode ierr; 197e39fd77fSBarry Smith 198e39fd77fSBarry Smith PetscFunctionBegin; 199e39fd77fSBarry Smith if (pdtype == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);} 200e39fd77fSBarry Smith else if (pdtype == PETSC_ENUM) {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);} 201acfcf0e5SJed Brown else if (pdtype == PETSC_BOOL) {ierr = PetscByteSwapBool((PetscBool*)data,count);CHKERRQ(ierr);} 202e39fd77fSBarry Smith else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);} 203e39fd77fSBarry Smith else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);} 204e95bf02fSSatish Balay else if (pdtype == PETSC_FLOAT) {ierr = PetscByteSwapFloat((float*)data,count);CHKERRQ(ierr);} 205e39fd77fSBarry Smith else if (pdtype == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);} 206e39fd77fSBarry Smith PetscFunctionReturn(0); 207e39fd77fSBarry Smith } 208e39fd77fSBarry Smith 209e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 210e5c89e4eSSatish Balay #undef __FUNCT__ 211e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead" 212e30d2299SSatish Balay /*@ 213e5c89e4eSSatish Balay PetscBinaryRead - Reads from a binary file. 214e5c89e4eSSatish Balay 215e5c89e4eSSatish Balay Not Collective 216e5c89e4eSSatish Balay 217e5c89e4eSSatish Balay Input Parameters: 218e5c89e4eSSatish Balay + fd - the file 219e5c89e4eSSatish Balay . n - the number of items to read 220e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 221e5c89e4eSSatish Balay 222e5c89e4eSSatish Balay Output Parameters: 223e5c89e4eSSatish Balay . p - the buffer 224e5c89e4eSSatish Balay 225e5c89e4eSSatish Balay 226e5c89e4eSSatish Balay 227e5c89e4eSSatish Balay Level: developer 228e5c89e4eSSatish Balay 229e5c89e4eSSatish Balay Notes: 230e5c89e4eSSatish Balay PetscBinaryRead() uses byte swapping to work on all machines; the files 231e5c89e4eSSatish Balay are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers 232e5c89e4eSSatish Balay are converted to the small-endian format when they are read in from the file. 233e2e64c6bSBarry Smith When PETSc is ./configure with --with-64bit-indices the integers are written to the 23454f21887SBarry Smith file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices 23554f21887SBarry Smith is used. 236e5c89e4eSSatish Balay 237e5c89e4eSSatish Balay Concepts: files^reading binary 238e5c89e4eSSatish Balay Concepts: binary files^reading 239e5c89e4eSSatish Balay 2404ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 2414ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 242e5c89e4eSSatish Balay @*/ 2437087cfbeSBarry Smith PetscErrorCode PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type) 244e5c89e4eSSatish Balay { 245e5c89e4eSSatish Balay int wsize,err; 246e5c89e4eSSatish Balay size_t m = (size_t) n,maxblock = 65536; 247e5c89e4eSSatish Balay char *pp = (char*)p; 2486de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 2496de02169SBarry Smith PetscErrorCode ierr; 250e5c89e4eSSatish Balay void *ptmp = p; 251e5c89e4eSSatish Balay #endif 252e5c89e4eSSatish Balay 253e5c89e4eSSatish Balay PetscFunctionBegin; 254e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 255e5c89e4eSSatish Balay 2566de02169SBarry Smith if (type == PETSC_INT) m *= sizeof(PetscInt); 257e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 258e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 259e95bf02fSSatish Balay else if (type == PETSC_FLOAT) m *= sizeof(float); 260e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 261e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 262e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 263ace3abfcSBarry Smith else if (type == PETSC_BOOL) m *= sizeof(PetscBool); 2649f7b6320SBarry Smith else if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char); 265e32f2f54SBarry Smith else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 266e5c89e4eSSatish Balay 267e5c89e4eSSatish Balay while (m) { 268e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 269e5c89e4eSSatish Balay err = read(fd,pp,wsize); 270e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 271e32f2f54SBarry Smith if (!err && wsize > 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Read past end of file"); 272e32f2f54SBarry Smith if (err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno); 273e5c89e4eSSatish Balay m -= err; 274e5c89e4eSSatish Balay pp += err; 275e5c89e4eSSatish Balay } 276e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 277e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 278e5c89e4eSSatish Balay #endif 279e5c89e4eSSatish Balay 280e5c89e4eSSatish Balay PetscFunctionReturn(0); 281e5c89e4eSSatish Balay } 282e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 283e5c89e4eSSatish Balay #undef __FUNCT__ 284e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite" 285e30d2299SSatish Balay /*@ 286e5c89e4eSSatish Balay PetscBinaryWrite - Writes to a binary file. 287e5c89e4eSSatish Balay 288e5c89e4eSSatish Balay Not Collective 289e5c89e4eSSatish Balay 290e5c89e4eSSatish Balay Input Parameters: 291e5c89e4eSSatish Balay + fd - the file 292e5c89e4eSSatish Balay . p - the buffer 293e5c89e4eSSatish Balay . n - the number of items to write 294e5c89e4eSSatish Balay . type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 295e5c89e4eSSatish Balay - istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise. 296e5c89e4eSSatish Balay 297e5c89e4eSSatish Balay Level: advanced 298e5c89e4eSSatish Balay 299e5c89e4eSSatish Balay Notes: 300e5c89e4eSSatish Balay PetscBinaryWrite() uses byte swapping to work on all machines; the files 301e5c89e4eSSatish Balay are written using big-endian ordering to the file. On small-endian machines the numbers 302e5c89e4eSSatish Balay are converted to the big-endian format when they are written to disk. 303e2e64c6bSBarry Smith When PETSc is ./configure with --with-64bit-indices the integers are written to the 30454f21887SBarry Smith file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices 30554f21887SBarry Smith is used. 306e5c89e4eSSatish Balay 307e5c89e4eSSatish Balay The Buffer p should be read-write buffer, and not static data. 308e5c89e4eSSatish Balay This way, byte-swapping is done in-place, and then the buffer is 309e5c89e4eSSatish Balay written to the file. 310e5c89e4eSSatish Balay 311e5c89e4eSSatish Balay This routine restores the original contents of the buffer, after 312e5c89e4eSSatish Balay it is written to the file. This is done by byte-swapping in-place 313e5c89e4eSSatish Balay the second time. If the flag istemp is set to PETSC_TRUE, the second 314e5c89e4eSSatish Balay byte-swapping operation is not done, thus saving some computation, 315*e5f36e38SBarry Smith but the buffer is left corrupted. 316e5c89e4eSSatish Balay 317300a7f5bSBarry Smith Because byte-swapping may be done on the values in data it cannot be declared const 318300a7f5bSBarry Smith 319e5c89e4eSSatish Balay Concepts: files^writing binary 320e5c89e4eSSatish Balay Concepts: binary files^writing 321e5c89e4eSSatish Balay 3224ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 3234ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 324e5c89e4eSSatish Balay @*/ 3257087cfbeSBarry Smith PetscErrorCode PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscBool istemp) 326e5c89e4eSSatish Balay { 327e5c89e4eSSatish Balay char *pp = (char*)p; 328e5c89e4eSSatish Balay int err,wsize; 329e5c89e4eSSatish Balay size_t m = (size_t)n,maxblock=65536; 3306de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 331e5c89e4eSSatish Balay PetscErrorCode ierr; 332e5c89e4eSSatish Balay void *ptmp = p; 333e5c89e4eSSatish Balay #endif 334e5c89e4eSSatish Balay 335e5c89e4eSSatish Balay PetscFunctionBegin; 336e32f2f54SBarry Smith if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n); 337e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 338e5c89e4eSSatish Balay 3396de02169SBarry Smith if (type == PETSC_INT) m *= sizeof(PetscInt); 340e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 341e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 342e95bf02fSSatish Balay else if (type == PETSC_FLOAT) m *= sizeof(float); 343e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 344e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 345e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 346ace3abfcSBarry Smith else if (type == PETSC_BOOL) m *= sizeof(PetscBool); 3479f7b6320SBarry Smith else if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char); 348e32f2f54SBarry Smith else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 349e5c89e4eSSatish Balay 350e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 351e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 352e5c89e4eSSatish Balay #endif 353e5c89e4eSSatish Balay 354e5c89e4eSSatish Balay while (m) { 355e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 356e5c89e4eSSatish Balay err = write(fd,pp,wsize); 357e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 358e32f2f54SBarry Smith if (err != wsize) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_WRITE,"Error writing to file."); 359e5c89e4eSSatish Balay m -= wsize; 360e5c89e4eSSatish Balay pp += wsize; 361e5c89e4eSSatish Balay } 362e5c89e4eSSatish Balay 363b659c0ceSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 364e5c89e4eSSatish Balay if (!istemp) { 365e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 366e5c89e4eSSatish Balay } 367e5c89e4eSSatish Balay #endif 368e5c89e4eSSatish Balay PetscFunctionReturn(0); 369e5c89e4eSSatish Balay } 370e5c89e4eSSatish Balay 371e5c89e4eSSatish Balay #undef __FUNCT__ 372e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen" 373e5c89e4eSSatish Balay /*@C 374e5c89e4eSSatish Balay PetscBinaryOpen - Opens a PETSc binary file. 375e5c89e4eSSatish Balay 376e5c89e4eSSatish Balay Not Collective 377e5c89e4eSSatish Balay 378e5c89e4eSSatish Balay Input Parameters: 379e5c89e4eSSatish Balay + name - filename 38045c64e65SBarry Smith - type - type of binary file, one of FILE_MODE_READ, FILE_MODE_APPEND, FILE_MODE_WRITE 381e5c89e4eSSatish Balay 382e5c89e4eSSatish Balay Output Parameter: 383e5c89e4eSSatish Balay . fd - the file 384e5c89e4eSSatish Balay 385e5c89e4eSSatish Balay Level: advanced 386e5c89e4eSSatish Balay 387e5c89e4eSSatish Balay Concepts: files^opening binary 388e5c89e4eSSatish Balay Concepts: binary files^opening 389e5c89e4eSSatish Balay 390e5c89e4eSSatish Balay Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in 391e5c89e4eSSatish Balay big-endian format. This means the file can be accessed using PetscBinaryOpen() and 392e5c89e4eSSatish Balay PetscBinaryRead() and PetscBinaryWrite() on any machine. 393e5c89e4eSSatish Balay 3944ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(), 3954ebed01fSBarry Smith PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 39645c64e65SBarry Smith 397e5c89e4eSSatish Balay @*/ 3987087cfbeSBarry Smith PetscErrorCode PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd) 399e5c89e4eSSatish Balay { 400e5c89e4eSSatish Balay PetscFunctionBegin; 401e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY) 40245c64e65SBarry Smith if (mode == FILE_MODE_WRITE) { 403e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) { 404e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 405e5c89e4eSSatish Balay } 40645c64e65SBarry Smith } else if (mode == FILE_MODE_READ) { 407e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) { 408e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 409e5c89e4eSSatish Balay } 41045c64e65SBarry Smith } else if (mode == FILE_MODE_APPEND) { 411e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) { 412e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 413e5c89e4eSSatish Balay } 414e5c89e4eSSatish Balay #else 41545c64e65SBarry Smith if (mode == FILE_MODE_WRITE) { 416e5c89e4eSSatish Balay if ((*fd = creat(name,0666)) == -1) { 417e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 418e5c89e4eSSatish Balay } 41945c64e65SBarry Smith } else if (mode == FILE_MODE_READ) { 420e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY,0)) == -1) { 421e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 422e5c89e4eSSatish Balay } 423e5c89e4eSSatish Balay } 42445c64e65SBarry Smith else if (mode == FILE_MODE_APPEND) { 425e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY,0)) == -1) { 426e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 427e5c89e4eSSatish Balay } 428e5c89e4eSSatish Balay #endif 429e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file mode"); 430e5c89e4eSSatish Balay PetscFunctionReturn(0); 431e5c89e4eSSatish Balay } 432e5c89e4eSSatish Balay 433e5c89e4eSSatish Balay #undef __FUNCT__ 434e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose" 435e30d2299SSatish Balay /*@ 436e5c89e4eSSatish Balay PetscBinaryClose - Closes a PETSc binary file. 437e5c89e4eSSatish Balay 438e5c89e4eSSatish Balay Not Collective 439e5c89e4eSSatish Balay 440e5c89e4eSSatish Balay Output Parameter: 441e5c89e4eSSatish Balay . fd - the file 442e5c89e4eSSatish Balay 443e5c89e4eSSatish Balay Level: advanced 444e5c89e4eSSatish Balay 4454ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 4464ebed01fSBarry Smith PetscBinarySynchronizedSeek() 447e5c89e4eSSatish Balay @*/ 4487087cfbeSBarry Smith PetscErrorCode PetscBinaryClose(int fd) 449e5c89e4eSSatish Balay { 450e5c89e4eSSatish Balay PetscFunctionBegin; 451e5c89e4eSSatish Balay close(fd); 452e5c89e4eSSatish Balay PetscFunctionReturn(0); 453e5c89e4eSSatish Balay } 454e5c89e4eSSatish Balay 455e5c89e4eSSatish Balay 456e5c89e4eSSatish Balay #undef __FUNCT__ 457e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek" 458e30d2299SSatish Balay /*@ 459e5c89e4eSSatish Balay PetscBinarySeek - Moves the file pointer on a PETSc binary file. 460e5c89e4eSSatish Balay 461e5c89e4eSSatish Balay Not Collective 462e5c89e4eSSatish Balay 463e5c89e4eSSatish Balay Input Parameters: 464e5c89e4eSSatish Balay + fd - the file 465ff553b35SMatthew Knepley . off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 466e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 467ff553b35SMatthew Knepley - whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file 468ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_CUR then off is an offset from the current location 469ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_END then off is an offset from the end of file 470e5c89e4eSSatish Balay 471e5c89e4eSSatish Balay Output Parameter: 472e5c89e4eSSatish Balay . offset - new offset in file 473e5c89e4eSSatish Balay 474e5c89e4eSSatish Balay Level: developer 475e5c89e4eSSatish Balay 476e5c89e4eSSatish Balay Notes: 477e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 478e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 479e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 480e5c89e4eSSatish Balay to determine the offset or location. 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay Concepts: files^binary seeking 483e5c89e4eSSatish Balay Concepts: binary files^seeking 484e5c89e4eSSatish Balay 4854ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 4864ebed01fSBarry Smith PetscBinarySynchronizedSeek() 487e5c89e4eSSatish Balay @*/ 4887087cfbeSBarry Smith PetscErrorCode PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 489e5c89e4eSSatish Balay { 490e5c89e4eSSatish Balay int iwhence = 0; 491e5c89e4eSSatish Balay 492e5c89e4eSSatish Balay PetscFunctionBegin; 493e5c89e4eSSatish Balay if (whence == PETSC_BINARY_SEEK_SET) { 494e5c89e4eSSatish Balay iwhence = SEEK_SET; 495e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_CUR) { 496e5c89e4eSSatish Balay iwhence = SEEK_CUR; 497e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_END) { 498e5c89e4eSSatish Balay iwhence = SEEK_END; 499e5c89e4eSSatish Balay } else { 500e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location"); 501e5c89e4eSSatish Balay } 502e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) 503e5c89e4eSSatish Balay *offset = lseek(fd,off,iwhence); 504e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK) 505e5c89e4eSSatish Balay *offset = _lseek(fd,(long)off,iwhence); 506e5c89e4eSSatish Balay #else 507e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file"); 508e5c89e4eSSatish Balay #endif 509e5c89e4eSSatish Balay PetscFunctionReturn(0); 510e5c89e4eSSatish Balay } 511e5c89e4eSSatish Balay 512e5c89e4eSSatish Balay #undef __FUNCT__ 5131d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedRead" 514e5c89e4eSSatish Balay /*@C 5151d280d73SBarry Smith PetscBinarySynchronizedRead - Reads from a binary file. 516e5c89e4eSSatish Balay 517e5c89e4eSSatish Balay Collective on MPI_Comm 518e5c89e4eSSatish Balay 519e5c89e4eSSatish Balay Input Parameters: 520e5c89e4eSSatish Balay + comm - the MPI communicator 521e5c89e4eSSatish Balay . fd - the file 522e5c89e4eSSatish Balay . n - the number of items to read 523e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 524e5c89e4eSSatish Balay 525e5c89e4eSSatish Balay Output Parameters: 526e5c89e4eSSatish Balay . p - the buffer 527e5c89e4eSSatish Balay 528e5c89e4eSSatish Balay Options Database Key: 529e5c89e4eSSatish Balay . -binary_longints - indicates the file was generated on a Cray vector 530e5c89e4eSSatish Balay machine (not the T3E/D) and the ints are stored as 64 bit 531e5c89e4eSSatish Balay quantities, otherwise they are stored as 32 bit 532e5c89e4eSSatish Balay 533e5c89e4eSSatish Balay Level: developer 534e5c89e4eSSatish Balay 535e5c89e4eSSatish Balay Notes: 536e5c89e4eSSatish Balay Does a PetscBinaryRead() followed by an MPI_Bcast() 537e5c89e4eSSatish Balay 5381d280d73SBarry Smith PetscBinarySynchronizedRead() uses byte swapping to work on all machines. 539e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 540e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 541e5c89e4eSSatish Balay binary file may be read on any machine. 542e5c89e4eSSatish Balay 543e5c89e4eSSatish Balay Concepts: files^synchronized reading of binary files 544e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 545e5c89e4eSSatish Balay 5464ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(), 5474ebed01fSBarry Smith PetscBinarySynchronizedSeek() 548e5c89e4eSSatish Balay @*/ 5497087cfbeSBarry Smith PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type) 550e5c89e4eSSatish Balay { 551e5c89e4eSSatish Balay PetscErrorCode ierr; 552e5c89e4eSSatish Balay PetscMPIInt rank; 553e5c89e4eSSatish Balay MPI_Datatype mtype; 554e5c89e4eSSatish Balay 555e5c89e4eSSatish Balay PetscFunctionBegin; 556e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 557e5c89e4eSSatish Balay if (!rank) { 558e5c89e4eSSatish Balay ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr); 559e5c89e4eSSatish Balay } 560e5c89e4eSSatish Balay ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr); 561e5c89e4eSSatish Balay ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr); 562e5c89e4eSSatish Balay PetscFunctionReturn(0); 563e5c89e4eSSatish Balay } 564e5c89e4eSSatish Balay 565e5c89e4eSSatish Balay #undef __FUNCT__ 5661d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedWrite" 567e5c89e4eSSatish Balay /*@C 5681d280d73SBarry Smith PetscBinarySynchronizedWrite - writes to a binary file. 569e5c89e4eSSatish Balay 570e5c89e4eSSatish Balay Collective on MPI_Comm 571e5c89e4eSSatish Balay 572e5c89e4eSSatish Balay Input Parameters: 573e5c89e4eSSatish Balay + comm - the MPI communicator 574e5c89e4eSSatish Balay . fd - the file 575e5c89e4eSSatish Balay . n - the number of items to write 576e5c89e4eSSatish Balay . p - the buffer 577e5c89e4eSSatish Balay . istemp - the buffer may be changed 578e5c89e4eSSatish Balay - type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 579e5c89e4eSSatish Balay 580e5c89e4eSSatish Balay Level: developer 581e5c89e4eSSatish Balay 582e5c89e4eSSatish Balay Notes: 583e5c89e4eSSatish Balay Process 0 does a PetscBinaryWrite() 584e5c89e4eSSatish Balay 5851d280d73SBarry Smith PetscBinarySynchronizedWrite() uses byte swapping to work on all machines. 586e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 587e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 588e5c89e4eSSatish Balay binary file may be read on any machine. 589e5c89e4eSSatish Balay 590300a7f5bSBarry Smith Notes: because byte-swapping may be done on the values in data it cannot be declared const 591300a7f5bSBarry Smith 5921d280d73SBarry Smith WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0, 5931d280d73SBarry Smith while PetscSynchronizedFPrintf() has all processes print their strings in order. 5941d280d73SBarry Smith 595e5c89e4eSSatish Balay Concepts: files^synchronized writing of binary files 596e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 597e5c89e4eSSatish Balay 5984ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(), 5994ebed01fSBarry Smith PetscBinarySynchronizedSeek() 600e5c89e4eSSatish Balay @*/ 6017087cfbeSBarry Smith PetscErrorCode PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscBool istemp) 602e5c89e4eSSatish Balay { 603e5c89e4eSSatish Balay PetscErrorCode ierr; 604e5c89e4eSSatish Balay PetscMPIInt rank; 605e5c89e4eSSatish Balay 606e5c89e4eSSatish Balay PetscFunctionBegin; 607e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 608e5c89e4eSSatish Balay if (!rank) { 609e5c89e4eSSatish Balay ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr); 610e5c89e4eSSatish Balay } 611e5c89e4eSSatish Balay PetscFunctionReturn(0); 612e5c89e4eSSatish Balay } 613e5c89e4eSSatish Balay 614e5c89e4eSSatish Balay #undef __FUNCT__ 6151d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedSeek" 616e5c89e4eSSatish Balay /*@C 6171d280d73SBarry Smith PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file. 618e5c89e4eSSatish Balay 619e5c89e4eSSatish Balay 620e5c89e4eSSatish Balay Input Parameters: 621e5c89e4eSSatish Balay + fd - the file 622e5c89e4eSSatish Balay . whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file 623e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_CUR then size is offset from current location 624e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_END then size is offset from end of file 625e5c89e4eSSatish Balay - off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 626e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 627e5c89e4eSSatish Balay 628e5c89e4eSSatish Balay Output Parameter: 629e5c89e4eSSatish Balay . offset - new offset in file 630e5c89e4eSSatish Balay 631e5c89e4eSSatish Balay Level: developer 632e5c89e4eSSatish Balay 633e5c89e4eSSatish Balay Notes: 634e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 635e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 636e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 637e5c89e4eSSatish Balay to determine the offset or location. 638e5c89e4eSSatish Balay 639e5c89e4eSSatish Balay Concepts: binary files^seeking 640e5c89e4eSSatish Balay Concepts: files^seeking in binary 641e5c89e4eSSatish Balay 6424ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 6434ebed01fSBarry Smith PetscBinarySynchronizedSeek() 644e5c89e4eSSatish Balay @*/ 6457087cfbeSBarry Smith PetscErrorCode PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 646e5c89e4eSSatish Balay { 647e5c89e4eSSatish Balay PetscErrorCode ierr; 648e5c89e4eSSatish Balay PetscMPIInt rank; 649e5c89e4eSSatish Balay 650e5c89e4eSSatish Balay PetscFunctionBegin; 651e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 652e5c89e4eSSatish Balay if (!rank) { 653e5c89e4eSSatish Balay ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr); 654e5c89e4eSSatish Balay } 655e5c89e4eSSatish Balay PetscFunctionReturn(0); 656e5c89e4eSSatish Balay } 657e5c89e4eSSatish Balay 6580fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO) 659e39fd77fSBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 660e39fd77fSBarry Smith 661951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32) 662e39fd77fSBarry Smith EXTERN_C_BEGIN 663e39fd77fSBarry Smith /* 664e39fd77fSBarry Smith MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions. 665e39fd77fSBarry Smith These are set into MPI in PetscInitialize() via MPI_Register_datarep() 666e39fd77fSBarry Smith 667e39fd77fSBarry Smith Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode) 668e39fd77fSBarry Smith 669951e3c8eSBarry Smith The next three routines are not used because MPICH does not support their use 670e39fd77fSBarry Smith 671e39fd77fSBarry Smith */ 672e39fd77fSBarry Smith PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state) 673e39fd77fSBarry Smith { 674e39fd77fSBarry Smith MPI_Aint ub; 675e39fd77fSBarry Smith PetscMPIInt ierr; 676e39fd77fSBarry Smith 677e39fd77fSBarry Smith ierr = MPI_Type_get_extent(datatype,&ub,file_extent); 678e39fd77fSBarry Smith return ierr; 679e39fd77fSBarry Smith } 680e39fd77fSBarry Smith 681e39fd77fSBarry Smith PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 682e39fd77fSBarry Smith { 683e39fd77fSBarry Smith PetscDataType pdtype; 684e39fd77fSBarry Smith PetscMPIInt ierr; 685e39fd77fSBarry Smith size_t dsize; 686e39fd77fSBarry Smith 687e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 688e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 689e39fd77fSBarry Smith 690e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 691e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize*position; 692e39fd77fSBarry Smith 693e39fd77fSBarry Smith ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr); 694e39fd77fSBarry Smith ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr); 695e39fd77fSBarry Smith return ierr; 696e39fd77fSBarry Smith } 697e39fd77fSBarry Smith 698e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 699e39fd77fSBarry Smith { 700e39fd77fSBarry Smith PetscDataType pdtype; 701e39fd77fSBarry Smith PetscMPIInt ierr; 702e39fd77fSBarry Smith size_t dsize; 703e39fd77fSBarry Smith 704e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 705e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 706e39fd77fSBarry Smith 707e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 708e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize*position; 709e39fd77fSBarry Smith 710e39fd77fSBarry Smith ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr); 711e39fd77fSBarry Smith ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr); 712e39fd77fSBarry Smith return ierr; 713e39fd77fSBarry Smith } 714e39fd77fSBarry Smith EXTERN_C_END 715951e3c8eSBarry Smith #endif 716e39fd77fSBarry Smith 7171c91e6bcSJed Brown #undef __FUNCT__ 7181c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_write_all" 719e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 720e39fd77fSBarry Smith { 721e39fd77fSBarry Smith PetscErrorCode ierr; 722e39fd77fSBarry Smith PetscDataType pdtype; 723e39fd77fSBarry Smith 724e39fd77fSBarry Smith PetscFunctionBegin; 725e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 726e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 727e39fd77fSBarry Smith ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 728e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 729a6796414SBarry Smith PetscFunctionReturn(0); 730e39fd77fSBarry Smith } 731e39fd77fSBarry Smith 7321c91e6bcSJed Brown #undef __FUNCT__ 7331c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_read_all" 734e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 735e39fd77fSBarry Smith { 736e39fd77fSBarry Smith PetscErrorCode ierr; 737e39fd77fSBarry Smith PetscDataType pdtype; 738e39fd77fSBarry Smith 739e39fd77fSBarry Smith PetscFunctionBegin; 740e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 741e39fd77fSBarry Smith ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 742e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 743a6796414SBarry Smith PetscFunctionReturn(0); 744e39fd77fSBarry Smith } 745e39fd77fSBarry Smith #endif 746951e3c8eSBarry Smith #endif 747