1e5c89e4eSSatish Balay #define PETSC_DLL 2e5c89e4eSSatish Balay 3e5c89e4eSSatish Balay /* 4e5c89e4eSSatish Balay This file contains simple binary read/write routines. 5e5c89e4eSSatish Balay */ 6e5c89e4eSSatish Balay 7*e30d2299SSatish Balay #include "petsc.h" 8*e30d2299SSatish 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 (PETSC_SIZEOF_INT == 8) 21e5c89e4eSSatish Balay #define PetscInt32 short 22e5c89e4eSSatish Balay #else 23e5c89e4eSSatish Balay #define PetscInt32 int 24e5c89e4eSSatish Balay #endif 25e5c89e4eSSatish Balay 26e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 27e5c89e4eSSatish Balay 28e5c89e4eSSatish Balay #undef __FUNCT__ 29e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapInt" 30e5c89e4eSSatish Balay /* 31e5c89e4eSSatish Balay PetscByteSwapInt - Swap bytes in a 32 bit integer. NOT a PetscInt! Note that PETSc binary read and write 32e5c89e4eSSatish Balay always store and read only 32 bit integers! (See PetscBinaryRead(), PetscBinaryWrite()). 33e5c89e4eSSatish Balay 34e5c89e4eSSatish Balay */ 35e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapInt(PetscInt32 *buff,PetscInt n) 36e5c89e4eSSatish Balay { 37e5c89e4eSSatish Balay PetscInt i,j,tmp = 0; 38e5c89e4eSSatish Balay PetscInt *tptr = &tmp; /* Need to access tmp indirectly to get */ 39e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; /* arround the bug in DEC-ALPHA g++ */ 40e5c89e4eSSatish Balay 41e5c89e4eSSatish Balay PetscFunctionBegin; 42e5c89e4eSSatish Balay for (j=0; j<n; j++) { 43e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 44e5c89e4eSSatish Balay for (i=0; i<(int)sizeof(PetscInt32); i++) { 45e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(PetscInt32)-1-i]; 46e5c89e4eSSatish Balay } 47e5c89e4eSSatish Balay buff[j] = *tptr; 48e5c89e4eSSatish Balay } 49e5c89e4eSSatish Balay PetscFunctionReturn(0); 50e5c89e4eSSatish Balay } 51e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 52e5c89e4eSSatish Balay #undef __FUNCT__ 53e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort" 54e5c89e4eSSatish Balay /* 55e5c89e4eSSatish Balay PetscByteSwapShort - Swap bytes in a short 56e5c89e4eSSatish Balay */ 57e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapShort(short *buff,PetscInt n) 58e5c89e4eSSatish Balay { 59e5c89e4eSSatish Balay PetscInt i,j; 60e5c89e4eSSatish Balay short tmp; 61e5c89e4eSSatish Balay short *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 62e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 63e5c89e4eSSatish Balay 64e5c89e4eSSatish Balay PetscFunctionBegin; 65e5c89e4eSSatish Balay for (j=0; j<n; j++) { 66e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 67e5c89e4eSSatish Balay for (i=0; i<(int) sizeof(short); i++) { 68e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(int)-1-i]; 69e5c89e4eSSatish Balay } 70e5c89e4eSSatish Balay buff[j] = *tptr; 71e5c89e4eSSatish Balay } 72e5c89e4eSSatish Balay PetscFunctionReturn(0); 73e5c89e4eSSatish Balay } 74e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 75e5c89e4eSSatish Balay #undef __FUNCT__ 76e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar" 77e5c89e4eSSatish Balay /* 78e5c89e4eSSatish Balay PetscByteSwapScalar - Swap bytes in a double 79e5c89e4eSSatish Balay Complex is dealt with as if array of double twice as long. 80e5c89e4eSSatish Balay */ 81e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapScalar(PetscScalar *buff,PetscInt n) 82e5c89e4eSSatish Balay { 83e5c89e4eSSatish Balay PetscInt i,j; 84e5c89e4eSSatish Balay PetscReal tmp,*buff1 = (PetscReal*)buff; 85e5c89e4eSSatish Balay PetscReal *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 86e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 87e5c89e4eSSatish Balay 88e5c89e4eSSatish Balay PetscFunctionBegin; 89e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 90e5c89e4eSSatish Balay n *= 2; 91e5c89e4eSSatish Balay #endif 92e5c89e4eSSatish Balay for (j=0; j<n; j++) { 93e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 94e5c89e4eSSatish Balay for (i=0; i<(int) sizeof(PetscReal); i++) { 95e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(PetscReal)-1-i]; 96e5c89e4eSSatish Balay } 97e5c89e4eSSatish Balay buff1[j] = *tptr; 98e5c89e4eSSatish Balay } 99e5c89e4eSSatish Balay PetscFunctionReturn(0); 100e5c89e4eSSatish Balay } 101e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 102e5c89e4eSSatish Balay #undef __FUNCT__ 103e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapDouble" 104e5c89e4eSSatish Balay /* 105e5c89e4eSSatish Balay PetscByteSwapDouble - Swap bytes in a double 106e5c89e4eSSatish Balay */ 107e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapDouble(double *buff,PetscInt n) 108e5c89e4eSSatish Balay { 109e5c89e4eSSatish Balay PetscInt i,j; 110e5c89e4eSSatish Balay double tmp,*buff1 = (double*)buff; 111e5c89e4eSSatish Balay double *tptr = &tmp; /* take care pf bug in DEC-ALPHA g++ */ 112e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 113e5c89e4eSSatish Balay 114e5c89e4eSSatish Balay PetscFunctionBegin; 115e5c89e4eSSatish Balay for (j=0; j<n; j++) { 116e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 117e5c89e4eSSatish Balay for (i=0; i<(int) sizeof(double); i++) { 118e5c89e4eSSatish Balay ptr2[i] = ptr1[sizeof(double)-1-i]; 119e5c89e4eSSatish Balay } 120e5c89e4eSSatish Balay buff1[j] = *tptr; 121e5c89e4eSSatish Balay } 122e5c89e4eSSatish Balay PetscFunctionReturn(0); 123e5c89e4eSSatish Balay } 124e5c89e4eSSatish Balay #endif 125e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 126e5c89e4eSSatish Balay #undef __FUNCT__ 127e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead" 128*e30d2299SSatish Balay /*@ 129e5c89e4eSSatish Balay PetscBinaryRead - Reads from a binary file. 130e5c89e4eSSatish Balay 131e5c89e4eSSatish Balay Not Collective 132e5c89e4eSSatish Balay 133e5c89e4eSSatish Balay Input Parameters: 134e5c89e4eSSatish Balay + fd - the file 135e5c89e4eSSatish Balay . n - the number of items to read 136e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 137e5c89e4eSSatish Balay 138e5c89e4eSSatish Balay Output Parameters: 139e5c89e4eSSatish Balay . p - the buffer 140e5c89e4eSSatish Balay 141e5c89e4eSSatish Balay 142e5c89e4eSSatish Balay 143e5c89e4eSSatish Balay Level: developer 144e5c89e4eSSatish Balay 145e5c89e4eSSatish Balay Notes: 146e5c89e4eSSatish Balay PetscBinaryRead() uses byte swapping to work on all machines; the files 147e5c89e4eSSatish Balay are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers 148e5c89e4eSSatish Balay are converted to the small-endian format when they are read in from the file. 149e5c89e4eSSatish Balay Integers are stored on the file as 32 bits long, regardless of whether 150e5c89e4eSSatish Balay they are stored in the machine as 32 bits or 64 bits, this means the same 151e5c89e4eSSatish Balay binary file may be read on any machine. 152e5c89e4eSSatish Balay 153e5c89e4eSSatish Balay Concepts: files^reading binary 154e5c89e4eSSatish Balay Concepts: binary files^reading 155e5c89e4eSSatish Balay 156e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose() 157e5c89e4eSSatish Balay @*/ 158e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type) 159e5c89e4eSSatish Balay { 160e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) || !defined(PETSC_WORDS_BIGENDIAN) 161e5c89e4eSSatish Balay PetscErrorCode ierr; 162e5c89e4eSSatish Balay #endif 163e5c89e4eSSatish Balay int wsize,err; 164e5c89e4eSSatish Balay size_t m = (size_t) n,maxblock = 65536; 165e5c89e4eSSatish Balay char *pp = (char*)p; 166e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || !defined(PETSC_WORDS_BIGENDIAN) || defined(PETSC_USE_64BIT_INT) 167e5c89e4eSSatish Balay void *ptmp = p; 168e5c89e4eSSatish Balay #endif 169e5c89e4eSSatish Balay 170e5c89e4eSSatish Balay PetscFunctionBegin; 171e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 172e5c89e4eSSatish Balay 173e5c89e4eSSatish Balay if (type == PETSC_INT){ 174e5c89e4eSSatish Balay m *= sizeof(PetscInt32); 175e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) 176e5c89e4eSSatish Balay /* read them in as 32 bit ints, later stretch into ints */ 177e5c89e4eSSatish Balay ierr = PetscMalloc(m,&pp);CHKERRQ(ierr); 178e5c89e4eSSatish Balay ptmp = (void*)pp; 179e5c89e4eSSatish Balay #endif 180e5c89e4eSSatish Balay } 181e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 182e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 183e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 184e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 185e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 186e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) m *= sizeof(PetscTruth); 187e5c89e4eSSatish Balay else if (type == PETSC_LOGICAL) m = PetscBTLength(m)*sizeof(char); 188e5c89e4eSSatish Balay else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 189e5c89e4eSSatish Balay 190e5c89e4eSSatish Balay while (m) { 191e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 192e5c89e4eSSatish Balay err = read(fd,pp,wsize); 193e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 194e5c89e4eSSatish Balay if (!err && wsize > 0) SETERRQ(PETSC_ERR_FILE_READ,"Read past end of file"); 195e5c89e4eSSatish Balay if (err < 0) SETERRQ(PETSC_ERR_FILE_READ,"Error reading from file"); 196e5c89e4eSSatish Balay m -= err; 197e5c89e4eSSatish Balay pp += err; 198e5c89e4eSSatish Balay } 199e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 200e5c89e4eSSatish Balay if (type == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 201e5c89e4eSSatish Balay else if (type == PETSC_ENUM) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 202e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 203e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);} 204e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)ptmp,n);CHKERRQ(ierr);} 205e5c89e4eSSatish Balay else if (type == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);} 206e5c89e4eSSatish Balay #endif 207e5c89e4eSSatish Balay 208e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) 209e5c89e4eSSatish Balay if (type == PETSC_INT) { 210e5c89e4eSSatish Balay PetscInt *p_int = (PetscInt*)p,i; 211e5c89e4eSSatish Balay PetscInt32 *p_short = (PetscInt32 *)ptmp; 212e5c89e4eSSatish Balay for (i=0; i<n; i++) { 213e5c89e4eSSatish Balay p_int[i] = (PetscInt)p_short[i]; 214e5c89e4eSSatish Balay } 215e5c89e4eSSatish Balay ierr = PetscFree(ptmp);CHKERRQ(ierr); 216e5c89e4eSSatish Balay } 217e5c89e4eSSatish Balay #endif 218e5c89e4eSSatish Balay PetscFunctionReturn(0); 219e5c89e4eSSatish Balay } 220e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 221e5c89e4eSSatish Balay #undef __FUNCT__ 222e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite" 223*e30d2299SSatish Balay /*@ 224e5c89e4eSSatish Balay PetscBinaryWrite - Writes to a binary file. 225e5c89e4eSSatish Balay 226e5c89e4eSSatish Balay Not Collective 227e5c89e4eSSatish Balay 228e5c89e4eSSatish Balay Input Parameters: 229e5c89e4eSSatish Balay + fd - the file 230e5c89e4eSSatish Balay . p - the buffer 231e5c89e4eSSatish Balay . n - the number of items to write 232e5c89e4eSSatish Balay . type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 233e5c89e4eSSatish Balay - istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise. 234e5c89e4eSSatish Balay 235e5c89e4eSSatish Balay Level: advanced 236e5c89e4eSSatish Balay 237e5c89e4eSSatish Balay Notes: 238e5c89e4eSSatish Balay PetscBinaryWrite() uses byte swapping to work on all machines; the files 239e5c89e4eSSatish Balay are written using big-endian ordering to the file. On small-endian machines the numbers 240e5c89e4eSSatish Balay are converted to the big-endian format when they are written to disk. 241e5c89e4eSSatish Balay Integers are stored on the file as 32 bits long, regardless of whether 242e5c89e4eSSatish Balay they are stored in the machine as 32 bits or 64 bits, this means the same 243e5c89e4eSSatish Balay binary file may be read on any machine. It also means that 64 bit integers larger than 244e5c89e4eSSatish Balay roughly 2 billion are TRUNCATED/WRONG when written to the file. 245e5c89e4eSSatish Balay 246e5c89e4eSSatish Balay The Buffer p should be read-write buffer, and not static data. 247e5c89e4eSSatish Balay This way, byte-swapping is done in-place, and then the buffer is 248e5c89e4eSSatish Balay written to the file. 249e5c89e4eSSatish Balay 250e5c89e4eSSatish Balay This routine restores the original contents of the buffer, after 251e5c89e4eSSatish Balay it is written to the file. This is done by byte-swapping in-place 252e5c89e4eSSatish Balay the second time. If the flag istemp is set to PETSC_TRUE, the second 253e5c89e4eSSatish Balay byte-swapping operation is not done, thus saving some computation, 254e5c89e4eSSatish Balay but the buffer corrupted is corrupted. 255e5c89e4eSSatish Balay 256e5c89e4eSSatish Balay Concepts: files^writing binary 257e5c89e4eSSatish Balay Concepts: binary files^writing 258e5c89e4eSSatish Balay 259e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose() 260e5c89e4eSSatish Balay @*/ 261e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp) 262e5c89e4eSSatish Balay { 263e5c89e4eSSatish Balay char *pp = (char*)p; 264e5c89e4eSSatish Balay int err,wsize; 265e5c89e4eSSatish Balay size_t m = (size_t)n,maxblock=65536; 266e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) || (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) 267e5c89e4eSSatish Balay PetscErrorCode ierr; 268e5c89e4eSSatish Balay void *ptmp = p; 269e5c89e4eSSatish Balay #endif 270e5c89e4eSSatish Balay 271e5c89e4eSSatish Balay PetscFunctionBegin; 272e5c89e4eSSatish Balay if (n < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n); 273e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 274e5c89e4eSSatish Balay 275e5c89e4eSSatish Balay if (type == PETSC_INT){ 276e5c89e4eSSatish Balay m *= sizeof(PetscInt32); 277e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) 278e5c89e4eSSatish Balay PetscInt *p_int = (PetscInt*)p,i; 279e5c89e4eSSatish Balay PetscInt32 *p_short; 280e5c89e4eSSatish Balay ierr = PetscMalloc(m,&pp);CHKERRQ(ierr); 281e5c89e4eSSatish Balay ptmp = (void*)pp; 282e5c89e4eSSatish Balay p_short = (PetscInt32*)pp; 283e5c89e4eSSatish Balay 284e5c89e4eSSatish Balay for (i=0; i<n; i++) { 285e5c89e4eSSatish Balay p_short[i] = (PetscInt32) p_int[i]; 286e5c89e4eSSatish Balay } 287e5c89e4eSSatish Balay istemp = PETSC_TRUE; 288e5c89e4eSSatish Balay #endif 289e5c89e4eSSatish Balay } 290e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) m *= sizeof(PetscScalar); 291e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) m *= sizeof(double); 292e5c89e4eSSatish Balay else if (type == PETSC_SHORT) m *= sizeof(short); 293e5c89e4eSSatish Balay else if (type == PETSC_CHAR) m *= sizeof(char); 294e5c89e4eSSatish Balay else if (type == PETSC_ENUM) m *= sizeof(PetscEnum); 295e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) m *= sizeof(PetscTruth); 296e5c89e4eSSatish Balay else if (type == PETSC_LOGICAL) m = PetscBTLength(m)*sizeof(char); 297e5c89e4eSSatish Balay else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 298e5c89e4eSSatish Balay 299e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) 300e5c89e4eSSatish Balay if (type == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 301e5c89e4eSSatish Balay else if (type == PETSC_ENUM) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 302e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 303e5c89e4eSSatish Balay else if (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);} 304e5c89e4eSSatish Balay else if (type == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)ptmp,n);CHKERRQ(ierr);} 305e5c89e4eSSatish Balay else if (type == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);} 306e5c89e4eSSatish Balay #endif 307e5c89e4eSSatish Balay 308e5c89e4eSSatish Balay while (m) { 309e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 310e5c89e4eSSatish Balay err = write(fd,pp,wsize); 311e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 312e5c89e4eSSatish Balay if (err != wsize) SETERRQ(PETSC_ERR_FILE_WRITE,"Error writing to file."); 313e5c89e4eSSatish Balay m -= wsize; 314e5c89e4eSSatish Balay pp += wsize; 315e5c89e4eSSatish Balay } 316e5c89e4eSSatish Balay 317e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) && !(PETSC_SIZEOF_INT == 8) && !defined(PETSC_USE_64BIT_INT) 318e5c89e4eSSatish Balay if (!istemp) { 319e5c89e4eSSatish Balay if (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);} 320e5c89e4eSSatish Balay else if (type == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);} 321e5c89e4eSSatish Balay else if (type == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 322e5c89e4eSSatish Balay else if (type == PETSC_ENUM) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 323e5c89e4eSSatish Balay else if (type == PETSC_TRUTH) {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);} 324e5c89e4eSSatish Balay } 325e5c89e4eSSatish Balay #endif 326e5c89e4eSSatish Balay 327e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) 328e5c89e4eSSatish Balay if (type == PETSC_INT){ 329e5c89e4eSSatish Balay ierr = PetscFree(ptmp);CHKERRQ(ierr); 330e5c89e4eSSatish Balay } 331e5c89e4eSSatish Balay #endif 332e5c89e4eSSatish Balay PetscFunctionReturn(0); 333e5c89e4eSSatish Balay } 334e5c89e4eSSatish Balay 335e5c89e4eSSatish Balay #undef __FUNCT__ 336e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen" 337e5c89e4eSSatish Balay /*@C 338e5c89e4eSSatish Balay PetscBinaryOpen - Opens a PETSc binary file. 339e5c89e4eSSatish Balay 340e5c89e4eSSatish Balay Not Collective 341e5c89e4eSSatish Balay 342e5c89e4eSSatish Balay Input Parameters: 343e5c89e4eSSatish Balay + name - filename 344e5c89e4eSSatish Balay - type - type of binary file, on of PETSC_FILE_RDONLY, PETSC_FILE_WRONLY, PETSC_FILE_CREATE 345e5c89e4eSSatish Balay 346e5c89e4eSSatish Balay Output Parameter: 347e5c89e4eSSatish Balay . fd - the file 348e5c89e4eSSatish Balay 349e5c89e4eSSatish Balay Level: advanced 350e5c89e4eSSatish Balay 351e5c89e4eSSatish Balay Concepts: files^opening binary 352e5c89e4eSSatish Balay Concepts: binary files^opening 353e5c89e4eSSatish Balay 354e5c89e4eSSatish Balay Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in 355e5c89e4eSSatish Balay big-endian format. This means the file can be accessed using PetscBinaryOpen() and 356e5c89e4eSSatish Balay PetscBinaryRead() and PetscBinaryWrite() on any machine. 357e5c89e4eSSatish Balay 358e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite() 359e5c89e4eSSatish Balay @*/ 360e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryOpen(const char name[],int type,int *fd) 361e5c89e4eSSatish Balay { 362e5c89e4eSSatish Balay PetscFunctionBegin; 363e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY) 364e5c89e4eSSatish Balay if (type == PETSC_FILE_CREATE) { 365e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) { 366e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 367e5c89e4eSSatish Balay } 368e5c89e4eSSatish Balay } else if (type == PETSC_FILE_RDONLY) { 369e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) { 370e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 371e5c89e4eSSatish Balay } 372e5c89e4eSSatish Balay } else if (type == PETSC_FILE_WRONLY) { 373e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) { 374e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 375e5c89e4eSSatish Balay } 376e5c89e4eSSatish Balay #else 377e5c89e4eSSatish Balay if (type == PETSC_FILE_CREATE) { 378e5c89e4eSSatish Balay if ((*fd = creat(name,0666)) == -1) { 379e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name); 380e5c89e4eSSatish Balay } 381e5c89e4eSSatish Balay } else if (type == PETSC_FILE_RDONLY) { 382e5c89e4eSSatish Balay if ((*fd = open(name,O_RDONLY,0)) == -1) { 383e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name); 384e5c89e4eSSatish Balay } 385e5c89e4eSSatish Balay } 386e5c89e4eSSatish Balay else if (type == PETSC_FILE_WRONLY) { 387e5c89e4eSSatish Balay if ((*fd = open(name,O_WRONLY,0)) == -1) { 388e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name); 389e5c89e4eSSatish Balay } 390e5c89e4eSSatish Balay #endif 391e5c89e4eSSatish Balay } else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type"); 392e5c89e4eSSatish Balay PetscFunctionReturn(0); 393e5c89e4eSSatish Balay } 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay #undef __FUNCT__ 396e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose" 397*e30d2299SSatish Balay /*@ 398e5c89e4eSSatish Balay PetscBinaryClose - Closes a PETSc binary file. 399e5c89e4eSSatish Balay 400e5c89e4eSSatish Balay Not Collective 401e5c89e4eSSatish Balay 402e5c89e4eSSatish Balay Output Parameter: 403e5c89e4eSSatish Balay . fd - the file 404e5c89e4eSSatish Balay 405e5c89e4eSSatish Balay Level: advanced 406e5c89e4eSSatish Balay 407e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen() 408e5c89e4eSSatish Balay @*/ 409e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryClose(int fd) 410e5c89e4eSSatish Balay { 411e5c89e4eSSatish Balay PetscFunctionBegin; 412e5c89e4eSSatish Balay close(fd); 413e5c89e4eSSatish Balay PetscFunctionReturn(0); 414e5c89e4eSSatish Balay } 415e5c89e4eSSatish Balay 416e5c89e4eSSatish Balay 417e5c89e4eSSatish Balay #undef __FUNCT__ 418e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek" 419*e30d2299SSatish Balay /*@ 420e5c89e4eSSatish Balay PetscBinarySeek - Moves the file pointer on a PETSc binary file. 421e5c89e4eSSatish Balay 422e5c89e4eSSatish Balay Not Collective 423e5c89e4eSSatish Balay 424e5c89e4eSSatish Balay Input Parameters: 425e5c89e4eSSatish Balay + fd - the file 426e5c89e4eSSatish Balay . whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file 427e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_CUR then size is offset from current location 428e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_END then size is offset from end of file 429e5c89e4eSSatish Balay - size - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 430e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 431e5c89e4eSSatish Balay 432e5c89e4eSSatish Balay Output Parameter: 433e5c89e4eSSatish Balay . offset - new offset in file 434e5c89e4eSSatish Balay 435e5c89e4eSSatish Balay Level: developer 436e5c89e4eSSatish Balay 437e5c89e4eSSatish Balay Notes: 438e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 439e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 440e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 441e5c89e4eSSatish Balay to determine the offset or location. 442e5c89e4eSSatish Balay 443e5c89e4eSSatish Balay Concepts: files^binary seeking 444e5c89e4eSSatish Balay Concepts: binary files^seeking 445e5c89e4eSSatish Balay 446e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen() 447e5c89e4eSSatish Balay @*/ 448e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 449e5c89e4eSSatish Balay { 450e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) || defined(PETSC_HAVE__LSEEK) 451e5c89e4eSSatish Balay int iwhence=0; 452e5c89e4eSSatish Balay #endif 453e5c89e4eSSatish Balay 454e5c89e4eSSatish Balay PetscFunctionBegin; 455e5c89e4eSSatish Balay if (whence == PETSC_BINARY_SEEK_SET) { 456e5c89e4eSSatish Balay iwhence = SEEK_SET; 457e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_CUR) { 458e5c89e4eSSatish Balay iwhence = SEEK_CUR; 459e5c89e4eSSatish Balay } else if (whence == PETSC_BINARY_SEEK_END) { 460e5c89e4eSSatish Balay iwhence = SEEK_END; 461e5c89e4eSSatish Balay } else { 462e5c89e4eSSatish Balay SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location"); 463e5c89e4eSSatish Balay } 464e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) 465e5c89e4eSSatish Balay *offset = lseek(fd,off,iwhence); 466e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK) 467e5c89e4eSSatish Balay *offset = _lseek(fd,(long)off,iwhence); 468e5c89e4eSSatish Balay #else 469e5c89e4eSSatish Balay SETERRQ(PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file"); 470e5c89e4eSSatish Balay #endif 471e5c89e4eSSatish Balay PetscFunctionReturn(0); 472e5c89e4eSSatish Balay } 473e5c89e4eSSatish Balay 474e5c89e4eSSatish Balay #undef __FUNCT__ 475e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinaryRead" 476e5c89e4eSSatish Balay /*@C 477e5c89e4eSSatish Balay PetscSynchronizedBinaryRead - Reads from a binary file. 478e5c89e4eSSatish Balay 479e5c89e4eSSatish Balay Collective on MPI_Comm 480e5c89e4eSSatish Balay 481e5c89e4eSSatish Balay Input Parameters: 482e5c89e4eSSatish Balay + comm - the MPI communicator 483e5c89e4eSSatish Balay . fd - the file 484e5c89e4eSSatish Balay . n - the number of items to read 485e5c89e4eSSatish Balay - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 486e5c89e4eSSatish Balay 487e5c89e4eSSatish Balay Output Parameters: 488e5c89e4eSSatish Balay . p - the buffer 489e5c89e4eSSatish Balay 490e5c89e4eSSatish Balay Options Database Key: 491e5c89e4eSSatish Balay . -binary_longints - indicates the file was generated on a Cray vector 492e5c89e4eSSatish Balay machine (not the T3E/D) and the ints are stored as 64 bit 493e5c89e4eSSatish Balay quantities, otherwise they are stored as 32 bit 494e5c89e4eSSatish Balay 495e5c89e4eSSatish Balay Level: developer 496e5c89e4eSSatish Balay 497e5c89e4eSSatish Balay Notes: 498e5c89e4eSSatish Balay Does a PetscBinaryRead() followed by an MPI_Bcast() 499e5c89e4eSSatish Balay 500e5c89e4eSSatish Balay PetscSynchronizedBinaryRead() uses byte swapping to work on all machines. 501e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 502e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 503e5c89e4eSSatish Balay binary file may be read on any machine. 504e5c89e4eSSatish Balay 505e5c89e4eSSatish Balay Concepts: files^synchronized reading of binary files 506e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 507e5c89e4eSSatish Balay 508e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead() 509e5c89e4eSSatish Balay @*/ 510e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinaryRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type) 511e5c89e4eSSatish Balay { 512e5c89e4eSSatish Balay PetscErrorCode ierr; 513e5c89e4eSSatish Balay PetscMPIInt rank; 514e5c89e4eSSatish Balay MPI_Datatype mtype; 515e5c89e4eSSatish Balay 516e5c89e4eSSatish Balay PetscFunctionBegin; 517e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 518e5c89e4eSSatish Balay if (!rank) { 519e5c89e4eSSatish Balay ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr); 520e5c89e4eSSatish Balay } 521e5c89e4eSSatish Balay ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr); 522e5c89e4eSSatish Balay ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr); 523e5c89e4eSSatish Balay PetscFunctionReturn(0); 524e5c89e4eSSatish Balay } 525e5c89e4eSSatish Balay 526e5c89e4eSSatish Balay #undef __FUNCT__ 527e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinaryWrite" 528e5c89e4eSSatish Balay /*@C 529e5c89e4eSSatish Balay PetscSynchronizedBinaryWrite - writes to a binary file. 530e5c89e4eSSatish Balay 531e5c89e4eSSatish Balay Collective on MPI_Comm 532e5c89e4eSSatish Balay 533e5c89e4eSSatish Balay Input Parameters: 534e5c89e4eSSatish Balay + comm - the MPI communicator 535e5c89e4eSSatish Balay . fd - the file 536e5c89e4eSSatish Balay . n - the number of items to write 537e5c89e4eSSatish Balay . p - the buffer 538e5c89e4eSSatish Balay . istemp - the buffer may be changed 539e5c89e4eSSatish Balay - type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 540e5c89e4eSSatish Balay 541e5c89e4eSSatish Balay Level: developer 542e5c89e4eSSatish Balay 543e5c89e4eSSatish Balay Notes: 544e5c89e4eSSatish Balay Process 0 does a PetscBinaryWrite() 545e5c89e4eSSatish Balay 546e5c89e4eSSatish Balay PetscSynchronizedBinaryWrite() uses byte swapping to work on all machines. 547e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 548e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 549e5c89e4eSSatish Balay binary file may be read on any machine. 550e5c89e4eSSatish Balay 551e5c89e4eSSatish Balay Concepts: files^synchronized writing of binary files 552e5c89e4eSSatish Balay Concepts: binary files^reading, synchronized 553e5c89e4eSSatish Balay 554e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead() 555e5c89e4eSSatish Balay @*/ 556e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinaryWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp) 557e5c89e4eSSatish Balay { 558e5c89e4eSSatish Balay PetscErrorCode ierr; 559e5c89e4eSSatish Balay PetscMPIInt rank; 560e5c89e4eSSatish Balay 561e5c89e4eSSatish Balay PetscFunctionBegin; 562e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 563e5c89e4eSSatish Balay if (!rank) { 564e5c89e4eSSatish Balay ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr); 565e5c89e4eSSatish Balay } 566e5c89e4eSSatish Balay PetscFunctionReturn(0); 567e5c89e4eSSatish Balay } 568e5c89e4eSSatish Balay 569e5c89e4eSSatish Balay #undef __FUNCT__ 570e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinarySeek" 571e5c89e4eSSatish Balay /*@C 572e5c89e4eSSatish Balay PetscSynchronizedBinarySeek - Moves the file pointer on a PETSc binary file. 573e5c89e4eSSatish Balay 574e5c89e4eSSatish Balay 575e5c89e4eSSatish Balay Input Parameters: 576e5c89e4eSSatish Balay + fd - the file 577e5c89e4eSSatish Balay . whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file 578e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_CUR then size is offset from current location 579e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_END then size is offset from end of file 580e5c89e4eSSatish Balay - off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 581e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 582e5c89e4eSSatish Balay 583e5c89e4eSSatish Balay Output Parameter: 584e5c89e4eSSatish Balay . offset - new offset in file 585e5c89e4eSSatish Balay 586e5c89e4eSSatish Balay Level: developer 587e5c89e4eSSatish Balay 588e5c89e4eSSatish Balay Notes: 589e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 590e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 591e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 592e5c89e4eSSatish Balay to determine the offset or location. 593e5c89e4eSSatish Balay 594e5c89e4eSSatish Balay Concepts: binary files^seeking 595e5c89e4eSSatish Balay Concepts: files^seeking in binary 596e5c89e4eSSatish Balay 597e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen() 598e5c89e4eSSatish Balay @*/ 599e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinarySeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 600e5c89e4eSSatish Balay { 601e5c89e4eSSatish Balay PetscErrorCode ierr; 602e5c89e4eSSatish Balay PetscMPIInt rank; 603e5c89e4eSSatish Balay 604e5c89e4eSSatish Balay PetscFunctionBegin; 605e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 606e5c89e4eSSatish Balay if (!rank) { 607e5c89e4eSSatish Balay ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr); 608e5c89e4eSSatish Balay } 609e5c89e4eSSatish Balay PetscFunctionReturn(0); 610e5c89e4eSSatish Balay } 611e5c89e4eSSatish Balay 612