xref: /petsc/src/sys/fileio/sysio.c (revision 9f7b63207eff1d49ba92c8dfbff6c97346b68fb7)
1e5c89e4eSSatish Balay #define PETSC_DLL
2e5c89e4eSSatish Balay 
3e5c89e4eSSatish Balay /*
4e5c89e4eSSatish Balay    This file contains simple binary read/write routines.
5e5c89e4eSSatish Balay  */
6e5c89e4eSSatish Balay 
7d382aafbSBarry Smith #include "petscsys.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
16e5c89e4eSSatish Balay #include "petscbt.h"
17e5c89e4eSSatish Balay 
18e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
19e5c89e4eSSatish Balay 
206de02169SBarry Smith /* --------------------------------------------------------- */
21e5c89e4eSSatish Balay #undef __FUNCT__
226de02169SBarry Smith #define __FUNCT__ "PetscByteSwapEnum"
23e5c89e4eSSatish Balay /*
246de02169SBarry Smith   PetscByteSwapEnum - Swap bytes in a  PETSc Enum
25e5c89e4eSSatish Balay 
26e5c89e4eSSatish Balay */
278738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT 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);
360b20345dSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) {
376de02169SBarry Smith       ptr2[i] = ptr1[sizeof(PetscEnum)-1-i];
38e5c89e4eSSatish Balay     }
39e0890e22SSatish Balay     for (i=0; i<(PetscInt)sizeof(PetscEnum); i++) {
40e0890e22SSatish Balay       ptr1[i] = ptr2[i];
41e0890e22SSatish Balay     }
42e5c89e4eSSatish Balay   }
43e5c89e4eSSatish Balay   PetscFunctionReturn(0);
44e5c89e4eSSatish Balay }
456de02169SBarry Smith 
466de02169SBarry Smith #undef __FUNCT__
476de02169SBarry Smith #define __FUNCT__ "PetscByteSwapTruth"
486de02169SBarry Smith /*
496de02169SBarry Smith   PetscByteSwapTruth - Swap bytes in a  PETSc Truth
506de02169SBarry Smith 
516de02169SBarry Smith */
528738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscByteSwapTruth(PetscTruth *buff,PetscInt n)
536de02169SBarry Smith {
546de02169SBarry Smith   PetscInt    i,j;
550b20345dSBarry Smith   PetscTruth  tmp = PETSC_FALSE;
56e0890e22SSatish Balay   char        *ptr1,*ptr2 = (char*)&tmp;
576de02169SBarry Smith 
586de02169SBarry Smith   PetscFunctionBegin;
596de02169SBarry Smith   for (j=0; j<n; j++) {
606de02169SBarry Smith     ptr1 = (char*)(buff + j);
610b20345dSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscTruth); i++) {
626de02169SBarry Smith       ptr2[i] = ptr1[sizeof(PetscTruth)-1-i];
636de02169SBarry Smith     }
64e0890e22SSatish Balay     for (i=0; i<(PetscInt)sizeof(PetscTruth); i++) {
65e0890e22SSatish Balay       ptr1[i] = ptr2[i];
66e0890e22SSatish Balay     }
676de02169SBarry Smith   }
686de02169SBarry Smith   PetscFunctionReturn(0);
696de02169SBarry Smith }
706de02169SBarry Smith 
71e5c89e4eSSatish Balay #undef __FUNCT__
72bd1d2e58SBarry Smith #define __FUNCT__ "PetscByteSwapInt"
73bd1d2e58SBarry Smith /*
746de02169SBarry Smith   PetscByteSwapInt - Swap bytes in a  PETSc integer (which may be 32 or 64 bits)
75bd1d2e58SBarry Smith 
76bd1d2e58SBarry Smith */
778738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscByteSwapInt(PetscInt *buff,PetscInt n)
78bd1d2e58SBarry Smith {
79bd1d2e58SBarry Smith   PetscInt  i,j,tmp = 0;
80e0890e22SSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
81bd1d2e58SBarry Smith 
82bd1d2e58SBarry Smith   PetscFunctionBegin;
83bd1d2e58SBarry Smith   for (j=0; j<n; j++) {
84bd1d2e58SBarry Smith     ptr1 = (char*)(buff + j);
850b20345dSBarry Smith     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) {
86bd1d2e58SBarry Smith       ptr2[i] = ptr1[sizeof(PetscInt)-1-i];
87bd1d2e58SBarry Smith     }
88e0890e22SSatish Balay     for (i=0; i<(PetscInt)sizeof(PetscInt); i++) {
89e0890e22SSatish Balay       ptr1[i] = ptr2[i];
90e0890e22SSatish Balay     }
91bd1d2e58SBarry Smith   }
92bd1d2e58SBarry Smith   PetscFunctionReturn(0);
93bd1d2e58SBarry Smith }
94bd1d2e58SBarry Smith /* --------------------------------------------------------- */
95bd1d2e58SBarry Smith #undef __FUNCT__
96e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort"
97e5c89e4eSSatish Balay /*
98e5c89e4eSSatish Balay   PetscByteSwapShort - Swap bytes in a short
99e5c89e4eSSatish Balay */
1008738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscByteSwapShort(short *buff,PetscInt n)
101e5c89e4eSSatish Balay {
102e5c89e4eSSatish Balay   PetscInt   i,j;
103e5c89e4eSSatish Balay   short      tmp;
104e5c89e4eSSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
105e5c89e4eSSatish Balay 
106e5c89e4eSSatish Balay   PetscFunctionBegin;
107e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
108e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
1090b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(short); i++) {
110e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(int)-1-i];
111e5c89e4eSSatish Balay     }
112e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(short); i++) {
113e0890e22SSatish Balay       ptr1[i] = ptr2[i];
114e0890e22SSatish Balay     }
115e5c89e4eSSatish Balay   }
116e5c89e4eSSatish Balay   PetscFunctionReturn(0);
117e5c89e4eSSatish Balay }
118e5c89e4eSSatish Balay /* --------------------------------------------------------- */
119e5c89e4eSSatish Balay #undef __FUNCT__
120e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar"
121e5c89e4eSSatish Balay /*
122e5c89e4eSSatish Balay   PetscByteSwapScalar - Swap bytes in a double
123e5c89e4eSSatish Balay   Complex is dealt with as if array of double twice as long.
124e5c89e4eSSatish Balay */
1258738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscByteSwapScalar(PetscScalar *buff,PetscInt n)
126e5c89e4eSSatish Balay {
127e5c89e4eSSatish Balay   PetscInt  i,j;
128e5c89e4eSSatish Balay   PetscReal tmp,*buff1 = (PetscReal*)buff;
129e5c89e4eSSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
130e5c89e4eSSatish Balay 
131e5c89e4eSSatish Balay   PetscFunctionBegin;
132e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX)
133e5c89e4eSSatish Balay   n *= 2;
134e5c89e4eSSatish Balay #endif
135e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
136e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
1370b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) {
138e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(PetscReal)-1-i];
139e5c89e4eSSatish Balay     }
140e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(PetscReal); i++) {
141e0890e22SSatish Balay       ptr1[i] = ptr2[i];
142e0890e22SSatish Balay     }
143e5c89e4eSSatish Balay   }
144e5c89e4eSSatish Balay   PetscFunctionReturn(0);
145e5c89e4eSSatish Balay }
146e5c89e4eSSatish Balay /* --------------------------------------------------------- */
147e5c89e4eSSatish Balay #undef __FUNCT__
148e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapDouble"
149e5c89e4eSSatish Balay /*
150e5c89e4eSSatish Balay   PetscByteSwapDouble - Swap bytes in a double
151e5c89e4eSSatish Balay */
1528738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscByteSwapDouble(double *buff,PetscInt n)
153e5c89e4eSSatish Balay {
154e5c89e4eSSatish Balay   PetscInt i,j;
155e5c89e4eSSatish Balay   double   tmp,*buff1 = (double*)buff;
156e5c89e4eSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
157e5c89e4eSSatish Balay 
158e5c89e4eSSatish Balay   PetscFunctionBegin;
159e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
160e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
1610b20345dSBarry Smith     for (i=0; i<(PetscInt) sizeof(double); i++) {
162e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(double)-1-i];
163e5c89e4eSSatish Balay     }
164e0890e22SSatish Balay     for (i=0; i<(PetscInt) sizeof(double); i++) {
165e0890e22SSatish Balay       ptr1[i] = ptr2[i];
166e0890e22SSatish Balay     }
167e5c89e4eSSatish Balay   }
168e5c89e4eSSatish Balay   PetscFunctionReturn(0);
169e5c89e4eSSatish Balay }
170e39fd77fSBarry Smith 
171e39fd77fSBarry Smith #undef __FUNCT__
172e39fd77fSBarry Smith #define __FUNCT__ "PetscByteSwap"
173e39fd77fSBarry Smith PetscErrorCode PetscByteSwap(void *data,PetscDataType pdtype,PetscInt count)
174e39fd77fSBarry Smith {
175e39fd77fSBarry Smith   PetscErrorCode ierr;
176e39fd77fSBarry Smith 
177e39fd77fSBarry Smith   PetscFunctionBegin;
178e39fd77fSBarry Smith   if      (pdtype == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt*)data,count);CHKERRQ(ierr);}
179e39fd77fSBarry Smith   else if (pdtype == PETSC_ENUM)   {ierr = PetscByteSwapEnum((PetscEnum*)data,count);CHKERRQ(ierr);}
180e39fd77fSBarry Smith   else if (pdtype == PETSC_TRUTH)  {ierr = PetscByteSwapTruth((PetscTruth*)data,count);CHKERRQ(ierr);}
181e39fd77fSBarry Smith   else if (pdtype == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)data,count);CHKERRQ(ierr);}
182e39fd77fSBarry Smith   else if (pdtype == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)data,count);CHKERRQ(ierr);}
183e39fd77fSBarry Smith   else if (pdtype == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)data,count);CHKERRQ(ierr);}
184e39fd77fSBarry Smith   PetscFunctionReturn(0);
185e39fd77fSBarry Smith }
186e39fd77fSBarry Smith 
187e5c89e4eSSatish Balay #endif
188e5c89e4eSSatish Balay /* --------------------------------------------------------- */
189e5c89e4eSSatish Balay #undef __FUNCT__
190e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead"
191e30d2299SSatish Balay /*@
192e5c89e4eSSatish Balay    PetscBinaryRead - Reads from a binary file.
193e5c89e4eSSatish Balay 
194e5c89e4eSSatish Balay    Not Collective
195e5c89e4eSSatish Balay 
196e5c89e4eSSatish Balay    Input Parameters:
197e5c89e4eSSatish Balay +  fd - the file
198e5c89e4eSSatish Balay .  n  - the number of items to read
199e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
200e5c89e4eSSatish Balay 
201e5c89e4eSSatish Balay    Output Parameters:
202e5c89e4eSSatish Balay .  p - the buffer
203e5c89e4eSSatish Balay 
204e5c89e4eSSatish Balay 
205e5c89e4eSSatish Balay 
206e5c89e4eSSatish Balay    Level: developer
207e5c89e4eSSatish Balay 
208e5c89e4eSSatish Balay    Notes:
209e5c89e4eSSatish Balay    PetscBinaryRead() uses byte swapping to work on all machines; the files
210e5c89e4eSSatish Balay    are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers
211e5c89e4eSSatish Balay    are converted to the small-endian format when they are read in from the file.
212e2e64c6bSBarry Smith    When PETSc is ./configure with --with-64bit-indices the integers are written to the
21354f21887SBarry Smith    file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices
21454f21887SBarry Smith    is used.
215e5c89e4eSSatish Balay 
216e5c89e4eSSatish Balay    Concepts: files^reading binary
217e5c89e4eSSatish Balay    Concepts: binary files^reading
218e5c89e4eSSatish Balay 
2194ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
2204ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
221e5c89e4eSSatish Balay @*/
2228738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type)
223e5c89e4eSSatish Balay {
224e5c89e4eSSatish Balay   int               wsize,err;
225e5c89e4eSSatish Balay   size_t            m = (size_t) n,maxblock = 65536;
226e5c89e4eSSatish Balay   char              *pp = (char*)p;
2276de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
2286de02169SBarry Smith   PetscErrorCode    ierr;
229e5c89e4eSSatish Balay   void              *ptmp = p;
230e5c89e4eSSatish Balay #endif
231e5c89e4eSSatish Balay 
232e5c89e4eSSatish Balay   PetscFunctionBegin;
233e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
234e5c89e4eSSatish Balay 
2356de02169SBarry Smith   if (type == PETSC_INT)          m *= sizeof(PetscInt);
236e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
237e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
238e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
239e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
240e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
241e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)   m *= sizeof(PetscTruth);
242*9f7b6320SBarry Smith   else if (type == PETSC_BIT_LOGICAL) m  = PetscBTLength(m)*sizeof(char);
243e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
244e5c89e4eSSatish Balay 
245e5c89e4eSSatish Balay   while (m) {
246e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
247e5c89e4eSSatish Balay     err = read(fd,pp,wsize);
248e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
249e32f2f54SBarry Smith     if (!err && wsize > 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Read past end of file");
250e32f2f54SBarry Smith     if (err < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Error reading from file, errno %d",errno);
251e5c89e4eSSatish Balay     m  -= err;
252e5c89e4eSSatish Balay     pp += err;
253e5c89e4eSSatish Balay   }
254e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
255e39fd77fSBarry Smith   ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
256e5c89e4eSSatish Balay #endif
257e5c89e4eSSatish Balay 
258e5c89e4eSSatish Balay   PetscFunctionReturn(0);
259e5c89e4eSSatish Balay }
260e5c89e4eSSatish Balay /* --------------------------------------------------------- */
261e5c89e4eSSatish Balay #undef __FUNCT__
262e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite"
263e30d2299SSatish Balay /*@
264e5c89e4eSSatish Balay    PetscBinaryWrite - Writes to a binary file.
265e5c89e4eSSatish Balay 
266e5c89e4eSSatish Balay    Not Collective
267e5c89e4eSSatish Balay 
268e5c89e4eSSatish Balay    Input Parameters:
269e5c89e4eSSatish Balay +  fd     - the file
270e5c89e4eSSatish Balay .  p      - the buffer
271e5c89e4eSSatish Balay .  n      - the number of items to write
272e5c89e4eSSatish Balay .  type   - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
273e5c89e4eSSatish Balay -  istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise.
274e5c89e4eSSatish Balay 
275e5c89e4eSSatish Balay    Level: advanced
276e5c89e4eSSatish Balay 
277e5c89e4eSSatish Balay    Notes:
278e5c89e4eSSatish Balay    PetscBinaryWrite() uses byte swapping to work on all machines; the files
279e5c89e4eSSatish Balay    are written using big-endian ordering to the file. On small-endian machines the numbers
280e5c89e4eSSatish Balay    are converted to the big-endian format when they are written to disk.
281e2e64c6bSBarry Smith    When PETSc is ./configure with --with-64bit-indices the integers are written to the
28254f21887SBarry Smith    file as 64 bit integers, this means they can only be read back in when the option --with-64bit-indices
28354f21887SBarry Smith    is used.
284e5c89e4eSSatish Balay 
285e5c89e4eSSatish Balay    The Buffer p should be read-write buffer, and not static data.
286e5c89e4eSSatish Balay    This way, byte-swapping is done in-place, and then the buffer is
287e5c89e4eSSatish Balay    written to the file.
288e5c89e4eSSatish Balay 
289e5c89e4eSSatish Balay    This routine restores the original contents of the buffer, after
290e5c89e4eSSatish Balay    it is written to the file. This is done by byte-swapping in-place
291e5c89e4eSSatish Balay    the second time. If the flag istemp is set to PETSC_TRUE, the second
292e5c89e4eSSatish Balay    byte-swapping operation is not done, thus saving some computation,
293e5c89e4eSSatish Balay    but the buffer corrupted is corrupted.
294e5c89e4eSSatish Balay 
295300a7f5bSBarry Smith    Because byte-swapping may be done on the values in data it cannot be declared const
296300a7f5bSBarry Smith 
297e5c89e4eSSatish Balay    Concepts: files^writing binary
298e5c89e4eSSatish Balay    Concepts: binary files^writing
299e5c89e4eSSatish Balay 
3004ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose(), PetscViewerBinaryGetDescriptor(), PetscBinarySynchronizedWrite(),
3014ebed01fSBarry Smith           PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
302e5c89e4eSSatish Balay @*/
3038738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp)
304e5c89e4eSSatish Balay {
305e5c89e4eSSatish Balay   char           *pp = (char*)p;
306e5c89e4eSSatish Balay   int            err,wsize;
307e5c89e4eSSatish Balay   size_t         m = (size_t)n,maxblock=65536;
3086de02169SBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
309e5c89e4eSSatish Balay   PetscErrorCode ierr;
310e5c89e4eSSatish Balay   void           *ptmp = p;
311e5c89e4eSSatish Balay #endif
312e5c89e4eSSatish Balay 
313e5c89e4eSSatish Balay   PetscFunctionBegin;
314e32f2f54SBarry Smith   if (n < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n);
315e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
316e5c89e4eSSatish Balay 
3176de02169SBarry Smith   if (type == PETSC_INT)          m *= sizeof(PetscInt);
318e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
319e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
320e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
321e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
322e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
323e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)   m *= sizeof(PetscTruth);
324*9f7b6320SBarry Smith   else if (type == PETSC_BIT_LOGICAL) m = PetscBTLength(m)*sizeof(char);
325e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
326e5c89e4eSSatish Balay 
327e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
328e39fd77fSBarry Smith   ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
329e5c89e4eSSatish Balay #endif
330e5c89e4eSSatish Balay 
331e5c89e4eSSatish Balay   while (m) {
332e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
333e5c89e4eSSatish Balay     err = write(fd,pp,wsize);
334e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
335e32f2f54SBarry Smith     if (err != wsize) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_WRITE,"Error writing to file.");
336e5c89e4eSSatish Balay     m -= wsize;
337e5c89e4eSSatish Balay     pp += wsize;
338e5c89e4eSSatish Balay   }
339e5c89e4eSSatish Balay 
340b659c0ceSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
341e5c89e4eSSatish Balay   if (!istemp) {
342e39fd77fSBarry Smith     ierr = PetscByteSwap(ptmp,type,n);CHKERRQ(ierr);
343e5c89e4eSSatish Balay   }
344e5c89e4eSSatish Balay #endif
345e5c89e4eSSatish Balay   PetscFunctionReturn(0);
346e5c89e4eSSatish Balay }
347e5c89e4eSSatish Balay 
348e5c89e4eSSatish Balay #undef __FUNCT__
349e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen"
350e5c89e4eSSatish Balay /*@C
351e5c89e4eSSatish Balay    PetscBinaryOpen - Opens a PETSc binary file.
352e5c89e4eSSatish Balay 
353e5c89e4eSSatish Balay    Not Collective
354e5c89e4eSSatish Balay 
355e5c89e4eSSatish Balay    Input Parameters:
356e5c89e4eSSatish Balay +  name - filename
35745c64e65SBarry Smith -  type - type of binary file, one of FILE_MODE_READ, FILE_MODE_APPEND, FILE_MODE_WRITE
358e5c89e4eSSatish Balay 
359e5c89e4eSSatish Balay    Output Parameter:
360e5c89e4eSSatish Balay .  fd - the file
361e5c89e4eSSatish Balay 
362e5c89e4eSSatish Balay    Level: advanced
363e5c89e4eSSatish Balay 
364e5c89e4eSSatish Balay   Concepts: files^opening binary
365e5c89e4eSSatish Balay   Concepts: binary files^opening
366e5c89e4eSSatish Balay 
367e5c89e4eSSatish Balay    Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in
368e5c89e4eSSatish Balay    big-endian format. This means the file can be accessed using PetscBinaryOpen() and
369e5c89e4eSSatish Balay    PetscBinaryRead() and PetscBinaryWrite() on any machine.
370e5c89e4eSSatish Balay 
3714ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscFileMode, PetscViewerFileSetMode(), PetscViewerBinaryGetDescriptor(),
3724ebed01fSBarry Smith           PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(), PetscBinarySynchronizedSeek()
37345c64e65SBarry Smith 
374e5c89e4eSSatish Balay @*/
3758738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinaryOpen(const char name[],PetscFileMode mode,int *fd)
376e5c89e4eSSatish Balay {
377e5c89e4eSSatish Balay   PetscFunctionBegin;
378e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY)
37945c64e65SBarry Smith   if (mode == FILE_MODE_WRITE) {
380e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) {
381e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
382e5c89e4eSSatish Balay     }
38345c64e65SBarry Smith   } else if (mode == FILE_MODE_READ) {
384e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) {
385e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
386e5c89e4eSSatish Balay     }
38745c64e65SBarry Smith   } else if (mode == FILE_MODE_APPEND) {
388e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) {
389e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
390e5c89e4eSSatish Balay     }
391e5c89e4eSSatish Balay #else
39245c64e65SBarry Smith   if (mode == FILE_MODE_WRITE) {
393e5c89e4eSSatish Balay     if ((*fd = creat(name,0666)) == -1) {
394e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
395e5c89e4eSSatish Balay     }
39645c64e65SBarry Smith   } else if (mode == FILE_MODE_READ) {
397e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY,0)) == -1) {
398e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
399e5c89e4eSSatish Balay     }
400e5c89e4eSSatish Balay   }
40145c64e65SBarry Smith   else if (mode == FILE_MODE_APPEND) {
402e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY,0)) == -1) {
403e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
404e5c89e4eSSatish Balay     }
405e5c89e4eSSatish Balay #endif
406e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file mode");
407e5c89e4eSSatish Balay   PetscFunctionReturn(0);
408e5c89e4eSSatish Balay }
409e5c89e4eSSatish Balay 
410e5c89e4eSSatish Balay #undef __FUNCT__
411e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose"
412e30d2299SSatish Balay /*@
413e5c89e4eSSatish Balay    PetscBinaryClose - Closes a PETSc binary file.
414e5c89e4eSSatish Balay 
415e5c89e4eSSatish Balay    Not Collective
416e5c89e4eSSatish Balay 
417e5c89e4eSSatish Balay    Output Parameter:
418e5c89e4eSSatish Balay .  fd - the file
419e5c89e4eSSatish Balay 
420e5c89e4eSSatish Balay    Level: advanced
421e5c89e4eSSatish Balay 
4224ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
4234ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
424e5c89e4eSSatish Balay @*/
4258738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinaryClose(int fd)
426e5c89e4eSSatish Balay {
427e5c89e4eSSatish Balay   PetscFunctionBegin;
428e5c89e4eSSatish Balay   close(fd);
429e5c89e4eSSatish Balay   PetscFunctionReturn(0);
430e5c89e4eSSatish Balay }
431e5c89e4eSSatish Balay 
432e5c89e4eSSatish Balay 
433e5c89e4eSSatish Balay #undef __FUNCT__
434e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek"
435e30d2299SSatish Balay /*@
436e5c89e4eSSatish Balay    PetscBinarySeek - Moves the file pointer on a PETSc binary file.
437e5c89e4eSSatish Balay 
438e5c89e4eSSatish Balay    Not Collective
439e5c89e4eSSatish Balay 
440e5c89e4eSSatish Balay    Input Parameters:
441e5c89e4eSSatish Balay +  fd - the file
442ff553b35SMatthew Knepley .  off - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
443e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
444ff553b35SMatthew Knepley -  whence - if PETSC_BINARY_SEEK_SET then off is an absolute location in the file
445ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_CUR then off is an offset from the current location
446ff553b35SMatthew Knepley             if PETSC_BINARY_SEEK_END then off is an offset from the end of file
447e5c89e4eSSatish Balay 
448e5c89e4eSSatish Balay    Output Parameter:
449e5c89e4eSSatish Balay .   offset - new offset in file
450e5c89e4eSSatish Balay 
451e5c89e4eSSatish Balay    Level: developer
452e5c89e4eSSatish Balay 
453e5c89e4eSSatish Balay    Notes:
454e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
455e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
456e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
457e5c89e4eSSatish Balay    to determine the offset or location.
458e5c89e4eSSatish Balay 
459e5c89e4eSSatish Balay    Concepts: files^binary seeking
460e5c89e4eSSatish Balay    Concepts: binary files^seeking
461e5c89e4eSSatish Balay 
4624ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
4634ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
464e5c89e4eSSatish Balay @*/
4658738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
466e5c89e4eSSatish Balay {
467e5c89e4eSSatish Balay   int iwhence = 0;
468e5c89e4eSSatish Balay 
469e5c89e4eSSatish Balay   PetscFunctionBegin;
470e5c89e4eSSatish Balay   if (whence == PETSC_BINARY_SEEK_SET) {
471e5c89e4eSSatish Balay     iwhence = SEEK_SET;
472e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_CUR) {
473e5c89e4eSSatish Balay     iwhence = SEEK_CUR;
474e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_END) {
475e5c89e4eSSatish Balay     iwhence = SEEK_END;
476e5c89e4eSSatish Balay   } else {
477e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location");
478e5c89e4eSSatish Balay   }
479e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK)
480e5c89e4eSSatish Balay   *offset = lseek(fd,off,iwhence);
481e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK)
482e5c89e4eSSatish Balay   *offset = _lseek(fd,(long)off,iwhence);
483e5c89e4eSSatish Balay #else
484e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file");
485e5c89e4eSSatish Balay #endif
486e5c89e4eSSatish Balay   PetscFunctionReturn(0);
487e5c89e4eSSatish Balay }
488e5c89e4eSSatish Balay 
489e5c89e4eSSatish Balay #undef __FUNCT__
4901d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedRead"
491e5c89e4eSSatish Balay /*@C
4921d280d73SBarry Smith    PetscBinarySynchronizedRead - Reads from a binary file.
493e5c89e4eSSatish Balay 
494e5c89e4eSSatish Balay    Collective on MPI_Comm
495e5c89e4eSSatish Balay 
496e5c89e4eSSatish Balay    Input Parameters:
497e5c89e4eSSatish Balay +  comm - the MPI communicator
498e5c89e4eSSatish Balay .  fd - the file
499e5c89e4eSSatish Balay .  n  - the number of items to read
500e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
501e5c89e4eSSatish Balay 
502e5c89e4eSSatish Balay    Output Parameters:
503e5c89e4eSSatish Balay .  p - the buffer
504e5c89e4eSSatish Balay 
505e5c89e4eSSatish Balay    Options Database Key:
506e5c89e4eSSatish Balay .   -binary_longints - indicates the file was generated on a Cray vector
507e5c89e4eSSatish Balay          machine (not the T3E/D) and the ints are stored as 64 bit
508e5c89e4eSSatish Balay          quantities, otherwise they are stored as 32 bit
509e5c89e4eSSatish Balay 
510e5c89e4eSSatish Balay    Level: developer
511e5c89e4eSSatish Balay 
512e5c89e4eSSatish Balay    Notes:
513e5c89e4eSSatish Balay    Does a PetscBinaryRead() followed by an MPI_Bcast()
514e5c89e4eSSatish Balay 
5151d280d73SBarry Smith    PetscBinarySynchronizedRead() uses byte swapping to work on all machines.
516e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
517e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
518e5c89e4eSSatish Balay    binary file may be read on any machine.
519e5c89e4eSSatish Balay 
520e5c89e4eSSatish Balay    Concepts: files^synchronized reading of binary files
521e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
522e5c89e4eSSatish Balay 
5234ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedWrite(),
5244ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
525e5c89e4eSSatish Balay @*/
5268738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinarySynchronizedRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type)
527e5c89e4eSSatish Balay {
528e5c89e4eSSatish Balay   PetscErrorCode ierr;
529e5c89e4eSSatish Balay   PetscMPIInt    rank;
530e5c89e4eSSatish Balay   MPI_Datatype   mtype;
531e5c89e4eSSatish Balay 
532e5c89e4eSSatish Balay   PetscFunctionBegin;
533e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
534e5c89e4eSSatish Balay   if (!rank) {
535e5c89e4eSSatish Balay     ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr);
536e5c89e4eSSatish Balay   }
537e5c89e4eSSatish Balay   ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
538e5c89e4eSSatish Balay   ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr);
539e5c89e4eSSatish Balay   PetscFunctionReturn(0);
540e5c89e4eSSatish Balay }
541e5c89e4eSSatish Balay 
542e5c89e4eSSatish Balay #undef __FUNCT__
5431d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedWrite"
544e5c89e4eSSatish Balay /*@C
5451d280d73SBarry Smith    PetscBinarySynchronizedWrite - writes to a binary file.
546e5c89e4eSSatish Balay 
547e5c89e4eSSatish Balay    Collective on MPI_Comm
548e5c89e4eSSatish Balay 
549e5c89e4eSSatish Balay    Input Parameters:
550e5c89e4eSSatish Balay +  comm - the MPI communicator
551e5c89e4eSSatish Balay .  fd - the file
552e5c89e4eSSatish Balay .  n  - the number of items to write
553e5c89e4eSSatish Balay .  p - the buffer
554e5c89e4eSSatish Balay .  istemp - the buffer may be changed
555e5c89e4eSSatish Balay -  type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
556e5c89e4eSSatish Balay 
557e5c89e4eSSatish Balay    Level: developer
558e5c89e4eSSatish Balay 
559e5c89e4eSSatish Balay    Notes:
560e5c89e4eSSatish Balay    Process 0 does a PetscBinaryWrite()
561e5c89e4eSSatish Balay 
5621d280d73SBarry Smith    PetscBinarySynchronizedWrite() uses byte swapping to work on all machines.
563e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
564e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
565e5c89e4eSSatish Balay    binary file may be read on any machine.
566e5c89e4eSSatish Balay 
567300a7f5bSBarry Smith    Notes: because byte-swapping may be done on the values in data it cannot be declared const
568300a7f5bSBarry Smith 
5691d280d73SBarry Smith    WARNING: This is NOT like PetscSynchronizedFPrintf()! This routine ignores calls on all but process 0,
5701d280d73SBarry Smith    while PetscSynchronizedFPrintf() has all processes print their strings in order.
5711d280d73SBarry Smith 
572e5c89e4eSSatish Balay    Concepts: files^synchronized writing of binary files
573e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
574e5c89e4eSSatish Balay 
5754ebed01fSBarry Smith .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead(), PetscBinarySynchronizedRead(),
5764ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
577e5c89e4eSSatish Balay @*/
5788738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinarySynchronizedWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp)
579e5c89e4eSSatish Balay {
580e5c89e4eSSatish Balay   PetscErrorCode ierr;
581e5c89e4eSSatish Balay   PetscMPIInt    rank;
582e5c89e4eSSatish Balay 
583e5c89e4eSSatish Balay   PetscFunctionBegin;
584e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
585e5c89e4eSSatish Balay   if (!rank) {
586e5c89e4eSSatish Balay     ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr);
587e5c89e4eSSatish Balay   }
588e5c89e4eSSatish Balay   PetscFunctionReturn(0);
589e5c89e4eSSatish Balay }
590e5c89e4eSSatish Balay 
591e5c89e4eSSatish Balay #undef __FUNCT__
5921d280d73SBarry Smith #define __FUNCT__ "PetscBinarySynchronizedSeek"
593e5c89e4eSSatish Balay /*@C
5941d280d73SBarry Smith    PetscBinarySynchronizedSeek - Moves the file pointer on a PETSc binary file.
595e5c89e4eSSatish Balay 
596e5c89e4eSSatish Balay 
597e5c89e4eSSatish Balay    Input Parameters:
598e5c89e4eSSatish Balay +  fd - the file
599e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
600e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
601e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
602e5c89e4eSSatish Balay -  off    - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
603e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
604e5c89e4eSSatish Balay 
605e5c89e4eSSatish Balay    Output Parameter:
606e5c89e4eSSatish Balay .   offset - new offset in file
607e5c89e4eSSatish Balay 
608e5c89e4eSSatish Balay    Level: developer
609e5c89e4eSSatish Balay 
610e5c89e4eSSatish Balay    Notes:
611e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
612e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
613e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
614e5c89e4eSSatish Balay    to determine the offset or location.
615e5c89e4eSSatish Balay 
616e5c89e4eSSatish Balay    Concepts: binary files^seeking
617e5c89e4eSSatish Balay    Concepts: files^seeking in binary
618e5c89e4eSSatish Balay 
6194ebed01fSBarry Smith .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen(), PetscBinarySynchronizedWrite(), PetscBinarySynchronizedRead(),
6204ebed01fSBarry Smith           PetscBinarySynchronizedSeek()
621e5c89e4eSSatish Balay @*/
6228738c821SJed Brown PetscErrorCode PETSCSYS_DLLEXPORT PetscBinarySynchronizedSeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
623e5c89e4eSSatish Balay {
624e5c89e4eSSatish Balay   PetscErrorCode ierr;
625e5c89e4eSSatish Balay   PetscMPIInt    rank;
626e5c89e4eSSatish Balay 
627e5c89e4eSSatish Balay   PetscFunctionBegin;
628e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
629e5c89e4eSSatish Balay   if (!rank) {
630e5c89e4eSSatish Balay     ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr);
631e5c89e4eSSatish Balay   }
632e5c89e4eSSatish Balay   PetscFunctionReturn(0);
633e5c89e4eSSatish Balay }
634e5c89e4eSSatish Balay 
6350fc9d207SBarry Smith #if defined(PETSC_HAVE_MPIIO)
636e39fd77fSBarry Smith #if !defined(PETSC_WORDS_BIGENDIAN)
637e39fd77fSBarry Smith 
638951e3c8eSBarry Smith #if defined(PETSC_USE_PETSC_MPI_EXTERNAL32)
639e39fd77fSBarry Smith EXTERN_C_BEGIN
640e39fd77fSBarry Smith /*
641e39fd77fSBarry Smith       MPICH does not provide the external32 representation for MPI_File_set_view() so we need to provide the functions.
642e39fd77fSBarry Smith     These are set into MPI in PetscInitialize() via MPI_Register_datarep()
643e39fd77fSBarry Smith 
644e39fd77fSBarry Smith     Note I use PetscMPIInt for the MPI error codes since that is what MPI uses (instead of the standard PetscErrorCode)
645e39fd77fSBarry Smith 
646951e3c8eSBarry Smith     The next three routines are not used because MPICH does not support their use
647e39fd77fSBarry Smith 
648e39fd77fSBarry Smith */
649e39fd77fSBarry Smith PetscMPIInt PetscDataRep_extent_fn(MPI_Datatype datatype,MPI_Aint *file_extent,void *extra_state)
650e39fd77fSBarry Smith {
651e39fd77fSBarry Smith   MPI_Aint    ub;
652e39fd77fSBarry Smith   PetscMPIInt ierr;
653e39fd77fSBarry Smith 
654e39fd77fSBarry Smith   ierr = MPI_Type_get_extent(datatype,&ub,file_extent);
655e39fd77fSBarry Smith   return ierr;
656e39fd77fSBarry Smith }
657e39fd77fSBarry Smith 
658e39fd77fSBarry Smith PetscMPIInt PetscDataRep_read_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
659e39fd77fSBarry Smith {
660e39fd77fSBarry Smith   PetscDataType pdtype;
661e39fd77fSBarry Smith   PetscMPIInt   ierr;
662e39fd77fSBarry Smith   size_t        dsize;
663e39fd77fSBarry Smith 
664e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
665e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
666e39fd77fSBarry Smith 
667e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
668e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize*position;
669e39fd77fSBarry Smith 
670e39fd77fSBarry Smith   ierr = PetscMemcpy(userbuf,filebuf,count*dsize);CHKERRQ(ierr);
671e39fd77fSBarry Smith   ierr = PetscByteSwap(userbuf,pdtype,count);CHKERRQ(ierr);
672e39fd77fSBarry Smith   return ierr;
673e39fd77fSBarry Smith }
674e39fd77fSBarry Smith 
675e39fd77fSBarry Smith PetscMPIInt PetscDataRep_write_conv_fn(void *userbuf, MPI_Datatype datatype,PetscMPIInt count,void *filebuf, MPI_Offset position,void *extra_state)
676e39fd77fSBarry Smith {
677e39fd77fSBarry Smith   PetscDataType pdtype;
678e39fd77fSBarry Smith   PetscMPIInt   ierr;
679e39fd77fSBarry Smith   size_t        dsize;
680e39fd77fSBarry Smith 
681e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(datatype,&pdtype);CHKERRQ(ierr);
682e39fd77fSBarry Smith   ierr = PetscDataTypeGetSize(pdtype,&dsize);CHKERRQ(ierr);
683e39fd77fSBarry Smith 
684e39fd77fSBarry Smith   /* offset is given in units of MPI_Datatype */
685e39fd77fSBarry Smith   userbuf = ((char *)userbuf) + dsize*position;
686e39fd77fSBarry Smith 
687e39fd77fSBarry Smith   ierr = PetscMemcpy(filebuf,userbuf,count*dsize);CHKERRQ(ierr);
688e39fd77fSBarry Smith   ierr = PetscByteSwap(filebuf,pdtype,count);CHKERRQ(ierr);
689e39fd77fSBarry Smith   return ierr;
690e39fd77fSBarry Smith }
691e39fd77fSBarry Smith EXTERN_C_END
692951e3c8eSBarry Smith #endif
693e39fd77fSBarry Smith 
6941c91e6bcSJed Brown #undef __FUNCT__
6951c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_write_all"
696e39fd77fSBarry Smith PetscErrorCode MPIU_File_write_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
697e39fd77fSBarry Smith {
698e39fd77fSBarry Smith   PetscErrorCode ierr;
699e39fd77fSBarry Smith   PetscDataType  pdtype;
700e39fd77fSBarry Smith 
701e39fd77fSBarry Smith   PetscFunctionBegin;
702e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
703e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
704e39fd77fSBarry Smith   ierr = MPI_File_write_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
705e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
706a6796414SBarry Smith   PetscFunctionReturn(0);
707e39fd77fSBarry Smith }
708e39fd77fSBarry Smith 
7091c91e6bcSJed Brown #undef __FUNCT__
7101c91e6bcSJed Brown #define __FUNCT__ "MPIU_File_read_all"
711e39fd77fSBarry Smith PetscErrorCode MPIU_File_read_all(MPI_File fd,void *data,PetscMPIInt cnt,MPI_Datatype dtype,MPI_Status *status)
712e39fd77fSBarry Smith {
713e39fd77fSBarry Smith   PetscErrorCode ierr;
714e39fd77fSBarry Smith   PetscDataType  pdtype;
715e39fd77fSBarry Smith 
716e39fd77fSBarry Smith   PetscFunctionBegin;
717e39fd77fSBarry Smith   ierr = PetscMPIDataTypeToPetscDataType(dtype,&pdtype);CHKERRQ(ierr);
718e39fd77fSBarry Smith   ierr = MPI_File_read_all(fd,data,cnt,dtype,status);CHKERRQ(ierr);
719e39fd77fSBarry Smith   ierr = PetscByteSwap(data,pdtype,cnt);CHKERRQ(ierr);
720a6796414SBarry Smith   PetscFunctionReturn(0);
721e39fd77fSBarry Smith }
722e39fd77fSBarry Smith #endif
723951e3c8eSBarry Smith #endif
724