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 PetscStopErrorHandler(), 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 PetscStopErrorHandler() 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]; 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 void PETSC_DLLEXPORT PetscExceptionPop(PetscErrorCode err) 401 { 402 /* if (PetscExceptionsCount <= 0)SETERRQ(PETSC_ERR_PLIB,"Stack for PetscExceptions is empty"); */ 403 if (PetscErrorIsCatchable(err)) PetscExceptionsCount--; 404 } 405 #endif 406 407 #undef __FUNCT__ 408 #define __FUNCT__ "PetscError" 409 /*@C 410 PetscError - Routine that is called when an error has been detected, 411 usually called through the macro SETERRQ(). 412 413 Not Collective 414 415 Input Parameters: 416 + line - the line number of the error (indicated by __LINE__) 417 . func - the function where the error occured (indicated by __FUNCT__) 418 . dir - the directory of file (indicated by __SDIR__) 419 . file - the file in which the error was detected (indicated by __FILE__) 420 . mess - an error text string, usually just printed to the screen 421 . n - the generic error number 422 . p - 1 indicates the error was initially detected, 0 indicates this is a traceback from a 423 previously detected error 424 - mess - formatted message string - aka printf 425 426 Level: intermediate 427 428 Notes: 429 Most users need not directly use this routine and the error handlers, but 430 can instead use the simplified interface SETERRQ, which has the calling 431 sequence 432 $ SETERRQ(n,mess) 433 434 Experienced users can set the error handler with PetscPushErrorHandler(). 435 436 Concepts: error^setting condition 437 438 .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 439 @*/ 440 PetscErrorCode PETSC_DLLEXPORT PetscError(int line,const char *func,const char* file,const char *dir,PetscErrorCode n,int p,const char *mess,...) 441 { 442 va_list Argp; 443 PetscErrorCode ierr; 444 char buf[2048],*lbuf = 0; 445 PetscTruth ismain,isunknown; 446 #if defined(PETSC_USE_ERRORCHECKING) 447 PetscInt i; 448 #endif 449 450 if (!func) func = "User provided function"; 451 if (!file) file = "User file"; 452 if (!dir) dir = " "; 453 454 PetscFunctionBegin; 455 /* Compose the message evaluating the print format */ 456 if (mess) { 457 va_start(Argp,mess); 458 PetscVSNPrintf(buf,2048,mess,Argp); 459 va_end(Argp); 460 lbuf = buf; 461 if (p == 1) { 462 PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 463 } 464 } 465 466 #if defined(PETSC_USE_ERRORCHECKING) 467 /* check if user is catching this exception */ 468 for (i=0; i<PetscExceptionsCount; i++) { 469 if (n == PetscExceptions[i]) PetscFunctionReturn(n); 470 } 471 #endif 472 473 if (!eh) ierr = PetscTraceBackErrorHandler(line,func,file,dir,n,p,lbuf,0); 474 else ierr = (*eh->handler)(line,func,file,dir,n,p,lbuf,eh->ctx); 475 476 /* 477 If this is called from the main() routine we call MPI_Abort() instead of 478 return to allow the parallel program to be properly shutdown. 479 480 Since this is in the error handler we don't check the errors below. Of course, 481 PetscStrncmp() does its own error checking which is problamatic 482 */ 483 PetscStrncmp(func,"main",4,&ismain); 484 PetscStrncmp(func,"unknown",7,&isunknown); 485 if (ismain || isunknown) { 486 MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 487 } 488 PetscFunctionReturn(ierr); 489 } 490 491 /* -------------------------------------------------------------------------*/ 492 493 #undef __FUNCT__ 494 #define __FUNCT__ "PetscIntView" 495 /*@C 496 PetscIntView - Prints an array of integers; useful for debugging. 497 498 Collective on PetscViewer 499 500 Input Parameters: 501 + N - number of integers in array 502 . idx - array of integers 503 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 504 505 Level: intermediate 506 507 .seealso: PetscRealView() 508 @*/ 509 PetscErrorCode PETSC_DLLEXPORT PetscIntView(PetscInt N,PetscInt idx[],PetscViewer viewer) 510 { 511 PetscErrorCode ierr; 512 PetscInt j,i,n = N/20,p = N % 20; 513 PetscTruth iascii,issocket; 514 MPI_Comm comm; 515 516 PetscFunctionBegin; 517 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 518 PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,3); 519 PetscValidIntPointer(idx,2); 520 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 521 522 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 523 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 524 if (iascii) { 525 for (i=0; i<n; i++) { 526 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 527 for (j=0; j<20; j++) { 528 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 529 } 530 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 531 } 532 if (p) { 533 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 534 for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 535 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 536 } 537 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 538 #if defined(PETSC_USE_SOCKET_VIEWER) 539 } else if (issocket) { 540 PetscMPIInt rank,size,*sizes,Ntotal,*displs, NN = (PetscMPIInt)N; 541 PetscInt *array; 542 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 543 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 544 545 if (size > 1) { 546 if (rank) { 547 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 548 ierr = MPI_Gatherv(idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 549 } else { 550 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 551 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPIU_INT,0,comm);CHKERRQ(ierr); 552 Ntotal = sizes[0]; 553 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 554 displs[0] = 0; 555 for (i=1; i<size; i++) { 556 Ntotal += sizes[i]; 557 displs[i] = displs[i-1] + sizes[i-1]; 558 } 559 ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&array);CHKERRQ(ierr); 560 ierr = MPI_Gatherv(idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 561 ierr = PetscViewerSocketPutInt(viewer,Ntotal,array);CHKERRQ(ierr); 562 ierr = PetscFree(sizes);CHKERRQ(ierr); 563 ierr = PetscFree(displs);CHKERRQ(ierr); 564 ierr = PetscFree(array);CHKERRQ(ierr); 565 } 566 } else { 567 ierr = PetscViewerSocketPutInt(viewer,N,idx);CHKERRQ(ierr); 568 } 569 #endif 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,issocket; 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_SOCKET,&issocket);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 #if defined(PETSC_USE_SOCKET_VIEWER) 624 } else if (issocket) { 625 PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN = (PetscMPIInt)N; 626 PetscReal *array; 627 628 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 629 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 630 631 if (size > 1) { 632 if (rank) { 633 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 634 ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,0,0,0,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 635 } else { 636 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 637 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 638 Ntotal = sizes[0]; 639 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 640 displs[0] = 0; 641 for (i=1; i<size; i++) { 642 Ntotal += sizes[i]; 643 displs[i] = displs[i-1] + sizes[i-1]; 644 } 645 ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 646 ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,array,sizes,displs,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 647 ierr = PetscViewerSocketPutReal(viewer,Ntotal,1,array);CHKERRQ(ierr); 648 ierr = PetscFree(sizes);CHKERRQ(ierr); 649 ierr = PetscFree(displs);CHKERRQ(ierr); 650 ierr = PetscFree(array);CHKERRQ(ierr); 651 } 652 } else { 653 ierr = PetscViewerSocketPutReal(viewer,N,1,idx);CHKERRQ(ierr); 654 } 655 #endif 656 } else { 657 const char *tname; 658 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 659 SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 660 } 661 PetscFunctionReturn(0); 662 } 663 664 #undef __FUNCT__ 665 #define __FUNCT__ "PetscScalarView" 666 /*@C 667 PetscScalarView - Prints an array of scalars; useful for debugging. 668 669 Collective on PetscViewer 670 671 Input Parameters: 672 + N - number of scalars in array 673 . idx - array of scalars 674 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 675 676 Level: intermediate 677 678 .seealso: PetscIntView(), PetscRealView() 679 @*/ 680 PetscErrorCode PETSC_DLLEXPORT PetscScalarView(PetscInt N,PetscScalar idx[],PetscViewer viewer) 681 { 682 PetscErrorCode ierr; 683 PetscInt j,i,n = N/3,p = N % 3; 684 PetscTruth iascii,issocket; 685 MPI_Comm comm; 686 687 PetscFunctionBegin; 688 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 689 PetscValidHeader(viewer,3); 690 PetscValidScalarPointer(idx,2); 691 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 692 693 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 694 ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 695 if (iascii) { 696 for (i=0; i<n; i++) { 697 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 698 for (j=0; j<3; j++) { 699 #if defined (PETSC_USE_COMPLEX) 700 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 701 PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 702 #else 703 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*3+j]);CHKERRQ(ierr); 704 #endif 705 } 706 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 707 } 708 if (p) { 709 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 710 for (i=0; i<p; i++) { 711 #if defined (PETSC_USE_COMPLEX) 712 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 713 PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 714 #else 715 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 716 #endif 717 } 718 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 719 } 720 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 721 #if defined(PETSC_USE_SOCKET_VIEWER) 722 } else if (issocket) { 723 PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN = (PetscMPIInt)N; 724 PetscScalar *array; 725 726 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 727 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 728 729 if (size > 1) { 730 if (rank) { 731 ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 732 ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 733 } else { 734 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 735 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 736 Ntotal = sizes[0]; 737 ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 738 displs[0] = 0; 739 for (i=1; i<size; i++) { 740 Ntotal += sizes[i]; 741 displs[i] = displs[i-1] + sizes[i-1]; 742 } 743 ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&array);CHKERRQ(ierr); 744 ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 745 ierr = PetscViewerSocketPutScalar(viewer,Ntotal,1,array);CHKERRQ(ierr); 746 ierr = PetscFree(sizes);CHKERRQ(ierr); 747 ierr = PetscFree(displs);CHKERRQ(ierr); 748 ierr = PetscFree(array);CHKERRQ(ierr); 749 } 750 } else { 751 ierr = PetscViewerSocketPutScalar(viewer,N,1,idx);CHKERRQ(ierr); 752 } 753 #endif 754 } else { 755 const char *tname; 756 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 757 SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 758 } 759 PetscFunctionReturn(0); 760 } 761 762 763 764 765