xref: /petsc/src/sys/classes/viewer/impls/ascii/filev.c (revision 1d031f67ac3d3f35cf7aa5933197a215ab0b045e)
1665c2dedSJed Brown #include <../src/sys/classes/viewer/impls/ascii/asciiimpl.h> /*I "petscviewer.h" I*/
25c6c1daeSBarry Smith 
35c6c1daeSBarry Smith #define QUEUESTRINGSIZE 8192
45c6c1daeSBarry Smith 
5d71ae5a4SJacob Faibussowitsch static PetscErrorCode PetscViewerFileClose_ASCII(PetscViewer viewer)
6d71ae5a4SJacob Faibussowitsch {
75c6c1daeSBarry Smith   PetscMPIInt        rank;
85c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
95c6c1daeSBarry Smith   int                err;
105c6c1daeSBarry Smith 
115c6c1daeSBarry Smith   PetscFunctionBegin;
1228b400f6SJacob Faibussowitsch   PetscCheck(!vascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
139566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
14dd400576SPatrick Sanan   if (rank == 0 && vascii->fd != stderr && vascii->fd != PETSC_STDOUT) {
155c6c1daeSBarry Smith     if (vascii->fd && vascii->closefile) {
165c6c1daeSBarry Smith       err = fclose(vascii->fd);
1728b400f6SJacob Faibussowitsch       PetscCheck(!err, PETSC_COMM_SELF, PETSC_ERR_SYS, "fclose() failed on file");
185c6c1daeSBarry Smith     }
195c6c1daeSBarry Smith     if (vascii->storecompressed) {
205c6c1daeSBarry Smith       char  par[PETSC_MAX_PATH_LEN], buf[PETSC_MAX_PATH_LEN];
215c6c1daeSBarry Smith       FILE *fp;
229566063dSJacob Faibussowitsch       PetscCall(PetscStrncpy(par, "gzip ", sizeof(par)));
239566063dSJacob Faibussowitsch       PetscCall(PetscStrlcat(par, vascii->filename, sizeof(par)));
245c6c1daeSBarry Smith #if defined(PETSC_HAVE_POPEN)
259566063dSJacob Faibussowitsch       PetscCall(PetscPOpen(PETSC_COMM_SELF, NULL, par, "r", &fp));
2600045ab3SPierre Jolivet       PetscCheck(!fgets(buf, 1024, fp), PETSC_COMM_SELF, PETSC_ERR_LIB, "Error from compression command %s %s", par, buf);
279566063dSJacob Faibussowitsch       PetscCall(PetscPClose(PETSC_COMM_SELF, fp));
285c6c1daeSBarry Smith #else
295c6c1daeSBarry Smith       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP_SYS, "Cannot run external programs on this machine");
305c6c1daeSBarry Smith #endif
315c6c1daeSBarry Smith     }
325c6c1daeSBarry Smith   }
339566063dSJacob Faibussowitsch   PetscCall(PetscFree(vascii->filename));
343ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
355c6c1daeSBarry Smith }
365c6c1daeSBarry Smith 
375c6c1daeSBarry Smith /* ----------------------------------------------------------------------*/
3834e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerDestroy_ASCII(PetscViewer viewer)
39d71ae5a4SJacob Faibussowitsch {
405c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
415c6c1daeSBarry Smith   PetscViewerLink   *vlink;
425c6c1daeSBarry Smith   PetscBool          flg;
435c6c1daeSBarry Smith 
445c6c1daeSBarry Smith   PetscFunctionBegin;
4528b400f6SJacob Faibussowitsch   PetscCheck(!vascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
469566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileClose_ASCII(viewer));
479566063dSJacob Faibussowitsch   PetscCall(PetscFree(vascii));
485c6c1daeSBarry Smith 
495c6c1daeSBarry Smith   /* remove the viewer from the list in the MPI Communicator */
5048a46eb9SPierre Jolivet   if (Petsc_Viewer_keyval == MPI_KEYVAL_INVALID) PetscCallMPI(MPI_Comm_create_keyval(MPI_COMM_NULL_COPY_FN, Petsc_DelViewer, &Petsc_Viewer_keyval, (void *)0));
515c6c1daeSBarry Smith 
529566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_get_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_keyval, (void **)&vlink, (PetscMPIInt *)&flg));
535c6c1daeSBarry Smith   if (flg) {
545c6c1daeSBarry Smith     if (vlink && vlink->viewer == viewer) {
55e5840a18SBarry Smith       if (vlink->next) {
569566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_set_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_keyval, vlink->next));
57e5840a18SBarry Smith       } else {
589566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Comm_delete_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_keyval));
59e5840a18SBarry Smith       }
609566063dSJacob Faibussowitsch       PetscCall(PetscFree(vlink));
615c6c1daeSBarry Smith     } else {
625c6c1daeSBarry Smith       while (vlink && vlink->next) {
635c6c1daeSBarry Smith         if (vlink->next->viewer == viewer) {
645c6c1daeSBarry Smith           PetscViewerLink *nv = vlink->next;
655c6c1daeSBarry Smith           vlink->next         = vlink->next->next;
669566063dSJacob Faibussowitsch           PetscCall(PetscFree(nv));
675c6c1daeSBarry Smith         }
685c6c1daeSBarry Smith         vlink = vlink->next;
695c6c1daeSBarry Smith       }
705c6c1daeSBarry Smith     }
715c6c1daeSBarry Smith   }
72aa139df6SJed Brown 
73aa139df6SJed Brown   if (Petsc_Viewer_Stdout_keyval != MPI_KEYVAL_INVALID) {
74aa139df6SJed Brown     PetscViewer aviewer;
759566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_get_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_Stdout_keyval, (void **)&aviewer, (PetscMPIInt *)&flg));
7648a46eb9SPierre Jolivet     if (flg && aviewer == viewer) PetscCallMPI(MPI_Comm_delete_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_Stdout_keyval));
77aa139df6SJed Brown   }
78aa139df6SJed Brown   if (Petsc_Viewer_Stderr_keyval != MPI_KEYVAL_INVALID) {
79aa139df6SJed Brown     PetscViewer aviewer;
809566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Comm_get_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_Stderr_keyval, (void **)&aviewer, (PetscMPIInt *)&flg));
8148a46eb9SPierre Jolivet     if (flg && aviewer == viewer) PetscCallMPI(MPI_Comm_delete_attr(PetscObjectComm((PetscObject)viewer), Petsc_Viewer_Stderr_keyval));
82aa139df6SJed Brown   }
832e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetName_C", NULL));
842e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetName_C", NULL));
852e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetMode_C", NULL));
862e956fe4SStefano Zampini   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetMode_C", NULL));
873ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
885c6c1daeSBarry Smith }
895c6c1daeSBarry Smith 
9034e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerDestroy_ASCII_SubViewer(PetscViewer viewer)
91d71ae5a4SJacob Faibussowitsch {
925c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
935fd66863SKarl Rupp 
945c6c1daeSBarry Smith   PetscFunctionBegin;
959566063dSJacob Faibussowitsch   PetscCall(PetscViewerRestoreSubViewer(vascii->bviewer, 0, &viewer));
963ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
975c6c1daeSBarry Smith }
985c6c1daeSBarry Smith 
9934e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerFlush_ASCII(PetscViewer viewer)
100d71ae5a4SJacob Faibussowitsch {
1015c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
102559f443fSBarry Smith   MPI_Comm           comm;
103559f443fSBarry Smith   PetscMPIInt        rank, size;
104559f443fSBarry Smith   FILE              *fd = vascii->fd;
1055c6c1daeSBarry Smith 
1065c6c1daeSBarry Smith   PetscFunctionBegin;
10728b400f6SJacob Faibussowitsch   PetscCheck(!vascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
1089566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)viewer, &comm));
1099566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
1109566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
111559f443fSBarry Smith 
112c69effb2SJacob Faibussowitsch   if (!vascii->bviewer && rank == 0 && (vascii->mode != FILE_MODE_READ)) PetscCall(PetscFFlush(vascii->fd));
1135c6c1daeSBarry Smith 
1145c6c1daeSBarry Smith   if (vascii->allowsynchronized) {
115559f443fSBarry Smith     PetscMPIInt tag, i, j, n = 0, dummy = 0;
116559f443fSBarry Smith     char       *message;
117559f443fSBarry Smith     MPI_Status  status;
118559f443fSBarry Smith 
1199566063dSJacob Faibussowitsch     PetscCall(PetscCommDuplicate(comm, &comm, &tag));
120559f443fSBarry Smith 
121559f443fSBarry Smith     /* First processor waits for messages from all other processors */
122dd400576SPatrick Sanan     if (rank == 0) {
123559f443fSBarry Smith       /* flush my own messages that I may have queued up */
124559f443fSBarry Smith       PrintfQueue next = vascii->petsc_printfqueuebase, previous;
125559f443fSBarry Smith       for (i = 0; i < vascii->petsc_printfqueuelength; i++) {
126559f443fSBarry Smith         if (!vascii->bviewer) {
1279566063dSJacob Faibussowitsch           PetscCall(PetscFPrintf(comm, fd, "%s", next->string));
128559f443fSBarry Smith         } else {
1299566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIISynchronizedPrintf(vascii->bviewer, "%s", next->string));
130559f443fSBarry Smith         }
131559f443fSBarry Smith         previous = next;
132559f443fSBarry Smith         next     = next->next;
1339566063dSJacob Faibussowitsch         PetscCall(PetscFree(previous->string));
1349566063dSJacob Faibussowitsch         PetscCall(PetscFree(previous));
135559f443fSBarry Smith       }
13602c9f0b5SLisandro Dalcin       vascii->petsc_printfqueue       = NULL;
137559f443fSBarry Smith       vascii->petsc_printfqueuelength = 0;
138559f443fSBarry Smith       for (i = 1; i < size; i++) {
139559f443fSBarry Smith         /* to prevent a flood of messages to process zero, request each message separately */
1409566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Send(&dummy, 1, MPI_INT, i, tag, comm));
1419566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Recv(&n, 1, MPI_INT, i, tag, comm, &status));
142559f443fSBarry Smith         for (j = 0; j < n; j++) {
143559f443fSBarry Smith           PetscMPIInt size = 0;
144559f443fSBarry Smith 
1459566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Recv(&size, 1, MPI_INT, i, tag, comm, &status));
1469566063dSJacob Faibussowitsch           PetscCall(PetscMalloc1(size, &message));
1479566063dSJacob Faibussowitsch           PetscCallMPI(MPI_Recv(message, size, MPI_CHAR, i, tag, comm, &status));
148559f443fSBarry Smith           if (!vascii->bviewer) {
1499566063dSJacob Faibussowitsch             PetscCall(PetscFPrintf(comm, fd, "%s", message));
150559f443fSBarry Smith           } else {
1519566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIISynchronizedPrintf(vascii->bviewer, "%s", message));
152559f443fSBarry Smith           }
1539566063dSJacob Faibussowitsch           PetscCall(PetscFree(message));
154559f443fSBarry Smith         }
155559f443fSBarry Smith       }
156559f443fSBarry Smith     } else { /* other processors send queue to processor 0 */
157559f443fSBarry Smith       PrintfQueue next = vascii->petsc_printfqueuebase, previous;
158559f443fSBarry Smith 
1599566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Recv(&dummy, 1, MPI_INT, 0, tag, comm, &status));
1609566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Send(&vascii->petsc_printfqueuelength, 1, MPI_INT, 0, tag, comm));
161559f443fSBarry Smith       for (i = 0; i < vascii->petsc_printfqueuelength; i++) {
1629566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Send(&next->size, 1, MPI_INT, 0, tag, comm));
1639566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Send(next->string, next->size, MPI_CHAR, 0, tag, comm));
164559f443fSBarry Smith         previous = next;
165559f443fSBarry Smith         next     = next->next;
1669566063dSJacob Faibussowitsch         PetscCall(PetscFree(previous->string));
1679566063dSJacob Faibussowitsch         PetscCall(PetscFree(previous));
168559f443fSBarry Smith       }
16902c9f0b5SLisandro Dalcin       vascii->petsc_printfqueue       = NULL;
170559f443fSBarry Smith       vascii->petsc_printfqueuelength = 0;
171559f443fSBarry Smith     }
1729566063dSJacob Faibussowitsch     PetscCall(PetscCommDestroy(&comm));
1735c6c1daeSBarry Smith   }
1743ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
1755c6c1daeSBarry Smith }
1765c6c1daeSBarry Smith 
1775c6c1daeSBarry Smith /*@C
178811af0c4SBarry Smith   PetscViewerASCIIGetPointer - Extracts the file pointer from an ASCII `PetscViewer`.
1795c6c1daeSBarry Smith 
18035cb6cd3SPierre Jolivet   Not Collective, depending on the viewer the value may be meaningless except for process 0 of the viewer; No Fortran Support
1815c6c1daeSBarry Smith 
182f8859db6SBarry Smith   Input Parameter:
1833f423023SBarry Smith . viewer - `PetscViewer` context, obtained from `PetscViewerASCIIOpen()`
184f8859db6SBarry Smith 
185f8859db6SBarry Smith   Output Parameter:
186f8859db6SBarry Smith . fd - file pointer
187f8859db6SBarry Smith 
1885c6c1daeSBarry Smith   Level: intermediate
1895c6c1daeSBarry Smith 
190811af0c4SBarry Smith   Note:
191c410d8ccSBarry Smith   For the standard `PETSCVIEWERASCII` the value is valid only on MPI rank 0 of the viewer
192811af0c4SBarry Smith 
1933f423023SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERASCII`, `PetscViewerASCIIOpen()`, `PetscViewerDestroy()`, `PetscViewerSetType()`,
1943f423023SBarry Smith           `PetscViewerCreate()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerFlush()`
1955c6c1daeSBarry Smith @*/
196d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIGetPointer(PetscViewer viewer, FILE **fd)
197d71ae5a4SJacob Faibussowitsch {
1985c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
1995c6c1daeSBarry Smith 
2005c6c1daeSBarry Smith   PetscFunctionBegin;
201c621c6acSBarry Smith   PetscCheck(!vascii->fileunit, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot request file pointer for viewers that use Fortran files");
2025c6c1daeSBarry Smith   *fd = vascii->fd;
2033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2045c6c1daeSBarry Smith }
2055c6c1daeSBarry Smith 
20634e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetMode_ASCII(PetscViewer viewer, PetscFileMode *mode)
207d71ae5a4SJacob Faibussowitsch {
2085c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
2095c6c1daeSBarry Smith 
2105c6c1daeSBarry Smith   PetscFunctionBegin;
2115c6c1daeSBarry Smith   *mode = vascii->mode;
2123ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2135c6c1daeSBarry Smith }
2145c6c1daeSBarry Smith 
21534e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetMode_ASCII(PetscViewer viewer, PetscFileMode mode)
216d71ae5a4SJacob Faibussowitsch {
2175c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
2185c6c1daeSBarry Smith 
2195c6c1daeSBarry Smith   PetscFunctionBegin;
2205c6c1daeSBarry Smith   vascii->mode = mode;
2213ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2225c6c1daeSBarry Smith }
2235c6c1daeSBarry Smith 
2245c6c1daeSBarry Smith /*
2255c6c1daeSBarry Smith    If petsc_history is on, then all Petsc*Printf() results are saved
2265c6c1daeSBarry Smith    if the appropriate (usually .petschistory) file.
2275c6c1daeSBarry Smith */
22895c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
2295c6c1daeSBarry Smith 
2305c6c1daeSBarry Smith /*@
2313f423023SBarry Smith   PetscViewerASCIISetTab - Causes `PetscViewer` to tab in a number of times before printing
2325c6c1daeSBarry Smith 
233cf53795eSBarry Smith   Not Collective, but only first processor in set has any effect; No Fortran Support
2345c6c1daeSBarry Smith 
2355c6c1daeSBarry Smith   Input Parameters:
236811af0c4SBarry Smith + viewer - obtained with `PetscViewerASCIIOpen()`
2375c6c1daeSBarry Smith - tabs   - number of tabs
2385c6c1daeSBarry Smith 
2395c6c1daeSBarry Smith   Level: developer
2405c6c1daeSBarry Smith 
2413f423023SBarry Smith   Note:
2423f423023SBarry Smith   `PetscViewerASCIIPushTab()` and `PetscViewerASCIIPopTab()` are the preferred usage
2433f423023SBarry Smith 
2443f423023SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERASCII`, `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
2453f423023SBarry Smith           `PetscViewerASCIIGetTab()`,
246db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
2473f423023SBarry Smith           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`,
2483f423023SBarry Smith           `PetscViewerASCIIPushTab()`
2495c6c1daeSBarry Smith @*/
250d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIISetTab(PetscViewer viewer, PetscInt tabs)
251d71ae5a4SJacob Faibussowitsch {
2525c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
2535c6c1daeSBarry Smith   PetscBool          iascii;
2545c6c1daeSBarry Smith 
2555c6c1daeSBarry Smith   PetscFunctionBegin;
2565c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
2579566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
258a297a907SKarl Rupp   if (iascii) ascii->tab = tabs;
2593ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2605c6c1daeSBarry Smith }
2615c6c1daeSBarry Smith 
2625c6c1daeSBarry Smith /*@
263811af0c4SBarry Smith   PetscViewerASCIIGetTab - Return the number of tabs used by `PetscViewer`.
2645c6c1daeSBarry Smith 
265cf53795eSBarry Smith   Not Collective, meaningful on first processor only; No Fortran Support
2665c6c1daeSBarry Smith 
26720f4b53cSBarry Smith   Input Parameter:
268811af0c4SBarry Smith . viewer - obtained with `PetscViewerASCIIOpen()`
269a2b725a8SWilliam Gropp 
27020f4b53cSBarry Smith   Output Parameter:
2715c6c1daeSBarry Smith . tabs - number of tabs
2725c6c1daeSBarry Smith 
2735c6c1daeSBarry Smith   Level: developer
2745c6c1daeSBarry Smith 
2753f423023SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERASCII`, `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
2763f423023SBarry Smith           `PetscViewerASCIISetTab()`,
277db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
278db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`, `PetscViewerASCIIPushTab()`
2795c6c1daeSBarry Smith @*/
280d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIGetTab(PetscViewer viewer, PetscInt *tabs)
281d71ae5a4SJacob Faibussowitsch {
2825c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
2835c6c1daeSBarry Smith   PetscBool          iascii;
2845c6c1daeSBarry Smith 
2855c6c1daeSBarry Smith   PetscFunctionBegin;
2865c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
2879566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
288a297a907SKarl Rupp   if (iascii && tabs) *tabs = ascii->tab;
2893ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
2905c6c1daeSBarry Smith }
2915c6c1daeSBarry Smith 
2925c6c1daeSBarry Smith /*@
2933f423023SBarry Smith   PetscViewerASCIIAddTab - Add to the number of times a `PETSCVIEWERASCII` viewer tabs before printing
2945c6c1daeSBarry Smith 
295cf53795eSBarry Smith   Not Collective, but only first processor in set has any effect; No Fortran Support
2965c6c1daeSBarry Smith 
2975c6c1daeSBarry Smith   Input Parameters:
298811af0c4SBarry Smith + viewer - obtained with `PetscViewerASCIIOpen()`
2995c6c1daeSBarry Smith - tabs   - number of tabs
3005c6c1daeSBarry Smith 
3015c6c1daeSBarry Smith   Level: developer
3025c6c1daeSBarry Smith 
3033f423023SBarry Smith   Note:
3043f423023SBarry Smith   `PetscViewerASCIIPushTab()` and `PetscViewerASCIIPopTab()` are the preferred usage
3053f423023SBarry Smith 
3063f423023SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERASCII`, `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
307db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
308db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`, `PetscViewerASCIIPushTab()`
3095c6c1daeSBarry Smith @*/
310d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIAddTab(PetscViewer viewer, PetscInt tabs)
311d71ae5a4SJacob Faibussowitsch {
3125c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
3135c6c1daeSBarry Smith   PetscBool          iascii;
3145c6c1daeSBarry Smith 
3155c6c1daeSBarry Smith   PetscFunctionBegin;
3165c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
3179566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
318a297a907SKarl Rupp   if (iascii) ascii->tab += tabs;
3193ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3205c6c1daeSBarry Smith }
3215c6c1daeSBarry Smith 
3225c6c1daeSBarry Smith /*@
3233f423023SBarry Smith   PetscViewerASCIISubtractTab - Subtracts from the number of times a `PETSCVIEWERASCII` viewer tabs before printing
3245c6c1daeSBarry Smith 
325cf53795eSBarry Smith   Not Collective, but only first processor in set has any effect; No Fortran Support
3265c6c1daeSBarry Smith 
3275c6c1daeSBarry Smith   Input Parameters:
328811af0c4SBarry Smith + viewer - obtained with `PetscViewerASCIIOpen()`
3295c6c1daeSBarry Smith - tabs   - number of tabs
3305c6c1daeSBarry Smith 
3315c6c1daeSBarry Smith   Level: developer
3325c6c1daeSBarry Smith 
3333f423023SBarry Smith   Note:
3343f423023SBarry Smith   `PetscViewerASCIIPushTab()` and `PetscViewerASCIIPopTab()` are the preferred usage
3353f423023SBarry Smith 
3363f423023SBarry Smith .seealso: [](sec_viewers), `PETSCVIEWERASCII`, `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
337db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
3383f423023SBarry Smith           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`,
3393f423023SBarry Smith           `PetscViewerASCIIPushTab()`
3405c6c1daeSBarry Smith @*/
341d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIISubtractTab(PetscViewer viewer, PetscInt tabs)
342d71ae5a4SJacob Faibussowitsch {
3435c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
3445c6c1daeSBarry Smith   PetscBool          iascii;
3455c6c1daeSBarry Smith 
3465c6c1daeSBarry Smith   PetscFunctionBegin;
3475c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
3489566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
349a297a907SKarl Rupp   if (iascii) ascii->tab -= tabs;
3503ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3515c6c1daeSBarry Smith }
3525c6c1daeSBarry Smith 
3535c6c1daeSBarry Smith /*@C
354811af0c4SBarry Smith   PetscViewerASCIIPushSynchronized - Allows calls to `PetscViewerASCIISynchronizedPrintf()` for this viewer
3555c6c1daeSBarry Smith 
356c3339decSBarry Smith   Collective
3575c6c1daeSBarry Smith 
35820f4b53cSBarry Smith   Input Parameter:
359811af0c4SBarry Smith . viewer - obtained with `PetscViewerASCIIOpen()`
3605c6c1daeSBarry Smith 
3615c6c1daeSBarry Smith   Level: intermediate
3625c6c1daeSBarry Smith 
363811af0c4SBarry Smith   Note:
364811af0c4SBarry Smith   See documentation of `PetscViewerASCIISynchronizedPrintf()` for more details how the synchronized output should be done properly.
3655c6c1daeSBarry Smith 
366d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerFlush()`, `PetscViewerASCIIPopSynchronized()`,
367db781477SPatrick Sanan           `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIIOpen()`,
368db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`
3695c6c1daeSBarry Smith @*/
370d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIPushSynchronized(PetscViewer viewer)
371d71ae5a4SJacob Faibussowitsch {
3725c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
3735c6c1daeSBarry Smith   PetscBool          iascii;
3745c6c1daeSBarry Smith 
3755c6c1daeSBarry Smith   PetscFunctionBegin;
3765c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
37728b400f6SJacob Faibussowitsch   PetscCheck(!ascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
3789566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
3791575c14dSBarry Smith   if (iascii) ascii->allowsynchronized++;
3803ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
3811575c14dSBarry Smith }
3821575c14dSBarry Smith 
3831575c14dSBarry Smith /*@C
384811af0c4SBarry Smith   PetscViewerASCIIPopSynchronized - Undoes most recent `PetscViewerASCIIPushSynchronized()` for this viewer
3851575c14dSBarry Smith 
386c3339decSBarry Smith   Collective
3871575c14dSBarry Smith 
38820f4b53cSBarry Smith   Input Parameter:
389811af0c4SBarry Smith . viewer - obtained with `PetscViewerASCIIOpen()`
3901575c14dSBarry Smith 
3911575c14dSBarry Smith   Level: intermediate
3921575c14dSBarry Smith 
393811af0c4SBarry Smith   Note:
394811af0c4SBarry Smith   See documentation of `PetscViewerASCIISynchronizedPrintf()` for more details how the synchronized output should be done properly.
3951575c14dSBarry Smith 
396d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerASCIIPushSynchronized()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerFlush()`,
397db781477SPatrick Sanan           `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIIOpen()`,
398db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`
3991575c14dSBarry Smith @*/
400d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIPopSynchronized(PetscViewer viewer)
401d71ae5a4SJacob Faibussowitsch {
4021575c14dSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
4031575c14dSBarry Smith   PetscBool          iascii;
4041575c14dSBarry Smith 
4051575c14dSBarry Smith   PetscFunctionBegin;
4061575c14dSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
40728b400f6SJacob Faibussowitsch   PetscCheck(!ascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
4089566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
4091575c14dSBarry Smith   if (iascii) {
4101575c14dSBarry Smith     ascii->allowsynchronized--;
41108401ef6SPierre Jolivet     PetscCheck(ascii->allowsynchronized >= 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called more times than PetscViewerASCIIPushSynchronized()");
4121575c14dSBarry Smith   }
4133ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4145c6c1daeSBarry Smith }
4155c6c1daeSBarry Smith 
4161c297824SMatthew G. Knepley /*@C
417811af0c4SBarry Smith   PetscViewerASCIIPushTab - Adds one more tab to the amount that `PetscViewerASCIIPrintf()`
4185c6c1daeSBarry Smith   lines are tabbed.
4195c6c1daeSBarry Smith 
420c410d8ccSBarry Smith   Not Collective, but only first MPI rank in the viewer has any effect; No Fortran Support
4215c6c1daeSBarry Smith 
42220f4b53cSBarry Smith   Input Parameter:
423811af0c4SBarry Smith . viewer - obtained with `PetscViewerASCIIOpen()`
4245c6c1daeSBarry Smith 
4255c6c1daeSBarry Smith   Level: developer
4265c6c1daeSBarry Smith 
427d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
428db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
429db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`
4305c6c1daeSBarry Smith @*/
431d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIPushTab(PetscViewer viewer)
432d71ae5a4SJacob Faibussowitsch {
4335c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
4345c6c1daeSBarry Smith   PetscBool          iascii;
4355c6c1daeSBarry Smith 
4365c6c1daeSBarry Smith   PetscFunctionBegin;
4375c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
4389566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
439a297a907SKarl Rupp   if (iascii) ascii->tab++;
4403ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4415c6c1daeSBarry Smith }
4425c6c1daeSBarry Smith 
4431c297824SMatthew G. Knepley /*@C
4443f423023SBarry Smith   PetscViewerASCIIPopTab - Removes one tab from the amount that `PetscViewerASCIIPrintf()` lines are tabbed that was provided by
4453f423023SBarry Smith   `PetscViewerASCIIPushTab()`
4465c6c1daeSBarry Smith 
447c410d8ccSBarry Smith   Not Collective, but only first MPI rank in the viewer has any effect; No Fortran Support
4485c6c1daeSBarry Smith 
44920f4b53cSBarry Smith   Input Parameter:
450811af0c4SBarry Smith . viewer - obtained with `PetscViewerASCIIOpen()`
4515c6c1daeSBarry Smith 
4525c6c1daeSBarry Smith   Level: developer
4535c6c1daeSBarry Smith 
454d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
455db781477SPatrick Sanan           `PetscViewerASCIIPushTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
456db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`
4575c6c1daeSBarry Smith @*/
458d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIPopTab(PetscViewer viewer)
459d71ae5a4SJacob Faibussowitsch {
4605c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
4615c6c1daeSBarry Smith   PetscBool          iascii;
4625c6c1daeSBarry Smith 
4635c6c1daeSBarry Smith   PetscFunctionBegin;
4645c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
4659566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
4665c6c1daeSBarry Smith   if (iascii) {
46708401ef6SPierre Jolivet     PetscCheck(ascii->tab > 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "More tabs popped than pushed");
4685c6c1daeSBarry Smith     ascii->tab--;
4695c6c1daeSBarry Smith   }
4703ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
4715c6c1daeSBarry Smith }
4725c6c1daeSBarry Smith 
4735c6c1daeSBarry Smith /*@
474c410d8ccSBarry Smith   PetscViewerASCIIUseTabs - Turns on or off the use of tabs with the `PETSCVIEWERASCII` `PetscViewer`
4755c6c1daeSBarry Smith 
476c410d8ccSBarry Smith   Not Collective, but only first MPI rank in the viewer has any effect; No Fortran Support
4775c6c1daeSBarry Smith 
4785c6c1daeSBarry Smith   Input Parameters:
479811af0c4SBarry Smith + viewer - obtained with `PetscViewerASCIIOpen()`
480811af0c4SBarry Smith - flg    - `PETSC_TRUE` or `PETSC_FALSE`
4815c6c1daeSBarry Smith 
4825c6c1daeSBarry Smith   Level: developer
4835c6c1daeSBarry Smith 
484d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
485db781477SPatrick Sanan           `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIPushTab()`, `PetscViewerASCIIOpen()`,
486db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`
4875c6c1daeSBarry Smith @*/
488d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIUseTabs(PetscViewer viewer, PetscBool flg)
489d71ae5a4SJacob Faibussowitsch {
4905c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
4915c6c1daeSBarry Smith   PetscBool          iascii;
4925c6c1daeSBarry Smith 
4935c6c1daeSBarry Smith   PetscFunctionBegin;
4945c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
4959566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
4965c6c1daeSBarry Smith   if (iascii) {
497a297a907SKarl Rupp     if (flg) ascii->tab = ascii->tab_store;
498a297a907SKarl Rupp     else {
4995c6c1daeSBarry Smith       ascii->tab_store = ascii->tab;
5005c6c1daeSBarry Smith       ascii->tab       = 0;
5015c6c1daeSBarry Smith     }
5025c6c1daeSBarry Smith   }
5033ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
5045c6c1daeSBarry Smith }
5055c6c1daeSBarry Smith 
506e4096674SBarry Smith #if defined(PETSC_USE_FORTRAN_BINDINGS)
507e4096674SBarry Smith 
508e4096674SBarry Smith   #if defined(PETSC_HAVE_FORTRAN_CAPS)
509e4096674SBarry Smith     #define petscviewerasciiopenwithfileunit_ PETSCVIEWERASCIIOPENWITHFILEUNIT
51057b1f488SBarry Smith     #define petscviewerasciisetfileunit_      PETSCVIEWERASCIISETFILEUNIT
51157b1f488SBarry Smith     #define petscviewerasciiworldsetfileunit_ PETSCVIEWERASCIIWORLDSETFILEUNIT
512e4096674SBarry Smith     #define petscfortranprinttounit_          PETSCFORTRANPRINTTOUNIT
513e4096674SBarry Smith   #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
514e4096674SBarry Smith     #define petscviewerasciiopenwithfileunit_ petscviewerasciiopenwithfileunit
515e4096674SBarry Smith     #define petscviewerasciisetfileunit_      petscviewerasciisetfileunit
51657b1f488SBarry Smith     #define petscviewerasciiworldsetfileunit_ petscviewerasciiworldsetfileunit
517e4096674SBarry Smith     #define petscfortranprinttounit_          petscfortranprinttounit
518e4096674SBarry Smith   #endif
519e4096674SBarry Smith 
520e4096674SBarry Smith   #if defined(__cplusplus)
521e4096674SBarry Smith extern "C" void petscfortranprinttounit_(PetscInt *, const char *, PetscErrorCode *, PETSC_FORTRAN_CHARLEN_T);
522e4096674SBarry Smith   #else
523e4096674SBarry Smith extern void petscfortranprinttounit_(PetscInt *, const char *, PetscErrorCode *, PETSC_FORTRAN_CHARLEN_T);
524e4096674SBarry Smith   #endif
525e4096674SBarry Smith 
526e4096674SBarry Smith   #define PETSCDEFAULTBUFFERSIZE 8 * 1024
527e4096674SBarry Smith 
52857b1f488SBarry Smith static PetscInt PETSC_VIEWER_ASCII_WORLD_fileunit = 0;
52957b1f488SBarry Smith 
53057b1f488SBarry Smith // PetscClangLinter pragma disable: -fdoc-synopsis-macro-explicit-synopsis-valid-header
53157b1f488SBarry Smith /*MC
53257b1f488SBarry Smith   PetscViewerASCIIWORLDSetFileUnit - sets `PETSC_VIEWER_STDOUT_WORLD` to write to a Fortran IO unit
53357b1f488SBarry Smith 
53457b1f488SBarry Smith   Synopsis:
53557b1f488SBarry Smith   #include <petscviewer.h>
53657b1f488SBarry Smith   void PetscViewerASCIIWORLDSetFileUnit(PetscInt unit, PetscErrorCode ierr)
53757b1f488SBarry Smith 
53857b1f488SBarry Smith   Input Parameter:
53957b1f488SBarry Smith . unit - the unit number
54057b1f488SBarry Smith 
54157b1f488SBarry Smith   Output Parameter:
54257b1f488SBarry Smith . ierr - the error code
54357b1f488SBarry Smith 
54457b1f488SBarry Smith   Level: intermediate
54557b1f488SBarry Smith 
54657b1f488SBarry Smith   Notes:
54757b1f488SBarry Smith   Must be called before `PetscInitialize()`
54857b1f488SBarry Smith 
549*1d031f67SBarry Smith   This may not work currently with some viewers that (improperly) use the `fd` directly instead of `PetscViewerASCIIPrintf()`
55057b1f488SBarry Smith 
55157b1f488SBarry Smith   With this option, for example, `-log_options` results will be saved to the Fortran file
55257b1f488SBarry Smith 
553*1d031f67SBarry Smith   Any process may call this but only the unit passed on the first process is used
55457b1f488SBarry Smith 
55557b1f488SBarry Smith   Fortran Note:
55657b1f488SBarry Smith   Only for Fortran
55757b1f488SBarry Smith 
55857b1f488SBarry Smith   Developer Note:
55957b1f488SBarry Smith   `PetscViewerASCIIWORLDSetFilename()` could be added in the future
56057b1f488SBarry Smith 
56157b1f488SBarry Smith .seealso: `PetscViewerASCIISetFILE()`, `PETSCVIEWERASCII`, `PetscViewerASCIIOpenWithFileUnit()`, `PetscViewerASCIIWORLDSetFileUnit()`
56257b1f488SBarry Smith M*/
56357b1f488SBarry Smith PETSC_EXTERN void petscviewerasciiworldsetfileunit_(PetscInt *unit, PetscErrorCode *ierr)
56457b1f488SBarry Smith {
56557b1f488SBarry Smith   PETSC_VIEWER_ASCII_WORLD_fileunit = *unit;
56657b1f488SBarry Smith }
56757b1f488SBarry Smith 
56857b1f488SBarry Smith   #include <petsc/private/fortranimpl.h>
56957b1f488SBarry Smith 
57010450e9eSJacob Faibussowitsch // PetscClangLinter pragma disable: -fdoc-synopsis-macro-explicit-synopsis-valid-header
571489d2c6aSPierre Jolivet /*MC
572baca6076SPierre Jolivet   PetscViewerASCIISetFileUnit - sets the `PETSCVIEWERASCII` to write to a Fortran IO unit
573e4096674SBarry Smith 
57410450e9eSJacob Faibussowitsch   Synopsis:
57510450e9eSJacob Faibussowitsch   #include <petscviewer.h>
576e4096674SBarry Smith   void PetscViewerASCIISetFileUnit(PetscViewer lab, PetscInt unit, PetscErrorCode ierr)
577e4096674SBarry Smith 
578e4096674SBarry Smith   Input Parameters:
579e4096674SBarry Smith + lab  - the viewer
580e4096674SBarry Smith - unit - the unit number
581e4096674SBarry Smith 
582e4096674SBarry Smith   Output Parameter:
583e4096674SBarry Smith . ierr - the error code
584e4096674SBarry Smith 
58510450e9eSJacob Faibussowitsch   Level: intermediate
58610450e9eSJacob Faibussowitsch 
587e4096674SBarry Smith   Note:
588e4096674SBarry Smith   `PetscViewerDestroy()` does not close the unit for this `PetscViewer`
589e4096674SBarry Smith 
590aec76313SJacob Faibussowitsch   Fortran Notes:
591e4096674SBarry Smith   Only for Fortran, use  `PetscViewerASCIISetFILE()` for C
592e4096674SBarry Smith 
59357b1f488SBarry Smith .seealso: `PetscViewerASCIISetFILE()`, `PETSCVIEWERASCII`, `PetscViewerASCIIOpenWithFileUnit()`, `PetscViewerASCIIWORLDSetFileUnit()`
594489d2c6aSPierre Jolivet M*/
595e4096674SBarry Smith PETSC_EXTERN void petscviewerasciisetfileunit_(PetscViewer *lab, PetscInt *unit, PetscErrorCode *ierr)
596e4096674SBarry Smith {
59757b1f488SBarry Smith   PetscViewer_ASCII *vascii;
59857b1f488SBarry Smith   PetscViewer        v;
599e4096674SBarry Smith 
60057b1f488SBarry Smith   PetscPatchDefaultViewers_Fortran(lab, v);
60157b1f488SBarry Smith   vascii = (PetscViewer_ASCII *)v->data;
602e4096674SBarry Smith   if (vascii->mode == FILE_MODE_READ) {
603e4096674SBarry Smith     *ierr = PETSC_ERR_ARG_WRONGSTATE;
604e4096674SBarry Smith     return;
605e4096674SBarry Smith   }
606e4096674SBarry Smith   vascii->fileunit = *unit;
607e4096674SBarry Smith }
608e4096674SBarry Smith 
60910450e9eSJacob Faibussowitsch // PetscClangLinter pragma disable: -fdoc-synopsis-macro-explicit-synopsis-valid-header
610489d2c6aSPierre Jolivet /*MC
611baca6076SPierre Jolivet   PetscViewerASCIIOpenWithFileUnit - opens a `PETSCVIEWERASCII` to write to a Fortran IO unit
612e4096674SBarry Smith 
61310450e9eSJacob Faibussowitsch   Synopsis:
61410450e9eSJacob Faibussowitsch   #include <petscviewer.h>
615e4096674SBarry Smith   void PetscViewerASCIIOpenWithFileUnit(MPI_Comm comm, PetscInt unit, PetscViewer viewer, PetscErrorCode ierr)
616e4096674SBarry Smith 
617e4096674SBarry Smith   Input Parameters:
618e4096674SBarry Smith + comm - the `MPI_Comm` to share the viewer
619e4096674SBarry Smith - unit - the unit number
620e4096674SBarry Smith 
621e4096674SBarry Smith   Output Parameters:
622e4096674SBarry Smith + lab  - the viewer
623e4096674SBarry Smith - ierr - the error code
624e4096674SBarry Smith 
62510450e9eSJacob Faibussowitsch   Level: intermediate
62610450e9eSJacob Faibussowitsch 
627e4096674SBarry Smith   Note:
628e4096674SBarry Smith   `PetscViewerDestroy()` does not close the unit for this `PetscViewer`
629e4096674SBarry Smith 
630aec76313SJacob Faibussowitsch   Fortran Notes:
631e4096674SBarry Smith   Only for Fortran, use  `PetscViewerASCIIOpenWithFILE()` for C
632e4096674SBarry Smith 
633e4096674SBarry Smith .seealso: `PetscViewerASCIISetFileUnit()`, `PetscViewerASCIISetFILE()`, `PETSCVIEWERASCII`, `PetscViewerASCIIOpenWithFILE()`
634489d2c6aSPierre Jolivet M*/
635e4096674SBarry Smith PETSC_EXTERN void petscviewerasciiopenwithfileunit_(MPI_Comm *comm, PetscInt *unit, PetscViewer *lab, PetscErrorCode *ierr)
636e4096674SBarry Smith {
637e4096674SBarry Smith   *ierr = PetscViewerCreate(MPI_Comm_f2c(*(MPI_Fint *)&*comm), lab);
638e4096674SBarry Smith   if (*ierr) return;
639e4096674SBarry Smith   *ierr = PetscViewerSetType(*lab, PETSCVIEWERASCII);
640e4096674SBarry Smith   if (*ierr) return;
641e4096674SBarry Smith   *ierr = PetscViewerFileSetMode(*lab, FILE_MODE_WRITE);
642e4096674SBarry Smith   if (*ierr) return;
643e4096674SBarry Smith   petscviewerasciisetfileunit_(lab, unit, ierr);
644e4096674SBarry Smith }
645e4096674SBarry Smith 
646e4096674SBarry Smith static PetscErrorCode PetscVFPrintfFortran(PetscInt unit, const char format[], va_list Argp)
647e4096674SBarry Smith {
648e4096674SBarry Smith   PetscErrorCode ierr;
649e4096674SBarry Smith   char           str[PETSCDEFAULTBUFFERSIZE];
650e4096674SBarry Smith   size_t         len;
651e4096674SBarry Smith 
652e4096674SBarry Smith   PetscFunctionBegin;
653e4096674SBarry Smith   PetscCall(PetscVSNPrintf(str, sizeof(str), format, NULL, Argp));
654e4096674SBarry Smith   PetscCall(PetscStrlen(str, &len));
655e4096674SBarry Smith   petscfortranprinttounit_(&unit, str, &ierr, (int)len);
656e4096674SBarry Smith   PetscFunctionReturn(PETSC_SUCCESS);
657e4096674SBarry Smith }
658e4096674SBarry Smith 
659e4096674SBarry Smith static PetscErrorCode PetscFPrintfFortran(PetscInt unit, const char str[])
660e4096674SBarry Smith {
661e4096674SBarry Smith   PetscErrorCode ierr;
662e4096674SBarry Smith   size_t         len;
663e4096674SBarry Smith 
664e4096674SBarry Smith   PetscFunctionBegin;
665e4096674SBarry Smith   PetscCall(PetscStrlen(str, &len));
666e4096674SBarry Smith   petscfortranprinttounit_(&unit, str, &ierr, (int)len);
667e4096674SBarry Smith   PetscFunctionReturn(PETSC_SUCCESS);
668e4096674SBarry Smith }
669e4096674SBarry Smith 
670e4096674SBarry Smith #else
671e4096674SBarry Smith 
672e4096674SBarry Smith /* these will never be used; but are needed to link with */
673e4096674SBarry Smith static PetscErrorCode PetscVFPrintfFortran(PetscInt unit, const char format[], va_list Argp)
674e4096674SBarry Smith {
675e4096674SBarry Smith   PetscFunctionBegin;
676e4096674SBarry Smith   PetscFunctionReturn(PETSC_SUCCESS);
677e4096674SBarry Smith }
678e4096674SBarry Smith 
679e4096674SBarry Smith static PetscErrorCode PetscFPrintfFortran(PetscInt unit, const char str[])
680e4096674SBarry Smith {
681e4096674SBarry Smith   PetscFunctionBegin;
682e4096674SBarry Smith   PetscFunctionReturn(PETSC_SUCCESS);
683e4096674SBarry Smith }
684e4096674SBarry Smith #endif
685e4096674SBarry Smith 
68657b1f488SBarry Smith /*@
687*1d031f67SBarry Smith   PetscViewerASCIIGetStdout - Creates a `PETSCVIEWERASCII` `PetscViewer` shared by all processes
68857b1f488SBarry Smith   in a communicator. Error returning version of `PETSC_VIEWER_STDOUT_()`
68957b1f488SBarry Smith 
69057b1f488SBarry Smith   Collective
69157b1f488SBarry Smith 
69257b1f488SBarry Smith   Input Parameter:
69357b1f488SBarry Smith . comm - the MPI communicator to share the `PetscViewer`
69457b1f488SBarry Smith 
69557b1f488SBarry Smith   Output Parameter:
69657b1f488SBarry Smith . viewer - the viewer
69757b1f488SBarry Smith 
69857b1f488SBarry Smith   Level: beginner
69957b1f488SBarry Smith 
70057b1f488SBarry Smith   Note:
70157b1f488SBarry Smith   This object is destroyed in `PetscFinalize()`, `PetscViewerDestroy()` should never be called on it
70257b1f488SBarry Smith 
70357b1f488SBarry Smith   Developer Note:
70457b1f488SBarry Smith   This should be used in all PETSc source code instead of `PETSC_VIEWER_STDOUT_()` since it allows error checking
70557b1f488SBarry Smith 
70657b1f488SBarry Smith .seealso: [](sec_viewers), `PETSC_VIEWER_DRAW_()`, `PetscViewerASCIIOpen()`, `PETSC_VIEWER_STDERR_`, `PETSC_VIEWER_STDOUT_WORLD`,
70757b1f488SBarry Smith           `PETSC_VIEWER_STDOUT_SELF`
70857b1f488SBarry Smith @*/
70957b1f488SBarry Smith PetscErrorCode PetscViewerASCIIGetStdout(MPI_Comm comm, PetscViewer *viewer)
71057b1f488SBarry Smith {
71157b1f488SBarry Smith   PetscBool flg;
71257b1f488SBarry Smith   MPI_Comm  ncomm;
71357b1f488SBarry Smith 
71457b1f488SBarry Smith   PetscFunctionBegin;
71557b1f488SBarry Smith   PetscCall(PetscSpinlockLock(&PetscViewerASCIISpinLockStdout));
71657b1f488SBarry Smith   PetscCall(PetscCommDuplicate(comm, &ncomm, NULL));
71757b1f488SBarry Smith   if (Petsc_Viewer_Stdout_keyval == MPI_KEYVAL_INVALID) PetscCallMPI(MPI_Comm_create_keyval(MPI_COMM_NULL_COPY_FN, MPI_COMM_NULL_DELETE_FN, &Petsc_Viewer_Stdout_keyval, NULL));
71857b1f488SBarry Smith   PetscCallMPI(MPI_Comm_get_attr(ncomm, Petsc_Viewer_Stdout_keyval, (void **)viewer, (PetscMPIInt *)&flg));
71957b1f488SBarry Smith   if (!flg) { /* PetscViewer not yet created */
72057b1f488SBarry Smith #if defined(PETSC_USE_FORTRAN_BINDINGS)
72157b1f488SBarry Smith     PetscMPIInt size, gsize;
72257b1f488SBarry Smith 
72357b1f488SBarry Smith     PetscCallMPI(MPI_Comm_size(comm, &size));
72457b1f488SBarry Smith     PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &gsize));
72557b1f488SBarry Smith     if (size == gsize) { PetscCallMPI(MPI_Bcast(&PETSC_VIEWER_ASCII_WORLD_fileunit, 1, MPIU_INT, 0, comm)); }
72657b1f488SBarry Smith     if (PETSC_VIEWER_ASCII_WORLD_fileunit) {
72757b1f488SBarry Smith       PetscErrorCode ierr;
72857b1f488SBarry Smith 
72957b1f488SBarry Smith       petscviewerasciiopenwithfileunit_(&ncomm, &PETSC_VIEWER_ASCII_WORLD_fileunit, viewer, &ierr);
73057b1f488SBarry Smith     } else
73157b1f488SBarry Smith #endif
73257b1f488SBarry Smith       PetscCall(PetscViewerASCIIOpen(ncomm, "stdout", viewer));
73357b1f488SBarry Smith     ((PetscObject)*viewer)->persistent = PETSC_TRUE;
73457b1f488SBarry Smith     PetscCall(PetscObjectRegisterDestroy((PetscObject)*viewer));
73557b1f488SBarry Smith     PetscCallMPI(MPI_Comm_set_attr(ncomm, Petsc_Viewer_Stdout_keyval, (void *)*viewer));
73657b1f488SBarry Smith   }
73757b1f488SBarry Smith   PetscCall(PetscCommDestroy(&ncomm));
73857b1f488SBarry Smith   PetscCall(PetscSpinlockUnlock(&PetscViewerASCIISpinLockStdout));
73957b1f488SBarry Smith   PetscFunctionReturn(PETSC_SUCCESS);
74057b1f488SBarry Smith }
74157b1f488SBarry Smith 
7425c6c1daeSBarry Smith /*@C
7435c6c1daeSBarry Smith   PetscViewerASCIIPrintf - Prints to a file, only from the first
7443f423023SBarry Smith   processor in the `PetscViewer` of type `PETSCVIEWERASCII`
7455c6c1daeSBarry Smith 
746c410d8ccSBarry Smith   Not Collective, but only the first MPI rank in the viewer has any effect
7475c6c1daeSBarry Smith 
7485c6c1daeSBarry Smith   Input Parameters:
749811af0c4SBarry Smith + viewer - obtained with `PetscViewerASCIIOpen()`
7505c6c1daeSBarry Smith - format - the usual printf() format string
7515c6c1daeSBarry Smith 
7525c6c1daeSBarry Smith   Level: developer
7535c6c1daeSBarry Smith 
754aec76313SJacob Faibussowitsch   Fortran Notes:
755c410d8ccSBarry Smith   The call sequence is `PetscViewerASCIIPrintf`(`PetscViewer`, character(*), int ierr) from Fortran.
7565c6c1daeSBarry Smith   That is, you can only pass a single character string from Fortran.
7575c6c1daeSBarry Smith 
758d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIOpen()`,
759db781477SPatrick Sanan           `PetscViewerASCIIPushTab()`, `PetscViewerASCIIPopTab()`, `PetscViewerASCIISynchronizedPrintf()`,
760db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`, `PetscViewerASCIIGetPointer()`, `PetscViewerASCIIPushSynchronized()`
7615c6c1daeSBarry Smith @*/
762d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIPrintf(PetscViewer viewer, const char format[], ...)
763d71ae5a4SJacob Faibussowitsch {
7645c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
7655c6c1daeSBarry Smith   PetscMPIInt        rank;
766fe8fb074SBarry Smith   PetscInt           tab = 0, intab = ascii->tab;
7675c6c1daeSBarry Smith   FILE              *fd = ascii->fd;
7683f08860eSBarry Smith   PetscBool          iascii;
7695c6c1daeSBarry Smith 
7705c6c1daeSBarry Smith   PetscFunctionBegin;
7715c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
77228b400f6SJacob Faibussowitsch   PetscCheck(!ascii->sviewer, PetscObjectComm((PetscObject)viewer), PETSC_ERR_ARG_WRONGSTATE, "Cannot call with outstanding call to PetscViewerRestoreSubViewer()");
7734f572ea9SToby Isaac   PetscAssertPointer(format, 2);
7749566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
77528b400f6SJacob Faibussowitsch   PetscCheck(iascii, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Not ASCII PetscViewer");
7769566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
7773ba16761SJacob Faibussowitsch   if (rank) PetscFunctionReturn(PETSC_SUCCESS);
7783f08860eSBarry Smith 
7793f08860eSBarry Smith   if (ascii->bviewer) { /* pass string up to parent viewer */
7803f08860eSBarry Smith     char   *string;
7813f08860eSBarry Smith     va_list Argp;
7823f08860eSBarry Smith     size_t  fullLength;
7833f08860eSBarry Smith 
7849566063dSJacob Faibussowitsch     PetscCall(PetscCalloc1(QUEUESTRINGSIZE, &string));
785fe8fb074SBarry Smith     for (; tab < ascii->tab; tab++) { string[2 * tab] = string[2 * tab + 1] = ' '; }
7863f08860eSBarry Smith     va_start(Argp, format);
787fe8fb074SBarry Smith     PetscCall(PetscVSNPrintf(string + 2 * intab, QUEUESTRINGSIZE - 2 * intab, format, &fullLength, Argp));
7883f08860eSBarry Smith     va_end(Argp);
789fe8fb074SBarry Smith     PetscCall(PetscViewerASCIISynchronizedPrintf(ascii->bviewer, "%s", string));
7909566063dSJacob Faibussowitsch     PetscCall(PetscFree(string));
7913f08860eSBarry Smith   } else { /* write directly to file */
7925c6c1daeSBarry Smith     va_list Argp;
793fe8fb074SBarry Smith 
794dd2fa690SBarry Smith     tab = intab;
795e4096674SBarry Smith     while (tab--) {
796e4096674SBarry Smith       if (!ascii->fileunit) PetscCall(PetscFPrintf(PETSC_COMM_SELF, fd, "  "));
797e4096674SBarry Smith       else PetscCall(PetscFPrintfFortran(ascii->fileunit, "   "));
798e4096674SBarry Smith     }
7995c6c1daeSBarry Smith 
8005c6c1daeSBarry Smith     va_start(Argp, format);
801e4096674SBarry Smith     if (!ascii->fileunit) PetscCall((*PetscVFPrintf)(fd, format, Argp));
802e4096674SBarry Smith     else PetscCall(PetscVFPrintfFortran(ascii->fileunit, format, Argp));
803eae3dc7dSJacob Faibussowitsch     va_end(Argp);
804c69effb2SJacob Faibussowitsch     PetscCall(PetscFFlush(fd));
8055c6c1daeSBarry Smith   }
8063ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
8075c6c1daeSBarry Smith }
8085c6c1daeSBarry Smith 
8095c6c1daeSBarry Smith /*@C
810c410d8ccSBarry Smith   PetscViewerFileSetName - Sets the name of the file the `PetscViewer` should use.
8115c6c1daeSBarry Smith 
812c3339decSBarry Smith   Collective
8135c6c1daeSBarry Smith 
8145c6c1daeSBarry Smith   Input Parameters:
8153f423023SBarry Smith + viewer - the `PetscViewer`; for example, of type `PETSCVIEWERASCII` or `PETSCVIEWERBINARY`
8165c6c1daeSBarry Smith - name   - the name of the file it should use
8175c6c1daeSBarry Smith 
8185c6c1daeSBarry Smith   Level: advanced
8195c6c1daeSBarry Smith 
820c410d8ccSBarry Smith   Note:
821c410d8ccSBarry Smith   This will have no effect on viewers that are not related to files
822c410d8ccSBarry Smith 
823d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerASCIIOpen()`, `PetscViewerBinaryOpen()`, `PetscViewerDestroy()`,
824db781477SPatrick Sanan           `PetscViewerASCIIGetPointer()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIISynchronizedPrintf()`
8255c6c1daeSBarry Smith @*/
826d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerFileSetName(PetscViewer viewer, const char name[])
827d71ae5a4SJacob Faibussowitsch {
828cc843e7aSLisandro Dalcin   char filename[PETSC_MAX_PATH_LEN];
8295c6c1daeSBarry Smith 
8305c6c1daeSBarry Smith   PetscFunctionBegin;
8315c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
8324f572ea9SToby Isaac   PetscAssertPointer(name, 2);
8339566063dSJacob Faibussowitsch   PetscCall(PetscStrreplace(PetscObjectComm((PetscObject)viewer), name, filename, sizeof(filename)));
834cac4c232SBarry Smith   PetscTryMethod(viewer, "PetscViewerFileSetName_C", (PetscViewer, const char[]), (viewer, filename));
8353ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
8365c6c1daeSBarry Smith }
8375c6c1daeSBarry Smith 
8385c6c1daeSBarry Smith /*@C
839c410d8ccSBarry Smith   PetscViewerFileGetName - Gets the name of the file the `PetscViewer` is using
8405c6c1daeSBarry Smith 
8415c6c1daeSBarry Smith   Not Collective
8425c6c1daeSBarry Smith 
8435c6c1daeSBarry Smith   Input Parameter:
8443f423023SBarry Smith . viewer - the `PetscViewer`
8455c6c1daeSBarry Smith 
8465c6c1daeSBarry Smith   Output Parameter:
8475c6c1daeSBarry Smith . name - the name of the file it is using
8485c6c1daeSBarry Smith 
8495c6c1daeSBarry Smith   Level: advanced
8505c6c1daeSBarry Smith 
851c410d8ccSBarry Smith   Note:
852c410d8ccSBarry Smith   This will have no effect on viewers that are not related to files
853c410d8ccSBarry Smith 
854d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerCreate()`, `PetscViewerSetType()`, `PetscViewerASCIIOpen()`, `PetscViewerBinaryOpen()`, `PetscViewerFileSetName()`
8555c6c1daeSBarry Smith @*/
856d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerFileGetName(PetscViewer viewer, const char **name)
857d71ae5a4SJacob Faibussowitsch {
8585c6c1daeSBarry Smith   PetscFunctionBegin;
8595c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
8604f572ea9SToby Isaac   PetscAssertPointer(name, 2);
861cac4c232SBarry Smith   PetscUseMethod(viewer, "PetscViewerFileGetName_C", (PetscViewer, const char **), (viewer, name));
8623ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
8635c6c1daeSBarry Smith }
8645c6c1daeSBarry Smith 
86534e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerFileGetName_ASCII(PetscViewer viewer, const char **name)
866d71ae5a4SJacob Faibussowitsch {
8675c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
8685c6c1daeSBarry Smith 
8695c6c1daeSBarry Smith   PetscFunctionBegin;
8705c6c1daeSBarry Smith   *name = vascii->filename;
8713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
8725c6c1daeSBarry Smith }
8735c6c1daeSBarry Smith 
874bf31d2d3SBarry Smith #include <errno.h>
87534e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerFileSetName_ASCII(PetscViewer viewer, const char name[])
876d71ae5a4SJacob Faibussowitsch {
8775c6c1daeSBarry Smith   size_t             len;
878bbcf679cSJacob Faibussowitsch   char               fname[PETSC_MAX_PATH_LEN], *gz = NULL;
8795c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
8805c6c1daeSBarry Smith   PetscBool          isstderr, isstdout;
8815c6c1daeSBarry Smith   PetscMPIInt        rank;
8825c6c1daeSBarry Smith 
8835c6c1daeSBarry Smith   PetscFunctionBegin;
8849566063dSJacob Faibussowitsch   PetscCall(PetscViewerFileClose_ASCII(viewer));
8853ba16761SJacob Faibussowitsch   if (!name) PetscFunctionReturn(PETSC_SUCCESS);
8869566063dSJacob Faibussowitsch   PetscCall(PetscStrallocpy(name, &vascii->filename));
8875c6c1daeSBarry Smith 
8885c6c1daeSBarry Smith   /* Is this file to be compressed */
8895c6c1daeSBarry Smith   vascii->storecompressed = PETSC_FALSE;
890a297a907SKarl Rupp 
8919566063dSJacob Faibussowitsch   PetscCall(PetscStrstr(vascii->filename, ".gz", &gz));
8925c6c1daeSBarry Smith   if (gz) {
8939566063dSJacob Faibussowitsch     PetscCall(PetscStrlen(gz, &len));
8945c6c1daeSBarry Smith     if (len == 3) {
89508401ef6SPierre Jolivet       PetscCheck(vascii->mode == FILE_MODE_WRITE, PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Cannot open ASCII PetscViewer file that is compressed; uncompress it manually first");
8965c6c1daeSBarry Smith       *gz                     = 0;
8975c6c1daeSBarry Smith       vascii->storecompressed = PETSC_TRUE;
8985c6c1daeSBarry Smith     }
8995c6c1daeSBarry Smith   }
9009566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
901dd400576SPatrick Sanan   if (rank == 0) {
9029566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "stderr", &isstderr));
9039566063dSJacob Faibussowitsch     PetscCall(PetscStrcmp(name, "stdout", &isstdout));
9045c6c1daeSBarry Smith     /* empty filename means stdout */
9055c6c1daeSBarry Smith     if (name[0] == 0) isstdout = PETSC_TRUE;
9065c6c1daeSBarry Smith     if (isstderr) vascii->fd = PETSC_STDERR;
9075c6c1daeSBarry Smith     else if (isstdout) vascii->fd = PETSC_STDOUT;
9085c6c1daeSBarry Smith     else {
9099566063dSJacob Faibussowitsch       PetscCall(PetscFixFilename(name, fname));
9105c6c1daeSBarry Smith       switch (vascii->mode) {
911d71ae5a4SJacob Faibussowitsch       case FILE_MODE_READ:
912d71ae5a4SJacob Faibussowitsch         vascii->fd = fopen(fname, "r");
913d71ae5a4SJacob Faibussowitsch         break;
914d71ae5a4SJacob Faibussowitsch       case FILE_MODE_WRITE:
915d71ae5a4SJacob Faibussowitsch         vascii->fd = fopen(fname, "w");
916d71ae5a4SJacob Faibussowitsch         break;
917d71ae5a4SJacob Faibussowitsch       case FILE_MODE_APPEND:
918d71ae5a4SJacob Faibussowitsch         vascii->fd = fopen(fname, "a");
919d71ae5a4SJacob Faibussowitsch         break;
9205c6c1daeSBarry Smith       case FILE_MODE_UPDATE:
9215c6c1daeSBarry Smith         vascii->fd = fopen(fname, "r+");
922a297a907SKarl Rupp         if (!vascii->fd) vascii->fd = fopen(fname, "w+");
9235c6c1daeSBarry Smith         break;
9245c6c1daeSBarry Smith       case FILE_MODE_APPEND_UPDATE:
9255c6c1daeSBarry Smith         /* I really want a file which is opened at the end for updating,
9265c6c1daeSBarry Smith            not a+, which opens at the beginning, but makes writes at the end.
9275c6c1daeSBarry Smith         */
9285c6c1daeSBarry Smith         vascii->fd = fopen(fname, "r+");
929a297a907SKarl Rupp         if (!vascii->fd) vascii->fd = fopen(fname, "w+");
9303ba16761SJacob Faibussowitsch         else {
9313ba16761SJacob Faibussowitsch           int ret = fseek(vascii->fd, 0, SEEK_END);
9323ba16761SJacob Faibussowitsch           PetscCheck(!ret, PETSC_COMM_SELF, PETSC_ERR_LIB, "fseek() failed with error code %d", ret);
9333ba16761SJacob Faibussowitsch         }
9345c6c1daeSBarry Smith         break;
935d71ae5a4SJacob Faibussowitsch       default:
936d71ae5a4SJacob Faibussowitsch         SETERRQ(PetscObjectComm((PetscObject)viewer), PETSC_ERR_SUP, "Unsupported file mode %s", PetscFileModes[vascii->mode]);
9375c6c1daeSBarry Smith       }
938bf31d2d3SBarry Smith       PetscCheck(vascii->fd, PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "Cannot open PetscViewer file: %s due to \"%s\"", fname, strerror(errno));
9395c6c1daeSBarry Smith     }
9405c6c1daeSBarry Smith   }
9413ba16761SJacob Faibussowitsch   PetscCall(PetscLogObjectState((PetscObject)viewer, "File: %s", name));
9423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
9435c6c1daeSBarry Smith }
9445c6c1daeSBarry Smith 
94534e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerGetSubViewer_ASCII(PetscViewer viewer, MPI_Comm subcomm, PetscViewer *outviewer)
946d71ae5a4SJacob Faibussowitsch {
9475c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data, *ovascii;
9485c6c1daeSBarry Smith 
9495c6c1daeSBarry Smith   PetscFunctionBegin;
95028b400f6SJacob Faibussowitsch   PetscCheck(!vascii->sviewer, PETSC_COMM_SELF, PETSC_ERR_ORDER, "SubViewer already obtained from PetscViewer and not restored");
951fe8fb074SBarry Smith   PetscCall(PetscViewerASCIIPushSynchronized(viewer));
952e5afcf28SBarry Smith   /*
9539530cbd7SBarry Smith      The following line is a bug; it does another PetscViewerASCIIPushSynchronized() on viewer, but if it is removed the code won't work
9549530cbd7SBarry Smith      because it relies on this behavior in other places. In particular this line causes the synchronized flush to occur when the viewer is destroyed
9559530cbd7SBarry Smith      (since the count never gets to zero) in some examples this displays information that otherwise would be lost
9569530cbd7SBarry Smith 
9579530cbd7SBarry Smith      This code also means another call to PetscViewerASCIIPopSynchronized() must be made after the PetscViewerRestoreSubViewer(), see, for example,
9589530cbd7SBarry Smith      PCView_GASM().
959e5afcf28SBarry Smith   */
9609566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPushSynchronized(viewer));
961b4025f61SBarry Smith   PetscCall(PetscViewerFlush(viewer));
9629566063dSJacob Faibussowitsch   PetscCall(PetscViewerCreate(subcomm, outviewer));
9639566063dSJacob Faibussowitsch   PetscCall(PetscViewerSetType(*outviewer, PETSCVIEWERASCII));
9649566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPushSynchronized(*outviewer));
9655c6c1daeSBarry Smith   ovascii            = (PetscViewer_ASCII *)(*outviewer)->data;
9665c6c1daeSBarry Smith   ovascii->fd        = vascii->fd;
967ba5a0b41SBarry Smith   ovascii->closefile = PETSC_FALSE;
9685c6c1daeSBarry Smith 
9695c6c1daeSBarry Smith   vascii->sviewer                                      = *outviewer;
9705c6c1daeSBarry Smith   (*outviewer)->format                                 = viewer->format;
9715c6c1daeSBarry Smith   ((PetscViewer_ASCII *)((*outviewer)->data))->bviewer = viewer;
9723f08860eSBarry Smith   (*outviewer)->ops->destroy                           = PetscViewerDestroy_ASCII_SubViewer;
9733ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
9745c6c1daeSBarry Smith }
9755c6c1daeSBarry Smith 
97634e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerRestoreSubViewer_ASCII(PetscViewer viewer, MPI_Comm comm, PetscViewer *outviewer)
977d71ae5a4SJacob Faibussowitsch {
9785c6c1daeSBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)viewer->data;
9795c6c1daeSBarry Smith 
9805c6c1daeSBarry Smith   PetscFunctionBegin;
98128b400f6SJacob Faibussowitsch   PetscCheck(ascii->sviewer, PETSC_COMM_SELF, PETSC_ERR_ORDER, "SubViewer never obtained from PetscViewer");
98208401ef6SPierre Jolivet   PetscCheck(ascii->sviewer == *outviewer, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "This PetscViewer did not generate this SubViewer");
9835c6c1daeSBarry Smith 
9849566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPopSynchronized(*outviewer));
985e5afcf28SBarry Smith   ascii->sviewer             = NULL;
9865c6c1daeSBarry Smith   (*outviewer)->ops->destroy = PetscViewerDestroy_ASCII;
9879566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(outviewer));
988fe8fb074SBarry Smith   PetscCall(PetscViewerFlush(viewer));
9899566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPopSynchronized(viewer));
9903ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
9915c6c1daeSBarry Smith }
9925c6c1daeSBarry Smith 
99334e79e72SJacob Faibussowitsch static PetscErrorCode PetscViewerView_ASCII(PetscViewer v, PetscViewer viewer)
994d71ae5a4SJacob Faibussowitsch {
9952bf49c77SBarry Smith   PetscViewer_ASCII *ascii = (PetscViewer_ASCII *)v->data;
9962bf49c77SBarry Smith 
9972bf49c77SBarry Smith   PetscFunctionBegin;
998*1d031f67SBarry Smith   if (ascii->fileunit) PetscCall(PetscViewerASCIIPrintf(viewer, "Fortran FILE UNIT: %" PetscInt_FMT "\n", ascii->fileunit));
99957b1f488SBarry Smith   else if (ascii->filename) PetscCall(PetscViewerASCIIPrintf(viewer, "Filename: %s\n", ascii->filename));
10003ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
10012bf49c77SBarry Smith }
10022bf49c77SBarry Smith 
10038556b5ebSBarry Smith /*MC
10048556b5ebSBarry Smith    PETSCVIEWERASCII - A viewer that prints to stdout or an ASCII file
10058556b5ebSBarry Smith 
1006811af0c4SBarry Smith   Level: beginner
1007811af0c4SBarry Smith 
1008d1f92df0SBarry Smith .seealso: [](sec_viewers), `PETSC_VIEWER_STDOUT_()`, `PETSC_VIEWER_STDOUT_SELF`, `PETSC_VIEWER_STDOUT_WORLD`, `PetscViewerCreate()`, `PetscViewerASCIIOpen()`,
1009db781477SPatrick Sanan           `PetscViewerMatlabOpen()`, `VecView()`, `DMView()`, `PetscViewerMatlabPutArray()`, `PETSCVIEWERBINARY`, `PETSCVIEWERMATLAB`,
1010db781477SPatrick Sanan           `PetscViewerFileSetName()`, `PetscViewerFileSetMode()`, `PetscViewerFormat`, `PetscViewerType`, `PetscViewerSetType()`
10118556b5ebSBarry Smith M*/
1012d71ae5a4SJacob Faibussowitsch PETSC_EXTERN PetscErrorCode PetscViewerCreate_ASCII(PetscViewer viewer)
1013d71ae5a4SJacob Faibussowitsch {
10145c6c1daeSBarry Smith   PetscViewer_ASCII *vascii;
10155c6c1daeSBarry Smith 
10165c6c1daeSBarry Smith   PetscFunctionBegin;
10174dfa11a4SJacob Faibussowitsch   PetscCall(PetscNew(&vascii));
10185c6c1daeSBarry Smith   viewer->data = (void *)vascii;
10195c6c1daeSBarry Smith 
10205c6c1daeSBarry Smith   viewer->ops->destroy          = PetscViewerDestroy_ASCII;
10215c6c1daeSBarry Smith   viewer->ops->flush            = PetscViewerFlush_ASCII;
1022559f443fSBarry Smith   viewer->ops->getsubviewer     = PetscViewerGetSubViewer_ASCII;
1023559f443fSBarry Smith   viewer->ops->restoresubviewer = PetscViewerRestoreSubViewer_ASCII;
10242bf49c77SBarry Smith   viewer->ops->view             = PetscViewerView_ASCII;
10251d641e7bSMichael Lange   viewer->ops->read             = PetscViewerASCIIRead;
10265c6c1daeSBarry Smith 
10275c6c1daeSBarry Smith   /* defaults to stdout unless set with PetscViewerFileSetName() */
10285c6c1daeSBarry Smith   vascii->fd        = PETSC_STDOUT;
10295c6c1daeSBarry Smith   vascii->mode      = FILE_MODE_WRITE;
103002c9f0b5SLisandro Dalcin   vascii->bviewer   = NULL;
103102c9f0b5SLisandro Dalcin   vascii->subviewer = NULL;
103202c9f0b5SLisandro Dalcin   vascii->sviewer   = NULL;
10335c6c1daeSBarry Smith   vascii->tab       = 0;
10345c6c1daeSBarry Smith   vascii->tab_store = 0;
103502c9f0b5SLisandro Dalcin   vascii->filename  = NULL;
10365c6c1daeSBarry Smith   vascii->closefile = PETSC_TRUE;
10375c6c1daeSBarry Smith 
10389566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetName_C", PetscViewerFileSetName_ASCII));
10399566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetName_C", PetscViewerFileGetName_ASCII));
10409566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileGetMode_C", PetscViewerFileGetMode_ASCII));
10419566063dSJacob Faibussowitsch   PetscCall(PetscObjectComposeFunction((PetscObject)viewer, "PetscViewerFileSetMode_C", PetscViewerFileSetMode_ASCII));
10423ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
10435c6c1daeSBarry Smith }
10445c6c1daeSBarry Smith 
10455c6c1daeSBarry Smith /*@C
1046c410d8ccSBarry Smith   PetscViewerASCIISynchronizedPrintf - Prints synchronized output to the specified `PETSCVIEWERASCII` file from
10475c6c1daeSBarry Smith   several processors.  Output of the first processor is followed by that of the
10485c6c1daeSBarry Smith   second, etc.
10495c6c1daeSBarry Smith 
1050c410d8ccSBarry Smith   Not Collective, must call collective `PetscViewerFlush()` to get the results flushed
10515c6c1daeSBarry Smith 
10525c6c1daeSBarry Smith   Input Parameters:
1053811af0c4SBarry Smith + viewer - the `PETSCVIEWERASCII` `PetscViewer`
10545c6c1daeSBarry Smith - format - the usual printf() format string
10555c6c1daeSBarry Smith 
10565c6c1daeSBarry Smith   Level: intermediate
10575c6c1daeSBarry Smith 
105895452b02SPatrick Sanan   Notes:
1059811af0c4SBarry Smith   You must have previously called `PetscViewerASCIIPushSynchronized()` to allow this routine to be called.
1060e6abc3ddSVáclav Hapla   Then you can do multiple independent calls to this routine.
1061811af0c4SBarry Smith 
1062811af0c4SBarry Smith   The actual synchronized print is then done using `PetscViewerFlush()`.
1063811af0c4SBarry Smith   `PetscViewerASCIIPopSynchronized()` should be then called if we are already done with the synchronized output
1064e6abc3ddSVáclav Hapla   to conclude the "synchronized session".
1065811af0c4SBarry Smith 
1066e6abc3ddSVáclav Hapla   So the typical calling sequence looks like
1067811af0c4SBarry Smith .vb
1068811af0c4SBarry Smith     PetscViewerASCIIPushSynchronized(viewer);
1069811af0c4SBarry Smith     PetscViewerASCIISynchronizedPrintf(viewer, ...);
1070811af0c4SBarry Smith     PetscViewerASCIISynchronizedPrintf(viewer, ...);
1071811af0c4SBarry Smith     ...
1072811af0c4SBarry Smith     PetscViewerFlush(viewer);
1073811af0c4SBarry Smith     PetscViewerASCIISynchronizedPrintf(viewer, ...);
1074811af0c4SBarry Smith     PetscViewerASCIISynchronizedPrintf(viewer, ...);
1075811af0c4SBarry Smith     ...
1076811af0c4SBarry Smith     PetscViewerFlush(viewer);
1077811af0c4SBarry Smith     PetscViewerASCIIPopSynchronized(viewer);
1078811af0c4SBarry Smith .ve
10795c6c1daeSBarry Smith 
1080aec76313SJacob Faibussowitsch   Fortran Notes:
10815c6c1daeSBarry Smith   Can only print a single character* string
10825c6c1daeSBarry Smith 
1083d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerASCIIPushSynchronized()`, `PetscViewerFlush()`, `PetscViewerASCIIPopSynchronized()`,
1084db781477SPatrick Sanan           `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIIOpen()`,
1085db781477SPatrick Sanan           `PetscViewerCreate()`, `PetscViewerDestroy()`, `PetscViewerSetType()`
10865c6c1daeSBarry Smith @*/
1087d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIISynchronizedPrintf(PetscViewer viewer, const char format[], ...)
1088d71ae5a4SJacob Faibussowitsch {
10895c6c1daeSBarry Smith   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
10903f08860eSBarry Smith   PetscMPIInt        rank;
1091fe8fb074SBarry Smith   PetscInt           tab = 0;
10925c6c1daeSBarry Smith   MPI_Comm           comm;
1093fe8fb074SBarry Smith   PetscBool          iascii;
10945c6c1daeSBarry Smith 
10955c6c1daeSBarry Smith   PetscFunctionBegin;
10965c6c1daeSBarry Smith   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
10974f572ea9SToby Isaac   PetscAssertPointer(format, 2);
10989566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &iascii));
109928b400f6SJacob Faibussowitsch   PetscCheck(iascii, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Not ASCII PetscViewer");
110028b400f6SJacob Faibussowitsch   PetscCheck(vascii->allowsynchronized, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "First call PetscViewerASCIIPushSynchronized() to allow this call");
11015c6c1daeSBarry Smith 
11029566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)viewer, &comm));
11039566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
11045c6c1daeSBarry Smith 
1105559f443fSBarry Smith   if (vascii->bviewer) {
1106fe8fb074SBarry Smith     char   *string;
11075c6c1daeSBarry Smith     va_list Argp;
1108fe8fb074SBarry Smith     size_t  fullLength;
11095c6c1daeSBarry Smith 
1110fe8fb074SBarry Smith     PetscCall(PetscCalloc1(QUEUESTRINGSIZE, &string));
1111fe8fb074SBarry Smith     for (; tab < vascii->tab; tab++) { string[2 * tab] = string[2 * tab + 1] = ' '; }
1112fe8fb074SBarry Smith     va_start(Argp, format);
1113fe8fb074SBarry Smith     PetscCall(PetscVSNPrintf(string + 2 * tab, QUEUESTRINGSIZE - 2 * tab, format, &fullLength, Argp));
1114fe8fb074SBarry Smith     va_end(Argp);
1115fe8fb074SBarry Smith     PetscCall(PetscViewerASCIISynchronizedPrintf(vascii->bviewer, "%s", string));
1116fe8fb074SBarry Smith     PetscCall(PetscFree(string));
1117fe8fb074SBarry Smith   } else if (rank == 0) { /* First processor prints immediately to fp */
1118fe8fb074SBarry Smith     va_list Argp;
1119fe8fb074SBarry Smith     FILE   *fp = vascii->fd;
1120fe8fb074SBarry Smith 
1121fe8fb074SBarry Smith     tab = vascii->tab;
112248a46eb9SPierre Jolivet     while (tab--) PetscCall(PetscFPrintf(PETSC_COMM_SELF, fp, "  "));
11235c6c1daeSBarry Smith 
11245c6c1daeSBarry Smith     va_start(Argp, format);
11259566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(fp, format, Argp));
1126eae3dc7dSJacob Faibussowitsch     va_end(Argp);
1127c69effb2SJacob Faibussowitsch     PetscCall(PetscFFlush(fp));
11285c6c1daeSBarry Smith     if (petsc_history) {
11295c6c1daeSBarry Smith       va_start(Argp, format);
11309566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
1131eae3dc7dSJacob Faibussowitsch       va_end(Argp);
1132c69effb2SJacob Faibussowitsch       PetscCall(PetscFFlush(petsc_history));
11335c6c1daeSBarry Smith     }
11345c6c1daeSBarry Smith     va_end(Argp);
1135559f443fSBarry Smith   } else { /* other processors add to queue */
11365c6c1daeSBarry Smith     char       *string;
11375c6c1daeSBarry Smith     va_list     Argp;
11385c6c1daeSBarry Smith     size_t      fullLength;
11395c6c1daeSBarry Smith     PrintfQueue next;
11405c6c1daeSBarry Smith 
11419566063dSJacob Faibussowitsch     PetscCall(PetscNew(&next));
1142559f443fSBarry Smith     if (vascii->petsc_printfqueue) {
1143559f443fSBarry Smith       vascii->petsc_printfqueue->next = next;
1144559f443fSBarry Smith       vascii->petsc_printfqueue       = next;
1145a297a907SKarl Rupp     } else {
1146559f443fSBarry Smith       vascii->petsc_printfqueuebase = vascii->petsc_printfqueue = next;
1147a297a907SKarl Rupp     }
1148559f443fSBarry Smith     vascii->petsc_printfqueuelength++;
11495c6c1daeSBarry Smith     next->size = QUEUESTRINGSIZE;
11509566063dSJacob Faibussowitsch     PetscCall(PetscCalloc1(next->size, &next->string));
11515c6c1daeSBarry Smith     string = next->string;
1152f15bb73eSStefano Zampini 
1153f15bb73eSStefano Zampini     tab = vascii->tab;
11545c6c1daeSBarry Smith     tab *= 2;
1155ad540459SPierre Jolivet     while (tab--) *string++ = ' ';
11565c6c1daeSBarry Smith     va_start(Argp, format);
11579566063dSJacob Faibussowitsch     PetscCall(PetscVSNPrintf(string, next->size - 2 * vascii->tab, format, &fullLength, Argp));
11585c6c1daeSBarry Smith     va_end(Argp);
1159cb500232SBarry Smith     if (fullLength > (size_t)(next->size - 2 * vascii->tab)) {
11609566063dSJacob Faibussowitsch       PetscCall(PetscFree(next->string));
116114416c0eSBarry Smith       next->size = fullLength + 2 * vascii->tab;
11629566063dSJacob Faibussowitsch       PetscCall(PetscCalloc1(next->size, &next->string));
116314416c0eSBarry Smith       string = next->string;
116414416c0eSBarry Smith       tab    = 2 * vascii->tab;
1165ad540459SPierre Jolivet       while (tab--) *string++ = ' ';
116614416c0eSBarry Smith       va_start(Argp, format);
11679566063dSJacob Faibussowitsch       PetscCall(PetscVSNPrintf(string, next->size - 2 * vascii->tab, format, NULL, Argp));
116814416c0eSBarry Smith       va_end(Argp);
116914416c0eSBarry Smith     }
11705c6c1daeSBarry Smith   }
11713ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
11725c6c1daeSBarry Smith }
11735c6c1daeSBarry Smith 
11742655f987SMichael Lange /*@C
1175c410d8ccSBarry Smith   PetscViewerASCIIRead - Reads from a `PETSCVIEWERASCII` file
11762655f987SMichael Lange 
1177c410d8ccSBarry Smith   Only MPI rank 0 in the `PetscViewer` may call this
11782655f987SMichael Lange 
11792655f987SMichael Lange   Input Parameters:
11803f423023SBarry Smith + viewer - the `PETSCVIEWERASCII` viewer
1181c410d8ccSBarry Smith . data   - location to write the data, treated as an array of type indicated by `datatype`
1182060da220SMatthew G. Knepley . num    - number of items of data to read
1183aec76313SJacob Faibussowitsch - dtype  - type of data to read
11842655f987SMichael Lange 
118520f4b53cSBarry Smith   Output Parameter:
11863f423023SBarry Smith . count - number of items of data actually read, or `NULL`
1187f8e4bde8SMatthew G. Knepley 
11882655f987SMichael Lange   Level: beginner
11892655f987SMichael Lange 
1190d1f92df0SBarry Smith .seealso: [](sec_viewers), `PetscViewerASCIIOpen()`, `PetscViewerPushFormat()`, `PetscViewerDestroy()`, `PetscViewerCreate()`, `PetscViewerFileSetMode()`, `PetscViewerFileSetName()`
1191db781477SPatrick Sanan           `VecView()`, `MatView()`, `VecLoad()`, `MatLoad()`, `PetscViewerBinaryGetDescriptor()`,
1192db781477SPatrick Sanan           `PetscViewerBinaryGetInfoPointer()`, `PetscFileMode`, `PetscViewer`, `PetscViewerBinaryRead()`
11932655f987SMichael Lange @*/
1194d71ae5a4SJacob Faibussowitsch PetscErrorCode PetscViewerASCIIRead(PetscViewer viewer, void *data, PetscInt num, PetscInt *count, PetscDataType dtype)
1195d71ae5a4SJacob Faibussowitsch {
11962655f987SMichael Lange   PetscViewer_ASCII *vascii = (PetscViewer_ASCII *)viewer->data;
11972655f987SMichael Lange   FILE              *fd     = vascii->fd;
11982655f987SMichael Lange   PetscInt           i;
11993b7fe8c3SMatthew G. Knepley   int                ret = 0;
1200f8859db6SBarry Smith   PetscMPIInt        rank;
12012655f987SMichael Lange 
12022655f987SMichael Lange   PetscFunctionBegin;
12032655f987SMichael Lange   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 1);
12049566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)viewer), &rank));
1205c5853193SPierre Jolivet   PetscCheck(rank == 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Can only be called from process 0 in the PetscViewer");
1206060da220SMatthew G. Knepley   for (i = 0; i < num; i++) {
1207f8e4bde8SMatthew G. Knepley     if (dtype == PETSC_CHAR) ret = fscanf(fd, "%c", &(((char *)data)[i]));
1208f8e4bde8SMatthew G. Knepley     else if (dtype == PETSC_STRING) ret = fscanf(fd, "%s", &(((char *)data)[i]));
1209a05e1a72SSatish Balay     else if (dtype == PETSC_INT) ret = fscanf(fd, "%" PetscInt_FMT, &(((PetscInt *)data)[i]));
1210f8e4bde8SMatthew G. Knepley     else if (dtype == PETSC_ENUM) ret = fscanf(fd, "%d", &(((int *)data)[i]));
12119e3e4c22SLisandro Dalcin     else if (dtype == PETSC_INT64) ret = fscanf(fd, "%" PetscInt64_FMT, &(((PetscInt64 *)data)[i]));
1212972064b6SLisandro Dalcin     else if (dtype == PETSC_LONG) ret = fscanf(fd, "%ld", &(((long *)data)[i]));
1213f8e4bde8SMatthew G. Knepley     else if (dtype == PETSC_FLOAT) ret = fscanf(fd, "%f", &(((float *)data)[i]));
1214f8e4bde8SMatthew G. Knepley     else if (dtype == PETSC_DOUBLE) ret = fscanf(fd, "%lg", &(((double *)data)[i]));
1215a6e181c6SToby Isaac #if defined(PETSC_USE_REAL___FLOAT128)
1216fba955ccSBarry Smith     else if (dtype == PETSC___FLOAT128) {
1217fba955ccSBarry Smith       double tmp;
1218fba955ccSBarry Smith       ret                     = fscanf(fd, "%lg", &tmp);
1219a6e181c6SToby Isaac       ((__float128 *)data)[i] = tmp;
1220a6e181c6SToby Isaac     }
1221fba955ccSBarry Smith #endif
12229371c9d4SSatish Balay     else
12239371c9d4SSatish Balay       SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Data type %d not supported", (int)dtype);
122428b400f6SJacob Faibussowitsch     PetscCheck(ret, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Conversion error for data type %d", (int)dtype);
1225f7d195e4SLawrence Mitchell     if (ret < 0) break; /* Proxy for EOF, need to check for it in configure */
12262655f987SMichael Lange   }
1227060da220SMatthew G. Knepley   if (count) *count = i;
122808401ef6SPierre Jolivet   else PetscCheck(ret >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Insufficient data, read only %" PetscInt_FMT " < %" PetscInt_FMT " items", i, num);
12293ba16761SJacob Faibussowitsch   PetscFunctionReturn(PETSC_SUCCESS);
12302655f987SMichael Lange }
1231