1e5c89e4eSSatish Balay #define PETSC_DLL 2e5c89e4eSSatish Balay /* 3e5c89e4eSSatish Balay Code that allows one to set the error handlers 4e5c89e4eSSatish Balay */ 5e5c89e4eSSatish Balay #include "petsc.h" /*I "petsc.h" I*/ 6e5c89e4eSSatish Balay #include "petscsys.h" 7e5c89e4eSSatish Balay #include <stdarg.h> 8e5c89e4eSSatish Balay #if defined(PETSC_HAVE_STDLIB_H) 9e5c89e4eSSatish Balay #include <stdlib.h> 10e5c89e4eSSatish Balay #endif 11e5c89e4eSSatish Balay 12e5c89e4eSSatish Balay typedef struct _EH *EH; 13e5c89e4eSSatish Balay struct _EH { 14e5c89e4eSSatish Balay int cookie; 15e5c89e4eSSatish Balay PetscErrorCode (*handler)(int,const char*,const char*,const char *,PetscErrorCode,int,const char*,void *); 16e5c89e4eSSatish Balay void *ctx; 17e5c89e4eSSatish Balay EH previous; 18e5c89e4eSSatish Balay }; 19e5c89e4eSSatish Balay 20e5c89e4eSSatish Balay static EH eh = 0; 21e5c89e4eSSatish Balay 22e5c89e4eSSatish Balay #undef __FUNCT__ 23e5c89e4eSSatish Balay #define __FUNCT__ "PetscEmacsClientErrorHandler" 24e5c89e4eSSatish Balay /*@C 25e5c89e4eSSatish Balay PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 26e5c89e4eSSatish Balay load the file where the error occured. Then calls the "previous" error handler. 27e5c89e4eSSatish Balay 28e5c89e4eSSatish Balay Not Collective 29e5c89e4eSSatish Balay 30e5c89e4eSSatish Balay Input Parameters: 31e5c89e4eSSatish Balay + 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 51e5c89e4eSSatish Balay $ SETERRQ(number,p,mess) 52e5c89e4eSSatish Balay 53e5c89e4eSSatish Balay Notes for experienced users: 54e5c89e4eSSatish Balay Use PetscPushErrorHandler() to set the desired error handler. The 55e5c89e4eSSatish Balay currently available PETSc error handlers include PetscTraceBackErrorHandler(), 56e5c89e4eSSatish Balay PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscStopErrorHandler() 57e5c89e4eSSatish Balay 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 @*/ 64e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscEmacsClientErrorHandler(int line,const char *fun,const char* file,const char *dir,PetscErrorCode n,int 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; 70e5c89e4eSSatish Balay 71e5c89e4eSSatish Balay PetscFunctionBegin; 72e5c89e4eSSatish Balay /* Note: don't check error codes since this an error handler :-) */ 73e5c89e4eSSatish Balay ierr = PetscGetPetscDir(&pdir);CHKERRQ(ierr); 74e5c89e4eSSatish Balay sprintf(command,"emacsclient +%d %s/%s%s\n",line,pdir,dir,file); 75e5c89e4eSSatish Balay #if defined(PETSC_HAVE_POPEN) 76e5c89e4eSSatish Balay ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp); 77e5c89e4eSSatish Balay ierr = PetscPClose(MPI_COMM_WORLD,fp); 78e5c89e4eSSatish Balay #else 79e5c89e4eSSatish Balay SETERRQ(PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 80e5c89e4eSSatish Balay #endif 81e5c89e4eSSatish Balay ierr = PetscPopErrorHandler(); /* remove this handler from the stack of handlers */ 82e5c89e4eSSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(line,fun,file,dir,n,p,mess,0); 83e5c89e4eSSatish Balay else ierr = (*eh->handler)(line,fun,file,dir,n,p,mess,eh->ctx); 84e5c89e4eSSatish Balay PetscFunctionReturn(ierr); 85e5c89e4eSSatish Balay } 86e5c89e4eSSatish Balay 87e5c89e4eSSatish Balay #undef __FUNCT__ 88e5c89e4eSSatish Balay #define __FUNCT__ "PetscPushErrorHandler" 89e5c89e4eSSatish Balay /*@C 90e5c89e4eSSatish Balay PetscPushErrorHandler - Sets a routine to be called on detection of errors. 91e5c89e4eSSatish Balay 92e5c89e4eSSatish Balay Not Collective 93e5c89e4eSSatish Balay 94e5c89e4eSSatish Balay Input Parameters: 95e5c89e4eSSatish Balay + handler - error handler routine 96e5c89e4eSSatish Balay - ctx - optional handler context that contains information needed by the handler (for 97e5c89e4eSSatish Balay example file pointers for error messages etc.) 98e5c89e4eSSatish Balay 99e5c89e4eSSatish Balay Calling sequence of handler: 100e5c89e4eSSatish Balay $ int handler(int line,char *func,char *file,char *dir,PetscErrorCode n,int p,char *mess,void *ctx); 101e5c89e4eSSatish Balay 102e5c89e4eSSatish Balay + 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) 107e5c89e4eSSatish Balay . p - the specific error number 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 117e5c89e4eSSatish Balay .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 118e5c89e4eSSatish Balay 119e5c89e4eSSatish Balay @*/ 120e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscPushErrorHandler(PetscErrorCode (*handler)(int,const char *,const char*,const char*,PetscErrorCode,int,const char*,void*),void *ctx) 121e5c89e4eSSatish Balay { 122e5c89e4eSSatish Balay EH neweh; 123e5c89e4eSSatish Balay PetscErrorCode ierr; 124e5c89e4eSSatish Balay 125e5c89e4eSSatish Balay PetscFunctionBegin; 126e5c89e4eSSatish Balay ierr = PetscNew(struct _EH,&neweh);CHKERRQ(ierr); 127e5c89e4eSSatish Balay if (eh) {neweh->previous = eh;} 128e5c89e4eSSatish Balay else {neweh->previous = 0;} 129e5c89e4eSSatish Balay neweh->handler = handler; 130e5c89e4eSSatish Balay neweh->ctx = ctx; 131e5c89e4eSSatish Balay eh = neweh; 132e5c89e4eSSatish Balay PetscFunctionReturn(0); 133e5c89e4eSSatish Balay } 134e5c89e4eSSatish Balay 135e5c89e4eSSatish Balay #undef __FUNCT__ 136e5c89e4eSSatish Balay #define __FUNCT__ "PetscPopErrorHandler" 137*e30d2299SSatish Balay /*@ 138e5c89e4eSSatish Balay PetscPopErrorHandler - Removes the latest error handler that was 139e5c89e4eSSatish Balay pushed with PetscPushErrorHandler(). 140e5c89e4eSSatish Balay 141e5c89e4eSSatish Balay Not Collective 142e5c89e4eSSatish Balay 143e5c89e4eSSatish Balay Level: intermediate 144e5c89e4eSSatish Balay 145e5c89e4eSSatish Balay Concepts: error handler^setting 146e5c89e4eSSatish Balay 147e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler() 148e5c89e4eSSatish Balay @*/ 149e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscPopErrorHandler(void) 150e5c89e4eSSatish Balay { 151e5c89e4eSSatish Balay EH tmp; 152e5c89e4eSSatish Balay PetscErrorCode ierr; 153e5c89e4eSSatish Balay 154e5c89e4eSSatish Balay PetscFunctionBegin; 155e5c89e4eSSatish Balay if (!eh) PetscFunctionReturn(0); 156e5c89e4eSSatish Balay tmp = eh; 157e5c89e4eSSatish Balay eh = eh->previous; 158e5c89e4eSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 159e5c89e4eSSatish Balay 160e5c89e4eSSatish Balay PetscFunctionReturn(0); 161e5c89e4eSSatish Balay } 162e5c89e4eSSatish Balay 163e5c89e4eSSatish Balay static char PetscErrorBaseMessage[1024]; 164e5c89e4eSSatish Balay /* 165e5c89e4eSSatish Balay The numerical values for these are defined in include/petscerror.h; any changes 166e5c89e4eSSatish Balay there must also be made here 167e5c89e4eSSatish Balay */ 168e5c89e4eSSatish Balay static const char *PetscErrorStrings[] = { 169e5c89e4eSSatish Balay /*55 */ "Out of memory", 170e5c89e4eSSatish Balay "No support for this operation for this object type", 171e5c89e4eSSatish Balay "No support for this operation on this system", 172e5c89e4eSSatish Balay /*58 */ "Operation done in wrong order", 173e5c89e4eSSatish Balay /*59 */ "Signal received", 174e5c89e4eSSatish Balay /*60 */ "Nonconforming object sizes", 175e5c89e4eSSatish Balay "Argument aliasing not permitted", 176e5c89e4eSSatish Balay "Invalid argument", 177e5c89e4eSSatish Balay /*63 */ "Argument out of range", 178e5c89e4eSSatish Balay "Corrupt argument: see http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#Corrupt", 179e5c89e4eSSatish Balay "Unable to open file", 180e5c89e4eSSatish Balay "Read from file failed", 181e5c89e4eSSatish Balay "Write to file failed", 182e5c89e4eSSatish Balay "Invalid pointer", 183e5c89e4eSSatish Balay /*69 */ "Arguments must have same type", 184e5c89e4eSSatish Balay "", 185e5c89e4eSSatish Balay /*71 */ "Detected zero pivot in LU factorization\nsee http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#ZeroPivot", 186e5c89e4eSSatish Balay /*72 */ "Floating point exception", 187e5c89e4eSSatish Balay /*73 */ "Object is in wrong state", 188e5c89e4eSSatish Balay "Corrupted Petsc object", 189e5c89e4eSSatish Balay "Arguments are incompatible", 190e5c89e4eSSatish Balay "Error in external library", 191e5c89e4eSSatish Balay /*77 */ "Petsc has generated inconsistent data", 192e5c89e4eSSatish Balay "Memory corruption", 193e5c89e4eSSatish Balay "Unexpected data in file", 194e5c89e4eSSatish Balay /*80 */ "Arguments must have same communicators", 195e5c89e4eSSatish Balay /*81 */ "Detected zero pivot in Cholesky factorization\nsee http://www.mcs.anl.gov/petsc/petsc-as/documentation/troubleshooting.html#ZeroPivot", 196e5c89e4eSSatish Balay " ", 197e5c89e4eSSatish Balay " ", 198e5c89e4eSSatish Balay " ", 199e5c89e4eSSatish Balay /*85 */ "Null argument, when expecting valid pointer", 200e5c89e4eSSatish Balay /*86 */ "Unknown type. Check for miss-spelling or missing external package needed for type"}; 201e5c89e4eSSatish Balay 202e5c89e4eSSatish Balay #undef __FUNCT__ 203e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorMessage" 204e5c89e4eSSatish Balay /*@C 205e5c89e4eSSatish Balay PetscErrorMessage - returns the text string associated with a PETSc error code. 206e5c89e4eSSatish Balay 207e5c89e4eSSatish Balay Not Collective 208e5c89e4eSSatish Balay 209e5c89e4eSSatish Balay Input Parameter: 210e5c89e4eSSatish Balay . errnum - the error code 211e5c89e4eSSatish Balay 212e5c89e4eSSatish Balay Output Parameter: 213e5c89e4eSSatish Balay + text - the error message (PETSC_NULL if not desired) 214e5c89e4eSSatish Balay - specific - the specific error message that was set with SETERRxxx() or PetscError(). (PETSC_NULL if not desired) 215e5c89e4eSSatish Balay 216e5c89e4eSSatish Balay Level: developer 217e5c89e4eSSatish Balay 218e5c89e4eSSatish Balay Concepts: error handler^messages 219e5c89e4eSSatish Balay 220e5c89e4eSSatish Balay .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), 221e5c89e4eSSatish Balay PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 222e5c89e4eSSatish Balay @*/ 223e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscErrorMessage(int errnum,const char *text[],char **specific) 224e5c89e4eSSatish Balay { 225e5c89e4eSSatish Balay PetscFunctionBegin; 226e5c89e4eSSatish Balay if (text && errnum >= PETSC_ERR_MEM && errnum <= PETSC_ERR_MEM_MALLOC_0) { 227e5c89e4eSSatish Balay *text = PetscErrorStrings[errnum-PETSC_ERR_MEM]; 228e5c89e4eSSatish Balay } else if (text) *text = 0; 229e5c89e4eSSatish Balay 230e5c89e4eSSatish Balay if (specific) { 231e5c89e4eSSatish Balay *specific = PetscErrorBaseMessage; 232e5c89e4eSSatish Balay } 233e5c89e4eSSatish Balay PetscFunctionReturn(0); 234e5c89e4eSSatish Balay } 235e5c89e4eSSatish Balay 236e5c89e4eSSatish Balay #if defined(PETSC_USE_ERRORCHECKING) 237e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscErrorUncatchable[PETSC_EXCEPTIONS_MAX] = {0}; 238e5c89e4eSSatish Balay PetscInt PETSC_DLLEXPORT PetscErrorUncatchableCount = 0; 239e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscExceptions[PETSC_EXCEPTIONS_MAX] = {0}; 240e5c89e4eSSatish Balay PetscInt PETSC_DLLEXPORT PetscExceptionsCount = 0; 241e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscExceptionTmp = 0; 242e5c89e4eSSatish Balay 243e5c89e4eSSatish Balay #undef __FUNCT__ 244e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorIsCatchable" 245e5c89e4eSSatish Balay static PetscTruth PetscErrorIsCatchable(PetscErrorCode err) 246e5c89e4eSSatish Balay { 247e5c89e4eSSatish Balay PetscInt i; 248e5c89e4eSSatish Balay for (i=0; i<PetscErrorUncatchableCount; i++) { 249e5c89e4eSSatish Balay if (err == PetscErrorUncatchable[i]) return PETSC_FALSE; 250e5c89e4eSSatish Balay } 251e5c89e4eSSatish Balay return PETSC_TRUE; 252e5c89e4eSSatish Balay } 253e5c89e4eSSatish Balay 254e5c89e4eSSatish Balay #undef __FUNCT__ 255e5c89e4eSSatish Balay #define __FUNCT__ "PetscErrorSetCatchable" 256e5c89e4eSSatish Balay /*@ 257e5c89e4eSSatish Balay PetscErrorSetCatchable - Sets if a PetscErrorCode can be caught with a PetscExceptionTry1() 258e5c89e4eSSatish Balay PetscExceptionCaught() pair. By default all errors are catchable. 259e5c89e4eSSatish Balay 260e5c89e4eSSatish Balay Input Parameters: 261e5c89e4eSSatish Balay + err - error code 262e5c89e4eSSatish Balay - flg - PETSC_TRUE means allow to be caught, PETSC_FALSE means do not allow to be caught 263e5c89e4eSSatish Balay 264e5c89e4eSSatish Balay Level: advanced 265e5c89e4eSSatish Balay 266e5c89e4eSSatish Balay Notes: 267e5c89e4eSSatish Balay PETSc must not be configured using the option --with-errorchecking=0 for this to work 268e5c89e4eSSatish Balay 269e5c89e4eSSatish Balay .seealso: PetscExceptionTry1(), PetscExceptionCaught() 270e5c89e4eSSatish Balay @*/ 271e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscErrorSetCatchable(PetscErrorCode err,PetscTruth flg) 272e5c89e4eSSatish Balay { 273e5c89e4eSSatish Balay PetscFunctionBegin; 274e5c89e4eSSatish Balay if (!flg && PetscErrorIsCatchable(err)) { 275e5c89e4eSSatish Balay /* add to list of uncatchable */ 2760e5e90baSSatish Balay if (PetscErrorUncatchableCount >= PETSC_EXCEPTIONS_MAX) SETERRQ(PETSC_ERR_PLIB,"Stack for PetscErrorUncatchable is overflowed, recompile \nsrc/sysd/error/err.c with a larger value for PETSC_EXCEPTIONS_MAX"); 277e5c89e4eSSatish Balay PetscErrorUncatchable[PetscErrorUncatchableCount++] = err; 278e5c89e4eSSatish Balay } else if (flg && !PetscErrorIsCatchable(err)) { 279e5c89e4eSSatish Balay /* remove from list of uncatchable */ 280e5c89e4eSSatish Balay PetscInt i; 281e5c89e4eSSatish Balay for (i=0; i<PetscErrorUncatchableCount; i++) { 282e5c89e4eSSatish Balay if (PetscErrorUncatchable[i] == err) break; 283e5c89e4eSSatish Balay } 284e5c89e4eSSatish Balay for (;i<PetscErrorUncatchableCount; i++) { 285e5c89e4eSSatish Balay PetscErrorUncatchable[i] = PetscErrorUncatchable[i+1]; 286e5c89e4eSSatish Balay } 287e5c89e4eSSatish Balay PetscErrorUncatchableCount--; 288e5c89e4eSSatish Balay } 289e5c89e4eSSatish Balay PetscFunctionReturn(0); 290e5c89e4eSSatish Balay } 291e5c89e4eSSatish Balay 292e5c89e4eSSatish Balay #undef __FUNCT__ 293e5c89e4eSSatish Balay #define __FUNCT__ "PetscExceptionPush" 294e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscExceptionPush(PetscErrorCode err) 295e5c89e4eSSatish Balay { 296e5c89e4eSSatish Balay PetscFunctionBegin; 2970e5e90baSSatish Balay if (PetscExceptionsCount >= PETSC_EXCEPTIONS_MAX) SETERRQ(PETSC_ERR_PLIB,"Stack for PetscExceptions is overflowed, recompile \nsrc/sysd/error/err.c with a larger value for PETSC_EXCEPTIONS_MAX"); 298e5c89e4eSSatish Balay if (PetscErrorIsCatchable(err)) PetscExceptions[PetscExceptionsCount++] = err; 299e5c89e4eSSatish Balay PetscFunctionReturn(0); 300e5c89e4eSSatish Balay } 301e5c89e4eSSatish Balay 302e5c89e4eSSatish Balay #undef __FUNCT__ 303e5c89e4eSSatish Balay #define __FUNCT__ "PetscExceptionPop" 304e5c89e4eSSatish Balay void PETSC_DLLEXPORT PetscExceptionPop(PetscErrorCode err) 305e5c89e4eSSatish Balay { 306e5c89e4eSSatish Balay /* if (PetscExceptionsCount <= 0)SETERRQ(PETSC_ERR_PLIB,"Stack for PetscExceptions is empty"); */ 307e5c89e4eSSatish Balay if (PetscErrorIsCatchable(err)) PetscExceptionsCount--; 308e5c89e4eSSatish Balay } 309e5c89e4eSSatish Balay #endif 310e5c89e4eSSatish Balay 311e5c89e4eSSatish Balay #undef __FUNCT__ 312e5c89e4eSSatish Balay #define __FUNCT__ "PetscError" 313e5c89e4eSSatish Balay /*@C 314e5c89e4eSSatish Balay PetscError - Routine that is called when an error has been detected, 315e5c89e4eSSatish Balay usually called through the macro SETERRQ(). 316e5c89e4eSSatish Balay 317e5c89e4eSSatish Balay Not Collective 318e5c89e4eSSatish Balay 319e5c89e4eSSatish Balay Input Parameters: 320e5c89e4eSSatish Balay + line - the line number of the error (indicated by __LINE__) 321e5c89e4eSSatish Balay . func - the function where the error occured (indicated by __FUNCT__) 322e5c89e4eSSatish Balay . dir - the directory of file (indicated by __SDIR__) 323e5c89e4eSSatish Balay . file - the file in which the error was detected (indicated by __FILE__) 324e5c89e4eSSatish Balay . mess - an error text string, usually just printed to the screen 325e5c89e4eSSatish Balay . n - the generic error number 326e5c89e4eSSatish Balay . p - 1 indicates the error was initially detected, 0 indicates this is a traceback from a 327e5c89e4eSSatish Balay previously detected error 328e5c89e4eSSatish Balay - mess - formatted message string - aka printf 329e5c89e4eSSatish Balay 330e5c89e4eSSatish Balay Level: intermediate 331e5c89e4eSSatish Balay 332e5c89e4eSSatish Balay Notes: 333e5c89e4eSSatish Balay Most users need not directly use this routine and the error handlers, but 334e5c89e4eSSatish Balay can instead use the simplified interface SETERRQ, which has the calling 335e5c89e4eSSatish Balay sequence 336e5c89e4eSSatish Balay $ SETERRQ(n,mess) 337e5c89e4eSSatish Balay 338e5c89e4eSSatish Balay Experienced users can set the error handler with PetscPushErrorHandler(). 339e5c89e4eSSatish Balay 340e5c89e4eSSatish Balay Concepts: error^setting condition 341e5c89e4eSSatish Balay 342e5c89e4eSSatish Balay .seealso: PetscTraceBackErrorHandler(), PetscPushErrorHandler(), SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2() 343e5c89e4eSSatish Balay @*/ 344e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscError(int line,const char *func,const char* file,const char *dir,PetscErrorCode n,int p,const char *mess,...) 345e5c89e4eSSatish Balay { 346e5c89e4eSSatish Balay va_list Argp; 347e5c89e4eSSatish Balay PetscErrorCode ierr; 348e5c89e4eSSatish Balay char buf[2048],*lbuf = 0; 349e5c89e4eSSatish Balay PetscTruth ismain,isunknown; 350e5c89e4eSSatish Balay #if defined(PETSC_USE_ERRORCHECKING) 351e5c89e4eSSatish Balay PetscInt i; 352e5c89e4eSSatish Balay #endif 353e5c89e4eSSatish Balay 354e5c89e4eSSatish Balay if (!func) func = "User provided function"; 355e5c89e4eSSatish Balay if (!file) file = "User file"; 356e5c89e4eSSatish Balay if (!dir) dir = " "; 357e5c89e4eSSatish Balay 358e5c89e4eSSatish Balay PetscFunctionBegin; 359e5c89e4eSSatish Balay /* Compose the message evaluating the print format */ 360e5c89e4eSSatish Balay if (mess) { 361e5c89e4eSSatish Balay va_start(Argp,mess); 362e5c89e4eSSatish Balay PetscVSNPrintf(buf,2048,mess,Argp); 363e5c89e4eSSatish Balay va_end(Argp); 364e5c89e4eSSatish Balay lbuf = buf; 365e5c89e4eSSatish Balay if (p == 1) { 366e5c89e4eSSatish Balay PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 367e5c89e4eSSatish Balay } 368e5c89e4eSSatish Balay } 369e5c89e4eSSatish Balay 370e5c89e4eSSatish Balay #if defined(PETSC_USE_ERRORCHECKING) 371e5c89e4eSSatish Balay /* check if user is catching this exception */ 372e5c89e4eSSatish Balay for (i=0; i<PetscExceptionsCount; i++) { 373e5c89e4eSSatish Balay if (n == PetscExceptions[i]) PetscFunctionReturn(n); 374e5c89e4eSSatish Balay } 375e5c89e4eSSatish Balay #endif 376e5c89e4eSSatish Balay 377e5c89e4eSSatish Balay if (!eh) ierr = PetscTraceBackErrorHandler(line,func,file,dir,n,p,lbuf,0); 378e5c89e4eSSatish Balay else ierr = (*eh->handler)(line,func,file,dir,n,p,lbuf,eh->ctx); 379e5c89e4eSSatish Balay 380e5c89e4eSSatish Balay /* 381e5c89e4eSSatish Balay If this is called from the main() routine we call MPI_Abort() instead of 382e5c89e4eSSatish Balay return to allow the parallel program to be properly shutdown. 383e5c89e4eSSatish Balay 384e5c89e4eSSatish Balay Since this is in the error handler we don't check the errors below. Of course, 385e5c89e4eSSatish Balay PetscStrncmp() does its own error checking which is problamatic 386e5c89e4eSSatish Balay */ 387e5c89e4eSSatish Balay PetscStrncmp(func,"main",4,&ismain); 388e5c89e4eSSatish Balay PetscStrncmp(func,"unknown",7,&isunknown); 389e5c89e4eSSatish Balay if (ismain || isunknown) { 390e5c89e4eSSatish Balay MPI_Abort(PETSC_COMM_WORLD,(int)ierr); 391e5c89e4eSSatish Balay } 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 411e5c89e4eSSatish Balay .seealso: PetscRealView() 412e5c89e4eSSatish Balay @*/ 413e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscIntView(PetscInt N,PetscInt idx[],PetscViewer viewer) 414e5c89e4eSSatish Balay { 415e5c89e4eSSatish Balay PetscErrorCode ierr; 416e5c89e4eSSatish Balay PetscInt j,i,n = N/20,p = N % 20; 417e5c89e4eSSatish Balay PetscTruth iascii,issocket; 418e5c89e4eSSatish Balay MPI_Comm comm; 419e5c89e4eSSatish Balay 420e5c89e4eSSatish Balay PetscFunctionBegin; 421e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 422e5c89e4eSSatish Balay PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,3); 423e5c89e4eSSatish Balay PetscValidIntPointer(idx,2); 424e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 425e5c89e4eSSatish Balay 426e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 427e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 428e5c89e4eSSatish Balay if (iascii) { 429e5c89e4eSSatish Balay for (i=0; i<n; i++) { 430e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 431e5c89e4eSSatish Balay for (j=0; j<20; j++) { 432e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 433e5c89e4eSSatish Balay } 434e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 435e5c89e4eSSatish Balay } 436e5c89e4eSSatish Balay if (p) { 437e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 438e5c89e4eSSatish Balay for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 439e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 440e5c89e4eSSatish Balay } 441e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 442e5c89e4eSSatish Balay #if defined(PETSC_USE_SOCKET_VIEWER) 443e5c89e4eSSatish Balay } else if (issocket) { 444e5c89e4eSSatish Balay PetscMPIInt rank,size,*sizes,Ntotal,*displs, NN = (PetscMPIInt)N; 445e5c89e4eSSatish Balay PetscInt *array; 446e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 447e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 448e5c89e4eSSatish Balay 449e5c89e4eSSatish Balay if (size > 1) { 450e5c89e4eSSatish Balay if (rank) { 451e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 452e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPIU_INT,0,0,0,MPIU_INT,0,comm);CHKERRQ(ierr); 453e5c89e4eSSatish Balay } else { 454e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 455e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPIU_INT,0,comm);CHKERRQ(ierr); 456e5c89e4eSSatish Balay Ntotal = sizes[0]; 457e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 458e5c89e4eSSatish Balay displs[0] = 0; 459e5c89e4eSSatish Balay for (i=1; i<size; i++) { 460e5c89e4eSSatish Balay Ntotal += sizes[i]; 461e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 462e5c89e4eSSatish Balay } 463e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscInt),&array);CHKERRQ(ierr); 464e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 465e5c89e4eSSatish Balay ierr = PetscViewerSocketPutInt(viewer,Ntotal,array);CHKERRQ(ierr); 466e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 467e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 468e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 469e5c89e4eSSatish Balay } 470e5c89e4eSSatish Balay } else { 471e5c89e4eSSatish Balay ierr = PetscViewerSocketPutInt(viewer,N,idx);CHKERRQ(ierr); 472e5c89e4eSSatish Balay } 473e5c89e4eSSatish Balay #endif 474e5c89e4eSSatish Balay } else { 475e5c89e4eSSatish Balay const char *tname; 476e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 477e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 478e5c89e4eSSatish Balay } 479e5c89e4eSSatish Balay PetscFunctionReturn(0); 480e5c89e4eSSatish Balay } 481e5c89e4eSSatish Balay 482e5c89e4eSSatish Balay #undef __FUNCT__ 483e5c89e4eSSatish Balay #define __FUNCT__ "PetscRealView" 484e5c89e4eSSatish Balay /*@C 485e5c89e4eSSatish Balay PetscRealView - Prints an array of doubles; useful for debugging. 486e5c89e4eSSatish Balay 487e5c89e4eSSatish Balay Collective on PetscViewer 488e5c89e4eSSatish Balay 489e5c89e4eSSatish Balay Input Parameters: 490e5c89e4eSSatish Balay + N - number of doubles in array 491e5c89e4eSSatish Balay . idx - array of doubles 492e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 493e5c89e4eSSatish Balay 494e5c89e4eSSatish Balay Level: intermediate 495e5c89e4eSSatish Balay 496e5c89e4eSSatish Balay .seealso: PetscIntView() 497e5c89e4eSSatish Balay @*/ 498e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscRealView(PetscInt N,PetscReal idx[],PetscViewer viewer) 499e5c89e4eSSatish Balay { 500e5c89e4eSSatish Balay PetscErrorCode ierr; 501e5c89e4eSSatish Balay PetscInt j,i,n = N/5,p = N % 5; 502e5c89e4eSSatish Balay PetscTruth iascii,issocket; 503e5c89e4eSSatish Balay MPI_Comm comm; 504e5c89e4eSSatish Balay 505e5c89e4eSSatish Balay PetscFunctionBegin; 506e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 507e5c89e4eSSatish Balay PetscValidHeaderSpecific(viewer,PETSC_VIEWER_COOKIE,3); 508e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 509e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 510e5c89e4eSSatish Balay 511e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 512e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 513e5c89e4eSSatish Balay if (iascii) { 514e5c89e4eSSatish Balay for (i=0; i<n; i++) { 515e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*i);CHKERRQ(ierr); 516e5c89e4eSSatish Balay for (j=0; j<5; j++) { 517e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*5+j]);CHKERRQ(ierr); 518e5c89e4eSSatish Balay } 519e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 520e5c89e4eSSatish Balay } 521e5c89e4eSSatish Balay if (p) { 522e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",5*n);CHKERRQ(ierr); 523e5c89e4eSSatish Balay for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[5*n+i]);CHKERRQ(ierr);} 524e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 525e5c89e4eSSatish Balay } 526e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 527e5c89e4eSSatish Balay #if defined(PETSC_USE_SOCKET_VIEWER) 528e5c89e4eSSatish Balay } else if (issocket) { 529e5c89e4eSSatish Balay PetscMPIInt rank,size,*sizes,*displs, Ntotal,NN = (PetscMPIInt)N; 530e5c89e4eSSatish Balay PetscReal *array; 531e5c89e4eSSatish Balay 532e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 533e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 534e5c89e4eSSatish Balay 535e5c89e4eSSatish Balay if (size > 1) { 536e5c89e4eSSatish Balay if (rank) { 537e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 538e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,0,0,0,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 539e5c89e4eSSatish Balay } else { 540e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 541e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 542e5c89e4eSSatish Balay Ntotal = sizes[0]; 543e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 544e5c89e4eSSatish Balay displs[0] = 0; 545e5c89e4eSSatish Balay for (i=1; i<size; i++) { 546e5c89e4eSSatish Balay Ntotal += sizes[i]; 547e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 548e5c89e4eSSatish Balay } 549e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscReal),&array);CHKERRQ(ierr); 550e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPI_DOUBLE,array,sizes,displs,MPI_DOUBLE,0,comm);CHKERRQ(ierr); 551e5c89e4eSSatish Balay ierr = PetscViewerSocketPutReal(viewer,Ntotal,1,array);CHKERRQ(ierr); 552e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 553e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 554e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 555e5c89e4eSSatish Balay } 556e5c89e4eSSatish Balay } else { 557e5c89e4eSSatish Balay ierr = PetscViewerSocketPutReal(viewer,N,1,idx);CHKERRQ(ierr); 558e5c89e4eSSatish Balay } 559e5c89e4eSSatish Balay #endif 560e5c89e4eSSatish Balay } else { 561e5c89e4eSSatish Balay const char *tname; 562e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 563e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 564e5c89e4eSSatish Balay } 565e5c89e4eSSatish Balay PetscFunctionReturn(0); 566e5c89e4eSSatish Balay } 567e5c89e4eSSatish Balay 568e5c89e4eSSatish Balay #undef __FUNCT__ 569e5c89e4eSSatish Balay #define __FUNCT__ "PetscScalarView" 570e5c89e4eSSatish Balay /*@C 571e5c89e4eSSatish Balay PetscScalarView - Prints an array of scalars; useful for debugging. 572e5c89e4eSSatish Balay 573e5c89e4eSSatish Balay Collective on PetscViewer 574e5c89e4eSSatish Balay 575e5c89e4eSSatish Balay Input Parameters: 576e5c89e4eSSatish Balay + N - number of scalars in array 577e5c89e4eSSatish Balay . idx - array of scalars 578e5c89e4eSSatish Balay - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 579e5c89e4eSSatish Balay 580e5c89e4eSSatish Balay Level: intermediate 581e5c89e4eSSatish Balay 582e5c89e4eSSatish Balay .seealso: PetscIntView(), PetscRealView() 583e5c89e4eSSatish Balay @*/ 584e5c89e4eSSatish Balay PetscErrorCode PETSC_DLLEXPORT PetscScalarView(PetscInt N,PetscScalar idx[],PetscViewer viewer) 585e5c89e4eSSatish Balay { 586e5c89e4eSSatish Balay PetscErrorCode ierr; 587e5c89e4eSSatish Balay PetscInt j,i,n = N/3,p = N % 3; 588e5c89e4eSSatish Balay PetscTruth iascii,issocket; 589e5c89e4eSSatish Balay MPI_Comm comm; 590e5c89e4eSSatish Balay 591e5c89e4eSSatish Balay PetscFunctionBegin; 592e5c89e4eSSatish Balay if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 593e5c89e4eSSatish Balay PetscValidHeader(viewer,3); 594e5c89e4eSSatish Balay PetscValidScalarPointer(idx,2); 595e5c89e4eSSatish Balay ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 596e5c89e4eSSatish Balay 597e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 598e5c89e4eSSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 599e5c89e4eSSatish Balay if (iascii) { 600e5c89e4eSSatish Balay for (i=0; i<n; i++) { 601e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 602e5c89e4eSSatish Balay for (j=0; j<3; j++) { 603e5c89e4eSSatish Balay #if defined (PETSC_USE_COMPLEX) 604e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 605e5c89e4eSSatish Balay PetscRealPart(idx[i*3+j]),PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 606e5c89e4eSSatish Balay #else 607e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[i*3+j]);CHKERRQ(ierr); 608e5c89e4eSSatish Balay #endif 609e5c89e4eSSatish Balay } 610e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 611e5c89e4eSSatish Balay } 612e5c89e4eSSatish Balay if (p) { 613e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 614e5c89e4eSSatish Balay for (i=0; i<p; i++) { 615e5c89e4eSSatish Balay #if defined (PETSC_USE_COMPLEX) 616e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", 617e5c89e4eSSatish Balay PetscRealPart(idx[n*3+i]),PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 618e5c89e4eSSatish Balay #else 619e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",idx[3*n+i]);CHKERRQ(ierr); 620e5c89e4eSSatish Balay #endif 621e5c89e4eSSatish Balay } 622e5c89e4eSSatish Balay ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 623e5c89e4eSSatish Balay } 624e5c89e4eSSatish Balay ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 625e5c89e4eSSatish Balay #if defined(PETSC_USE_SOCKET_VIEWER) 626e5c89e4eSSatish Balay } else if (issocket) { 627e5c89e4eSSatish Balay PetscMPIInt size,rank,*sizes,Ntotal,*displs,NN = (PetscMPIInt)N; 628e5c89e4eSSatish Balay PetscScalar *array; 629e5c89e4eSSatish Balay 630e5c89e4eSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 631e5c89e4eSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 632e5c89e4eSSatish Balay 633e5c89e4eSSatish Balay if (size > 1) { 634e5c89e4eSSatish Balay if (rank) { 635e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,0,0,MPI_INT,0,comm);CHKERRQ(ierr); 636e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,0,0,0,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 637e5c89e4eSSatish Balay } else { 638e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sizes);CHKERRQ(ierr); 639e5c89e4eSSatish Balay ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 640e5c89e4eSSatish Balay Ntotal = sizes[0]; 641e5c89e4eSSatish Balay ierr = PetscMalloc(size*sizeof(PetscMPIInt),&displs);CHKERRQ(ierr); 642e5c89e4eSSatish Balay displs[0] = 0; 643e5c89e4eSSatish Balay for (i=1; i<size; i++) { 644e5c89e4eSSatish Balay Ntotal += sizes[i]; 645e5c89e4eSSatish Balay displs[i] = displs[i-1] + sizes[i-1]; 646e5c89e4eSSatish Balay } 647e5c89e4eSSatish Balay ierr = PetscMalloc(Ntotal*sizeof(PetscScalar),&array);CHKERRQ(ierr); 648e5c89e4eSSatish Balay ierr = MPI_Gatherv(idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 649e5c89e4eSSatish Balay ierr = PetscViewerSocketPutScalar(viewer,Ntotal,1,array);CHKERRQ(ierr); 650e5c89e4eSSatish Balay ierr = PetscFree(sizes);CHKERRQ(ierr); 651e5c89e4eSSatish Balay ierr = PetscFree(displs);CHKERRQ(ierr); 652e5c89e4eSSatish Balay ierr = PetscFree(array);CHKERRQ(ierr); 653e5c89e4eSSatish Balay } 654e5c89e4eSSatish Balay } else { 655e5c89e4eSSatish Balay ierr = PetscViewerSocketPutScalar(viewer,N,1,idx);CHKERRQ(ierr); 656e5c89e4eSSatish Balay } 657e5c89e4eSSatish Balay #endif 658e5c89e4eSSatish Balay } else { 659e5c89e4eSSatish Balay const char *tname; 660e5c89e4eSSatish Balay ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 661e5c89e4eSSatish Balay SETERRQ1(PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 662e5c89e4eSSatish Balay } 663e5c89e4eSSatish Balay PetscFunctionReturn(0); 664e5c89e4eSSatish Balay } 665e5c89e4eSSatish Balay 666e5c89e4eSSatish Balay 667e5c89e4eSSatish Balay 668e5c89e4eSSatish Balay 669