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 { 130700a824SBarry Smith int classid; 14668f157eSBarry Smith PetscErrorCode (*handler)(MPI_Comm,int,const char*,const char*,const char *,PetscErrorCode,PetscErrorType,const char*,void *); 15e5c89e4eSSatish Balay void *ctx; 16e5c89e4eSSatish Balay EH previous; 17e5c89e4eSSatish Balay }; 18e5c89e4eSSatish Balay 19e5c89e4eSSatish Balay static EH eh = 0; 20e5c89e4eSSatish Balay 21e5c89e4eSSatish Balay #undef __FUNCT__ 22e5c89e4eSSatish Balay #define __FUNCT__ "PetscEmacsClientErrorHandler" 23e5c89e4eSSatish Balay /*@C 24e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 25e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 26e5c89e4eSSatish Balay 27e5c89e4eSSatish Balay Not Collective 28e5c89e4eSSatish Balay 29e5c89e4eSSatish Balay Input Parameters: 30e32f2f54SBarry Smith + comm - communicator over which error occured 31e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 32e5c89e4eSSatish Balay . func - the function where error is detected (indicated by __FUNCT__) 33e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 34e5c89e4eSSatish Balay . dir - the directory of the file (indicated by __SDIR__) 35e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 36e5c89e4eSSatish Balay . n - the generic error number 37e5c89e4eSSatish Balay . p - specific error number 38e5c89e4eSSatish Balay - ctx - error handler context 39e5c89e4eSSatish Balay 40e5c89e4eSSatish Balay Options Database Key: 41e5c89e4eSSatish Balay . -on_error_emacs <machinename> 42e5c89e4eSSatish Balay 43e5c89e4eSSatish Balay Level: developer 44e5c89e4eSSatish Balay 45e5c89e4eSSatish Balay Notes: 46e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 47e5c89e4eSSatish Balay 48e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 49e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 50e5c89e4eSSatish Balay the calling sequence 51e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 52e5c89e4eSSatish Balay 53e5c89e4eSSatish Balay Notes for experienced users: 54e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 55e5c89e4eSSatish Balay 56*3bf036e2SBarry Smith Developer Note: Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 57*3bf036e2SBarry Smith 58e5c89e4eSSatish Balay Concepts: emacs^going to on error 59e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 60e5c89e4eSSatish Balay 61e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 62e5c89e4eSSatish Balay PetscAbortErrorHandler() 63e5c89e4eSSatish Balay @*/ 647087cfbeSBarry 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) 65e5c89e4eSSatish Balay { 66e5c89e4eSSatish Balay PetscErrorCode ierr; 67e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 68e5c89e4eSSatish Balay const char *pdir; 69e5c89e4eSSatish Balay FILE *fp; 70784ad74dSJed Brown PetscInt rval; 71e5c89e4eSSatish Balay 72e5c89e4eSSatish Balay PetscFunctionBegin; 73*3bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 744b89aa1eSJed Brown sprintf(command,"cd %s; emacsclient --no-wait +%d %s%s\n",pdir,line,dir,file); 75e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 76*3bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 77*3bf036e2SBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp,&rval);if (ierr) PetscFunctionReturn(ierr); 78e5c89e4eSSatish Balay #else 79e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 80e5c89e4eSSatish Balay #endif 81*3bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 82*3bf036e2SBarry Smith if (!eh) { 83*3bf036e2SBarry Smith ierr = PetscTraceBackErrorHandler(comm,line,fun,file,dir,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 84*3bf036e2SBarry Smith } else { 85*3bf036e2SBarry Smith ierr = (*eh->handler)(comm,line,fun,file,dir,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 86*3bf036e2SBarry Smith } 87e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 88e5c89e4eSSatish Balay } 89e5c89e4eSSatish Balay 90e5c89e4eSSatish Balay #undef __FUNCT__ 91e5c89e4eSSatish Balay #define __FUNCT__ "PetscPushErrorHandler" 92e5c89e4eSSatish Balay /*@C 93e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 94e5c89e4eSSatish Balay 95e5c89e4eSSatish Balay Not Collective 96e5c89e4eSSatish Balay 97e5c89e4eSSatish Balay Input Parameters: 98e5c89e4eSSatish Balay + handler - error handler routine 99e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 100e5c89e4eSSatish Balay example file pointers for error messages etc.) 101e5c89e4eSSatish Balay 102e5c89e4eSSatish Balay Calling sequence of handler: 103e32f2f54SBarry Smith $ int handler(MPI_Comm comm,int line,char *func,char *file,char *dir,PetscErrorCode n,int p,char *mess,void *ctx); 104e5c89e4eSSatish Balay 105e32f2f54SBarry Smith + comm - communicator over which error occured 106e32f2f54SBarry Smith . func - the function where the error occured (indicated by __FUNCT__) 107e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 108e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 109e5c89e4eSSatish Balay . dir - the directory of the file (indicated by __SDIR__) 110e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 111668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 112e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 113e5c89e4eSSatish Balay - ctx - the error handler context 114e5c89e4eSSatish Balay 115e5c89e4eSSatish Balay Options Database Keys: 116e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 117e5c89e4eSSatish Balay - -on_error_abort 118e5c89e4eSSatish Balay 119e5c89e4eSSatish Balay Level: intermediate 120e5c89e4eSSatish Balay 121e93bc3c1Svictor Notes: 1227850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 123e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 124e93bc3c1Svictor 1257850c7c0SBarry Smith Fortran Notes: You can only push one error handler from Fortran before poping it. 1267850c7c0SBarry Smith 12734b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 128e5c89e4eSSatish Balay 129e5c89e4eSSatish Balay @*/ 1307087cfbeSBarry Smith PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char *,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 131e5c89e4eSSatish Balay { 132e5c89e4eSSatish Balay EH neweh; 133e5c89e4eSSatish Balay PetscErrorCode ierr; 134e5c89e4eSSatish Balay 135e5c89e4eSSatish Balay PetscFunctionBegin; 136e5c89e4eSSatish Balay ierr = PetscNew(struct _EH,&neweh);CHKERRQ(ierr); 137e5c89e4eSSatish Balay if (eh) {neweh->previous = eh;} 138e5c89e4eSSatish Balay else {neweh->previous = 0;} 139e5c89e4eSSatish Balay neweh->handler = handler; 140e5c89e4eSSatish Balay neweh->ctx = ctx; 141e5c89e4eSSatish Balay eh = neweh; 142e5c89e4eSSatish Balay PetscFunctionReturn(0); 143e5c89e4eSSatish Balay } 144e5c89e4eSSatish Balay 145e5c89e4eSSatish Balay #undef __FUNCT__ 146e5c89e4eSSatish Balay #define __FUNCT__ "PetscPopErrorHandler" 147e30d2299SSatish Balay /*@ 148e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 149e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 150e5c89e4eSSatish Balay 151e5c89e4eSSatish Balay Not Collective 152e5c89e4eSSatish Balay 153e5c89e4eSSatish Balay Level: intermediate 154e5c89e4eSSatish Balay 155e5c89e4eSSatish Balay Concepts: error handler^setting 156e5c89e4eSSatish Balay 157e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 158e5c89e4eSSatish Balay @*/ 1597087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 160e5c89e4eSSatish Balay { 161e5c89e4eSSatish Balay EH tmp; 162e5c89e4eSSatish Balay PetscErrorCode ierr; 163e5c89e4eSSatish Balay 164e5c89e4eSSatish Balay PetscFunctionBegin; 165e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 166e5c89e4eSSatish Balay tmp = eh; 167e5c89e4eSSatish Balay eh = eh->previous; 168e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 169e5c89e4eSSatish Balay 170e5c89e4eSSatish Balay PetscFunctionReturn(0); 171e5c89e4eSSatish Balay } 172e5c89e4eSSatish Balay 173e93bc3c1Svictor #undef __FUNCT__ 174e93bc3c1Svictor #define __FUNCT__ "PetscReturnErrorHandler" 175e93bc3c1Svictor /*@C 176e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 177e93bc3c1Svictor level. 178e93bc3c1Svictor 179e93bc3c1Svictor Not Collective 180e93bc3c1Svictor 181e93bc3c1Svictor Input Parameters: 182e32f2f54SBarry Smith + comm - communicator over which error occurred 183e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 184e93bc3c1Svictor . func - the function where error is detected (indicated by __FUNCT__) 185e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 186e93bc3c1Svictor . dir - the directory of the file (indicated by __SDIR__) 187e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 188e93bc3c1Svictor . n - the generic error number 189e93bc3c1Svictor . p - specific error number 190e93bc3c1Svictor - ctx - error handler context 191e93bc3c1Svictor 192e93bc3c1Svictor Level: developer 193e93bc3c1Svictor 194e93bc3c1Svictor Notes: 195e93bc3c1Svictor Most users need not directly employ this routine and the other error 196e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 197e93bc3c1Svictor the calling sequence 198e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 199e93bc3c1Svictor 200e93bc3c1Svictor Notes for experienced users: 201e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 202e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 203e93bc3c1Svictor and similar errors. 204e93bc3c1Svictor 205e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 206e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 207e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 208e93bc3c1Svictor 209e93bc3c1Svictor Concepts: error handler 210e93bc3c1Svictor 211e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 212e93bc3c1Svictor @*/ 213e93bc3c1Svictor 2147087cfbeSBarry 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) 215e93bc3c1Svictor { 216e93bc3c1Svictor PetscFunctionBegin; 217e93bc3c1Svictor PetscFunctionReturn(n); 218e93bc3c1Svictor } 219e93bc3c1Svictor 220e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 221e5c89e4eSSatish Balay /* 222e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 223e5c89e4eSSatish Balay there must also be made here 224e5c89e4eSSatish Balay */ 225e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 226e5c89e4eSSatish Balay /*55 */ "Out of memory", 227e5c89e4eSSatish Balay "No support for this operation for this object type", 228e5c89e4eSSatish Balay "No support for this operation on this system", 229e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 230e5c89e4eSSatish Balay /*59 */ "Signal received", 231e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 232e5c89e4eSSatish Balay "Argument aliasing not permitted", 233e5c89e4eSSatish Balay "Invalid argument", 234e5c89e4eSSatish Balay /*63 */ "Argument out of range", 235f08646a8SSatish Balay "Corrupt argument:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 236e5c89e4eSSatish Balay "Unable to open file", 237e5c89e4eSSatish Balay "Read from file failed", 238e5c89e4eSSatish Balay "Write to file failed", 239e5c89e4eSSatish Balay "Invalid pointer", 240e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 241a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 242f08646a8SSatish Balay /*71 */ "Detected zero pivot in LU factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 243e5c89e4eSSatish Balay /*72 */ "Floating point exception", 244e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 245e5c89e4eSSatish Balay "Corrupted Petsc object", 246e5c89e4eSSatish Balay "Arguments are incompatible", 247e5c89e4eSSatish Balay "Error in external library", 248e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 249e5c89e4eSSatish Balay "Memory corruption", 250e5c89e4eSSatish Balay "Unexpected data in file", 251e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 252f08646a8SSatish Balay /*81 */ "Detected zero pivot in Cholesky factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 253e5c89e4eSSatish Balay " ", 254e5c89e4eSSatish Balay " ", 255f08646a8SSatish Balay "Overflow in integer operation:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 256e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 257f08646a8SSatish 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", 2588cda6cd7SBarry Smith /*87 */ "Not used", 2598cda6cd7SBarry Smith /*88 */ "Error in system call", 260f08646a8SSatish Balay /*89 */ "Object Type not set:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset" 26173260a9bSLisandro Dalcin /*90 */ " ", 26273260a9bSLisandro Dalcin /* */ " ", 26373260a9bSLisandro Dalcin /* */ " ", 26473260a9bSLisandro Dalcin /* */ " ", 26573260a9bSLisandro Dalcin /* */ " ", 26673260a9bSLisandro Dalcin /*95 */ " ", 26773260a9bSLisandro Dalcin }; 268e5c89e4eSSatish Balay 269e5c89e4eSSatish Balay #undef __FUNCT__ 270e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorMessage" 271e5c89e4eSSatish Balay /*@C 272e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 273e5c89e4eSSatish Balay 274e5c89e4eSSatish Balay Not Collective 275e5c89e4eSSatish Balay 276e5c89e4eSSatish Balay Input Parameter: 277e5c89e4eSSatish Balay . errnum - the error code 278e5c89e4eSSatish Balay 279e5c89e4eSSatish Balay Output Parameter: 280e5c89e4eSSatish Balay + text - the error message (PETSC_NULL if not desired) 281e5c89e4eSSatish Balay - specific - the specific error message that was set with SETERRxxx() or PetscError(). (PETSC_NULL if not desired) 282e5c89e4eSSatish Balay 283e5c89e4eSSatish Balay Level: developer 284e5c89e4eSSatish Balay 285e5c89e4eSSatish Balay Concepts: error handler^messages 286e5c89e4eSSatish Balay 287e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 288e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 289e5c89e4eSSatish Balay @*/ 2907087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 291e5c89e4eSSatish Balay { 292e5c89e4eSSatish Balay PetscFunctionBegin; 2932a6744ebSBarry Smith if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) { 294c2f951e6SSatish Balay *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 295e5c89e4eSSatish Balay } else if (text) *text = 0; 296e5c89e4eSSatish Balay 297e5c89e4eSSatish Balay if (specific) { 298e5c89e4eSSatish Balay *specific = PetscErrorBaseMessage; 299e5c89e4eSSatish Balay } 300e5c89e4eSSatish Balay PetscFunctionReturn(0); 301e5c89e4eSSatish Balay } 302e5c89e4eSSatish Balay 303e5c89e4eSSatish Balay #undef __FUNCT__ 304e5c89e4eSSatish Balay #define __FUNCT__ "PetscError" 305e5c89e4eSSatish Balay /*@C 306e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 307e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 308e5c89e4eSSatish Balay 309e5c89e4eSSatish Balay Not Collective 310e5c89e4eSSatish Balay 311e5c89e4eSSatish Balay Input Parameters: 312e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 313e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 314e5c89e4eSSatish Balay . func - the function where the error occured (indicated by __FUNCT__) 315e5c89e4eSSatish Balay . dir - the directory of file (indicated by __SDIR__) 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 332*3bf036e2SBarry 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) 333*3bf036e2SBarry 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 334*3bf036e2SBarry Smith but this annoying. 335*3bf036e2SBarry Smith 336e5c89e4eSSatish Balay Concepts: error^setting condition 337e5c89e4eSSatish Balay 338e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 339e5c89e4eSSatish Balay @*/ 3407087cfbeSBarry Smith PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char* file,const char *dir,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; 345ace3abfcSBarry Smith PetscBool ismain,isunknown; 346*3bf036e2SBarry Smith PetscErrorCode ierr; 347e5c89e4eSSatish Balay 348e5c89e4eSSatish Balay if (!func) func = "User provided function"; 349e5c89e4eSSatish Balay if (!file) file = "User file"; 350e5c89e4eSSatish Balay if (!dir) dir = " "; 351e5c89e4eSSatish Balay 352e5c89e4eSSatish Balay PetscFunctionBegin; 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; 359e5c89e4eSSatish Balay if (p == 1) { 360e5c89e4eSSatish Balay PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 361e5c89e4eSSatish Balay } 362e5c89e4eSSatish Balay } 363e5c89e4eSSatish Balay 364e32f2f54SBarry Smith if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,dir,n,p,lbuf,0); 365e32f2f54SBarry Smith else ierr = (*eh->handler)(comm,line,func,file,dir,n,p,lbuf,eh->ctx); 366e5c89e4eSSatish Balay 367e5c89e4eSSatish Balay /* 368e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 369e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 370e5c89e4eSSatish Balay 371e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 372e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 373e5c89e4eSSatish Balay */ 374e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 375e5c89e4eSSatish Balay PetscStrncmp(func,"unknown",7,&isunknown); 376e5c89e4eSSatish Balay if (ismain || isunknown) { 377e5c89e4eSSatish Balay MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 378e5c89e4eSSatish Balay } 379d736bfebSBarry Smith #if defined(PETSC_CLANGUAGE_CXX) && !defined(PETSC_USE_EXTERN_CXX) 380d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 381d736bfebSBarry Smith const char *str; 3828fbb6cefSJed Brown if (eh && eh->ctx) { 383d736bfebSBarry Smith std::ostringstream *msg; 384d736bfebSBarry Smith msg = (std::ostringstream*) eh->ctx; 385d736bfebSBarry Smith str = msg->str().c_str(); 386d736bfebSBarry Smith } else { 387d736bfebSBarry Smith str = "Error detected in C PETSc"; 388d736bfebSBarry Smith } 389d736bfebSBarry Smith throw PETSc::Exception(str); 390d736bfebSBarry Smith } 391d736bfebSBarry Smith #endif 392e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 393e5c89e4eSSatish Balay } 394e5c89e4eSSatish Balay 395e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 396e5c89e4eSSatish Balay 397e5c89e4eSSatish Balay #undef __FUNCT__ 398e5c89e4eSSatish Balay #define __FUNCT__ "PetscIntView" 399e5c89e4eSSatish Balay /*@C 400e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 401e5c89e4eSSatish Balay 402e5c89e4eSSatish Balay Collective on PetscViewer 403e5c89e4eSSatish Balay 404e5c89e4eSSatish Balay Input Parameters: 405e5c89e4eSSatish Balay + N - number of integers in array 406e5c89e4eSSatish Balay . idx - array of integers 407e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 408e5c89e4eSSatish Balay 409e5c89e4eSSatish Balay Level: intermediate 410e5c89e4eSSatish Balay 411300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 412300a7f5bSBarry Smith 413e5c89e4eSSatish Balay .seealso: PetscRealView() 414e5c89e4eSSatish Balay @*/ 4157087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 416e5c89e4eSSatish Balay { 417e5c89e4eSSatish Balay PetscErrorCode ierr; 418e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 419ace3abfcSBarry Smith PetscBool iascii,isbinary; 420e5c89e4eSSatish Balay MPI_Comm comm; 421e5c89e4eSSatish Balay 422e5c89e4eSSatish Balay PetscFunctionBegin; 423e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4240700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4255ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 426e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 427e5c89e4eSSatish Balay 428251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 429251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 430e5c89e4eSSatish Balay if (iascii) { 4317b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 432e5c89e4eSSatish Balay for (i=0; i<n; i++) { 433e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 434e5c89e4eSSatish Balay for (j=0; j<20; j++) { 435e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 436e5c89e4eSSatish Balay } 437e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 438e5c89e4eSSatish Balay } 439e5c89e4eSSatish Balay if (p) { 440e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 441e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 442e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 443e5c89e4eSSatish Balay } 444e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4457b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 4466805f65bSBarry Smith } else if (isbinary) { 4471cbcd9c4SBarry Smith PetscMPIInt rank,size,*sizes,Ntotal,*displs, NN = PetscMPIIntCast(N); 448e5c89e4eSSatish Balay PetscInt *array; 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 { 457e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&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]; 460e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&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 } 466e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&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: 492e5c89e4eSSatish Balay + N - number of doubles in array 493e5c89e4eSSatish Balay . idx - array of doubles 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++) { 522e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",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); 528e5c89e4eSSatish Balay for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",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) { 5341cbcd9c4SBarry Smith PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN = PetscMPIIntCast(N); 535e5c89e4eSSatish Balay PetscReal *array; 536e5c89e4eSSatish Balay 537e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 538e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 539e5c89e4eSSatish Balay 540e5c89e4eSSatish Balay if (size > 1) { 541e5c89e4eSSatish Balay if (rank) { 542e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 543300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPI_DOUBLE,0,0,0,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 544e5c89e4eSSatish Balay } else { 545e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 546e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 547e5c89e4eSSatish Balay Ntotal = sizes[0]; 548e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 549e5c89e4eSSatish Balay displs[0] = 0; 550e5c89e4eSSatish Balay for (i=1; i<size; i++) { 551e5c89e4eSSatish Balay Ntotal += sizes[i]; 552e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 553e5c89e4eSSatish Balay } 554e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 555300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPI_DOUBLE,array,sizes,displs,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 5566805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 557e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 558e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 559e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 560e5c89e4eSSatish Balay } 561e5c89e4eSSatish Balay } else { 56247b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void *) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 563e5c89e4eSSatish Balay } 564e5c89e4eSSatish Balay } else { 565e5c89e4eSSatish Balay const char *tname; 566e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 567e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 568e5c89e4eSSatish Balay } 569e5c89e4eSSatish Balay PetscFunctionReturn(0); 570e5c89e4eSSatish Balay } 571e5c89e4eSSatish Balay 572e5c89e4eSSatish Balay #undef __FUNCT__ 573e5c89e4eSSatish Balay #define __FUNCT__ "PetscScalarView" 574e5c89e4eSSatish Balay /*@C 575e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 576e5c89e4eSSatish Balay 577e5c89e4eSSatish Balay Collective on PetscViewer 578e5c89e4eSSatish Balay 579e5c89e4eSSatish Balay Input Parameters: 580e5c89e4eSSatish Balay + N - number of scalars in array 581e5c89e4eSSatish Balay . idx - array of scalars 582e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 583e5c89e4eSSatish Balay 584e5c89e4eSSatish Balay Level: intermediate 585e5c89e4eSSatish Balay 586300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 587300a7f5bSBarry Smith 588e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 589e5c89e4eSSatish Balay @*/ 5907087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 591e5c89e4eSSatish Balay { 592e5c89e4eSSatish Balay PetscErrorCode ierr; 593e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 594ace3abfcSBarry Smith PetscBool iascii,isbinary; 595e5c89e4eSSatish Balay MPI_Comm comm; 596e5c89e4eSSatish Balay 597e5c89e4eSSatish Balay PetscFunctionBegin; 598e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 599e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 600e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 601e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 602e5c89e4eSSatish Balay 603251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 604251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 605e5c89e4eSSatish Balay if (iascii) { 6067b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 607e5c89e4eSSatish Balay for (i=0; i<n; i++) { 608e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 609e5c89e4eSSatish Balay for (j=0; j<3; j++) { 610e5c89e4eSSatish Balay #if defined (PETSC_USE_COMPLEX) 611e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 612e5c89e4eSSatish Balay PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 613e5c89e4eSSatish Balay #else 614e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",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) 623e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 624e5c89e4eSSatish Balay PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 625e5c89e4eSSatish Balay #else 626e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 627e5c89e4eSSatish Balay #endif 628e5c89e4eSSatish Balay } 629e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 630e5c89e4eSSatish Balay } 631e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6327b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 6336805f65bSBarry Smith } else if (isbinary) { 6341cbcd9c4SBarry Smith PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN = PetscMPIIntCast(N); 635e5c89e4eSSatish Balay PetscScalar *array; 636e5c89e4eSSatish Balay 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 { 645e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&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]; 648e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&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 } 654e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&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