xref: /petsc/src/sys/fileio/mprint.c (revision 48a46eb9bd028bec07ec0f396b1a3abb43f14558)
1e5c89e4eSSatish Balay /*
2e5c89e4eSSatish Balay       Utilites routines to add simple ASCII IO capability.
3e5c89e4eSSatish Balay */
4c6db04a5SJed Brown #include <../src/sys/fileio/mprint.h>
579c0e996SJed Brown #include <errno.h>
6e5c89e4eSSatish Balay /*
7e5c89e4eSSatish Balay    If petsc_history is on, then all Petsc*Printf() results are saved
8e5c89e4eSSatish Balay    if the appropriate (usually .petschistory) file.
9e5c89e4eSSatish Balay */
1095c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
11e5c89e4eSSatish Balay /*
12e5c89e4eSSatish Balay      Allows one to overwrite where standard out is sent. For example
135106ddf5SBarry Smith      PETSC_STDOUT = fopen("/dev/ttyXX","w") will cause all standard out
14e5c89e4eSSatish Balay      writes to go to terminal XX; assuming you have write permission there
15e5c89e4eSSatish Balay */
1602c9f0b5SLisandro Dalcin FILE              *PETSC_STDOUT = NULL;
17ae9b4142SLisandro Dalcin /*
18ae9b4142SLisandro Dalcin      Allows one to overwrite where standard error is sent. For example
19ae9b4142SLisandro Dalcin      PETSC_STDERR = fopen("/dev/ttyXX","w") will cause all standard error
20ae9b4142SLisandro Dalcin      writes to go to terminal XX; assuming you have write permission there
21ae9b4142SLisandro Dalcin */
2202c9f0b5SLisandro Dalcin FILE              *PETSC_STDERR = NULL;
23b13499bfSbcordonn 
24c9a19010SBarry Smith /*@C
25d781fa04SBarry Smith      PetscFormatConvertGetSize - Gets the length of a string needed to hold format converted with PetscFormatConvert()
26c9a19010SBarry Smith 
27d781fa04SBarry Smith    Input Parameter:
28d781fa04SBarry Smith .   format - the PETSc format string
29c9a19010SBarry Smith 
30d781fa04SBarry Smith    Output Parameter:
31d781fa04SBarry Smith .   size - the needed length of the new format
32c9a19010SBarry Smith 
33c9a19010SBarry Smith  Level: developer
34c9a19010SBarry Smith 
35db781477SPatrick Sanan .seealso: `PetscFormatConvert()`, `PetscVSNPrintf()`, `PetscVFPrintf()`
36d781fa04SBarry Smith 
37c9a19010SBarry Smith @*/
389371c9d4SSatish Balay PetscErrorCode PetscFormatConvertGetSize(const char *format, size_t *size) {
393ca90d2dSJacob Faibussowitsch   size_t   sz = 0;
40d781fa04SBarry Smith   PetscInt i  = 0;
41d781fa04SBarry Smith 
42d781fa04SBarry Smith   PetscFunctionBegin;
433ca90d2dSJacob Faibussowitsch   PetscValidCharPointer(format, 1);
443ca90d2dSJacob Faibussowitsch   PetscValidPointer(size, 2);
45d781fa04SBarry Smith   while (format[i]) {
463ca90d2dSJacob Faibussowitsch     if (format[i] == '%') {
473ca90d2dSJacob Faibussowitsch       if (format[i + 1] == '%') {
483ca90d2dSJacob Faibussowitsch         i += 2;
493ca90d2dSJacob Faibussowitsch         sz += 2;
503ca90d2dSJacob Faibussowitsch         continue;
513ca90d2dSJacob Faibussowitsch       }
52d781fa04SBarry Smith       /* Find the letter */
539371c9d4SSatish Balay       while (format[i] && (format[i] <= '9')) {
549371c9d4SSatish Balay         ++i;
559371c9d4SSatish Balay         ++sz;
569371c9d4SSatish Balay       }
57d781fa04SBarry Smith       switch (format[i]) {
583ca90d2dSJacob Faibussowitsch #if PetscDefined(USE_64BIT_INDICES)
599371c9d4SSatish Balay       case 'D': sz += 2; break;
60d781fa04SBarry Smith #endif
619371c9d4SSatish Balay       case 'g': sz += 4;
629371c9d4SSatish Balay       default: break;
63d781fa04SBarry Smith       }
64d781fa04SBarry Smith     }
653ca90d2dSJacob Faibussowitsch     ++i;
663ca90d2dSJacob Faibussowitsch     ++sz;
67d781fa04SBarry Smith   }
683ca90d2dSJacob Faibussowitsch   *size = sz + 1; /* space for NULL character */
69d781fa04SBarry Smith   PetscFunctionReturn(0);
70d781fa04SBarry Smith }
71d781fa04SBarry Smith 
72d781fa04SBarry Smith /*@C
73d781fa04SBarry Smith      PetscFormatConvert - Takes a PETSc format string and converts the %D to %d for 32 bit PETSc indices and %lld for 64 bit PETSc indices. Also
74d781fa04SBarry Smith                         converts %g to [|%g|] so that PetscVSNPrintf() can easily insure all %g formatted numbers have a decimal point when printed.
75d781fa04SBarry Smith 
76d781fa04SBarry Smith    Input Parameters:
77d781fa04SBarry Smith +   format - the PETSc format string
78d781fa04SBarry Smith .   newformat - the location to put the new format
79d781fa04SBarry Smith -   size - the length of newformat, you can use PetscFormatConvertGetSize() to compute the needed size
80d781fa04SBarry Smith 
81d781fa04SBarry Smith     Note: this exists so we can have the same code when PetscInt is either int or long long int
82d781fa04SBarry Smith 
83d781fa04SBarry Smith  Level: developer
84d781fa04SBarry Smith 
85db781477SPatrick Sanan .seealso: `PetscFormatConvertGetSize()`, `PetscVSNPrintf()`, `PetscVFPrintf()`
86d781fa04SBarry Smith 
87d781fa04SBarry Smith @*/
889371c9d4SSatish Balay PetscErrorCode PetscFormatConvert(const char *format, char *newformat) {
89e5c89e4eSSatish Balay   PetscInt i = 0, j = 0;
90e5c89e4eSSatish Balay 
91eed5747fSBarry Smith   PetscFunctionBegin;
92d781fa04SBarry Smith   while (format[i]) {
932a1ad9caSBarry Smith     if (format[i] == '%' && format[i + 1] == '%') {
942a1ad9caSBarry Smith       newformat[j++] = format[i++];
952a1ad9caSBarry Smith       newformat[j++] = format[i++];
962a1ad9caSBarry Smith     } else if (format[i] == '%') {
978627564fSBarry Smith       if (format[i + 1] == 'g') {
988627564fSBarry Smith         newformat[j++] = '[';
998627564fSBarry Smith         newformat[j++] = '|';
1008627564fSBarry Smith       }
1017bc47156SJose Roman       /* Find the letter */
1027bc47156SJose Roman       for (; format[i] && format[i] <= '9'; i++) newformat[j++] = format[i];
1037bc47156SJose Roman       switch (format[i]) {
1047bc47156SJose Roman       case 'D':
1056de02169SBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
106e5c89e4eSSatish Balay         newformat[j++] = 'd';
107e5c89e4eSSatish Balay #else
108e5c89e4eSSatish Balay         newformat[j++] = 'l';
109e5c89e4eSSatish Balay         newformat[j++] = 'l';
110e5c89e4eSSatish Balay         newformat[j++] = 'd';
111e5c89e4eSSatish Balay #endif
1127bc47156SJose Roman         break;
1138627564fSBarry Smith       case 'g':
1148627564fSBarry Smith         newformat[j++] = format[i];
1158627564fSBarry Smith         if (format[i - 1] == '%') {
1168627564fSBarry Smith           newformat[j++] = '|';
1178627564fSBarry Smith           newformat[j++] = ']';
1188627564fSBarry Smith         }
1198627564fSBarry Smith         break;
1209371c9d4SSatish Balay       case 'G': SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "%%G format is no longer supported, use %%g and cast the argument to double");
1219371c9d4SSatish Balay       case 'F': SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "%%F format is no longer supported, use %%f and cast the argument to double");
1229371c9d4SSatish Balay       default: newformat[j++] = format[i]; break;
1237bc47156SJose Roman       }
1247bc47156SJose Roman       i++;
125a297a907SKarl Rupp     } else newformat[j++] = format[i++];
126e5c89e4eSSatish Balay   }
127e5c89e4eSSatish Balay   newformat[j] = 0;
128eed5747fSBarry Smith   PetscFunctionReturn(0);
129e5c89e4eSSatish Balay }
130e5c89e4eSSatish Balay 
13114416c0eSBarry Smith #define PETSCDEFAULTBUFFERSIZE 8 * 1024
132d781fa04SBarry Smith 
133c9a19010SBarry Smith /*@C
134c9a19010SBarry Smith      PetscVSNPrintf - The PETSc version of vsnprintf(). Converts a PETSc format string into a standard C format string and then puts all the
135c9a19010SBarry Smith        function arguments into a string using the format statement.
136c9a19010SBarry Smith 
137c9a19010SBarry Smith    Input Parameters:
138c9a19010SBarry Smith +   str - location to put result
139c9a19010SBarry Smith .   len - the amount of space in str
140c9a19010SBarry Smith +   format - the PETSc format string
141c9a19010SBarry Smith -   fullLength - the amount of space in str actually used.
142c9a19010SBarry Smith 
14395452b02SPatrick Sanan     Developer Notes:
14495452b02SPatrick Sanan     this function may be called from an error handler, if an error occurs when it is called by the error handler than likely
145eed5747fSBarry Smith       a recursion will occur and possible crash.
146c9a19010SBarry Smith 
147c9a19010SBarry Smith  Level: developer
148c9a19010SBarry Smith 
149db781477SPatrick Sanan .seealso: `PetscVSNPrintf()`, `PetscErrorPrintf()`, `PetscVPrintf()`
150d781fa04SBarry Smith 
151c9a19010SBarry Smith @*/
1529371c9d4SSatish Balay PetscErrorCode PetscVSNPrintf(char *str, size_t len, const char *format, size_t *fullLength, va_list Argp) {
153d781fa04SBarry Smith   char  *newformat = NULL;
15414416c0eSBarry Smith   char   formatbuf[PETSCDEFAULTBUFFERSIZE];
155d781fa04SBarry Smith   size_t newLength;
15614416c0eSBarry Smith   int    flen;
157e5c89e4eSSatish Balay 
158eed5747fSBarry Smith   PetscFunctionBegin;
1599566063dSJacob Faibussowitsch   PetscCall(PetscFormatConvertGetSize(format, &newLength));
16094217ebdSBarry Smith   if (newLength < sizeof(formatbuf)) {
161e2135aedSMatthew Knepley     newformat = formatbuf;
16294217ebdSBarry Smith     newLength = sizeof(formatbuf) - 1;
163e2135aedSMatthew Knepley   } else {
1649566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(newLength, &newformat));
165e2135aedSMatthew Knepley   }
1669566063dSJacob Faibussowitsch   PetscCall(PetscFormatConvert(format, newformat));
1677b9a2d1bSSatish Balay #if defined(PETSC_HAVE_VSNPRINTF)
168152b30f0SSatish Balay   flen = vsnprintf(str, len, newformat, Argp);
169e5c89e4eSSatish Balay #else
17089b07760SSatish Balay #error "vsnprintf not found"
171e5c89e4eSSatish Balay #endif
172*48a46eb9SPierre Jolivet   if (newLength > sizeof(formatbuf) - 1) PetscCall(PetscFree(newformat));
1738627564fSBarry Smith   {
1748627564fSBarry Smith     PetscBool foundedot;
1758627564fSBarry Smith     size_t    cnt = 0, ncnt = 0, leng;
1769566063dSJacob Faibussowitsch     PetscCall(PetscStrlen(str, &leng));
17717ca8410SBarry Smith     if (leng > 4) {
1788627564fSBarry Smith       for (cnt = 0; cnt < leng - 4; cnt++) {
1798627564fSBarry Smith         if (str[cnt] == '[' && str[cnt + 1] == '|') {
180c540d043SBarry Smith           flen -= 4;
1819371c9d4SSatish Balay           cnt++;
1829371c9d4SSatish Balay           cnt++;
1838627564fSBarry Smith           foundedot = PETSC_FALSE;
1848627564fSBarry Smith           for (; cnt < leng - 1; cnt++) {
1858627564fSBarry Smith             if (str[cnt] == '|' && str[cnt + 1] == ']') {
1868627564fSBarry Smith               cnt++;
1878627564fSBarry Smith               if (!foundedot) str[ncnt++] = '.';
1888627564fSBarry Smith               ncnt--;
1898627564fSBarry Smith               break;
1908627564fSBarry Smith             } else {
1918627564fSBarry Smith               if (str[cnt] == 'e' || str[cnt] == '.') foundedot = PETSC_TRUE;
1928627564fSBarry Smith               str[ncnt++] = str[cnt];
1938627564fSBarry Smith             }
1948627564fSBarry Smith           }
1958627564fSBarry Smith         } else {
1968627564fSBarry Smith           str[ncnt] = str[cnt];
1978627564fSBarry Smith         }
1988627564fSBarry Smith         ncnt++;
1998627564fSBarry Smith       }
2008627564fSBarry Smith       while (cnt < leng) {
2019371c9d4SSatish Balay         str[ncnt] = str[cnt];
2029371c9d4SSatish Balay         ncnt++;
2039371c9d4SSatish Balay         cnt++;
2048627564fSBarry Smith       }
2058627564fSBarry Smith       str[ncnt] = 0;
2068627564fSBarry Smith     }
2078627564fSBarry Smith   }
208748e1b9dSBarry Smith #if defined(PETSC_HAVE_WINDOWS_H) && !defined(PETSC_HAVE__SET_OUTPUT_FORMAT)
209e51f71cfSBarry Smith   /* older Windows OS always produces e-+0np for floating point output; remove the extra 0 */
210748e1b9dSBarry Smith   {
211748e1b9dSBarry Smith     size_t cnt = 0, ncnt = 0, leng;
2129566063dSJacob Faibussowitsch     PetscCall(PetscStrlen(str, &leng));
213748e1b9dSBarry Smith     if (leng > 5) {
214748e1b9dSBarry Smith       for (cnt = 0; cnt < leng - 4; cnt++) {
215e51f71cfSBarry Smith         if (str[cnt] == 'e' && (str[cnt + 1] == '-' || str[cnt + 1] == '+') && str[cnt + 2] == '0' && str[cnt + 3] >= '0' && str[cnt + 3] <= '9' && str[cnt + 4] >= '0' && str[cnt + 4] <= '9') {
2169371c9d4SSatish Balay           str[ncnt] = str[cnt];
2179371c9d4SSatish Balay           ncnt++;
2189371c9d4SSatish Balay           cnt++;
2199371c9d4SSatish Balay           str[ncnt] = str[cnt];
2209371c9d4SSatish Balay           ncnt++;
2219371c9d4SSatish Balay           cnt++;
2229371c9d4SSatish Balay           cnt++;
223e51f71cfSBarry Smith           str[ncnt] = str[cnt];
224748e1b9dSBarry Smith         } else {
225748e1b9dSBarry Smith           str[ncnt] = str[cnt];
226748e1b9dSBarry Smith         }
227748e1b9dSBarry Smith         ncnt++;
228748e1b9dSBarry Smith       }
229748e1b9dSBarry Smith       while (cnt < leng) {
2309371c9d4SSatish Balay         str[ncnt] = str[cnt];
2319371c9d4SSatish Balay         ncnt++;
2329371c9d4SSatish Balay         cnt++;
233748e1b9dSBarry Smith       }
234748e1b9dSBarry Smith       str[ncnt] = 0;
235748e1b9dSBarry Smith     }
236748e1b9dSBarry Smith   }
237748e1b9dSBarry Smith #endif
238c540d043SBarry Smith   if (fullLength) *fullLength = 1 + (size_t)flen;
239eed5747fSBarry Smith   PetscFunctionReturn(0);
240e5c89e4eSSatish Balay }
241e5c89e4eSSatish Balay 
242c9a19010SBarry Smith /*@C
243c9a19010SBarry Smith      PetscVFPrintf -  All PETSc standard out and error messages are sent through this function; so, in theory, this can
244e5c89e4eSSatish Balay         can be replaced with something that does not simply write to a file.
245e5c89e4eSSatish Balay 
246c9a19010SBarry Smith       To use, write your own function for example,
247c9a19010SBarry Smith $PetscErrorCode mypetscvfprintf(FILE *fd,const char format[],va_list Argp)
248c9a19010SBarry Smith ${
249c9a19010SBarry Smith $  PetscErrorCode ierr;
250c9a19010SBarry Smith $
251c9a19010SBarry Smith $  PetscFunctionBegin;
252c9a19010SBarry Smith $   if (fd != stdout && fd != stderr) {  handle regular files
2535f80ce2aSJacob Faibussowitsch $      CHKERR(PetscVFPrintfDefault(fd,format,Argp));
254c9a19010SBarry Smith $  } else {
255c9a19010SBarry Smith $     char   buff[BIG];
256c9a19010SBarry Smith $     size_t length;
2579566063dSJacob Faibussowitsch $     PetscCall(PetscVSNPrintf(buff,BIG,format,&length,Argp));
258c9a19010SBarry Smith $     now send buff to whatever stream or whatever you want
259c9a19010SBarry Smith $ }
260c9a19010SBarry Smith $ PetscFunctionReturn(0);
261c9a19010SBarry Smith $}
262c9a19010SBarry Smith then before the call to PetscInitialize() do the assignment
263c9a19010SBarry Smith $    PetscVFPrintf = mypetscvfprintf;
264c9a19010SBarry Smith 
26595452b02SPatrick Sanan       Notes:
26695452b02SPatrick Sanan     For error messages this may be called by any process, for regular standard out it is
267e5c89e4eSSatish Balay           called only by process 0 of a given communicator
268e5c89e4eSSatish Balay 
26995452b02SPatrick Sanan       Developer Notes:
27095452b02SPatrick Sanan     this could be called by an error handler, if that happens then a recursion of the error handler may occur
271eed5747fSBarry Smith                        and a crash
272c9a19010SBarry Smith 
273c9a19010SBarry Smith   Level:  developer
274c9a19010SBarry Smith 
275db781477SPatrick Sanan .seealso: `PetscVSNPrintf()`, `PetscErrorPrintf()`
276c9a19010SBarry Smith 
277c9a19010SBarry Smith @*/
2789371c9d4SSatish Balay PetscErrorCode PetscVFPrintfDefault(FILE *fd, const char *format, va_list Argp) {
27914416c0eSBarry Smith   char   str[PETSCDEFAULTBUFFERSIZE];
28014416c0eSBarry Smith   char  *buff = str;
28114416c0eSBarry Smith   size_t fullLength;
2821531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
28314416c0eSBarry Smith   va_list Argpcopy;
2841531940fSBarry Smith #endif
2851179db26SBarry Smith 
286eed5747fSBarry Smith   PetscFunctionBegin;
2871531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
28814416c0eSBarry Smith   va_copy(Argpcopy, Argp);
2891531940fSBarry Smith #endif
2909566063dSJacob Faibussowitsch   PetscCall(PetscVSNPrintf(str, sizeof(str), format, &fullLength, Argp));
29114416c0eSBarry Smith   if (fullLength > sizeof(str)) {
2929566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(fullLength, &buff));
2931531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
2949566063dSJacob Faibussowitsch     PetscCall(PetscVSNPrintf(buff, fullLength, format, NULL, Argpcopy));
2951531940fSBarry Smith #else
2961531940fSBarry Smith     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "C89 does not support va_copy() hence cannot print long strings with PETSc printing routines");
2971531940fSBarry Smith #endif
29814416c0eSBarry Smith   }
2992f613bf5SBarry Smith   fprintf(fd, "%s", buff);
3005a058713SBarry Smith   fflush(fd);
301*48a46eb9SPierre Jolivet   if (buff != str) PetscCall(PetscFree(buff));
302eed5747fSBarry Smith   PetscFunctionReturn(0);
303e5c89e4eSSatish Balay }
304e5c89e4eSSatish Balay 
3055b5bc046SBarry Smith /*@C
3065b5bc046SBarry Smith     PetscSNPrintf - Prints to a string of given length
3075b5bc046SBarry Smith 
3085b5bc046SBarry Smith     Not Collective
3095b5bc046SBarry Smith 
3105b5bc046SBarry Smith     Input Parameters:
3115b5bc046SBarry Smith +   str - the string to print to
3125b5bc046SBarry Smith .   len - the length of str
3135b5bc046SBarry Smith .   format - the usual printf() format string
31410699b91SBarry Smith -   ... - any arguments that are to be printed, each much have an appropriate symbol in the format argument
3155b5bc046SBarry Smith 
3165b5bc046SBarry Smith    Level: intermediate
3175b5bc046SBarry Smith 
318db781477SPatrick Sanan .seealso: `PetscSynchronizedFlush()`, `PetscSynchronizedFPrintf()`, `PetscFPrintf()`, `PetscVSNPrintf()`,
319db781477SPatrick Sanan           `PetscPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscVFPrintf()`
3205b5bc046SBarry Smith @*/
3219371c9d4SSatish Balay PetscErrorCode PetscSNPrintf(char *str, size_t len, const char format[], ...) {
322c9a19010SBarry Smith   size_t  fullLength;
3235b5bc046SBarry Smith   va_list Argp;
3245b5bc046SBarry Smith 
3255b5bc046SBarry Smith   PetscFunctionBegin;
3265b5bc046SBarry Smith   va_start(Argp, format);
3279566063dSJacob Faibussowitsch   PetscCall(PetscVSNPrintf(str, len, format, &fullLength, Argp));
3285b5bc046SBarry Smith   PetscFunctionReturn(0);
3295b5bc046SBarry Smith }
3305b5bc046SBarry Smith 
331257d2499SJed Brown /*@C
332257d2499SJed Brown     PetscSNPrintfCount - Prints to a string of given length, returns count
333257d2499SJed Brown 
334257d2499SJed Brown     Not Collective
335257d2499SJed Brown 
336257d2499SJed Brown     Input Parameters:
337257d2499SJed Brown +   str - the string to print to
338257d2499SJed Brown .   len - the length of str
339257d2499SJed Brown .   format - the usual printf() format string
34010699b91SBarry Smith -   ... - any arguments that are to be printed, each much have an appropriate symbol in the format argument
341257d2499SJed Brown 
342cb398dd3SBarry Smith     Output Parameter:
343cb398dd3SBarry Smith .   countused - number of characters used
344cb398dd3SBarry Smith 
345257d2499SJed Brown    Level: intermediate
346257d2499SJed Brown 
347db781477SPatrick Sanan .seealso: `PetscSynchronizedFlush()`, `PetscSynchronizedFPrintf()`, `PetscFPrintf()`, `PetscVSNPrintf()`,
348db781477SPatrick Sanan           `PetscPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscSNPrintf()`, `PetscVFPrintf()`
349257d2499SJed Brown @*/
3509371c9d4SSatish Balay PetscErrorCode PetscSNPrintfCount(char *str, size_t len, const char format[], size_t *countused, ...) {
351257d2499SJed Brown   va_list Argp;
352257d2499SJed Brown 
353257d2499SJed Brown   PetscFunctionBegin;
354257d2499SJed Brown   va_start(Argp, countused);
3559566063dSJacob Faibussowitsch   PetscCall(PetscVSNPrintf(str, len, format, countused, Argp));
356257d2499SJed Brown   PetscFunctionReturn(0);
357257d2499SJed Brown }
358257d2499SJed Brown 
359e5c89e4eSSatish Balay /* ----------------------------------------------------------------------- */
360e5c89e4eSSatish Balay 
36102c9f0b5SLisandro Dalcin PrintfQueue petsc_printfqueue = NULL, petsc_printfqueuebase = NULL;
362d30b0576SJed Brown int         petsc_printfqueuelength = 0;
363e5c89e4eSSatish Balay 
364e5c89e4eSSatish Balay /*@C
365e5c89e4eSSatish Balay     PetscSynchronizedPrintf - Prints synchronized output from several processors.
366e5c89e4eSSatish Balay     Output of the first processor is followed by that of the second, etc.
367e5c89e4eSSatish Balay 
368e5c89e4eSSatish Balay     Not Collective
369e5c89e4eSSatish Balay 
370e5c89e4eSSatish Balay     Input Parameters:
371e5c89e4eSSatish Balay +   comm - the communicator
372e5c89e4eSSatish Balay -   format - the usual printf() format string
373e5c89e4eSSatish Balay 
374e5c89e4eSSatish Balay    Level: intermediate
375e5c89e4eSSatish Balay 
376e5c89e4eSSatish Balay     Notes:
3777889ec69SBarry Smith     REQUIRES a call to PetscSynchronizedFlush() by all the processes after the completion of the calls to PetscSynchronizedPrintf() for the information
378e5c89e4eSSatish Balay     from all the processors to be printed.
379e5c89e4eSSatish Balay 
380e5c89e4eSSatish Balay     Fortran Note:
381d60d70ccSBarry Smith     The call sequence is PetscSynchronizedPrintf(MPI_Comm, character(*), PetscErrorCode ierr) from Fortran.
382e5c89e4eSSatish Balay     That is, you can only pass a single character string from Fortran.
383e5c89e4eSSatish Balay 
384db781477SPatrick Sanan .seealso: `PetscSynchronizedFlush()`, `PetscSynchronizedFPrintf()`, `PetscFPrintf()`,
385db781477SPatrick Sanan           `PetscPrintf()`, `PetscViewerASCIIPrintf()`, `PetscViewerASCIISynchronizedPrintf()`
386e5c89e4eSSatish Balay @*/
3879371c9d4SSatish Balay PetscErrorCode PetscSynchronizedPrintf(MPI_Comm comm, const char format[], ...) {
388e5c89e4eSSatish Balay   PetscMPIInt rank;
389e5c89e4eSSatish Balay 
390e5c89e4eSSatish Balay   PetscFunctionBegin;
39108401ef6SPierre Jolivet   PetscCheck(comm != MPI_COMM_NULL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
3929566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
393e5c89e4eSSatish Balay 
394e5c89e4eSSatish Balay   /* First processor prints immediately to stdout */
395dd400576SPatrick Sanan   if (rank == 0) {
396e5c89e4eSSatish Balay     va_list Argp;
397e5c89e4eSSatish Balay     va_start(Argp, format);
3989566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(PETSC_STDOUT, format, Argp));
399e5c89e4eSSatish Balay     if (petsc_history) {
400cdc7d174SSatish Balay       va_start(Argp, format);
4019566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
402e5c89e4eSSatish Balay     }
403e5c89e4eSSatish Balay     va_end(Argp);
404e5c89e4eSSatish Balay   } else { /* other processors add to local queue */
405e5c89e4eSSatish Balay     va_list     Argp;
406e5c89e4eSSatish Balay     PrintfQueue next;
40714416c0eSBarry Smith     size_t      fullLength = PETSCDEFAULTBUFFERSIZE;
408e5c89e4eSSatish Balay 
4099566063dSJacob Faibussowitsch     PetscCall(PetscNew(&next));
410a297a907SKarl Rupp     if (petsc_printfqueue) {
411a297a907SKarl Rupp       petsc_printfqueue->next = next;
412a297a907SKarl Rupp       petsc_printfqueue       = next;
41302c9f0b5SLisandro Dalcin       petsc_printfqueue->next = NULL;
414a297a907SKarl Rupp     } else petsc_printfqueuebase = petsc_printfqueue = next;
415d30b0576SJed Brown     petsc_printfqueuelength++;
41694217ebdSBarry Smith     next->size   = 0;
41714416c0eSBarry Smith     next->string = NULL;
41894217ebdSBarry Smith     while (fullLength >= next->size) {
4192d609e63SMatthew Knepley       next->size = fullLength + 1;
4209566063dSJacob Faibussowitsch       PetscCall(PetscFree(next->string));
4219566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(next->size, &next->string));
422e5c89e4eSSatish Balay       va_start(Argp, format);
4239566063dSJacob Faibussowitsch       PetscCall(PetscArrayzero(next->string, next->size));
4249566063dSJacob Faibussowitsch       PetscCall(PetscVSNPrintf(next->string, next->size, format, &fullLength, Argp));
425e5c89e4eSSatish Balay       va_end(Argp);
426e5c89e4eSSatish Balay     }
4272d609e63SMatthew Knepley   }
428e5c89e4eSSatish Balay   PetscFunctionReturn(0);
429e5c89e4eSSatish Balay }
430e5c89e4eSSatish Balay 
431e5c89e4eSSatish Balay /*@C
432e5c89e4eSSatish Balay     PetscSynchronizedFPrintf - Prints synchronized output to the specified file from
433e5c89e4eSSatish Balay     several processors.  Output of the first processor is followed by that of the
434e5c89e4eSSatish Balay     second, etc.
435e5c89e4eSSatish Balay 
436e5c89e4eSSatish Balay     Not Collective
437e5c89e4eSSatish Balay 
438e5c89e4eSSatish Balay     Input Parameters:
439e5c89e4eSSatish Balay +   comm - the communicator
440e5c89e4eSSatish Balay .   fd - the file pointer
441e5c89e4eSSatish Balay -   format - the usual printf() format string
442e5c89e4eSSatish Balay 
443e5c89e4eSSatish Balay     Level: intermediate
444e5c89e4eSSatish Balay 
445e5c89e4eSSatish Balay     Notes:
446e5c89e4eSSatish Balay     REQUIRES a intervening call to PetscSynchronizedFlush() for the information
447e5c89e4eSSatish Balay     from all the processors to be printed.
448e5c89e4eSSatish Balay 
449db781477SPatrick Sanan .seealso: `PetscSynchronizedPrintf()`, `PetscSynchronizedFlush()`, `PetscFPrintf()`,
450db781477SPatrick Sanan           `PetscFOpen()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIPrintf()`
451e5c89e4eSSatish Balay 
452e5c89e4eSSatish Balay @*/
4539371c9d4SSatish Balay PetscErrorCode PetscSynchronizedFPrintf(MPI_Comm comm, FILE *fp, const char format[], ...) {
454e5c89e4eSSatish Balay   PetscMPIInt rank;
455e5c89e4eSSatish Balay 
456e5c89e4eSSatish Balay   PetscFunctionBegin;
45708401ef6SPierre Jolivet   PetscCheck(comm != MPI_COMM_NULL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
4589566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
459e5c89e4eSSatish Balay 
460e5c89e4eSSatish Balay   /* First processor prints immediately to fp */
461dd400576SPatrick Sanan   if (rank == 0) {
462e5c89e4eSSatish Balay     va_list Argp;
463e5c89e4eSSatish Balay     va_start(Argp, format);
4649566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(fp, format, Argp));
465cdc7d174SSatish Balay     if (petsc_history && (fp != petsc_history)) {
466cdc7d174SSatish Balay       va_start(Argp, format);
4679566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
468e5c89e4eSSatish Balay     }
469e5c89e4eSSatish Balay     va_end(Argp);
470e5c89e4eSSatish Balay   } else { /* other processors add to local queue */
471e5c89e4eSSatish Balay     va_list     Argp;
472e5c89e4eSSatish Balay     PrintfQueue next;
47314416c0eSBarry Smith     size_t      fullLength = PETSCDEFAULTBUFFERSIZE;
47414416c0eSBarry Smith 
4759566063dSJacob Faibussowitsch     PetscCall(PetscNew(&next));
476a297a907SKarl Rupp     if (petsc_printfqueue) {
477a297a907SKarl Rupp       petsc_printfqueue->next = next;
478a297a907SKarl Rupp       petsc_printfqueue       = next;
47902c9f0b5SLisandro Dalcin       petsc_printfqueue->next = NULL;
480a297a907SKarl Rupp     } else petsc_printfqueuebase = petsc_printfqueue = next;
481d30b0576SJed Brown     petsc_printfqueuelength++;
48294217ebdSBarry Smith     next->size   = 0;
48314416c0eSBarry Smith     next->string = NULL;
48494217ebdSBarry Smith     while (fullLength >= next->size) {
4852d609e63SMatthew Knepley       next->size = fullLength + 1;
4869566063dSJacob Faibussowitsch       PetscCall(PetscFree(next->string));
4879566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(next->size, &next->string));
488e5c89e4eSSatish Balay       va_start(Argp, format);
4899566063dSJacob Faibussowitsch       PetscCall(PetscArrayzero(next->string, next->size));
4909566063dSJacob Faibussowitsch       PetscCall(PetscVSNPrintf(next->string, next->size, format, &fullLength, Argp));
491e5c89e4eSSatish Balay       va_end(Argp);
492e5c89e4eSSatish Balay     }
4932d609e63SMatthew Knepley   }
494e5c89e4eSSatish Balay   PetscFunctionReturn(0);
495e5c89e4eSSatish Balay }
496e5c89e4eSSatish Balay 
4970ec8b6e3SBarry Smith /*@C
498e5c89e4eSSatish Balay     PetscSynchronizedFlush - Flushes to the screen output from all processors
4997889ec69SBarry Smith     involved in previous PetscSynchronizedPrintf()/PetscSynchronizedFPrintf() calls.
500e5c89e4eSSatish Balay 
501d083f849SBarry Smith     Collective
502e5c89e4eSSatish Balay 
503e5c89e4eSSatish Balay     Input Parameters:
5040ec8b6e3SBarry Smith +   comm - the communicator
5050ec8b6e3SBarry Smith -   fd - the file pointer (valid on process 0 of the communicator)
506e5c89e4eSSatish Balay 
507e5c89e4eSSatish Balay     Level: intermediate
508e5c89e4eSSatish Balay 
509e5c89e4eSSatish Balay     Notes:
5107889ec69SBarry Smith     If PetscSynchronizedPrintf() and/or PetscSynchronizedFPrintf() are called with
5117889ec69SBarry Smith     different MPI communicators there must be an intervening call to PetscSynchronizedFlush() between the calls with different MPI communicators.
512e5c89e4eSSatish Balay 
513e50bf69fSBarry Smith     From Fortran pass PETSC_STDOUT if the flush is for standard out; otherwise pass a value obtained from PetscFOpen()
514e50bf69fSBarry Smith 
515db781477SPatrick Sanan .seealso: `PetscSynchronizedPrintf()`, `PetscFPrintf()`, `PetscPrintf()`, `PetscViewerASCIIPrintf()`,
516db781477SPatrick Sanan           `PetscViewerASCIISynchronizedPrintf()`
5170087d953SMatthew G. Knepley @*/
5189371c9d4SSatish Balay PetscErrorCode PetscSynchronizedFlush(MPI_Comm comm, FILE *fd) {
51929a5cbdcSMatthew G. Knepley   PetscMPIInt rank, size, tag, i, j, n = 0, dummy = 0;
5202d609e63SMatthew Knepley   char       *message;
521e5c89e4eSSatish Balay   MPI_Status  status;
522e5c89e4eSSatish Balay 
523e5c89e4eSSatish Balay   PetscFunctionBegin;
5249566063dSJacob Faibussowitsch   PetscCall(PetscCommDuplicate(comm, &comm, &tag));
5259566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
5269566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(comm, &size));
527e5c89e4eSSatish Balay 
528e5c89e4eSSatish Balay   /* First processor waits for messages from all other processors */
529dd400576SPatrick Sanan   if (rank == 0) {
5300ec8b6e3SBarry Smith     if (!fd) fd = PETSC_STDOUT;
531e5c89e4eSSatish Balay     for (i = 1; i < size; i++) {
5329f73f8ecSBarry Smith       /* to prevent a flood of messages to process zero, request each message separately */
5339566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Send(&dummy, 1, MPI_INT, i, tag, comm));
5349566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Recv(&n, 1, MPI_INT, i, tag, comm, &status));
535e5c89e4eSSatish Balay       for (j = 0; j < n; j++) {
53629a5cbdcSMatthew G. Knepley         PetscMPIInt size = 0;
5372d609e63SMatthew Knepley 
5389566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Recv(&size, 1, MPI_INT, i, tag, comm, &status));
5399566063dSJacob Faibussowitsch         PetscCall(PetscMalloc1(size, &message));
5409566063dSJacob Faibussowitsch         PetscCallMPI(MPI_Recv(message, size, MPI_CHAR, i, tag, comm, &status));
5419566063dSJacob Faibussowitsch         PetscCall(PetscFPrintf(comm, fd, "%s", message));
5429566063dSJacob Faibussowitsch         PetscCall(PetscFree(message));
543e5c89e4eSSatish Balay       }
544e5c89e4eSSatish Balay     }
545e5c89e4eSSatish Balay   } else { /* other processors send queue to processor 0 */
546d30b0576SJed Brown     PrintfQueue next = petsc_printfqueuebase, previous;
547e5c89e4eSSatish Balay 
5489566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Recv(&dummy, 1, MPI_INT, 0, tag, comm, &status));
5499566063dSJacob Faibussowitsch     PetscCallMPI(MPI_Send(&petsc_printfqueuelength, 1, MPI_INT, 0, tag, comm));
550d30b0576SJed Brown     for (i = 0; i < petsc_printfqueuelength; i++) {
5519566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Send(&next->size, 1, MPI_INT, 0, tag, comm));
5529566063dSJacob Faibussowitsch       PetscCallMPI(MPI_Send(next->string, next->size, MPI_CHAR, 0, tag, comm));
553e5c89e4eSSatish Balay       previous = next;
554e5c89e4eSSatish Balay       next     = next->next;
5559566063dSJacob Faibussowitsch       PetscCall(PetscFree(previous->string));
5569566063dSJacob Faibussowitsch       PetscCall(PetscFree(previous));
557e5c89e4eSSatish Balay     }
55802c9f0b5SLisandro Dalcin     petsc_printfqueue       = NULL;
559d30b0576SJed Brown     petsc_printfqueuelength = 0;
560e5c89e4eSSatish Balay   }
5619566063dSJacob Faibussowitsch   PetscCall(PetscCommDestroy(&comm));
562e5c89e4eSSatish Balay   PetscFunctionReturn(0);
563e5c89e4eSSatish Balay }
564e5c89e4eSSatish Balay 
565e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------------------*/
566e5c89e4eSSatish Balay 
567e5c89e4eSSatish Balay /*@C
568e5c89e4eSSatish Balay     PetscFPrintf - Prints to a file, only from the first
569e5c89e4eSSatish Balay     processor in the communicator.
570e5c89e4eSSatish Balay 
571e5c89e4eSSatish Balay     Not Collective
572e5c89e4eSSatish Balay 
573e5c89e4eSSatish Balay     Input Parameters:
574e5c89e4eSSatish Balay +   comm - the communicator
575e5c89e4eSSatish Balay .   fd - the file pointer
576e5c89e4eSSatish Balay -   format - the usual printf() format string
577e5c89e4eSSatish Balay 
578e5c89e4eSSatish Balay     Level: intermediate
579e5c89e4eSSatish Balay 
580e5c89e4eSSatish Balay     Fortran Note:
581e5c89e4eSSatish Balay     This routine is not supported in Fortran.
582e5c89e4eSSatish Balay 
583db781477SPatrick Sanan .seealso: `PetscPrintf()`, `PetscSynchronizedPrintf()`, `PetscViewerASCIIPrintf()`,
584db781477SPatrick Sanan           `PetscViewerASCIISynchronizedPrintf()`, `PetscSynchronizedFlush()`
585e5c89e4eSSatish Balay @*/
5869371c9d4SSatish Balay PetscErrorCode PetscFPrintf(MPI_Comm comm, FILE *fd, const char format[], ...) {
587e5c89e4eSSatish Balay   PetscMPIInt rank;
588e5c89e4eSSatish Balay 
589e5c89e4eSSatish Balay   PetscFunctionBegin;
59008401ef6SPierre Jolivet   PetscCheck(comm != MPI_COMM_NULL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
5919566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
592dd400576SPatrick Sanan   if (rank == 0) {
593e5c89e4eSSatish Balay     va_list Argp;
594e5c89e4eSSatish Balay     va_start(Argp, format);
5959566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(fd, format, Argp));
596cdc7d174SSatish Balay     if (petsc_history && (fd != petsc_history)) {
597cdc7d174SSatish Balay       va_start(Argp, format);
5989566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
599e5c89e4eSSatish Balay     }
600e5c89e4eSSatish Balay     va_end(Argp);
601e5c89e4eSSatish Balay   }
602e5c89e4eSSatish Balay   PetscFunctionReturn(0);
603e5c89e4eSSatish Balay }
604e5c89e4eSSatish Balay 
605e5c89e4eSSatish Balay /*@C
606e5c89e4eSSatish Balay     PetscPrintf - Prints to standard out, only from the first
607eed5747fSBarry Smith     processor in the communicator. Calls from other processes are ignored.
608e5c89e4eSSatish Balay 
609e5c89e4eSSatish Balay     Not Collective
610e5c89e4eSSatish Balay 
611e5c89e4eSSatish Balay     Input Parameters:
612e5c89e4eSSatish Balay +   comm - the communicator
613e5c89e4eSSatish Balay -   format - the usual printf() format string
614e5c89e4eSSatish Balay 
615e5c89e4eSSatish Balay     Level: intermediate
616e5c89e4eSSatish Balay 
617b2706f25SRichard Tran Mills     Notes:
618b2706f25SRichard Tran Mills     PetscPrintf() supports some format specifiers that are unique to PETSc.
619b2706f25SRichard Tran Mills     See the manual page for PetscFormatConvert() for details.
620b2706f25SRichard Tran Mills 
621e5c89e4eSSatish Balay     Fortran Note:
622d60d70ccSBarry Smith     The call sequence is PetscPrintf(MPI_Comm, character(*), PetscErrorCode ierr) from Fortran.
623e5c89e4eSSatish Balay     That is, you can only pass a single character string from Fortran.
624e5c89e4eSSatish Balay 
625db781477SPatrick Sanan .seealso: `PetscFPrintf()`, `PetscSynchronizedPrintf()`, `PetscFormatConvert()`
626e5c89e4eSSatish Balay @*/
6279371c9d4SSatish Balay PetscErrorCode PetscPrintf(MPI_Comm comm, const char format[], ...) {
628e5c89e4eSSatish Balay   PetscMPIInt rank;
629e5c89e4eSSatish Balay 
630e5c89e4eSSatish Balay   PetscFunctionBegin;
63108401ef6SPierre Jolivet   PetscCheck(comm != MPI_COMM_NULL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
6329566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
633dd400576SPatrick Sanan   if (rank == 0) {
634e5c89e4eSSatish Balay     va_list Argp;
635e5c89e4eSSatish Balay     va_start(Argp, format);
6369566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(PETSC_STDOUT, format, Argp));
637e5c89e4eSSatish Balay     if (petsc_history) {
638cdc7d174SSatish Balay       va_start(Argp, format);
6399566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
640e5c89e4eSSatish Balay     }
641e5c89e4eSSatish Balay     va_end(Argp);
642e5c89e4eSSatish Balay   }
643e5c89e4eSSatish Balay   PetscFunctionReturn(0);
644e5c89e4eSSatish Balay }
645e5c89e4eSSatish Balay 
6469371c9d4SSatish Balay PetscErrorCode PetscHelpPrintfDefault(MPI_Comm comm, const char format[], ...) {
647e5c89e4eSSatish Balay   PetscMPIInt rank;
648e5c89e4eSSatish Balay 
649e5c89e4eSSatish Balay   PetscFunctionBegin;
65008401ef6SPierre Jolivet   PetscCheck(comm != MPI_COMM_NULL, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
6519566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
652dd400576SPatrick Sanan   if (rank == 0) {
653e5c89e4eSSatish Balay     va_list Argp;
654e5c89e4eSSatish Balay     va_start(Argp, format);
6559566063dSJacob Faibussowitsch     PetscCall((*PetscVFPrintf)(PETSC_STDOUT, format, Argp));
656e5c89e4eSSatish Balay     if (petsc_history) {
657cdc7d174SSatish Balay       va_start(Argp, format);
6589566063dSJacob Faibussowitsch       PetscCall((*PetscVFPrintf)(petsc_history, format, Argp));
659e5c89e4eSSatish Balay     }
660e5c89e4eSSatish Balay     va_end(Argp);
661e5c89e4eSSatish Balay   }
662e5c89e4eSSatish Balay   PetscFunctionReturn(0);
663e5c89e4eSSatish Balay }
664e5c89e4eSSatish Balay 
665e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------------------*/
666e5c89e4eSSatish Balay 
667e5c89e4eSSatish Balay /*@C
668e5c89e4eSSatish Balay     PetscSynchronizedFGets - Several processors all get the same line from a file.
669e5c89e4eSSatish Balay 
670d083f849SBarry Smith     Collective
671e5c89e4eSSatish Balay 
672e5c89e4eSSatish Balay     Input Parameters:
673e5c89e4eSSatish Balay +   comm - the communicator
674e5c89e4eSSatish Balay .   fd - the file pointer
675e5c89e4eSSatish Balay -   len - the length of the output buffer
676e5c89e4eSSatish Balay 
677e5c89e4eSSatish Balay     Output Parameter:
678e31d4fa4SJed Brown .   string - the line read from the file, at end of file string[0] == 0
679e5c89e4eSSatish Balay 
680e5c89e4eSSatish Balay     Level: intermediate
681e5c89e4eSSatish Balay 
682db781477SPatrick Sanan .seealso: `PetscSynchronizedPrintf()`, `PetscSynchronizedFlush()`,
683db781477SPatrick Sanan           `PetscFOpen()`, `PetscViewerASCIISynchronizedPrintf()`, `PetscViewerASCIIPrintf()`
684e5c89e4eSSatish Balay 
685e5c89e4eSSatish Balay @*/
6869371c9d4SSatish Balay PetscErrorCode PetscSynchronizedFGets(MPI_Comm comm, FILE *fp, size_t len, char string[]) {
687e5c89e4eSSatish Balay   PetscMPIInt rank;
688e5c89e4eSSatish Balay 
689e5c89e4eSSatish Balay   PetscFunctionBegin;
6909566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
691e5c89e4eSSatish Balay 
692dd400576SPatrick Sanan   if (rank == 0) {
693047b9c12SMatthew G Knepley     char *ptr = fgets(string, len, fp);
694047b9c12SMatthew G Knepley 
695047b9c12SMatthew G Knepley     if (!ptr) {
696e31d4fa4SJed Brown       string[0] = 0;
69708401ef6SPierre Jolivet       PetscCheck(feof(fp), PETSC_COMM_SELF, PETSC_ERR_FILE_READ, "Error reading from file: %d", errno);
698047b9c12SMatthew G Knepley     }
699e5c89e4eSSatish Balay   }
7009566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Bcast(string, len, MPI_BYTE, 0, comm));
701e5c89e4eSSatish Balay   PetscFunctionReturn(0);
702e5c89e4eSSatish Balay }
703238ccf28SShri Abhyankar 
7042475b7caSBarry Smith #if defined(PETSC_HAVE_CLOSURE)
705f1d7fe2eSBarry Smith int (^SwiftClosure)(const char *) = 0;
706f1d7fe2eSBarry Smith 
7079371c9d4SSatish Balay PetscErrorCode PetscVFPrintfToString(FILE *fd, const char format[], va_list Argp) {
708f1d7fe2eSBarry Smith   PetscFunctionBegin;
709f1d7fe2eSBarry Smith   if (fd != stdout && fd != stderr) { /* handle regular files */
7109566063dSJacob Faibussowitsch     PetscCall(PetscVFPrintfDefault(fd, format, Argp));
711f1d7fe2eSBarry Smith   } else {
71214416c0eSBarry Smith     size_t length;
71314416c0eSBarry Smith     char   buff[PETSCDEFAULTBUFFERSIZE];
714f1d7fe2eSBarry Smith 
7159566063dSJacob Faibussowitsch     PetscCall(PetscVSNPrintf(buff, sizeof(buff), format, &length, Argp));
7169566063dSJacob Faibussowitsch     PetscCall(SwiftClosure(buff));
717f1d7fe2eSBarry Smith   }
718f1d7fe2eSBarry Smith   PetscFunctionReturn(0);
719f1d7fe2eSBarry Smith }
720f1d7fe2eSBarry Smith 
721f1d7fe2eSBarry Smith /*
722f1d7fe2eSBarry Smith    Provide a Swift function that processes all the PETSc calls to PetscVFPrintf()
723f1d7fe2eSBarry Smith */
7249371c9d4SSatish Balay PetscErrorCode PetscVFPrintfSetClosure(int (^closure)(const char *)) {
725f1d7fe2eSBarry Smith   PetscVFPrintf = PetscVFPrintfToString;
726f1d7fe2eSBarry Smith   SwiftClosure  = closure;
727f1d7fe2eSBarry Smith   return 0;
728f1d7fe2eSBarry Smith }
729f1d7fe2eSBarry Smith #endif
730f1d7fe2eSBarry Smith 
7318c74ee41SBarry Smith /*@C
7328c74ee41SBarry Smith      PetscFormatStrip - Takes a PETSc format string and removes all numerical modifiers to % operations
7338c74ee41SBarry Smith 
7348c74ee41SBarry Smith    Input Parameters:
7358c74ee41SBarry Smith .   format - the PETSc format string
7368c74ee41SBarry Smith 
7378c74ee41SBarry Smith  Level: developer
7388c74ee41SBarry Smith 
7398c74ee41SBarry Smith @*/
7409371c9d4SSatish Balay PetscErrorCode PetscFormatStrip(char *format) {
7418c74ee41SBarry Smith   size_t loc1 = 0, loc2 = 0;
7428c74ee41SBarry Smith 
7438c74ee41SBarry Smith   PetscFunctionBegin;
7448c74ee41SBarry Smith   while (format[loc2]) {
7458c74ee41SBarry Smith     if (format[loc2] == '%') {
7468c74ee41SBarry Smith       format[loc1++] = format[loc2++];
7478c74ee41SBarry Smith       while (format[loc2] && ((format[loc2] >= '0' && format[loc2] <= '9') || format[loc2] == '.')) loc2++;
7488c74ee41SBarry Smith     }
7498c74ee41SBarry Smith     format[loc1++] = format[loc2++];
7508c74ee41SBarry Smith   }
7518c74ee41SBarry Smith   PetscFunctionReturn(0);
7528c74ee41SBarry Smith }
7538c74ee41SBarry Smith 
7549371c9d4SSatish Balay PetscErrorCode PetscFormatRealArray(char buf[], size_t len, const char *fmt, PetscInt n, const PetscReal x[]) {
7551b5687a1SBarry Smith   PetscInt i;
7561b5687a1SBarry Smith   size_t   left, count;
7571b5687a1SBarry Smith   char    *p;
7581b5687a1SBarry Smith 
7591b5687a1SBarry Smith   PetscFunctionBegin;
7601b5687a1SBarry Smith   for (i = 0, p = buf, left = len; i < n; i++) {
7619566063dSJacob Faibussowitsch     PetscCall(PetscSNPrintfCount(p, left, fmt, &count, (double)x[i]));
76208401ef6SPierre Jolivet     PetscCheck(count < left, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Insufficient space in buffer");
7631b5687a1SBarry Smith     left -= count;
7641b5687a1SBarry Smith     p += count - 1;
7651b5687a1SBarry Smith     *p++ = ' ';
7661b5687a1SBarry Smith   }
7671b5687a1SBarry Smith   p[i ? 0 : -1] = 0;
7681b5687a1SBarry Smith   PetscFunctionReturn(0);
7691b5687a1SBarry Smith }
770