xref: /petsc/src/sys/fileio/sysio.c (revision 28b400f66ebc7ae0049166a2294dfcd3df27e64b)
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 */
277087cfbeSBarry Smith PetscErrorCode  PetscByteSwapEnum(PetscEnum *buff,PetscInt n)
28e5c89e4eSSatish Balay {
296de02169SBarry Smith   PetscInt  i,j;
300b20345dSBarry Smith   PetscEnum tmp = ENUM_DUMMY;
31e0890e22SSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
32e5c89e4eSSatish Balay 
33e5c89e4eSSatish Balay   PetscFunctionBegin;
34e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
35e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
36a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) ptr2[i] = ptr1[sizeof(PetscEnum)-1-i];
37a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) ptr1[i] = ptr2[i];
38e5c89e4eSSatish Balay   }
39e5c89e4eSSatish Balay   PetscFunctionReturn(0);
40e5c89e4eSSatish Balay }
416de02169SBarry Smith 
426de02169SBarry Smith /*
43acfcf0e5SJed Brown   PetscByteSwapBool - Swap bytes in a  PETSc Bool
446de02169SBarry Smith 
456de02169SBarry Smith */
467087cfbeSBarry Smith PetscErrorCode  PetscByteSwapBool(PetscBool *buff,PetscInt n)
476de02169SBarry Smith {
486de02169SBarry Smith   PetscInt  i,j;
49ace3abfcSBarry Smith   PetscBool tmp = PETSC_FALSE;
50e0890e22SSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
516de02169SBarry Smith 
526de02169SBarry Smith   PetscFunctionBegin;
536de02169SBarry Smith   for (j=0; j<n; j++) {
546de02169SBarry Smith     ptr1 = (char*)(buff + j);
55a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscBool); i++) ptr2[i] = ptr1[sizeof(PetscBool)-1-i];
56a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscBool); i++) ptr1[i] = ptr2[i];
576de02169SBarry Smith   }
586de02169SBarry Smith   PetscFunctionReturn(0);
596de02169SBarry Smith }
606de02169SBarry Smith 
61bd1d2e58SBarry Smith /*
626de02169SBarry Smith   PetscByteSwapInt - Swap bytes in a  PETSc integer (which may be 32 or 64 bits)
63bd1d2e58SBarry Smith 
64bd1d2e58SBarry Smith */
657087cfbeSBarry Smith PetscErrorCode  PetscByteSwapInt(PetscInt *buff,PetscInt n)
66bd1d2e58SBarry Smith {
67bd1d2e58SBarry Smith   PetscInt i,j,tmp = 0;
68e0890e22SSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
69bd1d2e58SBarry Smith 
70bd1d2e58SBarry Smith   PetscFunctionBegin;
71bd1d2e58SBarry Smith   for (j=0; j<n; j++) {
72bd1d2e58SBarry Smith     ptr1 = (char*)(buff + j);
73a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) ptr2[i] = ptr1[sizeof(PetscInt)-1-i];
74a297a907SKarl Rupp     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) ptr1[i] = ptr2[i];
75bd1d2e58SBarry Smith   }
76bd1d2e58SBarry Smith   PetscFunctionReturn(0);
77bd1d2e58SBarry Smith }
789e3e4c22SLisandro Dalcin 
799e3e4c22SLisandro Dalcin /*
809e3e4c22SLisandro Dalcin   PetscByteSwapInt64 - Swap bytes in a  PETSc integer (64 bits)
819e3e4c22SLisandro Dalcin 
829e3e4c22SLisandro Dalcin */
839e3e4c22SLisandro Dalcin PetscErrorCode  PetscByteSwapInt64(PetscInt64 *buff,PetscInt n)
849e3e4c22SLisandro Dalcin {
859e3e4c22SLisandro Dalcin   PetscInt   i,j;
869e3e4c22SLisandro Dalcin   PetscInt64 tmp = 0;
879e3e4c22SLisandro Dalcin   char       *ptr1,*ptr2 = (char*)&tmp;
889e3e4c22SLisandro Dalcin 
899e3e4c22SLisandro Dalcin   PetscFunctionBegin;
909e3e4c22SLisandro Dalcin   for (j=0; j<n; j++) {
919e3e4c22SLisandro Dalcin     ptr1 = (char*)(buff + j);
929e3e4c22SLisandro Dalcin     for (i=0; i<(PetscInt)sizeof(PetscInt64); i++) ptr2[i] = ptr1[sizeof(PetscInt64)-1-i];
939e3e4c22SLisandro Dalcin     for (i=0; i<(PetscInt)sizeof(PetscInt64); i++) ptr1[i] = ptr2[i];
949e3e4c22SLisandro Dalcin   }
959e3e4c22SLisandro Dalcin   PetscFunctionReturn(0);
969e3e4c22SLisandro Dalcin }
979e3e4c22SLisandro Dalcin 
98bd1d2e58SBarry Smith /* --------------------------------------------------------- */
99e5c89e4eSSatish Balay /*
100e5c89e4eSSatish Balay   PetscByteSwapShort - Swap bytes in a short
101e5c89e4eSSatish Balay */
1027087cfbeSBarry Smith PetscErrorCode  PetscByteSwapShort(short *buff,PetscInt n)
103e5c89e4eSSatish Balay {
104e5c89e4eSSatish Balay   PetscInt i,j;
105e5c89e4eSSatish Balay   short    tmp;
106e5c89e4eSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
107e5c89e4eSSatish Balay 
108e5c89e4eSSatish Balay   PetscFunctionBegin;
109e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
110e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
111a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(short); i++) ptr2[i] = ptr1[sizeof(short)-1-i];
112a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(short); i++) ptr1[i] = ptr2[i];
113e5c89e4eSSatish Balay   }
114e5c89e4eSSatish Balay   PetscFunctionReturn(0);
115e5c89e4eSSatish Balay }
116972064b6SLisandro Dalcin /*
117972064b6SLisandro Dalcin   PetscByteSwapLong - Swap bytes in a long
118972064b6SLisandro Dalcin */
119972064b6SLisandro Dalcin PetscErrorCode  PetscByteSwapLong(long *buff,PetscInt n)
120972064b6SLisandro Dalcin {
121972064b6SLisandro Dalcin   PetscInt i,j;
122972064b6SLisandro Dalcin   long     tmp;
123972064b6SLisandro Dalcin   char     *ptr1,*ptr2 = (char*)&tmp;
124972064b6SLisandro Dalcin 
125972064b6SLisandro Dalcin   PetscFunctionBegin;
126972064b6SLisandro Dalcin   for (j=0; j<n; j++) {
127972064b6SLisandro Dalcin     ptr1 = (char*)(buff + j);
128972064b6SLisandro Dalcin     for (i=0; i<(PetscInt) sizeof(long); i++) ptr2[i] = ptr1[sizeof(long)-1-i];
129972064b6SLisandro Dalcin     for (i=0; i<(PetscInt) sizeof(long); i++) ptr1[i] = ptr2[i];
130972064b6SLisandro Dalcin   }
131972064b6SLisandro Dalcin   PetscFunctionReturn(0);
132972064b6SLisandro Dalcin }
133e5c89e4eSSatish Balay /* --------------------------------------------------------- */
134e5c89e4eSSatish Balay /*
1354caf0332SSatish Balay   PetscByteSwapReal - Swap bytes in a PetscReal
1364caf0332SSatish Balay */
1374caf0332SSatish Balay PetscErrorCode  PetscByteSwapReal(PetscReal *buff,PetscInt n)
1384caf0332SSatish Balay {
1394caf0332SSatish Balay   PetscInt  i,j;
1404caf0332SSatish Balay   PetscReal tmp,*buff1 = (PetscReal*)buff;
1414caf0332SSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
1424caf0332SSatish Balay 
1434caf0332SSatish Balay   PetscFunctionBegin;
1444caf0332SSatish Balay   for (j=0; j<n; j++) {
1454caf0332SSatish Balay     ptr1 = (char*)(buff1 + j);
1464caf0332SSatish Balay     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr2[i] = ptr1[sizeof(PetscReal)-1-i];
1474caf0332SSatish Balay     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr1[i] = ptr2[i];
1484caf0332SSatish Balay   }
1494caf0332SSatish Balay   PetscFunctionReturn(0);
1504caf0332SSatish Balay }
1514caf0332SSatish Balay /* --------------------------------------------------------- */
1524caf0332SSatish Balay /*
15341f502e3SPatrick Sanan   PetscByteSwapScalar - Swap bytes in a PetscScalar
15441f502e3SPatrick Sanan   The complex case is dealt with with an array of PetscReal, twice as long.
155e5c89e4eSSatish Balay */
1567087cfbeSBarry Smith PetscErrorCode  PetscByteSwapScalar(PetscScalar *buff,PetscInt n)
157e5c89e4eSSatish Balay {
158e5c89e4eSSatish Balay   PetscInt  i,j;
159e5c89e4eSSatish Balay   PetscReal tmp,*buff1 = (PetscReal*)buff;
160e5c89e4eSSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
161e5c89e4eSSatish Balay 
162e5c89e4eSSatish Balay   PetscFunctionBegin;
163e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX)
164e5c89e4eSSatish Balay   n *= 2;
165e5c89e4eSSatish Balay #endif
166e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
167e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
168a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr2[i] = ptr1[sizeof(PetscReal)-1-i];
169a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) ptr1[i] = ptr2[i];
170e5c89e4eSSatish Balay   }
171e5c89e4eSSatish Balay   PetscFunctionReturn(0);
172e5c89e4eSSatish Balay }
173e5c89e4eSSatish Balay /* --------------------------------------------------------- */
174e5c89e4eSSatish Balay /*
175e5c89e4eSSatish Balay   PetscByteSwapDouble - Swap bytes in a double
176e5c89e4eSSatish Balay */
1777087cfbeSBarry Smith PetscErrorCode  PetscByteSwapDouble(double *buff,PetscInt n)
178e5c89e4eSSatish Balay {
179e5c89e4eSSatish Balay   PetscInt i,j;
180e5c89e4eSSatish Balay   double   tmp,*buff1 = (double*)buff;
181e5c89e4eSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
182e5c89e4eSSatish Balay 
183e5c89e4eSSatish Balay   PetscFunctionBegin;
184e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
185e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
186a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(double); i++) ptr2[i] = ptr1[sizeof(double)-1-i];
187a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(double); i++) ptr1[i] = ptr2[i];
188e5c89e4eSSatish Balay   }
189e5c89e4eSSatish Balay   PetscFunctionReturn(0);
190e5c89e4eSSatish Balay }
191e39fd77fSBarry Smith 
192e95bf02fSSatish Balay /*
193e95bf02fSSatish Balay   PetscByteSwapFloat - Swap bytes in a float
194e95bf02fSSatish Balay */
195e95bf02fSSatish Balay PetscErrorCode PetscByteSwapFloat(float *buff,PetscInt n)
196e95bf02fSSatish Balay {
197e95bf02fSSatish Balay   PetscInt i,j;
198e95bf02fSSatish Balay   float    tmp,*buff1 = (float*)buff;
199e95bf02fSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
200e95bf02fSSatish Balay 
201e95bf02fSSatish Balay   PetscFunctionBegin;
202e95bf02fSSatish Balay   for (j=0; j<n; j++) {
203e95bf02fSSatish Balay     ptr1 = (char*)(buff1 + j);
204a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(float); i++) ptr2[i] = ptr1[sizeof(float)-1-i];
205a297a907SKarl Rupp     for (i=0; i<(PetscInt) sizeof(float); i++) ptr1[i] = ptr2[i];
206e95bf02fSSatish Balay   }
207e95bf02fSSatish Balay   PetscFunctionReturn(0);
208e95bf02fSSatish Balay }
209e95bf02fSSatish Balay 
210e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count)
211e39fd77fSBarry Smith {
212e39fd77fSBarry Smith   PetscFunctionBegin;
2135f80ce2aSJacob Faibussowitsch   if      (pdtype == PETSC_INT)    CHKERRQ(PetscByteSwapInt((PetscInt*)data,count));
2145f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_ENUM)   CHKERRQ(PetscByteSwapEnum((PetscEnum*)data,count));
2155f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_BOOL)   CHKERRQ(PetscByteSwapBool((PetscBool*)data,count));
2165f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_SCALAR) CHKERRQ(PetscByteSwapScalar((PetscScalar*)data,count));
2175f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_REAL)   CHKERRQ(PetscByteSwapReal((PetscReal*)data,count));
2185f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_COMPLEX)CHKERRQ(PetscByteSwapReal((PetscReal*)data,2*count));
2195f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_INT64)  CHKERRQ(PetscByteSwapInt64((PetscInt64*)data,count));
2205f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_DOUBLE) CHKERRQ(PetscByteSwapDouble((double*)data,count));
2215f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_FLOAT)  CHKERRQ(PetscByteSwapFloat((float*)data,count));
2225f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_SHORT)  CHKERRQ(PetscByteSwapShort((short*)data,count));
2235f80ce2aSJacob Faibussowitsch   else if (pdtype == PETSC_LONG)   CHKERRQ(PetscByteSwapLong((long*)data,count));
224e39fd77fSBarry Smith   PetscFunctionReturn(0);
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
2359860990eSLisandro Dalcin -  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 
247e5c89e4eSSatish Balay    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
251f253e43cSLisandro Dalcin    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 
2544ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
2554ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
256e5c89e4eSSatish Balay @*/
2579860990eSLisandro Dalcin PetscErrorCode  PetscBinaryRead(int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type)
258e5c89e4eSSatish Balay {
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;
2702c71b3e2SJacob Faibussowitsch   PetscCheckFalse(num < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to read a negative amount of data %" PetscInt_FMT,num);
2719860990eSLisandro Dalcin   if (!num) PetscFunctionReturn(0);
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;
279*28b400f6SJacob 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 
2835f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDataTypeGetSize(type,&typesize));
284e5c89e4eSSatish Balay 
2857a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
2865f80ce2aSJacob Faibussowitsch   CHKERRQ(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);
2905f80ce2aSJacob Faibussowitsch     CHKERRQ(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 */
3032c71b3e2SJacob Faibussowitsch     PetscCheckFalse(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   }
3082c71b3e2SJacob Faibussowitsch   PetscCheckFalse(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;
3175f80ce2aSJacob Faibussowitsch     if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwapDouble(pdouble,cnt));
3189860990eSLisandro Dalcin     for (i=0; i<cnt; i++) preal[i] = pdouble[i];
3195f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree(pdouble));
3207a881295SBarry Smith     PetscFunctionReturn(0);
3217a881295SBarry Smith   }
3227a881295SBarry Smith #endif
3237a881295SBarry Smith 
3245f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(ptmp,type,num));
325e5c89e4eSSatish Balay 
32605acbc63SBarry Smith   if (type == PETSC_FUNCTION) {
3272d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
3285f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDLSym(NULL,fname,(void**)data));
3292d53ad75SBarry Smith #else
3309860990eSLisandro Dalcin     *(void**)data = NULL;
3312d53ad75SBarry Smith #endif
33205acbc63SBarry Smith     free(fname);
3332d53ad75SBarry Smith   }
334e5c89e4eSSatish Balay   PetscFunctionReturn(0);
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
346f253e43cSLisandro Dalcin -  type   - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
347e5c89e4eSSatish Balay 
348e5c89e4eSSatish Balay    Level: advanced
349e5c89e4eSSatish Balay 
350e5c89e4eSSatish Balay    Notes:
351e5c89e4eSSatish Balay    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
355f253e43cSLisandro Dalcin    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 
35841f502e3SPatrick Sanan    If running with __float128 precision the output is in __float128 unless one uses the -binary_write_double option
3590da86b62SBarry Smith 
360e5c89e4eSSatish Balay    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 
3704ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
3714ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
372e5c89e4eSSatish Balay @*/
373f253e43cSLisandro Dalcin PetscErrorCode  PetscBinaryWrite(int fd,const void *p,PetscInt n,PetscDataType type)
374e5c89e4eSSatish Balay {
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;
3892c71b3e2SJacob Faibussowitsch   PetscCheckFalse(n < 0,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %" PetscInt_FMT,n);
390e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
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));
398*28b400f6SJacob Faibussowitsch     PetscCheck(fname,PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
39905acbc63SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
4002c71b3e2SJacob Faibussowitsch     PetscCheckFalse(n > 1,PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only binary view a single function at a time");
4015f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFPTFind(*(void**)p,&fnametmp));
4025f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscStrncpy(fname,fnametmp,m));
40305acbc63SBarry Smith #else
4045f80ce2aSJacob Faibussowitsch     CHKERRQ(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)
4125f80ce2aSJacob Faibussowitsch   CHKERRQ(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;
4165f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscMalloc1(n,&ppp));
4170da86b62SBarry Smith     pv = (PetscReal*)pp;
4180da86b62SBarry Smith     for (i=0; i<n; i++) {
4190da86b62SBarry Smith       ppp[i] = (double) pv[i];
4200da86b62SBarry Smith     }
4210da86b62SBarry Smith     pp   = (char*)ppp;
4220da86b62SBarry Smith     ptmp = (char*)ppp;
4230da86b62SBarry Smith   }
4240da86b62SBarry Smith #endif
4250da86b62SBarry Smith 
42641f502e3SPatrick Sanan   if (wtype == PETSC_INT)          m *= sizeof(PetscInt);
42741f502e3SPatrick Sanan   else if (wtype == PETSC_SCALAR)  m *= sizeof(PetscScalar);
428501ae109SJose E. Roman #if defined(PETSC_HAVE_COMPLEX)
429501ae109SJose E. Roman   else if (wtype == PETSC_COMPLEX) m *= sizeof(PetscComplex);
430501ae109SJose E. Roman #endif
4314caf0332SSatish Balay   else if (wtype == PETSC_REAL)    m *= sizeof(PetscReal);
43241f502e3SPatrick Sanan   else if (wtype == PETSC_DOUBLE)  m *= sizeof(double);
43341f502e3SPatrick Sanan   else if (wtype == PETSC_FLOAT)   m *= sizeof(float);
43441f502e3SPatrick Sanan   else if (wtype == PETSC_SHORT)   m *= sizeof(short);
435972064b6SLisandro Dalcin   else if (wtype == PETSC_LONG)    m *= sizeof(long);
43641f502e3SPatrick Sanan   else if (wtype == PETSC_CHAR)    m *= sizeof(char);
43741f502e3SPatrick Sanan   else if (wtype == PETSC_ENUM)    m *= sizeof(PetscEnum);
43841f502e3SPatrick Sanan   else if (wtype == PETSC_BOOL)    m *= sizeof(PetscBool);
4399e3e4c22SLisandro Dalcin   else if (wtype == PETSC_INT64)   m *= sizeof(PetscInt64);
44041f502e3SPatrick Sanan   else if (wtype == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char);
441e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
442e5c89e4eSSatish Balay 
4435f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap((void*)ptmp,wtype,n));
444e5c89e4eSSatish Balay 
445e5c89e4eSSatish Balay   while (m) {
446e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
447e5c89e4eSSatish Balay     err   = write(fd,pp,wsize);
448e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
4492c71b3e2SJacob Faibussowitsch     PetscCheckFalse(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);
450e5c89e4eSSatish Balay     m  -= wsize;
451e5c89e4eSSatish Balay     pp += wsize;
452e5c89e4eSSatish Balay   }
453e5c89e4eSSatish Balay 
4545f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap((void*)ptmp,wtype,n));
455f253e43cSLisandro Dalcin 
45605acbc63SBarry Smith   if (type == PETSC_FUNCTION) {
45705acbc63SBarry Smith     free(fname);
45805acbc63SBarry Smith   }
4590da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
460501ae109SJose E. Roman   if ((type == PETSC_SCALAR || type == PETSC_REAL || type == PETSC_COMPLEX) && writedouble) {
4615f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscFree(ppp));
4620da86b62SBarry Smith   }
4630da86b62SBarry Smith #endif
464e5c89e4eSSatish Balay   PetscFunctionReturn(0);
465e5c89e4eSSatish Balay }
466e5c89e4eSSatish Balay 
467e5c89e4eSSatish Balay /*@C
468e5c89e4eSSatish Balay    PetscBinaryOpen - Opens a PETSc binary file.
469e5c89e4eSSatish Balay 
470e5c89e4eSSatish Balay    Not Collective
471e5c89e4eSSatish Balay 
472e5c89e4eSSatish Balay    Input Parameters:
473e5c89e4eSSatish Balay +  name - filename
474cc843e7aSLisandro Dalcin -  mode - open mode of binary file, one of FILE_MODE_READ, FILE_MODE_WRITE, FILE_MODE_APPEND
475e5c89e4eSSatish Balay 
476e5c89e4eSSatish Balay    Output Parameter:
477e5c89e4eSSatish Balay .  fd - the file
478e5c89e4eSSatish Balay 
479e5c89e4eSSatish Balay    Level: advanced
480e5c89e4eSSatish Balay 
48195452b02SPatrick Sanan    Notes:
48295452b02SPatrick Sanan     Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in
483e5c89e4eSSatish Balay    big-endian format. This means the file can be accessed using PetscBinaryOpen() and
484e5c89e4eSSatish Balay    PetscBinaryRead() and PetscBinaryWrite() on any machine.
485e5c89e4eSSatish Balay 
4864ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(),
4874ebed01fSBarry Smith           PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
48845c64e65SBarry Smith 
489e5c89e4eSSatish Balay @*/
4907087cfbeSBarry Smith PetscErrorCode  PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd)
491e5c89e4eSSatish Balay {
492e5c89e4eSSatish Balay   PetscFunctionBegin;
493cc843e7aSLisandro Dalcin   switch (mode) {
494cc843e7aSLisandro Dalcin   case FILE_MODE_READ:   *fd = open(name,O_BINARY|O_RDONLY,0); break;
495cc843e7aSLisandro Dalcin   case FILE_MODE_WRITE:  *fd = open(name,O_BINARY|O_WRONLY|O_CREAT|O_TRUNC,0666); break;
496cc843e7aSLisandro Dalcin   case FILE_MODE_APPEND: *fd = open(name,O_BINARY|O_WRONLY|O_APPEND,0); break;
49798921bdaSJacob Faibussowitsch   default: SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported file mode %s",PetscFileModes[mode]);
498e5c89e4eSSatish Balay   }
4992c71b3e2SJacob Faibussowitsch   PetscCheckFalse(*fd == -1,PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for %s",name,PetscFileModes[mode]);
500e5c89e4eSSatish Balay   PetscFunctionReturn(0);
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 
5134ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
5144ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
515e5c89e4eSSatish Balay @*/
5167087cfbeSBarry Smith PetscErrorCode  PetscBinaryClose(int fd)
517e5c89e4eSSatish Balay {
518e5c89e4eSSatish Balay   PetscFunctionBegin;
5192c71b3e2SJacob Faibussowitsch   PetscCheckFalse(close(fd),PETSC_COMM_SELF,PETSC_ERR_SYS,"close() failed on file descriptor");
520e5c89e4eSSatish Balay   PetscFunctionReturn(0);
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
530ff553b35SMatthew Knepley .  off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
531e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
532ff553b35SMatthew Knepley -  whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file
533ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_CUR then off is an offset from the current location
534ff553b35SMatthew Knepley             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 
541e5c89e4eSSatish Balay    Notes:
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
544e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
545e5c89e4eSSatish Balay    to determine the offset or location.
546e5c89e4eSSatish Balay 
5474ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
5484ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
549e5c89e4eSSatish Balay @*/
5507087cfbeSBarry Smith PetscErrorCode  PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
551e5c89e4eSSatish Balay {
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
566e5c89e4eSSatish Balay   PetscFunctionReturn(0);
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
5789860990eSLisandro Dalcin -  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:
587e5c89e4eSSatish Balay    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 
5921d280d73SBarry 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 
5974ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
5984ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
599e5c89e4eSSatish Balay @*/
6009860990eSLisandro Dalcin PetscErrorCode  PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type)
601e5c89e4eSSatish Balay {
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;
615*28b400f6SJacob Faibussowitsch     PetscCheck(fname,PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
6162d53ad75SBarry Smith   }
6172d53ad75SBarry Smith 
6185f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(comm,&rank));
6195f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(comm,&size));
620dd400576SPatrick Sanan   if (rank == 0) {
6219860990eSLisandro Dalcin     ibuf[0] = PetscBinaryRead(fd,data,num,count?&ibuf[1]:NULL,type);
622e5c89e4eSSatish Balay   }
6235f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Bcast(ibuf,2,MPIU_INT,0,comm));
6245f80ce2aSJacob Faibussowitsch   CHKERRQ((PetscErrorCode)ibuf[0]);
6258b8bf82cSBarry Smith 
6268b8bf82cSBarry 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 */
6278b8bf82cSBarry Smith   if (size > 1) {
6285f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDataTypeToMPIDataType(type,&mtype));
6295f80ce2aSJacob Faibussowitsch     CHKERRMPI(MPI_Bcast(data,count?ibuf[1]:num,mtype,0,comm));
6308b8bf82cSBarry Smith   }
6319860990eSLisandro Dalcin   if (count) *count = ibuf[1];
6322d53ad75SBarry Smith 
633e366c363SBarry Smith   if (type == PETSC_FUNCTION) {
6342d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
6355f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)fptr));
6362d53ad75SBarry Smith #else
6379860990eSLisandro Dalcin     *(void**)fptr = NULL;
6382d53ad75SBarry Smith #endif
639e366c363SBarry Smith     free(fname);
6402d53ad75SBarry Smith   }
641e5c89e4eSSatish Balay   PetscFunctionReturn(0);
642e5c89e4eSSatish Balay }
643e5c89e4eSSatish Balay 
644e5c89e4eSSatish Balay /*@C
6451d280d73SBarry Smith    PetscBinarySynchronizedWrite - writes to a binary file.
646e5c89e4eSSatish Balay 
647d083f849SBarry Smith    Collective
648e5c89e4eSSatish Balay 
649e5c89e4eSSatish Balay    Input Parameters:
650e5c89e4eSSatish Balay +  comm - the MPI communicator
651e5c89e4eSSatish Balay .  fd - the file
652e5c89e4eSSatish Balay .  n  - the number of items to write
653e5c89e4eSSatish Balay .  p - the buffer
654e5c89e4eSSatish Balay -  type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
655e5c89e4eSSatish Balay 
656e5c89e4eSSatish Balay    Level: developer
657e5c89e4eSSatish Balay 
658e5c89e4eSSatish Balay    Notes:
659e5c89e4eSSatish Balay    Process 0 does a PetscBinaryWrite()
660e5c89e4eSSatish Balay 
6611d280d73SBarry Smith    PetscBinarySynchronizedWrite() uses byte swapping to work on all machines.
662e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
663e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
664e5c89e4eSSatish Balay    binary file may be read on any machine.
665e5c89e4eSSatish Balay 
66695452b02SPatrick Sanan    Notes:
66795452b02SPatrick Sanan     because byte-swapping may be done on the values in data it cannot be declared const
668300a7f5bSBarry Smith 
6691d280d73SBarry Smith    WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0,
6701d280d73SBarry Smith    while PetscSynchronizedFPrintf() has all processes print their strings in order.
6711d280d73SBarry Smith 
6724ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(),
6734ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
674e5c89e4eSSatish Balay @*/
675f253e43cSLisandro Dalcin PetscErrorCode  PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,const void *p,PetscInt n,PetscDataType type)
676e5c89e4eSSatish Balay {
677e5c89e4eSSatish Balay   PetscMPIInt    rank;
678e5c89e4eSSatish Balay 
679e5c89e4eSSatish Balay   PetscFunctionBegin;
6805f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(comm,&rank));
681dd400576SPatrick Sanan   if (rank == 0) {
6825f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscBinaryWrite(fd,p,n,type));
683e5c89e4eSSatish Balay   }
684e5c89e4eSSatish Balay   PetscFunctionReturn(0);
685e5c89e4eSSatish Balay }
686e5c89e4eSSatish Balay 
687e5c89e4eSSatish Balay /*@C
6881d280d73SBarry Smith    PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file.
689e5c89e4eSSatish Balay 
690e5c89e4eSSatish Balay    Input Parameters:
691e5c89e4eSSatish Balay +  fd - the file
692e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
693e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
694e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
695e5c89e4eSSatish Balay -  off    - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
696e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
697e5c89e4eSSatish Balay 
698e5c89e4eSSatish Balay    Output Parameter:
699e5c89e4eSSatish Balay .   offset - new offset in file
700e5c89e4eSSatish Balay 
701e5c89e4eSSatish Balay    Level: developer
702e5c89e4eSSatish Balay 
703e5c89e4eSSatish Balay    Notes:
704e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
705e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
706e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
707e5c89e4eSSatish Balay    to determine the offset or location.
708e5c89e4eSSatish Balay 
7094ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
7104ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
711e5c89e4eSSatish Balay @*/
7127087cfbeSBarry Smith PetscErrorCode  PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
713e5c89e4eSSatish Balay {
714e5c89e4eSSatish Balay   PetscMPIInt    rank;
715e5c89e4eSSatish Balay 
716e5c89e4eSSatish Balay   PetscFunctionBegin;
7175f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_rank(comm,&rank));
718dd400576SPatrick Sanan   if (rank == 0) {
7195f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscBinarySeek(fd,off,whence,offset));
720e5c89e4eSSatish Balay   }
721e5c89e4eSSatish Balay   PetscFunctionReturn(0);
722e5c89e4eSSatish Balay }
723e5c89e4eSSatish Balay 
7240fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO)
725e39fd77fSBarry Smith 
726951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32)
727e39fd77fSBarry Smith /*
728e39fd77fSBarry Smith       MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions.
729e39fd77fSBarry Smith     These are set into MPI in PetscInitialize() via MPI_Register_datarep()
730e39fd77fSBarry Smith 
731e39fd77fSBarry Smith     Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode)
732e39fd77fSBarry Smith 
733951e3c8eSBarry Smith     The next three routines are not used because MPICH does not support their use
734e39fd77fSBarry Smith 
735e39fd77fSBarry Smith */
7368cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state)
737e39fd77fSBarry Smith {
738e39fd77fSBarry Smith   MPI_Aint    ub;
739e39fd77fSBarry Smith   PetscMPIInt ierr;
740e39fd77fSBarry Smith 
741e39fd77fSBarry Smith   ierr = MPI_Type_get_extent(datatype,&ub,file_extent);
742e39fd77fSBarry Smith   return ierr;
743e39fd77fSBarry Smith }
744e39fd77fSBarry Smith 
7458cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
746e39fd77fSBarry Smith {
747e39fd77fSBarry Smith   PetscDataType pdtype;
748e39fd77fSBarry Smith   PetscMPIInt   ierr;
749e39fd77fSBarry Smith   size_t        dsize;
750e39fd77fSBarry Smith 
7515f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(datatype,&pdtype));
7525f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDataTypeGetSize(pdtype,&dsize));
753e39fd77fSBarry Smith 
754e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
755e39fd77fSBarry Smith   userbuf = ((char*)userbuf) + dsize*position;
756e39fd77fSBarry Smith 
7575f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemcpy(userbuf,filebuf,count*dsize));
7585f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(userbuf,pdtype,count));
759e39fd77fSBarry Smith   return ierr;
760e39fd77fSBarry Smith }
761e39fd77fSBarry Smith 
762e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
763e39fd77fSBarry Smith {
764e39fd77fSBarry Smith   PetscDataType pdtype;
765e39fd77fSBarry Smith   PetscMPIInt   ierr;
766e39fd77fSBarry Smith   size_t        dsize;
767e39fd77fSBarry Smith 
7685f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(datatype,&pdtype));
7695f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscDataTypeGetSize(pdtype,&dsize));
770e39fd77fSBarry Smith 
771e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
772e39fd77fSBarry Smith   userbuf = ((char*)userbuf) + dsize*position;
773e39fd77fSBarry Smith 
7745f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemcpy(filebuf,userbuf,count*dsize));
7755f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(filebuf,pdtype,count));
776e39fd77fSBarry Smith   return ierr;
777e39fd77fSBarry Smith }
778951e3c8eSBarry Smith #endif
779e39fd77fSBarry Smith 
780e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
781e39fd77fSBarry Smith {
782e39fd77fSBarry Smith   PetscDataType  pdtype;
78330815ce0SLisandro Dalcin 
784e39fd77fSBarry Smith   PetscFunctionBegin;
7855f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
7865f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
7875f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_write_all(fd,data,cnt,dtype,status));
7885f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
789a6796414SBarry Smith   PetscFunctionReturn(0);
790e39fd77fSBarry Smith }
791e39fd77fSBarry Smith 
792e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
793e39fd77fSBarry Smith {
794e39fd77fSBarry Smith   PetscDataType  pdtype;
795e39fd77fSBarry Smith 
796e39fd77fSBarry Smith   PetscFunctionBegin;
7975f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
7985f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_read_all(fd,data,cnt,dtype,status));
7995f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
800a6796414SBarry Smith   PetscFunctionReturn(0);
801e39fd77fSBarry Smith }
80230815ce0SLisandro Dalcin 
803c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
804c4e82887SLisandro Dalcin {
805c4e82887SLisandro Dalcin   PetscDataType  pdtype;
806c4e82887SLisandro Dalcin 
807c4e82887SLisandro Dalcin   PetscFunctionBegin;
8085f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
8095f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
8105f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_write_at(fd,off,data,cnt,dtype,status));
8115f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
812c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
813c4e82887SLisandro Dalcin }
814c4e82887SLisandro Dalcin 
815c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
816c4e82887SLisandro Dalcin {
817c4e82887SLisandro Dalcin   PetscDataType  pdtype;
818c4e82887SLisandro Dalcin 
819c4e82887SLisandro Dalcin   PetscFunctionBegin;
8205f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
8215f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_read_at(fd,off,data,cnt,dtype,status));
8225f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
823c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
824c4e82887SLisandro Dalcin }
825c4e82887SLisandro Dalcin 
826c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
827c4e82887SLisandro Dalcin {
828c4e82887SLisandro Dalcin   PetscDataType  pdtype;
829c4e82887SLisandro Dalcin 
830c4e82887SLisandro Dalcin   PetscFunctionBegin;
8315f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
8325f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
8335f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_write_at_all(fd,off,data,cnt,dtype,status));
8345f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
835c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
836c4e82887SLisandro Dalcin }
837c4e82887SLisandro Dalcin 
838c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
839c4e82887SLisandro Dalcin {
840c4e82887SLisandro Dalcin   PetscDataType  pdtype;
841c4e82887SLisandro Dalcin 
842c4e82887SLisandro Dalcin   PetscFunctionBegin;
8435f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMPIDataTypeToPetscDataType(dtype,&pdtype));
8445f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_File_read_at_all(fd,off,data,cnt,dtype,status));
8455f80ce2aSJacob Faibussowitsch   if (!PetscBinaryBigEndian()) CHKERRQ(PetscByteSwap(data,pdtype,cnt));
846c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
847c4e82887SLisandro Dalcin }
848c4e82887SLisandro Dalcin 
849951e3c8eSBarry Smith #endif
850