1e5c89e4eSSatish Balay #define PETSC_DLL 2e5c89e4eSSatish Balay 3e5c89e4eSSatish Balay /* 4e5c89e4eSSatish Balay This file contains simple binary read/write routines. 5e5c89e4eSSatish Balay */ 6e5c89e4eSSatish Balay 7e30d2299SSatish Balay #include "petsc.h" 8e30d2299SSatish Balay #include "petscsys.h" /*I "petscsys.h" I*/ 9e5c89e4eSSatish Balay 10e5c89e4eSSatish Balay #include <errno.h> 11e5c89e4eSSatish Balay #include <fcntl.h> 12e5c89e4eSSatish Balay #if defined(PETSC_HAVE_UNISTD_H) 13e5c89e4eSSatish Balay #include <unistd.h> 14e5c89e4eSSatish Balay #endif 15e5c89e4eSSatish Balay #if defined (PETSC_HAVE_IO_H) 16e5c89e4eSSatish Balay #include <io.h> 17e5c89e4eSSatish Balay #endif 18e5c89e4eSSatish Balay #include "petscbt.h" 19e5c89e4eSSatish Balay 20e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 21e5c89e4eSSatish Balay 226de02169SBarry Smith /* --------------------------------------------------------- */ 23e5c89e4eSSatish Balay #undef __FUNCT__ 246de02169SBarry Smith #define __FUNCT__ "PetscByteSwapEnum" 25e5c89e4eSSatish Balay /* 266de02169SBarry Smith PetscByteSwapEnum - Swap bytes in a PETSc Enum 27e5c89e4eSSatish Balay 28e5c89e4eSSatish Balay */ 296de02169SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscByteSwapEnum(PetscEnum *buff,PetscInt n) 30e5c89e4eSSatish Balay { 316de02169SBarry Smith PetscInt i,j; 320b20345dSBarry Smith PetscEnum tmp = ENUM_DUMMY; 336de02169SBarry Smith PetscEnum *tptr = &tmp; /* Need to access tmp indirectly to get */ 34e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; /* arround the bug in DEC-ALPHA g++ */ 35e5c89e4eSSatish Balay 36e5c89e4eSSatish Balay PetscFunctionBegin; 37e5c89e4eSSatish Balay for (j=0; j<n; j++) { 38e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 390b20345dSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) { 406de02169SBarry Smith ptr2[i] = ptr1[sizeof(PetscEnum)-1-i]; 41e5c89e4eSSatish Balay } 42e5c89e4eSSatish Balay buff[j] = *tptr; 43e5c89e4eSSatish Balay } 44e5c89e4eSSatish Balay PetscFunctionReturn(0); 45e5c89e4eSSatish Balay } 466de02169SBarry Smith 476de02169SBarry Smith #undef __FUNCT__ 486de02169SBarry Smith #define __FUNCT__ "PetscByteSwapTruth" 496de02169SBarry Smith /* 506de02169SBarry Smith PetscByteSwapTruth - Swap bytes in a PETSc Truth 516de02169SBarry Smith 526de02169SBarry Smith */ 536de02169SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscByteSwapTruth(PetscTruth *buff,PetscInt n) 546de02169SBarry Smith { 556de02169SBarry Smith PetscInt i,j; 560b20345dSBarry Smith PetscTruth tmp = PETSC_FALSE; 576de02169SBarry Smith PetscTruth *tptr = &tmp; /* Need to access tmp indirectly to get */ 586de02169SBarry Smith char *ptr1,*ptr2 = (char*)&tmp; /* arround the bug in DEC-ALPHA g++ */ 596de02169SBarry Smith 606de02169SBarry Smith PetscFunctionBegin; 616de02169SBarry Smith for (j=0; j<n; j++) { 626de02169SBarry Smith ptr1 = (char*)(buff + j); 630b20345dSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscTruth); i++) { 646de02169SBarry Smith ptr2[i] = ptr1[sizeof(PetscTruth)-1-i]; 656de02169SBarry Smith } 666de02169SBarry Smith buff[j] = *tptr; 676de02169SBarry Smith } 686de02169SBarry Smith PetscFunctionReturn(0); 696de02169SBarry Smith } 706de02169SBarry Smith 71e5c89e4eSSatish Balay #undef __FUNCT__ 72bd1d2e58SBarry Smith #define __FUNCT__ "PetscByteSwapInt" 73bd1d2e58SBarry Smith /* 746de02169SBarry Smith PetscByteSwapInt - Swap bytes in a PETSc integer (which may be 32 or 64 bits) 75bd1d2e58SBarry Smith 76bd1d2e58SBarry Smith */ 776de02169SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscByteSwapInt(PetscInt *buff,PetscInt n) 78bd1d2e58SBarry Smith { 79bd1d2e58SBarry Smith PetscInt i,j,tmp = 0; 80bd1d2e58SBarry Smith PetscInt *tptr = &tmp; /* Need to access tmp indirectly to get */ 81bd1d2e58SBarry Smith char *ptr1,*ptr2 = (char*)&tmp; /* arround the bug in DEC-ALPHA g++ */ 82bd1d2e58SBarry Smith 83bd1d2e58SBarry Smith PetscFunctionBegin; 84bd1d2e58SBarry Smith for (j=0; j<n; j++) { 85bd1d2e58SBarry Smith ptr1 = (char*)(buff + j); 860b20345dSBarry Smith for (i=0; i<(PetscInt)sizeof(PetscInt); i++) { 87bd1d2e58SBarry Smith ptr2[i] = ptr1[sizeof(PetscInt)-1-i]; 88bd1d2e58SBarry Smith } 89bd1d2e58SBarry Smith buff[j] = *tptr; 90bd1d2e58SBarry Smith } 91bd1d2e58SBarry Smith PetscFunctionReturn(0); 92bd1d2e58SBarry Smith } 93bd1d2e58SBarry Smith /* --------------------------------------------------------- */ 94bd1d2e58SBarry Smith #undef __FUNCT__ 95e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort" 96e5c89e4eSSatish Balay /* 97e5c89e4eSSatish Balay PetscByteSwapShort - Swap bytes in a short 98e5c89e4eSSatish Balay */ 99e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapShort(short *buff,PetscInt n) 100e5c89e4eSSatish Balay { 101e5c89e4eSSatish Balay PetscInt i,j; 102e5c89e4eSSatish Balay short tmp; 103e5c89e4eSSatish Balay short *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 104e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 105e5c89e4eSSatish Balay 106e5c89e4eSSatish Balay PetscFunctionBegin; 107e5c89e4eSSatish Balay for (j=0; j<n; j++) { 108e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 1090b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(short); i++) { 110e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(int)-1-i]; 111e5c89e4eSSatish Balay } 112e5c89e4eSSatish Balay buff[j] = *tptr; 113e5c89e4eSSatish Balay } 114e5c89e4eSSatish Balay PetscFunctionReturn(0); 115e5c89e4eSSatish Balay } 116e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 117e5c89e4eSSatish Balay #undef __FUNCT__ 118e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar" 119e5c89e4eSSatish Balay /* 120e5c89e4eSSatish Balay PetscByteSwapScalar - Swap bytes in a double 121e5c89e4eSSatish Balay Complex is dealt with as if array of double twice as long. 122e5c89e4eSSatish Balay */ 123e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapScalar(PetscScalar *buff,PetscInt n) 124e5c89e4eSSatish Balay { 125e5c89e4eSSatish Balay PetscInt i,j; 126e5c89e4eSSatish Balay PetscReal tmp,*buff1 = (PetscReal*)buff; 127e5c89e4eSSatish Balay PetscReal *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 128e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 129e5c89e4eSSatish Balay 130e5c89e4eSSatish Balay PetscFunctionBegin; 131e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 132e5c89e4eSSatish Balay n *= 2; 133e5c89e4eSSatish Balay #endif 134e5c89e4eSSatish Balay for (j=0; j<n; j++) { 135e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 1360b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(PetscReal); i++) { 137e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(PetscReal)-1-i]; 138e5c89e4eSSatish Balay } 139e5c89e4eSSatish Balay buff1[j] = *tptr; 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 */ 149e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapDouble(double *buff,PetscInt n) 150e5c89e4eSSatish Balay { 151e5c89e4eSSatish Balay PetscInt i,j; 152e5c89e4eSSatish Balay double tmp,*buff1 = (double*)buff; 153e5c89e4eSSatish Balay double *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 154e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 155e5c89e4eSSatish Balay 156e5c89e4eSSatish Balay PetscFunctionBegin; 157e5c89e4eSSatish Balay for (j=0; j<n; j++) { 158e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 1590b20345dSBarry Smith for (i=0; i<(PetscInt) sizeof(double); i++) { 160e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(double)-1-i]; 161e5c89e4eSSatish Balay } 162e5c89e4eSSatish Balay buff1[j] = *tptr; 163e5c89e4eSSatish Balay } 164e5c89e4eSSatish Balay PetscFunctionReturn(0); 165e5c89e4eSSatish Balay } 166*e39fd77fSBarry Smith 167*e39fd77fSBarry Smith #undef __FUNCT__ 168*e39fd77fSBarry Smith #define __FUNCT__ "PetscByteSwap" 169*e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count) 170*e39fd77fSBarry Smith { 171*e39fd77fSBarry Smith PetscErrorCode ierr; 172*e39fd77fSBarry Smith 173*e39fd77fSBarry Smith PetscFunctionBegin; 174*e39fd77fSBarry Smith if (pdtype == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);} 175*e39fd77fSBarry Smith else if (pdtype == PETSC_ENUM) {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);} 176*e39fd77fSBarry Smith else if (pdtype == PETSC_TRUTH) {ierr = PetscByteSwapTruth((PetscTruth*)data,count);CHKERRQ(ierr);} 177*e39fd77fSBarry Smith else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);} 178*e39fd77fSBarry Smith else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);} 179*e39fd77fSBarry Smith else if (pdtype == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);} 180*e39fd77fSBarry Smith PetscFunctionReturn(0); 181*e39fd77fSBarry Smith } 182*e39fd77fSBarry Smith 183e5c89e4eSSatish Balay #endif 184e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 185e5c89e4eSSatish Balay #undef __FUNCT__ 186e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead" 187e30d2299SSatish Balay /*@ 188e5c89e4eSSatish Balay PetscBinaryRead - Reads from a binary file. 189e5c89e4eSSatish Balay 190e5c89e4eSSatish Balay Not Collective 191e5c89e4eSSatish Balay 192e5c89e4eSSatish Balay Input Parameters: 193e5c89e4eSSatish Balay + fd - the file 194e5c89e4eSSatish Balay . n - the number of items to read 195e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 196e5c89e4eSSatish Balay 197e5c89e4eSSatish Balay Output Parameters: 198e5c89e4eSSatish Balay . p - the buffer 199e5c89e4eSSatish Balay 200e5c89e4eSSatish Balay 201e5c89e4eSSatish Balay 202e5c89e4eSSatish Balay Level: developer 203e5c89e4eSSatish Balay 204e5c89e4eSSatish Balay Notes: 205e5c89e4eSSatish Balay PetscBinaryRead() uses byte swapping to work on all machines; the files 206e5c89e4eSSatish Balay are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers 207e5c89e4eSSatish Balay are converted to the small-endian format when they are read in from the file. 20854f21887SBarry Smith When PETSc is config/configure.py with --with-64bit-indices the integers are written to the 20954f21887SBarry Smith file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices 21054f21887SBarry Smith is used. 211e5c89e4eSSatish Balay 212e5c89e4eSSatish Balay Concepts: files^reading binary 213e5c89e4eSSatish Balay Concepts: binary files^reading 214e5c89e4eSSatish Balay 2154ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 2164ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 217e5c89e4eSSatish Balay @*/ 218e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type) 219e5c89e4eSSatish Balay { 220e5c89e4eSSatish Balay int wsize,err; 221e5c89e4eSSatish Balay size_t m = (size_t) n,maxblock = 65536; 222e5c89e4eSSatish Balay char *pp = (char*)p; 2236de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 2246de02169SBarry Smith PetscErrorCode ierr; 225e5c89e4eSSatish Balay void *ptmp = p; 226e5c89e4eSSatish Balay #endif 227e5c89e4eSSatish Balay 228e5c89e4eSSatish Balay PetscFunctionBegin; 229e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 230e5c89e4eSSatish Balay 2316de02169SBarry Smith if (type == PETSC_INT) m *= sizeof(PetscInt); 232e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 233e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 234e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 235e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 236e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 237e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) m *= sizeof(PetscTruth); 238e5c89e4eSSatish Balay else if (type == PETSC_LOGICAL) m = PetscBTLength(m)*sizeof(char); 239e5c89e4eSSatish Balay else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 240e5c89e4eSSatish Balay 241e5c89e4eSSatish Balay while (m) { 242e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 243e5c89e4eSSatish Balay err = read(fd,pp,wsize); 244e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 245e5c89e4eSSatish Balay if (!err && wsize > 0) SETERRQ(PETSC_ERR_FILE_READ,"Read past end of file"); 246a1f04975SBarry Smith if (err < 0) SETERRQ1(PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno); 247e5c89e4eSSatish Balay m -= err; 248e5c89e4eSSatish Balay pp += err; 249e5c89e4eSSatish Balay } 250e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 251*e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 252e5c89e4eSSatish Balay #endif 253e5c89e4eSSatish Balay 254e5c89e4eSSatish Balay PetscFunctionReturn(0); 255e5c89e4eSSatish Balay } 256e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 257e5c89e4eSSatish Balay #undef __FUNCT__ 258e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite" 259e30d2299SSatish Balay /*@ 260e5c89e4eSSatish Balay PetscBinaryWrite - Writes to a binary file. 261e5c89e4eSSatish Balay 262e5c89e4eSSatish Balay Not Collective 263e5c89e4eSSatish Balay 264e5c89e4eSSatish Balay Input Parameters: 265e5c89e4eSSatish Balay + fd - the file 266e5c89e4eSSatish Balay . p - the buffer 267e5c89e4eSSatish Balay . n - the number of items to write 268e5c89e4eSSatish Balay . type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 269e5c89e4eSSatish Balay - istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise. 270e5c89e4eSSatish Balay 271e5c89e4eSSatish Balay Level: advanced 272e5c89e4eSSatish Balay 273e5c89e4eSSatish Balay Notes: 274e5c89e4eSSatish Balay PetscBinaryWrite() uses byte swapping to work on all machines; the files 275e5c89e4eSSatish Balay are written using big-endian ordering to the file. On small-endian machines the numbers 276e5c89e4eSSatish Balay are converted to the big-endian format when they are written to disk. 27754f21887SBarry Smith When PETSc is config/configure.py with --with-64bit-indices the integers are written to the 27854f21887SBarry Smith file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices 27954f21887SBarry Smith is used. 280e5c89e4eSSatish Balay 281e5c89e4eSSatish Balay The Buffer p should be read-write buffer, and not static data. 282e5c89e4eSSatish Balay This way, byte-swapping is done in-place, and then the buffer is 283e5c89e4eSSatish Balay written to the file. 284e5c89e4eSSatish Balay 285e5c89e4eSSatish Balay This routine restores the original contents of the buffer, after 286e5c89e4eSSatish Balay it is written to the file. This is done by byte-swapping in-place 287e5c89e4eSSatish Balay the second time. If the flag istemp is set to PETSC_TRUE, the second 288e5c89e4eSSatish Balay byte-swapping operation is not done, thus saving some computation, 289e5c89e4eSSatish Balay but the buffer corrupted is corrupted. 290e5c89e4eSSatish Balay 291e5c89e4eSSatish Balay Concepts: files^writing binary 292e5c89e4eSSatish Balay Concepts: binary files^writing 293e5c89e4eSSatish Balay 2944ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 2954ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 296e5c89e4eSSatish Balay @*/ 297e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp) 298e5c89e4eSSatish Balay { 299e5c89e4eSSatish Balay char *pp = (char*)p; 300e5c89e4eSSatish Balay int err,wsize; 301e5c89e4eSSatish Balay size_t m = (size_t)n,maxblock=65536; 3026de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 303e5c89e4eSSatish Balay PetscErrorCode ierr; 304e5c89e4eSSatish Balay void *ptmp = p; 305e5c89e4eSSatish Balay #endif 306e5c89e4eSSatish Balay 307e5c89e4eSSatish Balay PetscFunctionBegin; 308e5c89e4eSSatish Balay if (n < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n); 309e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 310e5c89e4eSSatish Balay 3116de02169SBarry Smith if (type == PETSC_INT) m *= sizeof(PetscInt); 312e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 313e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 314e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 315e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 316e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 317e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) m *= sizeof(PetscTruth); 318e5c89e4eSSatish Balay else if (type == PETSC_LOGICAL) m = PetscBTLength(m)*sizeof(char); 319e5c89e4eSSatish Balay else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 320e5c89e4eSSatish Balay 321e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 322*e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 323e5c89e4eSSatish Balay #endif 324e5c89e4eSSatish Balay 325e5c89e4eSSatish Balay while (m) { 326e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 327e5c89e4eSSatish Balay err = write(fd,pp,wsize); 328e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 329e5c89e4eSSatish Balay if (err != wsize) SETERRQ(PETSC_ERR_FILE_WRITE,"Error writing to file."); 330e5c89e4eSSatish Balay m -= wsize; 331e5c89e4eSSatish Balay pp += wsize; 332e5c89e4eSSatish Balay } 333e5c89e4eSSatish Balay 334b659c0ceSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 335e5c89e4eSSatish Balay if (!istemp) { 336*e39fd77fSBarry Smith ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr); 337e5c89e4eSSatish Balay } 338e5c89e4eSSatish Balay #endif 339e5c89e4eSSatish Balay PetscFunctionReturn(0); 340e5c89e4eSSatish Balay } 341e5c89e4eSSatish Balay 342e5c89e4eSSatish Balay #undef __FUNCT__ 343e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen" 344e5c89e4eSSatish Balay /*@C 345e5c89e4eSSatish Balay PetscBinaryOpen - Opens a PETSc binary file. 346e5c89e4eSSatish Balay 347e5c89e4eSSatish Balay Not Collective 348e5c89e4eSSatish Balay 349e5c89e4eSSatish Balay Input Parameters: 350e5c89e4eSSatish Balay + name - filename 35145c64e65SBarry Smith - type - type of binary file, one of FILE_MODE_READ, FILE_MODE_APPEND, FILE_MODE_WRITE 352e5c89e4eSSatish Balay 353e5c89e4eSSatish Balay Output Parameter: 354e5c89e4eSSatish Balay . fd - the file 355e5c89e4eSSatish Balay 356e5c89e4eSSatish Balay Level: advanced 357e5c89e4eSSatish Balay 358e5c89e4eSSatish Balay Concepts: files^opening binary 359e5c89e4eSSatish Balay Concepts: binary files^opening 360e5c89e4eSSatish Balay 361e5c89e4eSSatish Balay Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in 362e5c89e4eSSatish Balay big-endian format. This means the file can be accessed using PetscBinaryOpen() and 363e5c89e4eSSatish Balay PetscBinaryRead() and PetscBinaryWrite() on any machine. 364e5c89e4eSSatish Balay 3654ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(), 3664ebed01fSBarry Smith PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 36745c64e65SBarry Smith 368e5c89e4eSSatish Balay @*/ 36945c64e65SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd) 370e5c89e4eSSatish Balay { 371e5c89e4eSSatish Balay PetscFunctionBegin; 372e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY) 37345c64e65SBarry Smith if (mode == FILE_MODE_WRITE) { 374e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) { 375e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 376e5c89e4eSSatish Balay } 37745c64e65SBarry Smith } else if (mode == FILE_MODE_READ) { 378e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) { 379e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 380e5c89e4eSSatish Balay } 38145c64e65SBarry Smith } else if (mode == FILE_MODE_APPEND) { 382e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) { 383e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 384e5c89e4eSSatish Balay } 385e5c89e4eSSatish Balay #else 38645c64e65SBarry Smith if (mode == FILE_MODE_WRITE) { 387e5c89e4eSSatish Balay if ((*fd = creat(name,0666)) == -1) { 388e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 389e5c89e4eSSatish Balay } 39045c64e65SBarry Smith } else if (mode == FILE_MODE_READ) { 391e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY,0)) == -1) { 392e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 393e5c89e4eSSatish Balay } 394e5c89e4eSSatish Balay } 39545c64e65SBarry Smith else if (mode == FILE_MODE_APPEND) { 396e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY,0)) == -1) { 397e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 398e5c89e4eSSatish Balay } 399e5c89e4eSSatish Balay #endif 40045c64e65SBarry Smith } else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown file mode"); 401e5c89e4eSSatish Balay PetscFunctionReturn(0); 402e5c89e4eSSatish Balay } 403e5c89e4eSSatish Balay 404e5c89e4eSSatish Balay #undef __FUNCT__ 405e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose" 406e30d2299SSatish Balay /*@ 407e5c89e4eSSatish Balay PetscBinaryClose - Closes a PETSc binary file. 408e5c89e4eSSatish Balay 409e5c89e4eSSatish Balay Not Collective 410e5c89e4eSSatish Balay 411e5c89e4eSSatish Balay Output Parameter: 412e5c89e4eSSatish Balay . fd - the file 413e5c89e4eSSatish Balay 414e5c89e4eSSatish Balay Level: advanced 415e5c89e4eSSatish Balay 4164ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 4174ebed01fSBarry Smith PetscBinarySynchronizedSeek() 418e5c89e4eSSatish Balay @*/ 419e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryClose(int fd) 420e5c89e4eSSatish Balay { 421e5c89e4eSSatish Balay PetscFunctionBegin; 422e5c89e4eSSatish Balay close(fd); 423e5c89e4eSSatish Balay PetscFunctionReturn(0); 424e5c89e4eSSatish Balay } 425e5c89e4eSSatish Balay 426e5c89e4eSSatish Balay 427e5c89e4eSSatish Balay #undef __FUNCT__ 428e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek" 429e30d2299SSatish Balay /*@ 430e5c89e4eSSatish Balay PetscBinarySeek - Moves the file pointer on a PETSc binary file. 431e5c89e4eSSatish Balay 432e5c89e4eSSatish Balay Not Collective 433e5c89e4eSSatish Balay 434e5c89e4eSSatish Balay Input Parameters: 435e5c89e4eSSatish Balay + fd - the file 436ff553b35SMatthew Knepley . off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 437e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 438ff553b35SMatthew Knepley - whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file 439ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_CUR then off is an offset from the current location 440ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_END then off is an offset from the end of file 441e5c89e4eSSatish Balay 442e5c89e4eSSatish Balay Output Parameter: 443e5c89e4eSSatish Balay . offset - new offset in file 444e5c89e4eSSatish Balay 445e5c89e4eSSatish Balay Level: developer 446e5c89e4eSSatish Balay 447e5c89e4eSSatish Balay Notes: 448e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 449e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 450e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 451e5c89e4eSSatish Balay to determine the offset or location. 452e5c89e4eSSatish Balay 453e5c89e4eSSatish Balay Concepts: files^binary seeking 454e5c89e4eSSatish Balay Concepts: binary files^seeking 455e5c89e4eSSatish Balay 4564ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 4574ebed01fSBarry Smith PetscBinarySynchronizedSeek() 458e5c89e4eSSatish Balay @*/ 459e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 460e5c89e4eSSatish Balay { 461e5c89e4eSSatish Balay int iwhence = 0; 462e5c89e4eSSatish Balay 463e5c89e4eSSatish Balay PetscFunctionBegin; 464e5c89e4eSSatish Balay if (whence == PETSC_BINARY_SEEK_SET) { 465e5c89e4eSSatish Balay iwhence = SEEK_SET; 466e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_CUR) { 467e5c89e4eSSatish Balay iwhence = SEEK_CUR; 468e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_END) { 469e5c89e4eSSatish Balay iwhence = SEEK_END; 470e5c89e4eSSatish Balay } else { 471e5c89e4eSSatish Balay SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location"); 472e5c89e4eSSatish Balay } 473e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) 474e5c89e4eSSatish Balay *offset = lseek(fd,off,iwhence); 475e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK) 476e5c89e4eSSatish Balay *offset = _lseek(fd,(long)off,iwhence); 477e5c89e4eSSatish Balay #else 478e5c89e4eSSatish Balay SETERRQ(PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file"); 479e5c89e4eSSatish Balay #endif 480e5c89e4eSSatish Balay PetscFunctionReturn(0); 481e5c89e4eSSatish Balay } 482e5c89e4eSSatish Balay 483e5c89e4eSSatish Balay #undef __FUNCT__ 4841d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedRead" 485e5c89e4eSSatish Balay /*@C 4861d280d73SBarry Smith PetscBinarySynchronizedRead - Reads from a binary file. 487e5c89e4eSSatish Balay 488e5c89e4eSSatish Balay Collective on MPI_Comm 489e5c89e4eSSatish Balay 490e5c89e4eSSatish Balay Input Parameters: 491e5c89e4eSSatish Balay + comm - the MPI communicator 492e5c89e4eSSatish Balay . fd - the file 493e5c89e4eSSatish Balay . n - the number of items to read 494e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 495e5c89e4eSSatish Balay 496e5c89e4eSSatish Balay Output Parameters: 497e5c89e4eSSatish Balay . p - the buffer 498e5c89e4eSSatish Balay 499e5c89e4eSSatish Balay Options Database Key: 500e5c89e4eSSatish Balay . -binary_longints - indicates the file was generated on a Cray vector 501e5c89e4eSSatish Balay machine (not the T3E/D) and the ints are stored as 64 bit 502e5c89e4eSSatish Balay quantities, otherwise they are stored as 32 bit 503e5c89e4eSSatish Balay 504e5c89e4eSSatish Balay Level: developer 505e5c89e4eSSatish Balay 506e5c89e4eSSatish Balay Notes: 507e5c89e4eSSatish Balay Does a PetscBinaryRead() followed by an MPI_Bcast() 508e5c89e4eSSatish Balay 5091d280d73SBarry Smith PetscBinarySynchronizedRead() uses byte swapping to work on all machines. 510e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 511e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 512e5c89e4eSSatish Balay binary file may be read on any machine. 513e5c89e4eSSatish Balay 514e5c89e4eSSatish Balay Concepts: files^synchronized reading of binary files 515e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 516e5c89e4eSSatish Balay 5174ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(), 5184ebed01fSBarry Smith PetscBinarySynchronizedSeek() 519e5c89e4eSSatish Balay @*/ 5201d280d73SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type) 521e5c89e4eSSatish Balay { 522e5c89e4eSSatish Balay PetscErrorCode ierr; 523e5c89e4eSSatish Balay PetscMPIInt rank; 524e5c89e4eSSatish Balay MPI_Datatype mtype; 525e5c89e4eSSatish Balay 526e5c89e4eSSatish Balay PetscFunctionBegin; 527e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 528e5c89e4eSSatish Balay if (!rank) { 529e5c89e4eSSatish Balay ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr); 530e5c89e4eSSatish Balay } 531e5c89e4eSSatish Balay ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr); 532e5c89e4eSSatish Balay ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr); 533e5c89e4eSSatish Balay PetscFunctionReturn(0); 534e5c89e4eSSatish Balay } 535e5c89e4eSSatish Balay 536e5c89e4eSSatish Balay #undef __FUNCT__ 5371d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedWrite" 538e5c89e4eSSatish Balay /*@C 5391d280d73SBarry Smith PetscBinarySynchronizedWrite - writes to a binary file. 540e5c89e4eSSatish Balay 541e5c89e4eSSatish Balay Collective on MPI_Comm 542e5c89e4eSSatish Balay 543e5c89e4eSSatish Balay Input Parameters: 544e5c89e4eSSatish Balay + comm - the MPI communicator 545e5c89e4eSSatish Balay . fd - the file 546e5c89e4eSSatish Balay . n - the number of items to write 547e5c89e4eSSatish Balay . p - the buffer 548e5c89e4eSSatish Balay . istemp - the buffer may be changed 549e5c89e4eSSatish Balay - type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 550e5c89e4eSSatish Balay 551e5c89e4eSSatish Balay Level: developer 552e5c89e4eSSatish Balay 553e5c89e4eSSatish Balay Notes: 554e5c89e4eSSatish Balay Process 0 does a PetscBinaryWrite() 555e5c89e4eSSatish Balay 5561d280d73SBarry Smith PetscBinarySynchronizedWrite() uses byte swapping to work on all machines. 557e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 558e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 559e5c89e4eSSatish Balay binary file may be read on any machine. 560e5c89e4eSSatish Balay 5611d280d73SBarry Smith WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0, 5621d280d73SBarry Smith while PetscSynchronizedFPrintf() has all processes print their strings in order. 5631d280d73SBarry Smith 564e5c89e4eSSatish Balay Concepts: files^synchronized writing of binary files 565e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 566e5c89e4eSSatish Balay 5674ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(), 5684ebed01fSBarry Smith PetscBinarySynchronizedSeek() 569e5c89e4eSSatish Balay @*/ 5701d280d73SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp) 571e5c89e4eSSatish Balay { 572e5c89e4eSSatish Balay PetscErrorCode ierr; 573e5c89e4eSSatish Balay PetscMPIInt rank; 574e5c89e4eSSatish Balay 575e5c89e4eSSatish Balay PetscFunctionBegin; 576e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 577e5c89e4eSSatish Balay if (!rank) { 578e5c89e4eSSatish Balay ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr); 579e5c89e4eSSatish Balay } 580e5c89e4eSSatish Balay PetscFunctionReturn(0); 581e5c89e4eSSatish Balay } 582e5c89e4eSSatish Balay 583e5c89e4eSSatish Balay #undef __FUNCT__ 5841d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedSeek" 585e5c89e4eSSatish Balay /*@C 5861d280d73SBarry Smith PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file. 587e5c89e4eSSatish Balay 588e5c89e4eSSatish Balay 589e5c89e4eSSatish Balay Input Parameters: 590e5c89e4eSSatish Balay + fd - the file 591e5c89e4eSSatish Balay . whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file 592e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_CUR then size is offset from current location 593e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_END then size is offset from end of file 594e5c89e4eSSatish Balay - off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 595e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 596e5c89e4eSSatish Balay 597e5c89e4eSSatish Balay Output Parameter: 598e5c89e4eSSatish Balay . offset - new offset in file 599e5c89e4eSSatish Balay 600e5c89e4eSSatish Balay Level: developer 601e5c89e4eSSatish Balay 602e5c89e4eSSatish Balay Notes: 603e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 604e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 605e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 606e5c89e4eSSatish Balay to determine the offset or location. 607e5c89e4eSSatish Balay 608e5c89e4eSSatish Balay Concepts: binary files^seeking 609e5c89e4eSSatish Balay Concepts: files^seeking in binary 610e5c89e4eSSatish Balay 6114ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 6124ebed01fSBarry Smith PetscBinarySynchronizedSeek() 613e5c89e4eSSatish Balay @*/ 6141d280d73SBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 615e5c89e4eSSatish Balay { 616e5c89e4eSSatish Balay PetscErrorCode ierr; 617e5c89e4eSSatish Balay PetscMPIInt rank; 618e5c89e4eSSatish Balay 619e5c89e4eSSatish Balay PetscFunctionBegin; 620e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 621e5c89e4eSSatish Balay if (!rank) { 622e5c89e4eSSatish Balay ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr); 623e5c89e4eSSatish Balay } 624e5c89e4eSSatish Balay PetscFunctionReturn(0); 625e5c89e4eSSatish Balay } 626e5c89e4eSSatish Balay 627*e39fd77fSBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) 628*e39fd77fSBarry Smith 629*e39fd77fSBarry Smith EXTERN_C_BEGIN 630*e39fd77fSBarry Smith /* 631*e39fd77fSBarry Smith MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions. 632*e39fd77fSBarry Smith These are set into MPI in PetscInitialize() via MPI_Register_datarep() 633*e39fd77fSBarry Smith 634*e39fd77fSBarry Smith Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode) 635*e39fd77fSBarry Smith 636*e39fd77fSBarry Smith This code is not used because MPICH does not support there use 637*e39fd77fSBarry Smith 638*e39fd77fSBarry Smith */ 639*e39fd77fSBarry Smith PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state) 640*e39fd77fSBarry Smith { 641*e39fd77fSBarry Smith MPI_Aint ub; 642*e39fd77fSBarry Smith PetscMPIInt ierr; 643*e39fd77fSBarry Smith 644*e39fd77fSBarry Smith ierr = MPI_Type_get_extent(datatype,&ub,file_extent); 645*e39fd77fSBarry Smith return ierr; 646*e39fd77fSBarry Smith } 647*e39fd77fSBarry Smith 648*e39fd77fSBarry Smith PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 649*e39fd77fSBarry Smith { 650*e39fd77fSBarry Smith PetscDataType pdtype; 651*e39fd77fSBarry Smith PetscMPIInt ierr; 652*e39fd77fSBarry Smith size_t dsize; 653*e39fd77fSBarry Smith 654*e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 655*e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 656*e39fd77fSBarry Smith 657*e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 658*e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize*position; 659*e39fd77fSBarry Smith 660*e39fd77fSBarry Smith ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr); 661*e39fd77fSBarry Smith ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr); 662*e39fd77fSBarry Smith return ierr; 663*e39fd77fSBarry Smith } 664*e39fd77fSBarry Smith 665*e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 666*e39fd77fSBarry Smith { 667*e39fd77fSBarry Smith PetscDataType pdtype; 668*e39fd77fSBarry Smith PetscMPIInt ierr; 669*e39fd77fSBarry Smith size_t dsize; 670*e39fd77fSBarry Smith 671*e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 672*e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 673*e39fd77fSBarry Smith 674*e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 675*e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize*position; 676*e39fd77fSBarry Smith 677*e39fd77fSBarry Smith ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr); 678*e39fd77fSBarry Smith ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr); 679*e39fd77fSBarry Smith return ierr; 680*e39fd77fSBarry Smith } 681*e39fd77fSBarry Smith EXTERN_C_END 682*e39fd77fSBarry Smith 683*e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 684*e39fd77fSBarry Smith { 685*e39fd77fSBarry Smith PetscErrorCode ierr; 686*e39fd77fSBarry Smith PetscDataType pdtype; 687*e39fd77fSBarry Smith 688*e39fd77fSBarry Smith PetscFunctionBegin; 689*e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 690*e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 691*e39fd77fSBarry Smith ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 692*e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 693*e39fd77fSBarry Smith return ierr; 694*e39fd77fSBarry Smith } 695*e39fd77fSBarry Smith 696*e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 697*e39fd77fSBarry Smith { 698*e39fd77fSBarry Smith PetscErrorCode ierr; 699*e39fd77fSBarry Smith PetscDataType pdtype; 700*e39fd77fSBarry Smith 701*e39fd77fSBarry Smith PetscFunctionBegin; 702*e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 703*e39fd77fSBarry Smith ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 704*e39fd77fSBarry Smith ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr); 705*e39fd77fSBarry Smith return ierr; 706*e39fd77fSBarry Smith } 707*e39fd77fSBarry Smith #endif 708