xref: /petsc/src/sys/classes/viewer/impls/binary/binv.c (revision 12801b3912e16a0709a58df21c3d316f86673188)
15c6c1daeSBarry Smith 
2af0996ceSBarry Smith #include <petsc/private/viewerimpl.h>    /*I   "petscviewer.h"   I*/
35c6c1daeSBarry Smith #include <fcntl.h>
45c6c1daeSBarry Smith #if defined(PETSC_HAVE_UNISTD_H)
55c6c1daeSBarry Smith #include <unistd.h>
65c6c1daeSBarry Smith #endif
75c6c1daeSBarry Smith #if defined(PETSC_HAVE_IO_H)
85c6c1daeSBarry Smith #include <io.h>
95c6c1daeSBarry Smith #endif
105c6c1daeSBarry Smith 
115c6c1daeSBarry Smith typedef struct  {
125c6c1daeSBarry Smith   int           fdes;                 /* file descriptor, ignored if using MPI IO */
135c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
14bc196f7cSDave May   PetscBool     usempiio;
155c6c1daeSBarry Smith   MPI_File      mfdes;                /* ignored unless using MPI IO */
165c6c1daeSBarry Smith   MPI_Offset    moff;
175c6c1daeSBarry Smith #endif
185c6c1daeSBarry Smith   PetscFileMode btype;                /* read or write? */
195c6c1daeSBarry Smith   FILE          *fdes_info;           /* optional file containing info on binary file*/
205c6c1daeSBarry Smith   PetscBool     storecompressed;      /* gzip the write binary file when closing it*/
215c6c1daeSBarry Smith   char          *filename;
22f90597f1SStefano Zampini   char          *ogzfilename;         /* gzip can be run after the filename has been updated */
235c6c1daeSBarry Smith   PetscBool     skipinfo;             /* Don't create info file for writing; don't use for reading */
245c6c1daeSBarry Smith   PetscBool     skipoptions;          /* don't use PETSc options database when loading */
255c6c1daeSBarry Smith   PetscInt      flowcontrol;          /* allow only <flowcontrol> messages outstanding at a time while doing IO */
265c6c1daeSBarry Smith   PetscBool     skipheader;           /* don't write header, only raw data */
27a261c58fSBarry Smith   PetscBool     matlabheaderwritten;  /* if format is PETSC_VIEWER_BINARY_MATLAB has the MATLAB .info header been written yet */
28c98fd787SBarry Smith   PetscBool     setfromoptionscalled;
295c6c1daeSBarry Smith } PetscViewer_Binary;
305c6c1daeSBarry Smith 
3181f0254dSBarry Smith static PetscErrorCode PetscViewerGetSubViewer_Binary(PetscViewer viewer,MPI_Comm comm,PetscViewer *outviewer)
325c6c1daeSBarry Smith {
335c6c1daeSBarry Smith   int                rank;
345c6c1daeSBarry Smith   PetscErrorCode     ierr;
355c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data,*obinary;
365c6c1daeSBarry Smith 
375c6c1daeSBarry Smith   PetscFunctionBegin;
38c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
3996509d17SLisandro Dalcin #if defined(PETSC_HAVE_MPIIO)
4096509d17SLisandro Dalcin   if (vbinary->usempiio) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Not yet implemented for binary viewers using MPI I/O");
4196509d17SLisandro Dalcin #endif
42ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
435c6c1daeSBarry Smith   if (!rank) {
445c6c1daeSBarry Smith     ierr    = PetscViewerCreate(PETSC_COMM_SELF,outviewer);CHKERRQ(ierr);
455c6c1daeSBarry Smith     ierr    = PetscViewerSetType(*outviewer,PETSCVIEWERBINARY);CHKERRQ(ierr);
465c6c1daeSBarry Smith     obinary = (PetscViewer_Binary*)(*outviewer)->data;
475c6c1daeSBarry Smith     ierr    = PetscMemcpy(obinary,vbinary,sizeof(PetscViewer_Binary));CHKERRQ(ierr);
4896509d17SLisandro Dalcin   } else {
4996509d17SLisandro Dalcin     *outviewer = NULL;
5096509d17SLisandro Dalcin   }
515c6c1daeSBarry Smith   PetscFunctionReturn(0);
525c6c1daeSBarry Smith }
535c6c1daeSBarry Smith 
5481f0254dSBarry Smith static PetscErrorCode PetscViewerRestoreSubViewer_Binary(PetscViewer viewer,MPI_Comm comm,PetscViewer *outviewer)
555c6c1daeSBarry Smith {
565c6c1daeSBarry Smith   PetscErrorCode ierr;
575c6c1daeSBarry Smith   PetscErrorCode rank;
585c6c1daeSBarry Smith 
595c6c1daeSBarry Smith   PetscFunctionBegin;
60ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
615c6c1daeSBarry Smith   if (!rank) {
625c6c1daeSBarry Smith     ierr = PetscFree((*outviewer)->data);CHKERRQ(ierr);
635c6c1daeSBarry Smith     ierr = PetscHeaderDestroy(outviewer);CHKERRQ(ierr);
645c6c1daeSBarry Smith   }
655c6c1daeSBarry Smith   PetscFunctionReturn(0);
665c6c1daeSBarry Smith }
675c6c1daeSBarry Smith 
685c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
695c6c1daeSBarry Smith /*@C
705c6c1daeSBarry Smith     PetscViewerBinaryGetMPIIOOffset - Gets the current offset that should be passed to MPI_File_set_view()
715c6c1daeSBarry Smith 
725c6c1daeSBarry Smith     Not Collective
735c6c1daeSBarry Smith 
745c6c1daeSBarry Smith     Input Parameter:
755c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
765c6c1daeSBarry Smith 
775c6c1daeSBarry Smith     Output Parameter:
785c6c1daeSBarry Smith .    off - the current offset
795c6c1daeSBarry Smith 
805c6c1daeSBarry Smith     Level: advanced
815c6c1daeSBarry Smith 
825c6c1daeSBarry Smith     Fortran Note:
835c6c1daeSBarry Smith     This routine is not supported in Fortran.
845c6c1daeSBarry Smith 
855c6c1daeSBarry Smith     Use PetscViewerBinaryAddMPIIOOffset() to increase this value after you have written a view.
865c6c1daeSBarry Smith 
875c6c1daeSBarry Smith   Concepts: file descriptor^getting
885c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
895c6c1daeSBarry Smith 
9067918a83SBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryAddMPIIOOffset()
915c6c1daeSBarry Smith @*/
925c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetMPIIOOffset(PetscViewer viewer,MPI_Offset *off)
935c6c1daeSBarry Smith {
945c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
955c6c1daeSBarry Smith 
965c6c1daeSBarry Smith   PetscFunctionBegin;
975c6c1daeSBarry Smith   *off = vbinary->moff;
985c6c1daeSBarry Smith   PetscFunctionReturn(0);
995c6c1daeSBarry Smith }
1005c6c1daeSBarry Smith 
1015c6c1daeSBarry Smith /*@C
1025c6c1daeSBarry Smith     PetscViewerBinaryAddMPIIOOffset - Adds to the current offset that should be passed to MPI_File_set_view()
1035c6c1daeSBarry Smith 
1045c6c1daeSBarry Smith     Not Collective
1055c6c1daeSBarry Smith 
1065c6c1daeSBarry Smith     Input Parameters:
1075c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1085c6c1daeSBarry Smith -    off - the addition to the offset
1095c6c1daeSBarry Smith 
1105c6c1daeSBarry Smith     Level: advanced
1115c6c1daeSBarry Smith 
1125c6c1daeSBarry Smith     Fortran Note:
1135c6c1daeSBarry Smith     This routine is not supported in Fortran.
1145c6c1daeSBarry Smith 
1155c6c1daeSBarry Smith     Use PetscViewerBinaryGetMPIIOOffset() to get the value that you should pass to MPI_File_set_view()
1165c6c1daeSBarry Smith 
1175c6c1daeSBarry Smith   Concepts: file descriptor^getting
1185c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1195c6c1daeSBarry Smith 
12067918a83SBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
1215c6c1daeSBarry Smith @*/
1225c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryAddMPIIOOffset(PetscViewer viewer,MPI_Offset off)
1235c6c1daeSBarry Smith {
1245c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1255c6c1daeSBarry Smith 
1265c6c1daeSBarry Smith   PetscFunctionBegin;
1275c6c1daeSBarry Smith   vbinary->moff += off;
1285c6c1daeSBarry Smith   PetscFunctionReturn(0);
1295c6c1daeSBarry Smith }
1305c6c1daeSBarry Smith 
1315c6c1daeSBarry Smith /*@C
1325c6c1daeSBarry Smith     PetscViewerBinaryGetMPIIODescriptor - Extracts the MPI IO file descriptor from a PetscViewer.
1335c6c1daeSBarry Smith 
1345c6c1daeSBarry Smith     Not Collective
1355c6c1daeSBarry Smith 
1365c6c1daeSBarry Smith     Input Parameter:
1375c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1385c6c1daeSBarry Smith 
1395c6c1daeSBarry Smith     Output Parameter:
1405c6c1daeSBarry Smith .   fdes - file descriptor
1415c6c1daeSBarry Smith 
1425c6c1daeSBarry Smith     Level: advanced
1435c6c1daeSBarry Smith 
1445c6c1daeSBarry Smith     Fortran Note:
1455c6c1daeSBarry Smith     This routine is not supported in Fortran.
1465c6c1daeSBarry Smith 
1475c6c1daeSBarry Smith   Concepts: file descriptor^getting
1485c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1495c6c1daeSBarry Smith 
15067918a83SBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
1515c6c1daeSBarry Smith @*/
1525c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetMPIIODescriptor(PetscViewer viewer,MPI_File *fdes)
1535c6c1daeSBarry Smith {
15403a1d59fSDave May   PetscErrorCode     ierr;
1555c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1565c6c1daeSBarry Smith 
1575c6c1daeSBarry Smith   PetscFunctionBegin;
158c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
1595c6c1daeSBarry Smith   *fdes = vbinary->mfdes;
1605c6c1daeSBarry Smith   PetscFunctionReturn(0);
1615c6c1daeSBarry Smith }
1625c6c1daeSBarry Smith 
16381f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetUseMPIIO_Binary(PetscViewer viewer,PetscBool  *flg)
164a8aae444SDave May {
165a8aae444SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
166a8aae444SDave May 
167a8aae444SDave May   PetscFunctionBegin;
168bc196f7cSDave May   *flg = vbinary->usempiio;
169a8aae444SDave May   PetscFunctionReturn(0);
170a8aae444SDave May }
171a8aae444SDave May #endif
172a8aae444SDave May 
173a8aae444SDave May 
1745c6c1daeSBarry Smith /*@C
175bc196f7cSDave May     PetscViewerBinaryGetUseMPIIO - Returns PETSC_TRUE if the binary viewer uses MPI-IO.
1765c6c1daeSBarry Smith 
1775c6c1daeSBarry Smith     Not Collective
1785c6c1daeSBarry Smith 
1795c6c1daeSBarry Smith     Input Parameter:
1805c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1815c6c1daeSBarry Smith 
1825c6c1daeSBarry Smith     Output Parameter:
183bc196f7cSDave May -   flg - PETSC_TRUE if MPI-IO is being used
1845c6c1daeSBarry Smith 
1855c6c1daeSBarry Smith     Options Database:
1865c6c1daeSBarry Smith     -viewer_binary_mpiio : Flag for using MPI-IO
1875c6c1daeSBarry Smith 
1885c6c1daeSBarry Smith     Level: advanced
1895c6c1daeSBarry Smith 
190bc196f7cSDave May     Note:
191bc196f7cSDave May     If MPI-IO is not available, this function will always return PETSC_FALSE
192bc196f7cSDave May 
1935c6c1daeSBarry Smith     Fortran Note:
1945c6c1daeSBarry Smith     This routine is not supported in Fortran.
1955c6c1daeSBarry Smith 
1965c6c1daeSBarry Smith   Concepts: file descriptor^getting
1975c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1985c6c1daeSBarry Smith 
19967918a83SBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetInfoPointer(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
2005c6c1daeSBarry Smith @*/
201bc196f7cSDave May PetscErrorCode PetscViewerBinaryGetUseMPIIO(PetscViewer viewer,PetscBool *flg)
2025c6c1daeSBarry Smith {
203a8aae444SDave May   PetscErrorCode ierr;
2045c6c1daeSBarry Smith 
2055c6c1daeSBarry Smith   PetscFunctionBegin;
206a8aae444SDave May   *flg = PETSC_FALSE;
207bc196f7cSDave May   ierr = PetscTryMethod(viewer,"PetscViewerBinaryGetUseMPIIO_C",(PetscViewer,PetscBool*),(viewer,flg));CHKERRQ(ierr);
2085c6c1daeSBarry Smith   PetscFunctionReturn(0);
2095c6c1daeSBarry Smith }
2105c6c1daeSBarry Smith 
21181f0254dSBarry Smith static PetscErrorCode  PetscViewerBinaryGetFlowControl_Binary(PetscViewer viewer,PetscInt *fc)
2125c6c1daeSBarry Smith {
2135c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
2145c6c1daeSBarry Smith 
2155c6c1daeSBarry Smith   PetscFunctionBegin;
2165c6c1daeSBarry Smith   *fc = vbinary->flowcontrol;
2175c6c1daeSBarry Smith   PetscFunctionReturn(0);
2185c6c1daeSBarry Smith }
2195c6c1daeSBarry Smith 
2205c6c1daeSBarry Smith /*@C
2215c6c1daeSBarry Smith     PetscViewerBinaryGetFlowControl - Returns how many messages are allowed to outstanding at the same time during parallel IO reads/writes
2225c6c1daeSBarry Smith 
2235c6c1daeSBarry Smith     Not Collective
2245c6c1daeSBarry Smith 
2255c6c1daeSBarry Smith     Input Parameter:
2265c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
2275c6c1daeSBarry Smith 
2285c6c1daeSBarry Smith     Output Parameter:
2295c6c1daeSBarry Smith .   fc - the number of messages
2305c6c1daeSBarry Smith 
2315c6c1daeSBarry Smith     Level: advanced
2325c6c1daeSBarry Smith 
2335c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinarySetFlowControl()
2345c6c1daeSBarry Smith 
2355c6c1daeSBarry Smith @*/
2365c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetFlowControl(PetscViewer viewer,PetscInt *fc)
2375c6c1daeSBarry Smith {
2385c6c1daeSBarry Smith   PetscErrorCode ierr;
2395c6c1daeSBarry Smith 
2405c6c1daeSBarry Smith   PetscFunctionBegin;
241163d334eSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetFlowControl_C",(PetscViewer,PetscInt*),(viewer,fc));CHKERRQ(ierr);
2425c6c1daeSBarry Smith   PetscFunctionReturn(0);
2435c6c1daeSBarry Smith }
2445c6c1daeSBarry Smith 
24581f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetFlowControl_Binary(PetscViewer viewer,PetscInt fc)
2465c6c1daeSBarry Smith {
2475c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
2485c6c1daeSBarry Smith 
2495c6c1daeSBarry Smith   PetscFunctionBegin;
250ce94432eSBarry Smith   if (fc <= 1) SETERRQ1(PetscObjectComm((PetscObject)viewer),PETSC_ERR_ARG_OUTOFRANGE,"Flow control count must be greater than 1, %D was set",fc);
2515c6c1daeSBarry Smith   vbinary->flowcontrol = fc;
2525c6c1daeSBarry Smith   PetscFunctionReturn(0);
2535c6c1daeSBarry Smith }
2545c6c1daeSBarry Smith 
2555c6c1daeSBarry Smith /*@C
256639ff905SBarry Smith     PetscViewerBinarySetFlowControl - Sets how many messages are allowed to outstanding at the same time during parallel IO reads/writes
2575c6c1daeSBarry Smith 
2585c6c1daeSBarry Smith     Not Collective
2595c6c1daeSBarry Smith 
2605c6c1daeSBarry Smith     Input Parameter:
2615c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
262639ff905SBarry Smith -   fc - the number of messages, defaults to 256 if this function was not called
2635c6c1daeSBarry Smith 
2645c6c1daeSBarry Smith     Level: advanced
2655c6c1daeSBarry Smith 
2665c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetFlowControl()
2675c6c1daeSBarry Smith 
2685c6c1daeSBarry Smith @*/
2695c6c1daeSBarry Smith PetscErrorCode  PetscViewerBinarySetFlowControl(PetscViewer viewer,PetscInt fc)
2705c6c1daeSBarry Smith {
2715c6c1daeSBarry Smith   PetscErrorCode ierr;
2725c6c1daeSBarry Smith 
2735c6c1daeSBarry Smith   PetscFunctionBegin;
2745c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetFlowControl_C",(PetscViewer,PetscInt),(viewer,fc));CHKERRQ(ierr);
2755c6c1daeSBarry Smith   PetscFunctionReturn(0);
2765c6c1daeSBarry Smith }
2775c6c1daeSBarry Smith 
2785c6c1daeSBarry Smith /*@C
2795c6c1daeSBarry Smith     PetscViewerBinaryGetDescriptor - Extracts the file descriptor from a PetscViewer.
2805c6c1daeSBarry Smith 
2815872f025SBarry Smith     Collective On PetscViewer
2825c6c1daeSBarry Smith 
2835c6c1daeSBarry Smith     Input Parameter:
2845c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
2855c6c1daeSBarry Smith 
2865c6c1daeSBarry Smith     Output Parameter:
2875c6c1daeSBarry Smith .   fdes - file descriptor
2885c6c1daeSBarry Smith 
2895c6c1daeSBarry Smith     Level: advanced
2905c6c1daeSBarry Smith 
2915c6c1daeSBarry Smith     Notes:
2925c6c1daeSBarry Smith       For writable binary PetscViewers, the descriptor will only be valid for the
2935c6c1daeSBarry Smith     first processor in the communicator that shares the PetscViewer. For readable
2945c6c1daeSBarry Smith     files it will only be valid on nodes that have the file. If node 0 does not
2955c6c1daeSBarry Smith     have the file it generates an error even if another node does have the file.
2965c6c1daeSBarry Smith 
2975c6c1daeSBarry Smith     Fortran Note:
2985c6c1daeSBarry Smith     This routine is not supported in Fortran.
2995c6c1daeSBarry Smith 
3005872f025SBarry Smith     Developer Notes: This must be called on all processes because Dave May changed
3015872f025SBarry Smith     the source code that this may be trigger a PetscViewerSetUp() call if it was not previously triggered.
3025872f025SBarry Smith 
3035872f025SBarry Smith 
3045c6c1daeSBarry Smith   Concepts: file descriptor^getting
3055c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
3065c6c1daeSBarry Smith 
3075c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer()
3085c6c1daeSBarry Smith @*/
3095c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetDescriptor(PetscViewer viewer,int *fdes)
3105c6c1daeSBarry Smith {
31103a1d59fSDave May   PetscErrorCode     ierr;
3125c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
3135c6c1daeSBarry Smith 
3145c6c1daeSBarry Smith   PetscFunctionBegin;
315c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
3165c6c1daeSBarry Smith   *fdes = vbinary->fdes;
3175c6c1daeSBarry Smith   PetscFunctionReturn(0);
3185c6c1daeSBarry Smith }
3195c6c1daeSBarry Smith 
3205c6c1daeSBarry Smith /*@
3215c6c1daeSBarry Smith     PetscViewerBinarySkipInfo - Binary file will not have .info file created with it
3225c6c1daeSBarry Smith 
3235c6c1daeSBarry Smith     Not Collective
3245c6c1daeSBarry Smith 
3255c6c1daeSBarry Smith     Input Paramter:
3265c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerCreate()
3275c6c1daeSBarry Smith 
3285c6c1daeSBarry Smith     Options Database Key:
3295c6c1daeSBarry Smith .   -viewer_binary_skip_info
3305c6c1daeSBarry Smith 
3315c6c1daeSBarry Smith     Level: advanced
3325c6c1daeSBarry Smith 
333a2d7db39SDave May     Notes: This must be called after PetscViewerSetType(). If you use PetscViewerBinaryOpen() then
3345c6c1daeSBarry Smith     you can only skip the info file with the -viewer_binary_skip_info flag. To use the function you must open the
335a2d7db39SDave May     viewer with PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinarySkipInfo().
3365c6c1daeSBarry Smith 
3375c6c1daeSBarry Smith     The .info contains meta information about the data in the binary file, for example the block size if it was
3385c6c1daeSBarry Smith     set for a vector or matrix.
3395c6c1daeSBarry Smith 
3405c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
3415c6c1daeSBarry Smith 
3425c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySetSkipOptions(),
343807ea322SDave May           PetscViewerBinaryGetSkipOptions(), PetscViewerBinaryGetSkipInfo()
3445c6c1daeSBarry Smith @*/
3455c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySkipInfo(PetscViewer viewer)
3465c6c1daeSBarry Smith {
3475c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
3485c6c1daeSBarry Smith 
3495c6c1daeSBarry Smith   PetscFunctionBegin;
3505c6c1daeSBarry Smith   vbinary->skipinfo = PETSC_TRUE;
3515c6c1daeSBarry Smith   PetscFunctionReturn(0);
3525c6c1daeSBarry Smith }
3535c6c1daeSBarry Smith 
35481f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipInfo_Binary(PetscViewer viewer,PetscBool skip)
355807ea322SDave May {
356807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
357807ea322SDave May 
358807ea322SDave May   PetscFunctionBegin;
359807ea322SDave May   vbinary->skipinfo = skip;
360807ea322SDave May   PetscFunctionReturn(0);
361807ea322SDave May }
362807ea322SDave May 
363807ea322SDave May /*@
364807ea322SDave May     PetscViewerBinarySetSkipInfo - Binary file will not have .info file created with it
365807ea322SDave May 
366807ea322SDave May     Not Collective
367807ea322SDave May 
368807ea322SDave May     Input Paramter:
369807ea322SDave May .   viewer - PetscViewer context, obtained from PetscViewerCreate()
370807ea322SDave May 
371807ea322SDave May     Options Database Key:
372807ea322SDave May .   -viewer_binary_skip_info
373807ea322SDave May 
374807ea322SDave May     Level: advanced
375807ea322SDave May 
376807ea322SDave May     Concepts: PetscViewerBinary^accessing info file
377807ea322SDave May 
378807ea322SDave May .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySetSkipOptions(),
379807ea322SDave May           PetscViewerBinaryGetSkipOptions(), PetscViewerBinaryGetSkipInfo()
380807ea322SDave May @*/
381807ea322SDave May PetscErrorCode PetscViewerBinarySetSkipInfo(PetscViewer viewer,PetscBool skip)
382807ea322SDave May {
383807ea322SDave May   PetscErrorCode ierr;
384807ea322SDave May 
385807ea322SDave May   PetscFunctionBegin;
386807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipInfo_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
387807ea322SDave May   PetscFunctionReturn(0);
388807ea322SDave May }
389807ea322SDave May 
39081f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipInfo_Binary(PetscViewer viewer,PetscBool *skip)
391807ea322SDave May {
392807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
393807ea322SDave May 
394807ea322SDave May   PetscFunctionBegin;
395807ea322SDave May   *skip  = vbinary->skipinfo;
396807ea322SDave May   PetscFunctionReturn(0);
397807ea322SDave May }
398807ea322SDave May 
399807ea322SDave May /*@
400807ea322SDave May     PetscViewerBinaryGetSkipInfo - check if viewer wrote a .info file
401807ea322SDave May 
402807ea322SDave May     Not Collective
403807ea322SDave May 
404807ea322SDave May     Input Parameter:
405807ea322SDave May .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
406807ea322SDave May 
407807ea322SDave May     Output Parameter:
408807ea322SDave May .   skip - PETSC_TRUE implies the .info file was not generated
409807ea322SDave May 
410807ea322SDave May     Level: advanced
411807ea322SDave May 
412807ea322SDave May     Notes: This must be called after PetscViewerSetType()
413807ea322SDave May 
414807ea322SDave May     Concepts: PetscViewerBinary^accessing info file
415807ea322SDave May 
416807ea322SDave May .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
417807ea322SDave May           PetscViewerBinarySetSkipOptions(), PetscViewerBinarySetSkipInfo()
418807ea322SDave May @*/
419807ea322SDave May PetscErrorCode PetscViewerBinaryGetSkipInfo(PetscViewer viewer,PetscBool *skip)
420807ea322SDave May {
421807ea322SDave May   PetscErrorCode ierr;
422807ea322SDave May 
423807ea322SDave May   PetscFunctionBegin;
424807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipInfo_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
425807ea322SDave May   PetscFunctionReturn(0);
426807ea322SDave May }
427807ea322SDave May 
42881f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipOptions_Binary(PetscViewer viewer,PetscBool skip)
429807ea322SDave May {
430807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
431807ea322SDave May 
432807ea322SDave May   PetscFunctionBegin;
433807ea322SDave May   vbinary->skipoptions = skip;
434807ea322SDave May   PetscFunctionReturn(0);
435807ea322SDave May }
436807ea322SDave May 
4375c6c1daeSBarry Smith /*@
4385c6c1daeSBarry Smith     PetscViewerBinarySetSkipOptions - do not use the PETSc options database when loading objects
4395c6c1daeSBarry Smith 
4405c6c1daeSBarry Smith     Not Collective
4415c6c1daeSBarry Smith 
4425c6c1daeSBarry Smith     Input Parameters:
4435c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
4445c6c1daeSBarry Smith -   skip - PETSC_TRUE means do not use
4455c6c1daeSBarry Smith 
4465c6c1daeSBarry Smith     Options Database Key:
4475c6c1daeSBarry Smith .   -viewer_binary_skip_options
4485c6c1daeSBarry Smith 
4495c6c1daeSBarry Smith     Level: advanced
4505c6c1daeSBarry Smith 
4515c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
4525c6c1daeSBarry Smith 
4535c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
4545c6c1daeSBarry Smith 
4555c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
4565c6c1daeSBarry Smith           PetscViewerBinaryGetSkipOptions()
4575c6c1daeSBarry Smith @*/
4585c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySetSkipOptions(PetscViewer viewer,PetscBool skip)
4595c6c1daeSBarry Smith {
460807ea322SDave May   PetscErrorCode ierr;
4615c6c1daeSBarry Smith 
4625c6c1daeSBarry Smith   PetscFunctionBegin;
463807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipOptions_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
464807ea322SDave May   PetscFunctionReturn(0);
465807ea322SDave May }
466807ea322SDave May 
46781f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipOptions_Binary(PetscViewer viewer,PetscBool *skip)
468807ea322SDave May {
469807ea322SDave May   PetscViewer_Binary *vbinary;
470807ea322SDave May 
471807ea322SDave May   PetscFunctionBegin;
472807ea322SDave May   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
473807ea322SDave May   vbinary = (PetscViewer_Binary*)viewer->data;
474807ea322SDave May   *skip   = vbinary->skipoptions;
4755c6c1daeSBarry Smith   PetscFunctionReturn(0);
4765c6c1daeSBarry Smith }
4775c6c1daeSBarry Smith 
4785c6c1daeSBarry Smith /*@
4795c6c1daeSBarry Smith     PetscViewerBinaryGetSkipOptions - checks if viewer uses the PETSc options database when loading objects
4805c6c1daeSBarry Smith 
4815c6c1daeSBarry Smith     Not Collective
4825c6c1daeSBarry Smith 
4835c6c1daeSBarry Smith     Input Parameter:
4845c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
4855c6c1daeSBarry Smith 
4865c6c1daeSBarry Smith     Output Parameter:
4875c6c1daeSBarry Smith .   skip - PETSC_TRUE means do not use
4885c6c1daeSBarry Smith 
4895c6c1daeSBarry Smith     Level: advanced
4905c6c1daeSBarry Smith 
4915c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
4925c6c1daeSBarry Smith 
4935c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
4945c6c1daeSBarry Smith 
4955c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
4965c6c1daeSBarry Smith           PetscViewerBinarySetSkipOptions()
4975c6c1daeSBarry Smith @*/
4985c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetSkipOptions(PetscViewer viewer,PetscBool *skip)
4995c6c1daeSBarry Smith {
500807ea322SDave May   PetscErrorCode ierr;
5015c6c1daeSBarry Smith 
5025c6c1daeSBarry Smith   PetscFunctionBegin;
503807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipOptions_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
5045c6c1daeSBarry Smith   PetscFunctionReturn(0);
5055c6c1daeSBarry Smith }
5065c6c1daeSBarry Smith 
50781f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipHeader_Binary(PetscViewer viewer,PetscBool skip)
5085c6c1daeSBarry Smith {
5095c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
5105c6c1daeSBarry Smith 
5115c6c1daeSBarry Smith   PetscFunctionBegin;
5125c6c1daeSBarry Smith   vbinary->skipheader = skip;
5135c6c1daeSBarry Smith   PetscFunctionReturn(0);
5145c6c1daeSBarry Smith }
5155c6c1daeSBarry Smith 
5165c6c1daeSBarry Smith /*@
5175c6c1daeSBarry Smith     PetscViewerBinarySetSkipHeader - do not write a header with size information on output, just raw data
5185c6c1daeSBarry Smith 
5195c6c1daeSBarry Smith     Not Collective
5205c6c1daeSBarry Smith 
5215c6c1daeSBarry Smith     Input Parameters:
5225c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
5235c6c1daeSBarry Smith -   skip - PETSC_TRUE means do not write header
5245c6c1daeSBarry Smith 
5255c6c1daeSBarry Smith     Options Database Key:
5265c6c1daeSBarry Smith .   -viewer_binary_skip_header
5275c6c1daeSBarry Smith 
5285c6c1daeSBarry Smith     Level: advanced
5295c6c1daeSBarry Smith 
5305c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
5315c6c1daeSBarry Smith 
5325c6c1daeSBarry Smith            Can ONLY be called on a binary viewer
5335c6c1daeSBarry Smith 
5345c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
5355c6c1daeSBarry Smith           PetscViewerBinaryGetSkipHeader()
5365c6c1daeSBarry Smith @*/
5375c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySetSkipHeader(PetscViewer viewer,PetscBool skip)
5385c6c1daeSBarry Smith {
5395c6c1daeSBarry Smith   PetscErrorCode ierr;
5405c6c1daeSBarry Smith 
5415c6c1daeSBarry Smith   PetscFunctionBegin;
5425c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipHeader_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
5435c6c1daeSBarry Smith   PetscFunctionReturn(0);
5445c6c1daeSBarry Smith }
5455c6c1daeSBarry Smith 
54681f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipHeader_Binary(PetscViewer viewer,PetscBool  *skip)
5475c6c1daeSBarry Smith {
5485c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
5495c6c1daeSBarry Smith 
5505c6c1daeSBarry Smith   PetscFunctionBegin;
5515c6c1daeSBarry Smith   *skip = vbinary->skipheader;
5525c6c1daeSBarry Smith   PetscFunctionReturn(0);
5535c6c1daeSBarry Smith }
5545c6c1daeSBarry Smith 
5555c6c1daeSBarry Smith /*@
5565c6c1daeSBarry Smith     PetscViewerBinaryGetSkipHeader - checks whether to write a header with size information on output, or just raw data
5575c6c1daeSBarry Smith 
5585c6c1daeSBarry Smith     Not Collective
5595c6c1daeSBarry Smith 
5605c6c1daeSBarry Smith     Input Parameter:
5615c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
5625c6c1daeSBarry Smith 
5635c6c1daeSBarry Smith     Output Parameter:
5645c6c1daeSBarry Smith .   skip - PETSC_TRUE means do not write header
5655c6c1daeSBarry Smith 
5665c6c1daeSBarry Smith     Level: advanced
5675c6c1daeSBarry Smith 
5685c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
5695c6c1daeSBarry Smith 
5705c6c1daeSBarry Smith             Returns false for PETSCSOCKETVIEWER, you cannot skip the header for it.
5715c6c1daeSBarry Smith 
5725c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
5735c6c1daeSBarry Smith           PetscViewerBinarySetSkipHeader()
5745c6c1daeSBarry Smith @*/
5755c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetSkipHeader(PetscViewer viewer,PetscBool  *skip)
5765c6c1daeSBarry Smith {
5775c6c1daeSBarry Smith   PetscErrorCode ierr;
5785c6c1daeSBarry Smith 
5795c6c1daeSBarry Smith   PetscFunctionBegin;
5805c6c1daeSBarry Smith   *skip = PETSC_FALSE;
581163d334eSBarry Smith   ierr  = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipHeader_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
5825c6c1daeSBarry Smith   PetscFunctionReturn(0);
5835c6c1daeSBarry Smith }
5845c6c1daeSBarry Smith 
58581f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetInfoPointer_Binary(PetscViewer viewer,FILE **file)
5865c6c1daeSBarry Smith {
587f3b211e4SSatish Balay   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
588a261c58fSBarry Smith   PetscErrorCode     ierr;
589a261c58fSBarry Smith   MPI_Comm           comm;
5905c6c1daeSBarry Smith 
5915c6c1daeSBarry Smith   PetscFunctionBegin;
592c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
5935c6c1daeSBarry Smith   *file = vbinary->fdes_info;
594a261c58fSBarry Smith   if (viewer->format == PETSC_VIEWER_BINARY_MATLAB && !vbinary->matlabheaderwritten) {
595a261c58fSBarry Smith     vbinary->matlabheaderwritten = PETSC_TRUE;
596a261c58fSBarry Smith     ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
597da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#--- begin code written by PetscViewerBinary for MATLAB format ---#\n");CHKERRQ(ierr);
598da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#$$ Set.filename = '%s';\n",vbinary->filename);CHKERRQ(ierr);
599da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#$$ fd = PetscOpenFile(Set.filename);\n");CHKERRQ(ierr);
600da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#--- end code written by PetscViewerBinary for MATLAB format ---#\n\n");CHKERRQ(ierr);
601a261c58fSBarry Smith   }
6025c6c1daeSBarry Smith   PetscFunctionReturn(0);
6035c6c1daeSBarry Smith }
6045c6c1daeSBarry Smith 
6055c6c1daeSBarry Smith /*@C
6065c6c1daeSBarry Smith     PetscViewerBinaryGetInfoPointer - Extracts the file pointer for the ASCII
6075c6c1daeSBarry Smith           info file associated with a binary file.
6085c6c1daeSBarry Smith 
6095c6c1daeSBarry Smith     Not Collective
6105c6c1daeSBarry Smith 
6115c6c1daeSBarry Smith     Input Parameter:
6125c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
6135c6c1daeSBarry Smith 
6145c6c1daeSBarry Smith     Output Parameter:
6150298fd71SBarry Smith .   file - file pointer  Always returns NULL if not a binary viewer
6165c6c1daeSBarry Smith 
6175c6c1daeSBarry Smith     Level: advanced
6185c6c1daeSBarry Smith 
6195c6c1daeSBarry Smith     Notes:
6205c6c1daeSBarry Smith       For writable binary PetscViewers, the descriptor will only be valid for the
6215c6c1daeSBarry Smith     first processor in the communicator that shares the PetscViewer.
6225c6c1daeSBarry Smith 
6235c6c1daeSBarry Smith     Fortran Note:
6245c6c1daeSBarry Smith     This routine is not supported in Fortran.
6255c6c1daeSBarry Smith 
6265c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing info file
6275c6c1daeSBarry Smith 
6285c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetDescriptor()
6295c6c1daeSBarry Smith @*/
6305c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetInfoPointer(PetscViewer viewer,FILE **file)
6315c6c1daeSBarry Smith {
6325c6c1daeSBarry Smith   PetscErrorCode ierr;
6335c6c1daeSBarry Smith 
6345c6c1daeSBarry Smith   PetscFunctionBegin;
6350298fd71SBarry Smith   *file = NULL;
6365c6c1daeSBarry Smith   ierr  = PetscTryMethod(viewer,"PetscViewerBinaryGetInfoPointer_C",(PetscViewer,FILE **),(viewer,file));CHKERRQ(ierr);
6375c6c1daeSBarry Smith   PetscFunctionReturn(0);
6385c6c1daeSBarry Smith }
6395c6c1daeSBarry Smith 
6405c6c1daeSBarry Smith static PetscErrorCode PetscViewerFileClose_Binary(PetscViewer v)
6415c6c1daeSBarry Smith {
6425c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
6435c6c1daeSBarry Smith   PetscErrorCode     ierr;
6445c6c1daeSBarry Smith   PetscMPIInt        rank;
6455c6c1daeSBarry Smith   int                err;
6465c6c1daeSBarry Smith 
6475c6c1daeSBarry Smith   PetscFunctionBegin;
648ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)v),&rank);CHKERRQ(ierr);
6495c6c1daeSBarry Smith   if ((!rank || vbinary->btype == FILE_MODE_READ) && vbinary->fdes) {
6505c6c1daeSBarry Smith     close(vbinary->fdes);
6515c6c1daeSBarry Smith     if (!rank && vbinary->storecompressed) {
6525c6c1daeSBarry Smith       char par[PETSC_MAX_PATH_LEN],buf[PETSC_MAX_PATH_LEN];
6535c6c1daeSBarry Smith       FILE *fp;
6545c6c1daeSBarry Smith       /* compress the file */
6555c6c1daeSBarry Smith       ierr = PetscStrcpy(par,"gzip -f ");CHKERRQ(ierr);
656f90597f1SStefano Zampini       ierr = PetscStrcat(par,vbinary->ogzfilename ? vbinary->ogzfilename : vbinary->filename);CHKERRQ(ierr);
657f90597f1SStefano Zampini       ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
6585c6c1daeSBarry Smith #if defined(PETSC_HAVE_POPEN)
6590298fd71SBarry Smith       ierr = PetscPOpen(PETSC_COMM_SELF,NULL,par,"r",&fp);CHKERRQ(ierr);
6605c6c1daeSBarry Smith       if (fgets(buf,1024,fp)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error from command %s\n%s",par,buf);
6610298fd71SBarry Smith       ierr = PetscPClose(PETSC_COMM_SELF,fp,NULL);CHKERRQ(ierr);
6625c6c1daeSBarry Smith #else
6635c6c1daeSBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
6645c6c1daeSBarry Smith #endif
6655c6c1daeSBarry Smith     }
6665c6c1daeSBarry Smith   }
6675c6c1daeSBarry Smith   if (vbinary->fdes_info) {
6685c6c1daeSBarry Smith     err = fclose(vbinary->fdes_info);
6695c6c1daeSBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
6705c6c1daeSBarry Smith   }
6715c6c1daeSBarry Smith   PetscFunctionReturn(0);
6725c6c1daeSBarry Smith }
6735c6c1daeSBarry Smith 
674e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
675e0385b85SDave May static PetscErrorCode PetscViewerFileClose_BinaryMPIIO(PetscViewer v)
676e0385b85SDave May {
677e0385b85SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
678e0385b85SDave May   int                err;
679e0385b85SDave May   PetscErrorCode     ierr;
680e0385b85SDave May 
681e0385b85SDave May   PetscFunctionBegin;
682e0385b85SDave May   if (vbinary->mfdes) {
683e0385b85SDave May     ierr = MPI_File_close(&vbinary->mfdes);CHKERRQ(ierr);
684e0385b85SDave May   }
685e0385b85SDave May   if (vbinary->fdes_info) {
686e0385b85SDave May     err = fclose(vbinary->fdes_info);
687e0385b85SDave May     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
688e0385b85SDave May   }
689e0385b85SDave May   PetscFunctionReturn(0);
690e0385b85SDave May }
691e0385b85SDave May #endif
692e0385b85SDave May 
69381f0254dSBarry Smith static PetscErrorCode PetscViewerDestroy_Binary(PetscViewer v)
6945c6c1daeSBarry Smith {
6955c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
6965c6c1daeSBarry Smith   PetscErrorCode     ierr;
6975c6c1daeSBarry Smith 
6985c6c1daeSBarry Smith   PetscFunctionBegin;
699a261c58fSBarry Smith   if (v->format == PETSC_VIEWER_BINARY_MATLAB) {
700a261c58fSBarry Smith     MPI_Comm comm;
701a261c58fSBarry Smith     FILE     *info;
702a261c58fSBarry Smith 
703a261c58fSBarry Smith     ierr = PetscObjectGetComm((PetscObject)v,&comm);CHKERRQ(ierr);
704a261c58fSBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(v,&info);CHKERRQ(ierr);
705da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#--- begin code written by PetscViewerBinary for MATLAB format ---#\n");CHKERRQ(ierr);
706da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#$$ close(fd);\n");CHKERRQ(ierr);
707da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#--- end code written by PetscViewerBinary for MATLAB format ---#\n\n");CHKERRQ(ierr);
708a261c58fSBarry Smith   }
7095c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
710bc196f7cSDave May   if (vbinary->usempiio) {
711e0385b85SDave May     ierr = PetscViewerFileClose_BinaryMPIIO(v);CHKERRQ(ierr);
712e0385b85SDave May   } else {
713e0385b85SDave May #endif
714e0385b85SDave May     ierr = PetscViewerFileClose_Binary(v);CHKERRQ(ierr);
715e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
7165c6c1daeSBarry Smith   }
7175c6c1daeSBarry Smith #endif
718f90597f1SStefano Zampini   ierr = PetscFree(vbinary->filename);CHKERRQ(ierr);
719f90597f1SStefano Zampini   ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
720e0385b85SDave May   ierr = PetscFree(vbinary);CHKERRQ(ierr);
721e0385b85SDave May   PetscFunctionReturn(0);
722e0385b85SDave May }
7235c6c1daeSBarry Smith 
7245c6c1daeSBarry Smith /*@C
7255c6c1daeSBarry Smith    PetscViewerBinaryOpen - Opens a file for binary input/output.
7265c6c1daeSBarry Smith 
7275c6c1daeSBarry Smith    Collective on MPI_Comm
7285c6c1daeSBarry Smith 
7295c6c1daeSBarry Smith    Input Parameters:
7305c6c1daeSBarry Smith +  comm - MPI communicator
7315c6c1daeSBarry Smith .  name - name of file
7325c6c1daeSBarry Smith -  type - type of file
7335c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
7345c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
7355c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
7365c6c1daeSBarry Smith 
7375c6c1daeSBarry Smith    Output Parameter:
7385c6c1daeSBarry Smith .  binv - PetscViewer for binary input/output to use with the specified file
7395c6c1daeSBarry Smith 
7405c6c1daeSBarry Smith     Options Database Keys:
741a2d7db39SDave May +    -viewer_binary_filename <name>
742a2d7db39SDave May .    -viewer_binary_skip_info
7435c6c1daeSBarry Smith .    -viewer_binary_skip_options
744a2d7db39SDave May .    -viewer_binary_skip_header
745a2d7db39SDave May -    -viewer_binary_mpiio
7465c6c1daeSBarry Smith 
7475c6c1daeSBarry Smith    Level: beginner
7485c6c1daeSBarry Smith 
7495c6c1daeSBarry Smith    Note:
7505c6c1daeSBarry Smith    This PetscViewer should be destroyed with PetscViewerDestroy().
7515c6c1daeSBarry Smith 
7525c6c1daeSBarry Smith     For reading files, the filename may begin with ftp:// or http:// and/or
7535c6c1daeSBarry Smith     end with .gz; in this case file is brought over and uncompressed.
7545c6c1daeSBarry Smith 
7555c6c1daeSBarry Smith     For creating files, if the file name ends with .gz it is automatically
7565c6c1daeSBarry Smith     compressed when closed.
7575c6c1daeSBarry Smith 
7585c6c1daeSBarry Smith     For writing files it only opens the file on processor 0 in the communicator.
7595c6c1daeSBarry Smith     For readable files it opens the file on all nodes that have the file. If
7605c6c1daeSBarry Smith     node 0 does not have the file it generates an error even if other nodes
7615c6c1daeSBarry Smith     do have the file.
7625c6c1daeSBarry Smith 
7635c6c1daeSBarry Smith    Concepts: binary files
7645c6c1daeSBarry Smith    Concepts: PetscViewerBinary^creating
7655c6c1daeSBarry Smith    Concepts: gzip
7665c6c1daeSBarry Smith    Concepts: accessing remote file
7675c6c1daeSBarry Smith    Concepts: remote file
7685c6c1daeSBarry Smith 
7696a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
7705c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
77167918a83SBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscViewerBinaryRead(), PetscViewerBinarySetUseMPIIO(),
77267918a83SBarry Smith           PetscViewerBinaryGetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
7735c6c1daeSBarry Smith @*/
7745c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryOpen(MPI_Comm comm,const char name[],PetscFileMode type,PetscViewer *binv)
7755c6c1daeSBarry Smith {
7765c6c1daeSBarry Smith   PetscErrorCode ierr;
7775c6c1daeSBarry Smith 
7785c6c1daeSBarry Smith   PetscFunctionBegin;
7795c6c1daeSBarry Smith   ierr = PetscViewerCreate(comm,binv);CHKERRQ(ierr);
7805c6c1daeSBarry Smith   ierr = PetscViewerSetType(*binv,PETSCVIEWERBINARY);CHKERRQ(ierr);
7815c6c1daeSBarry Smith   ierr = PetscViewerFileSetMode(*binv,type);CHKERRQ(ierr);
7825c6c1daeSBarry Smith   ierr = PetscViewerFileSetName(*binv,name);CHKERRQ(ierr);
78303a1d59fSDave May   ierr = PetscViewerSetFromOptions(*binv);CHKERRQ(ierr);
7845c6c1daeSBarry Smith   PetscFunctionReturn(0);
7855c6c1daeSBarry Smith }
7865c6c1daeSBarry Smith 
7875c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
788060da220SMatthew G. Knepley static PetscErrorCode PetscViewerBinaryWriteReadMPIIO(PetscViewer viewer,void *data,PetscInt num,PetscInt *count,PetscDataType dtype,PetscBool write)
7895c6c1daeSBarry Smith {
7905c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
7915c6c1daeSBarry Smith   PetscErrorCode     ierr;
7925c6c1daeSBarry Smith   MPI_Datatype       mdtype;
793e597bc1dSJed Brown   PetscMPIInt        cnt;
7945c6c1daeSBarry Smith   MPI_Status         status;
7955c6c1daeSBarry Smith   MPI_Aint           ul,dsize;
7965c6c1daeSBarry Smith 
7975c6c1daeSBarry Smith   PetscFunctionBegin;
798060da220SMatthew G. Knepley   ierr = PetscMPIIntCast(num,&cnt);CHKERRQ(ierr);
7995c6c1daeSBarry Smith   ierr = PetscDataTypeToMPIDataType(dtype,&mdtype);CHKERRQ(ierr);
8005c6c1daeSBarry Smith   ierr = MPI_File_set_view(vbinary->mfdes,vbinary->moff,mdtype,mdtype,(char*)"native",MPI_INFO_NULL);CHKERRQ(ierr);
8015c6c1daeSBarry Smith   if (write) {
8025c6c1daeSBarry Smith     ierr = MPIU_File_write_all(vbinary->mfdes,data,cnt,mdtype,&status);CHKERRQ(ierr);
8035c6c1daeSBarry Smith   } else {
8045c6c1daeSBarry Smith     ierr = MPIU_File_read_all(vbinary->mfdes,data,cnt,mdtype,&status);CHKERRQ(ierr);
8055c6c1daeSBarry Smith   }
8065c6c1daeSBarry Smith   ierr = MPI_Type_get_extent(mdtype,&ul,&dsize);CHKERRQ(ierr);
807a297a907SKarl Rupp 
8085c6c1daeSBarry Smith   vbinary->moff += dsize*cnt;
809060da220SMatthew G. Knepley   if (count) *count = num;
8105c6c1daeSBarry Smith   PetscFunctionReturn(0);
8115c6c1daeSBarry Smith }
8125c6c1daeSBarry Smith #endif
8135c6c1daeSBarry Smith 
8145c6c1daeSBarry Smith /*@C
8155c6c1daeSBarry Smith    PetscViewerBinaryRead - Reads from a binary file, all processors get the same result
8165c6c1daeSBarry Smith 
8175c6c1daeSBarry Smith    Collective on MPI_Comm
8185c6c1daeSBarry Smith 
8195c6c1daeSBarry Smith    Input Parameters:
8205c6c1daeSBarry Smith +  viewer - the binary viewer
821907376e6SBarry Smith .  data - location of the data to be written
822060da220SMatthew G. Knepley .  num - number of items of data to read
823907376e6SBarry Smith -  dtype - type of data to read
8245c6c1daeSBarry Smith 
825f8e4bde8SMatthew G. Knepley    Output Parameters:
8260780a867SBarry Smith .  count - number of items of data actually read, or NULL. Unless an error is generated this is always set to the input parameter num.
827f8e4bde8SMatthew G. Knepley 
8285c6c1daeSBarry Smith    Level: beginner
8295c6c1daeSBarry Smith 
8305c6c1daeSBarry Smith    Concepts: binary files
8315c6c1daeSBarry Smith 
8320780a867SBarry Smith    Developer Note: Since count is always set to num it is not clear what purpose the output argument count serves.
8330780a867SBarry Smith 
8346a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
8355c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
8365c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
8375c6c1daeSBarry Smith @*/
838060da220SMatthew G. Knepley PetscErrorCode PetscViewerBinaryRead(PetscViewer viewer,void *data,PetscInt num,PetscInt *count,PetscDataType dtype)
8395c6c1daeSBarry Smith {
8405c6c1daeSBarry Smith   PetscErrorCode     ierr;
8415c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
8425c6c1daeSBarry Smith 
843c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
8445c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
845bc196f7cSDave May   if (vbinary->usempiio) {
846060da220SMatthew G. Knepley     ierr = PetscViewerBinaryWriteReadMPIIO(viewer,data,num,count,dtype,PETSC_FALSE);CHKERRQ(ierr);
8475c6c1daeSBarry Smith   } else {
8485c6c1daeSBarry Smith #endif
849060da220SMatthew G. Knepley     ierr = PetscBinarySynchronizedRead(PetscObjectComm((PetscObject)viewer),vbinary->fdes,data,num,dtype);CHKERRQ(ierr);
8500780a867SBarry Smith     if (count) *count = num;
8515c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
8525c6c1daeSBarry Smith   }
8535c6c1daeSBarry Smith #endif
8545c6c1daeSBarry Smith   PetscFunctionReturn(0);
8555c6c1daeSBarry Smith }
8565c6c1daeSBarry Smith 
8575c6c1daeSBarry Smith /*@C
8585c6c1daeSBarry Smith    PetscViewerBinaryWrite - writes to a binary file, only from the first process
8595c6c1daeSBarry Smith 
8605c6c1daeSBarry Smith    Collective on MPI_Comm
8615c6c1daeSBarry Smith 
8625c6c1daeSBarry Smith    Input Parameters:
8635c6c1daeSBarry Smith +  viewer - the binary viewer
8645c6c1daeSBarry Smith .  data - location of data
8655824c9d0SPatrick Sanan .  count - number of items of data to write
8665824c9d0SPatrick Sanan .  dtype - type of data to write
8675c6c1daeSBarry Smith -  istemp - data may be overwritten
8685c6c1daeSBarry Smith 
8695c6c1daeSBarry Smith    Level: beginner
8705c6c1daeSBarry Smith 
8715c6c1daeSBarry Smith    Notes: because byte-swapping may be done on the values in data it cannot be declared const
8725c6c1daeSBarry Smith 
8735c6c1daeSBarry Smith    Concepts: binary files
8745c6c1daeSBarry Smith 
8756a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
8765c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(), PetscDataType
8775c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
8785c6c1daeSBarry Smith @*/
8795c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryWrite(PetscViewer viewer,void *data,PetscInt count,PetscDataType dtype,PetscBool istemp)
8805c6c1daeSBarry Smith {
8815c6c1daeSBarry Smith   PetscErrorCode     ierr;
8825c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
8835c6c1daeSBarry Smith 
8845c6c1daeSBarry Smith   PetscFunctionBegin;
885c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
8865c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
887bc196f7cSDave May   if (vbinary->usempiio) {
888060da220SMatthew G. Knepley     ierr = PetscViewerBinaryWriteReadMPIIO(viewer,data,count,NULL,dtype,PETSC_TRUE);CHKERRQ(ierr);
8895c6c1daeSBarry Smith   } else {
8905c6c1daeSBarry Smith #endif
891ce94432eSBarry Smith     ierr = PetscBinarySynchronizedWrite(PetscObjectComm((PetscObject)viewer),vbinary->fdes,data,count,dtype,istemp);CHKERRQ(ierr);
8925c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
8935c6c1daeSBarry Smith   }
8945c6c1daeSBarry Smith #endif
8955c6c1daeSBarry Smith   PetscFunctionReturn(0);
8965c6c1daeSBarry Smith }
8975c6c1daeSBarry Smith 
8985c6c1daeSBarry Smith /*@C
8995c6c1daeSBarry Smith    PetscViewerBinaryWriteStringArray - writes to a binary file, only from the first process an array of strings
9005c6c1daeSBarry Smith 
9015c6c1daeSBarry Smith    Collective on MPI_Comm
9025c6c1daeSBarry Smith 
9035c6c1daeSBarry Smith    Input Parameters:
9045c6c1daeSBarry Smith +  viewer - the binary viewer
9055c6c1daeSBarry Smith -  data - location of the array of strings
9065c6c1daeSBarry Smith 
9075c6c1daeSBarry Smith 
9085c6c1daeSBarry Smith    Level: intermediate
9095c6c1daeSBarry Smith 
9105c6c1daeSBarry Smith    Concepts: binary files
9115c6c1daeSBarry Smith 
9125c6c1daeSBarry Smith     Notes: array of strings is null terminated
9135c6c1daeSBarry Smith 
9146a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
9155c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
9165c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
9175c6c1daeSBarry Smith @*/
91878fbdcc8SBarry Smith PetscErrorCode PetscViewerBinaryWriteStringArray(PetscViewer viewer,const char * const *data)
9195c6c1daeSBarry Smith {
9205c6c1daeSBarry Smith   PetscErrorCode ierr;
9215c6c1daeSBarry Smith   PetscInt       i,n = 0,*sizes;
9225c6c1daeSBarry Smith 
923c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
9245c6c1daeSBarry Smith   /* count number of strings */
9255c6c1daeSBarry Smith   while (data[n++]) ;
9265c6c1daeSBarry Smith   n--;
927854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&sizes);CHKERRQ(ierr);
9285c6c1daeSBarry Smith   sizes[0] = n;
9295c6c1daeSBarry Smith   for (i=0; i<n; i++) {
9305c6c1daeSBarry Smith     size_t tmp;
9315c6c1daeSBarry Smith     ierr       = PetscStrlen(data[i],&tmp);CHKERRQ(ierr);
9325c6c1daeSBarry Smith     sizes[i+1] = tmp + 1;   /* size includes space for the null terminator */
9335c6c1daeSBarry Smith   }
9345c6c1daeSBarry Smith   ierr = PetscViewerBinaryWrite(viewer,sizes,n+1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
9355c6c1daeSBarry Smith   for (i=0; i<n; i++) {
93678fbdcc8SBarry Smith     ierr = PetscViewerBinaryWrite(viewer,(void*)data[i],sizes[i+1],PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr);
9375c6c1daeSBarry Smith   }
9385c6c1daeSBarry Smith   ierr = PetscFree(sizes);CHKERRQ(ierr);
9395c6c1daeSBarry Smith   PetscFunctionReturn(0);
9405c6c1daeSBarry Smith }
9415c6c1daeSBarry Smith 
9425c6c1daeSBarry Smith /*@C
9435c6c1daeSBarry Smith    PetscViewerBinaryReadStringArray - reads a binary file an array of strings
9445c6c1daeSBarry Smith 
9455c6c1daeSBarry Smith    Collective on MPI_Comm
9465c6c1daeSBarry Smith 
9475c6c1daeSBarry Smith    Input Parameter:
9485c6c1daeSBarry Smith .  viewer - the binary viewer
9495c6c1daeSBarry Smith 
9505c6c1daeSBarry Smith    Output Parameter:
9515c6c1daeSBarry Smith .  data - location of the array of strings
9525c6c1daeSBarry Smith 
9535c6c1daeSBarry Smith    Level: intermediate
9545c6c1daeSBarry Smith 
9555c6c1daeSBarry Smith    Concepts: binary files
9565c6c1daeSBarry Smith 
9575c6c1daeSBarry Smith     Notes: array of strings is null terminated
9585c6c1daeSBarry Smith 
9596a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
9605c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
9615c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
9625c6c1daeSBarry Smith @*/
9635c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryReadStringArray(PetscViewer viewer,char ***data)
9645c6c1daeSBarry Smith {
9655c6c1daeSBarry Smith   PetscErrorCode ierr;
966060da220SMatthew G. Knepley   PetscInt       i,n,*sizes,N = 0;
9675c6c1daeSBarry Smith 
968c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
9695c6c1daeSBarry Smith   /* count number of strings */
970060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,&n,1,NULL,PETSC_INT);CHKERRQ(ierr);
971785e854fSJed Brown   ierr = PetscMalloc1(n,&sizes);CHKERRQ(ierr);
972060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,sizes,n,NULL,PETSC_INT);CHKERRQ(ierr);
973a297a907SKarl Rupp   for (i=0; i<n; i++) N += sizes[i];
974854ce69bSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(char*) + N*sizeof(char),data);CHKERRQ(ierr);
9755c6c1daeSBarry Smith   (*data)[0] = (char*)((*data) + n + 1);
976a297a907SKarl Rupp   for (i=1; i<n; i++) (*data)[i] = (*data)[i-1] + sizes[i-1];
977060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,(*data)[0],N,NULL,PETSC_CHAR);CHKERRQ(ierr);
978a297a907SKarl Rupp 
9795c6c1daeSBarry Smith   (*data)[n] = 0;
980a297a907SKarl Rupp 
9815c6c1daeSBarry Smith   ierr = PetscFree(sizes);CHKERRQ(ierr);
9825c6c1daeSBarry Smith   PetscFunctionReturn(0);
9835c6c1daeSBarry Smith }
9845c6c1daeSBarry Smith 
98581f0254dSBarry Smith static PetscErrorCode PetscViewerFileGetName_Binary(PetscViewer viewer,const char **name)
9865c6c1daeSBarry Smith {
9875c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
9885c6c1daeSBarry Smith 
9895c6c1daeSBarry Smith   PetscFunctionBegin;
9905c6c1daeSBarry Smith   *name = vbinary->filename;
9915c6c1daeSBarry Smith   PetscFunctionReturn(0);
9925c6c1daeSBarry Smith }
9935c6c1daeSBarry Smith 
9945c6c1daeSBarry Smith /*@C
9955c6c1daeSBarry Smith      PetscViewerFileGetMode - Gets the type of file to be open
9965c6c1daeSBarry Smith 
9975c6c1daeSBarry Smith     Not Collective
9985c6c1daeSBarry Smith 
9995c6c1daeSBarry Smith   Input Parameter:
1000232ac771SBarry Smith .  viewer - the PetscViewer; must be a PETSCVIEWERBINARY, PETSCVIEWERMATLAB, PETSCVIEWERHDF5, or PETSCVIEWERASCII  PetscViewer
10015c6c1daeSBarry Smith 
10025c6c1daeSBarry Smith   Output Parameter:
10035c6c1daeSBarry Smith .  type - type of file
10045c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
10055c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
10065c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
10075c6c1daeSBarry Smith 
10085c6c1daeSBarry Smith   Level: advanced
10095c6c1daeSBarry Smith 
10105c6c1daeSBarry Smith .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen()
10115c6c1daeSBarry Smith 
10125c6c1daeSBarry Smith @*/
10135c6c1daeSBarry Smith PetscErrorCode PetscViewerFileGetMode(PetscViewer viewer,PetscFileMode *type)
10145c6c1daeSBarry Smith {
10155c6c1daeSBarry Smith   PetscErrorCode ierr;
10165c6c1daeSBarry Smith 
10175c6c1daeSBarry Smith   PetscFunctionBegin;
10185c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
10195c6c1daeSBarry Smith   PetscValidPointer(type,2);
10205c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerFileGetMode_C",(PetscViewer,PetscFileMode*),(viewer,type));CHKERRQ(ierr);
10215c6c1daeSBarry Smith   PetscFunctionReturn(0);
10225c6c1daeSBarry Smith }
10235c6c1daeSBarry Smith 
10245c6c1daeSBarry Smith /*@
1025bc196f7cSDave May     PetscViewerBinarySetUseMPIIO - Sets a binary viewer to use MPI-IO for reading/writing. Must be called
10265c6c1daeSBarry Smith         before PetscViewerFileSetName()
10275c6c1daeSBarry Smith 
10285c6c1daeSBarry Smith     Logically Collective on PetscViewer
10295c6c1daeSBarry Smith 
10305c6c1daeSBarry Smith     Input Parameters:
1031bc196f7cSDave May +   viewer - the PetscViewer; must be a binary
1032bc196f7cSDave May -   flg - PETSC_TRUE means MPI-IO will be used
10335c6c1daeSBarry Smith 
10345c6c1daeSBarry Smith     Options Database:
10355c6c1daeSBarry Smith     -viewer_binary_mpiio : Flag for using MPI-IO
10365c6c1daeSBarry Smith 
10375c6c1daeSBarry Smith     Level: advanced
10385c6c1daeSBarry Smith 
1039bc196f7cSDave May .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen(),
1040bc196f7cSDave May           PetscViewerBinaryGetUseMPIIO()
10415c6c1daeSBarry Smith 
10425c6c1daeSBarry Smith @*/
1043bc196f7cSDave May PetscErrorCode PetscViewerBinarySetUseMPIIO(PetscViewer viewer,PetscBool flg)
10445c6c1daeSBarry Smith {
10455c6c1daeSBarry Smith   PetscErrorCode ierr;
10465c6c1daeSBarry Smith 
10475c6c1daeSBarry Smith   PetscFunctionBegin;
10485c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
1049bc196f7cSDave May   ierr = PetscTryMethod(viewer,"PetscViewerBinarySetUseMPIIO_C",(PetscViewer,PetscBool),(viewer,flg));CHKERRQ(ierr);
10505c6c1daeSBarry Smith   PetscFunctionReturn(0);
10515c6c1daeSBarry Smith }
10525c6c1daeSBarry Smith 
10535c6c1daeSBarry Smith /*@C
10545c6c1daeSBarry Smith      PetscViewerFileSetMode - Sets the type of file to be open
10555c6c1daeSBarry Smith 
10565c6c1daeSBarry Smith     Logically Collective on PetscViewer
10575c6c1daeSBarry Smith 
10585c6c1daeSBarry Smith   Input Parameters:
1059232ac771SBarry Smith +  viewer - the PetscViewer; must be a a PETSCVIEWERBINARY, PETSCVIEWERMATLAB, PETSCVIEWERHDF5, or PETSCVIEWERASCII  PetscViewer
10605c6c1daeSBarry Smith -  type - type of file
10615c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
10625c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
10635c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
10645c6c1daeSBarry Smith 
10655c6c1daeSBarry Smith   Level: advanced
10665c6c1daeSBarry Smith 
10675c6c1daeSBarry Smith .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen()
10685c6c1daeSBarry Smith 
10695c6c1daeSBarry Smith @*/
10705c6c1daeSBarry Smith PetscErrorCode PetscViewerFileSetMode(PetscViewer viewer,PetscFileMode type)
10715c6c1daeSBarry Smith {
10725c6c1daeSBarry Smith   PetscErrorCode ierr;
10735c6c1daeSBarry Smith 
10745c6c1daeSBarry Smith   PetscFunctionBegin;
10755c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
10765c6c1daeSBarry Smith   PetscValidLogicalCollectiveEnum(viewer,type,2);
10775c6c1daeSBarry Smith   ierr = PetscTryMethod(viewer,"PetscViewerFileSetMode_C",(PetscViewer,PetscFileMode),(viewer,type));CHKERRQ(ierr);
10785c6c1daeSBarry Smith   PetscFunctionReturn(0);
10795c6c1daeSBarry Smith }
10805c6c1daeSBarry Smith 
108181f0254dSBarry Smith static PetscErrorCode PetscViewerFileGetMode_Binary(PetscViewer viewer,PetscFileMode *type)
10825c6c1daeSBarry Smith {
10835c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
10845c6c1daeSBarry Smith 
10855c6c1daeSBarry Smith   PetscFunctionBegin;
10865c6c1daeSBarry Smith   *type = vbinary->btype;
10875c6c1daeSBarry Smith   PetscFunctionReturn(0);
10885c6c1daeSBarry Smith }
10895c6c1daeSBarry Smith 
109081f0254dSBarry Smith static PetscErrorCode PetscViewerFileSetMode_Binary(PetscViewer viewer,PetscFileMode type)
10915c6c1daeSBarry Smith {
10925c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
10935c6c1daeSBarry Smith 
10945c6c1daeSBarry Smith   PetscFunctionBegin;
10955c6c1daeSBarry Smith   vbinary->btype = type;
10965c6c1daeSBarry Smith   PetscFunctionReturn(0);
10975c6c1daeSBarry Smith }
10985c6c1daeSBarry Smith 
109981f0254dSBarry Smith static PetscErrorCode PetscViewerFileSetName_Binary(PetscViewer viewer,const char name[])
1100e0385b85SDave May {
1101e0385b85SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1102e0385b85SDave May   PetscErrorCode     ierr;
1103e0385b85SDave May 
1104e0385b85SDave May   PetscFunctionBegin;
1105f90597f1SStefano Zampini   if (vbinary->filename) {
1106f90597f1SStefano Zampini     /* gzip can be run after the file with the previous filename has been closed */
1107f90597f1SStefano Zampini     ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
1108f90597f1SStefano Zampini     ierr = PetscStrallocpy(vbinary->filename,&vbinary->ogzfilename);CHKERRQ(ierr);
1109f90597f1SStefano Zampini     ierr = PetscFree(vbinary->filename);CHKERRQ(ierr);
1110f90597f1SStefano Zampini     viewer->setupcalled = PETSC_FALSE;
1111f90597f1SStefano Zampini   }
1112e0385b85SDave May   ierr = PetscStrallocpy(name,&vbinary->filename);CHKERRQ(ierr);
1113e0385b85SDave May   PetscFunctionReturn(0);
1114e0385b85SDave May }
11155c6c1daeSBarry Smith /*
11165c6c1daeSBarry Smith         Actually opens the file
11175c6c1daeSBarry Smith */
1118e0877f53SBarry Smith static PetscErrorCode PetscViewerFileSetUp_Binary(PetscViewer viewer)
11195c6c1daeSBarry Smith {
11205c6c1daeSBarry Smith   PetscMPIInt        rank;
11215c6c1daeSBarry Smith   PetscErrorCode     ierr;
11225c6c1daeSBarry Smith   size_t             len;
11235c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
11245c6c1daeSBarry Smith   const char         *fname;
11252259a519SDave May   char               bname[PETSC_MAX_PATH_LEN],*gz;
11265c6c1daeSBarry Smith   PetscBool          found;
11275c6c1daeSBarry Smith   PetscFileMode      type = vbinary->btype;
11285c6c1daeSBarry Smith 
11295c6c1daeSBarry Smith   PetscFunctionBegin;
1130e0385b85SDave May   if (type == (PetscFileMode) -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetMode()");
11312259a519SDave May   if (!vbinary->filename) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetName()");
11325c6c1daeSBarry Smith   ierr = PetscViewerFileClose_Binary(viewer);CHKERRQ(ierr);
11335c6c1daeSBarry Smith 
1134ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
11355c6c1daeSBarry Smith 
11365c6c1daeSBarry Smith   /* if ends in .gz strip that off and note user wants file compressed */
11375c6c1daeSBarry Smith   vbinary->storecompressed = PETSC_FALSE;
11385c6c1daeSBarry Smith   if (!rank && type == FILE_MODE_WRITE) {
11395c6c1daeSBarry Smith     /* remove .gz if it ends library name */
11405c6c1daeSBarry Smith     ierr = PetscStrstr(vbinary->filename,".gz",&gz);CHKERRQ(ierr);
11415c6c1daeSBarry Smith     if (gz) {
11425c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
11435c6c1daeSBarry Smith       if (len == 3) {
11445c6c1daeSBarry Smith         *gz = 0;
11455c6c1daeSBarry Smith         vbinary->storecompressed = PETSC_TRUE;
11465c6c1daeSBarry Smith       }
11475c6c1daeSBarry Smith     }
11485c6c1daeSBarry Smith   }
11495c6c1daeSBarry Smith 
11505c6c1daeSBarry Smith   /* only first processor opens file if writeable */
11515c6c1daeSBarry Smith   if (!rank || type == FILE_MODE_READ) {
11525c6c1daeSBarry Smith 
11535c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
11545c6c1daeSBarry Smith       /* possibly get the file from remote site or compressed file */
1155ce94432eSBarry Smith       ierr  = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),vbinary->filename,bname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
11565c6c1daeSBarry Smith       fname = bname;
11575c6c1daeSBarry Smith       if (!rank && !found) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot locate file: %s on node zero",vbinary->filename);
11585c6c1daeSBarry Smith       else if (!found) {
11595c6c1daeSBarry Smith         ierr  = PetscInfo(viewer,"Nonzero processor did not locate readonly file\n");CHKERRQ(ierr);
11605c6c1daeSBarry Smith         fname = 0;
11615c6c1daeSBarry Smith       }
1162a297a907SKarl Rupp     } else fname = vbinary->filename;
11635c6c1daeSBarry Smith 
11645c6c1daeSBarry Smith #if defined(PETSC_HAVE_O_BINARY)
11655c6c1daeSBarry Smith     if (type == FILE_MODE_WRITE) {
11665c6c1daeSBarry Smith       if ((vbinary->fdes = open(fname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY,0666)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file %s for writing",fname);
11675c6c1daeSBarry Smith     } else if (type == FILE_MODE_READ && fname) {
11685c6c1daeSBarry Smith       if ((vbinary->fdes = open(fname,O_RDONLY|O_BINARY,0)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for reading",fname);
11695c6c1daeSBarry Smith     } else if (type == FILE_MODE_APPEND) {
11705c6c1daeSBarry Smith       if ((vbinary->fdes = open(fname,O_WRONLY|O_APPEND|O_BINARY,0)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for writing",fname);
11715c6c1daeSBarry Smith     } else if (fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type");
11725c6c1daeSBarry Smith #else
11735c6c1daeSBarry Smith     if (type == FILE_MODE_WRITE) {
11745c6c1daeSBarry Smith       if ((vbinary->fdes = creat(fname,0666)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file %s for writing",fname);
11755c6c1daeSBarry Smith     } else if (type == FILE_MODE_READ && fname) {
11765c6c1daeSBarry Smith       if ((vbinary->fdes = open(fname,O_RDONLY,0)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for reading",fname);
11775c6c1daeSBarry Smith     } else if (type == FILE_MODE_APPEND) {
11785c6c1daeSBarry Smith       if ((vbinary->fdes = open(fname,O_WRONLY|O_APPEND,0)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file %s for writing",fname);
11795c6c1daeSBarry Smith     } else if (fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type");
11805c6c1daeSBarry Smith #endif
11815c6c1daeSBarry Smith   } else vbinary->fdes = -1;
11825c6c1daeSBarry Smith 
11835c6c1daeSBarry Smith   /*
11845c6c1daeSBarry Smith       try to open info file: all processors open this file if read only
11855c6c1daeSBarry Smith   */
11865c6c1daeSBarry Smith   if (!vbinary->skipinfo && (!rank || type == FILE_MODE_READ)) {
11875c6c1daeSBarry Smith     char infoname[PETSC_MAX_PATH_LEN],iname[PETSC_MAX_PATH_LEN];
11885c6c1daeSBarry Smith 
1189e0385b85SDave May     ierr = PetscStrcpy(infoname,vbinary->filename);CHKERRQ(ierr);
11905c6c1daeSBarry Smith     /* remove .gz if it ends library name */
11915c6c1daeSBarry Smith     ierr = PetscStrstr(infoname,".gz",&gz);CHKERRQ(ierr);
11925c6c1daeSBarry Smith     if (gz) {
11935c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
1194a297a907SKarl Rupp       if (len == 3) *gz = 0;
11955c6c1daeSBarry Smith     }
11965c6c1daeSBarry Smith 
11975c6c1daeSBarry Smith     ierr = PetscStrcat(infoname,".info");CHKERRQ(ierr);
11985c6c1daeSBarry Smith     ierr = PetscFixFilename(infoname,iname);CHKERRQ(ierr);
11995c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
1200ce94432eSBarry Smith       ierr = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),iname,infoname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
1201c5929fdfSBarry Smith       ierr = PetscOptionsInsertFile(PetscObjectComm((PetscObject)viewer),((PetscObject)viewer)->options,infoname,PETSC_FALSE);CHKERRQ(ierr);
12025c6c1daeSBarry Smith     } else {
12035c6c1daeSBarry Smith       vbinary->fdes_info = fopen(infoname,"w");
12045c6c1daeSBarry Smith       if (!vbinary->fdes_info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open .info file %s for writing",infoname);
12055c6c1daeSBarry Smith     }
12065c6c1daeSBarry Smith   }
12075c6c1daeSBarry Smith #if defined(PETSC_USE_LOG)
1208e0385b85SDave May   PetscLogObjectState((PetscObject)viewer,"File: %s",vbinary->filename);
12095c6c1daeSBarry Smith #endif
12105c6c1daeSBarry Smith   PetscFunctionReturn(0);
12115c6c1daeSBarry Smith }
12125c6c1daeSBarry Smith 
12135c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
1214e0385b85SDave May static PetscErrorCode PetscViewerFileSetUp_BinaryMPIIO(PetscViewer viewer)
12155c6c1daeSBarry Smith {
12165c6c1daeSBarry Smith   PetscMPIInt        rank;
12175c6c1daeSBarry Smith   PetscErrorCode     ierr;
12185c6c1daeSBarry Smith   size_t             len;
12195c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
12202259a519SDave May   char               *gz;
12215c6c1daeSBarry Smith   PetscBool          found;
12225c6c1daeSBarry Smith   PetscFileMode      type = vbinary->btype;
12235c6c1daeSBarry Smith 
12245c6c1daeSBarry Smith   PetscFunctionBegin;
1225e0385b85SDave May   if (type == (PetscFileMode) -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetMode()");
12262259a519SDave May   if (!vbinary->filename) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetName()");
1227e0385b85SDave May   ierr = PetscViewerFileClose_BinaryMPIIO(viewer);CHKERRQ(ierr);
12285c6c1daeSBarry Smith 
1229ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
12305c6c1daeSBarry Smith 
1231a297a907SKarl Rupp   vbinary->storecompressed = PETSC_FALSE;
12325c6c1daeSBarry Smith 
12335c6c1daeSBarry Smith   /* only first processor opens file if writeable */
12345c6c1daeSBarry Smith   if (type == FILE_MODE_READ) {
1235ce94432eSBarry Smith     MPI_File_open(PetscObjectComm((PetscObject)viewer),vbinary->filename,MPI_MODE_RDONLY,MPI_INFO_NULL,&vbinary->mfdes);CHKERRQ(ierr);
12365c6c1daeSBarry Smith   } else if (type == FILE_MODE_WRITE) {
1237ce94432eSBarry Smith     MPI_File_open(PetscObjectComm((PetscObject)viewer),vbinary->filename,MPI_MODE_WRONLY | MPI_MODE_CREATE,MPI_INFO_NULL,&vbinary->mfdes);CHKERRQ(ierr);
12385c6c1daeSBarry Smith   }
12395c6c1daeSBarry Smith 
12405c6c1daeSBarry Smith   /*
12415c6c1daeSBarry Smith       try to open info file: all processors open this file if read only
12425c6c1daeSBarry Smith 
1243bebe2cf6SSatish Balay       Below is identical code to the code for Binary above, should be put in separate routine
12445c6c1daeSBarry Smith   */
12455c6c1daeSBarry Smith   if (!vbinary->skipinfo && (!rank || type == FILE_MODE_READ)) {
12465c6c1daeSBarry Smith     char infoname[PETSC_MAX_PATH_LEN],iname[PETSC_MAX_PATH_LEN];
12475c6c1daeSBarry Smith 
1248e0385b85SDave May     ierr = PetscStrcpy(infoname,vbinary->filename);CHKERRQ(ierr);
12495c6c1daeSBarry Smith     /* remove .gz if it ends library name */
12505c6c1daeSBarry Smith     ierr = PetscStrstr(infoname,".gz",&gz);CHKERRQ(ierr);
12515c6c1daeSBarry Smith     if (gz) {
12525c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
1253a297a907SKarl Rupp       if (len == 3) *gz = 0;
12545c6c1daeSBarry Smith     }
12555c6c1daeSBarry Smith 
12565c6c1daeSBarry Smith     ierr = PetscStrcat(infoname,".info");CHKERRQ(ierr);
12575c6c1daeSBarry Smith     ierr = PetscFixFilename(infoname,iname);CHKERRQ(ierr);
12585c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
1259ce94432eSBarry Smith       ierr = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),iname,infoname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
1260c5929fdfSBarry Smith       ierr = PetscOptionsInsertFile(PetscObjectComm((PetscObject)viewer),((PetscObject)viewer)->options,infoname,PETSC_FALSE);CHKERRQ(ierr);
12615c6c1daeSBarry Smith     } else {
12625c6c1daeSBarry Smith       vbinary->fdes_info = fopen(infoname,"w");
12635c6c1daeSBarry Smith       if (!vbinary->fdes_info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open .info file %s for writing",infoname);
12645c6c1daeSBarry Smith     }
12655c6c1daeSBarry Smith   }
12665c6c1daeSBarry Smith #if defined(PETSC_USE_LOG)
1267e0385b85SDave May   PetscLogObjectState((PetscObject)viewer,"File: %s",vbinary->filename);
12685c6c1daeSBarry Smith #endif
12695c6c1daeSBarry Smith   PetscFunctionReturn(0);
12705c6c1daeSBarry Smith }
12715c6c1daeSBarry Smith 
127281f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetUseMPIIO_Binary(PetscViewer viewer,PetscBool flg)
12735c6c1daeSBarry Smith {
12745c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
12755c6c1daeSBarry Smith   PetscFunctionBegin;
1276bc196f7cSDave May   vbinary->usempiio = flg;
12775c6c1daeSBarry Smith   PetscFunctionReturn(0);
12785c6c1daeSBarry Smith }
12795c6c1daeSBarry Smith #endif
12805c6c1daeSBarry Smith 
128181f0254dSBarry Smith static PetscErrorCode PetscViewerView_Binary(PetscViewer v,PetscViewer viewer)
12822bf49c77SBarry Smith {
12832bf49c77SBarry Smith   PetscErrorCode     ierr;
12842bf49c77SBarry Smith   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
12852bf49c77SBarry Smith 
12862bf49c77SBarry Smith   PetscFunctionBegin;
12872bf49c77SBarry Smith   if (binary->filename) {
12882bf49c77SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Filename: %s\n",binary->filename);CHKERRQ(ierr);
12892bf49c77SBarry Smith   }
12902bf49c77SBarry Smith   PetscFunctionReturn(0);
12912bf49c77SBarry Smith }
12922bf49c77SBarry Smith 
129303a1d59fSDave May static PetscErrorCode PetscViewerSetUp_Binary(PetscViewer v)
129403a1d59fSDave May {
129503a1d59fSDave May   PetscErrorCode     ierr;
129603a1d59fSDave May   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
1297e0385b85SDave May 
129803a1d59fSDave May   PetscFunctionBegin;
1299da07bb01SDave May   if (!binary->setfromoptionscalled) { ierr = PetscViewerSetFromOptions(v);CHKERRQ(ierr); }
130003a1d59fSDave May 
1301e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
1302bc196f7cSDave May   if (binary->usempiio) {
1303e0385b85SDave May     ierr = PetscViewerFileSetUp_BinaryMPIIO(v);CHKERRQ(ierr);
1304e0385b85SDave May   } else {
1305e0385b85SDave May #endif
1306e0385b85SDave May     ierr = PetscViewerFileSetUp_Binary(v);CHKERRQ(ierr);
1307e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
1308e0385b85SDave May   }
1309e0385b85SDave May #endif
131003a1d59fSDave May   PetscFunctionReturn(0);
131103a1d59fSDave May }
131203a1d59fSDave May 
13134416b707SBarry Smith static PetscErrorCode PetscViewerSetFromOptions_Binary(PetscOptionItems *PetscOptionsObject,PetscViewer v)
131403a1d59fSDave May {
131503a1d59fSDave May   PetscErrorCode     ierr;
131603a1d59fSDave May   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
1317d22fd6bcSDave May   char               defaultname[PETSC_MAX_PATH_LEN];
131803a1d59fSDave May   PetscBool          flg;
1319e0385b85SDave May 
132003a1d59fSDave May   PetscFunctionBegin;
132103a1d59fSDave May   ierr = PetscOptionsHead(PetscOptionsObject,"Binary PetscViewer Options");CHKERRQ(ierr);
1322d22fd6bcSDave May   ierr = PetscSNPrintf(defaultname,PETSC_MAX_PATH_LEN-1,"binaryoutput");CHKERRQ(ierr);
1323d22fd6bcSDave May   ierr = PetscOptionsString("-viewer_binary_filename","Specify filename","PetscViewerFileSetName",defaultname,defaultname,PETSC_MAX_PATH_LEN-1,&flg);CHKERRQ(ierr);
1324d22fd6bcSDave May   if (flg) { ierr = PetscViewerFileSetName_Binary(v,defaultname);CHKERRQ(ierr); }
132534a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_info","Skip writing/reading .info file","PetscViewerBinarySetSkipInfo",PETSC_FALSE,&binary->skipinfo,NULL);CHKERRQ(ierr);
132634a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_options","Skip parsing vec load options","PetscViewerBinarySetSkipOptions",PETSC_TRUE,&binary->skipoptions,NULL);CHKERRQ(ierr);
132734a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_header","Skip writing/reading header information","PetscViewerBinarySetSkipHeader",PETSC_FALSE,&binary->skipheader,NULL);CHKERRQ(ierr);
132803a1d59fSDave May #if defined(PETSC_HAVE_MPIIO)
132934a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_mpiio","Use MPI-IO functionality to write/read binary file","PetscViewerBinarySetUseMPIIO",PETSC_FALSE,&binary->usempiio,NULL);CHKERRQ(ierr);
133034a9cc2cSBarry Smith #elif defined(PETSC_HAVE_MPIUNI)
133134a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_mpiio","Use MPI-IO functionality to write/read binary file","PetscViewerBinarySetUseMPIIO",PETSC_FALSE,NULL,NULL);CHKERRQ(ierr);
133203a1d59fSDave May #endif
133303a1d59fSDave May   ierr = PetscOptionsTail();CHKERRQ(ierr);
1334bc196f7cSDave May   binary->setfromoptionscalled = PETSC_TRUE;
133503a1d59fSDave May   PetscFunctionReturn(0);
133603a1d59fSDave May }
133703a1d59fSDave May 
13388556b5ebSBarry Smith /*MC
13398556b5ebSBarry Smith    PETSCVIEWERBINARY - A viewer that saves to binary files
13408556b5ebSBarry Smith 
13418556b5ebSBarry Smith 
13428556b5ebSBarry Smith .seealso:  PetscViewerBinaryOpen(), PETSC_VIEWER_STDOUT_(),PETSC_VIEWER_STDOUT_SELF, PETSC_VIEWER_STDOUT_WORLD, PetscViewerCreate(), PetscViewerASCIIOpen(),
13438556b5ebSBarry Smith            PetscViewerMatlabOpen(), VecView(), DMView(), PetscViewerMatlabPutArray(), PETSCVIEWERASCII, PETSCVIEWERMATLAB, PETSCVIEWERDRAW,
134467918a83SBarry Smith            PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType(),
134567918a83SBarry Smith            PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO()
13468556b5ebSBarry Smith 
13471b266c99SBarry Smith   Level: beginner
13481b266c99SBarry Smith 
13498556b5ebSBarry Smith M*/
13508556b5ebSBarry Smith 
13518cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerCreate_Binary(PetscViewer v)
13525c6c1daeSBarry Smith {
13535c6c1daeSBarry Smith   PetscErrorCode     ierr;
13545c6c1daeSBarry Smith   PetscViewer_Binary *vbinary;
13555c6c1daeSBarry Smith 
13565c6c1daeSBarry Smith   PetscFunctionBegin;
1357b00a9115SJed Brown   ierr                     = PetscNewLog(v,&vbinary);CHKERRQ(ierr);
13585c6c1daeSBarry Smith   v->data                  = (void*)vbinary;
135903a1d59fSDave May   v->ops->setfromoptions   = PetscViewerSetFromOptions_Binary;
13605c6c1daeSBarry Smith   v->ops->destroy          = PetscViewerDestroy_Binary;
13612bf49c77SBarry Smith   v->ops->view             = PetscViewerView_Binary;
1362c98fd787SBarry Smith   v->ops->setup            = PetscViewerSetUp_Binary;
1363c98fd787SBarry Smith   v->ops->flush            = NULL;
13645c6c1daeSBarry Smith   vbinary->fdes_info       = 0;
13655c6c1daeSBarry Smith   vbinary->fdes            = 0;
13665c6c1daeSBarry Smith   vbinary->skipinfo        = PETSC_FALSE;
13675c6c1daeSBarry Smith   vbinary->skipoptions     = PETSC_TRUE;
13685c6c1daeSBarry Smith   vbinary->skipheader      = PETSC_FALSE;
1369da07bb01SDave May   vbinary->setfromoptionscalled = PETSC_FALSE;
1370559f443fSBarry Smith   v->ops->getsubviewer     = PetscViewerGetSubViewer_Binary;
1371559f443fSBarry Smith   v->ops->restoresubviewer = PetscViewerRestoreSubViewer_Binary;
13721d641e7bSMichael Lange   v->ops->read             = PetscViewerBinaryRead;
13735c6c1daeSBarry Smith   vbinary->btype           = (PetscFileMode) -1;
13745c6c1daeSBarry Smith   vbinary->storecompressed = PETSC_FALSE;
1375f90597f1SStefano Zampini   vbinary->filename        = NULL;
1376f90597f1SStefano Zampini   vbinary->ogzfilename     = NULL;
13775c6c1daeSBarry Smith   vbinary->flowcontrol     = 256; /* seems a good number for Cray XT-5 */
13785c6c1daeSBarry Smith 
1379bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetFlowControl_C",PetscViewerBinaryGetFlowControl_Binary);CHKERRQ(ierr);
1380bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetFlowControl_C",PetscViewerBinarySetFlowControl_Binary);CHKERRQ(ierr);
1381bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipHeader_C",PetscViewerBinarySetSkipHeader_Binary);CHKERRQ(ierr);
1382bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipHeader_C",PetscViewerBinaryGetSkipHeader_Binary);CHKERRQ(ierr);
1383807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipOptions_C",PetscViewerBinaryGetSkipOptions_Binary);CHKERRQ(ierr);
1384807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipOptions_C",PetscViewerBinarySetSkipOptions_Binary);CHKERRQ(ierr);
1385807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipInfo_C",PetscViewerBinaryGetSkipInfo_Binary);CHKERRQ(ierr);
1386807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipInfo_C",PetscViewerBinarySetSkipInfo_Binary);CHKERRQ(ierr);
1387bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetInfoPointer_C",PetscViewerBinaryGetInfoPointer_Binary);CHKERRQ(ierr);
1388bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_Binary);CHKERRQ(ierr);
1389bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_Binary);CHKERRQ(ierr);
1390bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetMode_C",PetscViewerFileGetMode_Binary);CHKERRQ(ierr);
1391bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_Binary);CHKERRQ(ierr);
13925c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
1393bc196f7cSDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetUseMPIIO_C",PetscViewerBinaryGetUseMPIIO_Binary);CHKERRQ(ierr);
1394bc196f7cSDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetUseMPIIO_C",PetscViewerBinarySetUseMPIIO_Binary);CHKERRQ(ierr);
13955c6c1daeSBarry Smith #endif
13965c6c1daeSBarry Smith   PetscFunctionReturn(0);
13975c6c1daeSBarry Smith }
13985c6c1daeSBarry Smith 
13995c6c1daeSBarry Smith /* ---------------------------------------------------------------------*/
14005c6c1daeSBarry Smith /*
14015c6c1daeSBarry Smith     The variable Petsc_Viewer_Binary_keyval is used to indicate an MPI attribute that
14025c6c1daeSBarry Smith   is attached to a communicator, in this case the attribute is a PetscViewer.
14035c6c1daeSBarry Smith */
1404d4c7638eSBarry Smith PetscMPIInt Petsc_Viewer_Binary_keyval = MPI_KEYVAL_INVALID;
14055c6c1daeSBarry Smith 
14065c6c1daeSBarry Smith /*@C
14075c6c1daeSBarry Smith      PETSC_VIEWER_BINARY_ - Creates a binary PetscViewer shared by all processors
14085c6c1daeSBarry Smith                      in a communicator.
14095c6c1daeSBarry Smith 
14105c6c1daeSBarry Smith      Collective on MPI_Comm
14115c6c1daeSBarry Smith 
14125c6c1daeSBarry Smith      Input Parameter:
14135c6c1daeSBarry Smith .    comm - the MPI communicator to share the binary PetscViewer
14145c6c1daeSBarry Smith 
14155c6c1daeSBarry Smith      Level: intermediate
14165c6c1daeSBarry Smith 
14175c6c1daeSBarry Smith    Options Database Keys:
14185c6c1daeSBarry Smith +    -viewer_binary_filename <name>
14195c6c1daeSBarry Smith .    -viewer_binary_skip_info
1420a2d7db39SDave May .    -viewer_binary_skip_options
1421a2d7db39SDave May .    -viewer_binary_skip_header
1422a2d7db39SDave May -    -viewer_binary_mpiio
14235c6c1daeSBarry Smith 
14245c6c1daeSBarry Smith    Environmental variables:
14255c6c1daeSBarry Smith -   PETSC_VIEWER_BINARY_FILENAME
14265c6c1daeSBarry Smith 
14275c6c1daeSBarry Smith      Notes:
14285c6c1daeSBarry Smith      Unlike almost all other PETSc routines, PETSC_VIEWER_BINARY_ does not return
14295c6c1daeSBarry Smith      an error code.  The binary PetscViewer is usually used in the form
14305c6c1daeSBarry Smith $       XXXView(XXX object,PETSC_VIEWER_BINARY_(comm));
14315c6c1daeSBarry Smith 
14325c6c1daeSBarry Smith .seealso: PETSC_VIEWER_BINARY_WORLD, PETSC_VIEWER_BINARY_SELF, PetscViewerBinaryOpen(), PetscViewerCreate(),
14335c6c1daeSBarry Smith           PetscViewerDestroy()
14345c6c1daeSBarry Smith @*/
14355c6c1daeSBarry Smith PetscViewer PETSC_VIEWER_BINARY_(MPI_Comm comm)
14365c6c1daeSBarry Smith {
14375c6c1daeSBarry Smith   PetscErrorCode ierr;
14385c6c1daeSBarry Smith   PetscBool      flg;
14395c6c1daeSBarry Smith   PetscViewer    viewer;
14405c6c1daeSBarry Smith   char           fname[PETSC_MAX_PATH_LEN];
14415c6c1daeSBarry Smith   MPI_Comm       ncomm;
14425c6c1daeSBarry Smith 
14435c6c1daeSBarry Smith   PetscFunctionBegin;
1444efca3c55SSatish Balay   ierr = PetscCommDuplicate(comm,&ncomm,NULL);if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14455c6c1daeSBarry Smith   if (Petsc_Viewer_Binary_keyval == MPI_KEYVAL_INVALID) {
1446*12801b39SBarry Smith     ierr = MPI_Comm_create_keyval(MPI_COMM_NULL_COPY_FN,MPI_COMM_NULL_DELETE_FN,&Petsc_Viewer_Binary_keyval,0);
1447efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14485c6c1daeSBarry Smith   }
14495c6c1daeSBarry Smith   ierr = MPI_Attr_get(ncomm,Petsc_Viewer_Binary_keyval,(void**)&viewer,(int*)&flg);
1450efca3c55SSatish Balay   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14515c6c1daeSBarry Smith   if (!flg) { /* PetscViewer not yet created */
14525c6c1daeSBarry Smith     ierr = PetscOptionsGetenv(ncomm,"PETSC_VIEWER_BINARY_FILENAME",fname,PETSC_MAX_PATH_LEN,&flg);
1453efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14545c6c1daeSBarry Smith     if (!flg) {
14555c6c1daeSBarry Smith       ierr = PetscStrcpy(fname,"binaryoutput");
1456efca3c55SSatish Balay       if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14575c6c1daeSBarry Smith     }
14585c6c1daeSBarry Smith     ierr = PetscViewerBinaryOpen(ncomm,fname,FILE_MODE_WRITE,&viewer);
1459efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14605c6c1daeSBarry Smith     ierr = PetscObjectRegisterDestroy((PetscObject)viewer);
1461efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14625c6c1daeSBarry Smith     ierr = MPI_Attr_put(ncomm,Petsc_Viewer_Binary_keyval,(void*)viewer);
1463efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14645c6c1daeSBarry Smith   }
14655c6c1daeSBarry Smith   ierr = PetscCommDestroy(&ncomm);
1466efca3c55SSatish Balay   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14675c6c1daeSBarry Smith   PetscFunctionReturn(viewer);
14685c6c1daeSBarry Smith }
1469