17d0a6c19SBarry Smith 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay Code that allows one to set the error handlers 4e5c89e4eSSatish Balay */ 5c6db04a5SJed Brown #include <petscsys.h> /*I "petscsys.h" I*/ 6e5c89e4eSSatish Balay #include <stdarg.h> 7e5c89e4eSSatish Balay #if defined(PETSC_HAVE_STDLIB_H) 8e5c89e4eSSatish Balay #include <stdlib.h> 9e5c89e4eSSatish Balay #endif 10e5c89e4eSSatish Balay 11e5c89e4eSSatish Balay typedef struct _EH *EH; 12e5c89e4eSSatish Balay struct _EH { 13668f157eSBarry Smith PetscErrorCode (*handler)(MPI_Comm,int,const char*,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*); 14e5c89e4eSSatish Balay void *ctx; 15e5c89e4eSSatish Balay EH previous; 16e5c89e4eSSatish Balay }; 17e5c89e4eSSatish Balay 18e5c89e4eSSatish Balay static EH eh = 0; 19e5c89e4eSSatish Balay 20e5c89e4eSSatish Balay #undef __FUNCT__ 21e5c89e4eSSatish Balay #define __FUNCT__ "PetscEmacsClientErrorHandler" 22e5c89e4eSSatish Balay /*@C 23e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 24e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 25e5c89e4eSSatish Balay 26e5c89e4eSSatish Balay Not Collective 27e5c89e4eSSatish Balay 28e5c89e4eSSatish Balay Input Parameters: 29e32f2f54SBarry Smith + comm - communicator over which error occured 30e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 31e5c89e4eSSatish Balay . func - the function where error is detected (indicated by __FUNCT__) 32e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 33e5c89e4eSSatish Balay . dir - the directory of the file (indicated by __SDIR__) 34e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 35e5c89e4eSSatish Balay . n - the generic error number 36e5c89e4eSSatish Balay . p - specific error number 37e5c89e4eSSatish Balay - ctx - error handler context 38e5c89e4eSSatish Balay 39e5c89e4eSSatish Balay Options Database Key: 40e5c89e4eSSatish Balay . -on_error_emacs <machinename> 41e5c89e4eSSatish Balay 42e5c89e4eSSatish Balay Level: developer 43e5c89e4eSSatish Balay 44e5c89e4eSSatish Balay Notes: 45e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 46e5c89e4eSSatish Balay 47e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 48e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 49e5c89e4eSSatish Balay the calling sequence 50e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 51e5c89e4eSSatish Balay 52e5c89e4eSSatish Balay Notes for experienced users: 53e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 54e5c89e4eSSatish Balay 553bf036e2SBarry Smith Developer Note: Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 563bf036e2SBarry Smith 57e5c89e4eSSatish Balay Concepts: emacs^going to on error 58e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 59e5c89e4eSSatish Balay 60e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 61e5c89e4eSSatish Balay PetscAbortErrorHandler() 62e5c89e4eSSatish Balay @*/ 637087cfbeSBarry Smith PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,const char *dir,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 64e5c89e4eSSatish Balay { 65e5c89e4eSSatish Balay PetscErrorCode ierr; 66e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 67e5c89e4eSSatish Balay const char *pdir; 68e5c89e4eSSatish Balay FILE *fp; 69784ad74dSJed Brown PetscInt rval; 70e5c89e4eSSatish Balay 71e5c89e4eSSatish Balay PetscFunctionBegin; 723bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 734b89aa1eSJed Brown sprintf(command,"cd %s; emacsclient --no-wait +%d %s%s\n",pdir,line,dir,file); 74e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 753bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 763bf036e2SBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp,&rval);if (ierr) PetscFunctionReturn(ierr); 77e5c89e4eSSatish Balay #else 78e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 79e5c89e4eSSatish Balay #endif 803bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 813bf036e2SBarry Smith if (!eh) { 823bf036e2SBarry Smith ierr = PetscTraceBackErrorHandler(comm,line,fun,file,dir,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 833bf036e2SBarry Smith } else { 843bf036e2SBarry Smith ierr = (*eh->handler)(comm,line,fun,file,dir,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 853bf036e2SBarry Smith } 86e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 87e5c89e4eSSatish Balay } 88e5c89e4eSSatish Balay 89e5c89e4eSSatish Balay #undef __FUNCT__ 90e5c89e4eSSatish Balay #define __FUNCT__ "PetscPushErrorHandler" 91e5c89e4eSSatish Balay /*@C 92e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 93e5c89e4eSSatish Balay 94e5c89e4eSSatish Balay Not Collective 95e5c89e4eSSatish Balay 96e5c89e4eSSatish Balay Input Parameters: 97e5c89e4eSSatish Balay + handler - error handler routine 98e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 99e5c89e4eSSatish Balay example file pointers for error messages etc.) 100e5c89e4eSSatish Balay 101e5c89e4eSSatish Balay Calling sequence of handler: 102e32f2f54SBarry Smith $ int handler(MPI_Comm comm,int line,char *func,char *file,char *dir,PetscErrorCode n,int p,char *mess,void *ctx); 103e5c89e4eSSatish Balay 104e32f2f54SBarry Smith + comm - communicator over which error occured 105e32f2f54SBarry Smith . func - the function where the error occured (indicated by __FUNCT__) 106e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 107e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 108e5c89e4eSSatish Balay . dir - the directory of the file (indicated by __SDIR__) 109e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 110668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 111e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 112e5c89e4eSSatish Balay - ctx - the error handler context 113e5c89e4eSSatish Balay 114e5c89e4eSSatish Balay Options Database Keys: 115e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 116e5c89e4eSSatish Balay - -on_error_abort 117e5c89e4eSSatish Balay 118e5c89e4eSSatish Balay Level: intermediate 119e5c89e4eSSatish Balay 120e93bc3c1Svictor Notes: 1217850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 122e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 123e93bc3c1Svictor 1247850c7c0SBarry Smith Fortran Notes: You can only push one error handler from Fortran before poping it. 1257850c7c0SBarry Smith 12634b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 127e5c89e4eSSatish Balay 128e5c89e4eSSatish Balay @*/ 1297087cfbeSBarry Smith PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 130e5c89e4eSSatish Balay { 131e5c89e4eSSatish Balay EH neweh; 132e5c89e4eSSatish Balay PetscErrorCode ierr; 133e5c89e4eSSatish Balay 134e5c89e4eSSatish Balay PetscFunctionBegin; 135e5c89e4eSSatish Balay ierr = PetscNew(struct _EH,&neweh);CHKERRQ(ierr); 136a297a907SKarl Rupp if (eh) neweh->previous = eh; 137a297a907SKarl Rupp else neweh->previous = 0; 138e5c89e4eSSatish Balay neweh->handler = handler; 139e5c89e4eSSatish Balay neweh->ctx = ctx; 140e5c89e4eSSatish Balay eh = neweh; 141e5c89e4eSSatish Balay PetscFunctionReturn(0); 142e5c89e4eSSatish Balay } 143e5c89e4eSSatish Balay 144e5c89e4eSSatish Balay #undef __FUNCT__ 145e5c89e4eSSatish Balay #define __FUNCT__ "PetscPopErrorHandler" 146e30d2299SSatish Balay /*@ 147e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 148e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 149e5c89e4eSSatish Balay 150e5c89e4eSSatish Balay Not Collective 151e5c89e4eSSatish Balay 152e5c89e4eSSatish Balay Level: intermediate 153e5c89e4eSSatish Balay 154e5c89e4eSSatish Balay Concepts: error handler^setting 155e5c89e4eSSatish Balay 156e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 157e5c89e4eSSatish Balay @*/ 1587087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 159e5c89e4eSSatish Balay { 160e5c89e4eSSatish Balay EH tmp; 161e5c89e4eSSatish Balay PetscErrorCode ierr; 162e5c89e4eSSatish Balay 163e5c89e4eSSatish Balay PetscFunctionBegin; 164e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 165e5c89e4eSSatish Balay tmp = eh; 166e5c89e4eSSatish Balay eh = eh->previous; 167e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 168e5c89e4eSSatish Balay PetscFunctionReturn(0); 169e5c89e4eSSatish Balay } 170e5c89e4eSSatish Balay 171e93bc3c1Svictor #undef __FUNCT__ 172e93bc3c1Svictor #define __FUNCT__ "PetscReturnErrorHandler" 173e93bc3c1Svictor /*@C 174e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 175e93bc3c1Svictor level. 176e93bc3c1Svictor 177e93bc3c1Svictor Not Collective 178e93bc3c1Svictor 179e93bc3c1Svictor Input Parameters: 180e32f2f54SBarry Smith + comm - communicator over which error occurred 181e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 182e93bc3c1Svictor . func - the function where error is detected (indicated by __FUNCT__) 183e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 184e93bc3c1Svictor . dir - the directory of the file (indicated by __SDIR__) 185e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 186e93bc3c1Svictor . n - the generic error number 187e93bc3c1Svictor . p - specific error number 188e93bc3c1Svictor - ctx - error handler context 189e93bc3c1Svictor 190e93bc3c1Svictor Level: developer 191e93bc3c1Svictor 192e93bc3c1Svictor Notes: 193e93bc3c1Svictor Most users need not directly employ this routine and the other error 194e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 195e93bc3c1Svictor the calling sequence 196e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 197e93bc3c1Svictor 198e93bc3c1Svictor Notes for experienced users: 199e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 200e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 201e93bc3c1Svictor and similar errors. 202e93bc3c1Svictor 203e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 204e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 205e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 206e93bc3c1Svictor 207e93bc3c1Svictor Concepts: error handler 208e93bc3c1Svictor 209e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 210e93bc3c1Svictor @*/ 211e93bc3c1Svictor 2127087cfbeSBarry Smith PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,const char *dir,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 213e93bc3c1Svictor { 214e93bc3c1Svictor PetscFunctionBegin; 215e93bc3c1Svictor PetscFunctionReturn(n); 216e93bc3c1Svictor } 217e93bc3c1Svictor 218e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 219e5c89e4eSSatish Balay /* 220e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 221e5c89e4eSSatish Balay there must also be made here 222e5c89e4eSSatish Balay */ 223e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 224e5c89e4eSSatish Balay /*55 */ "Out of memory", 225e5c89e4eSSatish Balay "No support for this operation for this object type", 226e5c89e4eSSatish Balay "No support for this operation on this system", 227e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 228e5c89e4eSSatish Balay /*59 */ "Signal received", 229e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 230e5c89e4eSSatish Balay "Argument aliasing not permitted", 231e5c89e4eSSatish Balay "Invalid argument", 232e5c89e4eSSatish Balay /*63 */ "Argument out of range", 233f08646a8SSatish Balay "Corrupt argument:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 234e5c89e4eSSatish Balay "Unable to open file", 235e5c89e4eSSatish Balay "Read from file failed", 236e5c89e4eSSatish Balay "Write to file failed", 237e5c89e4eSSatish Balay "Invalid pointer", 238e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 239a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 240f08646a8SSatish Balay /*71 */ "Detected zero pivot in LU factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 241e5c89e4eSSatish Balay /*72 */ "Floating point exception", 242e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 243e5c89e4eSSatish Balay "Corrupted Petsc object", 244e5c89e4eSSatish Balay "Arguments are incompatible", 245e5c89e4eSSatish Balay "Error in external library", 246e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 247e5c89e4eSSatish Balay "Memory corruption", 248e5c89e4eSSatish Balay "Unexpected data in file", 249e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 250f08646a8SSatish Balay /*81 */ "Detected zero pivot in Cholesky factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 251e5c89e4eSSatish Balay " ", 252e5c89e4eSSatish Balay " ", 253f08646a8SSatish Balay "Overflow in integer operation:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 254e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 255f08646a8SSatish 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", 2568cda6cd7SBarry Smith /*87 */ "Not used", 2578cda6cd7SBarry Smith /*88 */ "Error in system call", 258f08646a8SSatish Balay /*89 */ "Object Type not set:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset" 25973260a9bSLisandro Dalcin /*90 */ " ", 26073260a9bSLisandro Dalcin /* */ " ", 26173260a9bSLisandro Dalcin /* */ " ", 26273260a9bSLisandro Dalcin /* */ " ", 26373260a9bSLisandro Dalcin /* */ " ", 26473260a9bSLisandro Dalcin /*95 */ " ", 26573260a9bSLisandro Dalcin }; 266e5c89e4eSSatish Balay 267e5c89e4eSSatish Balay #undef __FUNCT__ 268e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorMessage" 269e5c89e4eSSatish Balay /*@C 270e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 271e5c89e4eSSatish Balay 272e5c89e4eSSatish Balay Not Collective 273e5c89e4eSSatish Balay 274e5c89e4eSSatish Balay Input Parameter: 275e5c89e4eSSatish Balay . errnum - the error code 276e5c89e4eSSatish Balay 277e5c89e4eSSatish Balay Output Parameter: 278*0298fd71SBarry Smith + text - the error message (NULL if not desired) 279*0298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 280e5c89e4eSSatish Balay 281e5c89e4eSSatish Balay Level: developer 282e5c89e4eSSatish Balay 283e5c89e4eSSatish Balay Concepts: error handler^messages 284e5c89e4eSSatish Balay 285e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 286e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 287e5c89e4eSSatish Balay @*/ 2887087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 289e5c89e4eSSatish Balay { 290e5c89e4eSSatish Balay PetscFunctionBegin; 291a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 292a297a907SKarl Rupp else if (text) *text = 0; 293e5c89e4eSSatish Balay 294a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 295e5c89e4eSSatish Balay PetscFunctionReturn(0); 296e5c89e4eSSatish Balay } 297e5c89e4eSSatish Balay 298e5c89e4eSSatish Balay #undef __FUNCT__ 299e5c89e4eSSatish Balay #define __FUNCT__ "PetscError" 300e5c89e4eSSatish Balay /*@C 301e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 302e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 303e5c89e4eSSatish Balay 304e5c89e4eSSatish Balay Not Collective 305e5c89e4eSSatish Balay 306e5c89e4eSSatish Balay Input Parameters: 307e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 308e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 309e5c89e4eSSatish Balay . func - the function where the error occured (indicated by __FUNCT__) 310e5c89e4eSSatish Balay . dir - the directory of file (indicated by __SDIR__) 311e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 312e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 313e5c89e4eSSatish Balay . n - the generic error number 314d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 315e5c89e4eSSatish Balay - mess - formatted message string - aka printf 316e5c89e4eSSatish Balay 317e5c89e4eSSatish Balay Level: intermediate 318e5c89e4eSSatish Balay 319e5c89e4eSSatish Balay Notes: 320e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 321e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 322e5c89e4eSSatish Balay sequence 323e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 324e5c89e4eSSatish Balay 325e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 326e5c89e4eSSatish Balay 3273bf036e2SBarry 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) 3283bf036e2SBarry 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 3293bf036e2SBarry Smith but this annoying. 3303bf036e2SBarry Smith 331e5c89e4eSSatish Balay Concepts: error^setting condition 332e5c89e4eSSatish Balay 333e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 334e5c89e4eSSatish Balay @*/ 3357087cfbeSBarry Smith PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,const char *dir,PetscErrorCode n,PetscErrorType p,const char *mess,...) 336e5c89e4eSSatish Balay { 337e5c89e4eSSatish Balay va_list Argp; 338c9a19010SBarry Smith size_t fullLength; 339e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 340ace3abfcSBarry Smith PetscBool ismain,isunknown; 3413bf036e2SBarry Smith PetscErrorCode ierr; 342e5c89e4eSSatish Balay 343e5c89e4eSSatish Balay if (!func) func = "User provided function"; 344e5c89e4eSSatish Balay if (!file) file = "User file"; 345e5c89e4eSSatish Balay if (!dir) dir = " "; 346e5c89e4eSSatish Balay 347e5c89e4eSSatish Balay PetscFunctionBegin; 348e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 349e5c89e4eSSatish Balay if (mess) { 350e5c89e4eSSatish Balay va_start(Argp,mess); 3512d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 352e5c89e4eSSatish Balay va_end(Argp); 353e5c89e4eSSatish Balay lbuf = buf; 354a297a907SKarl Rupp if (p == 1) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 355e5c89e4eSSatish Balay } 356e5c89e4eSSatish Balay 357e32f2f54SBarry Smith if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,dir,n,p,lbuf,0); 358e32f2f54SBarry Smith else ierr = (*eh->handler)(comm,line,func,file,dir,n,p,lbuf,eh->ctx); 359e5c89e4eSSatish Balay 360e5c89e4eSSatish Balay /* 361e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 362e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 363e5c89e4eSSatish Balay 364e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 365e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 366e5c89e4eSSatish Balay */ 367e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 368e5c89e4eSSatish Balay PetscStrncmp(func,"unknown",7,&isunknown); 369a297a907SKarl Rupp if (ismain || isunknown) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 370a297a907SKarl Rupp 371d736bfebSBarry Smith #if defined(PETSC_CLANGUAGE_CXX) && !defined(PETSC_USE_EXTERN_CXX) 372d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 373d736bfebSBarry Smith const char *str; 3748fbb6cefSJed Brown if (eh && eh->ctx) { 375d736bfebSBarry Smith std::ostringstream *msg; 376d736bfebSBarry Smith msg = (std::ostringstream*) eh->ctx; 377d736bfebSBarry Smith str = msg->str().c_str(); 378a297a907SKarl Rupp } else str = "Error detected in C PETSc"; 379a297a907SKarl Rupp 380d736bfebSBarry Smith throw PETSc::Exception(str); 381d736bfebSBarry Smith } 382d736bfebSBarry Smith #endif 383e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 384e5c89e4eSSatish Balay } 385e5c89e4eSSatish Balay 386e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 387e5c89e4eSSatish Balay 388e5c89e4eSSatish Balay #undef __FUNCT__ 389e5c89e4eSSatish Balay #define __FUNCT__ "PetscIntView" 390e5c89e4eSSatish Balay /*@C 391e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 392e5c89e4eSSatish Balay 393e5c89e4eSSatish Balay Collective on PetscViewer 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay Input Parameters: 396e5c89e4eSSatish Balay + N - number of integers in array 397e5c89e4eSSatish Balay . idx - array of integers 398e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 399e5c89e4eSSatish Balay 400e5c89e4eSSatish Balay Level: intermediate 401e5c89e4eSSatish Balay 402300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 403300a7f5bSBarry Smith 404e5c89e4eSSatish Balay .seealso: PetscRealView() 405e5c89e4eSSatish Balay @*/ 4067087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 407e5c89e4eSSatish Balay { 408e5c89e4eSSatish Balay PetscErrorCode ierr; 409e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 410ace3abfcSBarry Smith PetscBool iascii,isbinary; 411e5c89e4eSSatish Balay MPI_Comm comm; 412e5c89e4eSSatish Balay 413e5c89e4eSSatish Balay PetscFunctionBegin; 414e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4150700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4165ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 417e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 418e5c89e4eSSatish Balay 419251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 420251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 421e5c89e4eSSatish Balay if (iascii) { 4227b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 423e5c89e4eSSatish Balay for (i=0; i<n; i++) { 424e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 425e5c89e4eSSatish Balay for (j=0; j<20; j++) { 426e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 427e5c89e4eSSatish Balay } 428e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 429e5c89e4eSSatish Balay } 430e5c89e4eSSatish Balay if (p) { 431e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 432e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 433e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 434e5c89e4eSSatish Balay } 435e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4367b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 4376805f65bSBarry Smith } else if (isbinary) { 438783b601eSJed Brown PetscMPIInt rank,size,*sizes,Ntotal,*displs,NN; 439e5c89e4eSSatish Balay PetscInt *array; 440783b601eSJed Brown 4414dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 442e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 443e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 444e5c89e4eSSatish Balay 445e5c89e4eSSatish Balay if (size > 1) { 446e5c89e4eSSatish Balay if (rank) { 447e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 448300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 449e5c89e4eSSatish Balay } else { 450e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 451adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 452e5c89e4eSSatish Balay Ntotal = sizes[0]; 453e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 454e5c89e4eSSatish Balay displs[0] = 0; 455e5c89e4eSSatish Balay for (i=1; i<size; i++) { 456e5c89e4eSSatish Balay Ntotal += sizes[i]; 457e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 458e5c89e4eSSatish Balay } 459e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&array);CHKERRQ(ierr); 460300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4616805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 462e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 463e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 464e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 465e5c89e4eSSatish Balay } 466e5c89e4eSSatish Balay } else { 46747b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 468e5c89e4eSSatish Balay } 469e5c89e4eSSatish Balay } else { 470e5c89e4eSSatish Balay const char *tname; 471e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 472e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 473e5c89e4eSSatish Balay } 474e5c89e4eSSatish Balay PetscFunctionReturn(0); 475e5c89e4eSSatish Balay } 476e5c89e4eSSatish Balay 477e5c89e4eSSatish Balay #undef __FUNCT__ 478e5c89e4eSSatish Balay #define __FUNCT__ "PetscRealView" 479e5c89e4eSSatish Balay /*@C 480e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay Collective on PetscViewer 483e5c89e4eSSatish Balay 484e5c89e4eSSatish Balay Input Parameters: 485e5c89e4eSSatish Balay + N - number of doubles in array 486e5c89e4eSSatish Balay . idx - array of doubles 487e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 488e5c89e4eSSatish Balay 489e5c89e4eSSatish Balay Level: intermediate 490e5c89e4eSSatish Balay 491300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 492300a7f5bSBarry Smith 493e5c89e4eSSatish Balay .seealso: PetscIntView() 494e5c89e4eSSatish Balay @*/ 4957087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 496e5c89e4eSSatish Balay { 497e5c89e4eSSatish Balay PetscErrorCode ierr; 498e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 499ace3abfcSBarry Smith PetscBool iascii,isbinary; 500e5c89e4eSSatish Balay MPI_Comm comm; 501e5c89e4eSSatish Balay 502e5c89e4eSSatish Balay PetscFunctionBegin; 503e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 5040700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 505e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 506e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 507e5c89e4eSSatish Balay 508251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 509251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 510e5c89e4eSSatish Balay if (iascii) { 5117b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 512e5c89e4eSSatish Balay for (i=0; i<n; i++) { 513e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*i);CHKERRQ(ierr); 514e5c89e4eSSatish Balay for (j=0; j<5; j++) { 515e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*5+j]);CHKERRQ(ierr); 516e5c89e4eSSatish Balay } 517e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 518e5c89e4eSSatish Balay } 519e5c89e4eSSatish Balay if (p) { 520e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*n);CHKERRQ(ierr); 521e5c89e4eSSatish Balay for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[5*n+i]);CHKERRQ(ierr);} 522e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 523e5c89e4eSSatish Balay } 524e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5257b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 5266805f65bSBarry Smith } else if (isbinary) { 527783b601eSJed Brown PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN; 528e5c89e4eSSatish Balay PetscReal *array; 529e5c89e4eSSatish Balay 5304dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 531e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 532e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 533e5c89e4eSSatish Balay 534e5c89e4eSSatish Balay if (size > 1) { 535e5c89e4eSSatish Balay if (rank) { 536e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 537300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPI_DOUBLE,0,0,0,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 538e5c89e4eSSatish Balay } else { 539e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 540e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 541e5c89e4eSSatish Balay Ntotal = sizes[0]; 542e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 543e5c89e4eSSatish Balay displs[0] = 0; 544e5c89e4eSSatish Balay for (i=1; i<size; i++) { 545e5c89e4eSSatish Balay Ntotal += sizes[i]; 546e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 547e5c89e4eSSatish Balay } 548e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 549300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPI_DOUBLE,array,sizes,displs,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 5506805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 551e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 552e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 553e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 554e5c89e4eSSatish Balay } 555e5c89e4eSSatish Balay } else { 55647b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 557e5c89e4eSSatish Balay } 558e5c89e4eSSatish Balay } else { 559e5c89e4eSSatish Balay const char *tname; 560e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 561e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 562e5c89e4eSSatish Balay } 563e5c89e4eSSatish Balay PetscFunctionReturn(0); 564e5c89e4eSSatish Balay } 565e5c89e4eSSatish Balay 566e5c89e4eSSatish Balay #undef __FUNCT__ 567e5c89e4eSSatish Balay #define __FUNCT__ "PetscScalarView" 568e5c89e4eSSatish Balay /*@C 569e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 570e5c89e4eSSatish Balay 571e5c89e4eSSatish Balay Collective on PetscViewer 572e5c89e4eSSatish Balay 573e5c89e4eSSatish Balay Input Parameters: 574e5c89e4eSSatish Balay + N - number of scalars in array 575e5c89e4eSSatish Balay . idx - array of scalars 576e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 577e5c89e4eSSatish Balay 578e5c89e4eSSatish Balay Level: intermediate 579e5c89e4eSSatish Balay 580300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 581300a7f5bSBarry Smith 582e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 583e5c89e4eSSatish Balay @*/ 5847087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 585e5c89e4eSSatish Balay { 586e5c89e4eSSatish Balay PetscErrorCode ierr; 587e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 588ace3abfcSBarry Smith PetscBool iascii,isbinary; 589e5c89e4eSSatish Balay MPI_Comm comm; 590e5c89e4eSSatish Balay 591e5c89e4eSSatish Balay PetscFunctionBegin; 592e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 593e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 594e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 595e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 596e5c89e4eSSatish Balay 597251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 598251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 599e5c89e4eSSatish Balay if (iascii) { 6007b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 601e5c89e4eSSatish Balay for (i=0; i<n; i++) { 602e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 603e5c89e4eSSatish Balay for (j=0; j<3; j++) { 604e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 605b0043f70SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 606e5c89e4eSSatish Balay #else 607e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*3+j]);CHKERRQ(ierr); 608e5c89e4eSSatish Balay #endif 609e5c89e4eSSatish Balay } 610e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 611e5c89e4eSSatish Balay } 612e5c89e4eSSatish Balay if (p) { 613e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 614e5c89e4eSSatish Balay for (i=0; i<p; i++) { 615e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 616b0043f70SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 617e5c89e4eSSatish Balay #else 618e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 619e5c89e4eSSatish Balay #endif 620e5c89e4eSSatish Balay } 621e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 622e5c89e4eSSatish Balay } 623e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6247b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 6256805f65bSBarry Smith } else if (isbinary) { 626783b601eSJed Brown PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN; 627e5c89e4eSSatish Balay PetscScalar *array; 628e5c89e4eSSatish Balay 6294dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 630e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 631e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 632e5c89e4eSSatish Balay 633e5c89e4eSSatish Balay if (size > 1) { 634e5c89e4eSSatish Balay if (rank) { 635e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 636300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 637e5c89e4eSSatish Balay } else { 638e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 639e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 640e5c89e4eSSatish Balay Ntotal = sizes[0]; 641e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 642e5c89e4eSSatish Balay displs[0] = 0; 643e5c89e4eSSatish Balay for (i=1; i<size; i++) { 644e5c89e4eSSatish Balay Ntotal += sizes[i]; 645e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 646e5c89e4eSSatish Balay } 647e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&array);CHKERRQ(ierr); 648300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 6496805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 650e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 651e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 652e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 653e5c89e4eSSatish Balay } 654e5c89e4eSSatish Balay } else { 65547b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 656e5c89e4eSSatish Balay } 657e5c89e4eSSatish Balay } else { 658e5c89e4eSSatish Balay const char *tname; 659e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 660e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 661e5c89e4eSSatish Balay } 662e5c89e4eSSatish Balay PetscFunctionReturn(0); 663e5c89e4eSSatish Balay } 664e5c89e4eSSatish Balay 665e5c89e4eSSatish Balay 666e5c89e4eSSatish Balay 667e5c89e4eSSatish Balay 668