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 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"); 124121a09c4SJose Roman case 'F': 125145505e7SDominic Meiser SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"%%F format is no longer supported, use %%f and cast the argument to double"); 1267bc47156SJose Roman default: 1277bc47156SJose Roman newformat[j++] = format[i]; 1287bc47156SJose Roman break; 1297bc47156SJose Roman } 1307bc47156SJose Roman i++; 131a297a907SKarl Rupp } else newformat[j++] = format[i++]; 132e5c89e4eSSatish Balay } 133e5c89e4eSSatish Balay newformat[j] = 0; 134eed5747fSBarry Smith PetscFunctionReturn(0); 135e5c89e4eSSatish Balay } 136e5c89e4eSSatish Balay 13714416c0eSBarry Smith #define PETSCDEFAULTBUFFERSIZE 8*1024 138d781fa04SBarry Smith 139c9a19010SBarry Smith /*@C 140c9a19010SBarry Smith PetscVSNPrintf - The PETSc version of vsnprintf(). Converts a PETSc format string into a standard C format string and then puts all the 141c9a19010SBarry Smith function arguments into a string using the format statement. 142c9a19010SBarry Smith 143c9a19010SBarry Smith Input Parameters: 144c9a19010SBarry Smith + str - location to put result 145c9a19010SBarry Smith . len - the amount of space in str 146c9a19010SBarry Smith + format - the PETSc format string 147c9a19010SBarry Smith - fullLength - the amount of space in str actually used. 148c9a19010SBarry Smith 14995452b02SPatrick Sanan Developer Notes: 15095452b02SPatrick Sanan this function may be called from an error handler, if an error occurs when it is called by the error handler than likely 151eed5747fSBarry Smith a recursion will occur and possible crash. 152c9a19010SBarry Smith 153c9a19010SBarry Smith Level: developer 154c9a19010SBarry Smith 155d781fa04SBarry Smith .seealso: PetscVSNPrintf(), PetscErrorPrintf(), PetscVPrintf() 156d781fa04SBarry Smith 157c9a19010SBarry Smith @*/ 1587087cfbeSBarry Smith PetscErrorCode PetscVSNPrintf(char *str,size_t len,const char *format,size_t *fullLength,va_list Argp) 159e5c89e4eSSatish Balay { 160d781fa04SBarry Smith char *newformat = NULL; 16114416c0eSBarry Smith char formatbuf[PETSCDEFAULTBUFFERSIZE]; 162d781fa04SBarry Smith size_t newLength; 1631a15ef5dSMatthew Knepley PetscErrorCode ierr; 16414416c0eSBarry Smith int flen; 165e5c89e4eSSatish Balay 166eed5747fSBarry Smith PetscFunctionBegin; 167d781fa04SBarry Smith ierr = PetscFormatConvertGetSize(format,&newLength);CHKERRQ(ierr); 16814416c0eSBarry Smith if (newLength < PETSCDEFAULTBUFFERSIZE) { 169e2135aedSMatthew Knepley newformat = formatbuf; 17014416c0eSBarry Smith newLength = PETSCDEFAULTBUFFERSIZE-1; 171e2135aedSMatthew Knepley } else { 172d781fa04SBarry Smith ierr = PetscMalloc1(newLength, &newformat);CHKERRQ(ierr); 173e2135aedSMatthew Knepley } 174d781fa04SBarry Smith ierr = PetscFormatConvert(format,newformat);CHKERRQ(ierr); 1757b9a2d1bSSatish Balay #if defined(PETSC_HAVE_VSNPRINTF) 176152b30f0SSatish Balay flen = vsnprintf(str,len,newformat,Argp); 177e5c89e4eSSatish Balay #else 17889b07760SSatish Balay #error "vsnprintf not found" 179e5c89e4eSSatish Balay #endif 18014416c0eSBarry Smith if (newLength > PETSCDEFAULTBUFFERSIZE-1) { 181e2135aedSMatthew Knepley ierr = PetscFree(newformat);CHKERRQ(ierr); 182e2135aedSMatthew Knepley } 1838627564fSBarry Smith { 1848627564fSBarry Smith PetscBool foundedot; 1858627564fSBarry Smith size_t cnt = 0,ncnt = 0,leng; 1868627564fSBarry Smith ierr = PetscStrlen(str,&leng);CHKERRQ(ierr); 18717ca8410SBarry Smith if (leng > 4) { 1888627564fSBarry Smith for (cnt=0; cnt<leng-4; cnt++) { 1898627564fSBarry Smith if (str[cnt] == '[' && str[cnt+1] == '|'){ 190c540d043SBarry Smith flen -= 4; 1918627564fSBarry Smith cnt++; cnt++; 1928627564fSBarry Smith foundedot = PETSC_FALSE; 1938627564fSBarry Smith for (; cnt<leng-1; cnt++) { 1948627564fSBarry Smith if (str[cnt] == '|' && str[cnt+1] == ']'){ 1958627564fSBarry Smith cnt++; 1968627564fSBarry Smith if (!foundedot) str[ncnt++] = '.'; 1978627564fSBarry Smith ncnt--; 1988627564fSBarry Smith break; 1998627564fSBarry Smith } else { 2008627564fSBarry Smith if (str[cnt] == 'e' || str[cnt] == '.') foundedot = PETSC_TRUE; 2018627564fSBarry Smith str[ncnt++] = str[cnt]; 2028627564fSBarry Smith } 2038627564fSBarry Smith } 2048627564fSBarry Smith } else { 2058627564fSBarry Smith str[ncnt] = str[cnt]; 2068627564fSBarry Smith } 2078627564fSBarry Smith ncnt++; 2088627564fSBarry Smith } 2098627564fSBarry Smith while (cnt < leng) { 2108627564fSBarry Smith str[ncnt] = str[cnt]; ncnt++; cnt++; 2118627564fSBarry Smith } 2128627564fSBarry Smith str[ncnt] = 0; 2138627564fSBarry Smith } 2148627564fSBarry Smith } 215748e1b9dSBarry Smith #if defined(PETSC_HAVE_WINDOWS_H) && !defined(PETSC_HAVE__SET_OUTPUT_FORMAT) 216e51f71cfSBarry Smith /* older Windows OS always produces e-+0np for floating point output; remove the extra 0 */ 217748e1b9dSBarry Smith { 218748e1b9dSBarry Smith size_t cnt = 0,ncnt = 0,leng; 219748e1b9dSBarry Smith ierr = PetscStrlen(str,&leng);CHKERRQ(ierr); 220748e1b9dSBarry Smith if (leng > 5) { 221748e1b9dSBarry Smith for (cnt=0; cnt<leng-4; cnt++) { 222e51f71cfSBarry 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') { 223748e1b9dSBarry Smith str[ncnt] = str[cnt]; ncnt++; cnt++; 224e51f71cfSBarry Smith str[ncnt] = str[cnt]; ncnt++; cnt++; cnt++; 225e51f71cfSBarry Smith str[ncnt] = str[cnt]; 226748e1b9dSBarry Smith } else { 227748e1b9dSBarry Smith str[ncnt] = str[cnt]; 228748e1b9dSBarry Smith } 229748e1b9dSBarry Smith ncnt++; 230748e1b9dSBarry Smith } 231748e1b9dSBarry Smith while (cnt < leng) { 232748e1b9dSBarry Smith str[ncnt] = str[cnt]; ncnt++; 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 253c9a19010SBarry Smith $ ierr = PetscVFPrintfDefault(fd,format,Argp);CHKERR(ierr); 254c9a19010SBarry Smith $ } else { 255c9a19010SBarry Smith $ char buff[BIG]; 256c9a19010SBarry Smith $ size_t length; 257c9a19010SBarry Smith $ ierr = PetscVSNPrintf(buff,BIG,format,&length,Argp);CHKERRQ(ierr); 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 275c9a19010SBarry Smith .seealso: PetscVSNPrintf(), PetscErrorPrintf() 276c9a19010SBarry Smith 277c9a19010SBarry Smith @*/ 2787087cfbeSBarry Smith PetscErrorCode PetscVFPrintfDefault(FILE *fd,const char *format,va_list Argp) 279e5c89e4eSSatish Balay { 28014416c0eSBarry Smith char str[PETSCDEFAULTBUFFERSIZE]; 28114416c0eSBarry Smith char *buff = str; 28214416c0eSBarry Smith size_t fullLength; 283eed5747fSBarry Smith PetscErrorCode ierr; 2841531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY) 28514416c0eSBarry Smith va_list Argpcopy; 2861531940fSBarry Smith #endif 2871179db26SBarry Smith 288eed5747fSBarry Smith PetscFunctionBegin; 2891531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY) 29014416c0eSBarry Smith va_copy(Argpcopy,Argp); 2911531940fSBarry Smith #endif 29214416c0eSBarry Smith ierr = PetscVSNPrintf(str,sizeof(str),format,&fullLength,Argp);CHKERRQ(ierr); 29314416c0eSBarry Smith if (fullLength > sizeof(str)) { 29414416c0eSBarry Smith ierr = PetscMalloc1(fullLength,&buff);CHKERRQ(ierr); 2951531940fSBarry Smith #if defined(PETSC_HAVE_VA_COPY) 29614416c0eSBarry Smith ierr = PetscVSNPrintf(buff,fullLength,format,NULL,Argpcopy);CHKERRQ(ierr); 2971531940fSBarry Smith #else 2981531940fSBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"C89 does not support va_copy() hence cannot print long strings with PETSc printing routines"); 2991531940fSBarry Smith #endif 30014416c0eSBarry Smith } 30114416c0eSBarry Smith fprintf(fd,"%s",buff);CHKERRQ(ierr); 3025a058713SBarry Smith fflush(fd); 30314416c0eSBarry Smith if (buff != str) { 30414416c0eSBarry Smith ierr = PetscFree(buff);CHKERRQ(ierr); 30514416c0eSBarry Smith } 306eed5747fSBarry Smith PetscFunctionReturn(0); 307e5c89e4eSSatish Balay } 308e5c89e4eSSatish Balay 3095b5bc046SBarry Smith /*@C 3105b5bc046SBarry Smith PetscSNPrintf - Prints to a string of given length 3115b5bc046SBarry Smith 3125b5bc046SBarry Smith Not Collective 3135b5bc046SBarry Smith 3145b5bc046SBarry Smith Input Parameters: 3155b5bc046SBarry Smith + str - the string to print to 3165b5bc046SBarry Smith . len - the length of str 3175b5bc046SBarry Smith . format - the usual printf() format string 3185b5bc046SBarry Smith - any arguments 3195b5bc046SBarry Smith 3205b5bc046SBarry Smith Level: intermediate 3215b5bc046SBarry Smith 3225b5bc046SBarry Smith .seealso: PetscSynchronizedFlush(), PetscSynchronizedFPrintf(), PetscFPrintf(), PetscVSNPrintf(), 323d781fa04SBarry Smith PetscPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf(), PetscVFPrintf() 3245b5bc046SBarry Smith @*/ 3257087cfbeSBarry Smith PetscErrorCode PetscSNPrintf(char *str,size_t len,const char format[],...) 3265b5bc046SBarry Smith { 3275b5bc046SBarry Smith PetscErrorCode ierr; 328c9a19010SBarry Smith size_t fullLength; 3295b5bc046SBarry Smith va_list Argp; 3305b5bc046SBarry Smith 3315b5bc046SBarry Smith PetscFunctionBegin; 3325b5bc046SBarry Smith va_start(Argp,format); 3332d609e63SMatthew Knepley ierr = PetscVSNPrintf(str,len,format,&fullLength,Argp);CHKERRQ(ierr); 3345b5bc046SBarry Smith PetscFunctionReturn(0); 3355b5bc046SBarry Smith } 3365b5bc046SBarry Smith 337257d2499SJed Brown /*@C 338257d2499SJed Brown PetscSNPrintfCount - Prints to a string of given length, returns count 339257d2499SJed Brown 340257d2499SJed Brown Not Collective 341257d2499SJed Brown 342257d2499SJed Brown Input Parameters: 343257d2499SJed Brown + str - the string to print to 344257d2499SJed Brown . len - the length of str 345257d2499SJed Brown . format - the usual printf() format string 346257d2499SJed Brown - any arguments 347257d2499SJed Brown 348cb398dd3SBarry Smith Output Parameter: 349cb398dd3SBarry Smith . countused - number of characters used 350cb398dd3SBarry Smith 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 36902c9f0b5SLisandro Dalcin PrintfQueue petsc_printfqueue = NULL,petsc_printfqueuebase = NULL; 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"); 402*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(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; 42302c9f0b5SLisandro Dalcin petsc_printfqueue->next = NULL; 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); 433580bdb30SBarry Smith ierr = PetscArrayzero(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"); 470*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(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; 49102c9f0b5SLisandro Dalcin petsc_printfqueue->next = NULL; 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); 501580bdb30SBarry Smith ierr = PetscArrayzero(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 513d083f849SBarry Smith Collective 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); 539*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 540*ffc4695bSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRMPI(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 */ 547*ffc4695bSBarry Smith ierr = MPI_Send(&dummy,1,MPI_INT,i,tag,comm);CHKERRMPI(ierr); 548*ffc4695bSBarry Smith ierr = MPI_Recv(&n,1,MPI_INT,i,tag,comm,&status);CHKERRMPI(ierr); 549e5c89e4eSSatish Balay for (j=0; j<n; j++) { 55029a5cbdcSMatthew G. Knepley PetscMPIInt size = 0; 5512d609e63SMatthew Knepley 552*ffc4695bSBarry Smith ierr = MPI_Recv(&size,1,MPI_INT,i,tag,comm,&status);CHKERRMPI(ierr); 553785e854fSJed Brown ierr = PetscMalloc1(size, &message);CHKERRQ(ierr); 554*ffc4695bSBarry Smith ierr = MPI_Recv(message,size,MPI_CHAR,i,tag,comm,&status);CHKERRMPI(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 562*ffc4695bSBarry Smith ierr = MPI_Recv(&dummy,1,MPI_INT,0,tag,comm,&status);CHKERRMPI(ierr); 563*ffc4695bSBarry Smith ierr = MPI_Send(&petsc_printfqueuelength,1,MPI_INT,0,tag,comm);CHKERRMPI(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 } 57202c9f0b5SLisandro Dalcin petsc_printfqueue = NULL; 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 598e5c89e4eSSatish Balay .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), 599e5c89e4eSSatish Balay PetscViewerASCIISynchronizedPrintf(), PetscSynchronizedFlush() 600e5c89e4eSSatish Balay @*/ 6017087cfbeSBarry Smith PetscErrorCode PetscFPrintf(MPI_Comm comm,FILE* fd,const char format[],...) 602e5c89e4eSSatish Balay { 603e5c89e4eSSatish Balay PetscErrorCode ierr; 604e5c89e4eSSatish Balay PetscMPIInt rank; 605e5c89e4eSSatish Balay 606e5c89e4eSSatish Balay PetscFunctionBegin; 6076180deefSBarry Smith if (comm == MPI_COMM_NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Called with MPI_COMM_NULL, likely PetscObjectComm() failed"); 608*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 609e5c89e4eSSatish Balay if (!rank) { 610e5c89e4eSSatish Balay va_list Argp; 611e5c89e4eSSatish Balay va_start(Argp,format); 6121179db26SBarry Smith ierr = (*PetscVFPrintf)(fd,format,Argp);CHKERRQ(ierr); 613cdc7d174SSatish Balay if (petsc_history && (fd !=petsc_history)) { 614cdc7d174SSatish Balay va_start(Argp,format); 6151179db26SBarry Smith ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr); 616e5c89e4eSSatish Balay } 617e5c89e4eSSatish Balay va_end(Argp); 618e5c89e4eSSatish Balay } 619e5c89e4eSSatish Balay PetscFunctionReturn(0); 620e5c89e4eSSatish Balay } 621e5c89e4eSSatish Balay 622e5c89e4eSSatish Balay /*@C 623e5c89e4eSSatish Balay PetscPrintf - Prints to standard out, only from the first 624eed5747fSBarry Smith processor in the communicator. Calls from other processes are ignored. 625e5c89e4eSSatish Balay 626e5c89e4eSSatish Balay Not Collective 627e5c89e4eSSatish Balay 628e5c89e4eSSatish Balay Input Parameters: 629e5c89e4eSSatish Balay + comm - the communicator 630e5c89e4eSSatish Balay - format - the usual printf() format string 631e5c89e4eSSatish Balay 632e5c89e4eSSatish Balay Level: intermediate 633e5c89e4eSSatish Balay 634b2706f25SRichard Tran Mills Notes: 635b2706f25SRichard Tran Mills PetscPrintf() supports some format specifiers that are unique to PETSc. 636b2706f25SRichard Tran Mills See the manual page for PetscFormatConvert() for details. 637b2706f25SRichard Tran Mills 638e5c89e4eSSatish Balay Fortran Note: 639d60d70ccSBarry Smith The call sequence is PetscPrintf(MPI_Comm, character(*), PetscErrorCode ierr) from Fortran. 640e5c89e4eSSatish Balay That is, you can only pass a single character string from Fortran. 641e5c89e4eSSatish Balay 642e5c89e4eSSatish Balay 643b2706f25SRichard Tran Mills .seealso: PetscFPrintf(), PetscSynchronizedPrintf(), PetscFormatConvert() 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"); 652*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(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"); 692*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(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 712d083f849SBarry Smith Collective 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; 734*ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(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 } 744*ffc4695bSBarry Smith ierr = MPI_Bcast(string,len,MPI_BYTE,0,comm);CHKERRMPI(ierr); 745e5c89e4eSSatish Balay PetscFunctionReturn(0); 746e5c89e4eSSatish Balay } 747238ccf28SShri Abhyankar 7482475b7caSBarry Smith #if defined(PETSC_HAVE_CLOSURE) 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 7622475b7caSBarry Smith ierr = PetscVSNPrintf(buff,sizeof(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 7798c74ee41SBarry Smith /*@C 7808c74ee41SBarry Smith PetscFormatStrip - Takes a PETSc format string and removes all numerical modifiers to % operations 7818c74ee41SBarry Smith 7828c74ee41SBarry Smith Input Parameters: 7838c74ee41SBarry Smith . format - the PETSc format string 7848c74ee41SBarry Smith 7858c74ee41SBarry Smith Level: developer 7868c74ee41SBarry Smith 7878c74ee41SBarry Smith @*/ 7888c74ee41SBarry Smith PetscErrorCode PetscFormatStrip(char *format) 7898c74ee41SBarry Smith { 7908c74ee41SBarry Smith size_t loc1 = 0, loc2 = 0; 7918c74ee41SBarry Smith 7928c74ee41SBarry Smith PetscFunctionBegin; 7938c74ee41SBarry Smith while (format[loc2]) { 7948c74ee41SBarry Smith if (format[loc2] == '%') { 7958c74ee41SBarry Smith format[loc1++] = format[loc2++]; 7968c74ee41SBarry Smith while (format[loc2] && ((format[loc2] >= '0' && format[loc2] <= '9') || format[loc2] == '.')) loc2++; 7978c74ee41SBarry Smith } 7988c74ee41SBarry Smith format[loc1++] = format[loc2++]; 7998c74ee41SBarry Smith } 8008c74ee41SBarry Smith PetscFunctionReturn(0); 8018c74ee41SBarry Smith } 8028c74ee41SBarry Smith 8031b5687a1SBarry Smith PetscErrorCode PetscFormatRealArray(char buf[],size_t len,const char *fmt,PetscInt n,const PetscReal x[]) 8041b5687a1SBarry Smith { 8051b5687a1SBarry Smith PetscErrorCode ierr; 8061b5687a1SBarry Smith PetscInt i; 8071b5687a1SBarry Smith size_t left,count; 8081b5687a1SBarry Smith char *p; 8091b5687a1SBarry Smith 8101b5687a1SBarry Smith PetscFunctionBegin; 8111b5687a1SBarry Smith for (i=0,p=buf,left=len; i<n; i++) { 8121b5687a1SBarry Smith ierr = PetscSNPrintfCount(p,left,fmt,&count,(double)x[i]);CHKERRQ(ierr); 8131b5687a1SBarry Smith if (count >= left) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Insufficient space in buffer"); 8141b5687a1SBarry Smith left -= count; 8151b5687a1SBarry Smith p += count-1; 8161b5687a1SBarry Smith *p++ = ' '; 8171b5687a1SBarry Smith } 8181b5687a1SBarry Smith p[i ? 0 : -1] = 0; 8191b5687a1SBarry Smith PetscFunctionReturn(0); 8201b5687a1SBarry Smith } 821