Actual source code: stringv.c
petsc-3.6.1 2015-08-06
2: #include <petsc/private/viewerimpl.h> /*I "petscsys.h" I*/
4: typedef struct {
5: char *string; /* string where info is stored */
6: char *head; /* pointer to begining of unused portion */
7: size_t curlen,maxlen;
8: } PetscViewer_String;
12: static PetscErrorCode PetscViewerDestroy_String(PetscViewer viewer)
13: {
14: PetscViewer_String *vstr = (PetscViewer_String*)viewer->data;
15: PetscErrorCode ierr;
18: PetscFree(vstr);
19: return(0);
20: }
24: /*@C
25: PetscViewerStringSPrintf - Prints information to a PetscViewer string.
27: Logically Collective on PetscViewer (Hmmm, each processor maintains a separate string)
29: Input Parameters:
30: + v - a string PetscViewer, formed by PetscViewerStringOpen()
31: - format - the format of the input
33: Level: developer
35: Fortran Note:
36: This routine is not supported in Fortran.
38: Concepts: printing^to string
40: .seealso: PetscViewerStringOpen()
41: @*/
42: PetscErrorCode PetscViewerStringSPrintf(PetscViewer viewer,const char format[],...)
43: {
44: va_list Argp;
45: size_t fullLength;
46: size_t shift,cshift;
47: PetscErrorCode ierr;
48: PetscBool isstring;
49: char tmp[4096];
50: PetscViewer_String *vstr = (PetscViewer_String*)viewer->data;
55: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);
56: if (!isstring) return(0);
57: if (!vstr->string) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call PetscViewerStringSetString() before using");
59: va_start(Argp,format);
60: PetscVSNPrintf(tmp,4096,format,&fullLength,Argp);
61: va_end(Argp);
62: PetscStrlen(tmp,&shift);
63: cshift = shift+1;
64: if (cshift >= vstr->maxlen - vstr->curlen - 1) cshift = vstr->maxlen - vstr->curlen - 1;
65: PetscStrncpy(vstr->head,tmp,cshift);
66: vstr->head += shift;
67: vstr->curlen += shift;
68: return(0);
69: }
73: /*@C
74: PetscViewerStringOpen - Opens a string as a PetscViewer. This is a very
75: simple PetscViewer; information on the object is simply stored into
76: the string in a fairly nice way.
78: Collective on MPI_Comm
80: Input Parameters:
81: + comm - the communicator
82: . string - the string to use
83: - len - the string length
85: Output Parameter:
86: . lab - the PetscViewer
88: Level: advanced
90: Fortran Note:
91: This routine is not supported in Fortran.
93: Concepts: PetscViewerString^creating
95: .seealso: PetscViewerDestroy(), PetscViewerStringSPrintf()
96: @*/
97: PetscErrorCode PetscViewerStringOpen(MPI_Comm comm,char string[],size_t len,PetscViewer *lab)
98: {
102: PetscViewerCreate(comm,lab);
103: PetscViewerSetType(*lab,PETSCVIEWERSTRING);
104: PetscViewerStringSetString(*lab,string,len);
105: return(0);
106: }
110: PetscErrorCode PetscViewerGetSingleton_String(PetscViewer viewer,PetscViewer *sviewer)
111: {
112: PetscViewer_String *vstr = (PetscViewer_String*)viewer->data;
113: PetscErrorCode ierr;
116: PetscViewerStringOpen(PETSC_COMM_SELF,vstr->head,vstr->maxlen-vstr->curlen,sviewer);
117: return(0);
118: }
122: PetscErrorCode PetscViewerRestoreSingleton_String(PetscViewer viewer,PetscViewer *sviewer)
123: {
124: PetscErrorCode ierr;
125: PetscViewer_String *iviewer = (PetscViewer_String*)(*sviewer)->data;
126: PetscViewer_String *vstr = (PetscViewer_String*)viewer->data;
129: vstr->head = iviewer->head;
130: vstr->curlen += iviewer->curlen;
131: PetscViewerDestroy(sviewer);
132: return(0);
133: }
137: PETSC_EXTERN PetscErrorCode PetscViewerCreate_String(PetscViewer v)
138: {
139: PetscViewer_String *vstr;
140: PetscErrorCode ierr;
143: v->ops->destroy = PetscViewerDestroy_String;
144: v->ops->view = 0;
145: v->ops->flush = 0;
146: v->ops->getsingleton = PetscViewerGetSingleton_String;
147: v->ops->restoresingleton = PetscViewerRestoreSingleton_String;
148: PetscNewLog(v,&vstr);
149: v->data = (void*)vstr;
150: vstr->string = 0;
151: return(0);
152: }
156: /*@C
158: PetscViewerStringSetString - sets the string that a string viewer will print to
160: Logically Collective on PetscViewer
162: Input Parameters:
163: + viewer - string viewer you wish to attach string to
164: . string - the string to print data into
165: - len - the length of the string
167: Level: advanced
169: .seealso: PetscViewerStringOpen()
170: @*/
171: PetscErrorCode PetscViewerStringSetString(PetscViewer viewer,char string[],PetscInt len)
172: {
173: PetscViewer_String *vstr = (PetscViewer_String*)viewer->data;
174: PetscErrorCode ierr;
175: PetscBool isstring;
180: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);
181: if (!isstring) return(0);
182: if (len <= 2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"String must have length at least 2");
184: PetscMemzero(string,len*sizeof(char));
185: vstr->string = string;
186: vstr->head = string;
187: vstr->curlen = 0;
188: vstr->maxlen = len;
189: return(0);
190: }