xref: /petsc/src/sys/classes/viewer/impls/binary/binv.c (revision f90597f179c57bf20ec57d6ac7534727446b7bd5)
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;
22*f90597f1SStefano 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);
39ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
405c6c1daeSBarry Smith   if (!rank) {
415c6c1daeSBarry Smith     ierr    = PetscViewerCreate(PETSC_COMM_SELF,outviewer);CHKERRQ(ierr);
425c6c1daeSBarry Smith     ierr    = PetscViewerSetType(*outviewer,PETSCVIEWERBINARY);CHKERRQ(ierr);
435c6c1daeSBarry Smith     obinary = (PetscViewer_Binary*)(*outviewer)->data;
445c6c1daeSBarry Smith     ierr    = PetscMemcpy(obinary,vbinary,sizeof(PetscViewer_Binary));CHKERRQ(ierr);
453f08860eSBarry Smith   } SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Cannot get subcomm viewer for binary files or sockets unless SubViewer contains the rank 0 process");
465c6c1daeSBarry Smith   PetscFunctionReturn(0);
475c6c1daeSBarry Smith }
485c6c1daeSBarry Smith 
4981f0254dSBarry Smith static PetscErrorCode PetscViewerRestoreSubViewer_Binary(PetscViewer viewer,MPI_Comm comm,PetscViewer *outviewer)
505c6c1daeSBarry Smith {
515c6c1daeSBarry Smith   PetscErrorCode ierr;
525c6c1daeSBarry Smith   PetscErrorCode rank;
535c6c1daeSBarry Smith 
545c6c1daeSBarry Smith   PetscFunctionBegin;
55ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
565c6c1daeSBarry Smith   if (!rank) {
575c6c1daeSBarry Smith     ierr = PetscFree((*outviewer)->data);CHKERRQ(ierr);
585c6c1daeSBarry Smith     ierr = PetscHeaderDestroy(outviewer);CHKERRQ(ierr);
595c6c1daeSBarry Smith   }
605c6c1daeSBarry Smith   PetscFunctionReturn(0);
615c6c1daeSBarry Smith }
625c6c1daeSBarry Smith 
635c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
645c6c1daeSBarry Smith /*@C
655c6c1daeSBarry Smith     PetscViewerBinaryGetMPIIOOffset - Gets the current offset that should be passed to MPI_File_set_view()
665c6c1daeSBarry Smith 
675c6c1daeSBarry Smith     Not Collective
685c6c1daeSBarry Smith 
695c6c1daeSBarry Smith     Input Parameter:
705c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
715c6c1daeSBarry Smith 
725c6c1daeSBarry Smith     Output Parameter:
735c6c1daeSBarry Smith .    off - the current offset
745c6c1daeSBarry Smith 
755c6c1daeSBarry Smith     Level: advanced
765c6c1daeSBarry Smith 
775c6c1daeSBarry Smith     Fortran Note:
785c6c1daeSBarry Smith     This routine is not supported in Fortran.
795c6c1daeSBarry Smith 
805c6c1daeSBarry Smith     Use PetscViewerBinaryAddMPIIOOffset() to increase this value after you have written a view.
815c6c1daeSBarry Smith 
825c6c1daeSBarry Smith   Concepts: file descriptor^getting
835c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
845c6c1daeSBarry Smith 
8567918a83SBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryAddMPIIOOffset()
865c6c1daeSBarry Smith @*/
875c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetMPIIOOffset(PetscViewer viewer,MPI_Offset *off)
885c6c1daeSBarry Smith {
895c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
905c6c1daeSBarry Smith 
915c6c1daeSBarry Smith   PetscFunctionBegin;
925c6c1daeSBarry Smith   *off = vbinary->moff;
935c6c1daeSBarry Smith   PetscFunctionReturn(0);
945c6c1daeSBarry Smith }
955c6c1daeSBarry Smith 
965c6c1daeSBarry Smith /*@C
975c6c1daeSBarry Smith     PetscViewerBinaryAddMPIIOOffset - Adds to the current offset that should be passed to MPI_File_set_view()
985c6c1daeSBarry Smith 
995c6c1daeSBarry Smith     Not Collective
1005c6c1daeSBarry Smith 
1015c6c1daeSBarry Smith     Input Parameters:
1025c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1035c6c1daeSBarry Smith -    off - the addition to the offset
1045c6c1daeSBarry Smith 
1055c6c1daeSBarry Smith     Level: advanced
1065c6c1daeSBarry Smith 
1075c6c1daeSBarry Smith     Fortran Note:
1085c6c1daeSBarry Smith     This routine is not supported in Fortran.
1095c6c1daeSBarry Smith 
1105c6c1daeSBarry Smith     Use PetscViewerBinaryGetMPIIOOffset() to get the value that you should pass to MPI_File_set_view()
1115c6c1daeSBarry Smith 
1125c6c1daeSBarry Smith   Concepts: file descriptor^getting
1135c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1145c6c1daeSBarry Smith 
11567918a83SBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
1165c6c1daeSBarry Smith @*/
1175c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryAddMPIIOOffset(PetscViewer viewer,MPI_Offset off)
1185c6c1daeSBarry Smith {
1195c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1205c6c1daeSBarry Smith 
1215c6c1daeSBarry Smith   PetscFunctionBegin;
1225c6c1daeSBarry Smith   vbinary->moff += off;
1235c6c1daeSBarry Smith   PetscFunctionReturn(0);
1245c6c1daeSBarry Smith }
1255c6c1daeSBarry Smith 
1265c6c1daeSBarry Smith /*@C
1275c6c1daeSBarry Smith     PetscViewerBinaryGetMPIIODescriptor - Extracts the MPI IO file descriptor from a PetscViewer.
1285c6c1daeSBarry Smith 
1295c6c1daeSBarry Smith     Not Collective
1305c6c1daeSBarry Smith 
1315c6c1daeSBarry Smith     Input Parameter:
1325c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1335c6c1daeSBarry Smith 
1345c6c1daeSBarry Smith     Output Parameter:
1355c6c1daeSBarry Smith .   fdes - file descriptor
1365c6c1daeSBarry Smith 
1375c6c1daeSBarry Smith     Level: advanced
1385c6c1daeSBarry Smith 
1395c6c1daeSBarry Smith     Fortran Note:
1405c6c1daeSBarry Smith     This routine is not supported in Fortran.
1415c6c1daeSBarry Smith 
1425c6c1daeSBarry Smith   Concepts: file descriptor^getting
1435c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1445c6c1daeSBarry Smith 
14567918a83SBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
1465c6c1daeSBarry Smith @*/
1475c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetMPIIODescriptor(PetscViewer viewer,MPI_File *fdes)
1485c6c1daeSBarry Smith {
14903a1d59fSDave May   PetscErrorCode     ierr;
1505c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1515c6c1daeSBarry Smith 
1525c6c1daeSBarry Smith   PetscFunctionBegin;
153c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
1545c6c1daeSBarry Smith   *fdes = vbinary->mfdes;
1555c6c1daeSBarry Smith   PetscFunctionReturn(0);
1565c6c1daeSBarry Smith }
1575c6c1daeSBarry Smith 
15881f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetUseMPIIO_Binary(PetscViewer viewer,PetscBool  *flg)
159a8aae444SDave May {
160a8aae444SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
161a8aae444SDave May 
162a8aae444SDave May   PetscFunctionBegin;
163bc196f7cSDave May   *flg = vbinary->usempiio;
164a8aae444SDave May   PetscFunctionReturn(0);
165a8aae444SDave May }
166a8aae444SDave May #endif
167a8aae444SDave May 
168a8aae444SDave May 
1695c6c1daeSBarry Smith /*@C
170bc196f7cSDave May     PetscViewerBinaryGetUseMPIIO - Returns PETSC_TRUE if the binary viewer uses MPI-IO.
1715c6c1daeSBarry Smith 
1725c6c1daeSBarry Smith     Not Collective
1735c6c1daeSBarry Smith 
1745c6c1daeSBarry Smith     Input Parameter:
1755c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
1765c6c1daeSBarry Smith 
1775c6c1daeSBarry Smith     Output Parameter:
178bc196f7cSDave May -   flg - PETSC_TRUE if MPI-IO is being used
1795c6c1daeSBarry Smith 
1805c6c1daeSBarry Smith     Options Database:
1815c6c1daeSBarry Smith     -viewer_binary_mpiio : Flag for using MPI-IO
1825c6c1daeSBarry Smith 
1835c6c1daeSBarry Smith     Level: advanced
1845c6c1daeSBarry Smith 
185bc196f7cSDave May     Note:
186bc196f7cSDave May     If MPI-IO is not available, this function will always return PETSC_FALSE
187bc196f7cSDave May 
1885c6c1daeSBarry Smith     Fortran Note:
1895c6c1daeSBarry Smith     This routine is not supported in Fortran.
1905c6c1daeSBarry Smith 
1915c6c1daeSBarry Smith   Concepts: file descriptor^getting
1925c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
1935c6c1daeSBarry Smith 
19467918a83SBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetInfoPointer(), PetscViewerBinarySetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
1955c6c1daeSBarry Smith @*/
196bc196f7cSDave May PetscErrorCode PetscViewerBinaryGetUseMPIIO(PetscViewer viewer,PetscBool *flg)
1975c6c1daeSBarry Smith {
198a8aae444SDave May   PetscErrorCode ierr;
1995c6c1daeSBarry Smith 
2005c6c1daeSBarry Smith   PetscFunctionBegin;
201a8aae444SDave May   *flg = PETSC_FALSE;
202bc196f7cSDave May   ierr = PetscTryMethod(viewer,"PetscViewerBinaryGetUseMPIIO_C",(PetscViewer,PetscBool*),(viewer,flg));CHKERRQ(ierr);
2035c6c1daeSBarry Smith   PetscFunctionReturn(0);
2045c6c1daeSBarry Smith }
2055c6c1daeSBarry Smith 
20681f0254dSBarry Smith static PetscErrorCode  PetscViewerBinaryGetFlowControl_Binary(PetscViewer viewer,PetscInt *fc)
2075c6c1daeSBarry Smith {
2085c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
2095c6c1daeSBarry Smith 
2105c6c1daeSBarry Smith   PetscFunctionBegin;
2115c6c1daeSBarry Smith   *fc = vbinary->flowcontrol;
2125c6c1daeSBarry Smith   PetscFunctionReturn(0);
2135c6c1daeSBarry Smith }
2145c6c1daeSBarry Smith 
2155c6c1daeSBarry Smith /*@C
2165c6c1daeSBarry Smith     PetscViewerBinaryGetFlowControl - Returns how many messages are allowed to outstanding at the same time during parallel IO reads/writes
2175c6c1daeSBarry Smith 
2185c6c1daeSBarry Smith     Not Collective
2195c6c1daeSBarry Smith 
2205c6c1daeSBarry Smith     Input Parameter:
2215c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
2225c6c1daeSBarry Smith 
2235c6c1daeSBarry Smith     Output Parameter:
2245c6c1daeSBarry Smith .   fc - the number of messages
2255c6c1daeSBarry Smith 
2265c6c1daeSBarry Smith     Level: advanced
2275c6c1daeSBarry Smith 
2285c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinarySetFlowControl()
2295c6c1daeSBarry Smith 
2305c6c1daeSBarry Smith @*/
2315c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetFlowControl(PetscViewer viewer,PetscInt *fc)
2325c6c1daeSBarry Smith {
2335c6c1daeSBarry Smith   PetscErrorCode ierr;
2345c6c1daeSBarry Smith 
2355c6c1daeSBarry Smith   PetscFunctionBegin;
236163d334eSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetFlowControl_C",(PetscViewer,PetscInt*),(viewer,fc));CHKERRQ(ierr);
2375c6c1daeSBarry Smith   PetscFunctionReturn(0);
2385c6c1daeSBarry Smith }
2395c6c1daeSBarry Smith 
24081f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetFlowControl_Binary(PetscViewer viewer,PetscInt fc)
2415c6c1daeSBarry Smith {
2425c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
2435c6c1daeSBarry Smith 
2445c6c1daeSBarry Smith   PetscFunctionBegin;
245ce94432eSBarry Smith   if (fc <= 1) SETERRQ1(PetscObjectComm((PetscObject)viewer),PETSC_ERR_ARG_OUTOFRANGE,"Flow control count must be greater than 1, %D was set",fc);
2465c6c1daeSBarry Smith   vbinary->flowcontrol = fc;
2475c6c1daeSBarry Smith   PetscFunctionReturn(0);
2485c6c1daeSBarry Smith }
2495c6c1daeSBarry Smith 
2505c6c1daeSBarry Smith /*@C
251639ff905SBarry Smith     PetscViewerBinarySetFlowControl - Sets how many messages are allowed to outstanding at the same time during parallel IO reads/writes
2525c6c1daeSBarry Smith 
2535c6c1daeSBarry Smith     Not Collective
2545c6c1daeSBarry Smith 
2555c6c1daeSBarry Smith     Input Parameter:
2565c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
257639ff905SBarry Smith -   fc - the number of messages, defaults to 256 if this function was not called
2585c6c1daeSBarry Smith 
2595c6c1daeSBarry Smith     Level: advanced
2605c6c1daeSBarry Smith 
2615c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer(), PetscViewerBinaryGetFlowControl()
2625c6c1daeSBarry Smith 
2635c6c1daeSBarry Smith @*/
2645c6c1daeSBarry Smith PetscErrorCode  PetscViewerBinarySetFlowControl(PetscViewer viewer,PetscInt fc)
2655c6c1daeSBarry Smith {
2665c6c1daeSBarry Smith   PetscErrorCode ierr;
2675c6c1daeSBarry Smith 
2685c6c1daeSBarry Smith   PetscFunctionBegin;
2695c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetFlowControl_C",(PetscViewer,PetscInt),(viewer,fc));CHKERRQ(ierr);
2705c6c1daeSBarry Smith   PetscFunctionReturn(0);
2715c6c1daeSBarry Smith }
2725c6c1daeSBarry Smith 
2735c6c1daeSBarry Smith /*@C
2745c6c1daeSBarry Smith     PetscViewerBinaryGetDescriptor - Extracts the file descriptor from a PetscViewer.
2755c6c1daeSBarry Smith 
2765872f025SBarry Smith     Collective On PetscViewer
2775c6c1daeSBarry Smith 
2785c6c1daeSBarry Smith     Input Parameter:
2795c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
2805c6c1daeSBarry Smith 
2815c6c1daeSBarry Smith     Output Parameter:
2825c6c1daeSBarry Smith .   fdes - file descriptor
2835c6c1daeSBarry Smith 
2845c6c1daeSBarry Smith     Level: advanced
2855c6c1daeSBarry Smith 
2865c6c1daeSBarry Smith     Notes:
2875c6c1daeSBarry Smith       For writable binary PetscViewers, the descriptor will only be valid for the
2885c6c1daeSBarry Smith     first processor in the communicator that shares the PetscViewer. For readable
2895c6c1daeSBarry Smith     files it will only be valid on nodes that have the file. If node 0 does not
2905c6c1daeSBarry Smith     have the file it generates an error even if another node does have the file.
2915c6c1daeSBarry Smith 
2925c6c1daeSBarry Smith     Fortran Note:
2935c6c1daeSBarry Smith     This routine is not supported in Fortran.
2945c6c1daeSBarry Smith 
2955872f025SBarry Smith     Developer Notes: This must be called on all processes because Dave May changed
2965872f025SBarry Smith     the source code that this may be trigger a PetscViewerSetUp() call if it was not previously triggered.
2975872f025SBarry Smith 
2985872f025SBarry Smith 
2995c6c1daeSBarry Smith   Concepts: file descriptor^getting
3005c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing file descriptor
3015c6c1daeSBarry Smith 
3025c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetInfoPointer()
3035c6c1daeSBarry Smith @*/
3045c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetDescriptor(PetscViewer viewer,int *fdes)
3055c6c1daeSBarry Smith {
30603a1d59fSDave May   PetscErrorCode     ierr;
3075c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
3085c6c1daeSBarry Smith 
3095c6c1daeSBarry Smith   PetscFunctionBegin;
310c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
3115c6c1daeSBarry Smith   *fdes = vbinary->fdes;
3125c6c1daeSBarry Smith   PetscFunctionReturn(0);
3135c6c1daeSBarry Smith }
3145c6c1daeSBarry Smith 
3155c6c1daeSBarry Smith /*@
3165c6c1daeSBarry Smith     PetscViewerBinarySkipInfo - Binary file will not have .info file created with it
3175c6c1daeSBarry Smith 
3185c6c1daeSBarry Smith     Not Collective
3195c6c1daeSBarry Smith 
3205c6c1daeSBarry Smith     Input Paramter:
3215c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerCreate()
3225c6c1daeSBarry Smith 
3235c6c1daeSBarry Smith     Options Database Key:
3245c6c1daeSBarry Smith .   -viewer_binary_skip_info
3255c6c1daeSBarry Smith 
3265c6c1daeSBarry Smith     Level: advanced
3275c6c1daeSBarry Smith 
328a2d7db39SDave May     Notes: This must be called after PetscViewerSetType(). If you use PetscViewerBinaryOpen() then
3295c6c1daeSBarry Smith     you can only skip the info file with the -viewer_binary_skip_info flag. To use the function you must open the
330a2d7db39SDave May     viewer with PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinarySkipInfo().
3315c6c1daeSBarry Smith 
3325c6c1daeSBarry Smith     The .info contains meta information about the data in the binary file, for example the block size if it was
3335c6c1daeSBarry Smith     set for a vector or matrix.
3345c6c1daeSBarry Smith 
3355c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
3365c6c1daeSBarry Smith 
3375c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySetSkipOptions(),
338807ea322SDave May           PetscViewerBinaryGetSkipOptions(), PetscViewerBinaryGetSkipInfo()
3395c6c1daeSBarry Smith @*/
3405c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySkipInfo(PetscViewer viewer)
3415c6c1daeSBarry Smith {
3425c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
3435c6c1daeSBarry Smith 
3445c6c1daeSBarry Smith   PetscFunctionBegin;
3455c6c1daeSBarry Smith   vbinary->skipinfo = PETSC_TRUE;
3465c6c1daeSBarry Smith   PetscFunctionReturn(0);
3475c6c1daeSBarry Smith }
3485c6c1daeSBarry Smith 
34981f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipInfo_Binary(PetscViewer viewer,PetscBool skip)
350807ea322SDave May {
351807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
352807ea322SDave May 
353807ea322SDave May   PetscFunctionBegin;
354807ea322SDave May   vbinary->skipinfo = skip;
355807ea322SDave May   PetscFunctionReturn(0);
356807ea322SDave May }
357807ea322SDave May 
358807ea322SDave May /*@
359807ea322SDave May     PetscViewerBinarySetSkipInfo - Binary file will not have .info file created with it
360807ea322SDave May 
361807ea322SDave May     Not Collective
362807ea322SDave May 
363807ea322SDave May     Input Paramter:
364807ea322SDave May .   viewer - PetscViewer context, obtained from PetscViewerCreate()
365807ea322SDave May 
366807ea322SDave May     Options Database Key:
367807ea322SDave May .   -viewer_binary_skip_info
368807ea322SDave May 
369807ea322SDave May     Level: advanced
370807ea322SDave May 
371807ea322SDave May     Concepts: PetscViewerBinary^accessing info file
372807ea322SDave May 
373807ea322SDave May .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySetSkipOptions(),
374807ea322SDave May           PetscViewerBinaryGetSkipOptions(), PetscViewerBinaryGetSkipInfo()
375807ea322SDave May @*/
376807ea322SDave May PetscErrorCode PetscViewerBinarySetSkipInfo(PetscViewer viewer,PetscBool skip)
377807ea322SDave May {
378807ea322SDave May   PetscErrorCode ierr;
379807ea322SDave May 
380807ea322SDave May   PetscFunctionBegin;
381807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipInfo_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
382807ea322SDave May   PetscFunctionReturn(0);
383807ea322SDave May }
384807ea322SDave May 
38581f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipInfo_Binary(PetscViewer viewer,PetscBool *skip)
386807ea322SDave May {
387807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
388807ea322SDave May 
389807ea322SDave May   PetscFunctionBegin;
390807ea322SDave May   *skip  = vbinary->skipinfo;
391807ea322SDave May   PetscFunctionReturn(0);
392807ea322SDave May }
393807ea322SDave May 
394807ea322SDave May /*@
395807ea322SDave May     PetscViewerBinaryGetSkipInfo - check if viewer wrote a .info file
396807ea322SDave May 
397807ea322SDave May     Not Collective
398807ea322SDave May 
399807ea322SDave May     Input Parameter:
400807ea322SDave May .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
401807ea322SDave May 
402807ea322SDave May     Output Parameter:
403807ea322SDave May .   skip - PETSC_TRUE implies the .info file was not generated
404807ea322SDave May 
405807ea322SDave May     Level: advanced
406807ea322SDave May 
407807ea322SDave May     Notes: This must be called after PetscViewerSetType()
408807ea322SDave May 
409807ea322SDave May     Concepts: PetscViewerBinary^accessing info file
410807ea322SDave May 
411807ea322SDave May .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
412807ea322SDave May           PetscViewerBinarySetSkipOptions(), PetscViewerBinarySetSkipInfo()
413807ea322SDave May @*/
414807ea322SDave May PetscErrorCode PetscViewerBinaryGetSkipInfo(PetscViewer viewer,PetscBool *skip)
415807ea322SDave May {
416807ea322SDave May   PetscErrorCode ierr;
417807ea322SDave May 
418807ea322SDave May   PetscFunctionBegin;
419807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipInfo_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
420807ea322SDave May   PetscFunctionReturn(0);
421807ea322SDave May }
422807ea322SDave May 
42381f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipOptions_Binary(PetscViewer viewer,PetscBool skip)
424807ea322SDave May {
425807ea322SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
426807ea322SDave May 
427807ea322SDave May   PetscFunctionBegin;
428807ea322SDave May   vbinary->skipoptions = skip;
429807ea322SDave May   PetscFunctionReturn(0);
430807ea322SDave May }
431807ea322SDave May 
4325c6c1daeSBarry Smith /*@
4335c6c1daeSBarry Smith     PetscViewerBinarySetSkipOptions - do not use the PETSc options database when loading objects
4345c6c1daeSBarry Smith 
4355c6c1daeSBarry Smith     Not Collective
4365c6c1daeSBarry Smith 
4375c6c1daeSBarry Smith     Input Parameters:
4385c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
4395c6c1daeSBarry Smith -   skip - PETSC_TRUE means do not use
4405c6c1daeSBarry Smith 
4415c6c1daeSBarry Smith     Options Database Key:
4425c6c1daeSBarry Smith .   -viewer_binary_skip_options
4435c6c1daeSBarry Smith 
4445c6c1daeSBarry Smith     Level: advanced
4455c6c1daeSBarry Smith 
4465c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
4475c6c1daeSBarry Smith 
4485c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
4495c6c1daeSBarry Smith 
4505c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
4515c6c1daeSBarry Smith           PetscViewerBinaryGetSkipOptions()
4525c6c1daeSBarry Smith @*/
4535c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySetSkipOptions(PetscViewer viewer,PetscBool skip)
4545c6c1daeSBarry Smith {
455807ea322SDave May   PetscErrorCode ierr;
4565c6c1daeSBarry Smith 
4575c6c1daeSBarry Smith   PetscFunctionBegin;
458807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipOptions_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
459807ea322SDave May   PetscFunctionReturn(0);
460807ea322SDave May }
461807ea322SDave May 
46281f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipOptions_Binary(PetscViewer viewer,PetscBool *skip)
463807ea322SDave May {
464807ea322SDave May   PetscViewer_Binary *vbinary;
465807ea322SDave May 
466807ea322SDave May   PetscFunctionBegin;
467807ea322SDave May   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
468807ea322SDave May   vbinary = (PetscViewer_Binary*)viewer->data;
469807ea322SDave May   *skip   = vbinary->skipoptions;
4705c6c1daeSBarry Smith   PetscFunctionReturn(0);
4715c6c1daeSBarry Smith }
4725c6c1daeSBarry Smith 
4735c6c1daeSBarry Smith /*@
4745c6c1daeSBarry Smith     PetscViewerBinaryGetSkipOptions - checks if viewer uses the PETSc options database when loading objects
4755c6c1daeSBarry Smith 
4765c6c1daeSBarry Smith     Not Collective
4775c6c1daeSBarry Smith 
4785c6c1daeSBarry Smith     Input Parameter:
4795c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
4805c6c1daeSBarry Smith 
4815c6c1daeSBarry Smith     Output Parameter:
4825c6c1daeSBarry Smith .   skip - PETSC_TRUE means do not use
4835c6c1daeSBarry Smith 
4845c6c1daeSBarry Smith     Level: advanced
4855c6c1daeSBarry Smith 
4865c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
4875c6c1daeSBarry Smith 
4885c6c1daeSBarry Smith    Concepts: PetscViewerBinary^accessing info file
4895c6c1daeSBarry Smith 
4905c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
4915c6c1daeSBarry Smith           PetscViewerBinarySetSkipOptions()
4925c6c1daeSBarry Smith @*/
4935c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetSkipOptions(PetscViewer viewer,PetscBool *skip)
4945c6c1daeSBarry Smith {
495807ea322SDave May   PetscErrorCode ierr;
4965c6c1daeSBarry Smith 
4975c6c1daeSBarry Smith   PetscFunctionBegin;
498807ea322SDave May   ierr = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipOptions_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
4995c6c1daeSBarry Smith   PetscFunctionReturn(0);
5005c6c1daeSBarry Smith }
5015c6c1daeSBarry Smith 
50281f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetSkipHeader_Binary(PetscViewer viewer,PetscBool skip)
5035c6c1daeSBarry Smith {
5045c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
5055c6c1daeSBarry Smith 
5065c6c1daeSBarry Smith   PetscFunctionBegin;
5075c6c1daeSBarry Smith   vbinary->skipheader = skip;
5085c6c1daeSBarry Smith   PetscFunctionReturn(0);
5095c6c1daeSBarry Smith }
5105c6c1daeSBarry Smith 
5115c6c1daeSBarry Smith /*@
5125c6c1daeSBarry Smith     PetscViewerBinarySetSkipHeader - do not write a header with size information on output, just raw data
5135c6c1daeSBarry Smith 
5145c6c1daeSBarry Smith     Not Collective
5155c6c1daeSBarry Smith 
5165c6c1daeSBarry Smith     Input Parameters:
5175c6c1daeSBarry Smith +   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
5185c6c1daeSBarry Smith -   skip - PETSC_TRUE means do not write header
5195c6c1daeSBarry Smith 
5205c6c1daeSBarry Smith     Options Database Key:
5215c6c1daeSBarry Smith .   -viewer_binary_skip_header
5225c6c1daeSBarry Smith 
5235c6c1daeSBarry Smith     Level: advanced
5245c6c1daeSBarry Smith 
5255c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
5265c6c1daeSBarry Smith 
5275c6c1daeSBarry Smith            Can ONLY be called on a binary viewer
5285c6c1daeSBarry Smith 
5295c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
5305c6c1daeSBarry Smith           PetscViewerBinaryGetSkipHeader()
5315c6c1daeSBarry Smith @*/
5325c6c1daeSBarry Smith PetscErrorCode PetscViewerBinarySetSkipHeader(PetscViewer viewer,PetscBool skip)
5335c6c1daeSBarry Smith {
5345c6c1daeSBarry Smith   PetscErrorCode ierr;
5355c6c1daeSBarry Smith 
5365c6c1daeSBarry Smith   PetscFunctionBegin;
5375c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerBinarySetSkipHeader_C",(PetscViewer,PetscBool),(viewer,skip));CHKERRQ(ierr);
5385c6c1daeSBarry Smith   PetscFunctionReturn(0);
5395c6c1daeSBarry Smith }
5405c6c1daeSBarry Smith 
54181f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetSkipHeader_Binary(PetscViewer viewer,PetscBool  *skip)
5425c6c1daeSBarry Smith {
5435c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
5445c6c1daeSBarry Smith 
5455c6c1daeSBarry Smith   PetscFunctionBegin;
5465c6c1daeSBarry Smith   *skip = vbinary->skipheader;
5475c6c1daeSBarry Smith   PetscFunctionReturn(0);
5485c6c1daeSBarry Smith }
5495c6c1daeSBarry Smith 
5505c6c1daeSBarry Smith /*@
5515c6c1daeSBarry Smith     PetscViewerBinaryGetSkipHeader - checks whether to write a header with size information on output, or just raw data
5525c6c1daeSBarry Smith 
5535c6c1daeSBarry Smith     Not Collective
5545c6c1daeSBarry Smith 
5555c6c1daeSBarry Smith     Input Parameter:
5565c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
5575c6c1daeSBarry Smith 
5585c6c1daeSBarry Smith     Output Parameter:
5595c6c1daeSBarry Smith .   skip - PETSC_TRUE means do not write header
5605c6c1daeSBarry Smith 
5615c6c1daeSBarry Smith     Level: advanced
5625c6c1daeSBarry Smith 
5635c6c1daeSBarry Smith     Notes: This must be called after PetscViewerSetType()
5645c6c1daeSBarry Smith 
5655c6c1daeSBarry Smith             Returns false for PETSCSOCKETVIEWER, you cannot skip the header for it.
5665c6c1daeSBarry Smith 
5675c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(), PetscViewerBinaryGetDescriptor(), PetscViewerBinarySkipInfo(),
5685c6c1daeSBarry Smith           PetscViewerBinarySetSkipHeader()
5695c6c1daeSBarry Smith @*/
5705c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetSkipHeader(PetscViewer viewer,PetscBool  *skip)
5715c6c1daeSBarry Smith {
5725c6c1daeSBarry Smith   PetscErrorCode ierr;
5735c6c1daeSBarry Smith 
5745c6c1daeSBarry Smith   PetscFunctionBegin;
5755c6c1daeSBarry Smith   *skip = PETSC_FALSE;
576163d334eSBarry Smith   ierr  = PetscUseMethod(viewer,"PetscViewerBinaryGetSkipHeader_C",(PetscViewer,PetscBool*),(viewer,skip));CHKERRQ(ierr);
5775c6c1daeSBarry Smith   PetscFunctionReturn(0);
5785c6c1daeSBarry Smith }
5795c6c1daeSBarry Smith 
58081f0254dSBarry Smith static PetscErrorCode PetscViewerBinaryGetInfoPointer_Binary(PetscViewer viewer,FILE **file)
5815c6c1daeSBarry Smith {
582f3b211e4SSatish Balay   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
583a261c58fSBarry Smith   PetscErrorCode     ierr;
584a261c58fSBarry Smith   MPI_Comm           comm;
5855c6c1daeSBarry Smith 
5865c6c1daeSBarry Smith   PetscFunctionBegin;
587c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
5885c6c1daeSBarry Smith   *file = vbinary->fdes_info;
589a261c58fSBarry Smith   if (viewer->format == PETSC_VIEWER_BINARY_MATLAB && !vbinary->matlabheaderwritten) {
590a261c58fSBarry Smith     vbinary->matlabheaderwritten = PETSC_TRUE;
591a261c58fSBarry Smith     ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
592da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#--- begin code written by PetscViewerBinary for MATLAB format ---#\n");CHKERRQ(ierr);
593da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#$$ Set.filename = '%s';\n",vbinary->filename);CHKERRQ(ierr);
594da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#$$ fd = PetscOpenFile(Set.filename);\n");CHKERRQ(ierr);
595da88d4d4SJed Brown     ierr = PetscFPrintf(comm,*file,"#--- end code written by PetscViewerBinary for MATLAB format ---#\n\n");CHKERRQ(ierr);
596a261c58fSBarry Smith   }
5975c6c1daeSBarry Smith   PetscFunctionReturn(0);
5985c6c1daeSBarry Smith }
5995c6c1daeSBarry Smith 
6005c6c1daeSBarry Smith /*@C
6015c6c1daeSBarry Smith     PetscViewerBinaryGetInfoPointer - Extracts the file pointer for the ASCII
6025c6c1daeSBarry Smith           info file associated with a binary file.
6035c6c1daeSBarry Smith 
6045c6c1daeSBarry Smith     Not Collective
6055c6c1daeSBarry Smith 
6065c6c1daeSBarry Smith     Input Parameter:
6075c6c1daeSBarry Smith .   viewer - PetscViewer context, obtained from PetscViewerBinaryOpen()
6085c6c1daeSBarry Smith 
6095c6c1daeSBarry Smith     Output Parameter:
6100298fd71SBarry Smith .   file - file pointer  Always returns NULL if not a binary viewer
6115c6c1daeSBarry Smith 
6125c6c1daeSBarry Smith     Level: advanced
6135c6c1daeSBarry Smith 
6145c6c1daeSBarry Smith     Notes:
6155c6c1daeSBarry Smith       For writable binary PetscViewers, the descriptor will only be valid for the
6165c6c1daeSBarry Smith     first processor in the communicator that shares the PetscViewer.
6175c6c1daeSBarry Smith 
6185c6c1daeSBarry Smith     Fortran Note:
6195c6c1daeSBarry Smith     This routine is not supported in Fortran.
6205c6c1daeSBarry Smith 
6215c6c1daeSBarry Smith   Concepts: PetscViewerBinary^accessing info file
6225c6c1daeSBarry Smith 
6235c6c1daeSBarry Smith .seealso: PetscViewerBinaryOpen(),PetscViewerBinaryGetDescriptor()
6245c6c1daeSBarry Smith @*/
6255c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryGetInfoPointer(PetscViewer viewer,FILE **file)
6265c6c1daeSBarry Smith {
6275c6c1daeSBarry Smith   PetscErrorCode ierr;
6285c6c1daeSBarry Smith 
6295c6c1daeSBarry Smith   PetscFunctionBegin;
6300298fd71SBarry Smith   *file = NULL;
6315c6c1daeSBarry Smith   ierr  = PetscTryMethod(viewer,"PetscViewerBinaryGetInfoPointer_C",(PetscViewer,FILE **),(viewer,file));CHKERRQ(ierr);
6325c6c1daeSBarry Smith   PetscFunctionReturn(0);
6335c6c1daeSBarry Smith }
6345c6c1daeSBarry Smith 
6355c6c1daeSBarry Smith static PetscErrorCode PetscViewerFileClose_Binary(PetscViewer v)
6365c6c1daeSBarry Smith {
6375c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
6385c6c1daeSBarry Smith   PetscErrorCode     ierr;
6395c6c1daeSBarry Smith   PetscMPIInt        rank;
6405c6c1daeSBarry Smith   int                err;
6415c6c1daeSBarry Smith 
6425c6c1daeSBarry Smith   PetscFunctionBegin;
643ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)v),&rank);CHKERRQ(ierr);
6445c6c1daeSBarry Smith   if ((!rank || vbinary->btype == FILE_MODE_READ) && vbinary->fdes) {
6455c6c1daeSBarry Smith     close(vbinary->fdes);
6465c6c1daeSBarry Smith     if (!rank && vbinary->storecompressed) {
6475c6c1daeSBarry Smith       char par[PETSC_MAX_PATH_LEN],buf[PETSC_MAX_PATH_LEN];
6485c6c1daeSBarry Smith       FILE *fp;
6495c6c1daeSBarry Smith       /* compress the file */
6505c6c1daeSBarry Smith       ierr = PetscStrcpy(par,"gzip -f ");CHKERRQ(ierr);
651*f90597f1SStefano Zampini       ierr = PetscStrcat(par,vbinary->ogzfilename ? vbinary->ogzfilename : vbinary->filename);CHKERRQ(ierr);
652*f90597f1SStefano Zampini       ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
6535c6c1daeSBarry Smith #if defined(PETSC_HAVE_POPEN)
6540298fd71SBarry Smith       ierr = PetscPOpen(PETSC_COMM_SELF,NULL,par,"r",&fp);CHKERRQ(ierr);
6555c6c1daeSBarry Smith       if (fgets(buf,1024,fp)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error from command %s\n%s",par,buf);
6560298fd71SBarry Smith       ierr = PetscPClose(PETSC_COMM_SELF,fp,NULL);CHKERRQ(ierr);
6575c6c1daeSBarry Smith #else
6585c6c1daeSBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine");
6595c6c1daeSBarry Smith #endif
6605c6c1daeSBarry Smith     }
6615c6c1daeSBarry Smith   }
6625c6c1daeSBarry Smith   if (vbinary->fdes_info) {
6635c6c1daeSBarry Smith     err = fclose(vbinary->fdes_info);
6645c6c1daeSBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
6655c6c1daeSBarry Smith   }
6665c6c1daeSBarry Smith   PetscFunctionReturn(0);
6675c6c1daeSBarry Smith }
6685c6c1daeSBarry Smith 
669e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
670e0385b85SDave May static PetscErrorCode PetscViewerFileClose_BinaryMPIIO(PetscViewer v)
671e0385b85SDave May {
672e0385b85SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
673e0385b85SDave May   int                err;
674e0385b85SDave May   PetscErrorCode     ierr;
675e0385b85SDave May 
676e0385b85SDave May   PetscFunctionBegin;
677e0385b85SDave May   if (vbinary->mfdes) {
678e0385b85SDave May     ierr = MPI_File_close(&vbinary->mfdes);CHKERRQ(ierr);
679e0385b85SDave May   }
680e0385b85SDave May   if (vbinary->fdes_info) {
681e0385b85SDave May     err = fclose(vbinary->fdes_info);
682e0385b85SDave May     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
683e0385b85SDave May   }
684e0385b85SDave May   PetscFunctionReturn(0);
685e0385b85SDave May }
686e0385b85SDave May #endif
687e0385b85SDave May 
68881f0254dSBarry Smith static PetscErrorCode PetscViewerDestroy_Binary(PetscViewer v)
6895c6c1daeSBarry Smith {
6905c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)v->data;
6915c6c1daeSBarry Smith   PetscErrorCode     ierr;
6925c6c1daeSBarry Smith 
6935c6c1daeSBarry Smith   PetscFunctionBegin;
694a261c58fSBarry Smith   if (v->format == PETSC_VIEWER_BINARY_MATLAB) {
695a261c58fSBarry Smith     MPI_Comm comm;
696a261c58fSBarry Smith     FILE     *info;
697a261c58fSBarry Smith 
698a261c58fSBarry Smith     ierr = PetscObjectGetComm((PetscObject)v,&comm);CHKERRQ(ierr);
699a261c58fSBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(v,&info);CHKERRQ(ierr);
700da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#--- begin code written by PetscViewerBinary for MATLAB format ---#\n");CHKERRQ(ierr);
701da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#$$ close(fd);\n");CHKERRQ(ierr);
702da88d4d4SJed Brown     ierr = PetscFPrintf(comm,info,"#--- end code written by PetscViewerBinary for MATLAB format ---#\n\n");CHKERRQ(ierr);
703a261c58fSBarry Smith   }
7045c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
705bc196f7cSDave May   if (vbinary->usempiio) {
706e0385b85SDave May     ierr = PetscViewerFileClose_BinaryMPIIO(v);CHKERRQ(ierr);
707e0385b85SDave May   } else {
708e0385b85SDave May #endif
709e0385b85SDave May     ierr = PetscViewerFileClose_Binary(v);CHKERRQ(ierr);
710e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
7115c6c1daeSBarry Smith   }
7125c6c1daeSBarry Smith #endif
713*f90597f1SStefano Zampini   ierr = PetscFree(vbinary->filename);CHKERRQ(ierr);
714*f90597f1SStefano Zampini   ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
715e0385b85SDave May   ierr = PetscFree(vbinary);CHKERRQ(ierr);
716e0385b85SDave May   PetscFunctionReturn(0);
717e0385b85SDave May }
7185c6c1daeSBarry Smith 
7195c6c1daeSBarry Smith /*@C
7205c6c1daeSBarry Smith    PetscViewerBinaryOpen - Opens a file for binary input/output.
7215c6c1daeSBarry Smith 
7225c6c1daeSBarry Smith    Collective on MPI_Comm
7235c6c1daeSBarry Smith 
7245c6c1daeSBarry Smith    Input Parameters:
7255c6c1daeSBarry Smith +  comm - MPI communicator
7265c6c1daeSBarry Smith .  name - name of file
7275c6c1daeSBarry Smith -  type - type of file
7285c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
7295c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
7305c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
7315c6c1daeSBarry Smith 
7325c6c1daeSBarry Smith    Output Parameter:
7335c6c1daeSBarry Smith .  binv - PetscViewer for binary input/output to use with the specified file
7345c6c1daeSBarry Smith 
7355c6c1daeSBarry Smith     Options Database Keys:
736a2d7db39SDave May +    -viewer_binary_filename <name>
737a2d7db39SDave May .    -viewer_binary_skip_info
7385c6c1daeSBarry Smith .    -viewer_binary_skip_options
739a2d7db39SDave May .    -viewer_binary_skip_header
740a2d7db39SDave May -    -viewer_binary_mpiio
7415c6c1daeSBarry Smith 
7425c6c1daeSBarry Smith    Level: beginner
7435c6c1daeSBarry Smith 
7445c6c1daeSBarry Smith    Note:
7455c6c1daeSBarry Smith    This PetscViewer should be destroyed with PetscViewerDestroy().
7465c6c1daeSBarry Smith 
7475c6c1daeSBarry Smith     For reading files, the filename may begin with ftp:// or http:// and/or
7485c6c1daeSBarry Smith     end with .gz; in this case file is brought over and uncompressed.
7495c6c1daeSBarry Smith 
7505c6c1daeSBarry Smith     For creating files, if the file name ends with .gz it is automatically
7515c6c1daeSBarry Smith     compressed when closed.
7525c6c1daeSBarry Smith 
7535c6c1daeSBarry Smith     For writing files it only opens the file on processor 0 in the communicator.
7545c6c1daeSBarry Smith     For readable files it opens the file on all nodes that have the file. If
7555c6c1daeSBarry Smith     node 0 does not have the file it generates an error even if other nodes
7565c6c1daeSBarry Smith     do have the file.
7575c6c1daeSBarry Smith 
7585c6c1daeSBarry Smith    Concepts: binary files
7595c6c1daeSBarry Smith    Concepts: PetscViewerBinary^creating
7605c6c1daeSBarry Smith    Concepts: gzip
7615c6c1daeSBarry Smith    Concepts: accessing remote file
7625c6c1daeSBarry Smith    Concepts: remote file
7635c6c1daeSBarry Smith 
7646a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
7655c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
76667918a83SBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscViewerBinaryRead(), PetscViewerBinarySetUseMPIIO(),
76767918a83SBarry Smith           PetscViewerBinaryGetUseMPIIO(), PetscViewerBinaryGetMPIIOOffset()
7685c6c1daeSBarry Smith @*/
7695c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryOpen(MPI_Comm comm,const char name[],PetscFileMode type,PetscViewer *binv)
7705c6c1daeSBarry Smith {
7715c6c1daeSBarry Smith   PetscErrorCode ierr;
7725c6c1daeSBarry Smith 
7735c6c1daeSBarry Smith   PetscFunctionBegin;
7745c6c1daeSBarry Smith   ierr = PetscViewerCreate(comm,binv);CHKERRQ(ierr);
7755c6c1daeSBarry Smith   ierr = PetscViewerSetType(*binv,PETSCVIEWERBINARY);CHKERRQ(ierr);
7765c6c1daeSBarry Smith   ierr = PetscViewerFileSetMode(*binv,type);CHKERRQ(ierr);
7775c6c1daeSBarry Smith   ierr = PetscViewerFileSetName(*binv,name);CHKERRQ(ierr);
77803a1d59fSDave May   ierr = PetscViewerSetFromOptions(*binv);CHKERRQ(ierr);
7795c6c1daeSBarry Smith   PetscFunctionReturn(0);
7805c6c1daeSBarry Smith }
7815c6c1daeSBarry Smith 
7825c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
783060da220SMatthew G. Knepley static PetscErrorCode PetscViewerBinaryWriteReadMPIIO(PetscViewer viewer,void *data,PetscInt num,PetscInt *count,PetscDataType dtype,PetscBool write)
7845c6c1daeSBarry Smith {
7855c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
7865c6c1daeSBarry Smith   PetscErrorCode     ierr;
7875c6c1daeSBarry Smith   MPI_Datatype       mdtype;
788e597bc1dSJed Brown   PetscMPIInt        cnt;
7895c6c1daeSBarry Smith   MPI_Status         status;
7905c6c1daeSBarry Smith   MPI_Aint           ul,dsize;
7915c6c1daeSBarry Smith 
7925c6c1daeSBarry Smith   PetscFunctionBegin;
793060da220SMatthew G. Knepley   ierr = PetscMPIIntCast(num,&cnt);CHKERRQ(ierr);
7945c6c1daeSBarry Smith   ierr = PetscDataTypeToMPIDataType(dtype,&mdtype);CHKERRQ(ierr);
7955c6c1daeSBarry Smith   ierr = MPI_File_set_view(vbinary->mfdes,vbinary->moff,mdtype,mdtype,(char*)"native",MPI_INFO_NULL);CHKERRQ(ierr);
7965c6c1daeSBarry Smith   if (write) {
7975c6c1daeSBarry Smith     ierr = MPIU_File_write_all(vbinary->mfdes,data,cnt,mdtype,&status);CHKERRQ(ierr);
7985c6c1daeSBarry Smith   } else {
7995c6c1daeSBarry Smith     ierr = MPIU_File_read_all(vbinary->mfdes,data,cnt,mdtype,&status);CHKERRQ(ierr);
8005c6c1daeSBarry Smith   }
8015c6c1daeSBarry Smith   ierr = MPI_Type_get_extent(mdtype,&ul,&dsize);CHKERRQ(ierr);
802a297a907SKarl Rupp 
8035c6c1daeSBarry Smith   vbinary->moff += dsize*cnt;
804060da220SMatthew G. Knepley   if (count) *count = num;
8055c6c1daeSBarry Smith   PetscFunctionReturn(0);
8065c6c1daeSBarry Smith }
8075c6c1daeSBarry Smith #endif
8085c6c1daeSBarry Smith 
8095c6c1daeSBarry Smith /*@C
8105c6c1daeSBarry Smith    PetscViewerBinaryRead - Reads from a binary file, all processors get the same result
8115c6c1daeSBarry Smith 
8125c6c1daeSBarry Smith    Collective on MPI_Comm
8135c6c1daeSBarry Smith 
8145c6c1daeSBarry Smith    Input Parameters:
8155c6c1daeSBarry Smith +  viewer - the binary viewer
816907376e6SBarry Smith .  data - location of the data to be written
817060da220SMatthew G. Knepley .  num - number of items of data to read
818907376e6SBarry Smith -  dtype - type of data to read
8195c6c1daeSBarry Smith 
820f8e4bde8SMatthew G. Knepley    Output Parameters:
8210780a867SBarry 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.
822f8e4bde8SMatthew G. Knepley 
8235c6c1daeSBarry Smith    Level: beginner
8245c6c1daeSBarry Smith 
8255c6c1daeSBarry Smith    Concepts: binary files
8265c6c1daeSBarry Smith 
8270780a867SBarry Smith    Developer Note: Since count is always set to num it is not clear what purpose the output argument count serves.
8280780a867SBarry Smith 
8296a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
8305c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
8315c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
8325c6c1daeSBarry Smith @*/
833060da220SMatthew G. Knepley PetscErrorCode PetscViewerBinaryRead(PetscViewer viewer,void *data,PetscInt num,PetscInt *count,PetscDataType dtype)
8345c6c1daeSBarry Smith {
8355c6c1daeSBarry Smith   PetscErrorCode     ierr;
8365c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
8375c6c1daeSBarry Smith 
838c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
8395c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
840bc196f7cSDave May   if (vbinary->usempiio) {
841060da220SMatthew G. Knepley     ierr = PetscViewerBinaryWriteReadMPIIO(viewer,data,num,count,dtype,PETSC_FALSE);CHKERRQ(ierr);
8425c6c1daeSBarry Smith   } else {
8435c6c1daeSBarry Smith #endif
844060da220SMatthew G. Knepley     ierr = PetscBinarySynchronizedRead(PetscObjectComm((PetscObject)viewer),vbinary->fdes,data,num,dtype);CHKERRQ(ierr);
8450780a867SBarry Smith     if (count) *count = num;
8465c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
8475c6c1daeSBarry Smith   }
8485c6c1daeSBarry Smith #endif
8495c6c1daeSBarry Smith   PetscFunctionReturn(0);
8505c6c1daeSBarry Smith }
8515c6c1daeSBarry Smith 
8525c6c1daeSBarry Smith /*@C
8535c6c1daeSBarry Smith    PetscViewerBinaryWrite - writes to a binary file, only from the first process
8545c6c1daeSBarry Smith 
8555c6c1daeSBarry Smith    Collective on MPI_Comm
8565c6c1daeSBarry Smith 
8575c6c1daeSBarry Smith    Input Parameters:
8585c6c1daeSBarry Smith +  viewer - the binary viewer
8595c6c1daeSBarry Smith .  data - location of data
8605824c9d0SPatrick Sanan .  count - number of items of data to write
8615824c9d0SPatrick Sanan .  dtype - type of data to write
8625c6c1daeSBarry Smith -  istemp - data may be overwritten
8635c6c1daeSBarry Smith 
8645c6c1daeSBarry Smith    Level: beginner
8655c6c1daeSBarry Smith 
8665c6c1daeSBarry Smith    Notes: because byte-swapping may be done on the values in data it cannot be declared const
8675c6c1daeSBarry Smith 
8685c6c1daeSBarry Smith    Concepts: binary files
8695c6c1daeSBarry Smith 
8706a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
8715c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(), PetscDataType
8725c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
8735c6c1daeSBarry Smith @*/
8745c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryWrite(PetscViewer viewer,void *data,PetscInt count,PetscDataType dtype,PetscBool istemp)
8755c6c1daeSBarry Smith {
8765c6c1daeSBarry Smith   PetscErrorCode     ierr;
8775c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
8785c6c1daeSBarry Smith 
8795c6c1daeSBarry Smith   PetscFunctionBegin;
880c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
8815c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
882bc196f7cSDave May   if (vbinary->usempiio) {
883060da220SMatthew G. Knepley     ierr = PetscViewerBinaryWriteReadMPIIO(viewer,data,count,NULL,dtype,PETSC_TRUE);CHKERRQ(ierr);
8845c6c1daeSBarry Smith   } else {
8855c6c1daeSBarry Smith #endif
886ce94432eSBarry Smith     ierr = PetscBinarySynchronizedWrite(PetscObjectComm((PetscObject)viewer),vbinary->fdes,data,count,dtype,istemp);CHKERRQ(ierr);
8875c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
8885c6c1daeSBarry Smith   }
8895c6c1daeSBarry Smith #endif
8905c6c1daeSBarry Smith   PetscFunctionReturn(0);
8915c6c1daeSBarry Smith }
8925c6c1daeSBarry Smith 
8935c6c1daeSBarry Smith /*@C
8945c6c1daeSBarry Smith    PetscViewerBinaryWriteStringArray - writes to a binary file, only from the first process an array of strings
8955c6c1daeSBarry Smith 
8965c6c1daeSBarry Smith    Collective on MPI_Comm
8975c6c1daeSBarry Smith 
8985c6c1daeSBarry Smith    Input Parameters:
8995c6c1daeSBarry Smith +  viewer - the binary viewer
9005c6c1daeSBarry Smith -  data - location of the array of strings
9015c6c1daeSBarry Smith 
9025c6c1daeSBarry Smith 
9035c6c1daeSBarry Smith    Level: intermediate
9045c6c1daeSBarry Smith 
9055c6c1daeSBarry Smith    Concepts: binary files
9065c6c1daeSBarry Smith 
9075c6c1daeSBarry Smith     Notes: array of strings is null terminated
9085c6c1daeSBarry Smith 
9096a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
9105c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
9115c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
9125c6c1daeSBarry Smith @*/
91378fbdcc8SBarry Smith PetscErrorCode PetscViewerBinaryWriteStringArray(PetscViewer viewer,const char * const *data)
9145c6c1daeSBarry Smith {
9155c6c1daeSBarry Smith   PetscErrorCode ierr;
9165c6c1daeSBarry Smith   PetscInt       i,n = 0,*sizes;
9175c6c1daeSBarry Smith 
918c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
9195c6c1daeSBarry Smith   /* count number of strings */
9205c6c1daeSBarry Smith   while (data[n++]) ;
9215c6c1daeSBarry Smith   n--;
922854ce69bSBarry Smith   ierr     = PetscMalloc1(n+1,&sizes);CHKERRQ(ierr);
9235c6c1daeSBarry Smith   sizes[0] = n;
9245c6c1daeSBarry Smith   for (i=0; i<n; i++) {
9255c6c1daeSBarry Smith     size_t tmp;
9265c6c1daeSBarry Smith     ierr       = PetscStrlen(data[i],&tmp);CHKERRQ(ierr);
9275c6c1daeSBarry Smith     sizes[i+1] = tmp + 1;   /* size includes space for the null terminator */
9285c6c1daeSBarry Smith   }
9295c6c1daeSBarry Smith   ierr = PetscViewerBinaryWrite(viewer,sizes,n+1,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
9305c6c1daeSBarry Smith   for (i=0; i<n; i++) {
93178fbdcc8SBarry Smith     ierr = PetscViewerBinaryWrite(viewer,(void*)data[i],sizes[i+1],PETSC_CHAR,PETSC_FALSE);CHKERRQ(ierr);
9325c6c1daeSBarry Smith   }
9335c6c1daeSBarry Smith   ierr = PetscFree(sizes);CHKERRQ(ierr);
9345c6c1daeSBarry Smith   PetscFunctionReturn(0);
9355c6c1daeSBarry Smith }
9365c6c1daeSBarry Smith 
9375c6c1daeSBarry Smith /*@C
9385c6c1daeSBarry Smith    PetscViewerBinaryReadStringArray - reads a binary file an array of strings
9395c6c1daeSBarry Smith 
9405c6c1daeSBarry Smith    Collective on MPI_Comm
9415c6c1daeSBarry Smith 
9425c6c1daeSBarry Smith    Input Parameter:
9435c6c1daeSBarry Smith .  viewer - the binary viewer
9445c6c1daeSBarry Smith 
9455c6c1daeSBarry Smith    Output Parameter:
9465c6c1daeSBarry Smith .  data - location of the array of strings
9475c6c1daeSBarry Smith 
9485c6c1daeSBarry Smith    Level: intermediate
9495c6c1daeSBarry Smith 
9505c6c1daeSBarry Smith    Concepts: binary files
9515c6c1daeSBarry Smith 
9525c6c1daeSBarry Smith     Notes: array of strings is null terminated
9535c6c1daeSBarry Smith 
9546a9046bcSBarry Smith .seealso: PetscViewerASCIIOpen(), PetscViewerPushFormat(), PetscViewerDestroy(),
9555c6c1daeSBarry Smith           VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(),
9565c6c1daeSBarry Smith           PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead()
9575c6c1daeSBarry Smith @*/
9585c6c1daeSBarry Smith PetscErrorCode PetscViewerBinaryReadStringArray(PetscViewer viewer,char ***data)
9595c6c1daeSBarry Smith {
9605c6c1daeSBarry Smith   PetscErrorCode ierr;
961060da220SMatthew G. Knepley   PetscInt       i,n,*sizes,N = 0;
9625c6c1daeSBarry Smith 
963c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
9645c6c1daeSBarry Smith   /* count number of strings */
965060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,&n,1,NULL,PETSC_INT);CHKERRQ(ierr);
966785e854fSJed Brown   ierr = PetscMalloc1(n,&sizes);CHKERRQ(ierr);
967060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,sizes,n,NULL,PETSC_INT);CHKERRQ(ierr);
968a297a907SKarl Rupp   for (i=0; i<n; i++) N += sizes[i];
969854ce69bSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(char*) + N*sizeof(char),data);CHKERRQ(ierr);
9705c6c1daeSBarry Smith   (*data)[0] = (char*)((*data) + n + 1);
971a297a907SKarl Rupp   for (i=1; i<n; i++) (*data)[i] = (*data)[i-1] + sizes[i-1];
972060da220SMatthew G. Knepley   ierr = PetscViewerBinaryRead(viewer,(*data)[0],N,NULL,PETSC_CHAR);CHKERRQ(ierr);
973a297a907SKarl Rupp 
9745c6c1daeSBarry Smith   (*data)[n] = 0;
975a297a907SKarl Rupp 
9765c6c1daeSBarry Smith   ierr = PetscFree(sizes);CHKERRQ(ierr);
9775c6c1daeSBarry Smith   PetscFunctionReturn(0);
9785c6c1daeSBarry Smith }
9795c6c1daeSBarry Smith 
98081f0254dSBarry Smith static PetscErrorCode PetscViewerFileGetName_Binary(PetscViewer viewer,const char **name)
9815c6c1daeSBarry Smith {
9825c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
9835c6c1daeSBarry Smith 
9845c6c1daeSBarry Smith   PetscFunctionBegin;
9855c6c1daeSBarry Smith   *name = vbinary->filename;
9865c6c1daeSBarry Smith   PetscFunctionReturn(0);
9875c6c1daeSBarry Smith }
9885c6c1daeSBarry Smith 
9895c6c1daeSBarry Smith /*@C
9905c6c1daeSBarry Smith      PetscViewerFileGetMode - Gets the type of file to be open
9915c6c1daeSBarry Smith 
9925c6c1daeSBarry Smith     Not Collective
9935c6c1daeSBarry Smith 
9945c6c1daeSBarry Smith   Input Parameter:
995232ac771SBarry Smith .  viewer - the PetscViewer; must be a PETSCVIEWERBINARY, PETSCVIEWERMATLAB, PETSCVIEWERHDF5, or PETSCVIEWERASCII  PetscViewer
9965c6c1daeSBarry Smith 
9975c6c1daeSBarry Smith   Output Parameter:
9985c6c1daeSBarry Smith .  type - type of file
9995c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
10005c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
10015c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
10025c6c1daeSBarry Smith 
10035c6c1daeSBarry Smith   Level: advanced
10045c6c1daeSBarry Smith 
10055c6c1daeSBarry Smith .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen()
10065c6c1daeSBarry Smith 
10075c6c1daeSBarry Smith @*/
10085c6c1daeSBarry Smith PetscErrorCode PetscViewerFileGetMode(PetscViewer viewer,PetscFileMode *type)
10095c6c1daeSBarry Smith {
10105c6c1daeSBarry Smith   PetscErrorCode ierr;
10115c6c1daeSBarry Smith 
10125c6c1daeSBarry Smith   PetscFunctionBegin;
10135c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
10145c6c1daeSBarry Smith   PetscValidPointer(type,2);
10155c6c1daeSBarry Smith   ierr = PetscUseMethod(viewer,"PetscViewerFileGetMode_C",(PetscViewer,PetscFileMode*),(viewer,type));CHKERRQ(ierr);
10165c6c1daeSBarry Smith   PetscFunctionReturn(0);
10175c6c1daeSBarry Smith }
10185c6c1daeSBarry Smith 
10195c6c1daeSBarry Smith /*@
1020bc196f7cSDave May     PetscViewerBinarySetUseMPIIO - Sets a binary viewer to use MPI-IO for reading/writing. Must be called
10215c6c1daeSBarry Smith         before PetscViewerFileSetName()
10225c6c1daeSBarry Smith 
10235c6c1daeSBarry Smith     Logically Collective on PetscViewer
10245c6c1daeSBarry Smith 
10255c6c1daeSBarry Smith     Input Parameters:
1026bc196f7cSDave May +   viewer - the PetscViewer; must be a binary
1027bc196f7cSDave May -   flg - PETSC_TRUE means MPI-IO will be used
10285c6c1daeSBarry Smith 
10295c6c1daeSBarry Smith     Options Database:
10305c6c1daeSBarry Smith     -viewer_binary_mpiio : Flag for using MPI-IO
10315c6c1daeSBarry Smith 
10325c6c1daeSBarry Smith     Level: advanced
10335c6c1daeSBarry Smith 
1034bc196f7cSDave May .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen(),
1035bc196f7cSDave May           PetscViewerBinaryGetUseMPIIO()
10365c6c1daeSBarry Smith 
10375c6c1daeSBarry Smith @*/
1038bc196f7cSDave May PetscErrorCode PetscViewerBinarySetUseMPIIO(PetscViewer viewer,PetscBool flg)
10395c6c1daeSBarry Smith {
10405c6c1daeSBarry Smith   PetscErrorCode ierr;
10415c6c1daeSBarry Smith 
10425c6c1daeSBarry Smith   PetscFunctionBegin;
10435c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
1044bc196f7cSDave May   ierr = PetscTryMethod(viewer,"PetscViewerBinarySetUseMPIIO_C",(PetscViewer,PetscBool),(viewer,flg));CHKERRQ(ierr);
10455c6c1daeSBarry Smith   PetscFunctionReturn(0);
10465c6c1daeSBarry Smith }
10475c6c1daeSBarry Smith 
10485c6c1daeSBarry Smith /*@C
10495c6c1daeSBarry Smith      PetscViewerFileSetMode - Sets the type of file to be open
10505c6c1daeSBarry Smith 
10515c6c1daeSBarry Smith     Logically Collective on PetscViewer
10525c6c1daeSBarry Smith 
10535c6c1daeSBarry Smith   Input Parameters:
1054232ac771SBarry Smith +  viewer - the PetscViewer; must be a a PETSCVIEWERBINARY, PETSCVIEWERMATLAB, PETSCVIEWERHDF5, or PETSCVIEWERASCII  PetscViewer
10555c6c1daeSBarry Smith -  type - type of file
10565c6c1daeSBarry Smith $    FILE_MODE_WRITE - create new file for binary output
10575c6c1daeSBarry Smith $    FILE_MODE_READ - open existing file for binary input
10585c6c1daeSBarry Smith $    FILE_MODE_APPEND - open existing file for binary output
10595c6c1daeSBarry Smith 
10605c6c1daeSBarry Smith   Level: advanced
10615c6c1daeSBarry Smith 
10625c6c1daeSBarry Smith .seealso: PetscViewerFileSetMode(), PetscViewerCreate(), PetscViewerSetType(), PetscViewerBinaryOpen()
10635c6c1daeSBarry Smith 
10645c6c1daeSBarry Smith @*/
10655c6c1daeSBarry Smith PetscErrorCode PetscViewerFileSetMode(PetscViewer viewer,PetscFileMode type)
10665c6c1daeSBarry Smith {
10675c6c1daeSBarry Smith   PetscErrorCode ierr;
10685c6c1daeSBarry Smith 
10695c6c1daeSBarry Smith   PetscFunctionBegin;
10705c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
10715c6c1daeSBarry Smith   PetscValidLogicalCollectiveEnum(viewer,type,2);
10725c6c1daeSBarry Smith   ierr = PetscTryMethod(viewer,"PetscViewerFileSetMode_C",(PetscViewer,PetscFileMode),(viewer,type));CHKERRQ(ierr);
10735c6c1daeSBarry Smith   PetscFunctionReturn(0);
10745c6c1daeSBarry Smith }
10755c6c1daeSBarry Smith 
107681f0254dSBarry Smith static PetscErrorCode PetscViewerFileGetMode_Binary(PetscViewer viewer,PetscFileMode *type)
10775c6c1daeSBarry Smith {
10785c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
10795c6c1daeSBarry Smith 
10805c6c1daeSBarry Smith   PetscFunctionBegin;
10815c6c1daeSBarry Smith   *type = vbinary->btype;
10825c6c1daeSBarry Smith   PetscFunctionReturn(0);
10835c6c1daeSBarry Smith }
10845c6c1daeSBarry Smith 
108581f0254dSBarry Smith static PetscErrorCode PetscViewerFileSetMode_Binary(PetscViewer viewer,PetscFileMode type)
10865c6c1daeSBarry Smith {
10875c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
10885c6c1daeSBarry Smith 
10895c6c1daeSBarry Smith   PetscFunctionBegin;
10905c6c1daeSBarry Smith   vbinary->btype = type;
10915c6c1daeSBarry Smith   PetscFunctionReturn(0);
10925c6c1daeSBarry Smith }
10935c6c1daeSBarry Smith 
109481f0254dSBarry Smith static PetscErrorCode PetscViewerFileSetName_Binary(PetscViewer viewer,const char name[])
1095e0385b85SDave May {
1096e0385b85SDave May   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
1097e0385b85SDave May   PetscErrorCode     ierr;
1098e0385b85SDave May 
1099e0385b85SDave May   PetscFunctionBegin;
1100*f90597f1SStefano Zampini   if (vbinary->filename) {
1101*f90597f1SStefano Zampini     /* gzip can be run after the file with the previous filename has been closed */
1102*f90597f1SStefano Zampini     ierr = PetscFree(vbinary->ogzfilename);CHKERRQ(ierr);
1103*f90597f1SStefano Zampini     ierr = PetscStrallocpy(vbinary->filename,&vbinary->ogzfilename);CHKERRQ(ierr);
1104*f90597f1SStefano Zampini     ierr = PetscFree(vbinary->filename);CHKERRQ(ierr);
1105*f90597f1SStefano Zampini     viewer->setupcalled = PETSC_FALSE;
1106*f90597f1SStefano Zampini   }
1107e0385b85SDave May   ierr = PetscStrallocpy(name,&vbinary->filename);CHKERRQ(ierr);
1108e0385b85SDave May   PetscFunctionReturn(0);
1109e0385b85SDave May }
11105c6c1daeSBarry Smith /*
11115c6c1daeSBarry Smith         Actually opens the file
11125c6c1daeSBarry Smith */
1113e0877f53SBarry Smith static PetscErrorCode PetscViewerFileSetUp_Binary(PetscViewer viewer)
11145c6c1daeSBarry Smith {
11155c6c1daeSBarry Smith   PetscMPIInt        rank;
11165c6c1daeSBarry Smith   PetscErrorCode     ierr;
11175c6c1daeSBarry Smith   size_t             len;
11185c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
11195c6c1daeSBarry Smith   const char         *fname;
11202259a519SDave May   char               bname[PETSC_MAX_PATH_LEN],*gz;
11215c6c1daeSBarry Smith   PetscBool          found;
11225c6c1daeSBarry Smith   PetscFileMode      type = vbinary->btype;
11235c6c1daeSBarry Smith 
11245c6c1daeSBarry Smith   PetscFunctionBegin;
1125e0385b85SDave May   if (type == (PetscFileMode) -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetMode()");
11262259a519SDave May   if (!vbinary->filename) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetName()");
11275c6c1daeSBarry Smith   ierr = PetscViewerFileClose_Binary(viewer);CHKERRQ(ierr);
11285c6c1daeSBarry Smith 
1129ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
11305c6c1daeSBarry Smith 
11315c6c1daeSBarry Smith   /* if ends in .gz strip that off and note user wants file compressed */
11325c6c1daeSBarry Smith   vbinary->storecompressed = PETSC_FALSE;
11335c6c1daeSBarry Smith   if (!rank && type == FILE_MODE_WRITE) {
11345c6c1daeSBarry Smith     /* remove .gz if it ends library name */
11355c6c1daeSBarry Smith     ierr = PetscStrstr(vbinary->filename,".gz",&gz);CHKERRQ(ierr);
11365c6c1daeSBarry Smith     if (gz) {
11375c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
11385c6c1daeSBarry Smith       if (len == 3) {
11395c6c1daeSBarry Smith         *gz = 0;
11405c6c1daeSBarry Smith         vbinary->storecompressed = PETSC_TRUE;
11415c6c1daeSBarry Smith       }
11425c6c1daeSBarry Smith     }
11435c6c1daeSBarry Smith   }
11445c6c1daeSBarry Smith 
11455c6c1daeSBarry Smith   /* only first processor opens file if writeable */
11465c6c1daeSBarry Smith   if (!rank || type == FILE_MODE_READ) {
11475c6c1daeSBarry Smith 
11485c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
11495c6c1daeSBarry Smith       /* possibly get the file from remote site or compressed file */
1150ce94432eSBarry Smith       ierr  = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),vbinary->filename,bname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
11515c6c1daeSBarry Smith       fname = bname;
11525c6c1daeSBarry Smith       if (!rank && !found) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot locate file: %s on node zero",vbinary->filename);
11535c6c1daeSBarry Smith       else if (!found) {
11545c6c1daeSBarry Smith         ierr  = PetscInfo(viewer,"Nonzero processor did not locate readonly file\n");CHKERRQ(ierr);
11555c6c1daeSBarry Smith         fname = 0;
11565c6c1daeSBarry Smith       }
1157a297a907SKarl Rupp     } else fname = vbinary->filename;
11585c6c1daeSBarry Smith 
11595c6c1daeSBarry Smith #if defined(PETSC_HAVE_O_BINARY)
11605c6c1daeSBarry Smith     if (type == FILE_MODE_WRITE) {
11615c6c1daeSBarry 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);
11625c6c1daeSBarry Smith     } else if (type == FILE_MODE_READ && fname) {
11635c6c1daeSBarry 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);
11645c6c1daeSBarry Smith     } else if (type == FILE_MODE_APPEND) {
11655c6c1daeSBarry 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);
11665c6c1daeSBarry Smith     } else if (fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type");
11675c6c1daeSBarry Smith #else
11685c6c1daeSBarry Smith     if (type == FILE_MODE_WRITE) {
11695c6c1daeSBarry Smith       if ((vbinary->fdes = creat(fname,0666)) == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot create file %s for writing",fname);
11705c6c1daeSBarry Smith     } else if (type == FILE_MODE_READ && fname) {
11715c6c1daeSBarry 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);
11725c6c1daeSBarry Smith     } else if (type == FILE_MODE_APPEND) {
11735c6c1daeSBarry 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);
11745c6c1daeSBarry Smith     } else if (fname) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Unknown file type");
11755c6c1daeSBarry Smith #endif
11765c6c1daeSBarry Smith   } else vbinary->fdes = -1;
11775c6c1daeSBarry Smith 
11785c6c1daeSBarry Smith   /*
11795c6c1daeSBarry Smith       try to open info file: all processors open this file if read only
11805c6c1daeSBarry Smith   */
11815c6c1daeSBarry Smith   if (!vbinary->skipinfo && (!rank || type == FILE_MODE_READ)) {
11825c6c1daeSBarry Smith     char infoname[PETSC_MAX_PATH_LEN],iname[PETSC_MAX_PATH_LEN];
11835c6c1daeSBarry Smith 
1184e0385b85SDave May     ierr = PetscStrcpy(infoname,vbinary->filename);CHKERRQ(ierr);
11855c6c1daeSBarry Smith     /* remove .gz if it ends library name */
11865c6c1daeSBarry Smith     ierr = PetscStrstr(infoname,".gz",&gz);CHKERRQ(ierr);
11875c6c1daeSBarry Smith     if (gz) {
11885c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
1189a297a907SKarl Rupp       if (len == 3) *gz = 0;
11905c6c1daeSBarry Smith     }
11915c6c1daeSBarry Smith 
11925c6c1daeSBarry Smith     ierr = PetscStrcat(infoname,".info");CHKERRQ(ierr);
11935c6c1daeSBarry Smith     ierr = PetscFixFilename(infoname,iname);CHKERRQ(ierr);
11945c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
1195ce94432eSBarry Smith       ierr = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),iname,infoname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
1196c5929fdfSBarry Smith       ierr = PetscOptionsInsertFile(PetscObjectComm((PetscObject)viewer),((PetscObject)viewer)->options,infoname,PETSC_FALSE);CHKERRQ(ierr);
11975c6c1daeSBarry Smith     } else {
11985c6c1daeSBarry Smith       vbinary->fdes_info = fopen(infoname,"w");
11995c6c1daeSBarry Smith       if (!vbinary->fdes_info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open .info file %s for writing",infoname);
12005c6c1daeSBarry Smith     }
12015c6c1daeSBarry Smith   }
12025c6c1daeSBarry Smith #if defined(PETSC_USE_LOG)
1203e0385b85SDave May   PetscLogObjectState((PetscObject)viewer,"File: %s",vbinary->filename);
12045c6c1daeSBarry Smith #endif
12055c6c1daeSBarry Smith   PetscFunctionReturn(0);
12065c6c1daeSBarry Smith }
12075c6c1daeSBarry Smith 
12085c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
1209e0385b85SDave May static PetscErrorCode PetscViewerFileSetUp_BinaryMPIIO(PetscViewer viewer)
12105c6c1daeSBarry Smith {
12115c6c1daeSBarry Smith   PetscMPIInt        rank;
12125c6c1daeSBarry Smith   PetscErrorCode     ierr;
12135c6c1daeSBarry Smith   size_t             len;
12145c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
12152259a519SDave May   char               *gz;
12165c6c1daeSBarry Smith   PetscBool          found;
12175c6c1daeSBarry Smith   PetscFileMode      type = vbinary->btype;
12185c6c1daeSBarry Smith 
12195c6c1daeSBarry Smith   PetscFunctionBegin;
1220e0385b85SDave May   if (type == (PetscFileMode) -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetMode()");
12212259a519SDave May   if (!vbinary->filename) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerFileSetName()");
1222e0385b85SDave May   ierr = PetscViewerFileClose_BinaryMPIIO(viewer);CHKERRQ(ierr);
12235c6c1daeSBarry Smith 
1224ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr);
12255c6c1daeSBarry Smith 
1226a297a907SKarl Rupp   vbinary->storecompressed = PETSC_FALSE;
12275c6c1daeSBarry Smith 
12285c6c1daeSBarry Smith   /* only first processor opens file if writeable */
12295c6c1daeSBarry Smith   if (type == FILE_MODE_READ) {
1230ce94432eSBarry Smith     MPI_File_open(PetscObjectComm((PetscObject)viewer),vbinary->filename,MPI_MODE_RDONLY,MPI_INFO_NULL,&vbinary->mfdes);CHKERRQ(ierr);
12315c6c1daeSBarry Smith   } else if (type == FILE_MODE_WRITE) {
1232ce94432eSBarry Smith     MPI_File_open(PetscObjectComm((PetscObject)viewer),vbinary->filename,MPI_MODE_WRONLY | MPI_MODE_CREATE,MPI_INFO_NULL,&vbinary->mfdes);CHKERRQ(ierr);
12335c6c1daeSBarry Smith   }
12345c6c1daeSBarry Smith 
12355c6c1daeSBarry Smith   /*
12365c6c1daeSBarry Smith       try to open info file: all processors open this file if read only
12375c6c1daeSBarry Smith 
1238bebe2cf6SSatish Balay       Below is identical code to the code for Binary above, should be put in separate routine
12395c6c1daeSBarry Smith   */
12405c6c1daeSBarry Smith   if (!vbinary->skipinfo && (!rank || type == FILE_MODE_READ)) {
12415c6c1daeSBarry Smith     char infoname[PETSC_MAX_PATH_LEN],iname[PETSC_MAX_PATH_LEN];
12425c6c1daeSBarry Smith 
1243e0385b85SDave May     ierr = PetscStrcpy(infoname,vbinary->filename);CHKERRQ(ierr);
12445c6c1daeSBarry Smith     /* remove .gz if it ends library name */
12455c6c1daeSBarry Smith     ierr = PetscStrstr(infoname,".gz",&gz);CHKERRQ(ierr);
12465c6c1daeSBarry Smith     if (gz) {
12475c6c1daeSBarry Smith       ierr = PetscStrlen(gz,&len);CHKERRQ(ierr);
1248a297a907SKarl Rupp       if (len == 3) *gz = 0;
12495c6c1daeSBarry Smith     }
12505c6c1daeSBarry Smith 
12515c6c1daeSBarry Smith     ierr = PetscStrcat(infoname,".info");CHKERRQ(ierr);
12525c6c1daeSBarry Smith     ierr = PetscFixFilename(infoname,iname);CHKERRQ(ierr);
12535c6c1daeSBarry Smith     if (type == FILE_MODE_READ) {
1254ce94432eSBarry Smith       ierr = PetscFileRetrieve(PetscObjectComm((PetscObject)viewer),iname,infoname,PETSC_MAX_PATH_LEN,&found);CHKERRQ(ierr);
1255c5929fdfSBarry Smith       ierr = PetscOptionsInsertFile(PetscObjectComm((PetscObject)viewer),((PetscObject)viewer)->options,infoname,PETSC_FALSE);CHKERRQ(ierr);
12565c6c1daeSBarry Smith     } else {
12575c6c1daeSBarry Smith       vbinary->fdes_info = fopen(infoname,"w");
12585c6c1daeSBarry Smith       if (!vbinary->fdes_info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open .info file %s for writing",infoname);
12595c6c1daeSBarry Smith     }
12605c6c1daeSBarry Smith   }
12615c6c1daeSBarry Smith #if defined(PETSC_USE_LOG)
1262e0385b85SDave May   PetscLogObjectState((PetscObject)viewer,"File: %s",vbinary->filename);
12635c6c1daeSBarry Smith #endif
12645c6c1daeSBarry Smith   PetscFunctionReturn(0);
12655c6c1daeSBarry Smith }
12665c6c1daeSBarry Smith 
126781f0254dSBarry Smith static PetscErrorCode PetscViewerBinarySetUseMPIIO_Binary(PetscViewer viewer,PetscBool flg)
12685c6c1daeSBarry Smith {
12695c6c1daeSBarry Smith   PetscViewer_Binary *vbinary = (PetscViewer_Binary*)viewer->data;
12705c6c1daeSBarry Smith   PetscFunctionBegin;
1271bc196f7cSDave May   vbinary->usempiio = flg;
12725c6c1daeSBarry Smith   PetscFunctionReturn(0);
12735c6c1daeSBarry Smith }
12745c6c1daeSBarry Smith #endif
12755c6c1daeSBarry Smith 
127681f0254dSBarry Smith static PetscErrorCode PetscViewerView_Binary(PetscViewer v,PetscViewer viewer)
12772bf49c77SBarry Smith {
12782bf49c77SBarry Smith   PetscErrorCode     ierr;
12792bf49c77SBarry Smith   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
12802bf49c77SBarry Smith 
12812bf49c77SBarry Smith   PetscFunctionBegin;
12822bf49c77SBarry Smith   if (binary->filename) {
12832bf49c77SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"Filename: %s\n",binary->filename);CHKERRQ(ierr);
12842bf49c77SBarry Smith   }
12852bf49c77SBarry Smith   PetscFunctionReturn(0);
12862bf49c77SBarry Smith }
12872bf49c77SBarry Smith 
128803a1d59fSDave May static PetscErrorCode PetscViewerSetUp_Binary(PetscViewer v)
128903a1d59fSDave May {
129003a1d59fSDave May   PetscErrorCode     ierr;
129103a1d59fSDave May   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
1292e0385b85SDave May 
129303a1d59fSDave May   PetscFunctionBegin;
1294da07bb01SDave May   if (!binary->setfromoptionscalled) { ierr = PetscViewerSetFromOptions(v);CHKERRQ(ierr); }
129503a1d59fSDave May 
1296e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
1297bc196f7cSDave May   if (binary->usempiio) {
1298e0385b85SDave May     ierr = PetscViewerFileSetUp_BinaryMPIIO(v);CHKERRQ(ierr);
1299e0385b85SDave May   } else {
1300e0385b85SDave May #endif
1301e0385b85SDave May     ierr = PetscViewerFileSetUp_Binary(v);CHKERRQ(ierr);
1302e0385b85SDave May #if defined(PETSC_HAVE_MPIIO)
1303e0385b85SDave May   }
1304e0385b85SDave May #endif
130503a1d59fSDave May   PetscFunctionReturn(0);
130603a1d59fSDave May }
130703a1d59fSDave May 
13084416b707SBarry Smith static PetscErrorCode PetscViewerSetFromOptions_Binary(PetscOptionItems *PetscOptionsObject,PetscViewer v)
130903a1d59fSDave May {
131003a1d59fSDave May   PetscErrorCode     ierr;
131103a1d59fSDave May   PetscViewer_Binary *binary = (PetscViewer_Binary*)v->data;
1312d22fd6bcSDave May   char               defaultname[PETSC_MAX_PATH_LEN];
131303a1d59fSDave May   PetscBool          flg;
1314e0385b85SDave May 
131503a1d59fSDave May   PetscFunctionBegin;
131603a1d59fSDave May   ierr = PetscOptionsHead(PetscOptionsObject,"Binary PetscViewer Options");CHKERRQ(ierr);
1317d22fd6bcSDave May   ierr = PetscSNPrintf(defaultname,PETSC_MAX_PATH_LEN-1,"binaryoutput");CHKERRQ(ierr);
1318d22fd6bcSDave May   ierr = PetscOptionsString("-viewer_binary_filename","Specify filename","PetscViewerFileSetName",defaultname,defaultname,PETSC_MAX_PATH_LEN-1,&flg);CHKERRQ(ierr);
1319d22fd6bcSDave May   if (flg) { ierr = PetscViewerFileSetName_Binary(v,defaultname);CHKERRQ(ierr); }
132034a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_info","Skip writing/reading .info file","PetscViewerBinarySetSkipInfo",PETSC_FALSE,&binary->skipinfo,NULL);CHKERRQ(ierr);
132134a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_options","Skip parsing vec load options","PetscViewerBinarySetSkipOptions",PETSC_TRUE,&binary->skipoptions,NULL);CHKERRQ(ierr);
132234a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_skip_header","Skip writing/reading header information","PetscViewerBinarySetSkipHeader",PETSC_FALSE,&binary->skipheader,NULL);CHKERRQ(ierr);
132303a1d59fSDave May #if defined(PETSC_HAVE_MPIIO)
132434a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_mpiio","Use MPI-IO functionality to write/read binary file","PetscViewerBinarySetUseMPIIO",PETSC_FALSE,&binary->usempiio,NULL);CHKERRQ(ierr);
132534a9cc2cSBarry Smith #elif defined(PETSC_HAVE_MPIUNI)
132634a9cc2cSBarry Smith   ierr = PetscOptionsBool("-viewer_binary_mpiio","Use MPI-IO functionality to write/read binary file","PetscViewerBinarySetUseMPIIO",PETSC_FALSE,NULL,NULL);CHKERRQ(ierr);
132703a1d59fSDave May #endif
132803a1d59fSDave May   ierr = PetscOptionsTail();CHKERRQ(ierr);
1329bc196f7cSDave May   binary->setfromoptionscalled = PETSC_TRUE;
133003a1d59fSDave May   PetscFunctionReturn(0);
133103a1d59fSDave May }
133203a1d59fSDave May 
13338556b5ebSBarry Smith /*MC
13348556b5ebSBarry Smith    PETSCVIEWERBINARY - A viewer that saves to binary files
13358556b5ebSBarry Smith 
13368556b5ebSBarry Smith 
13378556b5ebSBarry Smith .seealso:  PetscViewerBinaryOpen(), PETSC_VIEWER_STDOUT_(),PETSC_VIEWER_STDOUT_SELF, PETSC_VIEWER_STDOUT_WORLD, PetscViewerCreate(), PetscViewerASCIIOpen(),
13388556b5ebSBarry Smith            PetscViewerMatlabOpen(), VecView(), DMView(), PetscViewerMatlabPutArray(), PETSCVIEWERASCII, PETSCVIEWERMATLAB, PETSCVIEWERDRAW,
133967918a83SBarry Smith            PetscViewerFileSetName(), PetscViewerFileSetMode(), PetscViewerFormat, PetscViewerType, PetscViewerSetType(),
134067918a83SBarry Smith            PetscViewerBinaryGetUseMPIIO(), PetscViewerBinarySetUseMPIIO()
13418556b5ebSBarry Smith 
13421b266c99SBarry Smith   Level: beginner
13431b266c99SBarry Smith 
13448556b5ebSBarry Smith M*/
13458556b5ebSBarry Smith 
13468cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PetscViewerCreate_Binary(PetscViewer v)
13475c6c1daeSBarry Smith {
13485c6c1daeSBarry Smith   PetscErrorCode     ierr;
13495c6c1daeSBarry Smith   PetscViewer_Binary *vbinary;
13505c6c1daeSBarry Smith 
13515c6c1daeSBarry Smith   PetscFunctionBegin;
1352b00a9115SJed Brown   ierr                     = PetscNewLog(v,&vbinary);CHKERRQ(ierr);
13535c6c1daeSBarry Smith   v->data                  = (void*)vbinary;
135403a1d59fSDave May   v->ops->setfromoptions   = PetscViewerSetFromOptions_Binary;
13555c6c1daeSBarry Smith   v->ops->destroy          = PetscViewerDestroy_Binary;
13562bf49c77SBarry Smith   v->ops->view             = PetscViewerView_Binary;
1357c98fd787SBarry Smith   v->ops->setup            = PetscViewerSetUp_Binary;
1358c98fd787SBarry Smith   v->ops->flush            = NULL;
13595c6c1daeSBarry Smith   vbinary->fdes_info       = 0;
13605c6c1daeSBarry Smith   vbinary->fdes            = 0;
13615c6c1daeSBarry Smith   vbinary->skipinfo        = PETSC_FALSE;
13625c6c1daeSBarry Smith   vbinary->skipoptions     = PETSC_TRUE;
13635c6c1daeSBarry Smith   vbinary->skipheader      = PETSC_FALSE;
1364da07bb01SDave May   vbinary->setfromoptionscalled = PETSC_FALSE;
1365559f443fSBarry Smith   v->ops->getsubviewer     = PetscViewerGetSubViewer_Binary;
1366559f443fSBarry Smith   v->ops->restoresubviewer = PetscViewerRestoreSubViewer_Binary;
13671d641e7bSMichael Lange   v->ops->read             = PetscViewerBinaryRead;
13685c6c1daeSBarry Smith   vbinary->btype           = (PetscFileMode) -1;
13695c6c1daeSBarry Smith   vbinary->storecompressed = PETSC_FALSE;
1370*f90597f1SStefano Zampini   vbinary->filename        = NULL;
1371*f90597f1SStefano Zampini   vbinary->ogzfilename     = NULL;
13725c6c1daeSBarry Smith   vbinary->flowcontrol     = 256; /* seems a good number for Cray XT-5 */
13735c6c1daeSBarry Smith 
1374bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetFlowControl_C",PetscViewerBinaryGetFlowControl_Binary);CHKERRQ(ierr);
1375bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetFlowControl_C",PetscViewerBinarySetFlowControl_Binary);CHKERRQ(ierr);
1376bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipHeader_C",PetscViewerBinarySetSkipHeader_Binary);CHKERRQ(ierr);
1377bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipHeader_C",PetscViewerBinaryGetSkipHeader_Binary);CHKERRQ(ierr);
1378807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipOptions_C",PetscViewerBinaryGetSkipOptions_Binary);CHKERRQ(ierr);
1379807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipOptions_C",PetscViewerBinarySetSkipOptions_Binary);CHKERRQ(ierr);
1380807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetSkipInfo_C",PetscViewerBinaryGetSkipInfo_Binary);CHKERRQ(ierr);
1381807ea322SDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetSkipInfo_C",PetscViewerBinarySetSkipInfo_Binary);CHKERRQ(ierr);
1382bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetInfoPointer_C",PetscViewerBinaryGetInfoPointer_Binary);CHKERRQ(ierr);
1383bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetName_C",PetscViewerFileSetName_Binary);CHKERRQ(ierr);
1384bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_Binary);CHKERRQ(ierr);
1385bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetMode_C",PetscViewerFileGetMode_Binary);CHKERRQ(ierr);
1386bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerFileGetName_C",PetscViewerFileGetName_Binary);CHKERRQ(ierr);
13875c6c1daeSBarry Smith #if defined(PETSC_HAVE_MPIIO)
1388bc196f7cSDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinaryGetUseMPIIO_C",PetscViewerBinaryGetUseMPIIO_Binary);CHKERRQ(ierr);
1389bc196f7cSDave May   ierr = PetscObjectComposeFunction((PetscObject)v,"PetscViewerBinarySetUseMPIIO_C",PetscViewerBinarySetUseMPIIO_Binary);CHKERRQ(ierr);
13905c6c1daeSBarry Smith #endif
13915c6c1daeSBarry Smith   PetscFunctionReturn(0);
13925c6c1daeSBarry Smith }
13935c6c1daeSBarry Smith 
13945c6c1daeSBarry Smith /* ---------------------------------------------------------------------*/
13955c6c1daeSBarry Smith /*
13965c6c1daeSBarry Smith     The variable Petsc_Viewer_Binary_keyval is used to indicate an MPI attribute that
13975c6c1daeSBarry Smith   is attached to a communicator, in this case the attribute is a PetscViewer.
13985c6c1daeSBarry Smith */
1399d4c7638eSBarry Smith PetscMPIInt Petsc_Viewer_Binary_keyval = MPI_KEYVAL_INVALID;
14005c6c1daeSBarry Smith 
14015c6c1daeSBarry Smith /*@C
14025c6c1daeSBarry Smith      PETSC_VIEWER_BINARY_ - Creates a binary PetscViewer shared by all processors
14035c6c1daeSBarry Smith                      in a communicator.
14045c6c1daeSBarry Smith 
14055c6c1daeSBarry Smith      Collective on MPI_Comm
14065c6c1daeSBarry Smith 
14075c6c1daeSBarry Smith      Input Parameter:
14085c6c1daeSBarry Smith .    comm - the MPI communicator to share the binary PetscViewer
14095c6c1daeSBarry Smith 
14105c6c1daeSBarry Smith      Level: intermediate
14115c6c1daeSBarry Smith 
14125c6c1daeSBarry Smith    Options Database Keys:
14135c6c1daeSBarry Smith +    -viewer_binary_filename <name>
14145c6c1daeSBarry Smith .    -viewer_binary_skip_info
1415a2d7db39SDave May .    -viewer_binary_skip_options
1416a2d7db39SDave May .    -viewer_binary_skip_header
1417a2d7db39SDave May -    -viewer_binary_mpiio
14185c6c1daeSBarry Smith 
14195c6c1daeSBarry Smith    Environmental variables:
14205c6c1daeSBarry Smith -   PETSC_VIEWER_BINARY_FILENAME
14215c6c1daeSBarry Smith 
14225c6c1daeSBarry Smith      Notes:
14235c6c1daeSBarry Smith      Unlike almost all other PETSc routines, PETSC_VIEWER_BINARY_ does not return
14245c6c1daeSBarry Smith      an error code.  The binary PetscViewer is usually used in the form
14255c6c1daeSBarry Smith $       XXXView(XXX object,PETSC_VIEWER_BINARY_(comm));
14265c6c1daeSBarry Smith 
14275c6c1daeSBarry Smith .seealso: PETSC_VIEWER_BINARY_WORLD, PETSC_VIEWER_BINARY_SELF, PetscViewerBinaryOpen(), PetscViewerCreate(),
14285c6c1daeSBarry Smith           PetscViewerDestroy()
14295c6c1daeSBarry Smith @*/
14305c6c1daeSBarry Smith PetscViewer PETSC_VIEWER_BINARY_(MPI_Comm comm)
14315c6c1daeSBarry Smith {
14325c6c1daeSBarry Smith   PetscErrorCode ierr;
14335c6c1daeSBarry Smith   PetscBool      flg;
14345c6c1daeSBarry Smith   PetscViewer    viewer;
14355c6c1daeSBarry Smith   char           fname[PETSC_MAX_PATH_LEN];
14365c6c1daeSBarry Smith   MPI_Comm       ncomm;
14375c6c1daeSBarry Smith 
14385c6c1daeSBarry Smith   PetscFunctionBegin;
1439efca3c55SSatish 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);}
14405c6c1daeSBarry Smith   if (Petsc_Viewer_Binary_keyval == MPI_KEYVAL_INVALID) {
14415c6c1daeSBarry Smith     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Viewer_Binary_keyval,0);
1442efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14435c6c1daeSBarry Smith   }
14445c6c1daeSBarry Smith   ierr = MPI_Attr_get(ncomm,Petsc_Viewer_Binary_keyval,(void**)&viewer,(int*)&flg);
1445efca3c55SSatish Balay   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14465c6c1daeSBarry Smith   if (!flg) { /* PetscViewer not yet created */
14475c6c1daeSBarry Smith     ierr = PetscOptionsGetenv(ncomm,"PETSC_VIEWER_BINARY_FILENAME",fname,PETSC_MAX_PATH_LEN,&flg);
1448efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14495c6c1daeSBarry Smith     if (!flg) {
14505c6c1daeSBarry Smith       ierr = PetscStrcpy(fname,"binaryoutput");
1451efca3c55SSatish Balay       if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14525c6c1daeSBarry Smith     }
14535c6c1daeSBarry Smith     ierr = PetscViewerBinaryOpen(ncomm,fname,FILE_MODE_WRITE,&viewer);
1454efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14555c6c1daeSBarry Smith     ierr = PetscObjectRegisterDestroy((PetscObject)viewer);
1456efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14575c6c1daeSBarry Smith     ierr = MPI_Attr_put(ncomm,Petsc_Viewer_Binary_keyval,(void*)viewer);
1458efca3c55SSatish Balay     if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14595c6c1daeSBarry Smith   }
14605c6c1daeSBarry Smith   ierr = PetscCommDestroy(&ncomm);
1461efca3c55SSatish Balay   if (ierr) {PetscError(PETSC_COMM_SELF,__LINE__,"PETSC_VIEWER_BINARY_",__FILE__,PETSC_ERR_PLIB,PETSC_ERROR_INITIAL," ");PetscFunctionReturn(0);}
14625c6c1daeSBarry Smith   PetscFunctionReturn(viewer);
14635c6c1daeSBarry Smith }
1464