17d0a6c19SBarry Smith 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay Code that allows one to set the error handlers 4e5c89e4eSSatish Balay */ 5af0996ceSBarry Smith #include <petsc/private/petscimpl.h> /*I "petscsys.h" I*/ 6665c2dedSJed Brown #include <petscviewer.h> 7e5c89e4eSSatish Balay 8*7c66cc67SJunchao Zhang /* A table of Petsc source files containing calls to PETSCABORT. We assume this table will 9*7c66cc67SJunchao Zhang stay stable for a while. When things changed, we just need to add new files to the table. 10*7c66cc67SJunchao Zhang */ 11*7c66cc67SJunchao Zhang static const char* PetscAbortSourceFiles[] = { 12*7c66cc67SJunchao Zhang "NotFound", /* 0, not found in petsc, but may be in users' code if they called PETSCABORT. */ 13*7c66cc67SJunchao Zhang "sys/error/adebug.c", 14*7c66cc67SJunchao Zhang "sys/error/err.c", 15*7c66cc67SJunchao Zhang "src/sys/error/errstop.c", 16*7c66cc67SJunchao Zhang "sys/error/fp.c", 17*7c66cc67SJunchao Zhang "sys/error/signal.c", /* 5 */ 18*7c66cc67SJunchao Zhang "sys/ftn-custom/zutils.c", 19*7c66cc67SJunchao Zhang "sys/logging/utils/stagelog.c", 20*7c66cc67SJunchao Zhang "sys/mpiuni/mpitime.c", 21*7c66cc67SJunchao Zhang "sys/objects/init.c", 22*7c66cc67SJunchao Zhang "sys/objects/pinit.c", /* 10 */ 23*7c66cc67SJunchao Zhang "vec/vec/interface/dlregisvec.c", 24*7c66cc67SJunchao Zhang "vec/vec/utils/comb.c" 25*7c66cc67SJunchao Zhang }; 26*7c66cc67SJunchao Zhang 27*7c66cc67SJunchao Zhang /* Find index of the soure file where a PETSCABORT was called. */ 28*7c66cc67SJunchao Zhang PetscErrorCode PetscAbortFindSourceFile_Private(const char* filepath, PetscInt *idx) 29*7c66cc67SJunchao Zhang { 30*7c66cc67SJunchao Zhang PetscErrorCode ierr; 31*7c66cc67SJunchao Zhang PetscInt i,n = sizeof(PetscAbortSourceFiles)/sizeof(PetscAbortSourceFiles[0]); 32*7c66cc67SJunchao Zhang PetscBool match; 33*7c66cc67SJunchao Zhang char subpath[256]; 34*7c66cc67SJunchao Zhang 35*7c66cc67SJunchao Zhang PetscFunctionBegin; 36*7c66cc67SJunchao Zhang PetscValidIntPointer(idx,2); 37*7c66cc67SJunchao Zhang *idx = 0; 38*7c66cc67SJunchao Zhang for (i=1; i<n; i++) { 39*7c66cc67SJunchao Zhang ierr = PetscFixFilename(PetscAbortSourceFiles[i],subpath);CHKERRQ(ierr); 40*7c66cc67SJunchao Zhang ierr = PetscStrendswith(filepath,subpath,&match);CHKERRQ(ierr); 41*7c66cc67SJunchao Zhang if (match) {*idx = i; break;} 42*7c66cc67SJunchao Zhang } 43*7c66cc67SJunchao Zhang PetscFunctionReturn(0); 44*7c66cc67SJunchao Zhang } 45*7c66cc67SJunchao Zhang 46e5c89e4eSSatish Balay typedef struct _EH *EH; 47e5c89e4eSSatish Balay struct _EH { 48efca3c55SSatish Balay PetscErrorCode (*handler)(MPI_Comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*); 49e5c89e4eSSatish Balay void *ctx; 50e5c89e4eSSatish Balay EH previous; 51e5c89e4eSSatish Balay }; 52e5c89e4eSSatish Balay 53e5c89e4eSSatish Balay static EH eh = 0; 54e5c89e4eSSatish Balay 55e5c89e4eSSatish Balay /*@C 56e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 57e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 58e5c89e4eSSatish Balay 59e5c89e4eSSatish Balay Not Collective 60e5c89e4eSSatish Balay 61e5c89e4eSSatish Balay Input Parameters: 62e32f2f54SBarry Smith + comm - communicator over which error occured 63e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 64e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 65e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 66e5c89e4eSSatish Balay . n - the generic error number 67e5c89e4eSSatish Balay . p - specific error number 68e5c89e4eSSatish Balay - ctx - error handler context 69e5c89e4eSSatish Balay 70e5c89e4eSSatish Balay Options Database Key: 71e5c89e4eSSatish Balay . -on_error_emacs <machinename> 72e5c89e4eSSatish Balay 73e5c89e4eSSatish Balay Level: developer 74e5c89e4eSSatish Balay 75e5c89e4eSSatish Balay Notes: 76e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 77e5c89e4eSSatish Balay 78e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 79e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 80e5c89e4eSSatish Balay the calling sequence 81e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 82e5c89e4eSSatish Balay 83e5c89e4eSSatish Balay Notes for experienced users: 84e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 85e5c89e4eSSatish Balay 86cc12936aSPatrick Sanan Developer Note: 87cc12936aSPatrick Sanan Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 883bf036e2SBarry Smith 89e5c89e4eSSatish Balay 90e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 91e5c89e4eSSatish Balay PetscAbortErrorHandler() 92e5c89e4eSSatish Balay @*/ 93efca3c55SSatish Balay PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 94e5c89e4eSSatish Balay { 95e5c89e4eSSatish Balay PetscErrorCode ierr; 96e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 97e5c89e4eSSatish Balay const char *pdir; 98e5c89e4eSSatish Balay FILE *fp; 99e5c89e4eSSatish Balay 100e5c89e4eSSatish Balay PetscFunctionBegin; 1013bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 102efca3c55SSatish Balay sprintf(command,"cd %s; emacsclient --no-wait +%d %s\n",pdir,line,file); 103e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 1043bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 105016831caSBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp);if (ierr) PetscFunctionReturn(ierr); 106e5c89e4eSSatish Balay #else 107e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 108e5c89e4eSSatish Balay #endif 1093bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 1103bf036e2SBarry Smith if (!eh) { 111efca3c55SSatish Balay ierr = PetscTraceBackErrorHandler(comm,line,fun,file,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 1123bf036e2SBarry Smith } else { 113efca3c55SSatish Balay ierr = (*eh->handler)(comm,line,fun,file,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 1143bf036e2SBarry Smith } 115e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 116e5c89e4eSSatish Balay } 117e5c89e4eSSatish Balay 118e5c89e4eSSatish Balay /*@C 119e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 120e5c89e4eSSatish Balay 121e5c89e4eSSatish Balay Not Collective 122e5c89e4eSSatish Balay 123e5c89e4eSSatish Balay Input Parameters: 124e5c89e4eSSatish Balay + handler - error handler routine 125e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 126e5c89e4eSSatish Balay example file pointers for error messages etc.) 127e5c89e4eSSatish Balay 128e5c89e4eSSatish Balay Calling sequence of handler: 129efca3c55SSatish Balay $ int handler(MPI_Comm comm,int line,char *func,char *file,PetscErrorCode n,int p,char *mess,void *ctx); 130e5c89e4eSSatish Balay 131e32f2f54SBarry Smith + comm - communicator over which error occured 132e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 133e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 134e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 135668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 136e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 137e5c89e4eSSatish Balay - ctx - the error handler context 138e5c89e4eSSatish Balay 139e5c89e4eSSatish Balay Options Database Keys: 140e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 141e5c89e4eSSatish Balay - -on_error_abort 142e5c89e4eSSatish Balay 143e5c89e4eSSatish Balay Level: intermediate 144e5c89e4eSSatish Balay 145e93bc3c1Svictor Notes: 1467850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 147e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 148e93bc3c1Svictor 14995452b02SPatrick Sanan Fortran Notes: 15095452b02SPatrick Sanan You can only push one error handler from Fortran before poping it. 1517850c7c0SBarry Smith 15234b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 153e5c89e4eSSatish Balay 154e5c89e4eSSatish Balay @*/ 155efca3c55SSatish Balay PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 156e5c89e4eSSatish Balay { 157e5c89e4eSSatish Balay EH neweh; 158e5c89e4eSSatish Balay PetscErrorCode ierr; 159e5c89e4eSSatish Balay 160e5c89e4eSSatish Balay PetscFunctionBegin; 161b00a9115SJed Brown ierr = PetscNew(&neweh);CHKERRQ(ierr); 162a297a907SKarl Rupp if (eh) neweh->previous = eh; 163a297a907SKarl Rupp else neweh->previous = 0; 164e5c89e4eSSatish Balay neweh->handler = handler; 165e5c89e4eSSatish Balay neweh->ctx = ctx; 166e5c89e4eSSatish Balay eh = neweh; 167e5c89e4eSSatish Balay PetscFunctionReturn(0); 168e5c89e4eSSatish Balay } 169e5c89e4eSSatish Balay 170e30d2299SSatish Balay /*@ 171e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 172e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 173e5c89e4eSSatish Balay 174e5c89e4eSSatish Balay Not Collective 175e5c89e4eSSatish Balay 176e5c89e4eSSatish Balay Level: intermediate 177e5c89e4eSSatish Balay 178e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 179e5c89e4eSSatish Balay @*/ 1807087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 181e5c89e4eSSatish Balay { 182e5c89e4eSSatish Balay EH tmp; 183e5c89e4eSSatish Balay PetscErrorCode ierr; 184e5c89e4eSSatish Balay 185e5c89e4eSSatish Balay PetscFunctionBegin; 186e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 187e5c89e4eSSatish Balay tmp = eh; 188e5c89e4eSSatish Balay eh = eh->previous; 189e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 190e5c89e4eSSatish Balay PetscFunctionReturn(0); 191e5c89e4eSSatish Balay } 192e5c89e4eSSatish Balay 193e93bc3c1Svictor /*@C 194e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 195e93bc3c1Svictor level. 196e93bc3c1Svictor 197e93bc3c1Svictor Not Collective 198e93bc3c1Svictor 199e93bc3c1Svictor Input Parameters: 200e32f2f54SBarry Smith + comm - communicator over which error occurred 201e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 202e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 203e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 204e93bc3c1Svictor . n - the generic error number 205e93bc3c1Svictor . p - specific error number 206e93bc3c1Svictor - ctx - error handler context 207e93bc3c1Svictor 208e93bc3c1Svictor Level: developer 209e93bc3c1Svictor 210e93bc3c1Svictor Notes: 211e93bc3c1Svictor Most users need not directly employ this routine and the other error 212e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 213e93bc3c1Svictor the calling sequence 214e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 215e93bc3c1Svictor 216e93bc3c1Svictor Notes for experienced users: 217e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 218e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 219e93bc3c1Svictor and similar errors. 220e93bc3c1Svictor 221e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 222e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 223e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 224e93bc3c1Svictor 225e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 226e93bc3c1Svictor @*/ 227e93bc3c1Svictor 228efca3c55SSatish Balay PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 229e93bc3c1Svictor { 230e93bc3c1Svictor PetscFunctionBegin; 231e93bc3c1Svictor PetscFunctionReturn(n); 232e93bc3c1Svictor } 233e93bc3c1Svictor 234e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 235e5c89e4eSSatish Balay /* 236e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 237e5c89e4eSSatish Balay there must also be made here 238e5c89e4eSSatish Balay */ 239e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 240e5c89e4eSSatish Balay /*55 */ "Out of memory", 241e5c89e4eSSatish Balay "No support for this operation for this object type", 242e5c89e4eSSatish Balay "No support for this operation on this system", 243e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 244e5c89e4eSSatish Balay /*59 */ "Signal received", 245e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 246e5c89e4eSSatish Balay "Argument aliasing not permitted", 247e5c89e4eSSatish Balay "Invalid argument", 248e5c89e4eSSatish Balay /*63 */ "Argument out of range", 249a8d69d7bSBarry Smith "Corrupt argument: https://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 250e5c89e4eSSatish Balay "Unable to open file", 251e5c89e4eSSatish Balay "Read from file failed", 252e5c89e4eSSatish Balay "Write to file failed", 253e5c89e4eSSatish Balay "Invalid pointer", 254e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 255a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 256a8d69d7bSBarry Smith /*71 */ "Zero pivot in LU factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 257e5c89e4eSSatish Balay /*72 */ "Floating point exception", 258e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 259e5c89e4eSSatish Balay "Corrupted Petsc object", 260e5c89e4eSSatish Balay "Arguments are incompatible", 261e5c89e4eSSatish Balay "Error in external library", 262e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 2633d40d40dSBarry Smith "Memory corruption: https://www.mcs.anl.gov/petsc/documentation/installation.html#valgrind", 264e5c89e4eSSatish Balay "Unexpected data in file", 265e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 266a8d69d7bSBarry Smith /*81 */ "Zero pivot in Cholesky factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 267e5c89e4eSSatish Balay " ", 268e5c89e4eSSatish Balay " ", 269a8d69d7bSBarry Smith "Overflow in integer operation: https://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 270e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 2713d40d40dSBarry Smith /*86 */ "Unknown type. Check for miss-spelling or missing package: https://www.mcs.anl.gov/petsc/documentation/installation.html#external", 2723d96e996SBarry Smith /*87 */ "MPI library at runtime is not compatible with MPI used at compile time", 2738cda6cd7SBarry Smith /*88 */ "Error in system call", 274a8d69d7bSBarry Smith /*89 */ "Object Type not set: https://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset", 27573260a9bSLisandro Dalcin /*90 */ " ", 27673260a9bSLisandro Dalcin /* */ " ", 2773d40d40dSBarry Smith /*92 */ "See https://www.mcs.anl.gov/petsc/documentation/linearsolvertable.html for possible LU and Cholesky solvers", 278f560318cSPatrick Sanan /*93 */ "You cannot overwrite this option since that will conflict with other previously set options", 27973260a9bSLisandro Dalcin }; 280e5c89e4eSSatish Balay 281e5c89e4eSSatish Balay /*@C 282e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 283e5c89e4eSSatish Balay 284e5c89e4eSSatish Balay Not Collective 285e5c89e4eSSatish Balay 286e5c89e4eSSatish Balay Input Parameter: 287e5c89e4eSSatish Balay . errnum - the error code 288e5c89e4eSSatish Balay 289e5c89e4eSSatish Balay Output Parameter: 2900298fd71SBarry Smith + text - the error message (NULL if not desired) 2910298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 292e5c89e4eSSatish Balay 293e5c89e4eSSatish Balay Level: developer 294e5c89e4eSSatish Balay 29578179f8bSBarry Smith .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscError(), SETERRQ(), CHKERRQ() 296e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 297e5c89e4eSSatish Balay @*/ 2987087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 299e5c89e4eSSatish Balay { 300e5c89e4eSSatish Balay PetscFunctionBegin; 301a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 302a297a907SKarl Rupp else if (text) *text = 0; 303e5c89e4eSSatish Balay 304a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 305e5c89e4eSSatish Balay PetscFunctionReturn(0); 306e5c89e4eSSatish Balay } 307e5c89e4eSSatish Balay 308984a1229SMatthew G. Knepley #if defined(PETSC_CLANGUAGE_CXX) 309984a1229SMatthew G. Knepley /* C++ exceptions are formally not allowed to propagate through extern "C" code. In practice, far too much software 310984a1229SMatthew G. Knepley * would be broken if implementations did not handle it it some common cases. However, keep in mind 311984a1229SMatthew G. Knepley * 312984a1229SMatthew G. Knepley * Rule 62. Don't allow exceptions to propagate across module boundaries 313984a1229SMatthew G. Knepley * 314984a1229SMatthew G. Knepley * in "C++ Coding Standards" by Sutter and Alexandrescu. (This accounts for part of the ongoing C++ binary interface 315984a1229SMatthew G. Knepley * instability.) Having PETSc raise errors as C++ exceptions was probably misguided and should eventually be removed. 316984a1229SMatthew G. Knepley * 317984a1229SMatthew G. Knepley * Here is the problem: You have a C++ function call a PETSc function, and you would like to maintain the error message 318984a1229SMatthew G. Knepley * and stack information from the PETSc error. You could make everyone write exactly this code in their C++, but that 319984a1229SMatthew G. Knepley * seems crazy to me. 320984a1229SMatthew G. Knepley */ 321984a1229SMatthew G. Knepley #include <sstream> 3224c94c282SMatthew G. Knepley #include <stdexcept> 323984a1229SMatthew G. Knepley static void PetscCxxErrorThrow() { 324984a1229SMatthew G. Knepley const char *str; 325984a1229SMatthew G. Knepley if (eh && eh->ctx) { 326984a1229SMatthew G. Knepley std::ostringstream *msg; 327984a1229SMatthew G. Knepley msg = (std::ostringstream*) eh->ctx; 328984a1229SMatthew G. Knepley str = msg->str().c_str(); 329984a1229SMatthew G. Knepley } else str = "Error detected in C PETSc"; 330984a1229SMatthew G. Knepley 331984a1229SMatthew G. Knepley throw std::runtime_error(str); 332984a1229SMatthew G. Knepley } 333984a1229SMatthew G. Knepley #endif 334984a1229SMatthew G. Knepley 335e5c89e4eSSatish Balay /*@C 336e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 337e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 338e5c89e4eSSatish Balay 339e5c89e4eSSatish Balay Not Collective 340e5c89e4eSSatish Balay 341e5c89e4eSSatish Balay Input Parameters: 342e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 343e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 3443de71b31SHong Zhang . func - the function name in which the error was detected 345e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 346e5c89e4eSSatish Balay . n - the generic error number 347d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 348e5c89e4eSSatish Balay - mess - formatted message string - aka printf 349e5c89e4eSSatish Balay 350e5c89e4eSSatish Balay Level: intermediate 351e5c89e4eSSatish Balay 352e5c89e4eSSatish Balay Notes: 353e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 354e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 355e5c89e4eSSatish Balay sequence 356e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 357e5c89e4eSSatish Balay 358e3081792SBarry Smith Fortran Note: 359e3081792SBarry Smith This routine is used differently from Fortran 360e3081792SBarry Smith $ PetscError(MPI_Comm comm,PetscErrorCode n,PetscErrorType p,char *message) 361e3081792SBarry Smith 362e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 363e5c89e4eSSatish Balay 3643bf036e2SBarry Smith Developer Note: Since this is called after an error condition it should not be calling any error handlers (currently it ignores any error codes) 3653bf036e2SBarry Smith BUT this routine does call regular PETSc functions that may call error handlers, this is problematic and could be fixed by never calling other PETSc routines 3663bf036e2SBarry Smith but this annoying. 3673bf036e2SBarry Smith 36878179f8bSBarry Smith .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2(), PetscErrorMessage() 369e5c89e4eSSatish Balay @*/ 370efca3c55SSatish Balay PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,...) 371e5c89e4eSSatish Balay { 372e5c89e4eSSatish Balay va_list Argp; 373c9a19010SBarry Smith size_t fullLength; 374e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 375fbfcfee5SBarry Smith PetscBool ismain; 3763bf036e2SBarry Smith PetscErrorCode ierr; 377e5c89e4eSSatish Balay 3780d577829SBarry Smith PetscFunctionBegin; 379e5c89e4eSSatish Balay if (!func) func = "User provided function"; 380e5c89e4eSSatish Balay if (!file) file = "User file"; 3810d577829SBarry Smith if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 382e5c89e4eSSatish Balay 383e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 384e5c89e4eSSatish Balay if (mess) { 385e5c89e4eSSatish Balay va_start(Argp,mess); 3862d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 387e5c89e4eSSatish Balay va_end(Argp); 388e5c89e4eSSatish Balay lbuf = buf; 38978179f8bSBarry Smith if (p == PETSC_ERROR_INITIAL) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 390e5c89e4eSSatish Balay } 391e5c89e4eSSatish Balay 392efca3c55SSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,n,p,lbuf,0); 393efca3c55SSatish Balay else ierr = (*eh->handler)(comm,line,func,file,n,p,lbuf,eh->ctx); 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay /* 396e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 397e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 398e5c89e4eSSatish Balay 399e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 400e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 401e5c89e4eSSatish Balay */ 402e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 403fbfcfee5SBarry Smith if (ismain) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 404a297a907SKarl Rupp 4052c280183SJed Brown #if defined(PETSC_CLANGUAGE_CXX) 406d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 4072c280183SJed Brown PetscCxxErrorThrow(); 408d736bfebSBarry Smith } 409d736bfebSBarry Smith #endif 410e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 411e5c89e4eSSatish Balay } 412e5c89e4eSSatish Balay 413e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 414e5c89e4eSSatish Balay 415e5c89e4eSSatish Balay /*@C 416e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 417e5c89e4eSSatish Balay 418e5c89e4eSSatish Balay Collective on PetscViewer 419e5c89e4eSSatish Balay 420e5c89e4eSSatish Balay Input Parameters: 421e5c89e4eSSatish Balay + N - number of integers in array 422e5c89e4eSSatish Balay . idx - array of integers 423e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 424e5c89e4eSSatish Balay 425e5c89e4eSSatish Balay Level: intermediate 426e5c89e4eSSatish Balay 42795452b02SPatrick Sanan Developer Notes: 42895452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 429300a7f5bSBarry Smith 430e5c89e4eSSatish Balay .seealso: PetscRealView() 431e5c89e4eSSatish Balay @*/ 4327087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 433e5c89e4eSSatish Balay { 434e5c89e4eSSatish Balay PetscErrorCode ierr; 435ca0c3be5SJacob Faibussowitsch PetscMPIInt rank,size; 436e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 437ace3abfcSBarry Smith PetscBool iascii,isbinary; 438e5c89e4eSSatish Balay MPI_Comm comm; 439e5c89e4eSSatish Balay 440e5c89e4eSSatish Balay PetscFunctionBegin; 441e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4420700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4435ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 444e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 445ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 446ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 447e5c89e4eSSatish Balay 448251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 449251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 450e5c89e4eSSatish Balay if (iascii) { 4511575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 452e5c89e4eSSatish Balay for (i=0; i<n; i++) { 453ca0c3be5SJacob Faibussowitsch if (size > 1) { 454ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D:", rank, 20*i);CHKERRQ(ierr); 455ca0c3be5SJacob Faibussowitsch } else { 456e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 457ca0c3be5SJacob Faibussowitsch } 458e5c89e4eSSatish Balay for (j=0; j<20; j++) { 459e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 460e5c89e4eSSatish Balay } 461e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 462e5c89e4eSSatish Balay } 463e5c89e4eSSatish Balay if (p) { 464ca0c3be5SJacob Faibussowitsch if (size > 1) { 465ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D:",rank ,20*n);CHKERRQ(ierr); 466ca0c3be5SJacob Faibussowitsch } else { 467e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 468ca0c3be5SJacob Faibussowitsch } 469e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 470e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 471e5c89e4eSSatish Balay } 472e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4731575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 4746805f65bSBarry Smith } else if (isbinary) { 475ca0c3be5SJacob Faibussowitsch PetscMPIInt *sizes,Ntotal,*displs,NN; 476e5c89e4eSSatish Balay PetscInt *array; 477783b601eSJed Brown 4784dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 479e5c89e4eSSatish Balay 480e5c89e4eSSatish Balay if (size > 1) { 481e5c89e4eSSatish Balay if (rank) { 482e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 483300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 484e5c89e4eSSatish Balay } else { 485785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 486adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 487e5c89e4eSSatish Balay Ntotal = sizes[0]; 488785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 489e5c89e4eSSatish Balay displs[0] = 0; 490e5c89e4eSSatish Balay for (i=1; i<size; i++) { 491e5c89e4eSSatish Balay Ntotal += sizes[i]; 492e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 493e5c89e4eSSatish Balay } 494785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 495300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4966805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 497e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 498e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 499e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 500e5c89e4eSSatish Balay } 501e5c89e4eSSatish Balay } else { 50247b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 503e5c89e4eSSatish Balay } 504e5c89e4eSSatish Balay } else { 505e5c89e4eSSatish Balay const char *tname; 506e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 507e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 508e5c89e4eSSatish Balay } 509e5c89e4eSSatish Balay PetscFunctionReturn(0); 510e5c89e4eSSatish Balay } 511e5c89e4eSSatish Balay 512e5c89e4eSSatish Balay /*@C 513e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 514e5c89e4eSSatish Balay 515e5c89e4eSSatish Balay Collective on PetscViewer 516e5c89e4eSSatish Balay 517e5c89e4eSSatish Balay Input Parameters: 518cba51d77SBarry Smith + N - number of PetscReal in array 519cba51d77SBarry Smith . idx - array of PetscReal 520e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 521e5c89e4eSSatish Balay 522e5c89e4eSSatish Balay Level: intermediate 523e5c89e4eSSatish Balay 52495452b02SPatrick Sanan Developer Notes: 52595452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 526300a7f5bSBarry Smith 527e5c89e4eSSatish Balay .seealso: PetscIntView() 528e5c89e4eSSatish Balay @*/ 5297087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 530e5c89e4eSSatish Balay { 531e5c89e4eSSatish Balay PetscErrorCode ierr; 532ca0c3be5SJacob Faibussowitsch PetscMPIInt rank,size; 533e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 534ace3abfcSBarry Smith PetscBool iascii,isbinary; 535e5c89e4eSSatish Balay MPI_Comm comm; 536e5c89e4eSSatish Balay 537e5c89e4eSSatish Balay PetscFunctionBegin; 538e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 5390700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 540e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 541e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 542ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 543ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 544e5c89e4eSSatish Balay 545251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 546251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 547e5c89e4eSSatish Balay if (iascii) { 5481575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 549e5c89e4eSSatish Balay for (i=0; i<n; i++) { 550ca0c3be5SJacob Faibussowitsch if (size > 1) { 551ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,(int)5*i);CHKERRQ(ierr); 552ca0c3be5SJacob Faibussowitsch } else { 5537824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*i);CHKERRQ(ierr); 554ca0c3be5SJacob Faibussowitsch } 555e5c89e4eSSatish Balay for (j=0; j<5; j++) { 556cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*5+j]);CHKERRQ(ierr); 557e5c89e4eSSatish Balay } 558e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 559e5c89e4eSSatish Balay } 560e5c89e4eSSatish Balay if (p) { 561ca0c3be5SJacob Faibussowitsch if (size > 1) { 562ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,(int)5*n);CHKERRQ(ierr); 563ca0c3be5SJacob Faibussowitsch } else { 5647824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*n);CHKERRQ(ierr); 565ca0c3be5SJacob Faibussowitsch } 566cba51d77SBarry Smith for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[5*n+i]);CHKERRQ(ierr);} 567e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 568e5c89e4eSSatish Balay } 569e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5701575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 5716805f65bSBarry Smith } else if (isbinary) { 572ca0c3be5SJacob Faibussowitsch PetscMPIInt *sizes,*displs, Ntotal,NN; 573e5c89e4eSSatish Balay PetscReal *array; 574e5c89e4eSSatish Balay 5754dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 576e5c89e4eSSatish Balay 577e5c89e4eSSatish Balay if (size > 1) { 578e5c89e4eSSatish Balay if (rank) { 579e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 5801d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,0,0,0,MPIU_REAL,0,comm);CHKERRQ(ierr); 581e5c89e4eSSatish Balay } else { 582785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 583e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 584e5c89e4eSSatish Balay Ntotal = sizes[0]; 585785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 586e5c89e4eSSatish Balay displs[0] = 0; 587e5c89e4eSSatish Balay for (i=1; i<size; i++) { 588e5c89e4eSSatish Balay Ntotal += sizes[i]; 589e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 590e5c89e4eSSatish Balay } 591785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 5921d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 5936805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 594e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 595e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 596e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 597e5c89e4eSSatish Balay } 598e5c89e4eSSatish Balay } else { 59947b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 600e5c89e4eSSatish Balay } 601e5c89e4eSSatish Balay } else { 602e5c89e4eSSatish Balay const char *tname; 603e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 604e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 605e5c89e4eSSatish Balay } 606e5c89e4eSSatish Balay PetscFunctionReturn(0); 607e5c89e4eSSatish Balay } 608e5c89e4eSSatish Balay 609e5c89e4eSSatish Balay /*@C 610e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 611e5c89e4eSSatish Balay 612e5c89e4eSSatish Balay Collective on PetscViewer 613e5c89e4eSSatish Balay 614e5c89e4eSSatish Balay Input Parameters: 615e5c89e4eSSatish Balay + N - number of scalars in array 616e5c89e4eSSatish Balay . idx - array of scalars 617e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 618e5c89e4eSSatish Balay 619e5c89e4eSSatish Balay Level: intermediate 620e5c89e4eSSatish Balay 62195452b02SPatrick Sanan Developer Notes: 62295452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 623300a7f5bSBarry Smith 624e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 625e5c89e4eSSatish Balay @*/ 6267087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 627e5c89e4eSSatish Balay { 628e5c89e4eSSatish Balay PetscErrorCode ierr; 629ca0c3be5SJacob Faibussowitsch PetscMPIInt rank,size; 630e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 631ace3abfcSBarry Smith PetscBool iascii,isbinary; 632e5c89e4eSSatish Balay MPI_Comm comm; 633e5c89e4eSSatish Balay 634e5c89e4eSSatish Balay PetscFunctionBegin; 635e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 636e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 6378c34849bSStefano Zampini if (N) PetscValidScalarPointer(idx,2); 638e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 639ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 640ca0c3be5SJacob Faibussowitsch ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 641e5c89e4eSSatish Balay 642251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 643251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 644e5c89e4eSSatish Balay if (iascii) { 6451575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 646e5c89e4eSSatish Balay for (i=0; i<n; i++) { 647ca0c3be5SJacob Faibussowitsch if (size > 1) { 648ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,3*i);CHKERRQ(ierr); 649ca0c3be5SJacob Faibussowitsch } else { 650e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 651ca0c3be5SJacob Faibussowitsch } 652e5c89e4eSSatish Balay for (j=0; j<3; j++) { 653e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 654cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[i*3+j]),(double)PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 655e5c89e4eSSatish Balay #else 656cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*3+j]);CHKERRQ(ierr); 657e5c89e4eSSatish Balay #endif 658e5c89e4eSSatish Balay } 659e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 660e5c89e4eSSatish Balay } 661e5c89e4eSSatish Balay if (p) { 662ca0c3be5SJacob Faibussowitsch if (size > 1) { 663ca0c3be5SJacob Faibussowitsch ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,3*n);CHKERRQ(ierr); 664ca0c3be5SJacob Faibussowitsch } else { 665e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 666ca0c3be5SJacob Faibussowitsch } 667e5c89e4eSSatish Balay for (i=0; i<p; i++) { 668e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 669cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[n*3+i]),(double)PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 670e5c89e4eSSatish Balay #else 671cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[3*n+i]);CHKERRQ(ierr); 672e5c89e4eSSatish Balay #endif 673e5c89e4eSSatish Balay } 674e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 675e5c89e4eSSatish Balay } 676e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6771575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 6786805f65bSBarry Smith } else if (isbinary) { 679ca0c3be5SJacob Faibussowitsch PetscMPIInt *sizes,Ntotal,*displs,NN; 680e5c89e4eSSatish Balay PetscScalar *array; 681e5c89e4eSSatish Balay 6824dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 683e5c89e4eSSatish Balay 684e5c89e4eSSatish Balay if (size > 1) { 685e5c89e4eSSatish Balay if (rank) { 686e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 687300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 688e5c89e4eSSatish Balay } else { 689785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 690e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 691e5c89e4eSSatish Balay Ntotal = sizes[0]; 692785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 693e5c89e4eSSatish Balay displs[0] = 0; 694e5c89e4eSSatish Balay for (i=1; i<size; i++) { 695e5c89e4eSSatish Balay Ntotal += sizes[i]; 696e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 697e5c89e4eSSatish Balay } 698785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 699300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 7006805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 701e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 702e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 703e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 704e5c89e4eSSatish Balay } 705e5c89e4eSSatish Balay } else { 70647b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 707e5c89e4eSSatish Balay } 708e5c89e4eSSatish Balay } else { 709e5c89e4eSSatish Balay const char *tname; 710e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 711e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 712e5c89e4eSSatish Balay } 713e5c89e4eSSatish Balay PetscFunctionReturn(0); 714e5c89e4eSSatish Balay } 715e5c89e4eSSatish Balay 716e5c89e4eSSatish Balay 717e5c89e4eSSatish Balay 718e5c89e4eSSatish Balay 719