1e5c89e4eSSatish Balay 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay This file contains simple binary read/write routines. 4e5c89e4eSSatish Balay */ 5e5c89e4eSSatish Balay 6c6db04a5SJed Brown #include <petscsys.h> 7*cc843e7aSLisandro Dalcin #include <petscbt.h> 8e5c89e4eSSatish Balay #include <errno.h> 9e5c89e4eSSatish Balay #include <fcntl.h> 10e5c89e4eSSatish Balay #if defined(PETSC_HAVE_UNISTD_H) 11e5c89e4eSSatish Balay #include <unistd.h> 12e5c89e4eSSatish Balay #endif 13e5c89e4eSSatish Balay #if defined(PETSC_HAVE_IO_H) 14e5c89e4eSSatish Balay #include <io.h> 15e5c89e4eSSatish Balay #endif 16*cc843e7aSLisandro Dalcin #if !defined(PETSC_HAVE_O_BINARY) 17*cc843e7aSLisandro Dalcin #define O_BINARY 0 18*cc843e7aSLisandro Dalcin #endif 19e5c89e4eSSatish Balay 2002c9f0b5SLisandro Dalcin const char *const PetscFileModes[] = {"READ","WRITE","APPEND","UPDATE","APPEND_UPDATE","PetscFileMode","PETSC_FILE_",NULL}; 21d6a4318aSJed Brown 226de02169SBarry Smith /* --------------------------------------------------------- */ 23e5c89e4eSSatish Balay /* 246de02169SBarry Smith PetscByteSwapEnum - Swap bytes in a PETSc Enum 25e5c89e4eSSatish Balay 26e5c89e4eSSatish Balay */ 277087cfbeSBarry Smith PetscErrorCode PetscByteSwapEnum(PetscEnum *buff,PetscInt n) 28e5c89e4eSSatish Balay { 296de02169SBarry Smith PetscInt i,j; 300b20345dSBarry Smith PetscEnum tmp = ENUM_DUMMY; 31e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 32e5c89e4eSSatish Balay 33e5c89e4eSSatish Balay PetscFunctionBegin; 34e5c89e4eSSatish Balay for (j=0; j<n; j++) { 35e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 36a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) ptr2[i] = ptr1[sizeof(PetscEnum)-1-i]; 37a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) ptr1[i] = ptr2[i]; 38e5c89e4eSSatish Balay } 39e5c89e4eSSatish Balay PetscFunctionReturn(0); 40e5c89e4eSSatish Balay } 416de02169SBarry Smith 426de02169SBarry Smith /* 43acfcf0e5SJed Brown PetscByteSwapBool - Swap bytes in a PETSc Bool 446de02169SBarry Smith 456de02169SBarry Smith */ 467087cfbeSBarry Smith PetscErrorCode PetscByteSwapBool(PetscBool *buff,PetscInt n) 476de02169SBarry Smith { 486de02169SBarry Smith PetscInt i,j; 49ace3abfcSBarry Smith PetscBool tmp = PETSC_FALSE; 50e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 516de02169SBarry Smith 526de02169SBarry Smith PetscFunctionBegin; 536de02169SBarry Smith for (j=0; j<n; j++) { 546de02169SBarry Smith ptr1 = (char*)(buff + j); 55a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscBool); i++) ptr2[i] = ptr1[sizeof(PetscBool)-1-i]; 56a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscBool); i++) ptr1[i] = ptr2[i]; 576de02169SBarry Smith } 586de02169SBarry Smith PetscFunctionReturn(0); 596de02169SBarry Smith } 606de02169SBarry Smith 61bd1d2e58SBarry Smith /* 626de02169SBarry Smith PetscByteSwapInt - Swap bytes in a PETSc integer (which may be 32 or 64 bits) 63bd1d2e58SBarry Smith 64bd1d2e58SBarry Smith */ 657087cfbeSBarry Smith PetscErrorCode PetscByteSwapInt(PetscInt *buff,PetscInt n) 66bd1d2e58SBarry Smith { 67bd1d2e58SBarry Smith PetscInt i,j,tmp = 0; 68e0890e22SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 69bd1d2e58SBarry Smith 70bd1d2e58SBarry Smith PetscFunctionBegin; 71bd1d2e58SBarry Smith for (j=0; j<n; j++) { 72bd1d2e58SBarry Smith ptr1 = (char*)(buff + j); 73a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscInt); i++) ptr2[i] = ptr1[sizeof(PetscInt)-1-i]; 74a297a907SKarl Rupp for (i=0; i<(PetscInt)sizeof(PetscInt); i++) ptr1[i] = ptr2[i]; 75bd1d2e58SBarry Smith } 76bd1d2e58SBarry Smith PetscFunctionReturn(0); 77bd1d2e58SBarry Smith } 789e3e4c22SLisandro Dalcin 799e3e4c22SLisandro Dalcin /* 809e3e4c22SLisandro Dalcin PetscByteSwapInt64 - Swap bytes in a PETSc integer (64 bits) 819e3e4c22SLisandro Dalcin 829e3e4c22SLisandro Dalcin */ 839e3e4c22SLisandro Dalcin PetscErrorCode PetscByteSwapInt64(PetscInt64 *buff,PetscInt n) 849e3e4c22SLisandro Dalcin { 859e3e4c22SLisandro Dalcin PetscInt i,j; 869e3e4c22SLisandro Dalcin PetscInt64 tmp = 0; 879e3e4c22SLisandro Dalcin char *ptr1,*ptr2 = (char*)&tmp; 889e3e4c22SLisandro Dalcin 899e3e4c22SLisandro Dalcin PetscFunctionBegin; 909e3e4c22SLisandro Dalcin for (j=0; j<n; j++) { 919e3e4c22SLisandro Dalcin ptr1 = (char*)(buff + j); 929e3e4c22SLisandro Dalcin for (i=0; i<(PetscInt)sizeof(PetscInt64); i++) ptr2[i] = ptr1[sizeof(PetscInt64)-1-i]; 939e3e4c22SLisandro Dalcin for (i=0; i<(PetscInt)sizeof(PetscInt64); i++) ptr1[i] = ptr2[i]; 949e3e4c22SLisandro Dalcin } 959e3e4c22SLisandro Dalcin PetscFunctionReturn(0); 969e3e4c22SLisandro Dalcin } 979e3e4c22SLisandro Dalcin 98bd1d2e58SBarry Smith /* --------------------------------------------------------- */ 99e5c89e4eSSatish Balay /* 100e5c89e4eSSatish Balay PetscByteSwapShort - Swap bytes in a short 101e5c89e4eSSatish Balay */ 1027087cfbeSBarry Smith PetscErrorCode PetscByteSwapShort(short *buff,PetscInt n) 103e5c89e4eSSatish Balay { 104e5c89e4eSSatish Balay PetscInt i,j; 105e5c89e4eSSatish Balay short tmp; 106e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 107e5c89e4eSSatish Balay 108e5c89e4eSSatish Balay PetscFunctionBegin; 109e5c89e4eSSatish Balay for (j=0; j<n; j++) { 110e5c89e4eSSatish Balay ptr1 = (char*)(buff + j); 111a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(short); i++) ptr2[i] = ptr1[sizeof(short)-1-i]; 112a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(short); i++) ptr1[i] = ptr2[i]; 113e5c89e4eSSatish Balay } 114e5c89e4eSSatish Balay PetscFunctionReturn(0); 115e5c89e4eSSatish Balay } 116972064b6SLisandro Dalcin /* 117972064b6SLisandro Dalcin PetscByteSwapLong - Swap bytes in a long 118972064b6SLisandro Dalcin */ 119972064b6SLisandro Dalcin PetscErrorCode PetscByteSwapLong(long *buff,PetscInt n) 120972064b6SLisandro Dalcin { 121972064b6SLisandro Dalcin PetscInt i,j; 122972064b6SLisandro Dalcin long tmp; 123972064b6SLisandro Dalcin char *ptr1,*ptr2 = (char*)&tmp; 124972064b6SLisandro Dalcin 125972064b6SLisandro Dalcin PetscFunctionBegin; 126972064b6SLisandro Dalcin for (j=0; j<n; j++) { 127972064b6SLisandro Dalcin ptr1 = (char*)(buff + j); 128972064b6SLisandro Dalcin for (i=0; i<(PetscInt) sizeof(long); i++) ptr2[i] = ptr1[sizeof(long)-1-i]; 129972064b6SLisandro Dalcin for (i=0; i<(PetscInt) sizeof(long); i++) ptr1[i] = ptr2[i]; 130972064b6SLisandro Dalcin } 131972064b6SLisandro Dalcin PetscFunctionReturn(0); 132972064b6SLisandro Dalcin } 133e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 134e5c89e4eSSatish Balay /* 1354caf0332SSatish Balay PetscByteSwapReal - Swap bytes in a PetscReal 1364caf0332SSatish Balay */ 1374caf0332SSatish Balay PetscErrorCode PetscByteSwapReal(PetscReal *buff,PetscInt n) 1384caf0332SSatish Balay { 1394caf0332SSatish Balay PetscInt i,j; 1404caf0332SSatish Balay PetscReal tmp,*buff1 = (PetscReal*)buff; 1414caf0332SSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 1424caf0332SSatish Balay 1434caf0332SSatish Balay PetscFunctionBegin; 1444caf0332SSatish Balay for (j=0; j<n; j++) { 1454caf0332SSatish Balay ptr1 = (char*)(buff1 + j); 1464caf0332SSatish Balay for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr2[i] = ptr1[sizeof(PetscReal)-1-i]; 1474caf0332SSatish Balay for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr1[i] = ptr2[i]; 1484caf0332SSatish Balay } 1494caf0332SSatish Balay PetscFunctionReturn(0); 1504caf0332SSatish Balay } 1514caf0332SSatish Balay /* --------------------------------------------------------- */ 1524caf0332SSatish Balay /* 15341f502e3SPatrick Sanan PetscByteSwapScalar - Swap bytes in a PetscScalar 15441f502e3SPatrick Sanan The complex case is dealt with with an array of PetscReal, twice as long. 155e5c89e4eSSatish Balay */ 1567087cfbeSBarry Smith PetscErrorCode PetscByteSwapScalar(PetscScalar *buff,PetscInt n) 157e5c89e4eSSatish Balay { 158e5c89e4eSSatish Balay PetscInt i,j; 159e5c89e4eSSatish Balay PetscReal tmp,*buff1 = (PetscReal*)buff; 160e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 161e5c89e4eSSatish Balay 162e5c89e4eSSatish Balay PetscFunctionBegin; 163e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 164e5c89e4eSSatish Balay n *= 2; 165e5c89e4eSSatish Balay #endif 166e5c89e4eSSatish Balay for (j=0; j<n; j++) { 167e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 168a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr2[i] = ptr1[sizeof(PetscReal)-1-i]; 169a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr1[i] = ptr2[i]; 170e5c89e4eSSatish Balay } 171e5c89e4eSSatish Balay PetscFunctionReturn(0); 172e5c89e4eSSatish Balay } 173e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 174e5c89e4eSSatish Balay /* 175e5c89e4eSSatish Balay PetscByteSwapDouble - Swap bytes in a double 176e5c89e4eSSatish Balay */ 1777087cfbeSBarry Smith PetscErrorCode PetscByteSwapDouble(double *buff,PetscInt n) 178e5c89e4eSSatish Balay { 179e5c89e4eSSatish Balay PetscInt i,j; 180e5c89e4eSSatish Balay double tmp,*buff1 = (double*)buff; 181e5c89e4eSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 182e5c89e4eSSatish Balay 183e5c89e4eSSatish Balay PetscFunctionBegin; 184e5c89e4eSSatish Balay for (j=0; j<n; j++) { 185e5c89e4eSSatish Balay ptr1 = (char*)(buff1 + j); 186a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(double); i++) ptr2[i] = ptr1[sizeof(double)-1-i]; 187a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(double); i++) ptr1[i] = ptr2[i]; 188e5c89e4eSSatish Balay } 189e5c89e4eSSatish Balay PetscFunctionReturn(0); 190e5c89e4eSSatish Balay } 191e39fd77fSBarry Smith 192e95bf02fSSatish Balay /* 193e95bf02fSSatish Balay PetscByteSwapFloat - Swap bytes in a float 194e95bf02fSSatish Balay */ 195e95bf02fSSatish Balay PetscErrorCode PetscByteSwapFloat(float *buff,PetscInt n) 196e95bf02fSSatish Balay { 197e95bf02fSSatish Balay PetscInt i,j; 198e95bf02fSSatish Balay float tmp,*buff1 = (float*)buff; 199e95bf02fSSatish Balay char *ptr1,*ptr2 = (char*)&tmp; 200e95bf02fSSatish Balay 201e95bf02fSSatish Balay PetscFunctionBegin; 202e95bf02fSSatish Balay for (j=0; j<n; j++) { 203e95bf02fSSatish Balay ptr1 = (char*)(buff1 + j); 204a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(float); i++) ptr2[i] = ptr1[sizeof(float)-1-i]; 205a297a907SKarl Rupp for (i=0; i<(PetscInt) sizeof(float); i++) ptr1[i] = ptr2[i]; 206e95bf02fSSatish Balay } 207e95bf02fSSatish Balay PetscFunctionReturn(0); 208e95bf02fSSatish Balay } 209e95bf02fSSatish Balay 210e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count) 211e39fd77fSBarry Smith { 212e39fd77fSBarry Smith PetscErrorCode ierr; 213e39fd77fSBarry Smith 214e39fd77fSBarry Smith PetscFunctionBegin; 215e39fd77fSBarry Smith if (pdtype == PETSC_INT) {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);} 216e39fd77fSBarry Smith else if (pdtype == PETSC_ENUM) {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);} 217acfcf0e5SJed Brown else if (pdtype == PETSC_BOOL) {ierr = PetscByteSwapBool((PetscBool*)data,count);CHKERRQ(ierr);} 218e39fd77fSBarry Smith else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);} 2194caf0332SSatish Balay else if (pdtype == PETSC_REAL) {ierr = PetscByteSwapReal((PetscReal*)data,count);CHKERRQ(ierr);} 2209860990eSLisandro Dalcin else if (pdtype == PETSC_COMPLEX){ierr = PetscByteSwapReal((PetscReal*)data,2*count);CHKERRQ(ierr);} 2219e3e4c22SLisandro Dalcin else if (pdtype == PETSC_INT64) {ierr = PetscByteSwapInt64((PetscInt64*)data,count);CHKERRQ(ierr);} 222e39fd77fSBarry Smith else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);} 223e95bf02fSSatish Balay else if (pdtype == PETSC_FLOAT) {ierr = PetscByteSwapFloat((float*)data,count);CHKERRQ(ierr);} 224e39fd77fSBarry Smith else if (pdtype == PETSC_SHORT) {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);} 225972064b6SLisandro Dalcin else if (pdtype == PETSC_LONG) {ierr = PetscByteSwapLong((long*)data,count);CHKERRQ(ierr);} 226e39fd77fSBarry Smith PetscFunctionReturn(0); 227e39fd77fSBarry Smith } 228e39fd77fSBarry Smith 2299860990eSLisandro Dalcin /*@C 230e5c89e4eSSatish Balay PetscBinaryRead - Reads from a binary file. 231e5c89e4eSSatish Balay 232e5c89e4eSSatish Balay Not Collective 233e5c89e4eSSatish Balay 234e5c89e4eSSatish Balay Input Parameters: 2359860990eSLisandro Dalcin + fd - the file descriptor 2369860990eSLisandro Dalcin . num - the maximum number of items to read 2379860990eSLisandro Dalcin - type - the type of items to read (PETSC_INT, PETSC_REAL, PETSC_SCALAR, etc.) 238e5c89e4eSSatish Balay 239e5c89e4eSSatish Balay Output Parameters: 2409860990eSLisandro Dalcin + data - the buffer 2419860990eSLisandro Dalcin - count - the number of items read, optional 242e5c89e4eSSatish Balay 243e5c89e4eSSatish Balay Level: developer 244e5c89e4eSSatish Balay 245e5c89e4eSSatish Balay Notes: 2469860990eSLisandro Dalcin If count is not provided and the number of items read is less than 2479860990eSLisandro Dalcin the maximum number of items to read, then this routine errors. 2489860990eSLisandro Dalcin 249e5c89e4eSSatish Balay PetscBinaryRead() uses byte swapping to work on all machines; the files 250f253e43cSLisandro Dalcin are written to file ALWAYS using big-endian ordering. On little-endian machines the numbers 251f253e43cSLisandro Dalcin are converted to the little-endian format when they are read in from the file. 252f253e43cSLisandro Dalcin When PETSc is ./configure with --with-64-bit-indices the integers are written to the 253f253e43cSLisandro Dalcin file as 64 bit integers, this means they can only be read back in when the option --with-64-bit-indices 25454f21887SBarry Smith is used. 255e5c89e4eSSatish Balay 2564ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 2574ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 258e5c89e4eSSatish Balay @*/ 2599860990eSLisandro Dalcin PetscErrorCode PetscBinaryRead(int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type) 260e5c89e4eSSatish Balay { 2619860990eSLisandro Dalcin size_t typesize, m = (size_t) num, n = 0, maxblock = 65536; 2629860990eSLisandro Dalcin char *p = (char*)data; 2637a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 264cba51d77SBarry Smith PetscBool readdouble = PETSC_FALSE; 2659860990eSLisandro Dalcin double *pdouble; 2667a881295SBarry Smith #endif 2679860990eSLisandro Dalcin void *ptmp = data; 26805acbc63SBarry Smith char *fname = NULL; 2699860990eSLisandro Dalcin PetscErrorCode ierr; 270e5c89e4eSSatish Balay 271e5c89e4eSSatish Balay PetscFunctionBegin; 2729860990eSLisandro Dalcin if (count) *count = 0; 2739860990eSLisandro Dalcin if (num < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to read a negative amount of data %D",num); 2749860990eSLisandro Dalcin if (!num) PetscFunctionReturn(0); 275e5c89e4eSSatish Balay 2762d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 2772d53ad75SBarry Smith m = 64; 2782d53ad75SBarry Smith type = PETSC_CHAR; 27905acbc63SBarry Smith fname = (char*)malloc(m*sizeof(char)); 2809860990eSLisandro Dalcin p = (char*)fname; 2812d53ad75SBarry Smith ptmp = (void*)fname; 2829860990eSLisandro Dalcin if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name"); 2832d53ad75SBarry Smith } 2849860990eSLisandro Dalcin if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m); 2852d53ad75SBarry Smith 2869860990eSLisandro Dalcin ierr = PetscDataTypeGetSize(type,&typesize);CHKERRQ(ierr); 287e5c89e4eSSatish Balay 2887a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 289c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-binary_read_double",&readdouble,NULL);CHKERRQ(ierr); 2907a881295SBarry Smith /* If using __float128 precision we still read in doubles from file */ 2919860990eSLisandro Dalcin if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) { 2929860990eSLisandro Dalcin PetscInt cnt = num * ((type == PETSC_REAL) ? 1 : 2); 2939860990eSLisandro Dalcin ierr = PetscMalloc1(cnt,&pdouble);CHKERRQ(ierr); 2949860990eSLisandro Dalcin p = (char*)pdouble; 2959860990eSLisandro Dalcin typesize /= 2; 2967a881295SBarry Smith } 2977a881295SBarry Smith #endif 2987a881295SBarry Smith 2999860990eSLisandro Dalcin m *= typesize; 3009860990eSLisandro Dalcin 301e5c89e4eSSatish Balay while (m) { 3029860990eSLisandro Dalcin size_t len = (m < maxblock) ? m : maxblock; 3039860990eSLisandro Dalcin int ret = (int)read(fd,p,len); 3049860990eSLisandro Dalcin if (ret < 0 && errno == EINTR) continue; 3059860990eSLisandro Dalcin if (!ret && len > 0) break; /* Proxy for EOF */ 3069860990eSLisandro Dalcin if (ret < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno); 3079860990eSLisandro Dalcin m -= ret; 3089860990eSLisandro Dalcin p += ret; 3099860990eSLisandro Dalcin n += ret; 310e5c89e4eSSatish Balay } 3119860990eSLisandro Dalcin if (m && !count) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Read past end of file"); 3129860990eSLisandro Dalcin 3139860990eSLisandro Dalcin num = (PetscInt)(n/typesize); /* Should we require `n % typesize == 0` ? */ 3149860990eSLisandro Dalcin if (count) *count = num; /* TODO: This is most likely wrong for PETSC_BIT_LOGICAL */ 3157a881295SBarry Smith 3167a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 3179860990eSLisandro Dalcin if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) { 3189860990eSLisandro Dalcin PetscInt i, cnt = num * ((type == PETSC_REAL) ? 1 : 2); 3199860990eSLisandro Dalcin PetscReal *preal = (PetscReal*)data; 32030815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwapDouble(pdouble,cnt);CHKERRQ(ierr);} 3219860990eSLisandro Dalcin for (i=0; i<cnt; i++) preal[i] = pdouble[i]; 3229860990eSLisandro Dalcin ierr = PetscFree(pdouble);CHKERRQ(ierr); 3237a881295SBarry Smith PetscFunctionReturn(0); 3247a881295SBarry Smith } 3257a881295SBarry Smith #endif 3267a881295SBarry Smith 32730815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(ptmp,type,num);CHKERRQ(ierr);} 328e5c89e4eSSatish Balay 32905acbc63SBarry Smith if (type == PETSC_FUNCTION) { 3302d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 3319860990eSLisandro Dalcin ierr = PetscDLSym(NULL,fname,(void**)data);CHKERRQ(ierr); 3322d53ad75SBarry Smith #else 3339860990eSLisandro Dalcin *(void**)data = NULL; 3342d53ad75SBarry Smith #endif 33505acbc63SBarry Smith free(fname); 3362d53ad75SBarry Smith } 337e5c89e4eSSatish Balay PetscFunctionReturn(0); 338e5c89e4eSSatish Balay } 3399860990eSLisandro Dalcin 340a9acdec7SBarry Smith /*@C 341e5c89e4eSSatish Balay PetscBinaryWrite - Writes to a binary file. 342e5c89e4eSSatish Balay 343e5c89e4eSSatish Balay Not Collective 344e5c89e4eSSatish Balay 345e5c89e4eSSatish Balay Input Parameters: 346e5c89e4eSSatish Balay + fd - the file 347e5c89e4eSSatish Balay . p - the buffer 348e5c89e4eSSatish Balay . n - the number of items to write 349f253e43cSLisandro Dalcin - type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 350e5c89e4eSSatish Balay 351e5c89e4eSSatish Balay Level: advanced 352e5c89e4eSSatish Balay 353e5c89e4eSSatish Balay Notes: 354e5c89e4eSSatish Balay PetscBinaryWrite() uses byte swapping to work on all machines; the files 355f253e43cSLisandro Dalcin are written using big-endian ordering to the file. On little-endian machines the numbers 356e5c89e4eSSatish Balay are converted to the big-endian format when they are written to disk. 357f253e43cSLisandro Dalcin When PETSc is ./configure with --with-64-bit-indices the integers are written to the 358f253e43cSLisandro Dalcin file as 64 bit integers, this means they can only be read back in when the option --with-64-bit-indices 35954f21887SBarry Smith is used. 360e5c89e4eSSatish Balay 36141f502e3SPatrick Sanan If running with __float128 precision the output is in __float128 unless one uses the -binary_write_double option 3620da86b62SBarry Smith 363e5c89e4eSSatish Balay The Buffer p should be read-write buffer, and not static data. 364e5c89e4eSSatish Balay This way, byte-swapping is done in-place, and then the buffer is 365e5c89e4eSSatish Balay written to the file. 366e5c89e4eSSatish Balay 367e5c89e4eSSatish Balay This routine restores the original contents of the buffer, after 368e5c89e4eSSatish Balay it is written to the file. This is done by byte-swapping in-place 369f253e43cSLisandro Dalcin the second time. 370e5c89e4eSSatish Balay 371300a7f5bSBarry Smith Because byte-swapping may be done on the values in data it cannot be declared const 372300a7f5bSBarry Smith 373e5c89e4eSSatish Balay 3744ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(), 3754ebed01fSBarry Smith PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 376e5c89e4eSSatish Balay @*/ 377f253e43cSLisandro Dalcin PetscErrorCode PetscBinaryWrite(int fd,const void *p,PetscInt n,PetscDataType type) 378e5c89e4eSSatish Balay { 379f253e43cSLisandro Dalcin const char *pp = (char*)p; 380e5c89e4eSSatish Balay int err,wsize; 381e5c89e4eSSatish Balay size_t m = (size_t)n,maxblock=65536; 382e5c89e4eSSatish Balay PetscErrorCode ierr; 383f253e43cSLisandro Dalcin const void *ptmp = p; 38405acbc63SBarry Smith char *fname = NULL; 3850da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 386df40af56SBarry Smith PetscBool writedouble = PETSC_FALSE; 3870da86b62SBarry Smith double *ppp; 3880da86b62SBarry Smith PetscReal *pv; 3890da86b62SBarry Smith PetscInt i; 3900da86b62SBarry Smith #endif 39141f502e3SPatrick Sanan PetscDataType wtype = type; 392e5c89e4eSSatish Balay 393e5c89e4eSSatish Balay PetscFunctionBegin; 394e32f2f54SBarry Smith if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n); 395e5c89e4eSSatish Balay if (!n) PetscFunctionReturn(0); 396e5c89e4eSSatish Balay 3972d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 3982d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 3992d53ad75SBarry Smith const char *fnametmp; 4002d53ad75SBarry Smith #endif 4012d53ad75SBarry Smith m = 64; 402e25ab156SSatish Balay fname = (char*)malloc(m*sizeof(char)); 40305acbc63SBarry Smith if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name"); 40405acbc63SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 40505acbc63SBarry Smith if (n > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only binary view a single function at a time"); 40605acbc63SBarry Smith ierr = PetscFPTFind(*(void**)p,&fnametmp);CHKERRQ(ierr); 40705acbc63SBarry Smith ierr = PetscStrncpy(fname,fnametmp,m);CHKERRQ(ierr); 40805acbc63SBarry Smith #else 40905acbc63SBarry Smith ierr = PetscStrncpy(fname,"",m);CHKERRQ(ierr); 41005acbc63SBarry Smith #endif 41101963f56SBarry Smith wtype = PETSC_CHAR; 4122d53ad75SBarry Smith pp = (char*)fname; 4132d53ad75SBarry Smith ptmp = (void*)fname; 4142d53ad75SBarry Smith } 4152d53ad75SBarry Smith 4160da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 4170da86b62SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-binary_write_double",&writedouble,NULL);CHKERRQ(ierr); 4180da86b62SBarry Smith /* If using __float128 precision we still write in doubles to file */ 4194caf0332SSatish Balay if ((type == PETSC_SCALAR || type == PETSC_REAL) && writedouble) { 42041f502e3SPatrick Sanan wtype = PETSC_DOUBLE; 4210da86b62SBarry Smith ierr = PetscMalloc1(n,&ppp);CHKERRQ(ierr); 4220da86b62SBarry Smith pv = (PetscReal*)pp; 4230da86b62SBarry Smith for (i=0; i<n; i++) { 4240da86b62SBarry Smith ppp[i] = (double) pv[i]; 4250da86b62SBarry Smith } 4260da86b62SBarry Smith pp = (char*)ppp; 4270da86b62SBarry Smith ptmp = (char*)ppp; 4280da86b62SBarry Smith } 4290da86b62SBarry Smith #endif 4300da86b62SBarry Smith 43141f502e3SPatrick Sanan if (wtype == PETSC_INT) m *= sizeof(PetscInt); 43241f502e3SPatrick Sanan else if (wtype == PETSC_SCALAR) m *= sizeof(PetscScalar); 4334caf0332SSatish Balay else if (wtype == PETSC_REAL) m *= sizeof(PetscReal); 43441f502e3SPatrick Sanan else if (wtype == PETSC_DOUBLE) m *= sizeof(double); 43541f502e3SPatrick Sanan else if (wtype == PETSC_FLOAT) m *= sizeof(float); 43641f502e3SPatrick Sanan else if (wtype == PETSC_SHORT) m *= sizeof(short); 437972064b6SLisandro Dalcin else if (wtype == PETSC_LONG) m *= sizeof(long); 43841f502e3SPatrick Sanan else if (wtype == PETSC_CHAR) m *= sizeof(char); 43941f502e3SPatrick Sanan else if (wtype == PETSC_ENUM) m *= sizeof(PetscEnum); 44041f502e3SPatrick Sanan else if (wtype == PETSC_BOOL) m *= sizeof(PetscBool); 4419e3e4c22SLisandro Dalcin else if (wtype == PETSC_INT64) m *= sizeof(PetscInt64); 44241f502e3SPatrick Sanan else if (wtype == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char); 443e32f2f54SBarry Smith else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type"); 444e5c89e4eSSatish Balay 445f253e43cSLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap((void*)ptmp,wtype,n);CHKERRQ(ierr);} 446e5c89e4eSSatish Balay 447e5c89e4eSSatish Balay while (m) { 448e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 449e5c89e4eSSatish Balay err = write(fd,pp,wsize); 450e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 45104102261SBarry Smith if (err != wsize) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_FILE_WRITE,"Error writing to file total size %d err %d wsize %d",(int)n,(int)err,(int)wsize); 452e5c89e4eSSatish Balay m -= wsize; 453e5c89e4eSSatish Balay pp += wsize; 454e5c89e4eSSatish Balay } 455e5c89e4eSSatish Balay 456f253e43cSLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap((void*)ptmp,wtype,n);CHKERRQ(ierr);} 457f253e43cSLisandro Dalcin 45805acbc63SBarry Smith if (type == PETSC_FUNCTION) { 45905acbc63SBarry Smith free(fname); 46005acbc63SBarry Smith } 4610da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 4624caf0332SSatish Balay if ((type == PETSC_SCALAR || type == PETSC_REAL) && writedouble) { 4630da86b62SBarry Smith ierr = PetscFree(ppp);CHKERRQ(ierr); 4640da86b62SBarry Smith } 4650da86b62SBarry Smith #endif 466e5c89e4eSSatish Balay PetscFunctionReturn(0); 467e5c89e4eSSatish Balay } 468e5c89e4eSSatish Balay 469e5c89e4eSSatish Balay /*@C 470e5c89e4eSSatish Balay PetscBinaryOpen - Opens a PETSc binary file. 471e5c89e4eSSatish Balay 472e5c89e4eSSatish Balay Not Collective 473e5c89e4eSSatish Balay 474e5c89e4eSSatish Balay Input Parameters: 475e5c89e4eSSatish Balay + name - filename 476*cc843e7aSLisandro Dalcin - mode - open mode of binary file, one of FILE_MODE_READ, FILE_MODE_WRITE, FILE_MODE_APPEND 477e5c89e4eSSatish Balay 478e5c89e4eSSatish Balay Output Parameter: 479e5c89e4eSSatish Balay . fd - the file 480e5c89e4eSSatish Balay 481e5c89e4eSSatish Balay Level: advanced 482e5c89e4eSSatish Balay 483e5c89e4eSSatish Balay 48495452b02SPatrick Sanan Notes: 48595452b02SPatrick Sanan Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in 486e5c89e4eSSatish Balay big-endian format. This means the file can be accessed using PetscBinaryOpen() and 487e5c89e4eSSatish Balay PetscBinaryRead() and PetscBinaryWrite() on any machine. 488e5c89e4eSSatish Balay 4894ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(), 4904ebed01fSBarry Smith PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek() 49145c64e65SBarry Smith 492e5c89e4eSSatish Balay @*/ 4937087cfbeSBarry Smith PetscErrorCode PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd) 494e5c89e4eSSatish Balay { 495e5c89e4eSSatish Balay PetscFunctionBegin; 496*cc843e7aSLisandro Dalcin switch (mode) { 497*cc843e7aSLisandro Dalcin case FILE_MODE_READ: *fd = open(name,O_BINARY|O_RDONLY,0); break; 498*cc843e7aSLisandro Dalcin case FILE_MODE_WRITE: *fd = open(name,O_BINARY|O_WRONLY|O_CREAT|O_TRUNC,0666); break; 499*cc843e7aSLisandro Dalcin case FILE_MODE_APPEND: *fd = open(name,O_BINARY|O_WRONLY|O_APPEND,0); break; 500*cc843e7aSLisandro Dalcin default: SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported file mode %s",PetscFileModes[mode]); 501e5c89e4eSSatish Balay } 502*cc843e7aSLisandro Dalcin if (*fd == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for %s: %s",name,PetscFileModes[mode]); 503e5c89e4eSSatish Balay PetscFunctionReturn(0); 504e5c89e4eSSatish Balay } 505e5c89e4eSSatish Balay 506e30d2299SSatish Balay /*@ 507e5c89e4eSSatish Balay PetscBinaryClose - Closes a PETSc binary file. 508e5c89e4eSSatish Balay 509e5c89e4eSSatish Balay Not Collective 510e5c89e4eSSatish Balay 511e5c89e4eSSatish Balay Output Parameter: 512e5c89e4eSSatish Balay . fd - the file 513e5c89e4eSSatish Balay 514e5c89e4eSSatish Balay Level: advanced 515e5c89e4eSSatish Balay 5164ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 5174ebed01fSBarry Smith PetscBinarySynchronizedSeek() 518e5c89e4eSSatish Balay @*/ 5197087cfbeSBarry Smith PetscErrorCode PetscBinaryClose(int fd) 520e5c89e4eSSatish Balay { 521e5c89e4eSSatish Balay PetscFunctionBegin; 522*cc843e7aSLisandro Dalcin if (close(fd)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"close() failed on file descriptor"); 523e5c89e4eSSatish Balay PetscFunctionReturn(0); 524e5c89e4eSSatish Balay } 525e5c89e4eSSatish Balay 526e5c89e4eSSatish Balay 527e8976759SBarry Smith /*@C 528e5c89e4eSSatish Balay PetscBinarySeek - Moves the file pointer on a PETSc binary file. 529e5c89e4eSSatish Balay 530e5c89e4eSSatish Balay Not Collective 531e5c89e4eSSatish Balay 532e5c89e4eSSatish Balay Input Parameters: 533e5c89e4eSSatish Balay + fd - the file 534ff553b35SMatthew Knepley . off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 535e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 536ff553b35SMatthew Knepley - whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file 537ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_CUR then off is an offset from the current location 538ff553b35SMatthew Knepley if PETSC_BINARY_SEEK_END then off is an offset from the end of file 539e5c89e4eSSatish Balay 540e5c89e4eSSatish Balay Output Parameter: 541e5c89e4eSSatish Balay . offset - new offset in file 542e5c89e4eSSatish Balay 543e5c89e4eSSatish Balay Level: developer 544e5c89e4eSSatish Balay 545e5c89e4eSSatish Balay Notes: 546e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 547e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 548e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 549e5c89e4eSSatish Balay to determine the offset or location. 550e5c89e4eSSatish Balay 551e5c89e4eSSatish Balay 5524ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 5534ebed01fSBarry Smith PetscBinarySynchronizedSeek() 554e5c89e4eSSatish Balay @*/ 5557087cfbeSBarry Smith PetscErrorCode PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 556e5c89e4eSSatish Balay { 557e5c89e4eSSatish Balay int iwhence = 0; 558e5c89e4eSSatish Balay 559e5c89e4eSSatish Balay PetscFunctionBegin; 560a297a907SKarl Rupp if (whence == PETSC_BINARY_SEEK_SET) iwhence = SEEK_SET; 561a297a907SKarl Rupp else if (whence == PETSC_BINARY_SEEK_CUR) iwhence = SEEK_CUR; 562a297a907SKarl Rupp else if (whence == PETSC_BINARY_SEEK_END) iwhence = SEEK_END; 563a297a907SKarl Rupp else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location"); 564e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) 565e5c89e4eSSatish Balay *offset = lseek(fd,off,iwhence); 566e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK) 567e5c89e4eSSatish Balay *offset = _lseek(fd,(long)off,iwhence); 568e5c89e4eSSatish Balay #else 569e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file"); 570e5c89e4eSSatish Balay #endif 571e5c89e4eSSatish Balay PetscFunctionReturn(0); 572e5c89e4eSSatish Balay } 573e5c89e4eSSatish Balay 574e5c89e4eSSatish Balay /*@C 5751d280d73SBarry Smith PetscBinarySynchronizedRead - Reads from a binary file. 576e5c89e4eSSatish Balay 577d083f849SBarry Smith Collective 578e5c89e4eSSatish Balay 579e5c89e4eSSatish Balay Input Parameters: 580e5c89e4eSSatish Balay + comm - the MPI communicator 5819860990eSLisandro Dalcin . fd - the file descriptor 5829860990eSLisandro Dalcin . num - the maximum number of items to read 5839860990eSLisandro Dalcin - type - the type of items to read (PETSC_INT, PETSC_REAL, PETSC_SCALAR, etc.) 584e5c89e4eSSatish Balay 585e5c89e4eSSatish Balay Output Parameters: 5869860990eSLisandro Dalcin + data - the buffer 5879860990eSLisandro Dalcin - count - the number of items read, optional 588e5c89e4eSSatish Balay 589e5c89e4eSSatish Balay Level: developer 590e5c89e4eSSatish Balay 591e5c89e4eSSatish Balay Notes: 592e5c89e4eSSatish Balay Does a PetscBinaryRead() followed by an MPI_Bcast() 593e5c89e4eSSatish Balay 5949860990eSLisandro Dalcin If count is not provided and the number of items read is less than 5959860990eSLisandro Dalcin the maximum number of items to read, then this routine errors. 5969860990eSLisandro Dalcin 5971d280d73SBarry Smith PetscBinarySynchronizedRead() uses byte swapping to work on all machines. 598e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 599e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 600e5c89e4eSSatish Balay binary file may be read on any machine. 601e5c89e4eSSatish Balay 602e5c89e4eSSatish Balay 6034ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(), 6044ebed01fSBarry Smith PetscBinarySynchronizedSeek() 605e5c89e4eSSatish Balay @*/ 6069860990eSLisandro Dalcin PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type) 607e5c89e4eSSatish Balay { 6089860990eSLisandro Dalcin PetscErrorCode ierr; 609e5c89e4eSSatish Balay PetscMPIInt rank; 610e5c89e4eSSatish Balay MPI_Datatype mtype; 6119860990eSLisandro Dalcin PetscInt ibuf[2] = {0, 0}; 61205acbc63SBarry Smith char *fname = NULL; 6139860990eSLisandro Dalcin void *fptr = NULL; 614e5c89e4eSSatish Balay 615e5c89e4eSSatish Balay PetscFunctionBegin; 6162d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 6179860990eSLisandro Dalcin num = 64; 6182d53ad75SBarry Smith type = PETSC_CHAR; 6199860990eSLisandro Dalcin fname = (char*)malloc(num*sizeof(char)); 6209860990eSLisandro Dalcin fptr = data; 6219860990eSLisandro Dalcin data = (void*)fname; 62205acbc63SBarry Smith if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name"); 6232d53ad75SBarry Smith } 6242d53ad75SBarry Smith 625e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 626e5c89e4eSSatish Balay if (!rank) { 6279860990eSLisandro Dalcin ibuf[0] = PetscBinaryRead(fd,data,num,count?&ibuf[1]:NULL,type); 628e5c89e4eSSatish Balay } 6299860990eSLisandro Dalcin ierr = MPI_Bcast(ibuf,2,MPIU_INT,0,comm);CHKERRQ(ierr); 6309860990eSLisandro Dalcin ierr = (PetscErrorCode)ibuf[0];CHKERRQ(ierr); 631e5c89e4eSSatish Balay ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr); 6329860990eSLisandro Dalcin ierr = MPI_Bcast(data,count?ibuf[1]:num,mtype,0,comm);CHKERRQ(ierr); 6339860990eSLisandro Dalcin if (count) *count = ibuf[1]; 6342d53ad75SBarry Smith 635e366c363SBarry Smith if (type == PETSC_FUNCTION) { 6362d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 6379860990eSLisandro Dalcin ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)fptr);CHKERRQ(ierr); 6382d53ad75SBarry Smith #else 6399860990eSLisandro Dalcin *(void**)fptr = NULL; 6402d53ad75SBarry Smith #endif 641e366c363SBarry Smith free(fname); 6422d53ad75SBarry Smith } 643e5c89e4eSSatish Balay PetscFunctionReturn(0); 644e5c89e4eSSatish Balay } 645e5c89e4eSSatish Balay 646e5c89e4eSSatish Balay /*@C 6471d280d73SBarry Smith PetscBinarySynchronizedWrite - writes to a binary file. 648e5c89e4eSSatish Balay 649d083f849SBarry Smith Collective 650e5c89e4eSSatish Balay 651e5c89e4eSSatish Balay Input Parameters: 652e5c89e4eSSatish Balay + comm - the MPI communicator 653e5c89e4eSSatish Balay . fd - the file 654e5c89e4eSSatish Balay . n - the number of items to write 655e5c89e4eSSatish Balay . p - the buffer 656e5c89e4eSSatish Balay - type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR) 657e5c89e4eSSatish Balay 658e5c89e4eSSatish Balay Level: developer 659e5c89e4eSSatish Balay 660e5c89e4eSSatish Balay Notes: 661e5c89e4eSSatish Balay Process 0 does a PetscBinaryWrite() 662e5c89e4eSSatish Balay 6631d280d73SBarry Smith PetscBinarySynchronizedWrite() uses byte swapping to work on all machines. 664e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 665e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 666e5c89e4eSSatish Balay binary file may be read on any machine. 667e5c89e4eSSatish Balay 66895452b02SPatrick Sanan Notes: 66995452b02SPatrick Sanan because byte-swapping may be done on the values in data it cannot be declared const 670300a7f5bSBarry Smith 6711d280d73SBarry Smith WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0, 6721d280d73SBarry Smith while PetscSynchronizedFPrintf() has all processes print their strings in order. 6731d280d73SBarry Smith 674e5c89e4eSSatish Balay 6754ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(), 6764ebed01fSBarry Smith PetscBinarySynchronizedSeek() 677e5c89e4eSSatish Balay @*/ 678f253e43cSLisandro Dalcin PetscErrorCode PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,const void *p,PetscInt n,PetscDataType type) 679e5c89e4eSSatish Balay { 680e5c89e4eSSatish Balay PetscErrorCode ierr; 681e5c89e4eSSatish Balay PetscMPIInt rank; 682e5c89e4eSSatish Balay 683e5c89e4eSSatish Balay PetscFunctionBegin; 684e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 685e5c89e4eSSatish Balay if (!rank) { 686f253e43cSLisandro Dalcin ierr = PetscBinaryWrite(fd,p,n,type);CHKERRQ(ierr); 687e5c89e4eSSatish Balay } 688e5c89e4eSSatish Balay PetscFunctionReturn(0); 689e5c89e4eSSatish Balay } 690e5c89e4eSSatish Balay 691e5c89e4eSSatish Balay /*@C 6921d280d73SBarry Smith PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file. 693e5c89e4eSSatish Balay 694e5c89e4eSSatish Balay 695e5c89e4eSSatish Balay Input Parameters: 696e5c89e4eSSatish Balay + fd - the file 697e5c89e4eSSatish Balay . whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file 698e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_CUR then size is offset from current location 699e5c89e4eSSatish Balay if PETSC_BINARY_SEEK_END then size is offset from end of file 700e5c89e4eSSatish Balay - off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE, 701e5c89e4eSSatish Balay etc. in your calculation rather than sizeof() to compute byte lengths. 702e5c89e4eSSatish Balay 703e5c89e4eSSatish Balay Output Parameter: 704e5c89e4eSSatish Balay . offset - new offset in file 705e5c89e4eSSatish Balay 706e5c89e4eSSatish Balay Level: developer 707e5c89e4eSSatish Balay 708e5c89e4eSSatish Balay Notes: 709e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 710e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 711e5c89e4eSSatish Balay binary file may be read on any machine. Hence you CANNOT use sizeof() 712e5c89e4eSSatish Balay to determine the offset or location. 713e5c89e4eSSatish Balay 714e5c89e4eSSatish Balay 7154ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), 7164ebed01fSBarry Smith PetscBinarySynchronizedSeek() 717e5c89e4eSSatish Balay @*/ 7187087cfbeSBarry Smith PetscErrorCode PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset) 719e5c89e4eSSatish Balay { 720e5c89e4eSSatish Balay PetscErrorCode ierr; 721e5c89e4eSSatish Balay PetscMPIInt rank; 722e5c89e4eSSatish Balay 723e5c89e4eSSatish Balay PetscFunctionBegin; 724e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 725e5c89e4eSSatish Balay if (!rank) { 726e5c89e4eSSatish Balay ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr); 727e5c89e4eSSatish Balay } 728e5c89e4eSSatish Balay PetscFunctionReturn(0); 729e5c89e4eSSatish Balay } 730e5c89e4eSSatish Balay 7310fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO) 732e39fd77fSBarry Smith 733951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32) 734e39fd77fSBarry Smith /* 735e39fd77fSBarry Smith MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions. 736e39fd77fSBarry Smith These are set into MPI in PetscInitialize() via MPI_Register_datarep() 737e39fd77fSBarry Smith 738e39fd77fSBarry Smith Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode) 739e39fd77fSBarry Smith 740951e3c8eSBarry Smith The next three routines are not used because MPICH does not support their use 741e39fd77fSBarry Smith 742e39fd77fSBarry Smith */ 7438cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state) 744e39fd77fSBarry Smith { 745e39fd77fSBarry Smith MPI_Aint ub; 746e39fd77fSBarry Smith PetscMPIInt ierr; 747e39fd77fSBarry Smith 748e39fd77fSBarry Smith ierr = MPI_Type_get_extent(datatype,&ub,file_extent); 749e39fd77fSBarry Smith return ierr; 750e39fd77fSBarry Smith } 751e39fd77fSBarry Smith 7528cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 753e39fd77fSBarry Smith { 754e39fd77fSBarry Smith PetscDataType pdtype; 755e39fd77fSBarry Smith PetscMPIInt ierr; 756e39fd77fSBarry Smith size_t dsize; 757e39fd77fSBarry Smith 758e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 759e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 760e39fd77fSBarry Smith 761e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 762e39fd77fSBarry Smith userbuf = ((char*)userbuf) + dsize*position; 763e39fd77fSBarry Smith 764e39fd77fSBarry Smith ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr); 76530815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr);} 766e39fd77fSBarry Smith return ierr; 767e39fd77fSBarry Smith } 768e39fd77fSBarry Smith 769e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state) 770e39fd77fSBarry Smith { 771e39fd77fSBarry Smith PetscDataType pdtype; 772e39fd77fSBarry Smith PetscMPIInt ierr; 773e39fd77fSBarry Smith size_t dsize; 774e39fd77fSBarry Smith 775e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr); 776e39fd77fSBarry Smith ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr); 777e39fd77fSBarry Smith 778e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 779e39fd77fSBarry Smith userbuf = ((char*)userbuf) + dsize*position; 780e39fd77fSBarry Smith 781e39fd77fSBarry Smith ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr); 78230815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr);} 783e39fd77fSBarry Smith return ierr; 784e39fd77fSBarry Smith } 785951e3c8eSBarry Smith #endif 786e39fd77fSBarry Smith 787e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 788e39fd77fSBarry Smith { 789e39fd77fSBarry Smith PetscDataType pdtype; 79030815ce0SLisandro Dalcin PetscErrorCode ierr; 79130815ce0SLisandro Dalcin 792e39fd77fSBarry Smith 793e39fd77fSBarry Smith PetscFunctionBegin; 794e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 79530815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 796e39fd77fSBarry Smith ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 79730815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 798a6796414SBarry Smith PetscFunctionReturn(0); 799e39fd77fSBarry Smith } 800e39fd77fSBarry Smith 801e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 802e39fd77fSBarry Smith { 803e39fd77fSBarry Smith PetscDataType pdtype; 80430815ce0SLisandro Dalcin PetscErrorCode ierr; 805e39fd77fSBarry Smith 806e39fd77fSBarry Smith PetscFunctionBegin; 807e39fd77fSBarry Smith ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 808e39fd77fSBarry Smith ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr); 80930815ce0SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 810a6796414SBarry Smith PetscFunctionReturn(0); 811e39fd77fSBarry Smith } 81230815ce0SLisandro Dalcin 813c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 814c4e82887SLisandro Dalcin { 815c4e82887SLisandro Dalcin PetscDataType pdtype; 816c4e82887SLisandro Dalcin PetscErrorCode ierr; 817c4e82887SLisandro Dalcin 818c4e82887SLisandro Dalcin 819c4e82887SLisandro Dalcin PetscFunctionBegin; 820c4e82887SLisandro Dalcin ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 821c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 822c4e82887SLisandro Dalcin ierr = MPI_File_write_at(fd,off,data,cnt,dtype,status);CHKERRQ(ierr); 823c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 824c4e82887SLisandro Dalcin PetscFunctionReturn(0); 825c4e82887SLisandro Dalcin } 826c4e82887SLisandro Dalcin 827c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 828c4e82887SLisandro Dalcin { 829c4e82887SLisandro Dalcin PetscDataType pdtype; 830c4e82887SLisandro Dalcin PetscErrorCode ierr; 831c4e82887SLisandro Dalcin 832c4e82887SLisandro Dalcin PetscFunctionBegin; 833c4e82887SLisandro Dalcin ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 834c4e82887SLisandro Dalcin ierr = MPI_File_read_at(fd,off,data,cnt,dtype,status);CHKERRQ(ierr); 835c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 836c4e82887SLisandro Dalcin PetscFunctionReturn(0); 837c4e82887SLisandro Dalcin } 838c4e82887SLisandro Dalcin 839c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 840c4e82887SLisandro Dalcin { 841c4e82887SLisandro Dalcin PetscDataType pdtype; 842c4e82887SLisandro Dalcin PetscErrorCode ierr; 843c4e82887SLisandro Dalcin 844c4e82887SLisandro Dalcin 845c4e82887SLisandro Dalcin PetscFunctionBegin; 846c4e82887SLisandro Dalcin ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 847c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 848c4e82887SLisandro Dalcin ierr = MPI_File_write_at_all(fd,off,data,cnt,dtype,status);CHKERRQ(ierr); 849c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 850c4e82887SLisandro Dalcin PetscFunctionReturn(0); 851c4e82887SLisandro Dalcin } 852c4e82887SLisandro Dalcin 853c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status) 854c4e82887SLisandro Dalcin { 855c4e82887SLisandro Dalcin PetscDataType pdtype; 856c4e82887SLisandro Dalcin PetscErrorCode ierr; 857c4e82887SLisandro Dalcin 858c4e82887SLisandro Dalcin PetscFunctionBegin; 859c4e82887SLisandro Dalcin ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr); 860c4e82887SLisandro Dalcin ierr = MPI_File_read_at_all(fd,off,data,cnt,dtype,status);CHKERRQ(ierr); 861c4e82887SLisandro Dalcin if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);} 862c4e82887SLisandro Dalcin PetscFunctionReturn(0); 863c4e82887SLisandro Dalcin } 864c4e82887SLisandro Dalcin 865951e3c8eSBarry Smith #endif 866