xref: /petsc/src/sys/fileio/sysio.c (revision e30d229923a696673d75fd4bbec7dc9405e48f2f)
1e5c89e4eSSatish Balay #define PETSC_DLL
2e5c89e4eSSatish Balay 
3e5c89e4eSSatish Balay /*
4e5c89e4eSSatish Balay    This file contains simple binary read/write routines.
5e5c89e4eSSatish Balay  */
6e5c89e4eSSatish Balay 
7*e30d2299SSatish Balay #include "petsc.h"
8*e30d2299SSatish Balay #include "petscsys.h"     /*I          "petscsys.h"    I*/
9e5c89e4eSSatish Balay 
10e5c89e4eSSatish Balay #include <errno.h>
11e5c89e4eSSatish Balay #include <fcntl.h>
12e5c89e4eSSatish Balay #if defined(PETSC_HAVE_UNISTD_H)
13e5c89e4eSSatish Balay #include <unistd.h>
14e5c89e4eSSatish Balay #endif
15e5c89e4eSSatish Balay #if defined (PETSC_HAVE_IO_H)
16e5c89e4eSSatish Balay #include <io.h>
17e5c89e4eSSatish Balay #endif
18e5c89e4eSSatish Balay #include "petscbt.h"
19e5c89e4eSSatish Balay 
20e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8)
21e5c89e4eSSatish Balay #define PetscInt32 short
22e5c89e4eSSatish Balay #else
23e5c89e4eSSatish Balay #define PetscInt32 int
24e5c89e4eSSatish Balay #endif
25e5c89e4eSSatish Balay 
26e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
27e5c89e4eSSatish Balay 
28e5c89e4eSSatish Balay #undef __FUNCT__
29e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapInt"
30e5c89e4eSSatish Balay /*
31e5c89e4eSSatish Balay   PetscByteSwapInt - Swap bytes in a 32 bit integer. NOT a PetscInt! Note that PETSc binary read and write
32e5c89e4eSSatish Balay       always store and read only 32 bit integers! (See PetscBinaryRead(), PetscBinaryWrite()).
33e5c89e4eSSatish Balay 
34e5c89e4eSSatish Balay */
35e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapInt(PetscInt32 *buff,PetscInt n)
36e5c89e4eSSatish Balay {
37e5c89e4eSSatish Balay   PetscInt  i,j,tmp = 0;
38e5c89e4eSSatish Balay   PetscInt  *tptr = &tmp;                /* Need to access tmp indirectly to get */
39e5c89e4eSSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp; /* arround the bug in DEC-ALPHA g++ */
40e5c89e4eSSatish Balay 
41e5c89e4eSSatish Balay   PetscFunctionBegin;
42e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
43e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
44e5c89e4eSSatish Balay     for (i=0; i<(int)sizeof(PetscInt32); i++) {
45e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(PetscInt32)-1-i];
46e5c89e4eSSatish Balay     }
47e5c89e4eSSatish Balay     buff[j] = *tptr;
48e5c89e4eSSatish Balay   }
49e5c89e4eSSatish Balay   PetscFunctionReturn(0);
50e5c89e4eSSatish Balay }
51e5c89e4eSSatish Balay /* --------------------------------------------------------- */
52e5c89e4eSSatish Balay #undef __FUNCT__
53e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapShort"
54e5c89e4eSSatish Balay /*
55e5c89e4eSSatish Balay   PetscByteSwapShort - Swap bytes in a short
56e5c89e4eSSatish Balay */
57e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapShort(short *buff,PetscInt n)
58e5c89e4eSSatish Balay {
59e5c89e4eSSatish Balay   PetscInt   i,j;
60e5c89e4eSSatish Balay   short      tmp;
61e5c89e4eSSatish Balay   short      *tptr = &tmp;           /* take care pf bug in DEC-ALPHA g++ */
62e5c89e4eSSatish Balay   char       *ptr1,*ptr2 = (char*)&tmp;
63e5c89e4eSSatish Balay 
64e5c89e4eSSatish Balay   PetscFunctionBegin;
65e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
66e5c89e4eSSatish Balay     ptr1 = (char*)(buff + j);
67e5c89e4eSSatish Balay     for (i=0; i<(int) sizeof(short); i++) {
68e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(int)-1-i];
69e5c89e4eSSatish Balay     }
70e5c89e4eSSatish Balay     buff[j] = *tptr;
71e5c89e4eSSatish Balay   }
72e5c89e4eSSatish Balay   PetscFunctionReturn(0);
73e5c89e4eSSatish Balay }
74e5c89e4eSSatish Balay /* --------------------------------------------------------- */
75e5c89e4eSSatish Balay #undef __FUNCT__
76e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapScalar"
77e5c89e4eSSatish Balay /*
78e5c89e4eSSatish Balay   PetscByteSwapScalar - Swap bytes in a double
79e5c89e4eSSatish Balay   Complex is dealt with as if array of double twice as long.
80e5c89e4eSSatish Balay */
81e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapScalar(PetscScalar *buff,PetscInt n)
82e5c89e4eSSatish Balay {
83e5c89e4eSSatish Balay   PetscInt  i,j;
84e5c89e4eSSatish Balay   PetscReal tmp,*buff1 = (PetscReal*)buff;
85e5c89e4eSSatish Balay   PetscReal *tptr = &tmp;          /* take care pf bug in DEC-ALPHA g++ */
86e5c89e4eSSatish Balay   char      *ptr1,*ptr2 = (char*)&tmp;
87e5c89e4eSSatish Balay 
88e5c89e4eSSatish Balay   PetscFunctionBegin;
89e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX)
90e5c89e4eSSatish Balay   n *= 2;
91e5c89e4eSSatish Balay #endif
92e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
93e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
94e5c89e4eSSatish Balay     for (i=0; i<(int) sizeof(PetscReal); i++) {
95e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(PetscReal)-1-i];
96e5c89e4eSSatish Balay     }
97e5c89e4eSSatish Balay     buff1[j] = *tptr;
98e5c89e4eSSatish Balay   }
99e5c89e4eSSatish Balay   PetscFunctionReturn(0);
100e5c89e4eSSatish Balay }
101e5c89e4eSSatish Balay /* --------------------------------------------------------- */
102e5c89e4eSSatish Balay #undef __FUNCT__
103e5c89e4eSSatish Balay #define __FUNCT__ "PetscByteSwapDouble"
104e5c89e4eSSatish Balay /*
105e5c89e4eSSatish Balay   PetscByteSwapDouble - Swap bytes in a double
106e5c89e4eSSatish Balay */
107e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscByteSwapDouble(double *buff,PetscInt n)
108e5c89e4eSSatish Balay {
109e5c89e4eSSatish Balay   PetscInt i,j;
110e5c89e4eSSatish Balay   double   tmp,*buff1 = (double*)buff;
111e5c89e4eSSatish Balay   double   *tptr = &tmp;          /* take care pf bug in DEC-ALPHA g++ */
112e5c89e4eSSatish Balay   char     *ptr1,*ptr2 = (char*)&tmp;
113e5c89e4eSSatish Balay 
114e5c89e4eSSatish Balay   PetscFunctionBegin;
115e5c89e4eSSatish Balay   for (j=0; j<n; j++) {
116e5c89e4eSSatish Balay     ptr1 = (char*)(buff1 + j);
117e5c89e4eSSatish Balay     for (i=0; i<(int) sizeof(double); i++) {
118e5c89e4eSSatish Balay       ptr2[i] = ptr1[sizeof(double)-1-i];
119e5c89e4eSSatish Balay     }
120e5c89e4eSSatish Balay     buff1[j] = *tptr;
121e5c89e4eSSatish Balay   }
122e5c89e4eSSatish Balay   PetscFunctionReturn(0);
123e5c89e4eSSatish Balay }
124e5c89e4eSSatish Balay #endif
125e5c89e4eSSatish Balay /* --------------------------------------------------------- */
126e5c89e4eSSatish Balay #undef __FUNCT__
127e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryRead"
128*e30d2299SSatish Balay /*@
129e5c89e4eSSatish Balay    PetscBinaryRead - Reads from a binary file.
130e5c89e4eSSatish Balay 
131e5c89e4eSSatish Balay    Not Collective
132e5c89e4eSSatish Balay 
133e5c89e4eSSatish Balay    Input Parameters:
134e5c89e4eSSatish Balay +  fd - the file
135e5c89e4eSSatish Balay .  n  - the number of items to read
136e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
137e5c89e4eSSatish Balay 
138e5c89e4eSSatish Balay    Output Parameters:
139e5c89e4eSSatish Balay .  p - the buffer
140e5c89e4eSSatish Balay 
141e5c89e4eSSatish Balay 
142e5c89e4eSSatish Balay 
143e5c89e4eSSatish Balay    Level: developer
144e5c89e4eSSatish Balay 
145e5c89e4eSSatish Balay    Notes:
146e5c89e4eSSatish Balay    PetscBinaryRead() uses byte swapping to work on all machines; the files
147e5c89e4eSSatish Balay    are written to file ALWAYS using big-endian ordering. On small-endian machines the numbers
148e5c89e4eSSatish Balay    are converted to the small-endian format when they are read in from the file.
149e5c89e4eSSatish Balay    Integers are stored on the file as 32 bits long, regardless of whether
150e5c89e4eSSatish Balay    they are stored in the machine as 32 bits or 64 bits, this means the same
151e5c89e4eSSatish Balay    binary file may be read on any machine.
152e5c89e4eSSatish Balay 
153e5c89e4eSSatish Balay    Concepts: files^reading binary
154e5c89e4eSSatish Balay    Concepts: binary files^reading
155e5c89e4eSSatish Balay 
156e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose()
157e5c89e4eSSatish Balay @*/
158e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryRead(int fd,void *p,PetscInt n,PetscDataType type)
159e5c89e4eSSatish Balay {
160e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT) || !defined(PETSC_WORDS_BIGENDIAN)
161e5c89e4eSSatish Balay   PetscErrorCode    ierr;
162e5c89e4eSSatish Balay #endif
163e5c89e4eSSatish Balay   int               wsize,err;
164e5c89e4eSSatish Balay   size_t            m = (size_t) n,maxblock = 65536;
165e5c89e4eSSatish Balay   char              *pp = (char*)p;
166e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || !defined(PETSC_WORDS_BIGENDIAN) || defined(PETSC_USE_64BIT_INT)
167e5c89e4eSSatish Balay   void              *ptmp = p;
168e5c89e4eSSatish Balay #endif
169e5c89e4eSSatish Balay 
170e5c89e4eSSatish Balay   PetscFunctionBegin;
171e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
172e5c89e4eSSatish Balay 
173e5c89e4eSSatish Balay   if (type == PETSC_INT){
174e5c89e4eSSatish Balay     m   *= sizeof(PetscInt32);
175e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT)
176e5c89e4eSSatish Balay     /* read them in as 32 bit ints, later stretch into ints */
177e5c89e4eSSatish Balay     ierr = PetscMalloc(m,&pp);CHKERRQ(ierr);
178e5c89e4eSSatish Balay     ptmp = (void*)pp;
179e5c89e4eSSatish Balay #endif
180e5c89e4eSSatish Balay   }
181e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
182e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
183e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
184e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
185e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
186e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)   m *= sizeof(PetscTruth);
187e5c89e4eSSatish Balay   else if (type == PETSC_LOGICAL) m  = PetscBTLength(m)*sizeof(char);
188e5c89e4eSSatish Balay   else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
189e5c89e4eSSatish Balay 
190e5c89e4eSSatish Balay   while (m) {
191e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
192e5c89e4eSSatish Balay     err = read(fd,pp,wsize);
193e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
194e5c89e4eSSatish Balay     if (!err && wsize > 0) SETERRQ(PETSC_ERR_FILE_READ,"Read past end of file");
195e5c89e4eSSatish Balay     if (err < 0) SETERRQ(PETSC_ERR_FILE_READ,"Error reading from file");
196e5c89e4eSSatish Balay     m  -= err;
197e5c89e4eSSatish Balay     pp += err;
198e5c89e4eSSatish Balay   }
199e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
200e5c89e4eSSatish Balay   if      (type == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
201e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)   {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
202e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)  {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
203e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);}
204e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)ptmp,n);CHKERRQ(ierr);}
205e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);}
206e5c89e4eSSatish Balay #endif
207e5c89e4eSSatish Balay 
208e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT)
209e5c89e4eSSatish Balay   if (type == PETSC_INT) {
210e5c89e4eSSatish Balay     PetscInt   *p_int = (PetscInt*)p,i;
211e5c89e4eSSatish Balay     PetscInt32 *p_short = (PetscInt32 *)ptmp;
212e5c89e4eSSatish Balay     for (i=0; i<n; i++) {
213e5c89e4eSSatish Balay       p_int[i] = (PetscInt)p_short[i];
214e5c89e4eSSatish Balay     }
215e5c89e4eSSatish Balay     ierr = PetscFree(ptmp);CHKERRQ(ierr);
216e5c89e4eSSatish Balay   }
217e5c89e4eSSatish Balay #endif
218e5c89e4eSSatish Balay   PetscFunctionReturn(0);
219e5c89e4eSSatish Balay }
220e5c89e4eSSatish Balay /* --------------------------------------------------------- */
221e5c89e4eSSatish Balay #undef __FUNCT__
222e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryWrite"
223*e30d2299SSatish Balay /*@
224e5c89e4eSSatish Balay    PetscBinaryWrite - Writes to a binary file.
225e5c89e4eSSatish Balay 
226e5c89e4eSSatish Balay    Not Collective
227e5c89e4eSSatish Balay 
228e5c89e4eSSatish Balay    Input Parameters:
229e5c89e4eSSatish Balay +  fd     - the file
230e5c89e4eSSatish Balay .  p      - the buffer
231e5c89e4eSSatish Balay .  n      - the number of items to write
232e5c89e4eSSatish Balay .  type   - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
233e5c89e4eSSatish Balay -  istemp - PETSC_FALSE if buffer data should be preserved, PETSC_TRUE otherwise.
234e5c89e4eSSatish Balay 
235e5c89e4eSSatish Balay    Level: advanced
236e5c89e4eSSatish Balay 
237e5c89e4eSSatish Balay    Notes:
238e5c89e4eSSatish Balay    PetscBinaryWrite() uses byte swapping to work on all machines; the files
239e5c89e4eSSatish Balay    are written using big-endian ordering to the file. On small-endian machines the numbers
240e5c89e4eSSatish Balay    are converted to the big-endian format when they are written to disk.
241e5c89e4eSSatish Balay    Integers are stored on the file as 32 bits long, regardless of whether
242e5c89e4eSSatish Balay    they are stored in the machine as 32 bits or 64 bits, this means the same
243e5c89e4eSSatish Balay    binary file may be read on any machine. It also means that 64 bit integers larger than
244e5c89e4eSSatish Balay    roughly 2 billion are TRUNCATED/WRONG when written to the file.
245e5c89e4eSSatish Balay 
246e5c89e4eSSatish Balay    The Buffer p should be read-write buffer, and not static data.
247e5c89e4eSSatish Balay    This way, byte-swapping is done in-place, and then the buffer is
248e5c89e4eSSatish Balay    written to the file.
249e5c89e4eSSatish Balay 
250e5c89e4eSSatish Balay    This routine restores the original contents of the buffer, after
251e5c89e4eSSatish Balay    it is written to the file. This is done by byte-swapping in-place
252e5c89e4eSSatish Balay    the second time. If the flag istemp is set to PETSC_TRUE, the second
253e5c89e4eSSatish Balay    byte-swapping operation is not done, thus saving some computation,
254e5c89e4eSSatish Balay    but the buffer corrupted is corrupted.
255e5c89e4eSSatish Balay 
256e5c89e4eSSatish Balay    Concepts: files^writing binary
257e5c89e4eSSatish Balay    Concepts: binary files^writing
258e5c89e4eSSatish Balay 
259e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryOpen(), PetscBinaryClose()
260e5c89e4eSSatish Balay @*/
261e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryWrite(int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp)
262e5c89e4eSSatish Balay {
263e5c89e4eSSatish Balay   char           *pp = (char*)p;
264e5c89e4eSSatish Balay   int            err,wsize;
265e5c89e4eSSatish Balay   size_t         m = (size_t)n,maxblock=65536;
266e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) || (PETSC_SIZEOF_INT == 8) ||  defined(PETSC_USE_64BIT_INT)
267e5c89e4eSSatish Balay   PetscErrorCode ierr;
268e5c89e4eSSatish Balay   void           *ptmp = p;
269e5c89e4eSSatish Balay #endif
270e5c89e4eSSatish Balay 
271e5c89e4eSSatish Balay   PetscFunctionBegin;
272e5c89e4eSSatish Balay   if (n < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Trying to write a negative amount of data %D",n);
273e5c89e4eSSatish Balay   if (!n) PetscFunctionReturn(0);
274e5c89e4eSSatish Balay 
275e5c89e4eSSatish Balay   if (type == PETSC_INT){
276e5c89e4eSSatish Balay     m   *= sizeof(PetscInt32);
277e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT)
278e5c89e4eSSatish Balay     PetscInt   *p_int = (PetscInt*)p,i;
279e5c89e4eSSatish Balay     PetscInt32 *p_short;
280e5c89e4eSSatish Balay     ierr    = PetscMalloc(m,&pp);CHKERRQ(ierr);
281e5c89e4eSSatish Balay     ptmp    = (void*)pp;
282e5c89e4eSSatish Balay     p_short = (PetscInt32*)pp;
283e5c89e4eSSatish Balay 
284e5c89e4eSSatish Balay     for (i=0; i<n; i++) {
285e5c89e4eSSatish Balay       p_short[i] = (PetscInt32) p_int[i];
286e5c89e4eSSatish Balay     }
287e5c89e4eSSatish Balay     istemp = PETSC_TRUE;
288e5c89e4eSSatish Balay #endif
289e5c89e4eSSatish Balay   }
290e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR)  m *= sizeof(PetscScalar);
291e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE)  m *= sizeof(double);
292e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)   m *= sizeof(short);
293e5c89e4eSSatish Balay   else if (type == PETSC_CHAR)    m *= sizeof(char);
294e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)    m *= sizeof(PetscEnum);
295e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)   m *= sizeof(PetscTruth);
296e5c89e4eSSatish Balay   else if (type == PETSC_LOGICAL) m = PetscBTLength(m)*sizeof(char);
297e5c89e4eSSatish Balay   else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown type");
298e5c89e4eSSatish Balay 
299e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN)
300e5c89e4eSSatish Balay   if      (type == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
301e5c89e4eSSatish Balay   else if (type == PETSC_ENUM)   {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
302e5c89e4eSSatish Balay   else if (type == PETSC_TRUTH)  {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
303e5c89e4eSSatish Balay   else if (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);}
304e5c89e4eSSatish Balay   else if (type == PETSC_DOUBLE) {ierr = PetscByteSwapDouble((double*)ptmp,n);CHKERRQ(ierr);}
305e5c89e4eSSatish Balay   else if (type == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);}
306e5c89e4eSSatish Balay #endif
307e5c89e4eSSatish Balay 
308e5c89e4eSSatish Balay   while (m) {
309e5c89e4eSSatish Balay     wsize = (m < maxblock) ? m : maxblock;
310e5c89e4eSSatish Balay     err = write(fd,pp,wsize);
311e5c89e4eSSatish Balay     if (err < 0 && errno == EINTR) continue;
312e5c89e4eSSatish Balay     if (err != wsize) SETERRQ(PETSC_ERR_FILE_WRITE,"Error writing to file.");
313e5c89e4eSSatish Balay     m -= wsize;
314e5c89e4eSSatish Balay     pp += wsize;
315e5c89e4eSSatish Balay   }
316e5c89e4eSSatish Balay 
317e5c89e4eSSatish Balay #if !defined(PETSC_WORDS_BIGENDIAN) && !(PETSC_SIZEOF_INT == 8) && !defined(PETSC_USE_64BIT_INT)
318e5c89e4eSSatish Balay   if (!istemp) {
319e5c89e4eSSatish Balay     if      (type == PETSC_SCALAR) {ierr = PetscByteSwapScalar((PetscScalar*)ptmp,n);CHKERRQ(ierr);}
320e5c89e4eSSatish Balay     else if (type == PETSC_SHORT)  {ierr = PetscByteSwapShort((short*)ptmp,n);CHKERRQ(ierr);}
321e5c89e4eSSatish Balay     else if (type == PETSC_INT)    {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
322e5c89e4eSSatish Balay     else if (type == PETSC_ENUM)   {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
323e5c89e4eSSatish Balay     else if (type == PETSC_TRUTH)  {ierr = PetscByteSwapInt((PetscInt32*)ptmp,n);CHKERRQ(ierr);}
324e5c89e4eSSatish Balay   }
325e5c89e4eSSatish Balay #endif
326e5c89e4eSSatish Balay 
327e5c89e4eSSatish Balay #if (PETSC_SIZEOF_INT == 8) || defined(PETSC_USE_64BIT_INT)
328e5c89e4eSSatish Balay   if (type == PETSC_INT){
329e5c89e4eSSatish Balay     ierr = PetscFree(ptmp);CHKERRQ(ierr);
330e5c89e4eSSatish Balay   }
331e5c89e4eSSatish Balay #endif
332e5c89e4eSSatish Balay   PetscFunctionReturn(0);
333e5c89e4eSSatish Balay }
334e5c89e4eSSatish Balay 
335e5c89e4eSSatish Balay #undef __FUNCT__
336e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryOpen"
337e5c89e4eSSatish Balay /*@C
338e5c89e4eSSatish Balay    PetscBinaryOpen - Opens a PETSc binary file.
339e5c89e4eSSatish Balay 
340e5c89e4eSSatish Balay    Not Collective
341e5c89e4eSSatish Balay 
342e5c89e4eSSatish Balay    Input Parameters:
343e5c89e4eSSatish Balay +  name - filename
344e5c89e4eSSatish Balay -  type - type of binary file, on of PETSC_FILE_RDONLY, PETSC_FILE_WRONLY, PETSC_FILE_CREATE
345e5c89e4eSSatish Balay 
346e5c89e4eSSatish Balay    Output Parameter:
347e5c89e4eSSatish Balay .  fd - the file
348e5c89e4eSSatish Balay 
349e5c89e4eSSatish Balay    Level: advanced
350e5c89e4eSSatish Balay 
351e5c89e4eSSatish Balay   Concepts: files^opening binary
352e5c89e4eSSatish Balay   Concepts: binary files^opening
353e5c89e4eSSatish Balay 
354e5c89e4eSSatish Balay    Notes: Files access with PetscBinaryRead() and PetscBinaryWrite() are ALWAYS written in
355e5c89e4eSSatish Balay    big-endian format. This means the file can be accessed using PetscBinaryOpen() and
356e5c89e4eSSatish Balay    PetscBinaryRead() and PetscBinaryWrite() on any machine.
357e5c89e4eSSatish Balay 
358e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite()
359e5c89e4eSSatish Balay @*/
360e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryOpen(const char name[],int type,int *fd)
361e5c89e4eSSatish Balay {
362e5c89e4eSSatish Balay   PetscFunctionBegin;
363e5c89e4eSSatish Balay #if defined(PETSC_HAVE_O_BINARY)
364e5c89e4eSSatish Balay   if (type == PETSC_FILE_CREATE) {
365e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) {
366e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
367e5c89e4eSSatish Balay     }
368e5c89e4eSSatish Balay   } else if (type == PETSC_FILE_RDONLY) {
369e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY|O_BINARY,0)) == -1) {
370e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
371e5c89e4eSSatish Balay     }
372e5c89e4eSSatish Balay   } else if (type == PETSC_FILE_WRONLY) {
373e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY|O_BINARY,0)) == -1) {
374e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
375e5c89e4eSSatish Balay     }
376e5c89e4eSSatish Balay #else
377e5c89e4eSSatish Balay   if (type == PETSC_FILE_CREATE) {
378e5c89e4eSSatish Balay     if ((*fd = creat(name,0666)) == -1) {
379e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot create file for writing: %s",name);
380e5c89e4eSSatish Balay     }
381e5c89e4eSSatish Balay   } else if (type == PETSC_FILE_RDONLY) {
382e5c89e4eSSatish Balay     if ((*fd = open(name,O_RDONLY,0)) == -1) {
383e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for reading: %s",name);
384e5c89e4eSSatish Balay     }
385e5c89e4eSSatish Balay   }
386e5c89e4eSSatish Balay   else if (type == PETSC_FILE_WRONLY) {
387e5c89e4eSSatish Balay     if ((*fd = open(name,O_WRONLY,0)) == -1) {
388e5c89e4eSSatish Balay       SETERRQ1(PETSC_ERR_FILE_OPEN,"Cannot open file for writing: %s",name);
389e5c89e4eSSatish Balay     }
390e5c89e4eSSatish Balay #endif
391e5c89e4eSSatish Balay   } else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type");
392e5c89e4eSSatish Balay   PetscFunctionReturn(0);
393e5c89e4eSSatish Balay }
394e5c89e4eSSatish Balay 
395e5c89e4eSSatish Balay #undef __FUNCT__
396e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinaryClose"
397*e30d2299SSatish Balay /*@
398e5c89e4eSSatish Balay    PetscBinaryClose - Closes a PETSc binary file.
399e5c89e4eSSatish Balay 
400e5c89e4eSSatish Balay    Not Collective
401e5c89e4eSSatish Balay 
402e5c89e4eSSatish Balay    Output Parameter:
403e5c89e4eSSatish Balay .  fd - the file
404e5c89e4eSSatish Balay 
405e5c89e4eSSatish Balay    Level: advanced
406e5c89e4eSSatish Balay 
407e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen()
408e5c89e4eSSatish Balay @*/
409e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinaryClose(int fd)
410e5c89e4eSSatish Balay {
411e5c89e4eSSatish Balay   PetscFunctionBegin;
412e5c89e4eSSatish Balay   close(fd);
413e5c89e4eSSatish Balay   PetscFunctionReturn(0);
414e5c89e4eSSatish Balay }
415e5c89e4eSSatish Balay 
416e5c89e4eSSatish Balay 
417e5c89e4eSSatish Balay #undef __FUNCT__
418e5c89e4eSSatish Balay #define __FUNCT__ "PetscBinarySeek"
419*e30d2299SSatish Balay /*@
420e5c89e4eSSatish Balay    PetscBinarySeek - Moves the file pointer on a PETSc binary file.
421e5c89e4eSSatish Balay 
422e5c89e4eSSatish Balay    Not Collective
423e5c89e4eSSatish Balay 
424e5c89e4eSSatish Balay    Input Parameters:
425e5c89e4eSSatish Balay +  fd - the file
426e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
427e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
428e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
429e5c89e4eSSatish Balay -  size - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
430e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
431e5c89e4eSSatish Balay 
432e5c89e4eSSatish Balay    Output Parameter:
433e5c89e4eSSatish Balay .   offset - new offset in file
434e5c89e4eSSatish Balay 
435e5c89e4eSSatish Balay    Level: developer
436e5c89e4eSSatish Balay 
437e5c89e4eSSatish Balay    Notes:
438e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
439e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
440e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
441e5c89e4eSSatish Balay    to determine the offset or location.
442e5c89e4eSSatish Balay 
443e5c89e4eSSatish Balay    Concepts: files^binary seeking
444e5c89e4eSSatish Balay    Concepts: binary files^seeking
445e5c89e4eSSatish Balay 
446e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen()
447e5c89e4eSSatish Balay @*/
448e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscBinarySeek(int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
449e5c89e4eSSatish Balay {
450e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK) || defined(PETSC_HAVE__LSEEK)
451e5c89e4eSSatish Balay   int iwhence=0;
452e5c89e4eSSatish Balay #endif
453e5c89e4eSSatish Balay 
454e5c89e4eSSatish Balay   PetscFunctionBegin;
455e5c89e4eSSatish Balay   if (whence == PETSC_BINARY_SEEK_SET) {
456e5c89e4eSSatish Balay     iwhence = SEEK_SET;
457e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_CUR) {
458e5c89e4eSSatish Balay     iwhence = SEEK_CUR;
459e5c89e4eSSatish Balay   } else if (whence == PETSC_BINARY_SEEK_END) {
460e5c89e4eSSatish Balay     iwhence = SEEK_END;
461e5c89e4eSSatish Balay   } else {
462e5c89e4eSSatish Balay     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown seek location");
463e5c89e4eSSatish Balay   }
464e5c89e4eSSatish Balay #if defined(PETSC_HAVE_LSEEK)
465e5c89e4eSSatish Balay   *offset = lseek(fd,off,iwhence);
466e5c89e4eSSatish Balay #elif defined(PETSC_HAVE__LSEEK)
467e5c89e4eSSatish Balay   *offset = _lseek(fd,(long)off,iwhence);
468e5c89e4eSSatish Balay #else
469e5c89e4eSSatish Balay   SETERRQ(PETSC_ERR_SUP_SYS,"System does not have a way of seeking on a file");
470e5c89e4eSSatish Balay #endif
471e5c89e4eSSatish Balay   PetscFunctionReturn(0);
472e5c89e4eSSatish Balay }
473e5c89e4eSSatish Balay 
474e5c89e4eSSatish Balay #undef __FUNCT__
475e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinaryRead"
476e5c89e4eSSatish Balay /*@C
477e5c89e4eSSatish Balay    PetscSynchronizedBinaryRead - Reads from a binary file.
478e5c89e4eSSatish Balay 
479e5c89e4eSSatish Balay    Collective on MPI_Comm
480e5c89e4eSSatish Balay 
481e5c89e4eSSatish Balay    Input Parameters:
482e5c89e4eSSatish Balay +  comm - the MPI communicator
483e5c89e4eSSatish Balay .  fd - the file
484e5c89e4eSSatish Balay .  n  - the number of items to read
485e5c89e4eSSatish Balay -  type - the type of items to read (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
486e5c89e4eSSatish Balay 
487e5c89e4eSSatish Balay    Output Parameters:
488e5c89e4eSSatish Balay .  p - the buffer
489e5c89e4eSSatish Balay 
490e5c89e4eSSatish Balay    Options Database Key:
491e5c89e4eSSatish Balay .   -binary_longints - indicates the file was generated on a Cray vector
492e5c89e4eSSatish Balay          machine (not the T3E/D) and the ints are stored as 64 bit
493e5c89e4eSSatish Balay          quantities, otherwise they are stored as 32 bit
494e5c89e4eSSatish Balay 
495e5c89e4eSSatish Balay    Level: developer
496e5c89e4eSSatish Balay 
497e5c89e4eSSatish Balay    Notes:
498e5c89e4eSSatish Balay    Does a PetscBinaryRead() followed by an MPI_Bcast()
499e5c89e4eSSatish Balay 
500e5c89e4eSSatish Balay    PetscSynchronizedBinaryRead() uses byte swapping to work on all machines.
501e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
502e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
503e5c89e4eSSatish Balay    binary file may be read on any machine.
504e5c89e4eSSatish Balay 
505e5c89e4eSSatish Balay    Concepts: files^synchronized reading of binary files
506e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
507e5c89e4eSSatish Balay 
508e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead()
509e5c89e4eSSatish Balay @*/
510e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinaryRead(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type)
511e5c89e4eSSatish Balay {
512e5c89e4eSSatish Balay   PetscErrorCode ierr;
513e5c89e4eSSatish Balay   PetscMPIInt    rank;
514e5c89e4eSSatish Balay   MPI_Datatype   mtype;
515e5c89e4eSSatish Balay 
516e5c89e4eSSatish Balay   PetscFunctionBegin;
517e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
518e5c89e4eSSatish Balay   if (!rank) {
519e5c89e4eSSatish Balay     ierr = PetscBinaryRead(fd,p,n,type);CHKERRQ(ierr);
520e5c89e4eSSatish Balay   }
521e5c89e4eSSatish Balay   ierr = PetscDataTypeToMPIDataType(type,&mtype);CHKERRQ(ierr);
522e5c89e4eSSatish Balay   ierr = MPI_Bcast(p,n,mtype,0,comm);CHKERRQ(ierr);
523e5c89e4eSSatish Balay   PetscFunctionReturn(0);
524e5c89e4eSSatish Balay }
525e5c89e4eSSatish Balay 
526e5c89e4eSSatish Balay #undef __FUNCT__
527e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinaryWrite"
528e5c89e4eSSatish Balay /*@C
529e5c89e4eSSatish Balay    PetscSynchronizedBinaryWrite - writes to a binary file.
530e5c89e4eSSatish Balay 
531e5c89e4eSSatish Balay    Collective on MPI_Comm
532e5c89e4eSSatish Balay 
533e5c89e4eSSatish Balay    Input Parameters:
534e5c89e4eSSatish Balay +  comm - the MPI communicator
535e5c89e4eSSatish Balay .  fd - the file
536e5c89e4eSSatish Balay .  n  - the number of items to write
537e5c89e4eSSatish Balay .  p - the buffer
538e5c89e4eSSatish Balay .  istemp - the buffer may be changed
539e5c89e4eSSatish Balay -  type - the type of items to write (PETSC_INT, PETSC_DOUBLE or PETSC_SCALAR)
540e5c89e4eSSatish Balay 
541e5c89e4eSSatish Balay    Level: developer
542e5c89e4eSSatish Balay 
543e5c89e4eSSatish Balay    Notes:
544e5c89e4eSSatish Balay    Process 0 does a PetscBinaryWrite()
545e5c89e4eSSatish Balay 
546e5c89e4eSSatish Balay    PetscSynchronizedBinaryWrite() uses byte swapping to work on all machines.
547e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
548e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
549e5c89e4eSSatish Balay    binary file may be read on any machine.
550e5c89e4eSSatish Balay 
551e5c89e4eSSatish Balay    Concepts: files^synchronized writing of binary files
552e5c89e4eSSatish Balay    Concepts: binary files^reading, synchronized
553e5c89e4eSSatish Balay 
554e5c89e4eSSatish Balay .seealso: PetscBinaryWrite(), PetscBinaryOpen(), PetscBinaryClose(), PetscBinaryRead()
555e5c89e4eSSatish Balay @*/
556e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinaryWrite(MPI_Comm comm,int fd,void *p,PetscInt n,PetscDataType type,PetscTruth istemp)
557e5c89e4eSSatish Balay {
558e5c89e4eSSatish Balay   PetscErrorCode ierr;
559e5c89e4eSSatish Balay   PetscMPIInt    rank;
560e5c89e4eSSatish Balay 
561e5c89e4eSSatish Balay   PetscFunctionBegin;
562e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
563e5c89e4eSSatish Balay   if (!rank) {
564e5c89e4eSSatish Balay     ierr = PetscBinaryWrite(fd,p,n,type,istemp);CHKERRQ(ierr);
565e5c89e4eSSatish Balay   }
566e5c89e4eSSatish Balay   PetscFunctionReturn(0);
567e5c89e4eSSatish Balay }
568e5c89e4eSSatish Balay 
569e5c89e4eSSatish Balay #undef __FUNCT__
570e5c89e4eSSatish Balay #define __FUNCT__ "PetscSynchronizedBinarySeek"
571e5c89e4eSSatish Balay /*@C
572e5c89e4eSSatish Balay    PetscSynchronizedBinarySeek - Moves the file pointer on a PETSc binary file.
573e5c89e4eSSatish Balay 
574e5c89e4eSSatish Balay 
575e5c89e4eSSatish Balay    Input Parameters:
576e5c89e4eSSatish Balay +  fd - the file
577e5c89e4eSSatish Balay .  whence - if PETSC_BINARY_SEEK_SET then size is an absolute location in the file
578e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_CUR then size is offset from current location
579e5c89e4eSSatish Balay             if PETSC_BINARY_SEEK_END then size is offset from end of file
580e5c89e4eSSatish Balay -  off    - number of bytes to move. Use PETSC_BINARY_INT_SIZE, PETSC_BINARY_SCALAR_SIZE,
581e5c89e4eSSatish Balay             etc. in your calculation rather than sizeof() to compute byte lengths.
582e5c89e4eSSatish Balay 
583e5c89e4eSSatish Balay    Output Parameter:
584e5c89e4eSSatish Balay .   offset - new offset in file
585e5c89e4eSSatish Balay 
586e5c89e4eSSatish Balay    Level: developer
587e5c89e4eSSatish Balay 
588e5c89e4eSSatish Balay    Notes:
589e5c89e4eSSatish Balay    Integers are stored on the file as 32 long, regardless of whether
590e5c89e4eSSatish Balay    they are stored in the machine as 32 or 64, this means the same
591e5c89e4eSSatish Balay    binary file may be read on any machine. Hence you CANNOT use sizeof()
592e5c89e4eSSatish Balay    to determine the offset or location.
593e5c89e4eSSatish Balay 
594e5c89e4eSSatish Balay    Concepts: binary files^seeking
595e5c89e4eSSatish Balay    Concepts: files^seeking in binary
596e5c89e4eSSatish Balay 
597e5c89e4eSSatish Balay .seealso: PetscBinaryRead(), PetscBinaryWrite(), PetscBinaryOpen()
598e5c89e4eSSatish Balay @*/
599e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscSynchronizedBinarySeek(MPI_Comm comm,int fd,off_t off,PetscBinarySeekType whence,off_t *offset)
600e5c89e4eSSatish Balay {
601e5c89e4eSSatish Balay   PetscErrorCode ierr;
602e5c89e4eSSatish Balay   PetscMPIInt    rank;
603e5c89e4eSSatish Balay 
604e5c89e4eSSatish Balay   PetscFunctionBegin;
605e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
606e5c89e4eSSatish Balay   if (!rank) {
607e5c89e4eSSatish Balay     ierr = PetscBinarySeek(fd,off,whence,offset);CHKERRQ(ierr);
608e5c89e4eSSatish Balay   }
609e5c89e4eSSatish Balay   PetscFunctionReturn(0);
610e5c89e4eSSatish Balay }
611e5c89e4eSSatish Balay 
612