xref: /petsc/src/sys/fileio/sysio.c (revision 2d53ad75667bd9af5ca32c6f75ade56f24a4fbe1)
1e5c89e4eSSatish Balay 
2e5c89e4eSSatish Balay /*
3e5c89e4eSSatish Balay    This file contains simple binary read/write routines.
4e5c89e4eSSatish Balay  */
5e5c89e4eSSatish Balay 
6c6db04a5SJed Brown #include <petscsys.h>
7e5c89e4eSSatish Balay #include <errno.h>
8e5c89e4eSSatish Balay #include <fcntl.h>
9e5c89e4eSSatish Balay #if defined(PETSC_HAVE_UNISTD_H)
10e5c89e4eSSatish Balay #include <unistd.h>
11e5c89e4eSSatish Balay #endif
12e5c89e4eSSatish Balay #if defined (PETSC_HAVE_IO_H)
13e5c89e4eSSatish Balay #include <io.h>
14e5c89e4eSSatish Balay #endif
15c6db04a5SJed Brown #include <petscbt.h>
16e5c89e4eSSatish Balay 
176de02169SBarry Smith /* --------------------------------------------------------- */
18e5c89e4eSSatish Balay #undef __FUNCT__
196de02169SBarry Smith #define __FUNCT__ "PetscByteSwapEnum"
20e5c89e4eSSatish Balay /*
216de02169SBarry Smith   PetscByteSwapEnum - Swap bytes in a  PETSc Enum
22e5c89e4eSSatish Balay 
23e5c89e4eSSatish Balay */
247087cfbeSBarry Smith PetscErrorCode  PetscByteSwapEnum(PetscEnum *buff,PetscInt n)
25e5c89e4eSSatish Balay {
266de02169SBarry Smith   PetscInt   i,j;
270b20345dSBarry Smith   PetscEnum   tmp = ENUM_DUMMY;
28e0890e22SSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
29e5c89e4eSSatish Balay 
30e5c89e4eSSatish Balay   PetscFunctionBegin;
31e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
32e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
330b20345dSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) {
346de02169SBarry Smith       ptr2[i] = ptr1[sizeof(PetscEnum)-1-i];
35e5c89e4eSSatish Balay     }
36e0890e22SSatish Balay     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) {
37e0890e22SSatish Balay       ptr1[i] = ptr2[i];
38e0890e22SSatish Balay     }
39e5c89e4eSSatish Balay   }
40e5c89e4eSSatish Balay   PetscFunctionReturn(0);
41e5c89e4eSSatish Balay }
426de02169SBarry Smith 
436de02169SBarry Smith #undef __FUNCT__
44acfcf0e5SJed Brown #define __FUNCT__ "PetscByteSwapBool"
456de02169SBarry Smith /*
46acfcf0e5SJed Brown   PetscByteSwapBool - Swap bytes in a  PETSc Bool
476de02169SBarry Smith 
486de02169SBarry Smith */
497087cfbeSBarry Smith PetscErrorCode  PetscByteSwapBool(PetscBool *buff,PetscInt n)
506de02169SBarry Smith {
516de02169SBarry Smith   PetscInt    i,j;
52ace3abfcSBarry Smith   PetscBool   tmp = PETSC_FALSE;
53e0890e22SSatish Balay   char        *ptr1,*ptr2 = (char*)&tmp;
546de02169SBarry Smith 
556de02169SBarry Smith   PetscFunctionBegin;
566de02169SBarry Smith   for (j=0; j<n; j++) {
576de02169SBarry Smith     ptr1 = (char*)(buff + j);
58ace3abfcSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscBool); i++) {
59ace3abfcSBarry Smith       ptr2[i] = ptr1[sizeof(PetscBool)-1-i];
606de02169SBarry Smith     }
61ace3abfcSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscBool); i++) {
62e0890e22SSatish Balay       ptr1[i] = ptr2[i];
63e0890e22SSatish Balay     }
646de02169SBarry Smith   }
656de02169SBarry Smith   PetscFunctionReturn(0);
666de02169SBarry Smith }
676de02169SBarry Smith 
68e5c89e4eSSatish Balay #undef __FUNCT__
69bd1d2e58SBarry Smith #define __FUNCT__ "PetscByteSwapInt"
70bd1d2e58SBarry Smith /*
716de02169SBarry Smith   PetscByteSwapInt - Swap bytes in a  PETSc integer (which may be 32 or 64 bits)
72bd1d2e58SBarry Smith 
73bd1d2e58SBarry Smith */
747087cfbeSBarry Smith PetscErrorCode  PetscByteSwapInt(PetscInt *buff,PetscInt n)
75bd1d2e58SBarry Smith {
76bd1d2e58SBarry Smith   PetscInt  i,j,tmp = 0;
77e0890e22SSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
78bd1d2e58SBarry Smith 
79bd1d2e58SBarry Smith   PetscFunctionBegin;
80bd1d2e58SBarry Smith   for (j=0; j<n; j++) {
81bd1d2e58SBarry Smith     ptr1 = (char*)(buff + j);
820b20345dSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) {
83bd1d2e58SBarry Smith       ptr2[i] = ptr1[sizeof(PetscInt)-1-i];
84bd1d2e58SBarry Smith     }
85e0890e22SSatish Balay     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) {
86e0890e22SSatish Balay       ptr1[i] = ptr2[i];
87e0890e22SSatish Balay     }
88bd1d2e58SBarry Smith   }
89bd1d2e58SBarry Smith   PetscFunctionReturn(0);
90bd1d2e58SBarry Smith }
91bd1d2e58SBarry Smith /* --------------------------------------------------------- */
92bd1d2e58SBarry Smith #undef __FUNCT__
93e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort"
94e5c89e4eSSatish Balay /*
95e5c89e4eSSatish Balay   PetscByteSwapShort - Swap bytes in a short
96e5c89e4eSSatish Balay */
977087cfbeSBarry Smith PetscErrorCode  PetscByteSwapShort(short *buff,PetscInt n)
98e5c89e4eSSatish Balay {
99e5c89e4eSSatish Balay   PetscInt   i,j;
100e5c89e4eSSatish Balay   short      tmp;
101e5c89e4eSSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
102e5c89e4eSSatish Balay 
103e5c89e4eSSatish Balay   PetscFunctionBegin;
104e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
105e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
1060b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(short); i++) {
107f348e1baSSatish Balay       ptr2[i] = ptr1[sizeof(short)-1-i];
108e5c89e4eSSatish Balay     }
109e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(short); i++) {
110e0890e22SSatish Balay       ptr1[i] = ptr2[i];
111e0890e22SSatish Balay     }
112e5c89e4eSSatish Balay   }
113e5c89e4eSSatish Balay   PetscFunctionReturn(0);
114e5c89e4eSSatish Balay }
115e5c89e4eSSatish Balay /* --------------------------------------------------------- */
116e5c89e4eSSatish Balay #undef __FUNCT__
117e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar"
118e5c89e4eSSatish Balay /*
119e5c89e4eSSatish Balay   PetscByteSwapScalar - Swap bytes in a double
120e5c89e4eSSatish Balay   Complex is dealt with as if array of double twice as long.
121e5c89e4eSSatish Balay */
1227087cfbeSBarry Smith PetscErrorCode  PetscByteSwapScalar(PetscScalar *buff,PetscInt n)
123e5c89e4eSSatish Balay {
124e5c89e4eSSatish Balay   PetscInt  i,j;
125e5c89e4eSSatish Balay   PetscReal tmp,*buff1 = (PetscReal*)buff;
126e5c89e4eSSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
127e5c89e4eSSatish Balay 
128e5c89e4eSSatish Balay   PetscFunctionBegin;
129e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX)
130e5c89e4eSSatish Balay   n *= 2;
131e5c89e4eSSatish Balay #endif
132e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
133e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
1340b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) {
135e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(PetscReal)-1-i];
136e5c89e4eSSatish Balay     }
137e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) {
138e0890e22SSatish Balay       ptr1[i] = ptr2[i];
139e0890e22SSatish Balay     }
140e5c89e4eSSatish Balay   }
141e5c89e4eSSatish Balay   PetscFunctionReturn(0);
142e5c89e4eSSatish Balay }
143e5c89e4eSSatish Balay /* --------------------------------------------------------- */
144e5c89e4eSSatish Balay #undef __FUNCT__
145e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapDouble"
146e5c89e4eSSatish Balay /*
147e5c89e4eSSatish Balay   PetscByteSwapDouble - Swap bytes in a double
148e5c89e4eSSatish Balay */
1497087cfbeSBarry Smith PetscErrorCode  PetscByteSwapDouble(double *buff,PetscInt n)
150e5c89e4eSSatish Balay {
151e5c89e4eSSatish Balay   PetscInt i,j;
152e5c89e4eSSatish Balay   double   tmp,*buff1 = (double*)buff;
153e5c89e4eSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
154e5c89e4eSSatish Balay 
155e5c89e4eSSatish Balay   PetscFunctionBegin;
156e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
157e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
1580b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(double); i++) {
159e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(double)-1-i];
160e5c89e4eSSatish Balay     }
161e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(double); i++) {
162e0890e22SSatish Balay       ptr1[i] = ptr2[i];
163e0890e22SSatish Balay     }
164e5c89e4eSSatish Balay   }
165e5c89e4eSSatish Balay   PetscFunctionReturn(0);
166e5c89e4eSSatish Balay }
167e39fd77fSBarry Smith 
168e39fd77fSBarry Smith #undef __FUNCT__
169e95bf02fSSatish Balay #define __FUNCT__ "PetscByteSwapFloat"
170e95bf02fSSatish Balay /*
171e95bf02fSSatish Balay   PetscByteSwapFloat - Swap bytes in a float
172e95bf02fSSatish Balay */
173e95bf02fSSatish Balay PetscErrorCode PetscByteSwapFloat(float *buff,PetscInt n)
174e95bf02fSSatish Balay {
175e95bf02fSSatish Balay   PetscInt i,j;
176e95bf02fSSatish Balay   float   tmp,*buff1 = (float*)buff;
177e95bf02fSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
178e95bf02fSSatish Balay 
179e95bf02fSSatish Balay   PetscFunctionBegin;
180e95bf02fSSatish Balay   for (j=0; j<n; j++) {
181e95bf02fSSatish Balay     ptr1 = (char*)(buff1 + j);
182e95bf02fSSatish Balay     for (i=0; i<(PetscInt) sizeof(float); i++) {
183e95bf02fSSatish Balay       ptr2[i] = ptr1[sizeof(float)-1-i];
184e95bf02fSSatish Balay     }
185e95bf02fSSatish Balay     for (i=0; i<(PetscInt) sizeof(float); i++) {
186e95bf02fSSatish Balay       ptr1[i] = ptr2[i];
187e95bf02fSSatish Balay     }
188e95bf02fSSatish Balay   }
189e95bf02fSSatish Balay   PetscFunctionReturn(0);
190e95bf02fSSatish Balay }
191e95bf02fSSatish Balay 
192e95bf02fSSatish Balay #undef __FUNCT__
193e39fd77fSBarry Smith #define __FUNCT__ "PetscByteSwap"
194e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count)
195e39fd77fSBarry Smith {
196e39fd77fSBarry Smith   PetscErrorCode ierr;
197e39fd77fSBarry Smith 
198e39fd77fSBarry Smith   PetscFunctionBegin;
199e39fd77fSBarry Smith   if      (pdtype == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);}
200e39fd77fSBarry Smith   else if (pdtype == PETSC_ENUM)   {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);}
201acfcf0e5SJed Brown   else if (pdtype == PETSC_BOOL)   {ierr = PetscByteSwapBool((PetscBool*)data,count);CHKERRQ(ierr);}
202e39fd77fSBarry Smith   else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);}
203e39fd77fSBarry Smith   else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);}
204e95bf02fSSatish Balay   else if (pdtype == PETSC_FLOAT) {ierr = PetscByteSwapFloat((float*)data,count);CHKERRQ(ierr);}
205e39fd77fSBarry Smith   else if (pdtype == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);}
206e39fd77fSBarry Smith   PetscFunctionReturn(0);
207e39fd77fSBarry Smith }
208e39fd77fSBarry Smith 
209e5c89e4eSSatish Balay /* --------------------------------------------------------- */
210e5c89e4eSSatish Balay #undef __FUNCT__
211e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead"
212e30d2299SSatish Balay /*@
213e5c89e4eSSatish Balay    PetscBinaryRead - Reads from a binary file.
214e5c89e4eSSatish Balay 
215e5c89e4eSSatish Balay    Not Collective
216e5c89e4eSSatish Balay 
217e5c89e4eSSatish Balay    Input Parameters:
218e5c89e4eSSatish Balay +  fd - the file
219e5c89e4eSSatish Balay .  n  - the number of items to read
220e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
221e5c89e4eSSatish Balay 
222e5c89e4eSSatish Balay    Output Parameters:
223e5c89e4eSSatish Balay .  p - the buffer
224e5c89e4eSSatish Balay 
225e5c89e4eSSatish Balay 
226e5c89e4eSSatish Balay 
227e5c89e4eSSatish Balay    Level: developer
228e5c89e4eSSatish Balay 
229e5c89e4eSSatish Balay    Notes:
230e5c89e4eSSatish Balay    PetscBinaryRead() uses byte swapping to work on all machines; the files
231e5c89e4eSSatish Balay    are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers
232e5c89e4eSSatish Balay    are converted to the small-endian format when they are read in from the file.
233e2e64c6bSBarry Smith    When PETSc is ./configure with --with-64bit-indices the integers are written to the
23454f21887SBarry Smith    file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices
23554f21887SBarry Smith    is used.
236e5c89e4eSSatish Balay 
237e5c89e4eSSatish Balay    Concepts: files^reading binary
238e5c89e4eSSatish Balay    Concepts: binary files^reading
239e5c89e4eSSatish Balay 
2404ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
2414ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
242e5c89e4eSSatish Balay @*/
2437087cfbeSBarry Smith PetscErrorCode  PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type)
244e5c89e4eSSatish Balay {
245e5c89e4eSSatish Balay   int               wsize,err;
246e5c89e4eSSatish Balay   size_t            m = (size_t) n,maxblock = 65536;
247e5c89e4eSSatish Balay   char              *pp = (char*)p;
2487a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
2497a881295SBarry Smith   double            *ppp;
2507a881295SBarry Smith #endif
2517a881295SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN) || defined(PETSC_USE_REAL___FLOAT128)
2526de02169SBarry Smith   PetscErrorCode    ierr;
2537a881295SBarry Smith #endif
2547a881295SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
255e5c89e4eSSatish Balay   void              *ptmp = p;
256e5c89e4eSSatish Balay #endif
257*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
258*2d53ad75SBarry Smith   char              fname[64];
259*2d53ad75SBarry Smith #endif
260*2d53ad75SBarry Smith   PetscBool         functionload = PETSC_FALSE;
261e5c89e4eSSatish Balay 
262e5c89e4eSSatish Balay   PetscFunctionBegin;
263*2d53ad75SBarry Smith   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n);
264e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
265e5c89e4eSSatish Balay 
266*2d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
267*2d53ad75SBarry Smith     functionload = PETSC_TRUE;
268*2d53ad75SBarry Smith     m    = 64;
269*2d53ad75SBarry Smith     type = PETSC_CHAR;
270*2d53ad75SBarry Smith     pp   = (char*)fname;
271*2d53ad75SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
272*2d53ad75SBarry Smith     ptmp = (void*)fname;
273*2d53ad75SBarry Smith #endif
274*2d53ad75SBarry Smith   }
275*2d53ad75SBarry Smith 
2766de02169SBarry Smith   if (type == PETSC_INT)          m *= sizeof(PetscInt);
277e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
278e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
279e95bf02fSSatish Balay   else if (type == PETSC_FLOAT)   m *= sizeof(float);
280e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
281e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
282e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
283ace3abfcSBarry Smith   else if (type == PETSC_BOOL)   m *= sizeof(PetscBool);
2849f7b6320SBarry Smith   else if (type == PETSC_BIT_LOGICAL) m  = PetscBTLength(m)*sizeof(char);
285e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
286e5c89e4eSSatish Balay 
2877a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
2887a881295SBarry Smith   /* If using __float128 precision we still read in doubles from file */
2897a881295SBarry Smith   if (type == PETSC_SCALAR) {
2907a881295SBarry Smith     m    = m/2;
2917a881295SBarry Smith     ierr = PetscMalloc(n*sizeof(double),&ppp);CHKERRQ(ierr);
2927a881295SBarry Smith     pp   = (char*)ppp;
2937a881295SBarry Smith   }
2947a881295SBarry Smith #endif
2957a881295SBarry Smith 
296e5c89e4eSSatish Balay   while (m) {
297e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
298e5c89e4eSSatish Balay     err = read(fd,pp,wsize);
299e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
300e32f2f54SBarry Smith     if (!err && wsize > 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Read past end of file");
301e32f2f54SBarry Smith     if (err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno);
302e5c89e4eSSatish Balay     m  -= err;
303e5c89e4eSSatish Balay     pp += err;
304e5c89e4eSSatish Balay   }
3057a881295SBarry Smith 
3067a881295SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
3077a881295SBarry Smith   if (type == PETSC_SCALAR) {
3087a881295SBarry Smith     PetscScalar *pv = (PetscScalar*) p;
3097a881295SBarry Smith     PetscInt    i;
3107a881295SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
3117a881295SBarry Smith     ierr = PetscByteSwapDouble(ppp,n);CHKERRQ(ierr);
3127a881295SBarry Smith #endif
3137a881295SBarry Smith     for (i=0; i<n; i++) {
3147a881295SBarry Smith       pv[i] = ppp[i];
3157a881295SBarry Smith     }
3167a881295SBarry Smith     ierr = PetscFree(ppp);CHKERRQ(ierr);
3177a881295SBarry Smith     PetscFunctionReturn(0);
3187a881295SBarry Smith   }
3197a881295SBarry Smith #endif
3207a881295SBarry Smith 
321e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
322e39fd77fSBarry Smith   ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
323e5c89e4eSSatish Balay #endif
324e5c89e4eSSatish Balay 
325*2d53ad75SBarry Smith   if (functionload) {
326*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
327*2d53ad75SBarry Smith     ierr = PetscDLSym(PETSC_NULL,fname,(void**)p);CHKERRQ(ierr);
328*2d53ad75SBarry Smith #else
329*2d53ad75SBarry Smith     *(void**)p = PETSC_NULL;
330*2d53ad75SBarry Smith #endif
331*2d53ad75SBarry Smith   }
332*2d53ad75SBarry Smith 
333e5c89e4eSSatish Balay   PetscFunctionReturn(0);
334e5c89e4eSSatish Balay }
335e5c89e4eSSatish Balay /* --------------------------------------------------------- */
336e5c89e4eSSatish Balay #undef __FUNCT__
337e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite"
338e30d2299SSatish Balay /*@
339e5c89e4eSSatish Balay    PetscBinaryWrite - Writes to a binary file.
340e5c89e4eSSatish Balay 
341e5c89e4eSSatish Balay    Not Collective
342e5c89e4eSSatish Balay 
343e5c89e4eSSatish Balay    Input Parameters:
344e5c89e4eSSatish Balay +  fd     - the file
345e5c89e4eSSatish Balay .  p      - the buffer
346e5c89e4eSSatish Balay .  n      - the number of items to write
347e5c89e4eSSatish Balay .  type   - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
348e5c89e4eSSatish Balay -  istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise.
349e5c89e4eSSatish Balay 
350e5c89e4eSSatish Balay    Level: advanced
351e5c89e4eSSatish Balay 
352e5c89e4eSSatish Balay    Notes:
353e5c89e4eSSatish Balay    PetscBinaryWrite() uses byte swapping to work on all machines; the files
354e5c89e4eSSatish Balay    are written using big-endian ordering to the file. On small-endian machines the numbers
355e5c89e4eSSatish Balay    are converted to the big-endian format when they are written to disk.
356e2e64c6bSBarry Smith    When PETSc is ./configure with --with-64bit-indices the integers are written to the
35754f21887SBarry Smith    file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices
35854f21887SBarry Smith    is used.
359e5c89e4eSSatish Balay 
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
366e5c89e4eSSatish Balay    the second time. If the flag istemp is set to PETSC_TRUE, the second
367e5c89e4eSSatish Balay    byte-swapping operation is not done, thus saving some computation,
368e5f36e38SBarry Smith    but the buffer is left corrupted.
369e5c89e4eSSatish Balay 
370300a7f5bSBarry Smith    Because byte-swapping may be done on the values in data it cannot be declared const
371300a7f5bSBarry Smith 
372e5c89e4eSSatish Balay    Concepts: files^writing binary
373e5c89e4eSSatish Balay    Concepts: binary files^writing
374e5c89e4eSSatish Balay 
3754ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
3764ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
377e5c89e4eSSatish Balay @*/
3787087cfbeSBarry Smith PetscErrorCode  PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscBool  istemp)
379e5c89e4eSSatish Balay {
380e5c89e4eSSatish Balay   char           *pp = (char*)p;
381e5c89e4eSSatish Balay   int            err,wsize;
382e5c89e4eSSatish Balay   size_t         m = (size_t)n,maxblock=65536;
3836de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
384e5c89e4eSSatish Balay   PetscErrorCode ierr;
385e5c89e4eSSatish Balay   void           *ptmp = p;
386e5c89e4eSSatish Balay #endif
387*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
388*2d53ad75SBarry Smith   char           fname[64];
389*2d53ad75SBarry Smith #endif
390e5c89e4eSSatish Balay 
391e5c89e4eSSatish Balay   PetscFunctionBegin;
392e32f2f54SBarry Smith   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n);
393e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
394e5c89e4eSSatish Balay 
395*2d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
396*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
397*2d53ad75SBarry Smith     const char *fnametmp;
398*2d53ad75SBarry Smith 
399*2d53ad75SBarry Smith     if (n > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only binary view a single function at a time");
400*2d53ad75SBarry Smith     ierr = PetscFPTFind(p,&fnametmp);CHKERRQ(ierr);
401*2d53ad75SBarry Smith     ierr = PetscStrncpy(fname,fnametmp,64);CHKERRQ(ierr);
402*2d53ad75SBarry Smith #else
403*2d53ad75SBarry Smith     ierr = PetscStrncpy(fname,"",64);CHKERRQ(ierr);
404*2d53ad75SBarry Smith #endif
405*2d53ad75SBarry Smith     m    = 64;
406*2d53ad75SBarry Smith     type = PETSC_CHAR;
407*2d53ad75SBarry Smith     pp   = (char*)fname;
408*2d53ad75SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
409*2d53ad75SBarry Smith     ptmp = (void*)fname;
410*2d53ad75SBarry Smith #endif
411*2d53ad75SBarry Smith   }
412*2d53ad75SBarry Smith 
4136de02169SBarry Smith   if (type == PETSC_INT)          m *= sizeof(PetscInt);
414e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
415e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
416e95bf02fSSatish Balay   else if (type == PETSC_FLOAT)   m *= sizeof(float);
417e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
418e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
419e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
420ace3abfcSBarry Smith   else if (type == PETSC_BOOL)   m *= sizeof(PetscBool);
4219f7b6320SBarry Smith   else if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char);
422e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
423e5c89e4eSSatish Balay 
424e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
425e39fd77fSBarry Smith   ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
426e5c89e4eSSatish Balay #endif
427e5c89e4eSSatish Balay 
428e5c89e4eSSatish Balay   while (m) {
429e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
430e5c89e4eSSatish Balay     err = write(fd,pp,wsize);
431e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
432e32f2f54SBarry Smith     if (err != wsize) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_WRITE,"Error writing to file.");
433e5c89e4eSSatish Balay     m -= wsize;
434e5c89e4eSSatish Balay     pp += wsize;
435e5c89e4eSSatish Balay   }
436e5c89e4eSSatish Balay 
437b659c0ceSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
438e5c89e4eSSatish Balay   if (!istemp) {
439e39fd77fSBarry Smith     ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
440e5c89e4eSSatish Balay   }
441e5c89e4eSSatish Balay #endif
442e5c89e4eSSatish Balay   PetscFunctionReturn(0);
443e5c89e4eSSatish Balay }
444e5c89e4eSSatish Balay 
445e5c89e4eSSatish Balay #undef __FUNCT__
446e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen"
447e5c89e4eSSatish Balay /*@C
448e5c89e4eSSatish Balay    PetscBinaryOpen - Opens a PETSc binary file.
449e5c89e4eSSatish Balay 
450e5c89e4eSSatish Balay    Not Collective
451e5c89e4eSSatish Balay 
452e5c89e4eSSatish Balay    Input Parameters:
453e5c89e4eSSatish Balay +  name - filename
45445c64e65SBarry Smith -  type - type of binary file, one of FILE_MODE_READ, FILE_MODE_APPEND, FILE_MODE_WRITE
455e5c89e4eSSatish Balay 
456e5c89e4eSSatish Balay    Output Parameter:
457e5c89e4eSSatish Balay .  fd - the file
458e5c89e4eSSatish Balay 
459e5c89e4eSSatish Balay    Level: advanced
460e5c89e4eSSatish Balay 
461e5c89e4eSSatish Balay   Concepts: files^opening binary
462e5c89e4eSSatish Balay   Concepts: binary files^opening
463e5c89e4eSSatish Balay 
464e5c89e4eSSatish Balay    Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in
465e5c89e4eSSatish Balay    big-endian format. This means the file can be accessed using PetscBinaryOpen() and
466e5c89e4eSSatish Balay    PetscBinaryRead() and PetscBinaryWrite() on any machine.
467e5c89e4eSSatish Balay 
4684ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(),
4694ebed01fSBarry Smith           PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
47045c64e65SBarry Smith 
471e5c89e4eSSatish Balay @*/
4727087cfbeSBarry Smith PetscErrorCode  PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd)
473e5c89e4eSSatish Balay {
474e5c89e4eSSatish Balay   PetscFunctionBegin;
475e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY)
47645c64e65SBarry Smith   if (mode == FILE_MODE_WRITE) {
477e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) {
478e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
479e5c89e4eSSatish Balay     }
48045c64e65SBarry Smith   } else if (mode == FILE_MODE_READ) {
481e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) {
482e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
483e5c89e4eSSatish Balay     }
48445c64e65SBarry Smith   } else if (mode == FILE_MODE_APPEND) {
485e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) {
486e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
487e5c89e4eSSatish Balay     }
488e5c89e4eSSatish Balay #else
48945c64e65SBarry Smith   if (mode == FILE_MODE_WRITE) {
490e5c89e4eSSatish Balay     if ((*fd = creat(name,0666)) == -1) {
491e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
492e5c89e4eSSatish Balay     }
49345c64e65SBarry Smith   } else if (mode == FILE_MODE_READ) {
494e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY,0)) == -1) {
495e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
496e5c89e4eSSatish Balay     }
497e5c89e4eSSatish Balay   }
49845c64e65SBarry Smith   else if (mode == FILE_MODE_APPEND) {
499e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY,0)) == -1) {
500e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
501e5c89e4eSSatish Balay     }
502e5c89e4eSSatish Balay #endif
503e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file mode");
504e5c89e4eSSatish Balay   PetscFunctionReturn(0);
505e5c89e4eSSatish Balay }
506e5c89e4eSSatish Balay 
507e5c89e4eSSatish Balay #undef __FUNCT__
508e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose"
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;
525e5c89e4eSSatish Balay   close(fd);
526e5c89e4eSSatish Balay   PetscFunctionReturn(0);
527e5c89e4eSSatish Balay }
528e5c89e4eSSatish Balay 
529e5c89e4eSSatish Balay 
530e5c89e4eSSatish Balay #undef __FUNCT__
531e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek"
532e30d2299SSatish Balay /*@
533e5c89e4eSSatish Balay    PetscBinarySeek - Moves the file pointer on a PETSc binary file.
534e5c89e4eSSatish Balay 
535e5c89e4eSSatish Balay    Not Collective
536e5c89e4eSSatish Balay 
537e5c89e4eSSatish Balay    Input Parameters:
538e5c89e4eSSatish Balay +  fd - the file
539ff553b35SMatthew Knepley .  off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
540e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
541ff553b35SMatthew Knepley -  whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file
542ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_CUR then off is an offset from the current location
543ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_END then off is an offset from the end of file
544e5c89e4eSSatish Balay 
545e5c89e4eSSatish Balay    Output Parameter:
546e5c89e4eSSatish Balay .   offset - new offset in file
547e5c89e4eSSatish Balay 
548e5c89e4eSSatish Balay    Level: developer
549e5c89e4eSSatish Balay 
550e5c89e4eSSatish Balay    Notes:
551e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
552e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
553e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
554e5c89e4eSSatish Balay    to determine the offset or location.
555e5c89e4eSSatish Balay 
556e5c89e4eSSatish Balay    Concepts: files^binary seeking
557e5c89e4eSSatish Balay    Concepts: binary files^seeking
558e5c89e4eSSatish Balay 
5594ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
5604ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
561e5c89e4eSSatish Balay @*/
5627087cfbeSBarry Smith PetscErrorCode  PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
563e5c89e4eSSatish Balay {
564e5c89e4eSSatish Balay   int iwhence = 0;
565e5c89e4eSSatish Balay 
566e5c89e4eSSatish Balay   PetscFunctionBegin;
567e5c89e4eSSatish Balay   if (whence == PETSC_BINARY_SEEK_SET) {
568e5c89e4eSSatish Balay     iwhence = SEEK_SET;
569e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_CUR) {
570e5c89e4eSSatish Balay     iwhence = SEEK_CUR;
571e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_END) {
572e5c89e4eSSatish Balay     iwhence = SEEK_END;
573e5c89e4eSSatish Balay   } else {
574e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location");
575e5c89e4eSSatish Balay   }
576e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK)
577e5c89e4eSSatish Balay   *offset = lseek(fd,off,iwhence);
578e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK)
579e5c89e4eSSatish Balay   *offset = _lseek(fd,(long)off,iwhence);
580e5c89e4eSSatish Balay #else
581e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file");
582e5c89e4eSSatish Balay #endif
583e5c89e4eSSatish Balay   PetscFunctionReturn(0);
584e5c89e4eSSatish Balay }
585e5c89e4eSSatish Balay 
586e5c89e4eSSatish Balay #undef __FUNCT__
5871d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedRead"
588e5c89e4eSSatish Balay /*@C
5891d280d73SBarry Smith    PetscBinarySynchronizedRead - Reads from a binary file.
590e5c89e4eSSatish Balay 
591e5c89e4eSSatish Balay    Collective on MPI_Comm
592e5c89e4eSSatish Balay 
593e5c89e4eSSatish Balay    Input Parameters:
594e5c89e4eSSatish Balay +  comm - the MPI communicator
595e5c89e4eSSatish Balay .  fd - the file
596e5c89e4eSSatish Balay .  n  - the number of items to read
597e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
598e5c89e4eSSatish Balay 
599e5c89e4eSSatish Balay    Output Parameters:
600e5c89e4eSSatish Balay .  p - the buffer
601e5c89e4eSSatish Balay 
602e5c89e4eSSatish Balay    Options Database Key:
603e5c89e4eSSatish Balay .   -binary_longints - indicates the file was generated on a Cray vector
604e5c89e4eSSatish Balay          machine (not the T3E/D) and the ints are stored as 64 bit
605e5c89e4eSSatish Balay          quantities, otherwise they are stored as 32 bit
606e5c89e4eSSatish Balay 
607e5c89e4eSSatish Balay    Level: developer
608e5c89e4eSSatish Balay 
609e5c89e4eSSatish Balay    Notes:
610e5c89e4eSSatish Balay    Does a PetscBinaryRead() followed by an MPI_Bcast()
611e5c89e4eSSatish Balay 
6121d280d73SBarry Smith    PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
613e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
614e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
615e5c89e4eSSatish Balay    binary file may be read on any machine.
616e5c89e4eSSatish Balay 
617e5c89e4eSSatish Balay    Concepts: files^synchronized reading of binary files
618e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
619e5c89e4eSSatish Balay 
6204ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
6214ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
622e5c89e4eSSatish Balay @*/
6237087cfbeSBarry Smith PetscErrorCode  PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type)
624e5c89e4eSSatish Balay {
625e5c89e4eSSatish Balay   PetscErrorCode ierr;
626e5c89e4eSSatish Balay   PetscMPIInt    rank;
627e5c89e4eSSatish Balay   MPI_Datatype   mtype;
628*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
629*2d53ad75SBarry Smith   char           *fname;
630*2d53ad75SBarry Smith #endif
631*2d53ad75SBarry Smith   PetscBool      functionload = PETSC_FALSE;
632*2d53ad75SBarry Smith   void           *ptmp;
633e5c89e4eSSatish Balay 
634e5c89e4eSSatish Balay   PetscFunctionBegin;
635*2d53ad75SBarry Smith   if (type == PETSC_FUNCTION) {
636*2d53ad75SBarry Smith     functionload = PETSC_TRUE;
637*2d53ad75SBarry Smith     n            = 64;
638*2d53ad75SBarry Smith     type         = PETSC_CHAR;
639*2d53ad75SBarry Smith     ptmp         = p;
640*2d53ad75SBarry Smith     /* warning memory leak */
641*2d53ad75SBarry Smith     fname        = malloc(64*sizeof(char));
642*2d53ad75SBarry Smith     p            = (void*)fname;
643*2d53ad75SBarry Smith   }
644*2d53ad75SBarry Smith 
645e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
646e5c89e4eSSatish Balay   if (!rank) {
647e5c89e4eSSatish Balay     ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr);
648e5c89e4eSSatish Balay   }
649e5c89e4eSSatish Balay   ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
650e5c89e4eSSatish Balay   ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr);
651*2d53ad75SBarry Smith 
652*2d53ad75SBarry Smith   if (functionload) {
653*2d53ad75SBarry Smith #if defined(PETSC_SERIALIZE_FUNCTIONS)
654*2d53ad75SBarry Smith     ierr = PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,PETSC_NULL,fname,(void**)ptmp);CHKERRQ(ierr);
655*2d53ad75SBarry Smith       /* ierr = PetscDLSym(PETSC_NULL,fname,(void**)ptmp);CHKERRQ(ierr);*/
656*2d53ad75SBarry Smith #else
657*2d53ad75SBarry Smith     *(void**)ptmp = PETSC_NULL;
658*2d53ad75SBarry Smith #endif
659*2d53ad75SBarry Smith   }
660e5c89e4eSSatish Balay   PetscFunctionReturn(0);
661e5c89e4eSSatish Balay }
662e5c89e4eSSatish Balay 
663e5c89e4eSSatish Balay #undef __FUNCT__
6641d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedWrite"
665e5c89e4eSSatish Balay /*@C
6661d280d73SBarry Smith    PetscBinarySynchronizedWrite - writes to a binary file.
667e5c89e4eSSatish Balay 
668e5c89e4eSSatish Balay    Collective on MPI_Comm
669e5c89e4eSSatish Balay 
670e5c89e4eSSatish Balay    Input Parameters:
671e5c89e4eSSatish Balay +  comm - the MPI communicator
672e5c89e4eSSatish Balay .  fd - the file
673e5c89e4eSSatish Balay .  n  - the number of items to write
674e5c89e4eSSatish Balay .  p - the buffer
675e5c89e4eSSatish Balay .  istemp - the buffer may be changed
676e5c89e4eSSatish Balay -  type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
677e5c89e4eSSatish Balay 
678e5c89e4eSSatish Balay    Level: developer
679e5c89e4eSSatish Balay 
680e5c89e4eSSatish Balay    Notes:
681e5c89e4eSSatish Balay    Process 0 does a PetscBinaryWrite()
682e5c89e4eSSatish Balay 
6831d280d73SBarry Smith    PetscBinarySynchronizedWrite() uses byte swapping to work on all machines.
684e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
685e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
686e5c89e4eSSatish Balay    binary file may be read on any machine.
687e5c89e4eSSatish Balay 
688300a7f5bSBarry Smith    Notes: because byte-swapping may be done on the values in data it cannot be declared const
689300a7f5bSBarry Smith 
6901d280d73SBarry Smith    WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0,
6911d280d73SBarry Smith    while PetscSynchronizedFPrintf() has all processes print their strings in order.
6921d280d73SBarry Smith 
693e5c89e4eSSatish Balay    Concepts: files^synchronized writing of binary files
694e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
695e5c89e4eSSatish Balay 
6964ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(),
6974ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
698e5c89e4eSSatish Balay @*/
6997087cfbeSBarry Smith PetscErrorCode  PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscBool  istemp)
700e5c89e4eSSatish Balay {
701e5c89e4eSSatish Balay   PetscErrorCode ierr;
702e5c89e4eSSatish Balay   PetscMPIInt    rank;
703e5c89e4eSSatish Balay 
704e5c89e4eSSatish Balay   PetscFunctionBegin;
705e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
706e5c89e4eSSatish Balay   if (!rank) {
707e5c89e4eSSatish Balay     ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr);
708e5c89e4eSSatish Balay   }
709e5c89e4eSSatish Balay   PetscFunctionReturn(0);
710e5c89e4eSSatish Balay }
711e5c89e4eSSatish Balay 
712e5c89e4eSSatish Balay #undef __FUNCT__
7131d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedSeek"
714e5c89e4eSSatish Balay /*@C
7151d280d73SBarry Smith    PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file.
716e5c89e4eSSatish Balay 
717e5c89e4eSSatish Balay 
718e5c89e4eSSatish Balay    Input Parameters:
719e5c89e4eSSatish Balay +  fd - the file
720e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
721e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
722e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
723e5c89e4eSSatish Balay -  off    - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
724e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
725e5c89e4eSSatish Balay 
726e5c89e4eSSatish Balay    Output Parameter:
727e5c89e4eSSatish Balay .   offset - new offset in file
728e5c89e4eSSatish Balay 
729e5c89e4eSSatish Balay    Level: developer
730e5c89e4eSSatish Balay 
731e5c89e4eSSatish Balay    Notes:
732e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
733e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
734e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
735e5c89e4eSSatish Balay    to determine the offset or location.
736e5c89e4eSSatish Balay 
737e5c89e4eSSatish Balay    Concepts: binary files^seeking
738e5c89e4eSSatish Balay    Concepts: files^seeking in binary
739e5c89e4eSSatish Balay 
7404ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
7414ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
742e5c89e4eSSatish Balay @*/
7437087cfbeSBarry Smith PetscErrorCode  PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
744e5c89e4eSSatish Balay {
745e5c89e4eSSatish Balay   PetscErrorCode ierr;
746e5c89e4eSSatish Balay   PetscMPIInt    rank;
747e5c89e4eSSatish Balay 
748e5c89e4eSSatish Balay   PetscFunctionBegin;
749e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
750e5c89e4eSSatish Balay   if (!rank) {
751e5c89e4eSSatish Balay     ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr);
752e5c89e4eSSatish Balay   }
753e5c89e4eSSatish Balay   PetscFunctionReturn(0);
754e5c89e4eSSatish Balay }
755e5c89e4eSSatish Balay 
7560fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO)
757e39fd77fSBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
758e39fd77fSBarry Smith 
759951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32)
760e39fd77fSBarry Smith EXTERN_C_BEGIN
761e39fd77fSBarry Smith /*
762e39fd77fSBarry Smith       MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions.
763e39fd77fSBarry Smith     These are set into MPI in PetscInitialize() via MPI_Register_datarep()
764e39fd77fSBarry Smith 
765e39fd77fSBarry Smith     Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode)
766e39fd77fSBarry Smith 
767951e3c8eSBarry Smith     The next three routines are not used because MPICH does not support their use
768e39fd77fSBarry Smith 
769e39fd77fSBarry Smith */
770e39fd77fSBarry Smith PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state)
771e39fd77fSBarry Smith {
772e39fd77fSBarry Smith   MPI_Aint    ub;
773e39fd77fSBarry Smith   PetscMPIInt ierr;
774e39fd77fSBarry Smith 
775e39fd77fSBarry Smith   ierr = MPI_Type_get_extent(datatype,&ub,file_extent);
776e39fd77fSBarry Smith   return ierr;
777e39fd77fSBarry Smith }
778e39fd77fSBarry Smith 
779e39fd77fSBarry Smith PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
780e39fd77fSBarry Smith {
781e39fd77fSBarry Smith   PetscDataType pdtype;
782e39fd77fSBarry Smith   PetscMPIInt   ierr;
783e39fd77fSBarry Smith   size_t        dsize;
784e39fd77fSBarry Smith 
785e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
786e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
787e39fd77fSBarry Smith 
788e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
789e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize*position;
790e39fd77fSBarry Smith 
791e39fd77fSBarry Smith   ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr);
792e39fd77fSBarry Smith   ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr);
793e39fd77fSBarry Smith   return ierr;
794e39fd77fSBarry Smith }
795e39fd77fSBarry Smith 
796e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
797e39fd77fSBarry Smith {
798e39fd77fSBarry Smith   PetscDataType pdtype;
799e39fd77fSBarry Smith   PetscMPIInt   ierr;
800e39fd77fSBarry Smith   size_t        dsize;
801e39fd77fSBarry Smith 
802e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
803e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
804e39fd77fSBarry Smith 
805e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
806e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize*position;
807e39fd77fSBarry Smith 
808e39fd77fSBarry Smith   ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr);
809e39fd77fSBarry Smith   ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr);
810e39fd77fSBarry Smith   return ierr;
811e39fd77fSBarry Smith }
812e39fd77fSBarry Smith EXTERN_C_END
813951e3c8eSBarry Smith #endif
814e39fd77fSBarry Smith 
8151c91e6bcSJed Brown #undef __FUNCT__
8161c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_write_all"
817e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
818e39fd77fSBarry Smith {
819e39fd77fSBarry Smith   PetscErrorCode ierr;
820e39fd77fSBarry Smith   PetscDataType  pdtype;
821e39fd77fSBarry Smith 
822e39fd77fSBarry Smith   PetscFunctionBegin;
823e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
824e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
825e39fd77fSBarry Smith   ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
826e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
827a6796414SBarry Smith   PetscFunctionReturn(0);
828e39fd77fSBarry Smith }
829e39fd77fSBarry Smith 
8301c91e6bcSJed Brown #undef __FUNCT__
8311c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_read_all"
832e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
833e39fd77fSBarry Smith {
834e39fd77fSBarry Smith   PetscErrorCode ierr;
835e39fd77fSBarry Smith   PetscDataType  pdtype;
836e39fd77fSBarry Smith 
837e39fd77fSBarry Smith   PetscFunctionBegin;
838e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
839e39fd77fSBarry Smith   ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
840e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
841a6796414SBarry Smith   PetscFunctionReturn(0);
842e39fd77fSBarry Smith }
843e39fd77fSBarry Smith #endif
844951e3c8eSBarry Smith #endif
845