xref: /petsc/src/sys/fileio/sysio.c (revision 8b8bf82c4aea97b8c649e849e281942e57140e67)
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   PetscErrorCode ierr;
213e39fd77fSBarry Smith 
214e39fd77fSBarry Smith   PetscFunctionBegin;
215e39fd77fSBarry Smith   if      (pdtype == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);}
216e39fd77fSBarry Smith   else if (pdtype == PETSC_ENUM)   {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);}
217acfcf0e5SJed Brown   else if (pdtype == PETSC_BOOL)   {ierr = PetscByteSwapBool((PetscBool*)data,count);CHKERRQ(ierr);}
218e39fd77fSBarry Smith   else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);}
2194caf0332SSatish Balay   else if (pdtype == PETSC_REAL)   {ierr = PetscByteSwapReal((PetscReal*)data,count);CHKERRQ(ierr);}
2209860990eSLisandro Dalcin   else if (pdtype == PETSC_COMPLEX){ierr = PetscByteSwapReal((PetscReal*)data,2*count);CHKERRQ(ierr);}
2219e3e4c22SLisandro Dalcin   else if (pdtype == PETSC_INT64)  {ierr = PetscByteSwapInt64((PetscInt64*)data,count);CHKERRQ(ierr);}
222e39fd77fSBarry Smith   else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);}
223e95bf02fSSatish Balay   else if (pdtype == PETSC_FLOAT)  {ierr = PetscByteSwapFloat((float*)data,count);CHKERRQ(ierr);}
224e39fd77fSBarry Smith   else if (pdtype == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);}
225972064b6SLisandro Dalcin   else if (pdtype == PETSC_LONG)   {ierr = PetscByteSwapLong((long*)data,count);CHKERRQ(ierr);}
226e39fd77fSBarry Smith   PetscFunctionReturn(0);
227e39fd77fSBarry Smith }
228e39fd77fSBarry Smith 
2299860990eSLisandro Dalcin /*@C
230e5c89e4eSSatish Balay    PetscBinaryRead - Reads from a binary file.
231e5c89e4eSSatish Balay 
232e5c89e4eSSatish Balay    Not Collective
233e5c89e4eSSatish Balay 
234e5c89e4eSSatish Balay    Input Parameters:
2359860990eSLisandro Dalcin +  fd - the file descriptor
2369860990eSLisandro Dalcin .  num  - the maximum number of items to read
2379860990eSLisandro Dalcin -  type - the type of items to read (PETSC_INT, PETSC_REAL, PETSC_SCALAR, etc.)
238e5c89e4eSSatish Balay 
239e5c89e4eSSatish Balay    Output Parameters:
2409860990eSLisandro Dalcin +  data - the buffer
2419860990eSLisandro Dalcin -  count - the number of items read, optional
242e5c89e4eSSatish Balay 
243e5c89e4eSSatish Balay    Level: developer
244e5c89e4eSSatish Balay 
245e5c89e4eSSatish Balay    Notes:
2469860990eSLisandro Dalcin    If count is not provided and the number of items read is less than
2479860990eSLisandro Dalcin    the maximum number of items to read, then this routine errors.
2489860990eSLisandro Dalcin 
249e5c89e4eSSatish Balay    PetscBinaryRead() uses byte swapping to work on all machines; the files
250f253e43cSLisandro Dalcin    are written to file ALWAYS using big-endian ordering. On little-endian machines the numbers
251f253e43cSLisandro Dalcin    are converted to the little-endian format when they are read in from the file.
252f253e43cSLisandro Dalcin    When PETSc is ./configure with --with-64-bit-indices the integers are written to the
253f253e43cSLisandro Dalcin    file as 64 bit integers, this means they can only be read back in when the option --with-64-bit-indices
25454f21887SBarry Smith    is used.
255e5c89e4eSSatish Balay 
2564ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
2574ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
258e5c89e4eSSatish Balay @*/
2599860990eSLisandro Dalcin PetscErrorCode  PetscBinaryRead(int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type)
260e5c89e4eSSatish Balay {
2619860990eSLisandro Dalcin   size_t            typesize, m = (size_t) num, n = 0, maxblock = 65536;
2629860990eSLisandro Dalcin   char              *p = (char*)data;
2637a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
264cba51d77SBarry Smith   PetscBool         readdouble = PETSC_FALSE;
2659860990eSLisandro Dalcin   double            *pdouble;
2667a881295SBarry Smith #endif
2679860990eSLisandro Dalcin   void              *ptmp = data;
26805acbc63SBarry Smith   char              *fname = NULL;
2699860990eSLisandro Dalcin   PetscErrorCode    ierr;
270e5c89e4eSSatish Balay 
271e5c89e4eSSatish Balay   PetscFunctionBegin;
2729860990eSLisandro Dalcin   if (count) *count = 0;
2739860990eSLisandro Dalcin   if (num < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to read a negative amount of data %D",num);
2749860990eSLisandro Dalcin   if (!num) PetscFunctionReturn(0);
275e5c89e4eSSatish Balay 
2762d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
2772d53ad75SBarry Smith     m     = 64;
2782d53ad75SBarry Smith     type  = PETSC_CHAR;
27905acbc63SBarry Smith     fname = (char*)malloc(m*sizeof(char));
2809860990eSLisandro Dalcin     p     = (char*)fname;
2812d53ad75SBarry Smith     ptmp  = (void*)fname;
2829860990eSLisandro Dalcin     if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
2832d53ad75SBarry Smith   }
2849860990eSLisandro Dalcin   if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m);
2852d53ad75SBarry Smith 
2869860990eSLisandro Dalcin   ierr = PetscDataTypeGetSize(type,&typesize);CHKERRQ(ierr);
287e5c89e4eSSatish Balay 
2887a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
289c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-binary_read_double",&readdouble,NULL);CHKERRQ(ierr);
2907a881295SBarry Smith   /* If using __float128 precision we still read in doubles from file */
2919860990eSLisandro Dalcin   if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) {
2929860990eSLisandro Dalcin     PetscInt cnt = num * ((type == PETSC_REAL) ? 1 : 2);
2939860990eSLisandro Dalcin     ierr = PetscMalloc1(cnt,&pdouble);CHKERRQ(ierr);
2949860990eSLisandro Dalcin     p = (char*)pdouble;
2959860990eSLisandro Dalcin     typesize /= 2;
2967a881295SBarry Smith   }
2977a881295SBarry Smith #endif
2987a881295SBarry Smith 
2999860990eSLisandro Dalcin   m *= typesize;
3009860990eSLisandro Dalcin 
301e5c89e4eSSatish Balay   while (m) {
3029860990eSLisandro Dalcin     size_t len = (m < maxblock) ? m : maxblock;
3039860990eSLisandro Dalcin     int    ret = (int)read(fd,p,len);
3049860990eSLisandro Dalcin     if (ret < 0 && errno == EINTR) continue;
3059860990eSLisandro Dalcin     if (!ret && len > 0) break; /* Proxy for EOF */
3069860990eSLisandro Dalcin     if (ret < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno);
3079860990eSLisandro Dalcin     m -= ret;
3089860990eSLisandro Dalcin     p += ret;
3099860990eSLisandro Dalcin     n += ret;
310e5c89e4eSSatish Balay   }
3119860990eSLisandro Dalcin   if (m && !count) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Read past end of file");
3129860990eSLisandro Dalcin 
3139860990eSLisandro Dalcin   num = (PetscInt)(n/typesize); /* Should we require `n % typesize == 0` ? */
3149860990eSLisandro Dalcin   if (count) *count = num;      /* TODO: This is most likely wrong for PETSC_BIT_LOGICAL */
3157a881295SBarry Smith 
3167a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
3179860990eSLisandro Dalcin   if ((type == PETSC_REAL || type == PETSC_COMPLEX) && readdouble) {
3189860990eSLisandro Dalcin     PetscInt  i, cnt = num * ((type == PETSC_REAL) ? 1 : 2);
3199860990eSLisandro Dalcin     PetscReal *preal = (PetscReal*)data;
32030815ce0SLisandro Dalcin     if (!PetscBinaryBigEndian()) {ierr = PetscByteSwapDouble(pdouble,cnt);CHKERRQ(ierr);}
3219860990eSLisandro Dalcin     for (i=0; i<cnt; i++) preal[i] = pdouble[i];
3229860990eSLisandro Dalcin     ierr = PetscFree(pdouble);CHKERRQ(ierr);
3237a881295SBarry Smith     PetscFunctionReturn(0);
3247a881295SBarry Smith   }
3257a881295SBarry Smith #endif
3267a881295SBarry Smith 
32730815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(ptmp,type,num);CHKERRQ(ierr);}
328e5c89e4eSSatish Balay 
32905acbc63SBarry Smith   if (type == PETSC_FUNCTION) {
3302d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
3319860990eSLisandro Dalcin     ierr = PetscDLSym(NULL,fname,(void**)data);CHKERRQ(ierr);
3322d53ad75SBarry Smith #else
3339860990eSLisandro Dalcin     *(void**)data = NULL;
3342d53ad75SBarry Smith #endif
33505acbc63SBarry Smith     free(fname);
3362d53ad75SBarry Smith   }
337e5c89e4eSSatish Balay   PetscFunctionReturn(0);
338e5c89e4eSSatish Balay }
3399860990eSLisandro Dalcin 
340a9acdec7SBarry Smith /*@C
341e5c89e4eSSatish Balay    PetscBinaryWrite - Writes to a binary file.
342e5c89e4eSSatish Balay 
343e5c89e4eSSatish Balay    Not Collective
344e5c89e4eSSatish Balay 
345e5c89e4eSSatish Balay    Input Parameters:
346e5c89e4eSSatish Balay +  fd     - the file
347e5c89e4eSSatish Balay .  p      - the buffer
348e5c89e4eSSatish Balay .  n      - the number of items to write
349f253e43cSLisandro Dalcin -  type   - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
350e5c89e4eSSatish Balay 
351e5c89e4eSSatish Balay    Level: advanced
352e5c89e4eSSatish Balay 
353e5c89e4eSSatish Balay    Notes:
354e5c89e4eSSatish Balay    PetscBinaryWrite() uses byte swapping to work on all machines; the files
355f253e43cSLisandro Dalcin    are written using big-endian ordering to the file. On little-endian machines the numbers
356e5c89e4eSSatish Balay    are converted to the big-endian format when they are written to disk.
357f253e43cSLisandro Dalcin    When PETSc is ./configure with --with-64-bit-indices the integers are written to the
358f253e43cSLisandro Dalcin    file as 64 bit integers, this means they can only be read back in when the option --with-64-bit-indices
35954f21887SBarry Smith    is used.
360e5c89e4eSSatish Balay 
36141f502e3SPatrick Sanan    If running with __float128 precision the output is in __float128 unless one uses the -binary_write_double option
3620da86b62SBarry Smith 
363e5c89e4eSSatish Balay    The Buffer p should be read-write buffer, and not static data.
364e5c89e4eSSatish Balay    This way, byte-swapping is done in-place, and then the buffer is
365e5c89e4eSSatish Balay    written to the file.
366e5c89e4eSSatish Balay 
367e5c89e4eSSatish Balay    This routine restores the original contents of the buffer, after
368e5c89e4eSSatish Balay    it is written to the file. This is done by byte-swapping in-place
369f253e43cSLisandro Dalcin    the second time.
370e5c89e4eSSatish Balay 
371300a7f5bSBarry Smith    Because byte-swapping may be done on the values in data it cannot be declared const
372300a7f5bSBarry Smith 
373e5c89e4eSSatish Balay 
3744ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
3754ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
376e5c89e4eSSatish Balay @*/
377f253e43cSLisandro Dalcin PetscErrorCode  PetscBinaryWrite(int fd,const void *p,PetscInt n,PetscDataType type)
378e5c89e4eSSatish Balay {
379f253e43cSLisandro Dalcin   const char     *pp = (char*)p;
380e5c89e4eSSatish Balay   int            err,wsize;
381e5c89e4eSSatish Balay   size_t         m = (size_t)n,maxblock=65536;
382e5c89e4eSSatish Balay   PetscErrorCode ierr;
383f253e43cSLisandro Dalcin   const void     *ptmp = p;
38405acbc63SBarry Smith   char           *fname = NULL;
3850da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
386df40af56SBarry Smith   PetscBool      writedouble = PETSC_FALSE;
3870da86b62SBarry Smith   double         *ppp;
3880da86b62SBarry Smith   PetscReal      *pv;
3890da86b62SBarry Smith   PetscInt       i;
3900da86b62SBarry Smith #endif
39141f502e3SPatrick Sanan   PetscDataType  wtype = type;
392e5c89e4eSSatish Balay 
393e5c89e4eSSatish Balay   PetscFunctionBegin;
394e32f2f54SBarry Smith   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n);
395e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
396e5c89e4eSSatish Balay 
3972d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
3982d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
3992d53ad75SBarry Smith     const char *fnametmp;
4002d53ad75SBarry Smith #endif
4012d53ad75SBarry Smith     m     = 64;
402e25ab156SSatish Balay     fname = (char*)malloc(m*sizeof(char));
40305acbc63SBarry Smith     if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
40405acbc63SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
40505acbc63SBarry Smith     if (n > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only binary view a single function at a time");
40605acbc63SBarry Smith     ierr = PetscFPTFind(*(void**)p,&fnametmp);CHKERRQ(ierr);
40705acbc63SBarry Smith     ierr = PetscStrncpy(fname,fnametmp,m);CHKERRQ(ierr);
40805acbc63SBarry Smith #else
40905acbc63SBarry Smith     ierr = PetscStrncpy(fname,"",m);CHKERRQ(ierr);
41005acbc63SBarry Smith #endif
41101963f56SBarry Smith     wtype = PETSC_CHAR;
4122d53ad75SBarry Smith     pp    = (char*)fname;
4132d53ad75SBarry Smith     ptmp  = (void*)fname;
4142d53ad75SBarry Smith   }
4152d53ad75SBarry Smith 
4160da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
4170da86b62SBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-binary_write_double",&writedouble,NULL);CHKERRQ(ierr);
4180da86b62SBarry Smith   /* If using __float128 precision we still write in doubles to file */
419501ae109SJose E. Roman   if ((type == PETSC_SCALAR || type == PETSC_REAL || type == PETSC_COMPLEX) && writedouble) {
42041f502e3SPatrick Sanan     wtype = PETSC_DOUBLE;
4210da86b62SBarry Smith     ierr = PetscMalloc1(n,&ppp);CHKERRQ(ierr);
4220da86b62SBarry Smith     pv = (PetscReal*)pp;
4230da86b62SBarry Smith     for (i=0; i<n; i++) {
4240da86b62SBarry Smith       ppp[i] = (double) pv[i];
4250da86b62SBarry Smith     }
4260da86b62SBarry Smith     pp   = (char*)ppp;
4270da86b62SBarry Smith     ptmp = (char*)ppp;
4280da86b62SBarry Smith   }
4290da86b62SBarry Smith #endif
4300da86b62SBarry Smith 
43141f502e3SPatrick Sanan   if (wtype == PETSC_INT)          m *= sizeof(PetscInt);
43241f502e3SPatrick Sanan   else if (wtype == PETSC_SCALAR)  m *= sizeof(PetscScalar);
433501ae109SJose E. Roman #if defined(PETSC_HAVE_COMPLEX)
434501ae109SJose E. Roman   else if (wtype == PETSC_COMPLEX) m *= sizeof(PetscComplex);
435501ae109SJose E. Roman #endif
4364caf0332SSatish Balay   else if (wtype == PETSC_REAL)    m *= sizeof(PetscReal);
43741f502e3SPatrick Sanan   else if (wtype == PETSC_DOUBLE)  m *= sizeof(double);
43841f502e3SPatrick Sanan   else if (wtype == PETSC_FLOAT)   m *= sizeof(float);
43941f502e3SPatrick Sanan   else if (wtype == PETSC_SHORT)   m *= sizeof(short);
440972064b6SLisandro Dalcin   else if (wtype == PETSC_LONG)    m *= sizeof(long);
44141f502e3SPatrick Sanan   else if (wtype == PETSC_CHAR)    m *= sizeof(char);
44241f502e3SPatrick Sanan   else if (wtype == PETSC_ENUM)    m *= sizeof(PetscEnum);
44341f502e3SPatrick Sanan   else if (wtype == PETSC_BOOL)    m *= sizeof(PetscBool);
4449e3e4c22SLisandro Dalcin   else if (wtype == PETSC_INT64)   m *= sizeof(PetscInt64);
44541f502e3SPatrick Sanan   else if (wtype == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char);
446e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
447e5c89e4eSSatish Balay 
448f253e43cSLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap((void*)ptmp,wtype,n);CHKERRQ(ierr);}
449e5c89e4eSSatish Balay 
450e5c89e4eSSatish Balay   while (m) {
451e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
452e5c89e4eSSatish Balay     err   = write(fd,pp,wsize);
453e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
45404102261SBarry Smith     if (err != wsize) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_FILE_WRITE,"Error writing to file total size %d err %d wsize %d",(int)n,(int)err,(int)wsize);
455e5c89e4eSSatish Balay     m  -= wsize;
456e5c89e4eSSatish Balay     pp += wsize;
457e5c89e4eSSatish Balay   }
458e5c89e4eSSatish Balay 
459f253e43cSLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap((void*)ptmp,wtype,n);CHKERRQ(ierr);}
460f253e43cSLisandro Dalcin 
46105acbc63SBarry Smith   if (type == PETSC_FUNCTION) {
46205acbc63SBarry Smith     free(fname);
46305acbc63SBarry Smith   }
4640da86b62SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
465501ae109SJose E. Roman   if ((type == PETSC_SCALAR || type == PETSC_REAL || type == PETSC_COMPLEX) && writedouble) {
4660da86b62SBarry Smith     ierr = PetscFree(ppp);CHKERRQ(ierr);
4670da86b62SBarry Smith   }
4680da86b62SBarry Smith #endif
469e5c89e4eSSatish Balay   PetscFunctionReturn(0);
470e5c89e4eSSatish Balay }
471e5c89e4eSSatish Balay 
472e5c89e4eSSatish Balay /*@C
473e5c89e4eSSatish Balay    PetscBinaryOpen - Opens a PETSc binary file.
474e5c89e4eSSatish Balay 
475e5c89e4eSSatish Balay    Not Collective
476e5c89e4eSSatish Balay 
477e5c89e4eSSatish Balay    Input Parameters:
478e5c89e4eSSatish Balay +  name - filename
479cc843e7aSLisandro Dalcin -  mode - open mode of binary file, one of FILE_MODE_READ, FILE_MODE_WRITE, FILE_MODE_APPEND
480e5c89e4eSSatish Balay 
481e5c89e4eSSatish Balay    Output Parameter:
482e5c89e4eSSatish Balay .  fd - the file
483e5c89e4eSSatish Balay 
484e5c89e4eSSatish Balay    Level: advanced
485e5c89e4eSSatish Balay 
486e5c89e4eSSatish Balay 
48795452b02SPatrick Sanan    Notes:
48895452b02SPatrick Sanan     Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in
489e5c89e4eSSatish Balay    big-endian format. This means the file can be accessed using PetscBinaryOpen() and
490e5c89e4eSSatish Balay    PetscBinaryRead() and PetscBinaryWrite() on any machine.
491e5c89e4eSSatish Balay 
4924ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(),
4934ebed01fSBarry Smith           PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
49445c64e65SBarry Smith 
495e5c89e4eSSatish Balay @*/
4967087cfbeSBarry Smith PetscErrorCode  PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd)
497e5c89e4eSSatish Balay {
498e5c89e4eSSatish Balay   PetscFunctionBegin;
499cc843e7aSLisandro Dalcin   switch (mode) {
500cc843e7aSLisandro Dalcin   case FILE_MODE_READ:   *fd = open(name,O_BINARY|O_RDONLY,0); break;
501cc843e7aSLisandro Dalcin   case FILE_MODE_WRITE:  *fd = open(name,O_BINARY|O_WRONLY|O_CREAT|O_TRUNC,0666); break;
502cc843e7aSLisandro Dalcin   case FILE_MODE_APPEND: *fd = open(name,O_BINARY|O_WRONLY|O_APPEND,0); break;
503cc843e7aSLisandro Dalcin   default: SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported file mode %s",PetscFileModes[mode]);
504e5c89e4eSSatish Balay   }
505cc843e7aSLisandro Dalcin   if (*fd == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for %s: %s",name,PetscFileModes[mode]);
506e5c89e4eSSatish Balay   PetscFunctionReturn(0);
507e5c89e4eSSatish Balay }
508e5c89e4eSSatish Balay 
509e30d2299SSatish Balay /*@
510e5c89e4eSSatish Balay    PetscBinaryClose - Closes a PETSc binary file.
511e5c89e4eSSatish Balay 
512e5c89e4eSSatish Balay    Not Collective
513e5c89e4eSSatish Balay 
514e5c89e4eSSatish Balay    Output Parameter:
515e5c89e4eSSatish Balay .  fd - the file
516e5c89e4eSSatish Balay 
517e5c89e4eSSatish Balay    Level: advanced
518e5c89e4eSSatish Balay 
5194ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
5204ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
521e5c89e4eSSatish Balay @*/
5227087cfbeSBarry Smith PetscErrorCode  PetscBinaryClose(int fd)
523e5c89e4eSSatish Balay {
524e5c89e4eSSatish Balay   PetscFunctionBegin;
525cc843e7aSLisandro Dalcin   if (close(fd)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"close() failed on file descriptor");
526e5c89e4eSSatish Balay   PetscFunctionReturn(0);
527e5c89e4eSSatish Balay }
528e5c89e4eSSatish Balay 
529e5c89e4eSSatish Balay 
530e8976759SBarry Smith /*@C
531e5c89e4eSSatish Balay    PetscBinarySeek - Moves the file pointer on a PETSc binary file.
532e5c89e4eSSatish Balay 
533e5c89e4eSSatish Balay    Not Collective
534e5c89e4eSSatish Balay 
535e5c89e4eSSatish Balay    Input Parameters:
536e5c89e4eSSatish Balay +  fd - the file
537ff553b35SMatthew Knepley .  off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
538e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
539ff553b35SMatthew Knepley -  whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file
540ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_CUR then off is an offset from the current location
541ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_END then off is an offset from the end of file
542e5c89e4eSSatish Balay 
543e5c89e4eSSatish Balay    Output Parameter:
544e5c89e4eSSatish Balay .   offset - new offset in file
545e5c89e4eSSatish Balay 
546e5c89e4eSSatish Balay    Level: developer
547e5c89e4eSSatish Balay 
548e5c89e4eSSatish Balay    Notes:
549e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
550e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
551e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
552e5c89e4eSSatish Balay    to determine the offset or location.
553e5c89e4eSSatish Balay 
554e5c89e4eSSatish Balay 
5554ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
5564ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
557e5c89e4eSSatish Balay @*/
5587087cfbeSBarry Smith PetscErrorCode  PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
559e5c89e4eSSatish Balay {
560e5c89e4eSSatish Balay   int iwhence = 0;
561e5c89e4eSSatish Balay 
562e5c89e4eSSatish Balay   PetscFunctionBegin;
563a297a907SKarl Rupp   if (whence == PETSC_BINARY_SEEK_SET) iwhence = SEEK_SET;
564a297a907SKarl Rupp   else if (whence == PETSC_BINARY_SEEK_CUR) iwhence = SEEK_CUR;
565a297a907SKarl Rupp   else if (whence == PETSC_BINARY_SEEK_END) iwhence = SEEK_END;
566a297a907SKarl Rupp   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location");
567e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK)
568e5c89e4eSSatish Balay   *offset = lseek(fd,off,iwhence);
569e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK)
570e5c89e4eSSatish Balay   *offset = _lseek(fd,(long)off,iwhence);
571e5c89e4eSSatish Balay #else
572e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file");
573e5c89e4eSSatish Balay #endif
574e5c89e4eSSatish Balay   PetscFunctionReturn(0);
575e5c89e4eSSatish Balay }
576e5c89e4eSSatish Balay 
577e5c89e4eSSatish Balay /*@C
5781d280d73SBarry Smith    PetscBinarySynchronizedRead - Reads from a binary file.
579e5c89e4eSSatish Balay 
580d083f849SBarry Smith    Collective
581e5c89e4eSSatish Balay 
582e5c89e4eSSatish Balay    Input Parameters:
583e5c89e4eSSatish Balay +  comm - the MPI communicator
5849860990eSLisandro Dalcin .  fd - the file descriptor
5859860990eSLisandro Dalcin .  num  - the maximum number of items to read
5869860990eSLisandro Dalcin -  type - the type of items to read (PETSC_INT, PETSC_REAL, PETSC_SCALAR, etc.)
587e5c89e4eSSatish Balay 
588e5c89e4eSSatish Balay    Output Parameters:
5899860990eSLisandro Dalcin +  data - the buffer
5909860990eSLisandro Dalcin -  count - the number of items read, optional
591e5c89e4eSSatish Balay 
592e5c89e4eSSatish Balay    Level: developer
593e5c89e4eSSatish Balay 
594e5c89e4eSSatish Balay    Notes:
595e5c89e4eSSatish Balay    Does a PetscBinaryRead() followed by an MPI_Bcast()
596e5c89e4eSSatish Balay 
5979860990eSLisandro Dalcin    If count is not provided and the number of items read is less than
5989860990eSLisandro Dalcin    the maximum number of items to read, then this routine errors.
5999860990eSLisandro Dalcin 
6001d280d73SBarry Smith    PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
601e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
602e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
603e5c89e4eSSatish Balay    binary file may be read on any machine.
604e5c89e4eSSatish Balay 
605e5c89e4eSSatish Balay 
6064ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
6074ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
608e5c89e4eSSatish Balay @*/
6099860990eSLisandro Dalcin PetscErrorCode  PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *data,PetscInt num,PetscInt *count,PetscDataType type)
610e5c89e4eSSatish Balay {
6119860990eSLisandro Dalcin   PetscErrorCode ierr;
612*8b8bf82cSBarry Smith   PetscMPIInt    rank,size;
613e5c89e4eSSatish Balay   MPI_Datatype   mtype;
6149860990eSLisandro Dalcin   PetscInt       ibuf[2] = {0, 0};
61505acbc63SBarry Smith   char           *fname = NULL;
6169860990eSLisandro Dalcin   void           *fptr = NULL;
617e5c89e4eSSatish Balay 
618e5c89e4eSSatish Balay   PetscFunctionBegin;
6192d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
6209860990eSLisandro Dalcin     num   = 64;
6212d53ad75SBarry Smith     type  = PETSC_CHAR;
6229860990eSLisandro Dalcin     fname = (char*)malloc(num*sizeof(char));
6239860990eSLisandro Dalcin     fptr  = data;
6249860990eSLisandro Dalcin     data  = (void*)fname;
62505acbc63SBarry Smith     if (!fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MEM,"Cannot allocate space for function name");
6262d53ad75SBarry Smith   }
6272d53ad75SBarry Smith 
628e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
629*8b8bf82cSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
630e5c89e4eSSatish Balay   if (!rank) {
6319860990eSLisandro Dalcin     ibuf[0] = PetscBinaryRead(fd,data,num,count?&ibuf[1]:NULL,type);
632e5c89e4eSSatish Balay   }
6339860990eSLisandro Dalcin   ierr = MPI_Bcast(ibuf,2,MPIU_INT,0,comm);CHKERRQ(ierr);
6349860990eSLisandro Dalcin   ierr = (PetscErrorCode)ibuf[0];CHKERRQ(ierr);
635*8b8bf82cSBarry Smith 
636*8b8bf82cSBarry 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 */
637*8b8bf82cSBarry Smith   if (size > 1) {
638e5c89e4eSSatish Balay     ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
6399860990eSLisandro Dalcin     ierr = MPI_Bcast(data,count?ibuf[1]:num,mtype,0,comm);CHKERRQ(ierr);
640*8b8bf82cSBarry Smith   }
6419860990eSLisandro Dalcin   if (count) *count = ibuf[1];
6422d53ad75SBarry Smith 
643e366c363SBarry Smith   if (type == PETSC_FUNCTION) {
6442d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
6459860990eSLisandro Dalcin     ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,fname,(void**)fptr);CHKERRQ(ierr);
6462d53ad75SBarry Smith #else
6479860990eSLisandro Dalcin     *(void**)fptr = NULL;
6482d53ad75SBarry Smith #endif
649e366c363SBarry Smith     free(fname);
6502d53ad75SBarry Smith   }
651e5c89e4eSSatish Balay   PetscFunctionReturn(0);
652e5c89e4eSSatish Balay }
653e5c89e4eSSatish Balay 
654e5c89e4eSSatish Balay /*@C
6551d280d73SBarry Smith    PetscBinarySynchronizedWrite - writes to a binary file.
656e5c89e4eSSatish Balay 
657d083f849SBarry Smith    Collective
658e5c89e4eSSatish Balay 
659e5c89e4eSSatish Balay    Input Parameters:
660e5c89e4eSSatish Balay +  comm - the MPI communicator
661e5c89e4eSSatish Balay .  fd - the file
662e5c89e4eSSatish Balay .  n  - the number of items to write
663e5c89e4eSSatish Balay .  p - the buffer
664e5c89e4eSSatish Balay -  type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
665e5c89e4eSSatish Balay 
666e5c89e4eSSatish Balay    Level: developer
667e5c89e4eSSatish Balay 
668e5c89e4eSSatish Balay    Notes:
669e5c89e4eSSatish Balay    Process 0 does a PetscBinaryWrite()
670e5c89e4eSSatish Balay 
6711d280d73SBarry Smith    PetscBinarySynchronizedWrite() uses byte swapping to work on all machines.
672e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
673e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
674e5c89e4eSSatish Balay    binary file may be read on any machine.
675e5c89e4eSSatish Balay 
67695452b02SPatrick Sanan    Notes:
67795452b02SPatrick Sanan     because byte-swapping may be done on the values in data it cannot be declared const
678300a7f5bSBarry Smith 
6791d280d73SBarry Smith    WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0,
6801d280d73SBarry Smith    while PetscSynchronizedFPrintf() has all processes print their strings in order.
6811d280d73SBarry Smith 
682e5c89e4eSSatish Balay 
6834ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(),
6844ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
685e5c89e4eSSatish Balay @*/
686f253e43cSLisandro Dalcin PetscErrorCode  PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,const void *p,PetscInt n,PetscDataType type)
687e5c89e4eSSatish Balay {
688e5c89e4eSSatish Balay   PetscErrorCode ierr;
689e5c89e4eSSatish Balay   PetscMPIInt    rank;
690e5c89e4eSSatish Balay 
691e5c89e4eSSatish Balay   PetscFunctionBegin;
692e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
693e5c89e4eSSatish Balay   if (!rank) {
694f253e43cSLisandro Dalcin     ierr = PetscBinaryWrite(fd,p,n,type);CHKERRQ(ierr);
695e5c89e4eSSatish Balay   }
696e5c89e4eSSatish Balay   PetscFunctionReturn(0);
697e5c89e4eSSatish Balay }
698e5c89e4eSSatish Balay 
699e5c89e4eSSatish Balay /*@C
7001d280d73SBarry Smith    PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file.
701e5c89e4eSSatish Balay 
702e5c89e4eSSatish Balay 
703e5c89e4eSSatish Balay    Input Parameters:
704e5c89e4eSSatish Balay +  fd - the file
705e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
706e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
707e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
708e5c89e4eSSatish Balay -  off    - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
709e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
710e5c89e4eSSatish Balay 
711e5c89e4eSSatish Balay    Output Parameter:
712e5c89e4eSSatish Balay .   offset - new offset in file
713e5c89e4eSSatish Balay 
714e5c89e4eSSatish Balay    Level: developer
715e5c89e4eSSatish Balay 
716e5c89e4eSSatish Balay    Notes:
717e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
718e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
719e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
720e5c89e4eSSatish Balay    to determine the offset or location.
721e5c89e4eSSatish Balay 
722e5c89e4eSSatish Balay 
7234ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
7244ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
725e5c89e4eSSatish Balay @*/
7267087cfbeSBarry Smith PetscErrorCode  PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
727e5c89e4eSSatish Balay {
728e5c89e4eSSatish Balay   PetscErrorCode ierr;
729e5c89e4eSSatish Balay   PetscMPIInt    rank;
730e5c89e4eSSatish Balay 
731e5c89e4eSSatish Balay   PetscFunctionBegin;
732e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
733e5c89e4eSSatish Balay   if (!rank) {
734e5c89e4eSSatish Balay     ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr);
735e5c89e4eSSatish Balay   }
736e5c89e4eSSatish Balay   PetscFunctionReturn(0);
737e5c89e4eSSatish Balay }
738e5c89e4eSSatish Balay 
7390fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO)
740e39fd77fSBarry Smith 
741951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32)
742e39fd77fSBarry Smith /*
743e39fd77fSBarry Smith       MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions.
744e39fd77fSBarry Smith     These are set into MPI in PetscInitialize() via MPI_Register_datarep()
745e39fd77fSBarry Smith 
746e39fd77fSBarry Smith     Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode)
747e39fd77fSBarry Smith 
748951e3c8eSBarry Smith     The next three routines are not used because MPICH does not support their use
749e39fd77fSBarry Smith 
750e39fd77fSBarry Smith */
7518cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state)
752e39fd77fSBarry Smith {
753e39fd77fSBarry Smith   MPI_Aint    ub;
754e39fd77fSBarry Smith   PetscMPIInt ierr;
755e39fd77fSBarry Smith 
756e39fd77fSBarry Smith   ierr = MPI_Type_get_extent(datatype,&ub,file_extent);
757e39fd77fSBarry Smith   return ierr;
758e39fd77fSBarry Smith }
759e39fd77fSBarry Smith 
7608cc058d9SJed Brown PETSC_EXTERN PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
761e39fd77fSBarry Smith {
762e39fd77fSBarry Smith   PetscDataType pdtype;
763e39fd77fSBarry Smith   PetscMPIInt   ierr;
764e39fd77fSBarry Smith   size_t        dsize;
765e39fd77fSBarry Smith 
766e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
767e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
768e39fd77fSBarry Smith 
769e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
770e39fd77fSBarry Smith   userbuf = ((char*)userbuf) + dsize*position;
771e39fd77fSBarry Smith 
772e39fd77fSBarry Smith   ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr);
77330815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr);}
774e39fd77fSBarry Smith   return ierr;
775e39fd77fSBarry Smith }
776e39fd77fSBarry Smith 
777e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
778e39fd77fSBarry Smith {
779e39fd77fSBarry Smith   PetscDataType pdtype;
780e39fd77fSBarry Smith   PetscMPIInt   ierr;
781e39fd77fSBarry Smith   size_t        dsize;
782e39fd77fSBarry Smith 
783e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
784e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
785e39fd77fSBarry Smith 
786e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
787e39fd77fSBarry Smith   userbuf = ((char*)userbuf) + dsize*position;
788e39fd77fSBarry Smith 
789e39fd77fSBarry Smith   ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr);
79030815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr);}
791e39fd77fSBarry Smith   return ierr;
792e39fd77fSBarry Smith }
793951e3c8eSBarry Smith #endif
794e39fd77fSBarry Smith 
795e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
796e39fd77fSBarry Smith {
797e39fd77fSBarry Smith   PetscDataType  pdtype;
79830815ce0SLisandro Dalcin   PetscErrorCode ierr;
79930815ce0SLisandro Dalcin 
800e39fd77fSBarry Smith 
801e39fd77fSBarry Smith   PetscFunctionBegin;
802e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
80330815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
804e39fd77fSBarry Smith   ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
80530815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
806a6796414SBarry Smith   PetscFunctionReturn(0);
807e39fd77fSBarry Smith }
808e39fd77fSBarry Smith 
809e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
810e39fd77fSBarry Smith {
811e39fd77fSBarry Smith   PetscDataType  pdtype;
81230815ce0SLisandro Dalcin   PetscErrorCode ierr;
813e39fd77fSBarry Smith 
814e39fd77fSBarry Smith   PetscFunctionBegin;
815e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
816e39fd77fSBarry Smith   ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
81730815ce0SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
818a6796414SBarry Smith   PetscFunctionReturn(0);
819e39fd77fSBarry Smith }
82030815ce0SLisandro Dalcin 
821c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
822c4e82887SLisandro Dalcin {
823c4e82887SLisandro Dalcin   PetscDataType  pdtype;
824c4e82887SLisandro Dalcin   PetscErrorCode ierr;
825c4e82887SLisandro Dalcin 
826c4e82887SLisandro Dalcin 
827c4e82887SLisandro Dalcin   PetscFunctionBegin;
828c4e82887SLisandro Dalcin   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
829c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
830c4e82887SLisandro Dalcin   ierr = MPI_File_write_at(fd,off,data,cnt,dtype,status);CHKERRQ(ierr);
831c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
832c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
833c4e82887SLisandro Dalcin }
834c4e82887SLisandro Dalcin 
835c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
836c4e82887SLisandro Dalcin {
837c4e82887SLisandro Dalcin   PetscDataType  pdtype;
838c4e82887SLisandro Dalcin   PetscErrorCode ierr;
839c4e82887SLisandro Dalcin 
840c4e82887SLisandro Dalcin   PetscFunctionBegin;
841c4e82887SLisandro Dalcin   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
842c4e82887SLisandro Dalcin   ierr = MPI_File_read_at(fd,off,data,cnt,dtype,status);CHKERRQ(ierr);
843c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
844c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
845c4e82887SLisandro Dalcin }
846c4e82887SLisandro Dalcin 
847c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_write_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
848c4e82887SLisandro Dalcin {
849c4e82887SLisandro Dalcin   PetscDataType  pdtype;
850c4e82887SLisandro Dalcin   PetscErrorCode ierr;
851c4e82887SLisandro Dalcin 
852c4e82887SLisandro Dalcin 
853c4e82887SLisandro Dalcin   PetscFunctionBegin;
854c4e82887SLisandro Dalcin   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
855c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
856c4e82887SLisandro Dalcin   ierr = MPI_File_write_at_all(fd,off,data,cnt,dtype,status);CHKERRQ(ierr);
857c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
858c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
859c4e82887SLisandro Dalcin }
860c4e82887SLisandro Dalcin 
861c4e82887SLisandro Dalcin PetscErrorCode MPIU_File_read_at_all(MPI_File fd,MPI_Offset off,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
862c4e82887SLisandro Dalcin {
863c4e82887SLisandro Dalcin   PetscDataType  pdtype;
864c4e82887SLisandro Dalcin   PetscErrorCode ierr;
865c4e82887SLisandro Dalcin 
866c4e82887SLisandro Dalcin   PetscFunctionBegin;
867c4e82887SLisandro Dalcin   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
868c4e82887SLisandro Dalcin   ierr = MPI_File_read_at_all(fd,off,data,cnt,dtype,status);CHKERRQ(ierr);
869c4e82887SLisandro Dalcin   if (!PetscBinaryBigEndian()) {ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);}
870c4e82887SLisandro Dalcin   PetscFunctionReturn(0);
871c4e82887SLisandro Dalcin }
872c4e82887SLisandro Dalcin 
873951e3c8eSBarry Smith #endif
874