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