1e5c89e4eSSatish Balay 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay This file contains simple binary read/write routines. 4e5c89e4eSSatish Balay */ 5e5c89e4eSSatish Balay 6c6db04a5SJed Brown #include <petscsys.h> 7cc843e7aSLisandro 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 16cc843e7aSLisandro Dalcin #if !defined(PETSC_HAVE_O_BINARY) 17cc843e7aSLisandro Dalcin #define O_BINARY 0 18cc843e7aSLisandro 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 */ 27d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapEnum(PetscEnum *buff, PetscInt n) 28d71ae5a4SJacob Faibussowitsch { 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 } 393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 40e5c89e4eSSatish Balay } 416de02169SBarry Smith 426de02169SBarry Smith /* 43acfcf0e5SJed Brown PetscByteSwapBool - Swap bytes in a PETSc Bool 446de02169SBarry Smith 456de02169SBarry Smith */ 46d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapBool(PetscBool *buff, PetscInt n) 47d71ae5a4SJacob Faibussowitsch { 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 } 583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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 */ 65d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapInt(PetscInt *buff, PetscInt n) 66d71ae5a4SJacob Faibussowitsch { 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 } 763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 77bd1d2e58SBarry Smith } 789e3e4c22SLisandro Dalcin 799e3e4c22SLisandro Dalcin /* 809e3e4c22SLisandro Dalcin PetscByteSwapInt64 - Swap bytes in a PETSc integer (64 bits) 819e3e4c22SLisandro Dalcin 829e3e4c22SLisandro Dalcin */ 83d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapInt64(PetscInt64 *buff, PetscInt n) 84d71ae5a4SJacob Faibussowitsch { 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 } 953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 969e3e4c22SLisandro Dalcin } 979e3e4c22SLisandro Dalcin 98bd1d2e58SBarry Smith /* --------------------------------------------------------- */ 99e5c89e4eSSatish Balay /* 100e5c89e4eSSatish Balay PetscByteSwapShort - Swap bytes in a short 101e5c89e4eSSatish Balay */ 102d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapShort(short *buff, PetscInt n) 103d71ae5a4SJacob Faibussowitsch { 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 } 1143ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 115e5c89e4eSSatish Balay } 116972064b6SLisandro Dalcin /* 117972064b6SLisandro Dalcin PetscByteSwapLong - Swap bytes in a long 118972064b6SLisandro Dalcin */ 119d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapLong(long *buff, PetscInt n) 120d71ae5a4SJacob Faibussowitsch { 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 } 1313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 132972064b6SLisandro Dalcin } 133e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 134e5c89e4eSSatish Balay /* 1354caf0332SSatish Balay PetscByteSwapReal - Swap bytes in a PetscReal 1364caf0332SSatish Balay */ 137d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapReal(PetscReal *buff, PetscInt n) 138d71ae5a4SJacob Faibussowitsch { 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 } 1493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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 */ 156d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapScalar(PetscScalar *buff, PetscInt n) 157d71ae5a4SJacob Faibussowitsch { 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 } 1713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 172e5c89e4eSSatish Balay } 173e5c89e4eSSatish Balay /* --------------------------------------------------------- */ 174e5c89e4eSSatish Balay /* 175e5c89e4eSSatish Balay PetscByteSwapDouble - Swap bytes in a double 176e5c89e4eSSatish Balay */ 177d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapDouble(double *buff, PetscInt n) 178d71ae5a4SJacob Faibussowitsch { 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 } 1893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 190e5c89e4eSSatish Balay } 191e39fd77fSBarry Smith 192e95bf02fSSatish Balay /* 193e95bf02fSSatish Balay PetscByteSwapFloat - Swap bytes in a float 194e95bf02fSSatish Balay */ 195d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwapFloat(float *buff, PetscInt n) 196d71ae5a4SJacob Faibussowitsch { 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 } 2073ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 208e95bf02fSSatish Balay } 209e95bf02fSSatish Balay 210d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscByteSwap(void *data, PetscDataType pdtype, PetscInt count) 211d71ae5a4SJacob Faibussowitsch { 212e39fd77fSBarry Smith PetscFunctionBegin; 2139566063dSJacob Faibussowitsch if (pdtype == PETSC_INT) PetscCall(PetscByteSwapInt((PetscInt *)data, count)); 2149566063dSJacob Faibussowitsch else if (pdtype == PETSC_ENUM) PetscCall(PetscByteSwapEnum((PetscEnum *)data, count)); 2159566063dSJacob Faibussowitsch else if (pdtype == PETSC_BOOL) PetscCall(PetscByteSwapBool((PetscBool *)data, count)); 2169566063dSJacob Faibussowitsch else if (pdtype == PETSC_SCALAR) PetscCall(PetscByteSwapScalar((PetscScalar *)data, count)); 2179566063dSJacob Faibussowitsch else if (pdtype == PETSC_REAL) PetscCall(PetscByteSwapReal((PetscReal *)data, count)); 2189566063dSJacob Faibussowitsch else if (pdtype == PETSC_COMPLEX) PetscCall(PetscByteSwapReal((PetscReal *)data, 2 * count)); 2199566063dSJacob Faibussowitsch else if (pdtype == PETSC_INT64) PetscCall(PetscByteSwapInt64((PetscInt64 *)data, count)); 2209566063dSJacob Faibussowitsch else if (pdtype == PETSC_DOUBLE) PetscCall(PetscByteSwapDouble((double *)data, count)); 2219566063dSJacob Faibussowitsch else if (pdtype == PETSC_FLOAT) PetscCall(PetscByteSwapFloat((float *)data, count)); 2229566063dSJacob Faibussowitsch else if (pdtype == PETSC_SHORT) PetscCall(PetscByteSwapShort((short *)data, count)); 2239566063dSJacob Faibussowitsch else if (pdtype == PETSC_LONG) PetscCall(PetscByteSwapLong((long *)data, count)); 2243ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 225e39fd77fSBarry Smith } 226e39fd77fSBarry Smith 2279860990eSLisandro Dalcin /*@C 228e5c89e4eSSatish Balay PetscBinaryRead - Reads from a binary file. 229e5c89e4eSSatish Balay 230e5c89e4eSSatish Balay Not Collective 231e5c89e4eSSatish Balay 232e5c89e4eSSatish Balay Input Parameters: 2339860990eSLisandro Dalcin + fd - the file descriptor 2349860990eSLisandro Dalcin . num - the maximum number of items to read 235811af0c4SBarry Smith - type - the type of items to read (`PETSC_INT`, `PETSC_REAL`, `PETSC_SCALAR`, etc.) 236e5c89e4eSSatish Balay 237e5c89e4eSSatish Balay Output Parameters: 2389860990eSLisandro Dalcin + data - the buffer 2399860990eSLisandro Dalcin - count - the number of items read, optional 240e5c89e4eSSatish Balay 241e5c89e4eSSatish Balay Level: developer 242e5c89e4eSSatish Balay 243e5c89e4eSSatish Balay Notes: 2449860990eSLisandro Dalcin If count is not provided and the number of items read is less than 2459860990eSLisandro Dalcin the maximum number of items to read, then this routine errors. 2469860990eSLisandro Dalcin 247811af0c4SBarry Smith `PetscBinaryRead()` uses byte swapping to work on all machines; the files 248f253e43cSLisandro Dalcin are written to file ALWAYS using big-endian ordering. On little-endian machines the numbers 249f253e43cSLisandro Dalcin are converted to the little-endian format when they are read in from the file. 250f253e43cSLisandro Dalcin When PETSc is ./configure with --with-64-bit-indices the integers are written to the 251*bfbbc7b7SBarry Smith file as 64 bit integers, this means they can only be read back in when the option `--with-64-bit-indices` 25254f21887SBarry Smith is used. 253e5c89e4eSSatish Balay 254db781477SPatrick Sanan .seealso: `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscViewerBinaryGetDescriptor()`, `PetscBinarySynchronizedWrite()`, 255db781477SPatrick Sanan `PetscBinarySynchronizedRead()`, `PetscBinarySynchronizedSeek()` 256e5c89e4eSSatish Balay @*/ 257d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinaryRead(int fd, void *data, PetscInt num, PetscInt *count, PetscDataType type) 258d71ae5a4SJacob Faibussowitsch { 2599860990eSLisandro Dalcin size_t typesize, m = (size_t)num, n = 0, maxblock = 65536; 2609860990eSLisandro Dalcin char *p = (char *)data; 2617a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 262cba51d77SBarry Smith PetscBool readdouble = PETSC_FALSE; 2639860990eSLisandro Dalcin double *pdouble; 2647a881295SBarry Smith #endif 2659860990eSLisandro Dalcin void *ptmp = data; 26605acbc63SBarry Smith char *fname = NULL; 267e5c89e4eSSatish Balay 268e5c89e4eSSatish Balay PetscFunctionBegin; 2699860990eSLisandro Dalcin if (count) *count = 0; 27008401ef6SPierre Jolivet PetscCheck(num >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Trying to read a negative amount of data %" PetscInt_FMT, num); 2713ba16761SJacob Faibussowitsch if (!num) PetscFunctionReturn(PETSC_SUCCESS); 272e5c89e4eSSatish Balay 2732d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 2742d53ad75SBarry Smith m = 64; 2752d53ad75SBarry Smith type = PETSC_CHAR; 27605acbc63SBarry Smith fname = (char *)malloc(m * sizeof(char)); 2779860990eSLisandro Dalcin p = (char *)fname; 2782d53ad75SBarry Smith ptmp = (void *)fname; 27928b400f6SJacob Faibussowitsch PetscCheck(fname, PETSC_COMM_SELF, PETSC_ERR_MEM, "Cannot allocate space for function name"); 2802d53ad75SBarry Smith } 2819860990eSLisandro Dalcin if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m); 2822d53ad75SBarry Smith 2839566063dSJacob Faibussowitsch PetscCall(PetscDataTypeGetSize(type, &typesize)); 284e5c89e4eSSatish Balay 2857a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 2869566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-binary_read_double", &readdouble, NULL)); 2877a881295SBarry Smith /* If using __float128 precision we still read in doubles from file */ 2889860990eSLisandro Dalcin if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) { 2899860990eSLisandro Dalcin PetscInt cnt = num * ((type == PETSC_REAL) ? 1 : 2); 2909566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(cnt, &pdouble)); 2919860990eSLisandro Dalcin p = (char *)pdouble; 2929860990eSLisandro Dalcin typesize /= 2; 2937a881295SBarry Smith } 2947a881295SBarry Smith #endif 2957a881295SBarry Smith 2969860990eSLisandro Dalcin m *= typesize; 2979860990eSLisandro Dalcin 298e5c89e4eSSatish Balay while (m) { 2999860990eSLisandro Dalcin size_t len = (m < maxblock) ? m : maxblock; 3009860990eSLisandro Dalcin int ret = (int)read(fd, p, len); 3019860990eSLisandro Dalcin if (ret < 0 && errno == EINTR) continue; 3029860990eSLisandro Dalcin if (!ret && len > 0) break; /* Proxy for EOF */ 30308401ef6SPierre Jolivet PetscCheck(ret >= 0, PETSC_COMM_SELF, PETSC_ERR_FILE_READ, "Error reading from file, errno %d", errno); 304ff9f933bSBarry Smith m -= (size_t)ret; 3059860990eSLisandro Dalcin p += ret; 306ff9f933bSBarry Smith n += (size_t)ret; 307e5c89e4eSSatish Balay } 30808401ef6SPierre Jolivet PetscCheck(!m || count, PETSC_COMM_SELF, PETSC_ERR_FILE_READ, "Read past end of file"); 3099860990eSLisandro Dalcin 3109860990eSLisandro Dalcin num = (PetscInt)(n / typesize); /* Should we require `n % typesize == 0` ? */ 3119860990eSLisandro Dalcin if (count) *count = num; /* TODO: This is most likely wrong for PETSC_BIT_LOGICAL */ 3127a881295SBarry Smith 3137a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 3149860990eSLisandro Dalcin if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) { 3159860990eSLisandro Dalcin PetscInt i, cnt = num * ((type == PETSC_REAL) ? 1 : 2); 3169860990eSLisandro Dalcin PetscReal *preal = (PetscReal *)data; 3179566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwapDouble(pdouble, cnt)); 3189860990eSLisandro Dalcin for (i = 0; i < cnt; i++) preal[i] = pdouble[i]; 3199566063dSJacob Faibussowitsch PetscCall(PetscFree(pdouble)); 3203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3217a881295SBarry Smith } 3227a881295SBarry Smith #endif 3237a881295SBarry Smith 3249566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(ptmp, type, num)); 325e5c89e4eSSatish Balay 32605acbc63SBarry Smith if (type == PETSC_FUNCTION) { 3272d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 3289566063dSJacob Faibussowitsch PetscCall(PetscDLSym(NULL, fname, (void **)data)); 3292d53ad75SBarry Smith #else 3309860990eSLisandro Dalcin *(void **)data = NULL; 3312d53ad75SBarry Smith #endif 33205acbc63SBarry Smith free(fname); 3332d53ad75SBarry Smith } 3343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 335e5c89e4eSSatish Balay } 3369860990eSLisandro Dalcin 337a9acdec7SBarry Smith /*@C 338e5c89e4eSSatish Balay PetscBinaryWrite - Writes to a binary file. 339e5c89e4eSSatish Balay 340e5c89e4eSSatish Balay Not Collective 341e5c89e4eSSatish Balay 342e5c89e4eSSatish Balay Input Parameters: 343e5c89e4eSSatish Balay + fd - the file 344e5c89e4eSSatish Balay . p - the buffer 345e5c89e4eSSatish Balay . n - the number of items to write 346811af0c4SBarry Smith - type - the type of items to read (`PETSC_INT`, `PETSC_REAL` or `PETSC_SCALAR`) 347e5c89e4eSSatish Balay 348e5c89e4eSSatish Balay Level: advanced 349e5c89e4eSSatish Balay 350e5c89e4eSSatish Balay Notes: 351811af0c4SBarry Smith `PetscBinaryWrite()` uses byte swapping to work on all machines; the files 352f253e43cSLisandro Dalcin are written using big-endian ordering to the file. On little-endian machines the numbers 353e5c89e4eSSatish Balay are converted to the big-endian format when they are written to disk. 354f253e43cSLisandro Dalcin When PETSc is ./configure with --with-64-bit-indices the integers are written to the 355*bfbbc7b7SBarry Smith file as 64 bit integers, this means they can only be read back in when the option `--with-64-bit-indices` 35654f21887SBarry Smith is used. 357e5c89e4eSSatish Balay 358*bfbbc7b7SBarry Smith If running with `__float128` precision the output is in `__float128` unless one uses the `-binary_write_double` option 3590da86b62SBarry Smith 360*bfbbc7b7SBarry Smith The buffer `p` should be read-write buffer, and not static data. 361e5c89e4eSSatish Balay This way, byte-swapping is done in-place, and then the buffer is 362e5c89e4eSSatish Balay written to the file. 363e5c89e4eSSatish Balay 364e5c89e4eSSatish Balay This routine restores the original contents of the buffer, after 365e5c89e4eSSatish Balay it is written to the file. This is done by byte-swapping in-place 366f253e43cSLisandro Dalcin the second time. 367e5c89e4eSSatish Balay 368300a7f5bSBarry Smith Because byte-swapping may be done on the values in data it cannot be declared const 369300a7f5bSBarry Smith 370db781477SPatrick Sanan .seealso: `PetscBinaryRead()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscViewerBinaryGetDescriptor()`, `PetscBinarySynchronizedWrite()`, 371db781477SPatrick Sanan `PetscBinarySynchronizedRead()`, `PetscBinarySynchronizedSeek()` 372e5c89e4eSSatish Balay @*/ 373d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinaryWrite(int fd, const void *p, PetscInt n, PetscDataType type) 374d71ae5a4SJacob Faibussowitsch { 375f253e43cSLisandro Dalcin const char *pp = (char *)p; 376e5c89e4eSSatish Balay int err, wsize; 377e5c89e4eSSatish Balay size_t m = (size_t)n, maxblock = 65536; 378f253e43cSLisandro Dalcin const void *ptmp = p; 37905acbc63SBarry Smith char *fname = NULL; 3800da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 381df40af56SBarry Smith PetscBool writedouble = PETSC_FALSE; 3820da86b62SBarry Smith double *ppp; 3830da86b62SBarry Smith PetscReal *pv; 3840da86b62SBarry Smith PetscInt i; 3850da86b62SBarry Smith #endif 38641f502e3SPatrick Sanan PetscDataType wtype = type; 387e5c89e4eSSatish Balay 388e5c89e4eSSatish Balay PetscFunctionBegin; 38908401ef6SPierre Jolivet PetscCheck(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Trying to write a negative amount of data %" PetscInt_FMT, n); 3903ba16761SJacob Faibussowitsch if (!n) PetscFunctionReturn(PETSC_SUCCESS); 391e5c89e4eSSatish Balay 3922d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 3932d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 3942d53ad75SBarry Smith const char *fnametmp; 3952d53ad75SBarry Smith #endif 3962d53ad75SBarry Smith m = 64; 397e25ab156SSatish Balay fname = (char *)malloc(m * sizeof(char)); 39828b400f6SJacob Faibussowitsch PetscCheck(fname, PETSC_COMM_SELF, PETSC_ERR_MEM, "Cannot allocate space for function name"); 39905acbc63SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 40008401ef6SPierre Jolivet PetscCheck(n <= 1, PETSC_COMM_SELF, PETSC_ERR_SUP, "Can only binary view a single function at a time"); 4019566063dSJacob Faibussowitsch PetscCall(PetscFPTFind(*(void **)p, &fnametmp)); 4029566063dSJacob Faibussowitsch PetscCall(PetscStrncpy(fname, fnametmp, m)); 40305acbc63SBarry Smith #else 4049566063dSJacob Faibussowitsch PetscCall(PetscStrncpy(fname, "", m)); 40505acbc63SBarry Smith #endif 40601963f56SBarry Smith wtype = PETSC_CHAR; 4072d53ad75SBarry Smith pp = (char *)fname; 4082d53ad75SBarry Smith ptmp = (void *)fname; 4092d53ad75SBarry Smith } 4102d53ad75SBarry Smith 4110da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 4129566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetBool(NULL, NULL, "-binary_write_double", &writedouble, NULL)); 4130da86b62SBarry Smith /* If using __float128 precision we still write in doubles to file */ 414501ae109SJose E. Roman if ((type == PETSC_SCALAR || type == PETSC_REAL || type == PETSC_COMPLEX) && writedouble) { 41541f502e3SPatrick Sanan wtype = PETSC_DOUBLE; 4169566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(n, &ppp)); 4170da86b62SBarry Smith pv = (PetscReal *)pp; 418ad540459SPierre Jolivet for (i = 0; i < n; i++) ppp[i] = (double)pv[i]; 4190da86b62SBarry Smith pp = (char *)ppp; 4200da86b62SBarry Smith ptmp = (char *)ppp; 4210da86b62SBarry Smith } 4220da86b62SBarry Smith #endif 4230da86b62SBarry Smith 42441f502e3SPatrick Sanan if (wtype == PETSC_INT) m *= sizeof(PetscInt); 42541f502e3SPatrick Sanan else if (wtype == PETSC_SCALAR) m *= sizeof(PetscScalar); 426501ae109SJose E. Roman #if defined(PETSC_HAVE_COMPLEX) 427501ae109SJose E. Roman else if (wtype == PETSC_COMPLEX) m *= sizeof(PetscComplex); 428501ae109SJose E. Roman #endif 4294caf0332SSatish Balay else if (wtype == PETSC_REAL) m *= sizeof(PetscReal); 43041f502e3SPatrick Sanan else if (wtype == PETSC_DOUBLE) m *= sizeof(double); 43141f502e3SPatrick Sanan else if (wtype == PETSC_FLOAT) m *= sizeof(float); 43241f502e3SPatrick Sanan else if (wtype == PETSC_SHORT) m *= sizeof(short); 433972064b6SLisandro Dalcin else if (wtype == PETSC_LONG) m *= sizeof(long); 43441f502e3SPatrick Sanan else if (wtype == PETSC_CHAR) m *= sizeof(char); 43541f502e3SPatrick Sanan else if (wtype == PETSC_ENUM) m *= sizeof(PetscEnum); 43641f502e3SPatrick Sanan else if (wtype == PETSC_BOOL) m *= sizeof(PetscBool); 4379e3e4c22SLisandro Dalcin else if (wtype == PETSC_INT64) m *= sizeof(PetscInt64); 43841f502e3SPatrick Sanan else if (wtype == PETSC_BIT_LOGICAL) m = PetscBTLength(m) * sizeof(char); 439e32f2f54SBarry Smith else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unknown type"); 440e5c89e4eSSatish Balay 4419566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap((void *)ptmp, wtype, n)); 442e5c89e4eSSatish Balay 443e5c89e4eSSatish Balay while (m) { 444e5c89e4eSSatish Balay wsize = (m < maxblock) ? m : maxblock; 445e5c89e4eSSatish Balay err = write(fd, pp, wsize); 446e5c89e4eSSatish Balay if (err < 0 && errno == EINTR) continue; 44708401ef6SPierre Jolivet PetscCheck(err == wsize, PETSC_COMM_SELF, PETSC_ERR_FILE_WRITE, "Error writing to file total size %d err %d wsize %d", (int)n, (int)err, (int)wsize); 448e5c89e4eSSatish Balay m -= wsize; 449e5c89e4eSSatish Balay pp += wsize; 450e5c89e4eSSatish Balay } 451e5c89e4eSSatish Balay 4529566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap((void *)ptmp, wtype, n)); 453f253e43cSLisandro Dalcin 454ad540459SPierre Jolivet if (type == PETSC_FUNCTION) free(fname); 4550da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 45648a46eb9SPierre Jolivet if ((type == PETSC_SCALAR || type == PETSC_REAL || type == PETSC_COMPLEX) && writedouble) PetscCall(PetscFree(ppp)); 4570da86b62SBarry Smith #endif 4583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 459e5c89e4eSSatish Balay } 460e5c89e4eSSatish Balay 461e5c89e4eSSatish Balay /*@C 462e5c89e4eSSatish Balay PetscBinaryOpen - Opens a PETSc binary file. 463e5c89e4eSSatish Balay 464e5c89e4eSSatish Balay Not Collective 465e5c89e4eSSatish Balay 466e5c89e4eSSatish Balay Input Parameters: 467e5c89e4eSSatish Balay + name - filename 468811af0c4SBarry Smith - mode - open mode of binary file, one of `FILE_MODE_READ`, `FILE_MODE_WRITE`, `FILE_MODE_APPEND`` 469e5c89e4eSSatish Balay 470e5c89e4eSSatish Balay Output Parameter: 471e5c89e4eSSatish Balay . fd - the file 472e5c89e4eSSatish Balay 473e5c89e4eSSatish Balay Level: advanced 474e5c89e4eSSatish Balay 475811af0c4SBarry Smith Note: 476811af0c4SBarry Smith Files access with PetscBinaryRead()` and `PetscBinaryWrite()` are ALWAYS written in 477811af0c4SBarry Smith big-endian format. This means the file can be accessed using `PetscBinaryOpen()` and 478811af0c4SBarry Smith `PetscBinaryRead()` and `PetscBinaryWrite()` on any machine. 479e5c89e4eSSatish Balay 480db781477SPatrick Sanan .seealso: `PetscBinaryRead()`, `PetscBinaryWrite()`, `PetscFileMode`, `PetscViewerFileSetMode()`, `PetscViewerBinaryGetDescriptor()`, 481db781477SPatrick Sanan `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`, `PetscBinarySynchronizedSeek()` 482e5c89e4eSSatish Balay @*/ 483d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinaryOpen(const char name[], PetscFileMode mode, int *fd) 484d71ae5a4SJacob Faibussowitsch { 485e5c89e4eSSatish Balay PetscFunctionBegin; 486cc843e7aSLisandro Dalcin switch (mode) { 487d71ae5a4SJacob Faibussowitsch case FILE_MODE_READ: 488d71ae5a4SJacob Faibussowitsch *fd = open(name, O_BINARY | O_RDONLY, 0); 489d71ae5a4SJacob Faibussowitsch break; 490d71ae5a4SJacob Faibussowitsch case FILE_MODE_WRITE: 491d71ae5a4SJacob Faibussowitsch *fd = open(name, O_BINARY | O_WRONLY | O_CREAT | O_TRUNC, 0666); 492d71ae5a4SJacob Faibussowitsch break; 493d71ae5a4SJacob Faibussowitsch case FILE_MODE_APPEND: 494d71ae5a4SJacob Faibussowitsch *fd = open(name, O_BINARY | O_WRONLY | O_APPEND, 0); 495d71ae5a4SJacob Faibussowitsch break; 496d71ae5a4SJacob Faibussowitsch default: 497d71ae5a4SJacob Faibussowitsch SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unsupported file mode %s", PetscFileModes[mode]); 498e5c89e4eSSatish Balay } 49908401ef6SPierre Jolivet PetscCheck(*fd != -1, PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "Cannot open file %s for %s", name, PetscFileModes[mode]); 5003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 501e5c89e4eSSatish Balay } 502e5c89e4eSSatish Balay 503e30d2299SSatish Balay /*@ 504e5c89e4eSSatish Balay PetscBinaryClose - Closes a PETSc binary file. 505e5c89e4eSSatish Balay 506e5c89e4eSSatish Balay Not Collective 507e5c89e4eSSatish Balay 508e5c89e4eSSatish Balay Output Parameter: 509e5c89e4eSSatish Balay . fd - the file 510e5c89e4eSSatish Balay 511e5c89e4eSSatish Balay Level: advanced 512e5c89e4eSSatish Balay 513db781477SPatrick Sanan .seealso: `PetscBinaryRead()`, `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`, 514db781477SPatrick Sanan `PetscBinarySynchronizedSeek()` 515e5c89e4eSSatish Balay @*/ 516d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinaryClose(int fd) 517d71ae5a4SJacob Faibussowitsch { 518e5c89e4eSSatish Balay PetscFunctionBegin; 519cc73adaaSBarry Smith PetscCheck(!close(fd), PETSC_COMM_SELF, PETSC_ERR_SYS, "close() failed on file descriptor"); 5203ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 521e5c89e4eSSatish Balay } 522e5c89e4eSSatish Balay 523e8976759SBarry Smith /*@C 524e5c89e4eSSatish Balay PetscBinarySeek - Moves the file pointer on a PETSc binary file. 525e5c89e4eSSatish Balay 526e5c89e4eSSatish Balay Not Collective 527e5c89e4eSSatish Balay 528e5c89e4eSSatish Balay Input Parameters: 529e5c89e4eSSatish Balay + fd - the file 530811af0c4SBarry Smith . off - number of bytes to move. Use `PETSC_BINARY_INT_SIZE`, `PETSC_BINARY_SCALAR_SIZE`, 531*bfbbc7b7SBarry Smith etc. in your calculation rather than `sizeof()` to compute byte lengths. 532811af0c4SBarry Smith - whence - if `PETSC_BINARY_SEEK_SET` then off is an absolute location in the file 533811af0c4SBarry Smith if `PETSC_BINARY_SEEK_CUR` then off is an offset from the current location 534811af0c4SBarry Smith if `PETSC_BINARY_SEEK_END` then off is an offset from the end of file 535e5c89e4eSSatish Balay 536e5c89e4eSSatish Balay Output Parameter: 537e5c89e4eSSatish Balay . offset - new offset in file 538e5c89e4eSSatish Balay 539e5c89e4eSSatish Balay Level: developer 540e5c89e4eSSatish Balay 541811af0c4SBarry Smith Note: 542e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 543e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 544811af0c4SBarry Smith binary file may be read on any machine. Hence you CANNOT use `sizeof()` 545e5c89e4eSSatish Balay to determine the offset or location. 546e5c89e4eSSatish Balay 547db781477SPatrick Sanan .seealso: `PetscBinaryRead()`, `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`, 548db781477SPatrick Sanan `PetscBinarySynchronizedSeek()` 549e5c89e4eSSatish Balay @*/ 550d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySeek(int fd, off_t off, PetscBinarySeekType whence, off_t *offset) 551d71ae5a4SJacob Faibussowitsch { 552e5c89e4eSSatish Balay int iwhence = 0; 553e5c89e4eSSatish Balay 554e5c89e4eSSatish Balay PetscFunctionBegin; 555a297a907SKarl Rupp if (whence == PETSC_BINARY_SEEK_SET) iwhence = SEEK_SET; 556a297a907SKarl Rupp else if (whence == PETSC_BINARY_SEEK_CUR) iwhence = SEEK_CUR; 557a297a907SKarl Rupp else if (whence == PETSC_BINARY_SEEK_END) iwhence = SEEK_END; 558a297a907SKarl Rupp else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unknown seek location"); 559e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) 560e5c89e4eSSatish Balay *offset = lseek(fd, off, iwhence); 561e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK) 562e5c89e4eSSatish Balay *offset = _lseek(fd, (long)off, iwhence); 563e5c89e4eSSatish Balay #else 564e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "System does not have a way of seeking on a file"); 565e5c89e4eSSatish Balay #endif 5663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 567e5c89e4eSSatish Balay } 568e5c89e4eSSatish Balay 569e5c89e4eSSatish Balay /*@C 5701d280d73SBarry Smith PetscBinarySynchronizedRead - Reads from a binary file. 571e5c89e4eSSatish Balay 572d083f849SBarry Smith Collective 573e5c89e4eSSatish Balay 574e5c89e4eSSatish Balay Input Parameters: 575e5c89e4eSSatish Balay + comm - the MPI communicator 5769860990eSLisandro Dalcin . fd - the file descriptor 5779860990eSLisandro Dalcin . num - the maximum number of items to read 578811af0c4SBarry Smith - type - the type of items to read (`PETSC_INT`, `PETSC_REAL`, `PETSC_SCALAR`, etc.) 579e5c89e4eSSatish Balay 580e5c89e4eSSatish Balay Output Parameters: 5819860990eSLisandro Dalcin + data - the buffer 5829860990eSLisandro Dalcin - count - the number of items read, optional 583e5c89e4eSSatish Balay 584e5c89e4eSSatish Balay Level: developer 585e5c89e4eSSatish Balay 586e5c89e4eSSatish Balay Notes: 587811af0c4SBarry Smith Does a `PetscBinaryRead()` followed by an `MPI_Bcast()` 588e5c89e4eSSatish Balay 5899860990eSLisandro Dalcin If count is not provided and the number of items read is less than 5909860990eSLisandro Dalcin the maximum number of items to read, then this routine errors. 5919860990eSLisandro Dalcin 592811af0c4SBarry Smith `PetscBinarySynchronizedRead()` uses byte swapping to work on all machines. 593e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 594e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 595e5c89e4eSSatish Balay binary file may be read on any machine. 596e5c89e4eSSatish Balay 597db781477SPatrick Sanan .seealso: `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscBinaryRead()`, `PetscBinarySynchronizedWrite()`, 598db781477SPatrick Sanan `PetscBinarySynchronizedSeek()` 599e5c89e4eSSatish Balay @*/ 600d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm, int fd, void *data, PetscInt num, PetscInt *count, PetscDataType type) 601d71ae5a4SJacob Faibussowitsch { 6028b8bf82cSBarry Smith PetscMPIInt rank, size; 603e5c89e4eSSatish Balay MPI_Datatype mtype; 6049860990eSLisandro Dalcin PetscInt ibuf[2] = {0, 0}; 60505acbc63SBarry Smith char *fname = NULL; 6069860990eSLisandro Dalcin void *fptr = NULL; 607e5c89e4eSSatish Balay 608e5c89e4eSSatish Balay PetscFunctionBegin; 6092d53ad75SBarry Smith if (type == PETSC_FUNCTION) { 6109860990eSLisandro Dalcin num = 64; 6112d53ad75SBarry Smith type = PETSC_CHAR; 6129860990eSLisandro Dalcin fname = (char *)malloc(num * sizeof(char)); 6139860990eSLisandro Dalcin fptr = data; 6149860990eSLisandro Dalcin data = (void *)fname; 61528b400f6SJacob Faibussowitsch PetscCheck(fname, PETSC_COMM_SELF, PETSC_ERR_MEM, "Cannot allocate space for function name"); 6162d53ad75SBarry Smith } 6172d53ad75SBarry Smith 6189566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 6199566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(comm, &size)); 620ad540459SPierre Jolivet if (rank == 0) ibuf[0] = PetscBinaryRead(fd, data, num, count ? &ibuf[1] : NULL, type); 6219566063dSJacob Faibussowitsch PetscCallMPI(MPI_Bcast(ibuf, 2, MPIU_INT, 0, comm)); 6229566063dSJacob Faibussowitsch PetscCall((PetscErrorCode)ibuf[0]); 6238b8bf82cSBarry Smith 6248b8bf82cSBarry Smith /* skip MPI call on potentially huge amounts of data when running with one process; this allows the amount of data to basically unlimited in that case */ 6258b8bf82cSBarry Smith if (size > 1) { 6269566063dSJacob Faibussowitsch PetscCall(PetscDataTypeToMPIDataType(type, &mtype)); 6279566063dSJacob Faibussowitsch PetscCallMPI(MPI_Bcast(data, count ? ibuf[1] : num, mtype, 0, comm)); 6288b8bf82cSBarry Smith } 6299860990eSLisandro Dalcin if (count) *count = ibuf[1]; 6302d53ad75SBarry Smith 631e366c363SBarry Smith if (type == PETSC_FUNCTION) { 6322d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS) 6339566063dSJacob Faibussowitsch PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, NULL, fname, (void **)fptr)); 6342d53ad75SBarry Smith #else 6359860990eSLisandro Dalcin *(void **)fptr = NULL; 6362d53ad75SBarry Smith #endif 637e366c363SBarry Smith free(fname); 6382d53ad75SBarry Smith } 6393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 640e5c89e4eSSatish Balay } 641e5c89e4eSSatish Balay 642e5c89e4eSSatish Balay /*@C 6431d280d73SBarry Smith PetscBinarySynchronizedWrite - writes to a binary file. 644e5c89e4eSSatish Balay 645d083f849SBarry Smith Collective 646e5c89e4eSSatish Balay 647e5c89e4eSSatish Balay Input Parameters: 648e5c89e4eSSatish Balay + comm - the MPI communicator 649e5c89e4eSSatish Balay . fd - the file 650e5c89e4eSSatish Balay . n - the number of items to write 651e5c89e4eSSatish Balay . p - the buffer 652811af0c4SBarry Smith - type - the type of items to write (`PETSC_INT`, `PETSC_REAL` or `PETSC_SCALAR`) 653e5c89e4eSSatish Balay 654e5c89e4eSSatish Balay Level: developer 655e5c89e4eSSatish Balay 656e5c89e4eSSatish Balay Notes: 657*bfbbc7b7SBarry Smith MPI rank 0 does a `PetscBinaryWrite()` 658e5c89e4eSSatish Balay 659811af0c4SBarry Smith `PetscBinarySynchronizedWrite()` uses byte swapping to work on all machines. 660e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 661e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 662e5c89e4eSSatish Balay binary file may be read on any machine. 663e5c89e4eSSatish Balay 664811af0c4SBarry Smith Because byte-swapping may be done on the values in data it cannot be declared const 665300a7f5bSBarry Smith 666811af0c4SBarry Smith WARNING: 667*bfbbc7b7SBarry Smith This is NOT like `PetscSynchronizedFPrintf()`! This routine ignores calls on all but MPI rank 0, 668811af0c4SBarry Smith while `PetscSynchronizedFPrintf()` has all processes print their strings in order. 6691d280d73SBarry Smith 670db781477SPatrick Sanan .seealso: `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscBinaryRead()`, `PetscBinarySynchronizedRead()`, 671db781477SPatrick Sanan `PetscBinarySynchronizedSeek()` 672e5c89e4eSSatish Balay @*/ 673d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedWrite(MPI_Comm comm, int fd, const void *p, PetscInt n, PetscDataType type) 674d71ae5a4SJacob Faibussowitsch { 675e5c89e4eSSatish Balay PetscMPIInt rank; 676e5c89e4eSSatish Balay 677e5c89e4eSSatish Balay PetscFunctionBegin; 6789566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 67948a46eb9SPierre Jolivet if (rank == 0) PetscCall(PetscBinaryWrite(fd, p, n, type)); 6803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 681e5c89e4eSSatish Balay } 682e5c89e4eSSatish Balay 683e5c89e4eSSatish Balay /*@C 6841d280d73SBarry Smith PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file. 685e5c89e4eSSatish Balay 686e5c89e4eSSatish Balay Input Parameters: 687e5c89e4eSSatish Balay + fd - the file 688811af0c4SBarry Smith . whence - if `PETSC_BINARY_SEEK_SET` then size is an absolute location in the file 689811af0c4SBarry Smith if `PETSC_BINARY_SEEK_CUR` then size is offset from current location 690811af0c4SBarry Smith if `PETSC_BINARY_SEEK_END` then size is offset from end of file 691811af0c4SBarry Smith - off - number of bytes to move. Use `PETSC_BINARY_INT_SIZE`, `PETSC_BINARY_SCALAR_SIZE`, 692811af0c4SBarry Smith etc. in your calculation rather than `sizeof()` to compute byte lengths. 693e5c89e4eSSatish Balay 694e5c89e4eSSatish Balay Output Parameter: 695e5c89e4eSSatish Balay . offset - new offset in file 696e5c89e4eSSatish Balay 697e5c89e4eSSatish Balay Level: developer 698e5c89e4eSSatish Balay 699811af0c4SBarry Smith Note: 700e5c89e4eSSatish Balay Integers are stored on the file as 32 long, regardless of whether 701e5c89e4eSSatish Balay they are stored in the machine as 32 or 64, this means the same 702811af0c4SBarry Smith binary file may be read on any machine. Hence you CANNOT use `sizeof()` 703e5c89e4eSSatish Balay to determine the offset or location. 704e5c89e4eSSatish Balay 705db781477SPatrick Sanan .seealso: `PetscBinaryRead()`, `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`, 706db781477SPatrick Sanan `PetscBinarySynchronizedSeek()` 707e5c89e4eSSatish Balay @*/ 708d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedSeek(MPI_Comm comm, int fd, off_t off, PetscBinarySeekType whence, off_t *offset) 709d71ae5a4SJacob Faibussowitsch { 710e5c89e4eSSatish Balay PetscMPIInt rank; 711e5c89e4eSSatish Balay 712e5c89e4eSSatish Balay PetscFunctionBegin; 7139566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(comm, &rank)); 71448a46eb9SPierre Jolivet if (rank == 0) PetscCall(PetscBinarySeek(fd, off, whence, offset)); 7153ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 716e5c89e4eSSatish Balay } 717e5c89e4eSSatish Balay 7180fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO) 719e39fd77fSBarry Smith 720951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32) 721e39fd77fSBarry Smith /* 722e39fd77fSBarry Smith MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions. 723e39fd77fSBarry Smith These are set into MPI in PetscInitialize() via MPI_Register_datarep() 724e39fd77fSBarry Smith 725e39fd77fSBarry Smith Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode) 726e39fd77fSBarry Smith 727951e3c8eSBarry Smith The next three routines are not used because MPICH does not support their use 728e39fd77fSBarry Smith 729e39fd77fSBarry Smith */ 730d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype, MPI_Aint *file_extent, void *extra_state) 731d71ae5a4SJacob Faibussowitsch { 732e39fd77fSBarry Smith MPI_Aint ub; 733e39fd77fSBarry Smith PetscMPIInt ierr; 734e39fd77fSBarry Smith 735e39fd77fSBarry Smith ierr = MPI_Type_get_extent(datatype, &ub, file_extent); 736e39fd77fSBarry Smith return ierr; 737e39fd77fSBarry Smith } 738e39fd77fSBarry Smith 739d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype, PetscMPIInt count, void *filebuf, MPI_Offset position, void *extra_state) 740d71ae5a4SJacob Faibussowitsch { 741e39fd77fSBarry Smith PetscDataType pdtype; 742e39fd77fSBarry Smith PetscMPIInt ierr; 743e39fd77fSBarry Smith size_t dsize; 744e39fd77fSBarry Smith 7459566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(datatype, &pdtype)); 7469566063dSJacob Faibussowitsch PetscCall(PetscDataTypeGetSize(pdtype, &dsize)); 747e39fd77fSBarry Smith 748e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 749e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize * position; 750e39fd77fSBarry Smith 7519566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(userbuf, filebuf, count * dsize)); 7529566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(userbuf, pdtype, count)); 753e39fd77fSBarry Smith return ierr; 754e39fd77fSBarry Smith } 755e39fd77fSBarry Smith 756d71ae5a4SJacob Faibussowitsch PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype, PetscMPIInt count, void *filebuf, MPI_Offset position, void *extra_state) 757d71ae5a4SJacob Faibussowitsch { 758e39fd77fSBarry Smith PetscDataType pdtype; 759e39fd77fSBarry Smith PetscMPIInt ierr; 760e39fd77fSBarry Smith size_t dsize; 761e39fd77fSBarry Smith 7629566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(datatype, &pdtype)); 7639566063dSJacob Faibussowitsch PetscCall(PetscDataTypeGetSize(pdtype, &dsize)); 764e39fd77fSBarry Smith 765e39fd77fSBarry Smith /* offset is given in units of MPI_Datatype */ 766e39fd77fSBarry Smith userbuf = ((char *)userbuf) + dsize * position; 767e39fd77fSBarry Smith 7689566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(filebuf, userbuf, count * dsize)); 7699566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(filebuf, pdtype, count)); 770e39fd77fSBarry Smith return ierr; 771e39fd77fSBarry Smith } 772951e3c8eSBarry Smith #endif 773e39fd77fSBarry Smith 774d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_all(MPI_File fd, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 775d71ae5a4SJacob Faibussowitsch { 776e39fd77fSBarry Smith PetscDataType pdtype; 77730815ce0SLisandro Dalcin 778e39fd77fSBarry Smith PetscFunctionBegin; 7799566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 7809566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 7819566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_write_all(fd, data, cnt, dtype, status)); 7829566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 7833ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 784e39fd77fSBarry Smith } 785e39fd77fSBarry Smith 786d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_all(MPI_File fd, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 787d71ae5a4SJacob Faibussowitsch { 788e39fd77fSBarry Smith PetscDataType pdtype; 789e39fd77fSBarry Smith 790e39fd77fSBarry Smith PetscFunctionBegin; 7919566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 7929566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_read_all(fd, data, cnt, dtype, status)); 7939566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 7943ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 795e39fd77fSBarry Smith } 79630815ce0SLisandro Dalcin 797d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_at(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 798d71ae5a4SJacob Faibussowitsch { 799c4e82887SLisandro Dalcin PetscDataType pdtype; 800c4e82887SLisandro Dalcin 801c4e82887SLisandro Dalcin PetscFunctionBegin; 8029566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 8039566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8049566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_write_at(fd, off, data, cnt, dtype, status)); 8059566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8063ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 807c4e82887SLisandro Dalcin } 808c4e82887SLisandro Dalcin 809d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_at(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 810d71ae5a4SJacob Faibussowitsch { 811c4e82887SLisandro Dalcin PetscDataType pdtype; 812c4e82887SLisandro Dalcin 813c4e82887SLisandro Dalcin PetscFunctionBegin; 8149566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 8159566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_read_at(fd, off, data, cnt, dtype, status)); 8169566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 818c4e82887SLisandro Dalcin } 819c4e82887SLisandro Dalcin 820d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_at_all(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 821d71ae5a4SJacob Faibussowitsch { 822c4e82887SLisandro Dalcin PetscDataType pdtype; 823c4e82887SLisandro Dalcin 824c4e82887SLisandro Dalcin PetscFunctionBegin; 8259566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 8269566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8279566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_write_at_all(fd, off, data, cnt, dtype, status)); 8289566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 830c4e82887SLisandro Dalcin } 831c4e82887SLisandro Dalcin 832d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_at_all(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status) 833d71ae5a4SJacob Faibussowitsch { 834c4e82887SLisandro Dalcin PetscDataType pdtype; 835c4e82887SLisandro Dalcin 836c4e82887SLisandro Dalcin PetscFunctionBegin; 8379566063dSJacob Faibussowitsch PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype)); 8389566063dSJacob Faibussowitsch PetscCallMPI(MPI_File_read_at_all(fd, off, data, cnt, dtype, status)); 8399566063dSJacob Faibussowitsch if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt)); 8403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 841c4e82887SLisandro Dalcin } 842c4e82887SLisandro Dalcin 843951e3c8eSBarry Smith #endif 844