1 #define PETSC_DLL 2 /* 3 Code that allows one to set the error handlers 4 */ 5 #include "petsc.h" /*I "petsc.h" I*/ 6 #include "petscsys.h" 7 #include <stdarg.h> 8 #if defined(PETSC_HAVE_STDLIB_H) 9 #include <stdlib.h> 10 #endif 11 12 typedef struct _EH *EH; 13 struct _EH { 14 int cookie; 15 PetscErrorCode (*handler)(int,const char*,const char*,const char *,PetscErrorCode,int,const char*,void *); 16 void *ctx; 17 EH previous; 18 }; 19 20 static EH eh = 0; 21 22 #undef __FUNCT__ 23 #define __FUNCT__ "PetscEmacsClientErrorHandler" 24 /*@C 25 PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 26 load the file where the error occured. Then calls the "previous" error handler. 27 28 Not Collective 29 30 Input Parameters: 31 + line - the line number of the error (indicated by __LINE__) 32 . func - the function where error is detected (indicated by __FUNCT__) 33 . file - the file in which the error was detected (indicated by __FILE__) 34 . dir - the directory of the file (indicated by __SDIR__) 35 . mess - an error text string, usually just printed to the screen 36 . n - the generic error number 37 . p - specific error number 38 - ctx - error handler context 39 40 Options Database Key: 41 . -on_error_emacs <machinename> 42 43 Level: developer 44 45 Notes: 46 You must put (server-start) in your .emacs file for the emacsclient software to work 47 48 Most users need not directly employ this routine and the other error 49 handlers, but can instead use the simplified interface SETERRQ, which has 50 the calling sequence 51 $ SETERRQ(number,p,mess) 52 53 Notes for experienced users: 54 Use PetscPushErrorHandler() to set the desired error handler. 55 56 Concepts: emacs^going to on error 57 Concepts: error handler^going to line in emacs 58 59 .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 60 PetscAbortErrorHandler() 61 @*/ 62 PetscErrorCode PETSC_DLLEXPORT PetscEmacsClientErrorHandler(int line,const char *fun,const char* file,const char *dir,PetscErrorCode n,int p,const char *mess,void *ctx) 63 { 64 PetscErrorCode ierr; 65 char command[PETSC_MAX_PATH_LEN]; 66 const char *pdir; 67 FILE *fp; 68 69 PetscFunctionBegin; 70 /* Note: don't check error codes since this an error handler :-) */ 71 ierr = PetscGetPetscDir(&pdir);CHKERRQ(ierr); 72 sprintf(command,"emacsclient +%d %s/%s%s\n",line,pdir,dir,file); 73 #if defined(PETSC_HAVE_POPEN) 74 ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp); 75 ierr = PetscPClose(MPI_COMM_WORLD,fp); 76 #else 77 SETERRQ(PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 78 #endif 79 ierr = PetscPopErrorHandler(); /* remove this handler from the stack of handlers */ 80 if (!eh) ierr = PetscTraceBackErrorHandler(line,fun,file,dir,n,p,mess,0); 81 else ierr = (*eh->handler)(line,fun,file,dir,n,p,mess,eh->ctx); 82 PetscFunctionReturn(ierr); 83 } 84 85 #undef __FUNCT__ 86 #define __FUNCT__ "PetscPushErrorHandler" 87 /*@C 88 PetscPushErrorHandler - Sets a routine to be called on detection of errors. 89 90 Not Collective 91 92 Input Parameters: 93 + handler - error handler routine 94 - ctx - optional handler context that contains information needed by the handler (for 95 example file pointers for error messages etc.) 96 97 Calling sequence of handler: 98 $ int handler(int line,char *func,char *file,char *dir,PetscErrorCode n,int p,char *mess,void *ctx); 99 100 + func - the function where the error occured (indicated by __FUNCT__) 101 . line - the line number of the error (indicated by __LINE__) 102 . file - the file in which the error was detected (indicated by __FILE__) 103 . dir - the directory of the file (indicated by __SDIR__) 104 . n - the generic error number (see list defined in include/petscerror.h) 105 . p - the specific error number 106 . mess - an error text string, usually just printed to the screen 107 - ctx - the error handler context 108 109 Options Database Keys: 110 + -on_error_attach_debugger <noxterm,gdb or dbx> 111 - -on_error_abort 112 113 Level: intermediate 114 115 Notes: 116 The 117 currently available PETSc error handlers include PetscTraceBackErrorHandler(), 118 PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 119 120 .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 121 122 @*/ 123 PetscErrorCode PETSC_DLLEXPORT PetscPushErrorHandler(PetscErrorCode (*handler)(int,const char *,const char*,const char*,PetscErrorCode,int,const char*,void*),void *ctx) 124 { 125 EH neweh; 126 PetscErrorCode ierr; 127 128 PetscFunctionBegin; 129 ierr = PetscNew(struct _EH,&neweh);CHKERRQ(ierr); 130 if (eh) {neweh->previous = eh;} 131 else {neweh->previous = 0;} 132 neweh->handler = handler; 133 neweh->ctx = ctx; 134 eh = neweh; 135 PetscFunctionReturn(0); 136 } 137 138 #undef __FUNCT__ 139 #define __FUNCT__ "PetscPopErrorHandler" 140 /*@ 141 PetscPopErrorHandler - Removes the latest error handler that was 142 pushed with PetscPushErrorHandler(). 143 144 Not Collective 145 146 Level: intermediate 147 148 Concepts: error handler^setting 149 150 .seealso: PetscPushErrorHandler() 151 @*/ 152 PetscErrorCode PETSC_DLLEXPORT PetscPopErrorHandler(void) 153 { 154 EH tmp; 155 PetscErrorCode ierr; 156 157 PetscFunctionBegin; 158 if (!eh) PetscFunctionReturn(0); 159 tmp = eh; 160 eh = eh->previous; 161 ierr = PetscFree(tmp);CHKERRQ(ierr); 162 163 PetscFunctionReturn(0); 164 } 165 166 #undef __FUNCT__ 167 #define __FUNCT__ "PetscReturnErrorHandler" 168 /*@C 169 PetscReturnErrorHandler - Error handler that causes a return to the current 170 level. 171 172 Not Collective 173 174 Input Parameters: 175 + line - the line number of the error (indicated by __LINE__) 176 . func - the function where error is detected (indicated by __FUNCT__) 177 . file - the file in which the error was detected (indicated by __FILE__) 178 . dir - the directory of the file (indicated by __SDIR__) 179 . mess - an error text string, usually just printed to the screen 180 . n - the generic error number 181 . p - specific error number 182 - ctx - error handler context 183 184 Level: developer 185 186 Notes: 187 Most users need not directly employ this routine and the other error 188 handlers, but can instead use the simplified interface SETERRQ, which has 189 the calling sequence 190 $ SETERRQ(number,p,mess) 191 192 Notes for experienced users: 193 This routine is good for catching errors such as zero pivots in preconditioners 194 or breakdown of iterative methods. It is not appropriate for memory violations 195 and similar errors. 196 197 Use PetscPushErrorHandler() to set the desired error handler. The 198 currently available PETSc error handlers include PetscTraceBackErrorHandler(), 199 PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 200 201 Concepts: error handler 202 203 .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 204 @*/ 205 206 PetscErrorCode PETSC_DLLEXPORT PetscReturnErrorHandler(int line,const char *fun,const char* file,const char *dir,PetscErrorCode n,int p,const char *mess,void *ctx) 207 { 208 PetscFunctionBegin; 209 PetscFunctionReturn(n); 210 } 211 212 static char PetscErrorBaseMessage[1024]; 213 /* 214 The numerical values for these are defined in include/petscerror.h; any changes 215 there must also be made here 216 */ 217 static const char *PetscErrorStrings[] = { 218 /*55 */ "Out of memory", 219 "No support for this operation for this object type", 220 "No support for this operation on this system", 221 /*58 */ "Operation done in wrong order", 222 /*59 */ "Signal received", 223 /*60 */ "Nonconforming object sizes", 224 "Argument aliasing not permitted", 225 "Invalid argument", 226 /*63 */ "Argument out of range", 227 "Corrupt argument: see http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#Corrupt", 228 "Unable to open file", 229 "Read from file failed", 230 "Write to file failed", 231 "Invalid pointer", 232 /*69 */ "Arguments must have same type", 233 "", 234 /*71 */ "Detected zero pivot in LU factorization\nsee http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#ZeroPivot", 235 /*72 */ "Floating point exception", 236 /*73 */ "Object is in wrong state", 237 "Corrupted Petsc object", 238 "Arguments are incompatible", 239 "Error in external library", 240 /*77 */ "Petsc has generated inconsistent data", 241 "Memory corruption", 242 "Unexpected data in file", 243 /*80 */ "Arguments must have same communicators", 244 /*81 */ "Detected zero pivot in Cholesky factorization\nsee http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#ZeroPivot", 245 " ", 246 " ", 247 " ", 248 /*85 */ "Null argument, when expecting valid pointer", 249 /*86 */ "Unknown type. Check for miss-spelling or missing external package needed for type"}; 250 251 #undef __FUNCT__ 252 #define __FUNCT__ "PetscErrorMessage" 253 /*@C 254 PetscErrorMessage - returns the text string associated with a PETSc error code. 255 256 Not Collective 257 258 Input Parameter: 259 . errnum - the error code 260 261 Output Parameter: 262 + text - the error message (PETSC_NULL if not desired) 263 - specific - the specific error message that was set with SETERRxxx() or PetscError(). (PETSC_NULL if not desired) 264 265 Level: developer 266 267 Concepts: error handler^messages 268 269 .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 270 PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 271 @*/ 272 PetscErrorCode PETSC_DLLEXPORT PetscErrorMessage(int errnum,const char *text[],char **specific) 273 { 274 PetscFunctionBegin; 275 if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) { 276 *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 277 } else if (text) *text = 0; 278 279 if (specific) { 280 *specific = PetscErrorBaseMessage; 281 } 282 PetscFunctionReturn(0); 283 } 284 285 #if defined(PETSC_USE_ERRORCHECKING) 286 PetscErrorCode PETSC_DLLEXPORT PetscErrorUncatchable[PETSC_EXCEPTIONS_MAX] = {0}; 287 PetscInt PETSC_DLLEXPORT PetscErrorUncatchableCount = 0; 288 PetscErrorCode PETSC_DLLEXPORT PetscExceptions[PETSC_EXCEPTIONS_MAX] = {0}; 289 PetscInt PETSC_DLLEXPORT PetscExceptionsCount = 0; 290 PetscErrorCode PETSC_DLLEXPORT PetscExceptionTmp = 0; 291 292 #undef __FUNCT__ 293 #define __FUNCT__ "PetscErrorIsCatchable" 294 /*@C 295 PetscErrorIsCatchable - Returns if a PetscErrorCode can be caught with a PetscExceptionTry1() or 296 PetscExceptionPush() 297 298 Input Parameters: 299 . err - error code 300 301 Level: advanced 302 303 Notes: 304 PETSc must not be configured using the option --with-errorchecking=0 for this to work 305 306 .seealso: PetscExceptionTry1(), PetscExceptionCaught(), PetscExceptionPush(), PetscExceptionPop(), PetscErrorSetCatchable() 307 @*/ 308 PetscTruth PETSC_DLLEXPORT PetscErrorIsCatchable(PetscErrorCode err) 309 { 310 PetscInt i; 311 for (i=0; i<PetscErrorUncatchableCount; i++) { 312 if (err == PetscErrorUncatchable[i]) return PETSC_FALSE; 313 } 314 return PETSC_TRUE; 315 } 316 317 #undef __FUNCT__ 318 #define __FUNCT__ "PetscErrorSetCatchable" 319 /*@ 320 PetscErrorSetCatchable - Sets if a PetscErrorCode can be caught with a PetscExceptionTry1() 321 PetscExceptionCaught() pair, or PetscExceptionPush(). By default all errors are catchable. 322 323 Input Parameters: 324 + err - error code 325 - flg - PETSC_TRUE means allow to be caught, PETSC_FALSE means do not allow to be caught 326 327 Level: advanced 328 329 Notes: 330 PETSc must not be configured using the option --with-errorchecking=0 for this to work 331 332 .seealso: PetscExceptionTry1(), PetscExceptionCaught(), PetscExceptionPush(), PetscExceptionPop(), PetscErrorIsCatchable() 333 @*/ 334 PetscErrorCode PETSC_DLLEXPORT PetscErrorSetCatchable(PetscErrorCode err,PetscTruth flg) 335 { 336 PetscFunctionBegin; 337 if (!flg && PetscErrorIsCatchable(err)) { 338 /* add to list of uncatchable */ 339 if (PetscErrorUncatchableCount >= PETSC_EXCEPTIONS_MAX) SETERRQ(PETSC_ERR_PLIB,"Stack for PetscErrorUncatchable is overflowed, recompile \nsrc/sysd/error/err.c with a larger value for PETSC_EXCEPTIONS_MAX"); 340 PetscErrorUncatchable[PetscErrorUncatchableCount++] = err; 341 } else if (flg && !PetscErrorIsCatchable(err)) { 342 /* remove from list of uncatchable */ 343 PetscInt i; 344 for (i=0; i<PetscErrorUncatchableCount; i++) { 345 if (PetscErrorUncatchable[i] == err) break; 346 } 347 for (;i<PetscErrorUncatchableCount; i++) { 348 PetscErrorUncatchable[i] = PetscErrorUncatchable[i+1]; 349 } 350 PetscErrorUncatchableCount--; 351 } 352 PetscFunctionReturn(0); 353 } 354 355 #undef __FUNCT__ 356 #define __FUNCT__ "PetscExceptionPush" 357 /*@ 358 PetscExceptionPush - Adds the exception as one to be caught and passed up. If passed up 359 can be checked with PetscExceptionCaught() or PetscExceptionValue() 360 361 Input Parameters: 362 . err - the exception to catch 363 364 Level: advanced 365 366 Notes: 367 PETSc must not be configured using the option --with-errorchecking=0 for this to work 368 369 Use PetscExceptionPop() to remove this as a value to be caught 370 371 This is not usually needed in C/C++ rather use PetscExceptionTry1() 372 373 .seealso: PetscExceptionTry1(), PetscExceptionCaught(), PetscExceptionPush(), PetscExceptionPop() 374 @*/ 375 PetscErrorCode PETSC_DLLEXPORT PetscExceptionPush(PetscErrorCode err) 376 { 377 PetscFunctionBegin; 378 if (PetscExceptionsCount >= PETSC_EXCEPTIONS_MAX) SETERRQ(PETSC_ERR_PLIB,"Stack for PetscExceptions is overflowed, recompile \nsrc/sysd/error/err.c with a larger value for PETSC_EXCEPTIONS_MAX"); 379 if (PetscErrorIsCatchable(err)) PetscExceptions[PetscExceptionsCount++] = err; 380 PetscFunctionReturn(0); 381 } 382 383 #undef __FUNCT__ 384 #define __FUNCT__ "PetscExceptionPop" 385 /*@ 386 PetscExceptionPop - Removes the most recent exception asked to be caught with PetscExceptionPush() 387 388 Input Parameters: 389 . err - the exception that was pushed 390 391 Level: advanced 392 393 Notes: 394 PETSc must not be configured using the option --with-errorchecking=0 for this to work 395 396 This is not usually needed in C/C++ rather use PetscExceptionTry1() 397 398 .seealso: PetscExceptionTry1(), PetscExceptionCaught(), PetscExceptionPush(), PetscExceptionPop() 399 @*/ 400 PetscErrorCode PETSC_DLLEXPORT PetscExceptionPop(PetscErrorCode err) 401 { 402 PetscFunctionBegin; 403 if (PetscExceptionsCount <= 0)SETERRQ(PETSC_ERR_PLIB,"Stack for PetscExceptions is empty"); 404 if (PetscErrorIsCatchable(err)) PetscExceptionsCount--; 405 PetscFunctionReturn(0); 406 } 407 #endif 408 409 #undef __FUNCT__ 410 #define __FUNCT__ "PetscError" 411 /*@C 412 PetscError - Routine that is called when an error has been detected, 413 usually called through the macro SETERRQ(). 414 415 Not Collective 416 417 Input Parameters: 418 + line - the line number of the error (indicated by __LINE__) 419 . func - the function where the error occured (indicated by __FUNCT__) 420 . dir - the directory of file (indicated by __SDIR__) 421 . file - the file in which the error was detected (indicated by __FILE__) 422 . mess - an error text string, usually just printed to the screen 423 . n - the generic error number 424 . p - 1 indicates the error was initially detected, 0 indicates this is a traceback from a 425 previously detected error 426 - mess - formatted message string - aka printf 427 428 Level: intermediate 429 430 Notes: 431 Most users need not directly use this routine and the error handlers, but 432 can instead use the simplified interface SETERRQ, which has the calling 433 sequence 434 $ SETERRQ(n,mess) 435 436 Experienced users can set the error handler with PetscPushErrorHandler(). 437 438 Concepts: error^setting condition 439 440 .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 441 @*/ 442 PetscErrorCode PETSC_DLLEXPORT PetscError(int line,const char *func,const char* file,const char *dir,PetscErrorCode n,int p,const char *mess,...) 443 { 444 va_list Argp; 445 PetscErrorCode ierr; 446 char buf[2048],*lbuf = 0; 447 PetscTruth ismain,isunknown; 448 #if defined(PETSC_USE_ERRORCHECKING) 449 PetscInt i; 450 #endif 451 452 if (!func) func = "User provided function"; 453 if (!file) file = "User file"; 454 if (!dir) dir = " "; 455 456 PetscFunctionBegin; 457 /* Compose the message evaluating the print format */ 458 if (mess) { 459 va_start(Argp,mess); 460 PetscVSNPrintf(buf,2048,mess,Argp); 461 va_end(Argp); 462 lbuf = buf; 463 if (p == 1) { 464 PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 465 } 466 } 467 468 #if defined(PETSC_USE_ERRORCHECKING) 469 /* check if user is catching this exception */ 470 for (i=0; i<PetscExceptionsCount; i++) { 471 if (n == PetscExceptions[i]) PetscFunctionReturn(n); 472 } 473 #endif 474 475 if (!eh) ierr = PetscTraceBackErrorHandler(line,func,file,dir,n,p,lbuf,0); 476 else ierr = (*eh->handler)(line,func,file,dir,n,p,lbuf,eh->ctx); 477 478 /* 479 If this is called from the main() routine we call MPI_Abort() instead of 480 return to allow the parallel program to be properly shutdown. 481 482 Since this is in the error handler we don't check the errors below. Of course, 483 PetscStrncmp() does its own error checking which is problamatic 484 */ 485 PetscStrncmp(func,"main",4,&ismain); 486 PetscStrncmp(func,"unknown",7,&isunknown); 487 if (ismain || isunknown) { 488 MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 489 } 490 PetscFunctionReturn(ierr); 491 } 492 493 /* -------------------------------------------------------------------------*/ 494 495 #undef __FUNCT__ 496 #define __FUNCT__ "PetscIntView" 497 /*@C 498 PetscIntView - Prints an array of integers; useful for debugging. 499 500 Collective on PetscViewer 501 502 Input Parameters: 503 + N - number of integers in array 504 . idx - array of integers 505 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 506 507 Level: intermediate 508 509 .seealso: PetscRealView() 510 @*/ 511 PetscErrorCode PETSC_DLLEXPORT PetscIntView(PetscInt N,PetscInt idx[],PetscViewer viewer) 512 { 513 PetscErrorCode ierr; 514 PetscInt j,i,n = N/20,p = N % 20; 515 PetscTruth iascii,isbinary; 516 MPI_Comm comm; 517 518 PetscFunctionBegin; 519 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 520 PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,3); 521 PetscValidIntPointer(idx,2); 522 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 523 524 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 525 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 526 if (iascii) { 527 for (i=0; i<n; i++) { 528 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 529 for (j=0; j<20; j++) { 530 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 531 } 532 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 533 } 534 if (p) { 535 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 536 for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 537 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 538 } 539 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 540 } else if (isbinary) { 541 PetscMPIInt rank,size,*sizes,Ntotal,*displs, NN = (PetscMPIInt)N; 542 PetscInt *array; 543 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 544 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 545 546 if (size > 1) { 547 if (rank) { 548 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 549 ierr = MPI_Gatherv(idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 550 } else { 551 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 552 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPIU_INT,0,comm);CHKERRQ(ierr); 553 Ntotal = sizes[0]; 554 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 555 displs[0] = 0; 556 for (i=1; i<size; i++) { 557 Ntotal += sizes[i]; 558 displs[i] = displs[i-1] + sizes[i-1]; 559 } 560 ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&array);CHKERRQ(ierr); 561 ierr = MPI_Gatherv(idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 562 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 563 ierr = PetscFree(sizes);CHKERRQ(ierr); 564 ierr = PetscFree(displs);CHKERRQ(ierr); 565 ierr = PetscFree(array);CHKERRQ(ierr); 566 } 567 } else { 568 ierr = PetscViewerBinaryWrite(viewer,idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 569 } 570 } else { 571 const char *tname; 572 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 573 SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 574 } 575 PetscFunctionReturn(0); 576 } 577 578 #undef __FUNCT__ 579 #define __FUNCT__ "PetscRealView" 580 /*@C 581 PetscRealView - Prints an array of doubles; useful for debugging. 582 583 Collective on PetscViewer 584 585 Input Parameters: 586 + N - number of doubles in array 587 . idx - array of doubles 588 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 589 590 Level: intermediate 591 592 .seealso: PetscIntView() 593 @*/ 594 PetscErrorCode PETSC_DLLEXPORT PetscRealView(PetscInt N,PetscReal idx[],PetscViewer viewer) 595 { 596 PetscErrorCode ierr; 597 PetscInt j,i,n = N/5,p = N % 5; 598 PetscTruth iascii,isbinary; 599 MPI_Comm comm; 600 601 PetscFunctionBegin; 602 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 603 PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,3); 604 PetscValidScalarPointer(idx,2); 605 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 606 607 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 608 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 609 if (iascii) { 610 for (i=0; i<n; i++) { 611 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*i);CHKERRQ(ierr); 612 for (j=0; j<5; j++) { 613 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*5+j]);CHKERRQ(ierr); 614 } 615 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 616 } 617 if (p) { 618 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*n);CHKERRQ(ierr); 619 for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[5*n+i]);CHKERRQ(ierr);} 620 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 621 } 622 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 623 } else if (isbinary) { 624 PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN = (PetscMPIInt)N; 625 PetscReal *array; 626 627 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 628 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 629 630 if (size > 1) { 631 if (rank) { 632 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 633 ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,0,0,0,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 634 } else { 635 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 636 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 637 Ntotal = sizes[0]; 638 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 639 displs[0] = 0; 640 for (i=1; i<size; i++) { 641 Ntotal += sizes[i]; 642 displs[i] = displs[i-1] + sizes[i-1]; 643 } 644 ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 645 ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,array,sizes,displs,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 646 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 647 ierr = PetscFree(sizes);CHKERRQ(ierr); 648 ierr = PetscFree(displs);CHKERRQ(ierr); 649 ierr = PetscFree(array);CHKERRQ(ierr); 650 } 651 } else { 652 ierr = PetscViewerBinaryWrite(viewer,idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 653 } 654 } else { 655 const char *tname; 656 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 657 SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 658 } 659 PetscFunctionReturn(0); 660 } 661 662 #undef __FUNCT__ 663 #define __FUNCT__ "PetscScalarView" 664 /*@C 665 PetscScalarView - Prints an array of scalars; useful for debugging. 666 667 Collective on PetscViewer 668 669 Input Parameters: 670 + N - number of scalars in array 671 . idx - array of scalars 672 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 673 674 Level: intermediate 675 676 .seealso: PetscIntView(), PetscRealView() 677 @*/ 678 PetscErrorCode PETSC_DLLEXPORT PetscScalarView(PetscInt N,PetscScalar idx[],PetscViewer viewer) 679 { 680 PetscErrorCode ierr; 681 PetscInt j,i,n = N/3,p = N % 3; 682 PetscTruth iascii,isbinary; 683 MPI_Comm comm; 684 685 PetscFunctionBegin; 686 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 687 PetscValidHeader(viewer,3); 688 PetscValidScalarPointer(idx,2); 689 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 690 691 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 692 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 693 if (iascii) { 694 for (i=0; i<n; i++) { 695 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 696 for (j=0; j<3; j++) { 697 #if defined (PETSC_USE_COMPLEX) 698 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 699 PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 700 #else 701 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*3+j]);CHKERRQ(ierr); 702 #endif 703 } 704 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 705 } 706 if (p) { 707 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 708 for (i=0; i<p; i++) { 709 #if defined (PETSC_USE_COMPLEX) 710 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 711 PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 712 #else 713 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 714 #endif 715 } 716 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 717 } 718 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 719 } else if (isbinary) { 720 PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN = (PetscMPIInt)N; 721 PetscScalar *array; 722 723 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 724 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 725 726 if (size > 1) { 727 if (rank) { 728 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 729 ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 730 } else { 731 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 732 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 733 Ntotal = sizes[0]; 734 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 735 displs[0] = 0; 736 for (i=1; i<size; i++) { 737 Ntotal += sizes[i]; 738 displs[i] = displs[i-1] + sizes[i-1]; 739 } 740 ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&array);CHKERRQ(ierr); 741 ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 742 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 743 ierr = PetscFree(sizes);CHKERRQ(ierr); 744 ierr = PetscFree(displs);CHKERRQ(ierr); 745 ierr = PetscFree(array);CHKERRQ(ierr); 746 } 747 } else { 748 ierr = PetscViewerBinaryWrite(viewer,idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 749 } 750 } else { 751 const char *tname; 752 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 753 SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 754 } 755 PetscFunctionReturn(0); 756 } 757 758 759 760 761