#include <../src/sys/classes/viewer/impls/ascii/asciiimpl.h> /*I "petscviewer.h" I*/ #define QUEUESTRINGSIZE 8192 #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileClose_ASCII" static PetscErrorCode PetscViewerFileClose_ASCII(PetscViewer viewer) { PetscErrorCode ierr; PetscMPIInt rank; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; int err; PetscFunctionBegin; ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr); if (!rank && vascii->fd != stderr && vascii->fd != PETSC_STDOUT) { if (vascii->fd && vascii->closefile) { err = fclose(vascii->fd); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file"); } if (vascii->storecompressed) { char par[PETSC_MAX_PATH_LEN],buf[PETSC_MAX_PATH_LEN]; FILE *fp; ierr = PetscStrcpy(par,"gzip ");CHKERRQ(ierr); ierr = PetscStrcat(par,vascii->filename);CHKERRQ(ierr); #if defined(PETSC_HAVE_POPEN) ierr = PetscPOpen(PETSC_COMM_SELF,NULL,par,"r",&fp);CHKERRQ(ierr); if (fgets(buf,1024,fp)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error from compression command %s\n%s",par,buf); ierr = PetscPClose(PETSC_COMM_SELF,fp,NULL);CHKERRQ(ierr); #else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); #endif } } ierr = PetscFree(vascii->filename);CHKERRQ(ierr); PetscFunctionReturn(0); } /* ----------------------------------------------------------------------*/ #undef __FUNCT__ #define __FUNCT__ "PetscViewerDestroy_ASCII" PetscErrorCode PetscViewerDestroy_ASCII(PetscViewer viewer) { PetscErrorCode ierr; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscViewerLink *vlink; PetscBool flg; PetscFunctionBegin; if (vascii->sviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"ASCII PetscViewer destroyed before restoring singleton or subcomm PetscViewer"); ierr = PetscViewerFileClose_ASCII(viewer);CHKERRQ(ierr); ierr = PetscFree(vascii);CHKERRQ(ierr); /* remove the viewer from the list in the MPI Communicator */ if (Petsc_Viewer_keyval == MPI_KEYVAL_INVALID) { ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,Petsc_DelViewer,&Petsc_Viewer_keyval,(void*)0);CHKERRQ(ierr); } ierr = MPI_Attr_get(PetscObjectComm((PetscObject)viewer),Petsc_Viewer_keyval,(void**)&vlink,(PetscMPIInt*)&flg);CHKERRQ(ierr); if (flg) { if (vlink && vlink->viewer == viewer) { ierr = MPI_Attr_put(PetscObjectComm((PetscObject)viewer),Petsc_Viewer_keyval,vlink->next);CHKERRQ(ierr); ierr = PetscFree(vlink);CHKERRQ(ierr); } else { while (vlink && vlink->next) { if (vlink->next->viewer == viewer) { PetscViewerLink *nv = vlink->next; vlink->next = vlink->next->next; ierr = PetscFree(nv);CHKERRQ(ierr); } vlink = vlink->next; } } } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerDestroy_ASCII_Singleton" PetscErrorCode PetscViewerDestroy_ASCII_Singleton(PetscViewer viewer) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscErrorCode ierr; PetscFunctionBegin; ierr = PetscViewerRestoreSingleton(vascii->bviewer,&viewer);CHKERRQ(ierr); vascii->bviewer = NULL; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerDestroy_ASCII_Subcomm" PetscErrorCode PetscViewerDestroy_ASCII_Subcomm(PetscViewer viewer) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscErrorCode ierr; PetscFunctionBegin; ierr = PetscViewerRestoreSubcomm(vascii->subviewer,PetscObjectComm((PetscObject)viewer),&viewer);CHKERRQ(ierr); vascii->subviewer = NULL; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFlush_ASCII_Singleton_0" PetscErrorCode PetscViewerFlush_ASCII_Singleton_0(PetscViewer viewer) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; int err; PetscFunctionBegin; err = fflush(vascii->fd); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFlush_ASCII" PetscErrorCode PetscViewerFlush_ASCII(PetscViewer viewer) { PetscMPIInt rank; PetscErrorCode ierr; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; int err; PetscFunctionBegin; ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr); /* fflush() fails on OSX for read-only descriptors */ if (!rank && (vascii->mode != FILE_MODE_READ)) { err = fflush(vascii->fd); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() call failed"); } if (vascii->allowsynchronized) { /* Also flush anything printed with PetscViewerASCIISynchronizedPrintf() */ ierr = PetscSynchronizedFlush(PetscObjectComm((PetscObject)viewer),vascii->fd);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIGetPointer" /*@C PetscViewerASCIIGetPointer - Extracts the file pointer from an ASCII PetscViewer. Not Collective + viewer - PetscViewer context, obtained from PetscViewerASCIIOpen() - fd - file pointer Level: intermediate Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewer^file pointer Concepts: file pointer^getting from PetscViewer .seealso: PetscViewerASCIIOpen(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerCreate(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf(), PetscViewerFlush() @*/ PetscErrorCode PetscViewerASCIIGetPointer(PetscViewer viewer,FILE **fd) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; *fd = vascii->fd; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileGetMode_ASCII" PetscErrorCode PetscViewerFileGetMode_ASCII(PetscViewer viewer, PetscFileMode *mode) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; *mode = vascii->mode; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileSetMode_ASCII" PetscErrorCode PetscViewerFileSetMode_ASCII(PetscViewer viewer, PetscFileMode mode) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; vascii->mode = mode; PetscFunctionReturn(0); } /* If petsc_history is on, then all Petsc*Printf() results are saved if the appropriate (usually .petschistory) file. */ extern FILE *petsc_history; #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIISetTab" /*@ PetscViewerASCIISetTab - Causes PetscViewer to tab in a number of times Not Collective, but only first processor in set has any effect Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - tabs - number of tabs Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIGetTab(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer(), PetscViewerASCIIPushTab() @*/ PetscErrorCode PetscViewerASCIISetTab(PetscViewer viewer,PetscInt tabs) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) ascii->tab = tabs; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIGetTab" /*@ PetscViewerASCIIGetTab - Return the number of tabs used by PetscViewer. Not Collective, meaningful on first processor only. Input Parameters: . viewer - optained with PetscViewerASCIIOpen() Output Parameters: . tabs - number of tabs Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^retrieval .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIISetTab(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer(), PetscViewerASCIIPushTab() @*/ PetscErrorCode PetscViewerASCIIGetTab(PetscViewer viewer,PetscInt *tabs) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii && tabs) *tabs = ascii->tab; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIAddTab" /*@ PetscViewerASCIIAddTab - Add to the number of times an ASCII viewer tabs before printing Not Collective, but only first processor in set has any effect Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - tabs - number of tabs Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer(), PetscViewerASCIIPushTab() @*/ PetscErrorCode PetscViewerASCIIAddTab(PetscViewer viewer,PetscInt tabs) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) ascii->tab += tabs; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIISubtractTab" /*@ PetscViewerASCIISubtractTab - Subtracts from the number of times an ASCII viewer tabs before printing Not Collective, but only first processor in set has any effect Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - tabs - number of tabs Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer(), PetscViewerASCIIPushTab() @*/ PetscErrorCode PetscViewerASCIISubtractTab(PetscViewer viewer,PetscInt tabs) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) ascii->tab -= tabs; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIISynchronizedAllow" /*@C PetscViewerASCIISynchronizedAllow - Allows calls to PetscViewerASCIISynchronizedPrintf() for this viewer Collective on PetscViewer Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - allow - PETSC_TRUE to allow the synchronized printing Level: intermediate Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer() @*/ PetscErrorCode PetscViewerASCIISynchronizedAllow(PetscViewer viewer,PetscBool allow) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) ascii->allowsynchronized = allow; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIPushTab" /*@ PetscViewerASCIIPushTab - Adds one more tab to the amount that PetscViewerASCIIPrintf() lines are tabbed. Not Collective, but only first processor in set has any effect Input Parameters: . viewer - optained with PetscViewerASCIIOpen() Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer() @*/ PetscErrorCode PetscViewerASCIIPushTab(PetscViewer viewer) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) ascii->tab++; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIPopTab" /*@ PetscViewerASCIIPopTab - Removes one tab from the amount that PetscViewerASCIIPrintf() lines are tabbed. Not Collective, but only first processor in set has any effect Input Parameters: . viewer - optained with PetscViewerASCIIOpen() Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPushTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer() @*/ PetscErrorCode PetscViewerASCIIPopTab(PetscViewer viewer) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscErrorCode ierr; PetscBool iascii; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) { if (ascii->tab <= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"More tabs popped than pushed"); ascii->tab--; } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIUseTabs" /*@ PetscViewerASCIIUseTabs - Turns on or off the use of tabs with the ASCII PetscViewer Not Collective, but only first processor in set has any effect Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - flg - PETSC_TRUE or PETSC_FALSE Level: developer Fortran Note: This routine is not supported in Fortran. Concepts: PetscViewerASCII^formating Concepts: tab^setting .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIPrintf(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerASCIIPushTab(), PetscViewerASCIIOpen(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer() @*/ PetscErrorCode PetscViewerASCIIUseTabs(PetscViewer viewer,PetscBool flg) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscBool iascii; PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (iascii) { if (flg) ascii->tab = ascii->tab_store; else { ascii->tab_store = ascii->tab; ascii->tab = 0; } } PetscFunctionReturn(0); } /* ----------------------------------------------------------------------- */ #include <../src/sys/fileio/mprint.h> /* defines the queue datastructures and variables */ #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIPrintf" /*@C PetscViewerASCIIPrintf - Prints to a file, only from the first processor in the PetscViewer Not Collective, but only first processor in set has any effect Input Parameters: + viewer - optained with PetscViewerASCIIOpen() - format - the usual printf() format string Level: developer Fortran Note: The call sequence is PetscViewerASCIIPrintf(PetscViewer, character(*), int ierr) from Fortran. That is, you can only pass a single character string from Fortran. Concepts: PetscViewerASCII^printing Concepts: printing^to file Concepts: printf .seealso: PetscPrintf(), PetscSynchronizedPrintf(), PetscViewerASCIIOpen(), PetscViewerASCIIPushTab(), PetscViewerASCIIPopTab(), PetscViewerASCIISynchronizedPrintf(), PetscViewerCreate(), PetscViewerDestroy(), PetscViewerSetType(), PetscViewerASCIIGetPointer(), PetscViewerASCIISynchronizedAllow() @*/ PetscErrorCode PetscViewerASCIIPrintf(PetscViewer viewer,const char format[],...) { PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscMPIInt rank; PetscInt tab,intab = ascii->tab; PetscErrorCode ierr; FILE *fd = ascii->fd; PetscBool iascii,issingleton = PETSC_FALSE; int err; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); PetscValidCharPointer(format,2); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (!iascii) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Not ASCII PetscViewer"); if (ascii->bviewer) { viewer = ascii->bviewer; ascii = (PetscViewer_ASCII*)viewer->data; fd = ascii->fd; issingleton = PETSC_TRUE; } ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr); if (!rank) { va_list Argp; tab = intab; while (tab--) { ierr = PetscFPrintf(PETSC_COMM_SELF,fd," ");CHKERRQ(ierr); } va_start(Argp,format); ierr = (*PetscVFPrintf)(fd,format,Argp);CHKERRQ(ierr); err = fflush(fd); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); if (petsc_history) { va_start(Argp,format); tab = intab; while (tab--) { ierr = PetscFPrintf(PETSC_COMM_SELF,petsc_history," ");CHKERRQ(ierr); } ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr); err = fflush(petsc_history); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); } va_end(Argp); } else if (issingleton) { char *string; va_list Argp; size_t fullLength; PrintfQueue next; ierr = PetscNew(&next);CHKERRQ(ierr); if (petsc_printfqueue) { petsc_printfqueue->next = next; petsc_printfqueue = next; } else { petsc_printfqueuebase = petsc_printfqueue = next; } petsc_printfqueuelength++; next->size = QUEUESTRINGSIZE; ierr = PetscCalloc1(next->size, &next->string);CHKERRQ(ierr); string = next->string; tab = intab; tab *= 2; while (tab--) { *string++ = ' '; } va_start(Argp,format); ierr = PetscVSNPrintf(string,next->size-2*ascii->tab,format,&fullLength,Argp);CHKERRQ(ierr); va_end(Argp); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileSetName" /*@C PetscViewerFileSetName - Sets the name of the file the PetscViewer uses. Collective on PetscViewer Input Parameters: + viewer - the PetscViewer; either ASCII or binary - name - the name of the file it should use Level: advanced .seealso: PetscViewerCreate(), PetscViewerSetType(), PetscViewerASCIIOpen(), PetscViewerBinaryOpen(), PetscViewerDestroy(), PetscViewerASCIIGetPointer(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedPrintf() @*/ PetscErrorCode PetscViewerFileSetName(PetscViewer viewer,const char name[]) { PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); PetscValidCharPointer(name,2); ierr = PetscTryMethod(viewer,"PetscViewerFileSetName_C",(PetscViewer,const char[]),(viewer,name));CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileGetName" /*@C PetscViewerFileGetName - Gets the name of the file the PetscViewer uses. Not Collective Input Parameter: . viewer - the PetscViewer; either ASCII or binary Output Parameter: . name - the name of the file it is using Level: advanced .seealso: PetscViewerCreate(), PetscViewerSetType(), PetscViewerASCIIOpen(), PetscViewerBinaryOpen(), PetscViewerFileSetName() @*/ PetscErrorCode PetscViewerFileGetName(PetscViewer viewer,const char **name) { PetscErrorCode ierr; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); ierr = PetscTryMethod(viewer,"PetscViewerFileGetName_C",(PetscViewer,const char**),(viewer,name));CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileGetName_ASCII" PetscErrorCode PetscViewerFileGetName_ASCII(PetscViewer viewer,const char **name) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; *name = vascii->filename; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerFileSetName_ASCII" PetscErrorCode PetscViewerFileSetName_ASCII(PetscViewer viewer,const char name[]) { PetscErrorCode ierr; size_t len; char fname[PETSC_MAX_PATH_LEN],*gz; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscBool isstderr,isstdout; PetscMPIInt rank; PetscFunctionBegin; ierr = PetscViewerFileClose_ASCII(viewer);CHKERRQ(ierr); if (!name) PetscFunctionReturn(0); ierr = PetscStrallocpy(name,&vascii->filename);CHKERRQ(ierr); /* Is this file to be compressed */ vascii->storecompressed = PETSC_FALSE; ierr = PetscStrstr(vascii->filename,".gz",&gz);CHKERRQ(ierr); if (gz) { ierr = PetscStrlen(gz,&len);CHKERRQ(ierr); if (len == 3) { *gz = 0; vascii->storecompressed = PETSC_TRUE; } } ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr); if (!rank) { ierr = PetscStrcmp(name,"stderr",&isstderr);CHKERRQ(ierr); ierr = PetscStrcmp(name,"stdout",&isstdout);CHKERRQ(ierr); /* empty filename means stdout */ if (name[0] == 0) isstdout = PETSC_TRUE; if (isstderr) vascii->fd = PETSC_STDERR; else if (isstdout) vascii->fd = PETSC_STDOUT; else { ierr = PetscFixFilename(name,fname);CHKERRQ(ierr); switch (vascii->mode) { case FILE_MODE_READ: vascii->fd = fopen(fname,"r"); break; case FILE_MODE_WRITE: vascii->fd = fopen(fname,"w"); break; case FILE_MODE_APPEND: vascii->fd = fopen(fname,"a"); break; case FILE_MODE_UPDATE: vascii->fd = fopen(fname,"r+"); if (!vascii->fd) vascii->fd = fopen(fname,"w+"); break; case FILE_MODE_APPEND_UPDATE: /* I really want a file which is opened at the end for updating, not a+, which opens at the beginning, but makes writes at the end. */ vascii->fd = fopen(fname,"r+"); if (!vascii->fd) vascii->fd = fopen(fname,"w+"); else { ierr = fseek(vascii->fd, 0, SEEK_END);CHKERRQ(ierr); } break; default: SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG, "Invalid file mode %d", vascii->mode); } if (!vascii->fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open PetscViewer file: %s",fname); } } #if defined(PETSC_USE_LOG) PetscLogObjectState((PetscObject)viewer,"File: %s",name); #endif PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerGetSingleton_ASCII" PetscErrorCode PetscViewerGetSingleton_ASCII(PetscViewer viewer,PetscViewer *outviewer) { PetscMPIInt rank; PetscErrorCode ierr; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data,*ovascii; const char *name; PetscFunctionBegin; if (vascii->sviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Singleton already obtained from PetscViewer and not restored"); ierr = PetscViewerCreate(PETSC_COMM_SELF,outviewer);CHKERRQ(ierr); ierr = PetscViewerSetType(*outviewer,PETSCVIEWERASCII);CHKERRQ(ierr); ovascii = (PetscViewer_ASCII*)(*outviewer)->data; ovascii->fd = vascii->fd; ovascii->tab = vascii->tab; vascii->sviewer = *outviewer; (*outviewer)->format = viewer->format; ierr = PetscObjectGetName((PetscObject)viewer,&name);CHKERRQ(ierr); ierr = PetscObjectSetName((PetscObject)(*outviewer),name);CHKERRQ(ierr); ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)viewer),&rank);CHKERRQ(ierr); ((PetscViewer_ASCII*)((*outviewer)->data))->bviewer = viewer; (*outviewer)->ops->destroy = PetscViewerDestroy_ASCII_Singleton; if (rank) (*outviewer)->ops->flush = 0; else (*outviewer)->ops->flush = PetscViewerFlush_ASCII_Singleton_0; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerRestoreSingleton_ASCII" PetscErrorCode PetscViewerRestoreSingleton_ASCII(PetscViewer viewer,PetscViewer *outviewer) { PetscErrorCode ierr; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)(*outviewer)->data; PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; if (!ascii->sviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Singleton never obtained from PetscViewer"); if (ascii->sviewer != *outviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"This PetscViewer did not generate singleton"); ascii->sviewer = 0; vascii->fd = PETSC_STDOUT; (*outviewer)->ops->destroy = PetscViewerDestroy_ASCII; ierr = PetscViewerDestroy(outviewer);CHKERRQ(ierr); ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerGetSubcomm_ASCII" PetscErrorCode PetscViewerGetSubcomm_ASCII(PetscViewer viewer,MPI_Comm subcomm,PetscViewer *outviewer) { PetscErrorCode ierr; PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data,*ovascii; const char *name; PetscFunctionBegin; if (vascii->sviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Subcomm viewer already obtained from PetscViewer and not restored"); /* Note that we need to open vascii->filename for the subcomm: we can't count on reusing viewer's fd since the root in comm and subcomm may differ. Further, if the subcomm happens to be the same as comm, PetscViewerASCIIOpen() will will return the current viewer, having increfed it. */ ierr = PetscViewerASCIIOpen(subcomm,vascii->filename, outviewer);CHKERRQ(ierr); if (*outviewer == viewer) PetscFunctionReturn(0); ovascii = (PetscViewer_ASCII*)(*outviewer)->data; ovascii->tab = vascii->tab; vascii->sviewer = *outviewer; (*outviewer)->format = viewer->format; ierr = PetscObjectGetName((PetscObject)viewer,&name);CHKERRQ(ierr); ierr = PetscObjectSetName((PetscObject)(*outviewer),name);CHKERRQ(ierr); ((PetscViewer_ASCII*)((*outviewer)->data))->subviewer = viewer; (*outviewer)->ops->destroy = PetscViewerDestroy_ASCII_Subcomm; PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerRestoreSubcomm_ASCII" PetscErrorCode PetscViewerRestoreSubcomm_ASCII(PetscViewer viewer,MPI_Comm subcomm,PetscViewer *outviewer) { PetscErrorCode ierr; PetscViewer_ASCII *oascii = (PetscViewer_ASCII*)(*outviewer)->data; PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)viewer->data; PetscFunctionBegin; if (!ascii->sviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Subcomm never obtained from PetscViewer"); if (ascii->sviewer != *outviewer) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"The given PetscViewer did not generate this subcomm viewer"); ascii->sviewer = 0; oascii->fd = PETSC_STDOUT; (*outviewer)->ops->destroy = PetscViewerDestroy_ASCII; ierr = PetscViewerDestroy(outviewer);CHKERRQ(ierr); ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerView_ASCII" PetscErrorCode PetscViewerView_ASCII(PetscViewer v,PetscViewer viewer) { PetscErrorCode ierr; PetscViewer_ASCII *ascii = (PetscViewer_ASCII*)v->data; PetscFunctionBegin; if (ascii->filename) { ierr = PetscViewerASCIIPrintf(viewer,"Filename: %s\n",ascii->filename);CHKERRQ(ierr); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerCreate_ASCII" PETSC_EXTERN PetscErrorCode PetscViewerCreate_ASCII(PetscViewer viewer) { PetscViewer_ASCII *vascii; PetscErrorCode ierr; PetscFunctionBegin; ierr = PetscNewLog(viewer,&vascii);CHKERRQ(ierr); viewer->data = (void*)vascii; viewer->ops->destroy = PetscViewerDestroy_ASCII; viewer->ops->flush = PetscViewerFlush_ASCII; viewer->ops->getsingleton = PetscViewerGetSingleton_ASCII; viewer->ops->restoresingleton = PetscViewerRestoreSingleton_ASCII; viewer->ops->getsubcomm = PetscViewerGetSubcomm_ASCII; viewer->ops->restoresubcomm = PetscViewerRestoreSubcomm_ASCII; viewer->ops->view = PetscViewerView_ASCII; viewer->ops->read = PetscViewerASCIIRead; /* defaults to stdout unless set with PetscViewerFileSetName() */ vascii->fd = PETSC_STDOUT; vascii->mode = FILE_MODE_WRITE; vascii->bviewer = 0; vascii->subviewer = 0; vascii->sviewer = 0; vascii->tab = 0; vascii->tab_store = 0; vascii->filename = 0; vascii->closefile = PETSC_TRUE; ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetName_C",PetscViewerFileSetName_ASCII);CHKERRQ(ierr); ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetName_C",PetscViewerFileGetName_ASCII);CHKERRQ(ierr); ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileGetMode_C",PetscViewerFileGetMode_ASCII);CHKERRQ(ierr); ierr = PetscObjectComposeFunction((PetscObject)viewer,"PetscViewerFileSetMode_C",PetscViewerFileSetMode_ASCII);CHKERRQ(ierr); PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIISynchronizedPrintf" /*@C PetscViewerASCIISynchronizedPrintf - Prints synchronized output to the specified file from several processors. Output of the first processor is followed by that of the second, etc. Not Collective, must call collective PetscViewerFlush() to get the results out Input Parameters: + viewer - the ASCII PetscViewer - format - the usual printf() format string Level: intermediate Notes: You must have previously called PetscViewerASCIISynchronizeAllow() to allow this routine to be called. Fortran Note: Can only print a single character* string .seealso: PetscSynchronizedPrintf(), PetscSynchronizedFlush(), PetscFPrintf(), PetscFOpen(), PetscViewerFlush(), PetscViewerASCIIGetPointer(), PetscViewerDestroy(), PetscViewerASCIIOpen(), PetscViewerASCIIPrintf(), PetscViewerASCIISynchronizedAllow() @*/ PetscErrorCode PetscViewerASCIISynchronizedPrintf(PetscViewer viewer,const char format[],...) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; PetscErrorCode ierr; PetscMPIInt rank,size; PetscInt tab = vascii->tab; MPI_Comm comm; FILE *fp; PetscBool iascii; int err; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); PetscValidCharPointer(format,2); ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); if (!iascii) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Not ASCII PetscViewer"); ierr = MPI_Comm_size(PetscObjectComm((PetscObject)viewer),&size);CHKERRQ(ierr); if (size > 1 && !vascii->allowsynchronized) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"First call PetscViewerASCIISynchronizedAllow() to allow this call"); if (!viewer->ops->flush) PetscFunctionReturn(0); /* This viewer obtained via PetscViewerGetSubcomm_ASCII(), should not participate. */ ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); fp = vascii->fd; ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); /* First processor prints immediately to fp */ if (!rank) { va_list Argp; while (tab--) { ierr = PetscFPrintf(PETSC_COMM_SELF,fp," ");CHKERRQ(ierr); } va_start(Argp,format); ierr = (*PetscVFPrintf)(fp,format,Argp);CHKERRQ(ierr); err = fflush(fp); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); if (petsc_history) { va_start(Argp,format); ierr = (*PetscVFPrintf)(petsc_history,format,Argp);CHKERRQ(ierr); err = fflush(petsc_history); if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); } va_end(Argp); } else { /* other processors add to local queue */ char *string; va_list Argp; size_t fullLength; PrintfQueue next; ierr = PetscNew(&next);CHKERRQ(ierr); if (petsc_printfqueue) { petsc_printfqueue->next = next; petsc_printfqueue = next; } else { petsc_printfqueuebase = petsc_printfqueue = next; } petsc_printfqueuelength++; next->size = QUEUESTRINGSIZE; ierr = PetscMalloc1(next->size, &next->string);CHKERRQ(ierr); ierr = PetscMemzero(next->string,next->size);CHKERRQ(ierr); string = next->string; tab *= 2; while (tab--) { *string++ = ' '; } va_start(Argp,format); ierr = PetscVSNPrintf(string,next->size-2*vascii->tab,format,&fullLength,Argp);CHKERRQ(ierr); va_end(Argp); } PetscFunctionReturn(0); } #undef __FUNCT__ #define __FUNCT__ "PetscViewerASCIIRead" /*@C PetscViewerASCIIRead - Reads from am ASCII file Collective on MPI_Comm Input Parameters: + viewer - the ascii viewer . data - location to write the data . num - number of items of data to read - datatype - type of data to read Output Parameters: . count - number of items of data actually read, or NULL Level: beginner Concepts: ascii files .seealso: PetscViewerASCIIOpen(), PetscViewerSetFormat(), PetscViewerDestroy(), VecView(), MatView(), VecLoad(), MatLoad(), PetscViewerBinaryGetDescriptor(), PetscViewerBinaryGetInfoPointer(), PetscFileMode, PetscViewer, PetscBinaryViewerRead() @*/ PetscErrorCode PetscViewerASCIIRead(PetscViewer viewer,void *data,PetscInt num,PetscInt *count,PetscDataType dtype) { PetscViewer_ASCII *vascii = (PetscViewer_ASCII*)viewer->data; FILE *fd = vascii->fd; PetscInt i; int ret = 0; PetscFunctionBegin; PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1); for (i=0; i