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 483bf036e2SBarry Smith Developer Note: Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 493bf036e2SBarry Smith 50e5c89e4eSSatish Balay Concepts: emacs^going to on error 51e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 52e5c89e4eSSatish Balay 53e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 54e5c89e4eSSatish Balay PetscAbortErrorHandler() 55e5c89e4eSSatish Balay @*/ 56efca3c55SSatish Balay PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 57e5c89e4eSSatish Balay { 58e5c89e4eSSatish Balay PetscErrorCode ierr; 59e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 60e5c89e4eSSatish Balay const char *pdir; 61e5c89e4eSSatish Balay FILE *fp; 6296a8e8cbSBarry Smith int rval; 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); 693bf036e2SBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp,&rval);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 1137850c7c0SBarry Smith Fortran Notes: You can only push one error handler from Fortran before poping it. 1147850c7c0SBarry Smith 11534b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 116e5c89e4eSSatish Balay 117e5c89e4eSSatish Balay @*/ 118efca3c55SSatish Balay PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 119e5c89e4eSSatish Balay { 120e5c89e4eSSatish Balay EH neweh; 121e5c89e4eSSatish Balay PetscErrorCode ierr; 122e5c89e4eSSatish Balay 123e5c89e4eSSatish Balay PetscFunctionBegin; 124b00a9115SJed Brown ierr = PetscNew(&neweh);CHKERRQ(ierr); 125a297a907SKarl Rupp if (eh) neweh->previous = eh; 126a297a907SKarl Rupp else neweh->previous = 0; 127e5c89e4eSSatish Balay neweh->handler = handler; 128e5c89e4eSSatish Balay neweh->ctx = ctx; 129e5c89e4eSSatish Balay eh = neweh; 130e5c89e4eSSatish Balay PetscFunctionReturn(0); 131e5c89e4eSSatish Balay } 132e5c89e4eSSatish Balay 133e30d2299SSatish Balay /*@ 134e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 135e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 136e5c89e4eSSatish Balay 137e5c89e4eSSatish Balay Not Collective 138e5c89e4eSSatish Balay 139e5c89e4eSSatish Balay Level: intermediate 140e5c89e4eSSatish Balay 141e5c89e4eSSatish Balay Concepts: error handler^setting 142e5c89e4eSSatish Balay 143e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 144e5c89e4eSSatish Balay @*/ 1457087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 146e5c89e4eSSatish Balay { 147e5c89e4eSSatish Balay EH tmp; 148e5c89e4eSSatish Balay PetscErrorCode ierr; 149e5c89e4eSSatish Balay 150e5c89e4eSSatish Balay PetscFunctionBegin; 151e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 152e5c89e4eSSatish Balay tmp = eh; 153e5c89e4eSSatish Balay eh = eh->previous; 154e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 155e5c89e4eSSatish Balay PetscFunctionReturn(0); 156e5c89e4eSSatish Balay } 157e5c89e4eSSatish Balay 158e93bc3c1Svictor /*@C 159e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 160e93bc3c1Svictor level. 161e93bc3c1Svictor 162e93bc3c1Svictor Not Collective 163e93bc3c1Svictor 164e93bc3c1Svictor Input Parameters: 165e32f2f54SBarry Smith + comm - communicator over which error occurred 166e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 167e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 168e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 169e93bc3c1Svictor . n - the generic error number 170e93bc3c1Svictor . p - specific error number 171e93bc3c1Svictor - ctx - error handler context 172e93bc3c1Svictor 173e93bc3c1Svictor Level: developer 174e93bc3c1Svictor 175e93bc3c1Svictor Notes: 176e93bc3c1Svictor Most users need not directly employ this routine and the other error 177e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 178e93bc3c1Svictor the calling sequence 179e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 180e93bc3c1Svictor 181e93bc3c1Svictor Notes for experienced users: 182e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 183e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 184e93bc3c1Svictor and similar errors. 185e93bc3c1Svictor 186e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 187e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 188e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 189e93bc3c1Svictor 190e93bc3c1Svictor Concepts: error handler 191e93bc3c1Svictor 192e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 193e93bc3c1Svictor @*/ 194e93bc3c1Svictor 195efca3c55SSatish Balay PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 196e93bc3c1Svictor { 197e93bc3c1Svictor PetscFunctionBegin; 198e93bc3c1Svictor PetscFunctionReturn(n); 199e93bc3c1Svictor } 200e93bc3c1Svictor 201e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 202e5c89e4eSSatish Balay /* 203e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 204e5c89e4eSSatish Balay there must also be made here 205e5c89e4eSSatish Balay */ 206e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 207e5c89e4eSSatish Balay /*55 */ "Out of memory", 208e5c89e4eSSatish Balay "No support for this operation for this object type", 209e5c89e4eSSatish Balay "No support for this operation on this system", 210e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 211e5c89e4eSSatish Balay /*59 */ "Signal received", 212e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 213e5c89e4eSSatish Balay "Argument aliasing not permitted", 214e5c89e4eSSatish Balay "Invalid argument", 215e5c89e4eSSatish Balay /*63 */ "Argument out of range", 216114011d0SBarry Smith "Corrupt argument: http://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 217e5c89e4eSSatish Balay "Unable to open file", 218e5c89e4eSSatish Balay "Read from file failed", 219e5c89e4eSSatish Balay "Write to file failed", 220e5c89e4eSSatish Balay "Invalid pointer", 221e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 222a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 2238a05737aSBarry Smith /*71 */ "Zero pivot in LU factorization: http://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 224e5c89e4eSSatish Balay /*72 */ "Floating point exception", 225e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 226e5c89e4eSSatish Balay "Corrupted Petsc object", 227e5c89e4eSSatish Balay "Arguments are incompatible", 228e5c89e4eSSatish Balay "Error in external library", 229e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 230114011d0SBarry Smith "Memory corruption: http://www.mcs.anl.gov/petsc/documentation/installation.html#valgrind", 231e5c89e4eSSatish Balay "Unexpected data in file", 232e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 2338a05737aSBarry Smith /*81 */ "Zero pivot in Cholesky factorization: http://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 234e5c89e4eSSatish Balay " ", 235e5c89e4eSSatish Balay " ", 236114011d0SBarry Smith "Overflow in integer operation: http://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 237e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 238114011d0SBarry Smith /*86 */ "Unknown type. Check for miss-spelling or missing package: http://www.mcs.anl.gov/petsc/documentation/installation.html#external", 2398cda6cd7SBarry Smith /*87 */ "Not used", 2408cda6cd7SBarry Smith /*88 */ "Error in system call", 241114011d0SBarry Smith /*89 */ "Object Type not set: http://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset" 24273260a9bSLisandro Dalcin /*90 */ " ", 24373260a9bSLisandro Dalcin /* */ " ", 244f051a2edSHong Zhang /* */ " ", 245f051a2edSHong Zhang /*93 */ "See http://www.mcs.anl.gov/petsc/documentation/linearsolvertable.html for possible LU and Cholesky solvers", 246c5a392abSBarry Smith /* */ "You cannot overwrite this option since that will conflict with other previously set options", 24773260a9bSLisandro Dalcin /* */ " ", 248f051a2edSHong Zhang /*96 */ " ", 24973260a9bSLisandro Dalcin }; 250e5c89e4eSSatish Balay 251e5c89e4eSSatish Balay /*@C 252e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 253e5c89e4eSSatish Balay 254e5c89e4eSSatish Balay Not Collective 255e5c89e4eSSatish Balay 256e5c89e4eSSatish Balay Input Parameter: 257e5c89e4eSSatish Balay . errnum - the error code 258e5c89e4eSSatish Balay 259e5c89e4eSSatish Balay Output Parameter: 2600298fd71SBarry Smith + text - the error message (NULL if not desired) 2610298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 262e5c89e4eSSatish Balay 263e5c89e4eSSatish Balay Level: developer 264e5c89e4eSSatish Balay 265e5c89e4eSSatish Balay Concepts: error handler^messages 266e5c89e4eSSatish Balay 267e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 268e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 269e5c89e4eSSatish Balay @*/ 2707087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 271e5c89e4eSSatish Balay { 272e5c89e4eSSatish Balay PetscFunctionBegin; 273a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 274a297a907SKarl Rupp else if (text) *text = 0; 275e5c89e4eSSatish Balay 276a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 277e5c89e4eSSatish Balay PetscFunctionReturn(0); 278e5c89e4eSSatish Balay } 279e5c89e4eSSatish Balay 280984a1229SMatthew G. Knepley #if defined(PETSC_CLANGUAGE_CXX) 281984a1229SMatthew G. Knepley /* C++ exceptions are formally not allowed to propagate through extern "C" code. In practice, far too much software 282984a1229SMatthew G. Knepley * would be broken if implementations did not handle it it some common cases. However, keep in mind 283984a1229SMatthew G. Knepley * 284984a1229SMatthew G. Knepley * Rule 62. Don't allow exceptions to propagate across module boundaries 285984a1229SMatthew G. Knepley * 286984a1229SMatthew G. Knepley * in "C++ Coding Standards" by Sutter and Alexandrescu. (This accounts for part of the ongoing C++ binary interface 287984a1229SMatthew G. Knepley * instability.) Having PETSc raise errors as C++ exceptions was probably misguided and should eventually be removed. 288984a1229SMatthew G. Knepley * 289984a1229SMatthew G. Knepley * Here is the problem: You have a C++ function call a PETSc function, and you would like to maintain the error message 290984a1229SMatthew G. Knepley * and stack information from the PETSc error. You could make everyone write exactly this code in their C++, but that 291984a1229SMatthew G. Knepley * seems crazy to me. 292984a1229SMatthew G. Knepley */ 293984a1229SMatthew G. Knepley #include <sstream> 2944c94c282SMatthew G. Knepley #include <stdexcept> 295984a1229SMatthew G. Knepley static void PetscCxxErrorThrow() { 296984a1229SMatthew G. Knepley const char *str; 297984a1229SMatthew G. Knepley if (eh && eh->ctx) { 298984a1229SMatthew G. Knepley std::ostringstream *msg; 299984a1229SMatthew G. Knepley msg = (std::ostringstream*) eh->ctx; 300984a1229SMatthew G. Knepley str = msg->str().c_str(); 301984a1229SMatthew G. Knepley } else str = "Error detected in C PETSc"; 302984a1229SMatthew G. Knepley 303984a1229SMatthew G. Knepley throw std::runtime_error(str); 304984a1229SMatthew G. Knepley } 305984a1229SMatthew G. Knepley #endif 306984a1229SMatthew G. Knepley 307e5c89e4eSSatish Balay /*@C 308e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 309e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 310e5c89e4eSSatish Balay 311e5c89e4eSSatish Balay Not Collective 312e5c89e4eSSatish Balay 313e5c89e4eSSatish Balay Input Parameters: 314e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 315e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 316e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 317e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 318e5c89e4eSSatish Balay . n - the generic error number 319d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 320e5c89e4eSSatish Balay - mess - formatted message string - aka printf 321e5c89e4eSSatish Balay 322e5c89e4eSSatish Balay Level: intermediate 323e5c89e4eSSatish Balay 324e5c89e4eSSatish Balay Notes: 325e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 326e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 327e5c89e4eSSatish Balay sequence 328e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 329e5c89e4eSSatish Balay 330e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 331e5c89e4eSSatish Balay 3323bf036e2SBarry 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) 3333bf036e2SBarry 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 3343bf036e2SBarry Smith but this annoying. 3353bf036e2SBarry Smith 336e5c89e4eSSatish Balay Concepts: error^setting condition 337e5c89e4eSSatish Balay 338e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 339e5c89e4eSSatish Balay @*/ 340efca3c55SSatish Balay PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,...) 341e5c89e4eSSatish Balay { 342e5c89e4eSSatish Balay va_list Argp; 343c9a19010SBarry Smith size_t fullLength; 344e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 345*fbfcfee5SBarry Smith PetscBool ismain; 3463bf036e2SBarry Smith PetscErrorCode ierr; 347e5c89e4eSSatish Balay 3480d577829SBarry Smith PetscFunctionBegin; 349e5c89e4eSSatish Balay if (!func) func = "User provided function"; 350e5c89e4eSSatish Balay if (!file) file = "User file"; 3510d577829SBarry Smith if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 352e5c89e4eSSatish Balay 353e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 354e5c89e4eSSatish Balay if (mess) { 355e5c89e4eSSatish Balay va_start(Argp,mess); 3562d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 357e5c89e4eSSatish Balay va_end(Argp); 358e5c89e4eSSatish Balay lbuf = buf; 359a297a907SKarl Rupp if (p == 1) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 360e5c89e4eSSatish Balay } 361e5c89e4eSSatish Balay 362efca3c55SSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,n,p,lbuf,0); 363efca3c55SSatish Balay else ierr = (*eh->handler)(comm,line,func,file,n,p,lbuf,eh->ctx); 364e5c89e4eSSatish Balay 365e5c89e4eSSatish Balay /* 366e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 367e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 368e5c89e4eSSatish Balay 369e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 370e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 371e5c89e4eSSatish Balay */ 372e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 373*fbfcfee5SBarry Smith if (ismain) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 374a297a907SKarl Rupp 3752c280183SJed Brown #if defined(PETSC_CLANGUAGE_CXX) 376d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 3772c280183SJed Brown PetscCxxErrorThrow(); 378d736bfebSBarry Smith } 379d736bfebSBarry Smith #endif 380e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 381e5c89e4eSSatish Balay } 382e5c89e4eSSatish Balay 383e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 384e5c89e4eSSatish Balay 385e5c89e4eSSatish Balay /*@C 386e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 387e5c89e4eSSatish Balay 388e5c89e4eSSatish Balay Collective on PetscViewer 389e5c89e4eSSatish Balay 390e5c89e4eSSatish Balay Input Parameters: 391e5c89e4eSSatish Balay + N - number of integers in array 392e5c89e4eSSatish Balay . idx - array of integers 393e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay Level: intermediate 396e5c89e4eSSatish Balay 397300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 398300a7f5bSBarry Smith 399e5c89e4eSSatish Balay .seealso: PetscRealView() 400e5c89e4eSSatish Balay @*/ 4017087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 402e5c89e4eSSatish Balay { 403e5c89e4eSSatish Balay PetscErrorCode ierr; 404e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 405ace3abfcSBarry Smith PetscBool iascii,isbinary; 406e5c89e4eSSatish Balay MPI_Comm comm; 407e5c89e4eSSatish Balay 408e5c89e4eSSatish Balay PetscFunctionBegin; 409e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4100700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4115ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 412e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 413e5c89e4eSSatish Balay 414251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 415251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 416e5c89e4eSSatish Balay if (iascii) { 4171575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 418e5c89e4eSSatish Balay for (i=0; i<n; i++) { 419e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 420e5c89e4eSSatish Balay for (j=0; j<20; j++) { 421e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 422e5c89e4eSSatish Balay } 423e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 424e5c89e4eSSatish Balay } 425e5c89e4eSSatish Balay if (p) { 426e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 427e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 428e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 429e5c89e4eSSatish Balay } 430e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4311575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 4326805f65bSBarry Smith } else if (isbinary) { 433783b601eSJed Brown PetscMPIInt rank,size,*sizes,Ntotal,*displs,NN; 434e5c89e4eSSatish Balay PetscInt *array; 435783b601eSJed Brown 4364dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 437e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 438e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 439e5c89e4eSSatish Balay 440e5c89e4eSSatish Balay if (size > 1) { 441e5c89e4eSSatish Balay if (rank) { 442e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 443300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 444e5c89e4eSSatish Balay } else { 445785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 446adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 447e5c89e4eSSatish Balay Ntotal = sizes[0]; 448785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 449e5c89e4eSSatish Balay displs[0] = 0; 450e5c89e4eSSatish Balay for (i=1; i<size; i++) { 451e5c89e4eSSatish Balay Ntotal += sizes[i]; 452e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 453e5c89e4eSSatish Balay } 454785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 455300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4566805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 457e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 458e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 459e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 460e5c89e4eSSatish Balay } 461e5c89e4eSSatish Balay } else { 46247b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 463e5c89e4eSSatish Balay } 464e5c89e4eSSatish Balay } else { 465e5c89e4eSSatish Balay const char *tname; 466e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 467e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 468e5c89e4eSSatish Balay } 469e5c89e4eSSatish Balay PetscFunctionReturn(0); 470e5c89e4eSSatish Balay } 471e5c89e4eSSatish Balay 472e5c89e4eSSatish Balay /*@C 473e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 474e5c89e4eSSatish Balay 475e5c89e4eSSatish Balay Collective on PetscViewer 476e5c89e4eSSatish Balay 477e5c89e4eSSatish Balay Input Parameters: 478cba51d77SBarry Smith + N - number of PetscReal in array 479cba51d77SBarry Smith . idx - array of PetscReal 480e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay Level: intermediate 483e5c89e4eSSatish Balay 484300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 485300a7f5bSBarry Smith 486e5c89e4eSSatish Balay .seealso: PetscIntView() 487e5c89e4eSSatish Balay @*/ 4887087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 489e5c89e4eSSatish Balay { 490e5c89e4eSSatish Balay PetscErrorCode ierr; 491e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 492ace3abfcSBarry Smith PetscBool iascii,isbinary; 493e5c89e4eSSatish Balay MPI_Comm comm; 494e5c89e4eSSatish Balay 495e5c89e4eSSatish Balay PetscFunctionBegin; 496e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4970700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 498e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 499e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 500e5c89e4eSSatish Balay 501251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 502251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 503e5c89e4eSSatish Balay if (iascii) { 5041575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 505e5c89e4eSSatish Balay for (i=0; i<n; i++) { 5067824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*i);CHKERRQ(ierr); 507e5c89e4eSSatish Balay for (j=0; j<5; j++) { 508cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*5+j]);CHKERRQ(ierr); 509e5c89e4eSSatish Balay } 510e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 511e5c89e4eSSatish Balay } 512e5c89e4eSSatish Balay if (p) { 5137824ef9bSBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*n);CHKERRQ(ierr); 514cba51d77SBarry Smith for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[5*n+i]);CHKERRQ(ierr);} 515e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 516e5c89e4eSSatish Balay } 517e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5181575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 5196805f65bSBarry Smith } else if (isbinary) { 520783b601eSJed Brown PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN; 521e5c89e4eSSatish Balay PetscReal *array; 522e5c89e4eSSatish Balay 5234dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 524e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 525e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 526e5c89e4eSSatish Balay 527e5c89e4eSSatish Balay if (size > 1) { 528e5c89e4eSSatish Balay if (rank) { 529e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 5301d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,0,0,0,MPIU_REAL,0,comm);CHKERRQ(ierr); 531e5c89e4eSSatish Balay } else { 532785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 533e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 534e5c89e4eSSatish Balay Ntotal = sizes[0]; 535785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 536e5c89e4eSSatish Balay displs[0] = 0; 537e5c89e4eSSatish Balay for (i=1; i<size; i++) { 538e5c89e4eSSatish Balay Ntotal += sizes[i]; 539e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 540e5c89e4eSSatish Balay } 541785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 5421d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 5436805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 544e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 545e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 546e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 547e5c89e4eSSatish Balay } 548e5c89e4eSSatish Balay } else { 54947b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 550e5c89e4eSSatish Balay } 551e5c89e4eSSatish Balay } else { 552e5c89e4eSSatish Balay const char *tname; 553e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 554e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 555e5c89e4eSSatish Balay } 556e5c89e4eSSatish Balay PetscFunctionReturn(0); 557e5c89e4eSSatish Balay } 558e5c89e4eSSatish Balay 559e5c89e4eSSatish Balay /*@C 560e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 561e5c89e4eSSatish Balay 562e5c89e4eSSatish Balay Collective on PetscViewer 563e5c89e4eSSatish Balay 564e5c89e4eSSatish Balay Input Parameters: 565e5c89e4eSSatish Balay + N - number of scalars in array 566e5c89e4eSSatish Balay . idx - array of scalars 567e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 568e5c89e4eSSatish Balay 569e5c89e4eSSatish Balay Level: intermediate 570e5c89e4eSSatish Balay 571300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 572300a7f5bSBarry Smith 573e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 574e5c89e4eSSatish Balay @*/ 5757087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 576e5c89e4eSSatish Balay { 577e5c89e4eSSatish Balay PetscErrorCode ierr; 578e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 579ace3abfcSBarry Smith PetscBool iascii,isbinary; 580e5c89e4eSSatish Balay MPI_Comm comm; 581e5c89e4eSSatish Balay 582e5c89e4eSSatish Balay PetscFunctionBegin; 583e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 584e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 5858c34849bSStefano Zampini if (N) PetscValidScalarPointer(idx,2); 586e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 587e5c89e4eSSatish Balay 588251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 589251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 590e5c89e4eSSatish Balay if (iascii) { 5911575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 592e5c89e4eSSatish Balay for (i=0; i<n; i++) { 593e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 594e5c89e4eSSatish Balay for (j=0; j<3; j++) { 595e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 596cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[i*3+j]),(double)PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 597e5c89e4eSSatish Balay #else 598cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*3+j]);CHKERRQ(ierr); 599e5c89e4eSSatish Balay #endif 600e5c89e4eSSatish Balay } 601e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 602e5c89e4eSSatish Balay } 603e5c89e4eSSatish Balay if (p) { 604e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 605e5c89e4eSSatish Balay for (i=0; i<p; i++) { 606e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 607cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[n*3+i]),(double)PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 608e5c89e4eSSatish Balay #else 609cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[3*n+i]);CHKERRQ(ierr); 610e5c89e4eSSatish Balay #endif 611e5c89e4eSSatish Balay } 612e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 613e5c89e4eSSatish Balay } 614e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6151575c14dSBarry Smith ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 6166805f65bSBarry Smith } else if (isbinary) { 617783b601eSJed Brown PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN; 618e5c89e4eSSatish Balay PetscScalar *array; 619e5c89e4eSSatish Balay 6204dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 621e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 622e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 623e5c89e4eSSatish Balay 624e5c89e4eSSatish Balay if (size > 1) { 625e5c89e4eSSatish Balay if (rank) { 626e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 627300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 628e5c89e4eSSatish Balay } else { 629785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 630e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 631e5c89e4eSSatish Balay Ntotal = sizes[0]; 632785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 633e5c89e4eSSatish Balay displs[0] = 0; 634e5c89e4eSSatish Balay for (i=1; i<size; i++) { 635e5c89e4eSSatish Balay Ntotal += sizes[i]; 636e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 637e5c89e4eSSatish Balay } 638785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 639300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 6406805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 641e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 642e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 643e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 644e5c89e4eSSatish Balay } 645e5c89e4eSSatish Balay } else { 64647b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 647e5c89e4eSSatish Balay } 648e5c89e4eSSatish Balay } else { 649e5c89e4eSSatish Balay const char *tname; 650e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 651e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 652e5c89e4eSSatish Balay } 653e5c89e4eSSatish Balay PetscFunctionReturn(0); 654e5c89e4eSSatish Balay } 655e5c89e4eSSatish Balay 656e5c89e4eSSatish Balay 657e5c89e4eSSatish Balay 658e5c89e4eSSatish Balay 659