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 8e5c89e4eSSatish Balay typedef struct _EH *EH; 9e5c89e4eSSatish Balay struct _EH { 10efca3c55SSatish Balay PetscErrorCode (*handler)(MPI_Comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*); 11e5c89e4eSSatish Balay void *ctx; 12e5c89e4eSSatish Balay EH previous; 13e5c89e4eSSatish Balay }; 14e5c89e4eSSatish Balay 15e5c89e4eSSatish Balay static EH eh = 0; 16e5c89e4eSSatish Balay 17e5c89e4eSSatish Balay /*@C 18e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 19e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 20e5c89e4eSSatish Balay 21e5c89e4eSSatish Balay Not Collective 22e5c89e4eSSatish Balay 23e5c89e4eSSatish Balay Input Parameters: 24e32f2f54SBarry Smith + comm - communicator over which error occured 25e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 26e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 27e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 28e5c89e4eSSatish Balay . n - the generic error number 29e5c89e4eSSatish Balay . p - specific error number 30e5c89e4eSSatish Balay - ctx - error handler context 31e5c89e4eSSatish Balay 32e5c89e4eSSatish Balay Options Database Key: 33e5c89e4eSSatish Balay . -on_error_emacs <machinename> 34e5c89e4eSSatish Balay 35e5c89e4eSSatish Balay Level: developer 36e5c89e4eSSatish Balay 37e5c89e4eSSatish Balay Notes: 38e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 39e5c89e4eSSatish Balay 40e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 41e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 42e5c89e4eSSatish Balay the calling sequence 43e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 44e5c89e4eSSatish Balay 45e5c89e4eSSatish Balay Notes for experienced users: 46e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 47e5c89e4eSSatish Balay 48cc12936aSPatrick Sanan Developer Note: 49cc12936aSPatrick Sanan Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 503bf036e2SBarry Smith 51e5c89e4eSSatish Balay Concepts: emacs^going to on error 52e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 53e5c89e4eSSatish Balay 54e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 55e5c89e4eSSatish Balay PetscAbortErrorHandler() 56e5c89e4eSSatish Balay @*/ 57efca3c55SSatish Balay PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 58e5c89e4eSSatish Balay { 59e5c89e4eSSatish Balay PetscErrorCode ierr; 60e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 61e5c89e4eSSatish Balay const char *pdir; 62e5c89e4eSSatish Balay FILE *fp; 63e5c89e4eSSatish Balay 64e5c89e4eSSatish Balay PetscFunctionBegin; 653bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 66efca3c55SSatish Balay sprintf(command,"cd %s; emacsclient --no-wait +%d %s\n",pdir,line,file); 67e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 683bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 69016831caSBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp);if (ierr) PetscFunctionReturn(ierr); 70e5c89e4eSSatish Balay #else 71e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 72e5c89e4eSSatish Balay #endif 733bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 743bf036e2SBarry Smith if (!eh) { 75efca3c55SSatish Balay ierr = PetscTraceBackErrorHandler(comm,line,fun,file,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 763bf036e2SBarry Smith } else { 77efca3c55SSatish Balay ierr = (*eh->handler)(comm,line,fun,file,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 783bf036e2SBarry Smith } 79e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 80e5c89e4eSSatish Balay } 81e5c89e4eSSatish Balay 82e5c89e4eSSatish Balay /*@C 83e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 84e5c89e4eSSatish Balay 85e5c89e4eSSatish Balay Not Collective 86e5c89e4eSSatish Balay 87e5c89e4eSSatish Balay Input Parameters: 88e5c89e4eSSatish Balay + handler - error handler routine 89e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 90e5c89e4eSSatish Balay example file pointers for error messages etc.) 91e5c89e4eSSatish Balay 92e5c89e4eSSatish Balay Calling sequence of handler: 93efca3c55SSatish Balay $ int handler(MPI_Comm comm,int line,char *func,char *file,PetscErrorCode n,int p,char *mess,void *ctx); 94e5c89e4eSSatish Balay 95e32f2f54SBarry Smith + comm - communicator over which error occured 96e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 97e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 98e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 99668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 100e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 101e5c89e4eSSatish Balay - ctx - the error handler context 102e5c89e4eSSatish Balay 103e5c89e4eSSatish Balay Options Database Keys: 104e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 105e5c89e4eSSatish Balay - -on_error_abort 106e5c89e4eSSatish Balay 107e5c89e4eSSatish Balay Level: intermediate 108e5c89e4eSSatish Balay 109e93bc3c1Svictor Notes: 1107850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 111e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 112e93bc3c1Svictor 11395452b02SPatrick Sanan Fortran Notes: 11495452b02SPatrick Sanan You can only push one error handler from Fortran before poping it. 1157850c7c0SBarry Smith 11634b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 117e5c89e4eSSatish Balay 118e5c89e4eSSatish Balay @*/ 119efca3c55SSatish Balay PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 120e5c89e4eSSatish Balay { 121e5c89e4eSSatish Balay EH neweh; 122e5c89e4eSSatish Balay PetscErrorCode ierr; 123e5c89e4eSSatish Balay 124e5c89e4eSSatish Balay PetscFunctionBegin; 125b00a9115SJed Brown ierr = PetscNew(&neweh);CHKERRQ(ierr); 126a297a907SKarl Rupp if (eh) neweh->previous = eh; 127a297a907SKarl Rupp else neweh->previous = 0; 128e5c89e4eSSatish Balay neweh->handler = handler; 129e5c89e4eSSatish Balay neweh->ctx = ctx; 130e5c89e4eSSatish Balay eh = neweh; 131e5c89e4eSSatish Balay PetscFunctionReturn(0); 132e5c89e4eSSatish Balay } 133e5c89e4eSSatish Balay 134e30d2299SSatish Balay /*@ 135e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 136e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 137e5c89e4eSSatish Balay 138e5c89e4eSSatish Balay Not Collective 139e5c89e4eSSatish Balay 140e5c89e4eSSatish Balay Level: intermediate 141e5c89e4eSSatish Balay 142e5c89e4eSSatish Balay Concepts: error handler^setting 143e5c89e4eSSatish Balay 144e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 145e5c89e4eSSatish Balay @*/ 1467087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 147e5c89e4eSSatish Balay { 148e5c89e4eSSatish Balay EH tmp; 149e5c89e4eSSatish Balay PetscErrorCode ierr; 150e5c89e4eSSatish Balay 151e5c89e4eSSatish Balay PetscFunctionBegin; 152e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 153e5c89e4eSSatish Balay tmp = eh; 154e5c89e4eSSatish Balay eh = eh->previous; 155e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 156e5c89e4eSSatish Balay PetscFunctionReturn(0); 157e5c89e4eSSatish Balay } 158e5c89e4eSSatish Balay 159e93bc3c1Svictor /*@C 160e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 161e93bc3c1Svictor level. 162e93bc3c1Svictor 163e93bc3c1Svictor Not Collective 164e93bc3c1Svictor 165e93bc3c1Svictor Input Parameters: 166e32f2f54SBarry Smith + comm - communicator over which error occurred 167e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 168e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 169e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 170e93bc3c1Svictor . n - the generic error number 171e93bc3c1Svictor . p - specific error number 172e93bc3c1Svictor - ctx - error handler context 173e93bc3c1Svictor 174e93bc3c1Svictor Level: developer 175e93bc3c1Svictor 176e93bc3c1Svictor Notes: 177e93bc3c1Svictor Most users need not directly employ this routine and the other error 178e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 179e93bc3c1Svictor the calling sequence 180e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 181e93bc3c1Svictor 182e93bc3c1Svictor Notes for experienced users: 183e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 184e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 185e93bc3c1Svictor and similar errors. 186e93bc3c1Svictor 187e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 188e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 189e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 190e93bc3c1Svictor 191e93bc3c1Svictor Concepts: error handler 192e93bc3c1Svictor 193e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 194e93bc3c1Svictor @*/ 195e93bc3c1Svictor 196efca3c55SSatish Balay PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 197e93bc3c1Svictor { 198e93bc3c1Svictor PetscFunctionBegin; 199e93bc3c1Svictor PetscFunctionReturn(n); 200e93bc3c1Svictor } 201e93bc3c1Svictor 202e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 203e5c89e4eSSatish Balay /* 204e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 205e5c89e4eSSatish Balay there must also be made here 206e5c89e4eSSatish Balay */ 207e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 208e5c89e4eSSatish Balay /*55 */ "Out of memory", 209e5c89e4eSSatish Balay "No support for this operation for this object type", 210e5c89e4eSSatish Balay "No support for this operation on this system", 211e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 212e5c89e4eSSatish Balay /*59 */ "Signal received", 213e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 214e5c89e4eSSatish Balay "Argument aliasing not permitted", 215e5c89e4eSSatish Balay "Invalid argument", 216e5c89e4eSSatish Balay /*63 */ "Argument out of range", 217*a8d69d7bSBarry Smith "Corrupt argument: https://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 218e5c89e4eSSatish Balay "Unable to open file", 219e5c89e4eSSatish Balay "Read from file failed", 220e5c89e4eSSatish Balay "Write to file failed", 221e5c89e4eSSatish Balay "Invalid pointer", 222e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 223a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 224*a8d69d7bSBarry Smith /*71 */ "Zero pivot in LU factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 225e5c89e4eSSatish Balay /*72 */ "Floating point exception", 226e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 227e5c89e4eSSatish Balay "Corrupted Petsc object", 228e5c89e4eSSatish Balay "Arguments are incompatible", 229e5c89e4eSSatish Balay "Error in external library", 230e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 231*a8d69d7bSBarry Smith "Memory corruption: https://www.mcs.anl.gov/petsc/documentation//installation.html#valgrind", 232e5c89e4eSSatish Balay "Unexpected data in file", 233e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 234*a8d69d7bSBarry Smith /*81 */ "Zero pivot in Cholesky factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 235e5c89e4eSSatish Balay " ", 236e5c89e4eSSatish Balay " ", 237*a8d69d7bSBarry Smith "Overflow in integer operation: https://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 238e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 239*a8d69d7bSBarry Smith /*86 */ "Unknown type. Check for miss-spelling or missing package: https://www.mcs.anl.gov/petsc/documentation//installation.html#external", 2403d96e996SBarry Smith /*87 */ "MPI library at runtime is not compatible with MPI used at compile time", 2418cda6cd7SBarry Smith /*88 */ "Error in system call", 242*a8d69d7bSBarry Smith /*89 */ "Object Type not set: https://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset", 24373260a9bSLisandro Dalcin /*90 */ " ", 24473260a9bSLisandro Dalcin /* */ " ", 245*a8d69d7bSBarry Smith /*92 */ "See https://www.mcs.anl.gov/petsc/documentation//linearsolvertable.html for possible LU and Cholesky solvers", 246f560318cSPatrick Sanan /*93 */ "You cannot overwrite this option since that will conflict with other previously set options", 24773260a9bSLisandro Dalcin }; 248e5c89e4eSSatish Balay 249e5c89e4eSSatish Balay /*@C 250e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 251e5c89e4eSSatish Balay 252e5c89e4eSSatish Balay Not Collective 253e5c89e4eSSatish Balay 254e5c89e4eSSatish Balay Input Parameter: 255e5c89e4eSSatish Balay . errnum - the error code 256e5c89e4eSSatish Balay 257e5c89e4eSSatish Balay Output Parameter: 2580298fd71SBarry Smith + text - the error message (NULL if not desired) 2590298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 260e5c89e4eSSatish Balay 261e5c89e4eSSatish Balay Level: developer 262e5c89e4eSSatish Balay 263e5c89e4eSSatish Balay Concepts: error handler^messages 264e5c89e4eSSatish Balay 26578179f8bSBarry Smith .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscError(), SETERRQ(), CHKERRQ() 266e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 267e5c89e4eSSatish Balay @*/ 2687087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 269e5c89e4eSSatish Balay { 270e5c89e4eSSatish Balay PetscFunctionBegin; 271a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 272a297a907SKarl Rupp else if (text) *text = 0; 273e5c89e4eSSatish Balay 274a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 275e5c89e4eSSatish Balay PetscFunctionReturn(0); 276e5c89e4eSSatish Balay } 277e5c89e4eSSatish Balay 278984a1229SMatthew G. Knepley #if defined(PETSC_CLANGUAGE_CXX) 279984a1229SMatthew G. Knepley /* C++ exceptions are formally not allowed to propagate through extern "C" code. In practice, far too much software 280984a1229SMatthew G. Knepley * would be broken if implementations did not handle it it some common cases. However, keep in mind 281984a1229SMatthew G. Knepley * 282984a1229SMatthew G. Knepley * Rule 62. Don't allow exceptions to propagate across module boundaries 283984a1229SMatthew G. Knepley * 284984a1229SMatthew G. Knepley * in "C++ Coding Standards" by Sutter and Alexandrescu. (This accounts for part of the ongoing C++ binary interface 285984a1229SMatthew G. Knepley * instability.) Having PETSc raise errors as C++ exceptions was probably misguided and should eventually be removed. 286984a1229SMatthew G. Knepley * 287984a1229SMatthew G. Knepley * Here is the problem: You have a C++ function call a PETSc function, and you would like to maintain the error message 288984a1229SMatthew G. Knepley * and stack information from the PETSc error. You could make everyone write exactly this code in their C++, but that 289984a1229SMatthew G. Knepley * seems crazy to me. 290984a1229SMatthew G. Knepley */ 291984a1229SMatthew G. Knepley #include <sstream> 2924c94c282SMatthew G. Knepley #include <stdexcept> 293984a1229SMatthew G. Knepley static void PetscCxxErrorThrow() { 294984a1229SMatthew G. Knepley const char *str; 295984a1229SMatthew G. Knepley if (eh && eh->ctx) { 296984a1229SMatthew G. Knepley std::ostringstream *msg; 297984a1229SMatthew G. Knepley msg = (std::ostringstream*) eh->ctx; 298984a1229SMatthew G. Knepley str = msg->str().c_str(); 299984a1229SMatthew G. Knepley } else str = "Error detected in C PETSc"; 300984a1229SMatthew G. Knepley 301984a1229SMatthew G. Knepley throw std::runtime_error(str); 302984a1229SMatthew G. Knepley } 303984a1229SMatthew G. Knepley #endif 304984a1229SMatthew G. Knepley 305e5c89e4eSSatish Balay /*@C 306e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 307e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 308e5c89e4eSSatish Balay 309e5c89e4eSSatish Balay Not Collective 310e5c89e4eSSatish Balay 311e5c89e4eSSatish Balay Input Parameters: 312e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 313e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 3143de71b31SHong Zhang . func - the function name in which the error was detected 315e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 316e5c89e4eSSatish Balay . n - the generic error number 317d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 318e5c89e4eSSatish Balay - mess - formatted message string - aka printf 319e5c89e4eSSatish Balay 320e5c89e4eSSatish Balay Level: intermediate 321e5c89e4eSSatish Balay 322e5c89e4eSSatish Balay Notes: 323e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 324e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 325e5c89e4eSSatish Balay sequence 326e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 327e5c89e4eSSatish Balay 328e3081792SBarry Smith Fortran Note: 329e3081792SBarry Smith This routine is used differently from Fortran 330e3081792SBarry Smith $ PetscError(MPI_Comm comm,PetscErrorCode n,PetscErrorType p,char *message) 331e3081792SBarry Smith 332e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 333e5c89e4eSSatish Balay 3343bf036e2SBarry 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) 3353bf036e2SBarry 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 3363bf036e2SBarry Smith but this annoying. 3373bf036e2SBarry Smith 338e5c89e4eSSatish Balay Concepts: error^setting condition 339e5c89e4eSSatish Balay 34078179f8bSBarry Smith .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2(), PetscErrorMessage() 341e5c89e4eSSatish Balay @*/ 342efca3c55SSatish Balay PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,...) 343e5c89e4eSSatish Balay { 344e5c89e4eSSatish Balay va_list Argp; 345c9a19010SBarry Smith size_t fullLength; 346e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 347fbfcfee5SBarry Smith PetscBool ismain; 3483bf036e2SBarry Smith PetscErrorCode ierr; 349e5c89e4eSSatish Balay 3500d577829SBarry Smith PetscFunctionBegin; 351e5c89e4eSSatish Balay if (!func) func = "User provided function"; 352e5c89e4eSSatish Balay if (!file) file = "User file"; 3530d577829SBarry Smith if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 354e5c89e4eSSatish Balay 355e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 356e5c89e4eSSatish Balay if (mess) { 357e5c89e4eSSatish Balay va_start(Argp,mess); 3582d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 359e5c89e4eSSatish Balay va_end(Argp); 360e5c89e4eSSatish Balay lbuf = buf; 36178179f8bSBarry Smith if (p == PETSC_ERROR_INITIAL) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 362e5c89e4eSSatish Balay } 363e5c89e4eSSatish Balay 364efca3c55SSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,n,p,lbuf,0); 365efca3c55SSatish Balay else ierr = (*eh->handler)(comm,line,func,file,n,p,lbuf,eh->ctx); 366e5c89e4eSSatish Balay 367e5c89e4eSSatish Balay /* 368e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 369e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 370e5c89e4eSSatish Balay 371e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 372e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 373e5c89e4eSSatish Balay */ 374e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 375fbfcfee5SBarry Smith if (ismain) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 376a297a907SKarl Rupp 3772c280183SJed Brown #if defined(PETSC_CLANGUAGE_CXX) 378d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 3792c280183SJed Brown PetscCxxErrorThrow(); 380d736bfebSBarry Smith } 381d736bfebSBarry Smith #endif 382e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 383e5c89e4eSSatish Balay } 384e5c89e4eSSatish Balay 385e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 386e5c89e4eSSatish Balay 387e5c89e4eSSatish Balay /*@C 388e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 389e5c89e4eSSatish Balay 390e5c89e4eSSatish Balay Collective on PetscViewer 391e5c89e4eSSatish Balay 392e5c89e4eSSatish Balay Input Parameters: 393e5c89e4eSSatish Balay + N - number of integers in array 394e5c89e4eSSatish Balay . idx - array of integers 395e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 396e5c89e4eSSatish Balay 397e5c89e4eSSatish Balay Level: intermediate 398e5c89e4eSSatish Balay 39995452b02SPatrick Sanan Developer Notes: 40095452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 401300a7f5bSBarry Smith 402e5c89e4eSSatish Balay .seealso: PetscRealView() 403e5c89e4eSSatish Balay @*/ 4047087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 405e5c89e4eSSatish Balay { 406e5c89e4eSSatish Balay PetscErrorCode ierr; 407e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 408ace3abfcSBarry Smith PetscBool iascii,isbinary; 409e5c89e4eSSatish Balay MPI_Comm comm; 410e5c89e4eSSatish Balay 411e5c89e4eSSatish Balay PetscFunctionBegin; 412e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4130700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4145ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 415e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 416e5c89e4eSSatish Balay 417251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 418251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 419e5c89e4eSSatish Balay if (iascii) { 4201575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 421e5c89e4eSSatish Balay for (i=0; i<n; i++) { 422e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 423e5c89e4eSSatish Balay for (j=0; j<20; j++) { 424e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 425e5c89e4eSSatish Balay } 426e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 427e5c89e4eSSatish Balay } 428e5c89e4eSSatish Balay if (p) { 429e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 430e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 431e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 432e5c89e4eSSatish Balay } 433e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4341575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 4356805f65bSBarry Smith } else if (isbinary) { 436783b601eSJed Brown PetscMPIInt rank,size,*sizes,Ntotal,*displs,NN; 437e5c89e4eSSatish Balay PetscInt *array; 438783b601eSJed Brown 4394dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 440e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 441e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 442e5c89e4eSSatish Balay 443e5c89e4eSSatish Balay if (size > 1) { 444e5c89e4eSSatish Balay if (rank) { 445e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 446300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 447e5c89e4eSSatish Balay } else { 448785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 449adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 450e5c89e4eSSatish Balay Ntotal = sizes[0]; 451785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 452e5c89e4eSSatish Balay displs[0] = 0; 453e5c89e4eSSatish Balay for (i=1; i<size; i++) { 454e5c89e4eSSatish Balay Ntotal += sizes[i]; 455e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 456e5c89e4eSSatish Balay } 457785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 458300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4596805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 460e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 461e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 462e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 463e5c89e4eSSatish Balay } 464e5c89e4eSSatish Balay } else { 46547b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 466e5c89e4eSSatish Balay } 467e5c89e4eSSatish Balay } else { 468e5c89e4eSSatish Balay const char *tname; 469e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 470e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 471e5c89e4eSSatish Balay } 472e5c89e4eSSatish Balay PetscFunctionReturn(0); 473e5c89e4eSSatish Balay } 474e5c89e4eSSatish Balay 475e5c89e4eSSatish Balay /*@C 476e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 477e5c89e4eSSatish Balay 478e5c89e4eSSatish Balay Collective on PetscViewer 479e5c89e4eSSatish Balay 480e5c89e4eSSatish Balay Input Parameters: 481cba51d77SBarry Smith + N - number of PetscReal in array 482cba51d77SBarry Smith . idx - array of PetscReal 483e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 484e5c89e4eSSatish Balay 485e5c89e4eSSatish Balay Level: intermediate 486e5c89e4eSSatish Balay 48795452b02SPatrick Sanan Developer Notes: 48895452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 489300a7f5bSBarry Smith 490e5c89e4eSSatish Balay .seealso: PetscIntView() 491e5c89e4eSSatish Balay @*/ 4927087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 493e5c89e4eSSatish Balay { 494e5c89e4eSSatish Balay PetscErrorCode ierr; 495e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 496ace3abfcSBarry Smith PetscBool iascii,isbinary; 497e5c89e4eSSatish Balay MPI_Comm comm; 498e5c89e4eSSatish Balay 499e5c89e4eSSatish Balay PetscFunctionBegin; 500e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 5010700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 502e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 503e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 504e5c89e4eSSatish Balay 505251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 506251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 507e5c89e4eSSatish Balay if (iascii) { 5081575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 509e5c89e4eSSatish Balay for (i=0; i<n; i++) { 5107824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*i);CHKERRQ(ierr); 511e5c89e4eSSatish Balay for (j=0; j<5; j++) { 512cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*5+j]);CHKERRQ(ierr); 513e5c89e4eSSatish Balay } 514e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 515e5c89e4eSSatish Balay } 516e5c89e4eSSatish Balay if (p) { 5177824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*n);CHKERRQ(ierr); 518cba51d77SBarry Smith for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[5*n+i]);CHKERRQ(ierr);} 519e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 520e5c89e4eSSatish Balay } 521e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5221575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 5236805f65bSBarry Smith } else if (isbinary) { 524783b601eSJed Brown PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN; 525e5c89e4eSSatish Balay PetscReal *array; 526e5c89e4eSSatish Balay 5274dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 528e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 529e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 530e5c89e4eSSatish Balay 531e5c89e4eSSatish Balay if (size > 1) { 532e5c89e4eSSatish Balay if (rank) { 533e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 5341d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,0,0,0,MPIU_REAL,0,comm);CHKERRQ(ierr); 535e5c89e4eSSatish Balay } else { 536785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 537e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 538e5c89e4eSSatish Balay Ntotal = sizes[0]; 539785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 540e5c89e4eSSatish Balay displs[0] = 0; 541e5c89e4eSSatish Balay for (i=1; i<size; i++) { 542e5c89e4eSSatish Balay Ntotal += sizes[i]; 543e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 544e5c89e4eSSatish Balay } 545785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 5461d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 5476805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 548e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 549e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 550e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 551e5c89e4eSSatish Balay } 552e5c89e4eSSatish Balay } else { 55347b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 554e5c89e4eSSatish Balay } 555e5c89e4eSSatish Balay } else { 556e5c89e4eSSatish Balay const char *tname; 557e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 558e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 559e5c89e4eSSatish Balay } 560e5c89e4eSSatish Balay PetscFunctionReturn(0); 561e5c89e4eSSatish Balay } 562e5c89e4eSSatish Balay 563e5c89e4eSSatish Balay /*@C 564e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 565e5c89e4eSSatish Balay 566e5c89e4eSSatish Balay Collective on PetscViewer 567e5c89e4eSSatish Balay 568e5c89e4eSSatish Balay Input Parameters: 569e5c89e4eSSatish Balay + N - number of scalars in array 570e5c89e4eSSatish Balay . idx - array of scalars 571e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 572e5c89e4eSSatish Balay 573e5c89e4eSSatish Balay Level: intermediate 574e5c89e4eSSatish Balay 57595452b02SPatrick Sanan Developer Notes: 57695452b02SPatrick Sanan idx cannot be const because may be passed to binary viewer where byte swapping is done 577300a7f5bSBarry Smith 578e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 579e5c89e4eSSatish Balay @*/ 5807087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 581e5c89e4eSSatish Balay { 582e5c89e4eSSatish Balay PetscErrorCode ierr; 583e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 584ace3abfcSBarry Smith PetscBool iascii,isbinary; 585e5c89e4eSSatish Balay MPI_Comm comm; 586e5c89e4eSSatish Balay 587e5c89e4eSSatish Balay PetscFunctionBegin; 588e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 589e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 5908c34849bSStefano Zampini if (N) PetscValidScalarPointer(idx,2); 591e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 592e5c89e4eSSatish Balay 593251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 594251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 595e5c89e4eSSatish Balay if (iascii) { 5961575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 597e5c89e4eSSatish Balay for (i=0; i<n; i++) { 598e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 599e5c89e4eSSatish Balay for (j=0; j<3; j++) { 600e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 601cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[i*3+j]),(double)PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 602e5c89e4eSSatish Balay #else 603cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*3+j]);CHKERRQ(ierr); 604e5c89e4eSSatish Balay #endif 605e5c89e4eSSatish Balay } 606e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 607e5c89e4eSSatish Balay } 608e5c89e4eSSatish Balay if (p) { 609e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 610e5c89e4eSSatish Balay for (i=0; i<p; i++) { 611e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 612cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[n*3+i]),(double)PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 613e5c89e4eSSatish Balay #else 614cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[3*n+i]);CHKERRQ(ierr); 615e5c89e4eSSatish Balay #endif 616e5c89e4eSSatish Balay } 617e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 618e5c89e4eSSatish Balay } 619e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6201575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 6216805f65bSBarry Smith } else if (isbinary) { 622783b601eSJed Brown PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN; 623e5c89e4eSSatish Balay PetscScalar *array; 624e5c89e4eSSatish Balay 6254dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 626e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 627e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 628e5c89e4eSSatish Balay 629e5c89e4eSSatish Balay if (size > 1) { 630e5c89e4eSSatish Balay if (rank) { 631e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 632300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 633e5c89e4eSSatish Balay } else { 634785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 635e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 636e5c89e4eSSatish Balay Ntotal = sizes[0]; 637785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 638e5c89e4eSSatish Balay displs[0] = 0; 639e5c89e4eSSatish Balay for (i=1; i<size; i++) { 640e5c89e4eSSatish Balay Ntotal += sizes[i]; 641e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 642e5c89e4eSSatish Balay } 643785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 644300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 6456805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 646e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 647e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 648e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 649e5c89e4eSSatish Balay } 650e5c89e4eSSatish Balay } else { 65147b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 652e5c89e4eSSatish Balay } 653e5c89e4eSSatish Balay } else { 654e5c89e4eSSatish Balay const char *tname; 655e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 656e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 657e5c89e4eSSatish Balay } 658e5c89e4eSSatish Balay PetscFunctionReturn(0); 659e5c89e4eSSatish Balay } 660e5c89e4eSSatish Balay 661e5c89e4eSSatish Balay 662e5c89e4eSSatish Balay 663e5c89e4eSSatish Balay 664