xref: /petsc/src/sys/fileio/mprint.c (revision 152b30f0700ecb9af2871cf9d7b123df7729bd86)
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 */
10e5c89e4eSSatish Balay extern 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 */
16e5c89e4eSSatish Balay FILE *PETSC_STDOUT = 0;
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 */
22ae9b4142SLisandro Dalcin FILE *PETSC_STDERR = 0;
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 
35d781fa04SBarry Smith .seealso: PetscFormatConvert(), PetscVSNPrintf(), PetscVFPrintf()
36d781fa04SBarry Smith 
37c9a19010SBarry Smith @*/
38d781fa04SBarry Smith PetscErrorCode PetscFormatConvertGetSize(const char *format,size_t *size)
39d781fa04SBarry Smith {
40d781fa04SBarry Smith   PetscInt i = 0;
41d781fa04SBarry Smith 
42d781fa04SBarry Smith   PetscFunctionBegin;
43d781fa04SBarry Smith   *size = 0;
44d781fa04SBarry Smith   while (format[i]) {
45d781fa04SBarry Smith     if (format[i] == '%' && format[i+1] == '%') {
46d781fa04SBarry Smith       i++; i++; *size += 2;
47d781fa04SBarry Smith     } else if (format[i] == '%') {
48d781fa04SBarry Smith       /* Find the letter */
49d781fa04SBarry Smith       for (; format[i] && format[i] <= '9'; i++,(*size += 1));
50d781fa04SBarry Smith       switch (format[i]) {
51d781fa04SBarry Smith       case 'D':
52d781fa04SBarry Smith #if defined(PETSC_USE_64BIT_INDICES)
53d781fa04SBarry Smith         *size += 2;
54d781fa04SBarry Smith #endif
55d781fa04SBarry Smith         break;
56d781fa04SBarry Smith       case 'g':
57d781fa04SBarry Smith         *size += 4;
58d781fa04SBarry Smith         break;
59d781fa04SBarry Smith       default:
60d781fa04SBarry Smith         break;
61d781fa04SBarry Smith       }
62d781fa04SBarry Smith       *size += 1;
63d781fa04SBarry Smith       i++;
64d781fa04SBarry Smith     } else {
65d781fa04SBarry Smith       i++;
66d781fa04SBarry Smith       *size += 1;
67d781fa04SBarry Smith     }
68d781fa04SBarry Smith   }
69d781fa04SBarry Smith   *size += 1; /* space for NULL character */
70d781fa04SBarry Smith   PetscFunctionReturn(0);
71d781fa04SBarry Smith }
72d781fa04SBarry Smith 
73d781fa04SBarry Smith /*@C
74d781fa04SBarry 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
75d781fa04SBarry Smith                         converts %g to [|%g|] so that PetscVSNPrintf() can easily insure all %g formatted numbers have a decimal point when printed.
76d781fa04SBarry Smith 
77d781fa04SBarry Smith    Input Parameters:
78d781fa04SBarry Smith +   format - the PETSc format string
79d781fa04SBarry Smith .   newformat - the location to put the new format
80d781fa04SBarry Smith -   size - the length of newformat, you can use PetscFormatConvertGetSize() to compute the needed size
81d781fa04SBarry Smith 
82d781fa04SBarry Smith     Note: this exists so we can have the same code when PetscInt is either int or long long int
83d781fa04SBarry Smith 
84d781fa04SBarry Smith  Level: developer
85d781fa04SBarry Smith 
86d781fa04SBarry Smith .seealso: PetscFormatConvertGetSize(), PetscVSNPrintf(), PetscVFPrintf()
87d781fa04SBarry Smith 
88d781fa04SBarry Smith @*/
89d781fa04SBarry Smith PetscErrorCode PetscFormatConvert(const char *format,char *newformat)
90e5c89e4eSSatish Balay {
91e5c89e4eSSatish Balay   PetscInt i = 0, j = 0;
92e5c89e4eSSatish Balay 
93eed5747fSBarry Smith   PetscFunctionBegin;
94d781fa04SBarry Smith   while (format[i]) {
952a1ad9caSBarry Smith     if (format[i] == '%' && format[i+1] == '%') {
962a1ad9caSBarry Smith       newformat[j++] = format[i++];
972a1ad9caSBarry Smith       newformat[j++] = format[i++];
982a1ad9caSBarry Smith     } else if (format[i] == '%') {
998627564fSBarry Smith       if (format[i+1] == 'g') {
1008627564fSBarry Smith         newformat[j++] = '[';
1018627564fSBarry Smith         newformat[j++] = '|';
1028627564fSBarry Smith       }
1037bc47156SJose Roman       /* Find the letter */
1047bc47156SJose Roman       for (; format[i] && format[i] <= '9'; i++) newformat[j++] = format[i];
1057bc47156SJose Roman       switch (format[i]) {
1067bc47156SJose Roman       case 'D':
1076de02169SBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
108e5c89e4eSSatish Balay         newformat[j++] = 'd';
109e5c89e4eSSatish Balay #else
110e5c89e4eSSatish Balay         newformat[j++] = 'l';
111e5c89e4eSSatish Balay         newformat[j++] = 'l';
112e5c89e4eSSatish Balay         newformat[j++] = 'd';
113e5c89e4eSSatish Balay #endif
1147bc47156SJose Roman         break;
1158627564fSBarry Smith       case 'g':
1168627564fSBarry Smith         newformat[j++] = format[i];
1178627564fSBarry Smith         if (format[i-1] == '%') {
1188627564fSBarry Smith           newformat[j++] = '|';
1198627564fSBarry Smith           newformat[j++] = ']';
1208627564fSBarry Smith         }
1218627564fSBarry Smith         break;
1227bc47156SJose Roman       case 'G':
123145505e7SDominic Meiser         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"%%G format is no longer supported, use %%g and cast the argument to double");
1247bc47156SJose Roman         break;
125121a09c4SJose Roman       case 'F':
126145505e7SDominic Meiser         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"%%F format is no longer supported, use %%f and cast the argument to double");
127121a09c4SJose Roman         break;
1287bc47156SJose Roman       default:
1297bc47156SJose Roman         newformat[j++] = format[i];
1307bc47156SJose Roman         break;
1317bc47156SJose Roman       }
1327bc47156SJose Roman       i++;
133a297a907SKarl Rupp     } else newformat[j++] = format[i++];
134e5c89e4eSSatish Balay   }
135e5c89e4eSSatish Balay   newformat[j] = 0;
136eed5747fSBarry Smith   PetscFunctionReturn(0);
137e5c89e4eSSatish Balay }
138e5c89e4eSSatish Balay 
13914416c0eSBarry Smith #define PETSCDEFAULTBUFFERSIZE 8*1024
140d781fa04SBarry Smith 
141c9a19010SBarry Smith /*@C
142c9a19010SBarry Smith      PetscVSNPrintf - The PETSc version of vsnprintf(). Converts a PETSc format string into a standard C format string and then puts all the
143c9a19010SBarry Smith        function arguments into a string using the format statement.
144c9a19010SBarry Smith 
145c9a19010SBarry Smith    Input Parameters:
146c9a19010SBarry Smith +   str - location to put result
147c9a19010SBarry Smith .   len - the amount of space in str
148c9a19010SBarry Smith +   format - the PETSc format string
149c9a19010SBarry Smith -   fullLength - the amount of space in str actually used.
150c9a19010SBarry Smith 
151eed5747fSBarry Smith     Developer Notes: this function may be called from an error handler, if an error occurs when it is called by the error handler than likely
152eed5747fSBarry Smith       a recursion will occur and possible crash.
153c9a19010SBarry Smith 
154c9a19010SBarry Smith  Level: developer
155c9a19010SBarry Smith 
156d781fa04SBarry Smith .seealso: PetscVSNPrintf(), PetscErrorPrintf(), PetscVPrintf()
157d781fa04SBarry Smith 
158c9a19010SBarry Smith @*/
1597087cfbeSBarry Smith PetscErrorCode PetscVSNPrintf(char *str,size_t len,const char *format,size_t *fullLength,va_list Argp)
160e5c89e4eSSatish Balay {
161d781fa04SBarry Smith   char           *newformat = NULL;
16214416c0eSBarry Smith   char           formatbuf[PETSCDEFAULTBUFFERSIZE];
163d781fa04SBarry Smith   size_t         newLength;
1641a15ef5dSMatthew Knepley   PetscErrorCode ierr;
16514416c0eSBarry Smith   int            flen;
166e5c89e4eSSatish Balay 
167eed5747fSBarry Smith   PetscFunctionBegin;
168d781fa04SBarry Smith   ierr = PetscFormatConvertGetSize(format,&newLength);CHKERRQ(ierr);
16914416c0eSBarry Smith   if (newLength < PETSCDEFAULTBUFFERSIZE) {
170e2135aedSMatthew Knepley     newformat = formatbuf;
17114416c0eSBarry Smith     newLength = PETSCDEFAULTBUFFERSIZE-1;
172e2135aedSMatthew Knepley   } else {
173d781fa04SBarry Smith     ierr      = PetscMalloc1(newLength, &newformat);CHKERRQ(ierr);
174e2135aedSMatthew Knepley   }
175d781fa04SBarry Smith   ierr = PetscFormatConvert(format,newformat);CHKERRQ(ierr);
1765a058713SBarry Smith #if defined(PETSC_HAVE_VSNPRINTF_CHAR)
177*152b30f0SSatish Balay   flen = vsnprintf(str,len,newformat,(char*)Argp);
17889b07760SSatish Balay #elif defined(PETSC_HAVE_VSNPRINTF)
179*152b30f0SSatish Balay   flen = vsnprintf(str,len,newformat,Argp);
180141c0d65SSatish Balay #elif defined(PETSC_HAVE__VSNPRINTF)
181*152b30f0SSatish Balay   flen = _vsnprintf(str,len,newformat,Argp);
182e5c89e4eSSatish Balay #else
18389b07760SSatish Balay #error "vsnprintf not found"
184e5c89e4eSSatish Balay #endif
18514416c0eSBarry Smith   if (fullLength) *fullLength = 1 + (size_t) flen;
18614416c0eSBarry Smith   if (newLength > PETSCDEFAULTBUFFERSIZE-1) {
187e2135aedSMatthew Knepley     ierr = PetscFree(newformat);CHKERRQ(ierr);
188e2135aedSMatthew Knepley   }
1898627564fSBarry Smith   {
1908627564fSBarry Smith     PetscBool foundedot;
1918627564fSBarry Smith     size_t cnt = 0,ncnt = 0,leng;
1928627564fSBarry Smith     ierr = PetscStrlen(str,&leng);CHKERRQ(ierr);
19317ca8410SBarry Smith     if (leng > 4) {
1948627564fSBarry Smith       for (cnt=0; cnt<leng-4; cnt++) {
1958627564fSBarry Smith         if (str[cnt] == '[' && str[cnt+1] == '|'){
1968627564fSBarry Smith            cnt++; cnt++;
1978627564fSBarry Smith            foundedot = PETSC_FALSE;
1988627564fSBarry Smith            for (; cnt<leng-1; cnt++) {
1998627564fSBarry Smith              if (str[cnt] == '|' && str[cnt+1] == ']'){
2008627564fSBarry Smith                cnt++;
2018627564fSBarry Smith                if (!foundedot) str[ncnt++] = '.';
2028627564fSBarry Smith                ncnt--;
2038627564fSBarry Smith                break;
2048627564fSBarry Smith              } else {
2058627564fSBarry Smith                if (str[cnt] == 'e' || str[cnt] == '.') foundedot = PETSC_TRUE;
2068627564fSBarry Smith                str[ncnt++] = str[cnt];
2078627564fSBarry Smith              }
2088627564fSBarry Smith            }
2098627564fSBarry Smith         } else {
2108627564fSBarry Smith           str[ncnt] = str[cnt];
2118627564fSBarry Smith         }
2128627564fSBarry Smith         ncnt++;
2138627564fSBarry Smith       }
2148627564fSBarry Smith       while (cnt < leng) {
2158627564fSBarry Smith         str[ncnt] = str[cnt]; ncnt++; cnt++;
2168627564fSBarry Smith       }
2178627564fSBarry Smith       str[ncnt] = 0;
2188627564fSBarry Smith     }
2198627564fSBarry Smith   }
220748e1b9dSBarry Smith #if defined(PETSC_HAVE_WINDOWS_H) && !defined(PETSC_HAVE__SET_OUTPUT_FORMAT)
221e51f71cfSBarry Smith   /* older Windows OS always produces e-+0np for floating point output; remove the extra 0 */
222748e1b9dSBarry Smith   {
223748e1b9dSBarry Smith     size_t cnt = 0,ncnt = 0,leng;
224748e1b9dSBarry Smith     ierr = PetscStrlen(str,&leng);CHKERRQ(ierr);
225748e1b9dSBarry Smith     if (leng > 5) {
226748e1b9dSBarry Smith       for (cnt=0; cnt<leng-4; cnt++) {
227e51f71cfSBarry 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') {
228748e1b9dSBarry Smith           str[ncnt] = str[cnt]; ncnt++; cnt++;
229e51f71cfSBarry Smith           str[ncnt] = str[cnt]; ncnt++; cnt++; cnt++;
230e51f71cfSBarry Smith           str[ncnt] = str[cnt];
231748e1b9dSBarry Smith         } else {
232748e1b9dSBarry Smith           str[ncnt] = str[cnt];
233748e1b9dSBarry Smith         }
234748e1b9dSBarry Smith         ncnt++;
235748e1b9dSBarry Smith       }
236748e1b9dSBarry Smith       while (cnt < leng) {
237748e1b9dSBarry Smith         str[ncnt] = str[cnt]; ncnt++; cnt++;
238748e1b9dSBarry Smith       }
239748e1b9dSBarry Smith       str[ncnt] = 0;
240748e1b9dSBarry Smith     }
241748e1b9dSBarry Smith   }
242748e1b9dSBarry Smith #endif
243eed5747fSBarry Smith   PetscFunctionReturn(0);
244e5c89e4eSSatish Balay }
245e5c89e4eSSatish Balay 
246c9a19010SBarry Smith /*@C
247c9a19010SBarry Smith      PetscVFPrintf -  All PETSc standard out and error messages are sent through this function; so, in theory, this can
248e5c89e4eSSatish Balay         can be replaced with something that does not simply write to a file.
249e5c89e4eSSatish Balay 
250c9a19010SBarry Smith       To use, write your own function for example,
251c9a19010SBarry Smith $PetscErrorCode mypetscvfprintf(FILE *fd,const char format[],va_list Argp)
252c9a19010SBarry Smith ${
253c9a19010SBarry Smith $  PetscErrorCode ierr;
254c9a19010SBarry Smith $
255c9a19010SBarry Smith $  PetscFunctionBegin;
256c9a19010SBarry Smith $   if (fd != stdout && fd != stderr) {  handle regular files
257c9a19010SBarry Smith $      ierr = PetscVFPrintfDefault(fd,format,Argp);CHKERR(ierr);
258c9a19010SBarry Smith $  } else {
259c9a19010SBarry Smith $     char   buff[BIG];
260c9a19010SBarry Smith $     size_t length;
261c9a19010SBarry Smith $     ierr = PetscVSNPrintf(buff,BIG,format,&length,Argp);CHKERRQ(ierr);
262c9a19010SBarry Smith $     now send buff to whatever stream or whatever you want
263c9a19010SBarry Smith $ }
264c9a19010SBarry Smith $ PetscFunctionReturn(0);
265c9a19010SBarry Smith $}
266c9a19010SBarry Smith then before the call to PetscInitialize() do the assignment
267c9a19010SBarry Smith $    PetscVFPrintf = mypetscvfprintf;
268c9a19010SBarry Smith 
269c9a19010SBarry Smith       Notes: For error messages this may be called by any process, for regular standard out it is
270e5c89e4eSSatish Balay           called only by process 0 of a given communicator
271e5c89e4eSSatish Balay 
272eed5747fSBarry Smith       Developer Notes: this could be called by an error handler, if that happens then a recursion of the error handler may occur
273eed5747fSBarry Smith                        and a crash
274c9a19010SBarry Smith 
275c9a19010SBarry Smith   Level:  developer
276c9a19010SBarry Smith 
277c9a19010SBarry Smith .seealso: PetscVSNPrintf(), PetscErrorPrintf()
278c9a19010SBarry Smith 
279c9a19010SBarry Smith @*/
2807087cfbeSBarry Smith PetscErrorCode PetscVFPrintfDefault(FILE *fd,const char *format,va_list Argp)
281e5c89e4eSSatish Balay {
28214416c0eSBarry Smith   char           str[PETSCDEFAULTBUFFERSIZE];
28314416c0eSBarry Smith   char           *buff = str;
28414416c0eSBarry Smith   size_t         fullLength;
285eed5747fSBarry Smith   PetscErrorCode ierr;
2861531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
28714416c0eSBarry Smith   va_list        Argpcopy;
2881531940fSBarry Smith #endif
2891179db26SBarry Smith 
290eed5747fSBarry Smith   PetscFunctionBegin;
2911531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
29214416c0eSBarry Smith   va_copy(Argpcopy,Argp);
2931531940fSBarry Smith #endif
29414416c0eSBarry Smith   ierr = PetscVSNPrintf(str,sizeof(str),format,&fullLength,Argp);CHKERRQ(ierr);
29514416c0eSBarry Smith   if (fullLength > sizeof(str)) {
29614416c0eSBarry Smith     ierr = PetscMalloc1(fullLength,&buff);CHKERRQ(ierr);
2971531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY)
29814416c0eSBarry Smith     ierr = PetscVSNPrintf(buff,fullLength,format,NULL,Argpcopy);CHKERRQ(ierr);
2991531940fSBarry Smith #else
3001531940fSBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"C89 does not support va_copy() hence cannot print long strings with PETSc printing routines");
3011531940fSBarry Smith #endif
30214416c0eSBarry Smith   }
30314416c0eSBarry Smith   fprintf(fd,"%s",buff);CHKERRQ(ierr);
3045a058713SBarry Smith   fflush(fd);
30514416c0eSBarry Smith   if (buff != str) {
30614416c0eSBarry Smith     ierr = PetscFree(buff);CHKERRQ(ierr);
30714416c0eSBarry Smith   }
308eed5747fSBarry Smith   PetscFunctionReturn(0);
309e5c89e4eSSatish Balay }
310e5c89e4eSSatish Balay 
3115b5bc046SBarry Smith /*@C
3125b5bc046SBarry Smith     PetscSNPrintf - Prints to a string of given length
3135b5bc046SBarry Smith 
3145b5bc046SBarry Smith     Not Collective
3155b5bc046SBarry Smith 
3165b5bc046SBarry Smith     Input Parameters:
3175b5bc046SBarry Smith +   str - the string to print to
3185b5bc046SBarry Smith .   len - the length of str
3195b5bc046SBarry Smith .   format - the usual printf() format string
3205b5bc046SBarry Smith -   any arguments
3215b5bc046SBarry Smith 
3225b5bc046SBarry Smith    Level: intermediate
3235b5bc046SBarry Smith 
3245b5bc046SBarry Smith .seealso: PetscSynchronizedFlush(), PetscSynchronizedFPrintf(), PetscFPrintf(), PetscVSNPrintf(),
325d781fa04SBarry Smith           PetscPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf(), PetscVFPrintf()
3265b5bc046SBarry Smith @*/
3277087cfbeSBarry Smith PetscErrorCode PetscSNPrintf(char *str,size_t len,const char format[],...)
3285b5bc046SBarry Smith {
3295b5bc046SBarry Smith   PetscErrorCode ierr;
330c9a19010SBarry Smith   size_t         fullLength;
3315b5bc046SBarry Smith   va_list        Argp;
3325b5bc046SBarry Smith 
3335b5bc046SBarry Smith   PetscFunctionBegin;
3345b5bc046SBarry Smith   va_start(Argp,format);
3352d609e63SMatthew Knepley   ierr = PetscVSNPrintf(str,len,format,&fullLength,Argp);CHKERRQ(ierr);
3365b5bc046SBarry Smith   PetscFunctionReturn(0);
3375b5bc046SBarry Smith }
3385b5bc046SBarry Smith 
339257d2499SJed Brown /*@C
340257d2499SJed Brown     PetscSNPrintfCount - Prints to a string of given length, returns count
341257d2499SJed Brown 
342257d2499SJed Brown     Not Collective
343257d2499SJed Brown 
344257d2499SJed Brown     Input Parameters:
345257d2499SJed Brown +   str - the string to print to
346257d2499SJed Brown .   len - the length of str
347257d2499SJed Brown .   format - the usual printf() format string
348257d2499SJed Brown .   countused - number of characters used
349257d2499SJed Brown -   any arguments
350257d2499SJed Brown 
351257d2499SJed Brown    Level: intermediate
352257d2499SJed Brown 
353257d2499SJed Brown .seealso: PetscSynchronizedFlush(), PetscSynchronizedFPrintf(), PetscFPrintf(), PetscVSNPrintf(),
354d781fa04SBarry Smith           PetscPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf(), PetscSNPrintf(), PetscVFPrintf()
355257d2499SJed Brown @*/
356257d2499SJed Brown PetscErrorCode PetscSNPrintfCount(char *str,size_t len,const char format[],size_t *countused,...)
357257d2499SJed Brown {
358257d2499SJed Brown   PetscErrorCode ierr;
359257d2499SJed Brown   va_list        Argp;
360257d2499SJed Brown 
361257d2499SJed Brown   PetscFunctionBegin;
362257d2499SJed Brown   va_start(Argp,countused);
363257d2499SJed Brown   ierr = PetscVSNPrintf(str,len,format,countused,Argp);CHKERRQ(ierr);
364257d2499SJed Brown   PetscFunctionReturn(0);
365257d2499SJed Brown }
366257d2499SJed Brown 
367e5c89e4eSSatish Balay /* ----------------------------------------------------------------------- */
368e5c89e4eSSatish Balay 
369d30b0576SJed Brown PrintfQueue petsc_printfqueue       = 0,petsc_printfqueuebase = 0;
370d30b0576SJed Brown int         petsc_printfqueuelength = 0;
371e5c89e4eSSatish Balay 
372e5c89e4eSSatish Balay /*@C
373e5c89e4eSSatish Balay     PetscSynchronizedPrintf - Prints synchronized output from several processors.
374e5c89e4eSSatish Balay     Output of the first processor is followed by that of the second, etc.
375e5c89e4eSSatish Balay 
376e5c89e4eSSatish Balay     Not Collective
377e5c89e4eSSatish Balay 
378e5c89e4eSSatish Balay     Input Parameters:
379e5c89e4eSSatish Balay +   comm - the communicator
380e5c89e4eSSatish Balay -   format - the usual printf() format string
381e5c89e4eSSatish Balay 
382e5c89e4eSSatish Balay    Level: intermediate
383e5c89e4eSSatish Balay 
384e5c89e4eSSatish Balay     Notes:
3857889ec69SBarry Smith     REQUIRES a call to PetscSynchronizedFlush() by all the processes after the completion of the calls to PetscSynchronizedPrintf() for the information
386e5c89e4eSSatish Balay     from all the processors to be printed.
387e5c89e4eSSatish Balay 
388e5c89e4eSSatish Balay     Fortran Note:
389d60d70ccSBarry Smith     The call sequence is PetscSynchronizedPrintf(MPI_Comm, character(*), PetscErrorCode ierr) from Fortran.
390e5c89e4eSSatish Balay     That is, you can only pass a single character string from Fortran.
391e5c89e4eSSatish Balay 
392e5c89e4eSSatish Balay .seealso: PetscSynchronizedFlush(), PetscSynchronizedFPrintf(), PetscFPrintf(),
393e5c89e4eSSatish Balay           PetscPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf()
394e5c89e4eSSatish Balay @*/
3957087cfbeSBarry Smith PetscErrorCode PetscSynchronizedPrintf(MPI_Comm comm,const char format[],...)
396e5c89e4eSSatish Balay {
397e5c89e4eSSatish Balay   PetscErrorCode ierr;
398e5c89e4eSSatish Balay   PetscMPIInt    rank;
399e5c89e4eSSatish Balay 
400e5c89e4eSSatish Balay   PetscFunctionBegin;
4016180deefSBarry Smith   if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
402e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
403e5c89e4eSSatish Balay 
404e5c89e4eSSatish Balay   /* First processor prints immediately to stdout */
405e5c89e4eSSatish Balay   if (!rank) {
406e5c89e4eSSatish Balay     va_list Argp;
407e5c89e4eSSatish Balay     va_start(Argp,format);
4081179db26SBarry Smith     ierr = (*PetscVFPrintf)(PETSC_STDOUT,format,Argp);CHKERRQ(ierr);
409e5c89e4eSSatish Balay     if (petsc_history) {
410cdc7d174SSatish Balay       va_start(Argp,format);
4111179db26SBarry Smith       ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr);
412e5c89e4eSSatish Balay     }
413e5c89e4eSSatish Balay     va_end(Argp);
414e5c89e4eSSatish Balay   } else { /* other processors add to local queue */
415e5c89e4eSSatish Balay     va_list     Argp;
416e5c89e4eSSatish Balay     PrintfQueue next;
41714416c0eSBarry Smith     size_t      fullLength = PETSCDEFAULTBUFFERSIZE;
418e5c89e4eSSatish Balay 
419b00a9115SJed Brown     ierr = PetscNew(&next);CHKERRQ(ierr);
420a297a907SKarl Rupp     if (petsc_printfqueue) {
421a297a907SKarl Rupp       petsc_printfqueue->next = next;
422a297a907SKarl Rupp       petsc_printfqueue       = next;
423a297a907SKarl Rupp       petsc_printfqueue->next = 0;
424a297a907SKarl Rupp     } else petsc_printfqueuebase = petsc_printfqueue = next;
425d30b0576SJed Brown     petsc_printfqueuelength++;
4262d609e63SMatthew Knepley     next->size   = -1;
42714416c0eSBarry Smith     next->string = NULL;
4289ad23270SJed Brown     while ((PetscInt)fullLength >= next->size) {
4292d609e63SMatthew Knepley       next->size = fullLength+1;
43014416c0eSBarry Smith       ierr = PetscFree(next->string);CHKERRQ(ierr);
431785e854fSJed Brown       ierr = PetscMalloc1(next->size, &next->string);CHKERRQ(ierr);
432e5c89e4eSSatish Balay       va_start(Argp,format);
4332d609e63SMatthew Knepley       ierr = PetscMemzero(next->string,next->size);CHKERRQ(ierr);
4342d609e63SMatthew Knepley       ierr = PetscVSNPrintf(next->string,next->size,format, &fullLength,Argp);CHKERRQ(ierr);
435e5c89e4eSSatish Balay       va_end(Argp);
436e5c89e4eSSatish Balay     }
4372d609e63SMatthew Knepley   }
438e5c89e4eSSatish Balay   PetscFunctionReturn(0);
439e5c89e4eSSatish Balay }
440e5c89e4eSSatish Balay 
441e5c89e4eSSatish Balay /*@C
442e5c89e4eSSatish Balay     PetscSynchronizedFPrintf - Prints synchronized output to the specified file from
443e5c89e4eSSatish Balay     several processors.  Output of the first processor is followed by that of the
444e5c89e4eSSatish Balay     second, etc.
445e5c89e4eSSatish Balay 
446e5c89e4eSSatish Balay     Not Collective
447e5c89e4eSSatish Balay 
448e5c89e4eSSatish Balay     Input Parameters:
449e5c89e4eSSatish Balay +   comm - the communicator
450e5c89e4eSSatish Balay .   fd - the file pointer
451e5c89e4eSSatish Balay -   format - the usual printf() format string
452e5c89e4eSSatish Balay 
453e5c89e4eSSatish Balay     Level: intermediate
454e5c89e4eSSatish Balay 
455e5c89e4eSSatish Balay     Notes:
456e5c89e4eSSatish Balay     REQUIRES a intervening call to PetscSynchronizedFlush() for the information
457e5c89e4eSSatish Balay     from all the processors to be printed.
458e5c89e4eSSatish Balay 
459e5c89e4eSSatish Balay .seealso: PetscSynchronizedPrintf(), PetscSynchronizedFlush(), PetscFPrintf(),
460e5c89e4eSSatish Balay           PetscFOpen(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIPrintf()
461e5c89e4eSSatish Balay 
462e5c89e4eSSatish Balay @*/
4637087cfbeSBarry Smith PetscErrorCode PetscSynchronizedFPrintf(MPI_Comm comm,FILE *fp,const char format[],...)
464e5c89e4eSSatish Balay {
465e5c89e4eSSatish Balay   PetscErrorCode ierr;
466e5c89e4eSSatish Balay   PetscMPIInt    rank;
467e5c89e4eSSatish Balay 
468e5c89e4eSSatish Balay   PetscFunctionBegin;
4696180deefSBarry Smith   if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
470e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
471e5c89e4eSSatish Balay 
472e5c89e4eSSatish Balay   /* First processor prints immediately to fp */
473e5c89e4eSSatish Balay   if (!rank) {
474e5c89e4eSSatish Balay     va_list Argp;
475e5c89e4eSSatish Balay     va_start(Argp,format);
4761179db26SBarry Smith     ierr = (*PetscVFPrintf)(fp,format,Argp);CHKERRQ(ierr);
477cdc7d174SSatish Balay     if (petsc_history && (fp !=petsc_history)) {
478cdc7d174SSatish Balay       va_start(Argp,format);
4791179db26SBarry Smith       ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr);
480e5c89e4eSSatish Balay     }
481e5c89e4eSSatish Balay     va_end(Argp);
482e5c89e4eSSatish Balay   } else { /* other processors add to local queue */
483e5c89e4eSSatish Balay     va_list     Argp;
484e5c89e4eSSatish Balay     PrintfQueue next;
48514416c0eSBarry Smith     size_t      fullLength = PETSCDEFAULTBUFFERSIZE;
48614416c0eSBarry Smith 
487b00a9115SJed Brown     ierr = PetscNew(&next);CHKERRQ(ierr);
488a297a907SKarl Rupp     if (petsc_printfqueue) {
489a297a907SKarl Rupp       petsc_printfqueue->next = next;
490a297a907SKarl Rupp       petsc_printfqueue       = next;
491a297a907SKarl Rupp       petsc_printfqueue->next = 0;
492a297a907SKarl Rupp     } else petsc_printfqueuebase = petsc_printfqueue = next;
493d30b0576SJed Brown     petsc_printfqueuelength++;
4942d609e63SMatthew Knepley     next->size   = -1;
49514416c0eSBarry Smith     next->string = NULL;
4969ad23270SJed Brown     while ((PetscInt)fullLength >= next->size) {
4972d609e63SMatthew Knepley       next->size = fullLength+1;
49814416c0eSBarry Smith       ierr = PetscFree(next->string);CHKERRQ(ierr);
499785e854fSJed Brown       ierr = PetscMalloc1(next->size, &next->string);CHKERRQ(ierr);
500e5c89e4eSSatish Balay       va_start(Argp,format);
5012d609e63SMatthew Knepley       ierr = PetscMemzero(next->string,next->size);CHKERRQ(ierr);
5022d609e63SMatthew Knepley       ierr = PetscVSNPrintf(next->string,next->size,format,&fullLength,Argp);CHKERRQ(ierr);
503e5c89e4eSSatish Balay       va_end(Argp);
504e5c89e4eSSatish Balay     }
5052d609e63SMatthew Knepley   }
506e5c89e4eSSatish Balay   PetscFunctionReturn(0);
507e5c89e4eSSatish Balay }
508e5c89e4eSSatish Balay 
5090ec8b6e3SBarry Smith /*@C
510e5c89e4eSSatish Balay     PetscSynchronizedFlush - Flushes to the screen output from all processors
5117889ec69SBarry Smith     involved in previous PetscSynchronizedPrintf()/PetscSynchronizedFPrintf() calls.
512e5c89e4eSSatish Balay 
513e5c89e4eSSatish Balay     Collective on MPI_Comm
514e5c89e4eSSatish Balay 
515e5c89e4eSSatish Balay     Input Parameters:
5160ec8b6e3SBarry Smith +   comm - the communicator
5170ec8b6e3SBarry Smith -   fd - the file pointer (valid on process 0 of the communicator)
518e5c89e4eSSatish Balay 
519e5c89e4eSSatish Balay     Level: intermediate
520e5c89e4eSSatish Balay 
521e5c89e4eSSatish Balay     Notes:
5227889ec69SBarry Smith     If PetscSynchronizedPrintf() and/or PetscSynchronizedFPrintf() are called with
5237889ec69SBarry Smith     different MPI communicators there must be an intervening call to PetscSynchronizedFlush() between the calls with different MPI communicators.
524e5c89e4eSSatish Balay 
525e50bf69fSBarry Smith     From Fortran pass PETSC_STDOUT if the flush is for standard out; otherwise pass a value obtained from PetscFOpen()
526e50bf69fSBarry Smith 
527e5c89e4eSSatish Balay .seealso: PetscSynchronizedPrintf(), PetscFPrintf(), PetscPrintf(), PetscViewerASCIIPrintf(),
528e5c89e4eSSatish Balay           PetscViewerASCIISynchronizedPrintf()
5290087d953SMatthew G. Knepley @*/
5300ec8b6e3SBarry Smith PetscErrorCode PetscSynchronizedFlush(MPI_Comm comm,FILE *fd)
531e5c89e4eSSatish Balay {
532e5c89e4eSSatish Balay   PetscErrorCode ierr;
53329a5cbdcSMatthew G. Knepley   PetscMPIInt    rank,size,tag,i,j,n = 0,dummy = 0;
5342d609e63SMatthew Knepley   char          *message;
535e5c89e4eSSatish Balay   MPI_Status     status;
536e5c89e4eSSatish Balay 
537e5c89e4eSSatish Balay   PetscFunctionBegin;
538e5c89e4eSSatish Balay   ierr = PetscCommDuplicate(comm,&comm,&tag);CHKERRQ(ierr);
539e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
540e5c89e4eSSatish Balay   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
541e5c89e4eSSatish Balay 
542e5c89e4eSSatish Balay   /* First processor waits for messages from all other processors */
543e5c89e4eSSatish Balay   if (!rank) {
5440ec8b6e3SBarry Smith     if (!fd) fd = PETSC_STDOUT;
545e5c89e4eSSatish Balay     for (i=1; i<size; i++) {
5469f73f8ecSBarry Smith       /* to prevent a flood of messages to process zero, request each message separately */
5479f73f8ecSBarry Smith       ierr = MPI_Send(&dummy,1,MPI_INT,i,tag,comm);CHKERRQ(ierr);
548e5c89e4eSSatish Balay       ierr = MPI_Recv(&n,1,MPI_INT,i,tag,comm,&status);CHKERRQ(ierr);
549e5c89e4eSSatish Balay       for (j=0; j<n; j++) {
55029a5cbdcSMatthew G. Knepley         PetscMPIInt size = 0;
5512d609e63SMatthew Knepley 
5522d609e63SMatthew Knepley         ierr = MPI_Recv(&size,1,MPI_INT,i,tag,comm,&status);CHKERRQ(ierr);
553785e854fSJed Brown         ierr = PetscMalloc1(size, &message);CHKERRQ(ierr);
5542d609e63SMatthew Knepley         ierr = MPI_Recv(message,size,MPI_CHAR,i,tag,comm,&status);CHKERRQ(ierr);
5553bf036e2SBarry Smith         ierr = PetscFPrintf(comm,fd,"%s",message);CHKERRQ(ierr);
5562d609e63SMatthew Knepley         ierr = PetscFree(message);CHKERRQ(ierr);
557e5c89e4eSSatish Balay       }
558e5c89e4eSSatish Balay     }
559e5c89e4eSSatish Balay   } else { /* other processors send queue to processor 0 */
560d30b0576SJed Brown     PrintfQueue next = petsc_printfqueuebase,previous;
561e5c89e4eSSatish Balay 
562b3ef9d35SBarry Smith     ierr = MPI_Recv(&dummy,1,MPI_INT,0,tag,comm,&status);CHKERRQ(ierr);
563d30b0576SJed Brown     ierr = MPI_Send(&petsc_printfqueuelength,1,MPI_INT,0,tag,comm);CHKERRQ(ierr);
564d30b0576SJed Brown     for (i=0; i<petsc_printfqueuelength; i++) {
5652d609e63SMatthew Knepley       ierr     = MPI_Send(&next->size,1,MPI_INT,0,tag,comm);CHKERRQ(ierr);
5662d609e63SMatthew Knepley       ierr     = MPI_Send(next->string,next->size,MPI_CHAR,0,tag,comm);CHKERRQ(ierr);
567e5c89e4eSSatish Balay       previous = next;
568e5c89e4eSSatish Balay       next     = next->next;
5692d609e63SMatthew Knepley       ierr     = PetscFree(previous->string);CHKERRQ(ierr);
570e5c89e4eSSatish Balay       ierr     = PetscFree(previous);CHKERRQ(ierr);
571e5c89e4eSSatish Balay     }
572d30b0576SJed Brown     petsc_printfqueue       = 0;
573d30b0576SJed Brown     petsc_printfqueuelength = 0;
574e5c89e4eSSatish Balay   }
575e5c89e4eSSatish Balay   ierr = PetscCommDestroy(&comm);CHKERRQ(ierr);
576e5c89e4eSSatish Balay   PetscFunctionReturn(0);
577e5c89e4eSSatish Balay }
578e5c89e4eSSatish Balay 
579e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------------------*/
580e5c89e4eSSatish Balay 
581e5c89e4eSSatish Balay /*@C
582e5c89e4eSSatish Balay     PetscFPrintf - Prints to a file, only from the first
583e5c89e4eSSatish Balay     processor in the communicator.
584e5c89e4eSSatish Balay 
585e5c89e4eSSatish Balay     Not Collective
586e5c89e4eSSatish Balay 
587e5c89e4eSSatish Balay     Input Parameters:
588e5c89e4eSSatish Balay +   comm - the communicator
589e5c89e4eSSatish Balay .   fd - the file pointer
590e5c89e4eSSatish Balay -   format - the usual printf() format string
591e5c89e4eSSatish Balay 
592e5c89e4eSSatish Balay     Level: intermediate
593e5c89e4eSSatish Balay 
594e5c89e4eSSatish Balay     Fortran Note:
595e5c89e4eSSatish Balay     This routine is not supported in Fortran.
596e5c89e4eSSatish Balay 
597e5c89e4eSSatish Balay    Concepts: printing^in parallel
598e5c89e4eSSatish Balay    Concepts: printf^in parallel
599e5c89e4eSSatish Balay 
600e5c89e4eSSatish Balay .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(),
601e5c89e4eSSatish Balay           PetscViewerASCIISynchronizedPrintf(), PetscSynchronizedFlush()
602e5c89e4eSSatish Balay @*/
6037087cfbeSBarry Smith PetscErrorCode PetscFPrintf(MPI_Comm comm,FILE* fd,const char format[],...)
604e5c89e4eSSatish Balay {
605e5c89e4eSSatish Balay   PetscErrorCode ierr;
606e5c89e4eSSatish Balay   PetscMPIInt    rank;
607e5c89e4eSSatish Balay 
608e5c89e4eSSatish Balay   PetscFunctionBegin;
6096180deefSBarry Smith   if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
610e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
611e5c89e4eSSatish Balay   if (!rank) {
612e5c89e4eSSatish Balay     va_list Argp;
613e5c89e4eSSatish Balay     va_start(Argp,format);
6141179db26SBarry Smith     ierr = (*PetscVFPrintf)(fd,format,Argp);CHKERRQ(ierr);
615cdc7d174SSatish Balay     if (petsc_history && (fd !=petsc_history)) {
616cdc7d174SSatish Balay       va_start(Argp,format);
6171179db26SBarry Smith       ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr);
618e5c89e4eSSatish Balay     }
619e5c89e4eSSatish Balay     va_end(Argp);
620e5c89e4eSSatish Balay   }
621e5c89e4eSSatish Balay   PetscFunctionReturn(0);
622e5c89e4eSSatish Balay }
623e5c89e4eSSatish Balay 
624e5c89e4eSSatish Balay /*@C
625e5c89e4eSSatish Balay     PetscPrintf - Prints to standard out, only from the first
626eed5747fSBarry Smith     processor in the communicator. Calls from other processes are ignored.
627e5c89e4eSSatish Balay 
628e5c89e4eSSatish Balay     Not Collective
629e5c89e4eSSatish Balay 
630e5c89e4eSSatish Balay     Input Parameters:
631e5c89e4eSSatish Balay +   comm - the communicator
632e5c89e4eSSatish Balay -   format - the usual printf() format string
633e5c89e4eSSatish Balay 
634e5c89e4eSSatish Balay    Level: intermediate
635e5c89e4eSSatish Balay 
636e5c89e4eSSatish Balay     Fortran Note:
637d60d70ccSBarry Smith     The call sequence is PetscPrintf(MPI_Comm, character(*), PetscErrorCode ierr) from Fortran.
638e5c89e4eSSatish Balay     That is, you can only pass a single character string from Fortran.
639e5c89e4eSSatish Balay 
640e5c89e4eSSatish Balay    Concepts: printing^in parallel
641e5c89e4eSSatish Balay    Concepts: printf^in parallel
642e5c89e4eSSatish Balay 
643e5c89e4eSSatish Balay .seealso: PetscFPrintf(), PetscSynchronizedPrintf()
644e5c89e4eSSatish Balay @*/
6457087cfbeSBarry Smith PetscErrorCode PetscPrintf(MPI_Comm comm,const char format[],...)
646e5c89e4eSSatish Balay {
647e5c89e4eSSatish Balay   PetscErrorCode ierr;
648e5c89e4eSSatish Balay   PetscMPIInt    rank;
649e5c89e4eSSatish Balay 
650e5c89e4eSSatish Balay   PetscFunctionBegin;
6516180deefSBarry Smith   if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
652e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
653e5c89e4eSSatish Balay   if (!rank) {
654e5c89e4eSSatish Balay     va_list Argp;
655e5c89e4eSSatish Balay     va_start(Argp,format);
656eed5747fSBarry Smith     ierr = (*PetscVFPrintf)(PETSC_STDOUT,format,Argp);CHKERRQ(ierr);
657e5c89e4eSSatish Balay     if (petsc_history) {
658cdc7d174SSatish Balay       va_start(Argp,format);
659eed5747fSBarry Smith       ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr);
660e5c89e4eSSatish Balay     }
661e5c89e4eSSatish Balay     va_end(Argp);
662e5c89e4eSSatish Balay   }
663e5c89e4eSSatish Balay   PetscFunctionReturn(0);
664e5c89e4eSSatish Balay }
665e5c89e4eSSatish Balay 
666e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------------------*/
667c9a19010SBarry Smith /*@C
668c9a19010SBarry Smith      PetscHelpPrintf -  All PETSc help messages are passing through this function. You can change how help messages are printed by
669c9a19010SBarry Smith         replacinng it  with something that does not simply write to a stdout.
670c9a19010SBarry Smith 
671c9a19010SBarry Smith       To use, write your own function for example,
672c9a19010SBarry Smith $PetscErrorCode mypetschelpprintf(MPI_Comm comm,const char format[],....)
673c9a19010SBarry Smith ${
674c9a19010SBarry Smith $ PetscFunctionReturn(0);
675c9a19010SBarry Smith $}
676c9a19010SBarry Smith then before the call to PetscInitialize() do the assignment
677c9a19010SBarry Smith $    PetscHelpPrintf = mypetschelpprintf;
678c9a19010SBarry Smith 
679c9a19010SBarry Smith   Note: the default routine used is called PetscHelpPrintfDefault().
680c9a19010SBarry Smith 
681c9a19010SBarry Smith   Level:  developer
682c9a19010SBarry Smith 
683c9a19010SBarry Smith .seealso: PetscVSNPrintf(), PetscVFPrintf(), PetscErrorPrintf()
684c9a19010SBarry Smith @*/
6857087cfbeSBarry Smith PetscErrorCode PetscHelpPrintfDefault(MPI_Comm comm,const char format[],...)
686e5c89e4eSSatish Balay {
687e5c89e4eSSatish Balay   PetscErrorCode ierr;
688e5c89e4eSSatish Balay   PetscMPIInt    rank;
689e5c89e4eSSatish Balay 
690e5c89e4eSSatish Balay   PetscFunctionBegin;
6916180deefSBarry Smith   if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed");
692e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
693e5c89e4eSSatish Balay   if (!rank) {
694e5c89e4eSSatish Balay     va_list Argp;
695e5c89e4eSSatish Balay     va_start(Argp,format);
6961179db26SBarry Smith     ierr = (*PetscVFPrintf)(PETSC_STDOUT,format,Argp);CHKERRQ(ierr);
697e5c89e4eSSatish Balay     if (petsc_history) {
698cdc7d174SSatish Balay       va_start(Argp,format);
6991179db26SBarry Smith       ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr);
700e5c89e4eSSatish Balay     }
701e5c89e4eSSatish Balay     va_end(Argp);
702e5c89e4eSSatish Balay   }
703e5c89e4eSSatish Balay   PetscFunctionReturn(0);
704e5c89e4eSSatish Balay }
705e5c89e4eSSatish Balay 
706e5c89e4eSSatish Balay /* ---------------------------------------------------------------------------------------*/
707e5c89e4eSSatish Balay 
708e5c89e4eSSatish Balay 
709e5c89e4eSSatish Balay /*@C
710e5c89e4eSSatish Balay     PetscSynchronizedFGets - Several processors all get the same line from a file.
711e5c89e4eSSatish Balay 
712e5c89e4eSSatish Balay     Collective on MPI_Comm
713e5c89e4eSSatish Balay 
714e5c89e4eSSatish Balay     Input Parameters:
715e5c89e4eSSatish Balay +   comm - the communicator
716e5c89e4eSSatish Balay .   fd - the file pointer
717e5c89e4eSSatish Balay -   len - the length of the output buffer
718e5c89e4eSSatish Balay 
719e5c89e4eSSatish Balay     Output Parameter:
720e31d4fa4SJed Brown .   string - the line read from the file, at end of file string[0] == 0
721e5c89e4eSSatish Balay 
722e5c89e4eSSatish Balay     Level: intermediate
723e5c89e4eSSatish Balay 
724e5c89e4eSSatish Balay .seealso: PetscSynchronizedPrintf(), PetscSynchronizedFlush(),
725e5c89e4eSSatish Balay           PetscFOpen(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIPrintf()
726e5c89e4eSSatish Balay 
727e5c89e4eSSatish Balay @*/
7287087cfbeSBarry Smith PetscErrorCode PetscSynchronizedFGets(MPI_Comm comm,FILE *fp,size_t len,char string[])
729e5c89e4eSSatish Balay {
730e5c89e4eSSatish Balay   PetscErrorCode ierr;
731e5c89e4eSSatish Balay   PetscMPIInt    rank;
732e5c89e4eSSatish Balay 
733e5c89e4eSSatish Balay   PetscFunctionBegin;
734e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
735e5c89e4eSSatish Balay 
736e5c89e4eSSatish Balay   if (!rank) {
737047b9c12SMatthew G Knepley     char *ptr = fgets(string, len, fp);
738047b9c12SMatthew G Knepley 
739047b9c12SMatthew G Knepley     if (!ptr) {
740e31d4fa4SJed Brown       string[0] = 0;
741e31d4fa4SJed Brown       if (!feof(fp)) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_FILE_READ, "Error reading from file: %d", errno);
742047b9c12SMatthew G Knepley     }
743e5c89e4eSSatish Balay   }
744e5c89e4eSSatish Balay   ierr = MPI_Bcast(string,len,MPI_BYTE,0,comm);CHKERRQ(ierr);
745e5c89e4eSSatish Balay   PetscFunctionReturn(0);
746e5c89e4eSSatish Balay }
747238ccf28SShri Abhyankar 
748f1d7fe2eSBarry Smith #if defined(PETSC_HAVE_CLOSURES)
749f1d7fe2eSBarry Smith int (^SwiftClosure)(const char*) = 0;
750f1d7fe2eSBarry Smith 
751f1d7fe2eSBarry Smith PetscErrorCode PetscVFPrintfToString(FILE *fd,const char format[],va_list Argp)
752f1d7fe2eSBarry Smith {
753f1d7fe2eSBarry Smith   PetscErrorCode ierr;
754f1d7fe2eSBarry Smith 
755f1d7fe2eSBarry Smith   PetscFunctionBegin;
756f1d7fe2eSBarry Smith   if (fd != stdout && fd != stderr) { /* handle regular files */
757f1d7fe2eSBarry Smith     ierr = PetscVFPrintfDefault(fd,format,Argp);CHKERRQ(ierr);
758f1d7fe2eSBarry Smith   } else {
75914416c0eSBarry Smith     size_t length;
76014416c0eSBarry Smith     char   buff[PETSCDEFAULTBUFFERSIZE];
761f1d7fe2eSBarry Smith 
76214416c0eSBarry Smith     ierr = PetscVSNPrintf(buf,size(buff),format,&length,Argp);CHKERRQ(ierr);
76314416c0eSBarry Smith     ierr = SwiftClosure(buff);CHKERRQ(ierr);
764f1d7fe2eSBarry Smith   }
765f1d7fe2eSBarry Smith   PetscFunctionReturn(0);
766f1d7fe2eSBarry Smith }
767f1d7fe2eSBarry Smith 
768f1d7fe2eSBarry Smith /*
769f1d7fe2eSBarry Smith    Provide a Swift function that processes all the PETSc calls to PetscVFPrintf()
770f1d7fe2eSBarry Smith */
771f1d7fe2eSBarry Smith PetscErrorCode PetscVFPrintfSetClosure(int (^closure)(const char*))
772f1d7fe2eSBarry Smith {
773f1d7fe2eSBarry Smith   PetscVFPrintf = PetscVFPrintfToString;
774f1d7fe2eSBarry Smith   SwiftClosure  = closure;
775f1d7fe2eSBarry Smith   return 0;
776f1d7fe2eSBarry Smith }
777f1d7fe2eSBarry Smith #endif
778f1d7fe2eSBarry Smith 
779238ccf28SShri Abhyankar #if defined(PETSC_HAVE_MATLAB_ENGINE)
780c6db04a5SJed Brown #include <mex.h>
7817087cfbeSBarry Smith PetscErrorCode PetscVFPrintf_Matlab(FILE *fd,const char format[],va_list Argp)
782238ccf28SShri Abhyankar {
783238ccf28SShri Abhyankar   PetscErrorCode ierr;
784238ccf28SShri Abhyankar 
785238ccf28SShri Abhyankar   PetscFunctionBegin;
786238ccf28SShri Abhyankar   if (fd != stdout && fd != stderr) { /* handle regular files */
787238ccf28SShri Abhyankar     ierr = PetscVFPrintfDefault(fd,format,Argp);CHKERRQ(ierr);
788238ccf28SShri Abhyankar   } else {
78914416c0eSBarry Smith     size_t length;
790cb500232SBarry Smith     char   buff[length];
791238ccf28SShri Abhyankar 
79214416c0eSBarry Smith     ierr = PetscVSNPrintf(buff,sizeof(buff),format,&length,Argp);CHKERRQ(ierr);
793cb500232SBarry Smith     mexPrintf("%s",buff);
794238ccf28SShri Abhyankar   }
795238ccf28SShri Abhyankar   PetscFunctionReturn(0);
796238ccf28SShri Abhyankar }
797238ccf28SShri Abhyankar #endif
7986fc7ef2bSBarry Smith 
7998c74ee41SBarry Smith /*@C
8008c74ee41SBarry Smith      PetscFormatStrip - Takes a PETSc format string and removes all numerical modifiers to % operations
8018c74ee41SBarry Smith 
8028c74ee41SBarry Smith    Input Parameters:
8038c74ee41SBarry Smith .   format - the PETSc format string
8048c74ee41SBarry Smith 
8058c74ee41SBarry Smith  Level: developer
8068c74ee41SBarry Smith 
8078c74ee41SBarry Smith @*/
8088c74ee41SBarry Smith PetscErrorCode PetscFormatStrip(char *format)
8098c74ee41SBarry Smith {
8108c74ee41SBarry Smith   size_t loc1 = 0, loc2 = 0;
8118c74ee41SBarry Smith 
8128c74ee41SBarry Smith   PetscFunctionBegin;
8138c74ee41SBarry Smith   while (format[loc2]) {
8148c74ee41SBarry Smith     if (format[loc2] == '%') {
8158c74ee41SBarry Smith       format[loc1++] = format[loc2++];
8168c74ee41SBarry Smith       while (format[loc2] && ((format[loc2] >= '0' && format[loc2] <= '9') || format[loc2] == '.')) loc2++;
8178c74ee41SBarry Smith     }
8188c74ee41SBarry Smith     format[loc1++] = format[loc2++];
8198c74ee41SBarry Smith   }
8208c74ee41SBarry Smith   PetscFunctionReturn(0);
8218c74ee41SBarry Smith }
8228c74ee41SBarry Smith 
8231b5687a1SBarry Smith PetscErrorCode PetscFormatRealArray(char buf[],size_t len,const char *fmt,PetscInt n,const PetscReal x[])
8241b5687a1SBarry Smith {
8251b5687a1SBarry Smith   PetscErrorCode ierr;
8261b5687a1SBarry Smith   PetscInt       i;
8271b5687a1SBarry Smith   size_t         left,count;
8281b5687a1SBarry Smith   char           *p;
8291b5687a1SBarry Smith 
8301b5687a1SBarry Smith   PetscFunctionBegin;
8311b5687a1SBarry Smith   for (i=0,p=buf,left=len; i<n; i++) {
8321b5687a1SBarry Smith     ierr = PetscSNPrintfCount(p,left,fmt,&count,(double)x[i]);CHKERRQ(ierr);
8331b5687a1SBarry Smith     if (count >= left) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Insufficient space in buffer");
8341b5687a1SBarry Smith     left -= count;
8351b5687a1SBarry Smith     p    += count-1;
8361b5687a1SBarry Smith     *p++  = ' ';
8371b5687a1SBarry Smith   }
8381b5687a1SBarry Smith   p[i ? 0 : -1] = 0;
8391b5687a1SBarry Smith   PetscFunctionReturn(0);
8401b5687a1SBarry Smith }
841