xref: /petsc/src/sys/fileio/sysio.c (revision 16a05f60a523f53ab316acaac9f77b7425611adc)
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
251bfbbc7b7SBarry 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
355bfbbc7b7SBarry 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 
358bfbbc7b7SBarry Smith    If running with `__float128` precision the output is in `__float128` unless one uses the `-binary_write_double` option
3590da86b62SBarry Smith 
360bfbbc7b7SBarry 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*16a05f60SBarry Smith             etc. in your calculation rather than sizeof() to compute byte lengths.
532*16a05f60SBarry Smith -  whence - see `PetscBinarySeekType` for possible values
533e5c89e4eSSatish Balay 
534e5c89e4eSSatish Balay    Output Parameter:
535e5c89e4eSSatish Balay .   offset - new offset in file
536e5c89e4eSSatish Balay 
537e5c89e4eSSatish Balay    Level: developer
538e5c89e4eSSatish Balay 
539811af0c4SBarry Smith    Note:
540e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
541e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
542811af0c4SBarry Smith    binary file may be read on any machine. Hence you CANNOT use `sizeof()`
543e5c89e4eSSatish Balay    to determine the offset or location.
544e5c89e4eSSatish Balay 
545*16a05f60SBarry Smith .seealso: `PetscBinaryRead()`, `PetscBinarySeekType`, `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`,
546db781477SPatrick Sanan           `PetscBinarySynchronizedSeek()`
547e5c89e4eSSatish Balay @*/
548d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySeek(int fd, off_t off, PetscBinarySeekType whence, off_t *offset)
549d71ae5a4SJacob Faibussowitsch {
550e5c89e4eSSatish Balay   int iwhence = 0;
551e5c89e4eSSatish Balay 
552e5c89e4eSSatish Balay   PetscFunctionBegin;
553a297a907SKarl Rupp   if (whence == PETSC_BINARY_SEEK_SET) iwhence = SEEK_SET;
554a297a907SKarl Rupp   else if (whence == PETSC_BINARY_SEEK_CUR) iwhence = SEEK_CUR;
555a297a907SKarl Rupp   else if (whence == PETSC_BINARY_SEEK_END) iwhence = SEEK_END;
556a297a907SKarl Rupp   else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unknown seek location");
557e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK)
558e5c89e4eSSatish Balay   *offset = lseek(fd, off, iwhence);
559e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK)
560e5c89e4eSSatish Balay   *offset = _lseek(fd, (long)off, iwhence);
561e5c89e4eSSatish Balay #else
562e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "System does not have a way of seeking on a file");
563e5c89e4eSSatish Balay #endif
5643ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
565e5c89e4eSSatish Balay }
566e5c89e4eSSatish Balay 
567e5c89e4eSSatish Balay /*@C
5681d280d73SBarry Smith    PetscBinarySynchronizedRead - Reads from a binary file.
569e5c89e4eSSatish Balay 
570d083f849SBarry Smith    Collective
571e5c89e4eSSatish Balay 
572e5c89e4eSSatish Balay    Input Parameters:
573e5c89e4eSSatish Balay +  comm - the MPI communicator
5749860990eSLisandro Dalcin .  fd - the file descriptor
5759860990eSLisandro Dalcin .  num  - the maximum number of items to read
576811af0c4SBarry Smith -  type - the type of items to read (`PETSC_INT`, `PETSC_REAL`, `PETSC_SCALAR`, etc.)
577e5c89e4eSSatish Balay 
578e5c89e4eSSatish Balay    Output Parameters:
5799860990eSLisandro Dalcin +  data - the buffer
5809860990eSLisandro Dalcin -  count - the number of items read, optional
581e5c89e4eSSatish Balay 
582e5c89e4eSSatish Balay    Level: developer
583e5c89e4eSSatish Balay 
584e5c89e4eSSatish Balay    Notes:
585811af0c4SBarry Smith    Does a `PetscBinaryRead()` followed by an `MPI_Bcast()`
586e5c89e4eSSatish Balay 
5879860990eSLisandro Dalcin    If count is not provided and the number of items read is less than
5889860990eSLisandro Dalcin    the maximum number of items to read, then this routine errors.
5899860990eSLisandro Dalcin 
590811af0c4SBarry Smith    `PetscBinarySynchronizedRead()` uses byte swapping to work on all machines.
591e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
592e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
593e5c89e4eSSatish Balay    binary file may be read on any machine.
594e5c89e4eSSatish Balay 
595db781477SPatrick Sanan .seealso: `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscBinaryRead()`, `PetscBinarySynchronizedWrite()`,
596db781477SPatrick Sanan           `PetscBinarySynchronizedSeek()`
597e5c89e4eSSatish Balay @*/
598d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedRead(MPI_Comm comm, int fd, void *data, PetscInt num, PetscInt *count, PetscDataType type)
599d71ae5a4SJacob Faibussowitsch {
6008b8bf82cSBarry Smith   PetscMPIInt  rank, size;
601e5c89e4eSSatish Balay   MPI_Datatype mtype;
6029860990eSLisandro Dalcin   PetscInt     ibuf[2] = {0, 0};
60305acbc63SBarry Smith   char        *fname   = NULL;
6049860990eSLisandro Dalcin   void        *fptr    = NULL;
605e5c89e4eSSatish Balay 
606e5c89e4eSSatish Balay   PetscFunctionBegin;
6072d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
6089860990eSLisandro Dalcin     num   = 64;
6092d53ad75SBarry Smith     type  = PETSC_CHAR;
6109860990eSLisandro Dalcin     fname = (char *)malloc(num * sizeof(char));
6119860990eSLisandro Dalcin     fptr  = data;
6129860990eSLisandro Dalcin     data  = (void *)fname;
61328b400f6SJacob Faibussowitsch     PetscCheck(fname, PETSC_COMM_SELF, PETSC_ERR_MEM, "Cannot allocate space for function name");
6142d53ad75SBarry Smith   }
6152d53ad75SBarry Smith 
6169566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
6179566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
618ad540459SPierre Jolivet   if (rank == 0) ibuf[0] = PetscBinaryRead(fd, data, num, count ? &ibuf[1] : NULL, type);
6199566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Bcast(ibuf, 2, MPIU_INT, 0, comm));
6209566063dSJacob Faibussowitsch   PetscCall((PetscErrorCode)ibuf[0]);
6218b8bf82cSBarry Smith 
6228b8bf82cSBarry 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 */
6238b8bf82cSBarry Smith   if (size > 1) {
6249566063dSJacob Faibussowitsch     PetscCall(PetscDataTypeToMPIDataType(type, &mtype));
6259566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Bcast(data, count ? ibuf[1] : num, mtype, 0, comm));
6268b8bf82cSBarry Smith   }
6279860990eSLisandro Dalcin   if (count) *count = ibuf[1];
6282d53ad75SBarry Smith 
629e366c363SBarry Smith   if (type == PETSC_FUNCTION) {
6302d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
6319566063dSJacob Faibussowitsch     PetscCall(PetscDLLibrarySym(PETSC_COMM_SELF, &PetscDLLibrariesLoaded, NULL, fname, (void **)fptr));
6322d53ad75SBarry Smith #else
6339860990eSLisandro Dalcin     *(void **)fptr = NULL;
6342d53ad75SBarry Smith #endif
635e366c363SBarry Smith     free(fname);
6362d53ad75SBarry Smith   }
6373ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
638e5c89e4eSSatish Balay }
639e5c89e4eSSatish Balay 
640e5c89e4eSSatish Balay /*@C
6411d280d73SBarry Smith    PetscBinarySynchronizedWrite - writes to a binary file.
642e5c89e4eSSatish Balay 
643d083f849SBarry Smith    Collective
644e5c89e4eSSatish Balay 
645e5c89e4eSSatish Balay    Input Parameters:
646e5c89e4eSSatish Balay +  comm - the MPI communicator
647e5c89e4eSSatish Balay .  fd - the file
648e5c89e4eSSatish Balay .  n  - the number of items to write
649e5c89e4eSSatish Balay .  p - the buffer
650811af0c4SBarry Smith -  type - the type of items to write (`PETSC_INT`, `PETSC_REAL` or `PETSC_SCALAR`)
651e5c89e4eSSatish Balay 
652e5c89e4eSSatish Balay    Level: developer
653e5c89e4eSSatish Balay 
654e5c89e4eSSatish Balay    Notes:
655bfbbc7b7SBarry Smith    MPI rank 0 does a `PetscBinaryWrite()`
656e5c89e4eSSatish Balay 
657811af0c4SBarry Smith    `PetscBinarySynchronizedWrite()` uses byte swapping to work on all machines.
658e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
659e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
660e5c89e4eSSatish Balay    binary file may be read on any machine.
661e5c89e4eSSatish Balay 
662811af0c4SBarry Smith    Because byte-swapping may be done on the values in data it cannot be declared const
663300a7f5bSBarry Smith 
664811af0c4SBarry Smith    WARNING:
665bfbbc7b7SBarry Smith    This is NOT like `PetscSynchronizedFPrintf()`! This routine ignores calls on all but MPI rank 0,
666811af0c4SBarry Smith    while `PetscSynchronizedFPrintf()` has all processes print their strings in order.
6671d280d73SBarry Smith 
668db781477SPatrick Sanan .seealso: `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinaryClose()`, `PetscBinaryRead()`, `PetscBinarySynchronizedRead()`,
669db781477SPatrick Sanan           `PetscBinarySynchronizedSeek()`
670e5c89e4eSSatish Balay @*/
671d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedWrite(MPI_Comm comm, int fd, const void *p, PetscInt n, PetscDataType type)
672d71ae5a4SJacob Faibussowitsch {
673e5c89e4eSSatish Balay   PetscMPIInt rank;
674e5c89e4eSSatish Balay 
675e5c89e4eSSatish Balay   PetscFunctionBegin;
6769566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
67748a46eb9SPierre Jolivet   if (rank == 0) PetscCall(PetscBinaryWrite(fd, p, n, type));
6783ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
679e5c89e4eSSatish Balay }
680e5c89e4eSSatish Balay 
681e5c89e4eSSatish Balay /*@C
6821d280d73SBarry Smith    PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file.
683e5c89e4eSSatish Balay 
684e5c89e4eSSatish Balay    Input Parameters:
685e5c89e4eSSatish Balay +  fd - the file
686*16a05f60SBarry Smith .  whence -  see `PetscBinarySeekType` for possible values
687811af0c4SBarry Smith -  off    - number of bytes to move. Use `PETSC_BINARY_INT_SIZE`, `PETSC_BINARY_SCALAR_SIZE`,
688811af0c4SBarry Smith             etc. in your calculation rather than `sizeof()` to compute byte lengths.
689e5c89e4eSSatish Balay 
690e5c89e4eSSatish Balay    Output Parameter:
691e5c89e4eSSatish Balay .   offset - new offset in file
692e5c89e4eSSatish Balay 
693e5c89e4eSSatish Balay    Level: developer
694e5c89e4eSSatish Balay 
695811af0c4SBarry Smith    Note:
696e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
697e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
698811af0c4SBarry Smith    binary file may be read on any machine. Hence you CANNOT use `sizeof()`
699e5c89e4eSSatish Balay    to determine the offset or location.
700e5c89e4eSSatish Balay 
701*16a05f60SBarry Smith .seealso: `PetscBinaryRead()`, `PetscBinarySeekType`, `PetscBinaryWrite()`, `PetscBinaryOpen()`, `PetscBinarySynchronizedWrite()`, `PetscBinarySynchronizedRead()`,
702db781477SPatrick Sanan           `PetscBinarySynchronizedSeek()`
703e5c89e4eSSatish Balay @*/
704d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscBinarySynchronizedSeek(MPI_Comm comm, int fd, off_t off, PetscBinarySeekType whence, off_t *offset)
705d71ae5a4SJacob Faibussowitsch {
706e5c89e4eSSatish Balay   PetscMPIInt rank;
707e5c89e4eSSatish Balay 
708e5c89e4eSSatish Balay   PetscFunctionBegin;
7099566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
71048a46eb9SPierre Jolivet   if (rank == 0) PetscCall(PetscBinarySeek(fd, off, whence, offset));
7113ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
712e5c89e4eSSatish Balay }
713e5c89e4eSSatish Balay 
7140fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO)
715e39fd77fSBarry Smith 
716951e3c8eSBarry Smith   #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32)
717e39fd77fSBarry Smith /*
718e39fd77fSBarry Smith       MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions.
719e39fd77fSBarry Smith     These are set into MPI in PetscInitialize() via MPI_Register_datarep()
720e39fd77fSBarry Smith 
721e39fd77fSBarry Smith     Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode)
722e39fd77fSBarry Smith 
723951e3c8eSBarry Smith     The next three routines are not used because MPICH does not support their use
724e39fd77fSBarry Smith 
725e39fd77fSBarry Smith */
726d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype, MPI_Aint *file_extent, void *extra_state)
727d71ae5a4SJacob Faibussowitsch {
728e39fd77fSBarry Smith   MPI_Aint    ub;
729e39fd77fSBarry Smith   PetscMPIInt ierr;
730e39fd77fSBarry Smith 
731e39fd77fSBarry Smith   ierr = MPI_Type_get_extent(datatype, &ub, file_extent);
732e39fd77fSBarry Smith   return ierr;
733e39fd77fSBarry Smith }
734e39fd77fSBarry Smith 
735d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype, PetscMPIInt count, void *filebuf, MPI_Offset position, void *extra_state)
736d71ae5a4SJacob Faibussowitsch {
737e39fd77fSBarry Smith   PetscDataType pdtype;
738e39fd77fSBarry Smith   PetscMPIInt   ierr;
739e39fd77fSBarry Smith   size_t        dsize;
740e39fd77fSBarry Smith 
7419566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(datatype, &pdtype));
7429566063dSJacob Faibussowitsch   PetscCall(PetscDataTypeGetSize(pdtype, &dsize));
743e39fd77fSBarry Smith 
744e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
745e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize * position;
746e39fd77fSBarry Smith 
7479566063dSJacob Faibussowitsch   PetscCall(PetscMemcpy(userbuf, filebuf, count * dsize));
7489566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(userbuf, pdtype, count));
749e39fd77fSBarry Smith   return ierr;
750e39fd77fSBarry Smith }
751e39fd77fSBarry Smith 
752d71ae5a4SJacob Faibussowitsch PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype, PetscMPIInt count, void *filebuf, MPI_Offset position, void *extra_state)
753d71ae5a4SJacob Faibussowitsch {
754e39fd77fSBarry Smith   PetscDataType pdtype;
755e39fd77fSBarry Smith   PetscMPIInt   ierr;
756e39fd77fSBarry Smith   size_t        dsize;
757e39fd77fSBarry Smith 
7589566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(datatype, &pdtype));
7599566063dSJacob Faibussowitsch   PetscCall(PetscDataTypeGetSize(pdtype, &dsize));
760e39fd77fSBarry Smith 
761e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
762e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize * position;
763e39fd77fSBarry Smith 
7649566063dSJacob Faibussowitsch   PetscCall(PetscMemcpy(filebuf, userbuf, count * dsize));
7659566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(filebuf, pdtype, count));
766e39fd77fSBarry Smith   return ierr;
767e39fd77fSBarry Smith }
768951e3c8eSBarry Smith   #endif
769e39fd77fSBarry Smith 
770d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_all(MPI_File fd, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
771d71ae5a4SJacob Faibussowitsch {
772e39fd77fSBarry Smith   PetscDataType pdtype;
77330815ce0SLisandro Dalcin 
774e39fd77fSBarry Smith   PetscFunctionBegin;
7759566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
7769566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
7779566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_write_all(fd, data, cnt, dtype, status));
7789566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
7793ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
780e39fd77fSBarry Smith }
781e39fd77fSBarry Smith 
782d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_all(MPI_File fd, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
783d71ae5a4SJacob Faibussowitsch {
784e39fd77fSBarry Smith   PetscDataType pdtype;
785e39fd77fSBarry Smith 
786e39fd77fSBarry Smith   PetscFunctionBegin;
7879566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
7889566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_read_all(fd, data, cnt, dtype, status));
7899566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
7903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
791e39fd77fSBarry Smith }
79230815ce0SLisandro Dalcin 
793d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_at(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
794d71ae5a4SJacob Faibussowitsch {
795c4e82887SLisandro Dalcin   PetscDataType pdtype;
796c4e82887SLisandro Dalcin 
797c4e82887SLisandro Dalcin   PetscFunctionBegin;
7989566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
7999566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8009566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_write_at(fd, off, data, cnt, dtype, status));
8019566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8023ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
803c4e82887SLisandro Dalcin }
804c4e82887SLisandro Dalcin 
805d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_at(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
806d71ae5a4SJacob Faibussowitsch {
807c4e82887SLisandro Dalcin   PetscDataType pdtype;
808c4e82887SLisandro Dalcin 
809c4e82887SLisandro Dalcin   PetscFunctionBegin;
8109566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
8119566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_read_at(fd, off, data, cnt, dtype, status));
8129566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
814c4e82887SLisandro Dalcin }
815c4e82887SLisandro Dalcin 
816d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_write_at_all(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
817d71ae5a4SJacob Faibussowitsch {
818c4e82887SLisandro Dalcin   PetscDataType pdtype;
819c4e82887SLisandro Dalcin 
820c4e82887SLisandro Dalcin   PetscFunctionBegin;
8219566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
8229566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8239566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_write_at_all(fd, off, data, cnt, dtype, status));
8249566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8253ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
826c4e82887SLisandro Dalcin }
827c4e82887SLisandro Dalcin 
828d71ae5a4SJacob Faibussowitsch PetscErrorCode MPIU_File_read_at_all(MPI_File fd, MPI_Offset off, void *data, PetscMPIInt cnt, MPI_Datatype dtype, MPI_Status *status)
829d71ae5a4SJacob Faibussowitsch {
830c4e82887SLisandro Dalcin   PetscDataType pdtype;
831c4e82887SLisandro Dalcin 
832c4e82887SLisandro Dalcin   PetscFunctionBegin;
8339566063dSJacob Faibussowitsch   PetscCall(PetscMPIDataTypeToPetscDataType(dtype, &pdtype));
8349566063dSJacob Faibussowitsch   PetscCallMPI(MPI_File_read_at_all(fd, off, data, cnt, dtype, status));
8359566063dSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) PetscCall(PetscByteSwap(data, pdtype, cnt));
8363ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
837c4e82887SLisandro Dalcin }
838c4e82887SLisandro Dalcin 
839951e3c8eSBarry Smith #endif
840