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*/ 6*665c2dedSJed Brown #include <petscviewer.h> 7e5c89e4eSSatish Balay 8e5c89e4eSSatish Balay typedef struct _EH *EH; 9e5c89e4eSSatish Balay struct _EH { 10668f157eSBarry Smith PetscErrorCode (*handler)(MPI_Comm,int,const char*,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 . dir - the directory of the file (indicated by __SDIR__) 31e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 32e5c89e4eSSatish Balay . n - the generic error number 33e5c89e4eSSatish Balay . p - specific error number 34e5c89e4eSSatish Balay - ctx - error handler context 35e5c89e4eSSatish Balay 36e5c89e4eSSatish Balay Options Database Key: 37e5c89e4eSSatish Balay . -on_error_emacs <machinename> 38e5c89e4eSSatish Balay 39e5c89e4eSSatish Balay Level: developer 40e5c89e4eSSatish Balay 41e5c89e4eSSatish Balay Notes: 42e5c89e4eSSatish Balay You must put (server-start) in your .emacs file for the emacsclient software to work 43e5c89e4eSSatish Balay 44e5c89e4eSSatish Balay Most users need not directly employ this routine and the other error 45e5c89e4eSSatish Balay handlers, but can instead use the simplified interface SETERRQ, which has 46e5c89e4eSSatish Balay the calling sequence 47e32f2f54SBarry Smith $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 48e5c89e4eSSatish Balay 49e5c89e4eSSatish Balay Notes for experienced users: 50e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. 51e5c89e4eSSatish Balay 523bf036e2SBarry Smith Developer Note: Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 533bf036e2SBarry Smith 54e5c89e4eSSatish Balay Concepts: emacs^going to on error 55e5c89e4eSSatish Balay Concepts: error handler^going to line in emacs 56e5c89e4eSSatish Balay 57e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 58e5c89e4eSSatish Balay PetscAbortErrorHandler() 59e5c89e4eSSatish Balay @*/ 607087cfbeSBarry 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) 61e5c89e4eSSatish Balay { 62e5c89e4eSSatish Balay PetscErrorCode ierr; 63e5c89e4eSSatish Balay char command[PETSC_MAX_PATH_LEN]; 64e5c89e4eSSatish Balay const char *pdir; 65e5c89e4eSSatish Balay FILE *fp; 66784ad74dSJed Brown PetscInt rval; 67e5c89e4eSSatish Balay 68e5c89e4eSSatish Balay PetscFunctionBegin; 693bf036e2SBarry Smith ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 704b89aa1eSJed Brown sprintf(command,"cd %s; emacsclient --no-wait +%d %s%s\n",pdir,line,dir,file); 71e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 723bf036e2SBarry Smith ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 733bf036e2SBarry Smith ierr = PetscPClose(MPI_COMM_WORLD,fp,&rval);if (ierr) PetscFunctionReturn(ierr); 74e5c89e4eSSatish Balay #else 75e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 76e5c89e4eSSatish Balay #endif 773bf036e2SBarry Smith ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 783bf036e2SBarry Smith if (!eh) { 793bf036e2SBarry Smith ierr = PetscTraceBackErrorHandler(comm,line,fun,file,dir,n,p,mess,0);if (ierr) PetscFunctionReturn(ierr); 803bf036e2SBarry Smith } else { 813bf036e2SBarry Smith ierr = (*eh->handler)(comm,line,fun,file,dir,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 823bf036e2SBarry Smith } 83e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 84e5c89e4eSSatish Balay } 85e5c89e4eSSatish Balay 86e5c89e4eSSatish Balay #undef __FUNCT__ 87e5c89e4eSSatish Balay #define __FUNCT__ "PetscPushErrorHandler" 88e5c89e4eSSatish Balay /*@C 89e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 90e5c89e4eSSatish Balay 91e5c89e4eSSatish Balay Not Collective 92e5c89e4eSSatish Balay 93e5c89e4eSSatish Balay Input Parameters: 94e5c89e4eSSatish Balay + handler - error handler routine 95e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 96e5c89e4eSSatish Balay example file pointers for error messages etc.) 97e5c89e4eSSatish Balay 98e5c89e4eSSatish Balay Calling sequence of handler: 99e32f2f54SBarry Smith $ int handler(MPI_Comm comm,int line,char *func,char *file,char *dir,PetscErrorCode n,int p,char *mess,void *ctx); 100e5c89e4eSSatish Balay 101e32f2f54SBarry Smith + comm - communicator over which error occured 102e32f2f54SBarry Smith . func - the function where the error occured (indicated by __FUNCT__) 103e5c89e4eSSatish Balay . line - the line number of the error (indicated by __LINE__) 104e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 105e5c89e4eSSatish Balay . dir - the directory of the file (indicated by __SDIR__) 106e5c89e4eSSatish Balay . n - the generic error number (see list defined in include/petscerror.h) 107668f157eSBarry Smith . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 108e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 109e5c89e4eSSatish Balay - ctx - the error handler context 110e5c89e4eSSatish Balay 111e5c89e4eSSatish Balay Options Database Keys: 112e5c89e4eSSatish Balay + -on_error_attach_debugger <noxterm,gdb or dbx> 113e5c89e4eSSatish Balay - -on_error_abort 114e5c89e4eSSatish Balay 115e5c89e4eSSatish Balay Level: intermediate 116e5c89e4eSSatish Balay 117e93bc3c1Svictor Notes: 1187850c7c0SBarry Smith The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 119e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 120e93bc3c1Svictor 1217850c7c0SBarry Smith Fortran Notes: You can only push one error handler from Fortran before poping it. 1227850c7c0SBarry Smith 12334b9b2edSBarry Smith .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 124e5c89e4eSSatish Balay 125e5c89e4eSSatish Balay @*/ 1267087cfbeSBarry Smith PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 127e5c89e4eSSatish Balay { 128e5c89e4eSSatish Balay EH neweh; 129e5c89e4eSSatish Balay PetscErrorCode ierr; 130e5c89e4eSSatish Balay 131e5c89e4eSSatish Balay PetscFunctionBegin; 132e5c89e4eSSatish Balay ierr = PetscNew(struct _EH,&neweh);CHKERRQ(ierr); 133a297a907SKarl Rupp if (eh) neweh->previous = eh; 134a297a907SKarl Rupp else neweh->previous = 0; 135e5c89e4eSSatish Balay neweh->handler = handler; 136e5c89e4eSSatish Balay neweh->ctx = ctx; 137e5c89e4eSSatish Balay eh = neweh; 138e5c89e4eSSatish Balay PetscFunctionReturn(0); 139e5c89e4eSSatish Balay } 140e5c89e4eSSatish Balay 141e5c89e4eSSatish Balay #undef __FUNCT__ 142e5c89e4eSSatish Balay #define __FUNCT__ "PetscPopErrorHandler" 143e30d2299SSatish Balay /*@ 144e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 145e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 146e5c89e4eSSatish Balay 147e5c89e4eSSatish Balay Not Collective 148e5c89e4eSSatish Balay 149e5c89e4eSSatish Balay Level: intermediate 150e5c89e4eSSatish Balay 151e5c89e4eSSatish Balay Concepts: error handler^setting 152e5c89e4eSSatish Balay 153e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 154e5c89e4eSSatish Balay @*/ 1557087cfbeSBarry Smith PetscErrorCode PetscPopErrorHandler(void) 156e5c89e4eSSatish Balay { 157e5c89e4eSSatish Balay EH tmp; 158e5c89e4eSSatish Balay PetscErrorCode ierr; 159e5c89e4eSSatish Balay 160e5c89e4eSSatish Balay PetscFunctionBegin; 161e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 162e5c89e4eSSatish Balay tmp = eh; 163e5c89e4eSSatish Balay eh = eh->previous; 164e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 165e5c89e4eSSatish Balay PetscFunctionReturn(0); 166e5c89e4eSSatish Balay } 167e5c89e4eSSatish Balay 168e93bc3c1Svictor #undef __FUNCT__ 169e93bc3c1Svictor #define __FUNCT__ "PetscReturnErrorHandler" 170e93bc3c1Svictor /*@C 171e93bc3c1Svictor PetscReturnErrorHandler - Error handler that causes a return to the current 172e93bc3c1Svictor level. 173e93bc3c1Svictor 174e93bc3c1Svictor Not Collective 175e93bc3c1Svictor 176e93bc3c1Svictor Input Parameters: 177e32f2f54SBarry Smith + comm - communicator over which error occurred 178e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 179e93bc3c1Svictor . func - the function where error is detected (indicated by __FUNCT__) 180e93bc3c1Svictor . file - the file in which the error was detected (indicated by __FILE__) 181e93bc3c1Svictor . dir - the directory of the file (indicated by __SDIR__) 182e93bc3c1Svictor . mess - an error text string, usually just printed to the screen 183e93bc3c1Svictor . n - the generic error number 184e93bc3c1Svictor . p - specific error number 185e93bc3c1Svictor - ctx - error handler context 186e93bc3c1Svictor 187e93bc3c1Svictor Level: developer 188e93bc3c1Svictor 189e93bc3c1Svictor Notes: 190e93bc3c1Svictor Most users need not directly employ this routine and the other error 191e93bc3c1Svictor handlers, but can instead use the simplified interface SETERRQ, which has 192e93bc3c1Svictor the calling sequence 193e32f2f54SBarry Smith $ SETERRQ(comm,number,mess) 194e93bc3c1Svictor 195e93bc3c1Svictor Notes for experienced users: 196e93bc3c1Svictor This routine is good for catching errors such as zero pivots in preconditioners 197e93bc3c1Svictor or breakdown of iterative methods. It is not appropriate for memory violations 198e93bc3c1Svictor and similar errors. 199e93bc3c1Svictor 200e93bc3c1Svictor Use PetscPushErrorHandler() to set the desired error handler. The 201e93bc3c1Svictor currently available PETSc error handlers include PetscTraceBackErrorHandler(), 202e8fb0fc0SBarry Smith PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscAbortErrorHandler() 203e93bc3c1Svictor 204e93bc3c1Svictor Concepts: error handler 205e93bc3c1Svictor 206e93bc3c1Svictor .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(). 207e93bc3c1Svictor @*/ 208e93bc3c1Svictor 2097087cfbeSBarry 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) 210e93bc3c1Svictor { 211e93bc3c1Svictor PetscFunctionBegin; 212e93bc3c1Svictor PetscFunctionReturn(n); 213e93bc3c1Svictor } 214e93bc3c1Svictor 215e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 216e5c89e4eSSatish Balay /* 217e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 218e5c89e4eSSatish Balay there must also be made here 219e5c89e4eSSatish Balay */ 220e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 221e5c89e4eSSatish Balay /*55 */ "Out of memory", 222e5c89e4eSSatish Balay "No support for this operation for this object type", 223e5c89e4eSSatish Balay "No support for this operation on this system", 224e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 225e5c89e4eSSatish Balay /*59 */ "Signal received", 226e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 227e5c89e4eSSatish Balay "Argument aliasing not permitted", 228e5c89e4eSSatish Balay "Invalid argument", 229e5c89e4eSSatish Balay /*63 */ "Argument out of range", 230f08646a8SSatish Balay "Corrupt argument:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 231e5c89e4eSSatish Balay "Unable to open file", 232e5c89e4eSSatish Balay "Read from file failed", 233e5c89e4eSSatish Balay "Write to file failed", 234e5c89e4eSSatish Balay "Invalid pointer", 235e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 236a8b45ee7SBarry Smith /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 237f08646a8SSatish Balay /*71 */ "Detected zero pivot in LU factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 238e5c89e4eSSatish Balay /*72 */ "Floating point exception", 239e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 240e5c89e4eSSatish Balay "Corrupted Petsc object", 241e5c89e4eSSatish Balay "Arguments are incompatible", 242e5c89e4eSSatish Balay "Error in external library", 243e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 244e5c89e4eSSatish Balay "Memory corruption", 245e5c89e4eSSatish Balay "Unexpected data in file", 246e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 247f08646a8SSatish Balay /*81 */ "Detected zero pivot in Cholesky factorization:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#ZeroPivot", 248e5c89e4eSSatish Balay " ", 249e5c89e4eSSatish Balay " ", 250f08646a8SSatish Balay "Overflow in integer operation:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 251e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 252f08646a8SSatish 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", 2538cda6cd7SBarry Smith /*87 */ "Not used", 2548cda6cd7SBarry Smith /*88 */ "Error in system call", 255f08646a8SSatish Balay /*89 */ "Object Type not set:\nsee http://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset" 25673260a9bSLisandro Dalcin /*90 */ " ", 25773260a9bSLisandro Dalcin /* */ " ", 25873260a9bSLisandro Dalcin /* */ " ", 25973260a9bSLisandro Dalcin /* */ " ", 26073260a9bSLisandro Dalcin /* */ " ", 26173260a9bSLisandro Dalcin /*95 */ " ", 26273260a9bSLisandro Dalcin }; 263e5c89e4eSSatish Balay 264e5c89e4eSSatish Balay #undef __FUNCT__ 265e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorMessage" 266e5c89e4eSSatish Balay /*@C 267e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 268e5c89e4eSSatish Balay 269e5c89e4eSSatish Balay Not Collective 270e5c89e4eSSatish Balay 271e5c89e4eSSatish Balay Input Parameter: 272e5c89e4eSSatish Balay . errnum - the error code 273e5c89e4eSSatish Balay 274e5c89e4eSSatish Balay Output Parameter: 2750298fd71SBarry Smith + text - the error message (NULL if not desired) 2760298fd71SBarry Smith - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 277e5c89e4eSSatish Balay 278e5c89e4eSSatish Balay Level: developer 279e5c89e4eSSatish Balay 280e5c89e4eSSatish Balay Concepts: error handler^messages 281e5c89e4eSSatish Balay 282e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 283e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 284e5c89e4eSSatish Balay @*/ 2857087cfbeSBarry Smith PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 286e5c89e4eSSatish Balay { 287e5c89e4eSSatish Balay PetscFunctionBegin; 288a297a907SKarl Rupp if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 289a297a907SKarl Rupp else if (text) *text = 0; 290e5c89e4eSSatish Balay 291a297a907SKarl Rupp if (specific) *specific = PetscErrorBaseMessage; 292e5c89e4eSSatish Balay PetscFunctionReturn(0); 293e5c89e4eSSatish Balay } 294e5c89e4eSSatish Balay 295e5c89e4eSSatish Balay #undef __FUNCT__ 296e5c89e4eSSatish Balay #define __FUNCT__ "PetscError" 297e5c89e4eSSatish Balay /*@C 298e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 299e32f2f54SBarry Smith usually called through the macro SETERRQ(PETSC_COMM_SELF,). 300e5c89e4eSSatish Balay 301e5c89e4eSSatish Balay Not Collective 302e5c89e4eSSatish Balay 303e5c89e4eSSatish Balay Input Parameters: 304e32f2f54SBarry Smith + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 305e32f2f54SBarry Smith . line - the line number of the error (indicated by __LINE__) 306e5c89e4eSSatish Balay . func - the function where the error occured (indicated by __FUNCT__) 307e5c89e4eSSatish Balay . dir - the directory of file (indicated by __SDIR__) 308e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 309e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 310e5c89e4eSSatish Balay . n - the generic error number 311d736bfebSBarry Smith . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 312e5c89e4eSSatish Balay - mess - formatted message string - aka printf 313e5c89e4eSSatish Balay 314e5c89e4eSSatish Balay Level: intermediate 315e5c89e4eSSatish Balay 316e5c89e4eSSatish Balay Notes: 317e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 318e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 319e5c89e4eSSatish Balay sequence 320e32f2f54SBarry Smith $ SETERRQ(comm,n,mess) 321e5c89e4eSSatish Balay 322e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 323e5c89e4eSSatish Balay 3243bf036e2SBarry 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) 3253bf036e2SBarry 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 3263bf036e2SBarry Smith but this annoying. 3273bf036e2SBarry Smith 328e5c89e4eSSatish Balay Concepts: error^setting condition 329e5c89e4eSSatish Balay 330e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 331e5c89e4eSSatish Balay @*/ 3327087cfbeSBarry Smith PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,const char *dir,PetscErrorCode n,PetscErrorType p,const char *mess,...) 333e5c89e4eSSatish Balay { 334e5c89e4eSSatish Balay va_list Argp; 335c9a19010SBarry Smith size_t fullLength; 336e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 337ace3abfcSBarry Smith PetscBool ismain,isunknown; 3383bf036e2SBarry Smith PetscErrorCode ierr; 339e5c89e4eSSatish Balay 3400d577829SBarry Smith PetscFunctionBegin; 341e5c89e4eSSatish Balay if (!func) func = "User provided function"; 342e5c89e4eSSatish Balay if (!file) file = "User file"; 343e5c89e4eSSatish Balay if (!dir) dir = " "; 3440d577829SBarry Smith if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 345e5c89e4eSSatish Balay 346e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 347e5c89e4eSSatish Balay if (mess) { 348e5c89e4eSSatish Balay va_start(Argp,mess); 3492d609e63SMatthew Knepley PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 350e5c89e4eSSatish Balay va_end(Argp); 351e5c89e4eSSatish Balay lbuf = buf; 352a297a907SKarl Rupp if (p == 1) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 353e5c89e4eSSatish Balay } 354e5c89e4eSSatish Balay 355e32f2f54SBarry Smith if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,dir,n,p,lbuf,0); 356e32f2f54SBarry Smith else ierr = (*eh->handler)(comm,line,func,file,dir,n,p,lbuf,eh->ctx); 357e5c89e4eSSatish Balay 358e5c89e4eSSatish Balay /* 359e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 360e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 361e5c89e4eSSatish Balay 362e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 363e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 364e5c89e4eSSatish Balay */ 365e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 366e5c89e4eSSatish Balay PetscStrncmp(func,"unknown",7,&isunknown); 367a297a907SKarl Rupp if (ismain || isunknown) MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 368a297a907SKarl Rupp 369d736bfebSBarry Smith #if defined(PETSC_CLANGUAGE_CXX) && !defined(PETSC_USE_EXTERN_CXX) 370d736bfebSBarry Smith if (p == PETSC_ERROR_IN_CXX) { 371d736bfebSBarry Smith const char *str; 3728fbb6cefSJed Brown if (eh && eh->ctx) { 373d736bfebSBarry Smith std::ostringstream *msg; 374d736bfebSBarry Smith msg = (std::ostringstream*) eh->ctx; 375d736bfebSBarry Smith str = msg->str().c_str(); 376a297a907SKarl Rupp } else str = "Error detected in C PETSc"; 377a297a907SKarl Rupp 378d736bfebSBarry Smith throw PETSc::Exception(str); 379d736bfebSBarry Smith } 380d736bfebSBarry Smith #endif 381e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 382e5c89e4eSSatish Balay } 383e5c89e4eSSatish Balay 384e5c89e4eSSatish Balay /* -------------------------------------------------------------------------*/ 385e5c89e4eSSatish Balay 386e5c89e4eSSatish Balay #undef __FUNCT__ 387e5c89e4eSSatish Balay #define __FUNCT__ "PetscIntView" 388e5c89e4eSSatish Balay /*@C 389e5c89e4eSSatish Balay PetscIntView - Prints an array of integers; useful for debugging. 390e5c89e4eSSatish Balay 391e5c89e4eSSatish Balay Collective on PetscViewer 392e5c89e4eSSatish Balay 393e5c89e4eSSatish Balay Input Parameters: 394e5c89e4eSSatish Balay + N - number of integers in array 395e5c89e4eSSatish Balay . idx - array of integers 396e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 397e5c89e4eSSatish Balay 398e5c89e4eSSatish Balay Level: intermediate 399e5c89e4eSSatish Balay 400300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 401300a7f5bSBarry Smith 402e5c89e4eSSatish Balay .seealso: PetscRealView() 403e5c89e4eSSatish Balay @*/ 4047087cfbeSBarry Smith PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 405e5c89e4eSSatish Balay { 406e5c89e4eSSatish Balay PetscErrorCode ierr; 407e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 408ace3abfcSBarry Smith PetscBool iascii,isbinary; 409e5c89e4eSSatish Balay MPI_Comm comm; 410e5c89e4eSSatish Balay 411e5c89e4eSSatish Balay PetscFunctionBegin; 412e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 4130700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 4145ab33896SBarry Smith if (N) PetscValidIntPointer(idx,2); 415e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 416e5c89e4eSSatish Balay 417251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 418251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 419e5c89e4eSSatish Balay if (iascii) { 4207b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 421e5c89e4eSSatish Balay for (i=0; i<n; i++) { 422e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 423e5c89e4eSSatish Balay for (j=0; j<20; j++) { 424e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 425e5c89e4eSSatish Balay } 426e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 427e5c89e4eSSatish Balay } 428e5c89e4eSSatish Balay if (p) { 429e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 430e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 431e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 432e5c89e4eSSatish Balay } 433e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 4347b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 4356805f65bSBarry Smith } else if (isbinary) { 436783b601eSJed Brown PetscMPIInt rank,size,*sizes,Ntotal,*displs,NN; 437e5c89e4eSSatish Balay PetscInt *array; 438783b601eSJed Brown 4394dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 440e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 441e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 442e5c89e4eSSatish Balay 443e5c89e4eSSatish Balay if (size > 1) { 444e5c89e4eSSatish Balay if (rank) { 445e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 446300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 447e5c89e4eSSatish Balay } else { 448e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 449adcec1e5SJed Brown ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 450e5c89e4eSSatish Balay Ntotal = sizes[0]; 451e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 452e5c89e4eSSatish Balay displs[0] = 0; 453e5c89e4eSSatish Balay for (i=1; i<size; i++) { 454e5c89e4eSSatish Balay Ntotal += sizes[i]; 455e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 456e5c89e4eSSatish Balay } 457e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&array);CHKERRQ(ierr); 458300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 4596805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 460e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 461e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 462e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 463e5c89e4eSSatish Balay } 464e5c89e4eSSatish Balay } else { 46547b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 466e5c89e4eSSatish Balay } 467e5c89e4eSSatish Balay } else { 468e5c89e4eSSatish Balay const char *tname; 469e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 470e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 471e5c89e4eSSatish Balay } 472e5c89e4eSSatish Balay PetscFunctionReturn(0); 473e5c89e4eSSatish Balay } 474e5c89e4eSSatish Balay 475e5c89e4eSSatish Balay #undef __FUNCT__ 476e5c89e4eSSatish Balay #define __FUNCT__ "PetscRealView" 477e5c89e4eSSatish Balay /*@C 478e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 479e5c89e4eSSatish Balay 480e5c89e4eSSatish Balay Collective on PetscViewer 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay Input Parameters: 483e5c89e4eSSatish Balay + N - number of doubles in array 484e5c89e4eSSatish Balay . idx - array of doubles 485e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 486e5c89e4eSSatish Balay 487e5c89e4eSSatish Balay Level: intermediate 488e5c89e4eSSatish Balay 489300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 490300a7f5bSBarry Smith 491e5c89e4eSSatish Balay .seealso: PetscIntView() 492e5c89e4eSSatish Balay @*/ 4937087cfbeSBarry Smith PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 494e5c89e4eSSatish Balay { 495e5c89e4eSSatish Balay PetscErrorCode ierr; 496e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 497ace3abfcSBarry Smith PetscBool iascii,isbinary; 498e5c89e4eSSatish Balay MPI_Comm comm; 499e5c89e4eSSatish Balay 500e5c89e4eSSatish Balay PetscFunctionBegin; 501e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 5020700a824SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 503e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 504e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 505e5c89e4eSSatish Balay 506251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 507251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 508e5c89e4eSSatish Balay if (iascii) { 5097b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 510e5c89e4eSSatish Balay for (i=0; i<n; i++) { 511e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*i);CHKERRQ(ierr); 512e5c89e4eSSatish Balay for (j=0; j<5; j++) { 513e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*5+j]);CHKERRQ(ierr); 514e5c89e4eSSatish Balay } 515e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 516e5c89e4eSSatish Balay } 517e5c89e4eSSatish Balay if (p) { 518e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*n);CHKERRQ(ierr); 519e5c89e4eSSatish Balay for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[5*n+i]);CHKERRQ(ierr);} 520e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 521e5c89e4eSSatish Balay } 522e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 5237b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 5246805f65bSBarry Smith } else if (isbinary) { 525783b601eSJed Brown PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN; 526e5c89e4eSSatish Balay PetscReal *array; 527e5c89e4eSSatish Balay 5284dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 529e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 530e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 531e5c89e4eSSatish Balay 532e5c89e4eSSatish Balay if (size > 1) { 533e5c89e4eSSatish Balay if (rank) { 534e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 5351d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,0,0,0,MPIU_REAL,0,comm);CHKERRQ(ierr); 536e5c89e4eSSatish Balay } else { 537e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 538e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 539e5c89e4eSSatish Balay Ntotal = sizes[0]; 540e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 541e5c89e4eSSatish Balay displs[0] = 0; 542e5c89e4eSSatish Balay for (i=1; i<size; i++) { 543e5c89e4eSSatish Balay Ntotal += sizes[i]; 544e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 545e5c89e4eSSatish Balay } 546e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 5471d36abb7SJed Brown ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 5486805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL,PETSC_TRUE);CHKERRQ(ierr); 549e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 550e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 551e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 552e5c89e4eSSatish Balay } 553e5c89e4eSSatish Balay } else { 55447b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL,PETSC_FALSE);CHKERRQ(ierr); 555e5c89e4eSSatish Balay } 556e5c89e4eSSatish Balay } else { 557e5c89e4eSSatish Balay const char *tname; 558e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 559e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 560e5c89e4eSSatish Balay } 561e5c89e4eSSatish Balay PetscFunctionReturn(0); 562e5c89e4eSSatish Balay } 563e5c89e4eSSatish Balay 564e5c89e4eSSatish Balay #undef __FUNCT__ 565e5c89e4eSSatish Balay #define __FUNCT__ "PetscScalarView" 566e5c89e4eSSatish Balay /*@C 567e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 568e5c89e4eSSatish Balay 569e5c89e4eSSatish Balay Collective on PetscViewer 570e5c89e4eSSatish Balay 571e5c89e4eSSatish Balay Input Parameters: 572e5c89e4eSSatish Balay + N - number of scalars in array 573e5c89e4eSSatish Balay . idx - array of scalars 574e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 575e5c89e4eSSatish Balay 576e5c89e4eSSatish Balay Level: intermediate 577e5c89e4eSSatish Balay 578300a7f5bSBarry Smith Developer Notes: idx cannot be const because may be passed to binary viewer where byte swappping is done 579300a7f5bSBarry Smith 580e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 581e5c89e4eSSatish Balay @*/ 5827087cfbeSBarry Smith PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 583e5c89e4eSSatish Balay { 584e5c89e4eSSatish Balay PetscErrorCode ierr; 585e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 586ace3abfcSBarry Smith PetscBool iascii,isbinary; 587e5c89e4eSSatish Balay MPI_Comm comm; 588e5c89e4eSSatish Balay 589e5c89e4eSSatish Balay PetscFunctionBegin; 590e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 591e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 592e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 593e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 594e5c89e4eSSatish Balay 595251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 596251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 597e5c89e4eSSatish Balay if (iascii) { 5987b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 599e5c89e4eSSatish Balay for (i=0; i<n; i++) { 600e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 601e5c89e4eSSatish Balay for (j=0; j<3; j++) { 602e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 603b0043f70SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 604e5c89e4eSSatish Balay #else 605e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*3+j]);CHKERRQ(ierr); 606e5c89e4eSSatish Balay #endif 607e5c89e4eSSatish Balay } 608e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 609e5c89e4eSSatish Balay } 610e5c89e4eSSatish Balay if (p) { 611e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 612e5c89e4eSSatish Balay for (i=0; i<p; i++) { 613e5c89e4eSSatish Balay #if defined(PETSC_USE_COMPLEX) 614b0043f70SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 615e5c89e4eSSatish Balay #else 616e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 617e5c89e4eSSatish Balay #endif 618e5c89e4eSSatish Balay } 619e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 620e5c89e4eSSatish Balay } 621e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 6227b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 6236805f65bSBarry Smith } else if (isbinary) { 624783b601eSJed Brown PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN; 625e5c89e4eSSatish Balay PetscScalar *array; 626e5c89e4eSSatish Balay 6274dc2109aSBarry Smith ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 628e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 629e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 630e5c89e4eSSatish Balay 631e5c89e4eSSatish Balay if (size > 1) { 632e5c89e4eSSatish Balay if (rank) { 633e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 634300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 635e5c89e4eSSatish Balay } else { 636e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 637e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 638e5c89e4eSSatish Balay Ntotal = sizes[0]; 639e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 640e5c89e4eSSatish Balay displs[0] = 0; 641e5c89e4eSSatish Balay for (i=1; i<size; i++) { 642e5c89e4eSSatish Balay Ntotal += sizes[i]; 643e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 644e5c89e4eSSatish Balay } 645e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&array);CHKERRQ(ierr); 646300a7f5bSBarry Smith ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 6476805f65bSBarry Smith ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR,PETSC_TRUE);CHKERRQ(ierr); 648e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 649e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 650e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 651e5c89e4eSSatish Balay } 652e5c89e4eSSatish Balay } else { 65347b06e05SMatthew Knepley ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 654e5c89e4eSSatish Balay } 655e5c89e4eSSatish Balay } else { 656e5c89e4eSSatish Balay const char *tname; 657e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 658e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 659e5c89e4eSSatish Balay } 660e5c89e4eSSatish Balay PetscFunctionReturn(0); 661e5c89e4eSSatish Balay } 662e5c89e4eSSatish Balay 663e5c89e4eSSatish Balay 664e5c89e4eSSatish Balay 665e5c89e4eSSatish Balay 666