17d0a6c19SBarry Smith 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay Code that allows one to set the error handlers 4e5c89e4eSSatish Balay */ 5afcb2eb5SJed Brown #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 #undef __FUNCT__ 18e5c89e4eSSatish Balay #define __FUNCT__ "PetscEmacsClientErrorHandler" 19e5c89e4eSSatish Balay /*@C 20e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 21e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 22e5c89e4eSSatish Balay 23e5c89e4eSSatish Balay Not Collective 24e5c89e4eSSatish Balay 25e5c89e4eSSatish Balay Input Parameters: 26e32f2f54SBarry Smith + comm - communicator over which error occured 27e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 28e5c89e4eSSatish Balay . func - the function where error is detected (indicated by __FUNCT__) 29e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 30e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 31e5c89e4eSSatish Balay . n - the generic error number 32e5c89e4eSSatish Balay . p - specific error number 33e5c89e4eSSatish Balay - ctx - error handler context 34e5c89e4eSSatish Balay 35e5c89e4eSSatish Balay Options Database Key: 36e5c89e4eSSatish Balay . -on_error_emacs <machinename> 37e5c89e4eSSatish Balay 38e5c89e4eSSatish Balay Level: developer 39e5c89e4eSSatish Balay 40e5c89e4eSSatish Balay Notes: 41e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 42e5c89e4eSSatish Balay 43e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 44e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 45e5c89e4eSSatish Balay the calling sequence 46e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 47e5c89e4eSSatish Balay 48e5c89e4eSSatish Balay Notes for experienced users: 49e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 50e5c89e4eSSatish Balay 513bf036e2SBarry Smith Developer Note: Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 523bf036e2SBarry Smith 53e5c89e4eSSatish Balay Concepts: emacs^going to on error 54e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 55e5c89e4eSSatish Balay 56e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 57e5c89e4eSSatish Balay PetscAbortErrorHandler() 58e5c89e4eSSatish Balay @*/ 59efca3c55SSatish Balay PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 60e5c89e4eSSatish Balay { 61e5c89e4eSSatish Balay PetscErrorCode ierr; 62e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 63e5c89e4eSSatish Balay const char *pdir; 64e5c89e4eSSatish Balay FILE *fp; 6596a8e8cbSBarry Smith int rval; 66e5c89e4eSSatish Balay 67e5c89e4eSSatish Balay PetscFunctionBegin; 683bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 69efca3c55SSatish Balay sprintf(command,"cd %s; emacsclient --no-wait +%d %s\n",pdir,line,file); 70e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 713bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 723bf036e2SBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp,&rval);if (ierr) PetscFunctionReturn(ierr); 73e5c89e4eSSatish Balay #else 74e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 75e5c89e4eSSatish Balay #endif 763bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 773bf036e2SBarry Smith if (!eh) { 78efca3c55SSatish Balay ierr = PetscTraceBackErrorHandler(comm,line,fun,file,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 793bf036e2SBarry Smith } else { 80efca3c55SSatish Balay ierr = (*eh->handler)(comm,line,fun,file,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 813bf036e2SBarry Smith } 82e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 83e5c89e4eSSatish Balay } 84e5c89e4eSSatish Balay 85e5c89e4eSSatish Balay #undef __FUNCT__ 86e5c89e4eSSatish Balay #define __FUNCT__ "PetscPushErrorHandler" 87e5c89e4eSSatish Balay /*@C 88e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 89e5c89e4eSSatish Balay 90e5c89e4eSSatish Balay Not Collective 91e5c89e4eSSatish Balay 92e5c89e4eSSatish Balay Input Parameters: 93e5c89e4eSSatish Balay + handler - error handler routine 94e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 95e5c89e4eSSatish Balay example file pointers for error messages etc.) 96e5c89e4eSSatish Balay 97e5c89e4eSSatish Balay Calling sequence of handler: 98efca3c55SSatish Balay $ int handler(MPI_Comm comm,int line,char *func,char *file,PetscErrorCode n,int p,char *mess,void *ctx); 99e5c89e4eSSatish Balay 100e32f2f54SBarry Smith + comm - communicator over which error occured 101e32f2f54SBarry Smith . func - the function where the error occured (indicated by __FUNCT__) 102e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 103e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 104e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 105668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 106e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 107e5c89e4eSSatish Balay - ctx - the error handler context 108e5c89e4eSSatish Balay 109e5c89e4eSSatish Balay Options Database Keys: 110e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 111e5c89e4eSSatish Balay - -on_error_abort 112e5c89e4eSSatish Balay 113e5c89e4eSSatish Balay Level: intermediate 114e5c89e4eSSatish Balay 115e93bc3c1Svictor Notes: 1167850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 117e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 118e93bc3c1Svictor 1197850c7c0SBarry Smith Fortran Notes: You can only push one error handler from Fortran before poping it. 1207850c7c0SBarry Smith 12134b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 122e5c89e4eSSatish Balay 123e5c89e4eSSatish Balay @*/ 124efca3c55SSatish Balay PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 125e5c89e4eSSatish Balay { 126e5c89e4eSSatish Balay EH neweh; 127e5c89e4eSSatish Balay PetscErrorCode ierr; 128e5c89e4eSSatish Balay 129e5c89e4eSSatish Balay PetscFunctionBegin; 130b00a9115SJed Brown ierr = PetscNew(&neweh);CHKERRQ(ierr); 131a297a907SKarl Rupp if (eh) neweh->previous = eh; 132a297a907SKarl Rupp else neweh->previous = 0; 133e5c89e4eSSatish Balay neweh->handler = handler; 134e5c89e4eSSatish Balay neweh->ctx = ctx; 135e5c89e4eSSatish Balay eh = neweh; 136e5c89e4eSSatish Balay PetscFunctionReturn(0); 137e5c89e4eSSatish Balay } 138e5c89e4eSSatish Balay 139e5c89e4eSSatish Balay #undef __FUNCT__ 140e5c89e4eSSatish Balay #define __FUNCT__ "PetscPopErrorHandler" 141e30d2299SSatish Balay /*@ 142e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 143e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 144e5c89e4eSSatish Balay 145e5c89e4eSSatish Balay Not Collective 146e5c89e4eSSatish Balay 147e5c89e4eSSatish Balay Level: intermediate 148e5c89e4eSSatish Balay 149e5c89e4eSSatish Balay Concepts: error handler^setting 150e5c89e4eSSatish Balay 151e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 152e5c89e4eSSatish Balay @*/ 1537087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 154e5c89e4eSSatish Balay { 155e5c89e4eSSatish Balay EH tmp; 156e5c89e4eSSatish Balay PetscErrorCode ierr; 157e5c89e4eSSatish Balay 158e5c89e4eSSatish Balay PetscFunctionBegin; 159e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 160e5c89e4eSSatish Balay tmp = eh; 161e5c89e4eSSatish Balay eh = eh->previous; 162e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 163e5c89e4eSSatish Balay PetscFunctionReturn(0); 164e5c89e4eSSatish Balay } 165e5c89e4eSSatish Balay 166e93bc3c1Svictor #undef __FUNCT__ 167e93bc3c1Svictor #define __FUNCT__ "PetscReturnErrorHandler" 168e93bc3c1Svictor /*@C 169e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 170e93bc3c1Svictor level. 171e93bc3c1Svictor 172e93bc3c1Svictor Not Collective 173e93bc3c1Svictor 174e93bc3c1Svictor Input Parameters: 175e32f2f54SBarry Smith + comm - communicator over which error occurred 176e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 177e93bc3c1Svictor . func - the function where error is detected (indicated by __FUNCT__) 178e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 179e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 180e93bc3c1Svictor . n - the generic error number 181e93bc3c1Svictor . p - specific error number 182e93bc3c1Svictor - ctx - error handler context 183e93bc3c1Svictor 184e93bc3c1Svictor Level: developer 185e93bc3c1Svictor 186e93bc3c1Svictor Notes: 187e93bc3c1Svictor Most users need not directly employ this routine and the other error 188e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 189e93bc3c1Svictor the calling sequence 190e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 191e93bc3c1Svictor 192e93bc3c1Svictor Notes for experienced users: 193e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 194e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 195e93bc3c1Svictor and similar errors. 196e93bc3c1Svictor 197e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 198e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 199e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 200e93bc3c1Svictor 201e93bc3c1Svictor Concepts: error handler 202e93bc3c1Svictor 203e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 204e93bc3c1Svictor @*/ 205e93bc3c1Svictor 206efca3c55SSatish Balay PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 207e93bc3c1Svictor { 208e93bc3c1Svictor PetscFunctionBegin; 209e93bc3c1Svictor PetscFunctionReturn(n); 210e93bc3c1Svictor } 211e93bc3c1Svictor 212e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 213e5c89e4eSSatish Balay /* 214e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 215e5c89e4eSSatish Balay there must also be made here 216e5c89e4eSSatish Balay */ 217e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 218e5c89e4eSSatish Balay /*55 */ "Out of memory", 219e5c89e4eSSatish Balay "No support for this operation for this object type", 220e5c89e4eSSatish Balay "No support for this operation on this system", 221e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 222e5c89e4eSSatish Balay /*59 */ "Signal received", 223e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 224e5c89e4eSSatish Balay "Argument aliasing not permitted", 225e5c89e4eSSatish Balay "Invalid argument", 226e5c89e4eSSatish Balay /*63 */ "Argument out of range", 227f08646a8SSatish Balay "Corrupt argument:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 228e5c89e4eSSatish Balay "Unable to open file", 229e5c89e4eSSatish Balay "Read from file failed", 230e5c89e4eSSatish Balay "Write to file failed", 231e5c89e4eSSatish Balay "Invalid pointer", 232e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 233a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 234f08646a8SSatish Balay /*71 */ "Detected zero pivot in LU factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 235e5c89e4eSSatish Balay /*72 */ "Floating point exception", 236e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 237e5c89e4eSSatish Balay "Corrupted Petsc object", 238e5c89e4eSSatish Balay "Arguments are incompatible", 239e5c89e4eSSatish Balay "Error in external library", 240e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 241e5c89e4eSSatish Balay "Memory corruption", 242e5c89e4eSSatish Balay "Unexpected data in file", 243e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 244f08646a8SSatish Balay /*81 */ "Detected zero pivot in Cholesky factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 245e5c89e4eSSatish Balay " ", 246e5c89e4eSSatish Balay " ", 247f08646a8SSatish Balay "Overflow in integer operation:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 248e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 249f08646a8SSatish Balay /*86 */ "Unknown type. Check for miss-spelling or missing external package needed for type:\nsee http://www.mcs.anl.gov/petsc/documentation/installation.html#external", 2508cda6cd7SBarry Smith /*87 */ "Not used", 2518cda6cd7SBarry Smith /*88 */ "Error in system call", 252f08646a8SSatish Balay /*89 */ "Object Type not set:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset" 25373260a9bSLisandro Dalcin /*90 */ " ", 25473260a9bSLisandro Dalcin /* */ " ", 25573260a9bSLisandro Dalcin /* */ " ", 25673260a9bSLisandro Dalcin /* */ " ", 25773260a9bSLisandro Dalcin /* */ " ", 25873260a9bSLisandro Dalcin /*95 */ " ", 25973260a9bSLisandro Dalcin }; 260e5c89e4eSSatish Balay 261e5c89e4eSSatish Balay #undef __FUNCT__ 262e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorMessage" 263e5c89e4eSSatish Balay /*@C 264e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 265e5c89e4eSSatish Balay 266e5c89e4eSSatish Balay Not Collective 267e5c89e4eSSatish Balay 268e5c89e4eSSatish Balay Input Parameter: 269e5c89e4eSSatish Balay . errnum - the error code 270e5c89e4eSSatish Balay 271e5c89e4eSSatish Balay Output Parameter: 2720298fd71SBarry Smith + text - the error message (NULL if not desired) 2730298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 274e5c89e4eSSatish Balay 275e5c89e4eSSatish Balay Level: developer 276e5c89e4eSSatish Balay 277e5c89e4eSSatish Balay Concepts: error handler^messages 278e5c89e4eSSatish Balay 279e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 280e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 281e5c89e4eSSatish Balay @*/ 2827087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 283e5c89e4eSSatish Balay { 284e5c89e4eSSatish Balay PetscFunctionBegin; 285a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 286a297a907SKarl Rupp else if (text) *text = 0; 287e5c89e4eSSatish Balay 288a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 289e5c89e4eSSatish Balay PetscFunctionReturn(0); 290e5c89e4eSSatish Balay } 291e5c89e4eSSatish Balay 2922c280183SJed Brown #if defined(PETSC_CLANGUAGE_CXX) 2932c280183SJed Brown /* C++ exceptions are formally not allowed to propagate through extern "C" code. In practice, far too much software 2942c280183SJed Brown * would be broken if implementations did not handle it it some common cases. However, keep in mind 2952c280183SJed Brown * 2962c280183SJed Brown * Rule 62. Don't allow exceptions to propagate across module boundaries 2972c280183SJed Brown * 2982c280183SJed Brown * in "C++ Coding Standards" by Sutter and Alexandrescu. (This accounts for part of the ongoing C++ binary interface 2992c280183SJed Brown * instability.) Having PETSc raise errors as C++ exceptions was probably misguided and should eventually be removed. 3002c280183SJed Brown */ 3012c280183SJed Brown static void PetscCxxErrorThrow() { 3022c280183SJed Brown const char *str; 3032c280183SJed Brown if (eh && eh->ctx) { 3042c280183SJed Brown std::ostringstream *msg; 3052c280183SJed Brown msg = (std::ostringstream*) eh->ctx; 3062c280183SJed Brown str = msg->str().c_str(); 3072c280183SJed Brown } else str = "Error detected in C PETSc"; 3082c280183SJed Brown 3092c280183SJed Brown throw PETSc::Exception(str); 3102c280183SJed Brown } 3112c280183SJed Brown #endif 3122c280183SJed Brown 313e5c89e4eSSatish Balay #undef __FUNCT__ 314e5c89e4eSSatish Balay #define __FUNCT__ "PetscError" 315e5c89e4eSSatish Balay /*@C 316e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 317e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 318e5c89e4eSSatish Balay 319e5c89e4eSSatish Balay Not Collective 320e5c89e4eSSatish Balay 321e5c89e4eSSatish Balay Input Parameters: 322e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 323e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 324e5c89e4eSSatish Balay . func - the function where the error occured (indicated by __FUNCT__) 325e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 326e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 327e5c89e4eSSatish Balay . n - the generic error number 328d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 329e5c89e4eSSatish Balay - mess - formatted message string - aka printf 330e5c89e4eSSatish Balay 331e5c89e4eSSatish Balay Level: intermediate 332e5c89e4eSSatish Balay 333e5c89e4eSSatish Balay Notes: 334e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 335e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 336e5c89e4eSSatish Balay sequence 337e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 338e5c89e4eSSatish Balay 339e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 340e5c89e4eSSatish Balay 3413bf036e2SBarry 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) 3423bf036e2SBarry 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 3433bf036e2SBarry Smith but this annoying. 3443bf036e2SBarry Smith 345e5c89e4eSSatish Balay Concepts: error^setting condition 346e5c89e4eSSatish Balay 347e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 348e5c89e4eSSatish Balay @*/ 349efca3c55SSatish Balay PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,...) 350e5c89e4eSSatish Balay { 351e5c89e4eSSatish Balay va_list Argp; 352c9a19010SBarry Smith size_t fullLength; 353e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 354ace3abfcSBarry Smith PetscBool ismain,isunknown; 3553bf036e2SBarry Smith PetscErrorCode ierr; 356e5c89e4eSSatish Balay 3570d577829SBarry Smith PetscFunctionBegin; 358e5c89e4eSSatish Balay if (!func) func = "User provided function"; 359e5c89e4eSSatish Balay if (!file) file = "User file"; 3600d577829SBarry Smith if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 361e5c89e4eSSatish Balay 362e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 363e5c89e4eSSatish Balay if (mess) { 364e5c89e4eSSatish Balay va_start(Argp,mess); 3652d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 366e5c89e4eSSatish Balay va_end(Argp); 367e5c89e4eSSatish Balay lbuf = buf; 368a297a907SKarl Rupp if (p == 1) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 369e5c89e4eSSatish Balay } 370e5c89e4eSSatish Balay 371efca3c55SSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,n,p,lbuf,0); 372efca3c55SSatish Balay else ierr = (*eh->handler)(comm,line,func,file,n,p,lbuf,eh->ctx); 373e5c89e4eSSatish Balay 374e5c89e4eSSatish Balay /* 375e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 376e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 377e5c89e4eSSatish Balay 378e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 379e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 380e5c89e4eSSatish Balay */ 381e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 382e5c89e4eSSatish Balay PetscStrncmp(func,"unknown",7,&isunknown); 383a297a907SKarl Rupp if (ismain || isunknown) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 384a297a907SKarl Rupp 3852c280183SJed Brown #if defined(PETSC_CLANGUAGE_CXX) 386d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 3872c280183SJed Brown PetscCxxErrorThrow(); 388d736bfebSBarry Smith } 389d736bfebSBarry Smith #endif 390e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 391e5c89e4eSSatish Balay } 392e5c89e4eSSatish Balay 393e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay #undef __FUNCT__ 396e5c89e4eSSatish Balay #define __FUNCT__ "PetscIntView" 397e5c89e4eSSatish Balay /*@C 398e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 399e5c89e4eSSatish Balay 400e5c89e4eSSatish Balay Collective on PetscViewer 401e5c89e4eSSatish Balay 402e5c89e4eSSatish Balay Input Parameters: 403e5c89e4eSSatish Balay + N - number of integers in array 404e5c89e4eSSatish Balay . idx - array of integers 405e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 406e5c89e4eSSatish Balay 407e5c89e4eSSatish Balay Level: intermediate 408e5c89e4eSSatish Balay 409300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 410300a7f5bSBarry Smith 411e5c89e4eSSatish Balay .seealso: PetscRealView() 412e5c89e4eSSatish Balay @*/ 4137087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 414e5c89e4eSSatish Balay { 415e5c89e4eSSatish Balay PetscErrorCode ierr; 416e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 417ace3abfcSBarry Smith PetscBool iascii,isbinary; 418e5c89e4eSSatish Balay MPI_Comm comm; 419e5c89e4eSSatish Balay 420e5c89e4eSSatish Balay PetscFunctionBegin; 421e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4220700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4235ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 424e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 425e5c89e4eSSatish Balay 426251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 427251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 428e5c89e4eSSatish Balay if (iascii) { 4297b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 430e5c89e4eSSatish Balay for (i=0; i<n; i++) { 431e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 432e5c89e4eSSatish Balay for (j=0; j<20; j++) { 433e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 434e5c89e4eSSatish Balay } 435e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 436e5c89e4eSSatish Balay } 437e5c89e4eSSatish Balay if (p) { 438e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 439e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 440e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 441e5c89e4eSSatish Balay } 442e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4437b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 4446805f65bSBarry Smith } else if (isbinary) { 445783b601eSJed Brown PetscMPIInt rank,size,*sizes,Ntotal,*displs,NN; 446e5c89e4eSSatish Balay PetscInt *array; 447783b601eSJed Brown 4484dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 449e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 450e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 451e5c89e4eSSatish Balay 452e5c89e4eSSatish Balay if (size > 1) { 453e5c89e4eSSatish Balay if (rank) { 454e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 455300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 456e5c89e4eSSatish Balay } else { 457785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 458adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 459e5c89e4eSSatish Balay Ntotal = sizes[0]; 460785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 461e5c89e4eSSatish Balay displs[0] = 0; 462e5c89e4eSSatish Balay for (i=1; i<size; i++) { 463e5c89e4eSSatish Balay Ntotal += sizes[i]; 464e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 465e5c89e4eSSatish Balay } 466785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 467300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4686805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 469e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 470e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 471e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 472e5c89e4eSSatish Balay } 473e5c89e4eSSatish Balay } else { 47447b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 475e5c89e4eSSatish Balay } 476e5c89e4eSSatish Balay } else { 477e5c89e4eSSatish Balay const char *tname; 478e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 479e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 480e5c89e4eSSatish Balay } 481e5c89e4eSSatish Balay PetscFunctionReturn(0); 482e5c89e4eSSatish Balay } 483e5c89e4eSSatish Balay 484e5c89e4eSSatish Balay #undef __FUNCT__ 485e5c89e4eSSatish Balay #define __FUNCT__ "PetscRealView" 486e5c89e4eSSatish Balay /*@C 487e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 488e5c89e4eSSatish Balay 489e5c89e4eSSatish Balay Collective on PetscViewer 490e5c89e4eSSatish Balay 491e5c89e4eSSatish Balay Input Parameters: 492*cba51d77SBarry Smith + N - number of PetscReal in array 493*cba51d77SBarry Smith . idx - array of PetscReal 494e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 495e5c89e4eSSatish Balay 496e5c89e4eSSatish Balay Level: intermediate 497e5c89e4eSSatish Balay 498300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 499300a7f5bSBarry Smith 500e5c89e4eSSatish Balay .seealso: PetscIntView() 501e5c89e4eSSatish Balay @*/ 5027087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 503e5c89e4eSSatish Balay { 504e5c89e4eSSatish Balay PetscErrorCode ierr; 505e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 506ace3abfcSBarry Smith PetscBool iascii,isbinary; 507e5c89e4eSSatish Balay MPI_Comm comm; 508e5c89e4eSSatish Balay 509e5c89e4eSSatish Balay PetscFunctionBegin; 510e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 5110700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 512e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 513e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 514e5c89e4eSSatish Balay 515251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 516251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 517e5c89e4eSSatish Balay if (iascii) { 5187b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 519e5c89e4eSSatish Balay for (i=0; i<n; i++) { 520e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*i);CHKERRQ(ierr); 521e5c89e4eSSatish Balay for (j=0; j<5; j++) { 522*cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*5+j]);CHKERRQ(ierr); 523e5c89e4eSSatish Balay } 524e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 525e5c89e4eSSatish Balay } 526e5c89e4eSSatish Balay if (p) { 527e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*n);CHKERRQ(ierr); 528*cba51d77SBarry Smith for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[5*n+i]);CHKERRQ(ierr);} 529e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 530e5c89e4eSSatish Balay } 531e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5327b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 5336805f65bSBarry Smith } else if (isbinary) { 534783b601eSJed Brown PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN; 535e5c89e4eSSatish Balay PetscReal *array; 536e5c89e4eSSatish Balay 5374dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 538e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 539e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 540e5c89e4eSSatish Balay 541e5c89e4eSSatish Balay if (size > 1) { 542e5c89e4eSSatish Balay if (rank) { 543e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 5441d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,0,0,0,MPIU_REAL,0,comm);CHKERRQ(ierr); 545e5c89e4eSSatish Balay } else { 546785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 547e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 548e5c89e4eSSatish Balay Ntotal = sizes[0]; 549785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 550e5c89e4eSSatish Balay displs[0] = 0; 551e5c89e4eSSatish Balay for (i=1; i<size; i++) { 552e5c89e4eSSatish Balay Ntotal += sizes[i]; 553e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 554e5c89e4eSSatish Balay } 555785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 5561d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 5576805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 558e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 559e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 560e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 561e5c89e4eSSatish Balay } 562e5c89e4eSSatish Balay } else { 56347b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 564e5c89e4eSSatish Balay } 565e5c89e4eSSatish Balay } else { 566e5c89e4eSSatish Balay const char *tname; 567e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 568e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 569e5c89e4eSSatish Balay } 570e5c89e4eSSatish Balay PetscFunctionReturn(0); 571e5c89e4eSSatish Balay } 572e5c89e4eSSatish Balay 573e5c89e4eSSatish Balay #undef __FUNCT__ 574e5c89e4eSSatish Balay #define __FUNCT__ "PetscScalarView" 575e5c89e4eSSatish Balay /*@C 576e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 577e5c89e4eSSatish Balay 578e5c89e4eSSatish Balay Collective on PetscViewer 579e5c89e4eSSatish Balay 580e5c89e4eSSatish Balay Input Parameters: 581e5c89e4eSSatish Balay + N - number of scalars in array 582e5c89e4eSSatish Balay . idx - array of scalars 583e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 584e5c89e4eSSatish Balay 585e5c89e4eSSatish Balay Level: intermediate 586e5c89e4eSSatish Balay 587300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 588300a7f5bSBarry Smith 589e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 590e5c89e4eSSatish Balay @*/ 5917087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 592e5c89e4eSSatish Balay { 593e5c89e4eSSatish Balay PetscErrorCode ierr; 594e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 595ace3abfcSBarry Smith PetscBool iascii,isbinary; 596e5c89e4eSSatish Balay MPI_Comm comm; 597e5c89e4eSSatish Balay 598e5c89e4eSSatish Balay PetscFunctionBegin; 599e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 600e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 601e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 602e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 603e5c89e4eSSatish Balay 604251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 605251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 606e5c89e4eSSatish Balay if (iascii) { 6077b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 608e5c89e4eSSatish Balay for (i=0; i<n; i++) { 609e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 610e5c89e4eSSatish Balay for (j=0; j<3; j++) { 611e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 612*cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[i*3+j]),(double)PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 613e5c89e4eSSatish Balay #else 614*cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*3+j]);CHKERRQ(ierr); 615e5c89e4eSSatish Balay #endif 616e5c89e4eSSatish Balay } 617e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 618e5c89e4eSSatish Balay } 619e5c89e4eSSatish Balay if (p) { 620e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 621e5c89e4eSSatish Balay for (i=0; i<p; i++) { 622e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 623*cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[n*3+i]),(double)PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 624e5c89e4eSSatish Balay #else 625*cba51d77SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[3*n+i]);CHKERRQ(ierr); 626e5c89e4eSSatish Balay #endif 627e5c89e4eSSatish Balay } 628e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 629e5c89e4eSSatish Balay } 630e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6317b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 6326805f65bSBarry Smith } else if (isbinary) { 633783b601eSJed Brown PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN; 634e5c89e4eSSatish Balay PetscScalar *array; 635e5c89e4eSSatish Balay 6364dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 637e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 638e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 639e5c89e4eSSatish Balay 640e5c89e4eSSatish Balay if (size > 1) { 641e5c89e4eSSatish Balay if (rank) { 642e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 643300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 644e5c89e4eSSatish Balay } else { 645785e854fSJed Brown ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 646e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 647e5c89e4eSSatish Balay Ntotal = sizes[0]; 648785e854fSJed Brown ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 649e5c89e4eSSatish Balay displs[0] = 0; 650e5c89e4eSSatish Balay for (i=1; i<size; i++) { 651e5c89e4eSSatish Balay Ntotal += sizes[i]; 652e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 653e5c89e4eSSatish Balay } 654785e854fSJed Brown ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 655300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 6566805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 657e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 658e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 659e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 660e5c89e4eSSatish Balay } 661e5c89e4eSSatish Balay } else { 66247b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 663e5c89e4eSSatish Balay } 664e5c89e4eSSatish Balay } else { 665e5c89e4eSSatish Balay const char *tname; 666e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 667e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 668e5c89e4eSSatish Balay } 669e5c89e4eSSatish Balay PetscFunctionReturn(0); 670e5c89e4eSSatish Balay } 671e5c89e4eSSatish Balay 672e5c89e4eSSatish Balay 673e5c89e4eSSatish Balay 674e5c89e4eSSatish Balay 675