Actual source code: viewreg.c
petsc-3.12.5 2020-03-29
2: #include <petsc/private/viewerimpl.h>
3: #include <petsc/private/hashtable.h>
4: #if defined(PETSC_HAVE_SAWS)
5: #include <petscviewersaws.h>
6: #endif
8: PetscFunctionList PetscViewerList = 0;
11: PetscOptionsHelpPrinted PetscOptionsHelpPrintedSingleton = NULL;
12: KHASH_SET_INIT_STR(HTPrinted)
13: struct _n_PetscOptionsHelpPrinted{
14: khash_t(HTPrinted) *printed;
15: PetscSegBuffer strings;
16: };
18: PetscErrorCode PetscOptionsHelpPrintedDestroy(PetscOptionsHelpPrinted *hp)
19: {
23: if (!*hp) return(0);
24: kh_destroy(HTPrinted,(*hp)->printed);
25: PetscSegBufferDestroy(&(*hp)->strings);
26: PetscFree(*hp);
27: return(0);
28: }
30: /*@C
31: PetscOptionsHelpPrintedCreate - Creates an object used to manage tracking which help messages have
32: been printed so they will not be printed again.
34: Not collective
36: Level: developer
38: .seealso: PetscOptionsHelpPrintedCheck(), PetscOptionsHelpPrintChecked()
39: @*/
40: PetscErrorCode PetscOptionsHelpPrintedCreate(PetscOptionsHelpPrinted *hp)
41: {
42: PetscErrorCode ierr;
45: PetscNew(hp);
46: (*hp)->printed = kh_init(HTPrinted);
47: PetscSegBufferCreate(sizeof(char),10000,&(*hp)->strings);
48: return(0);
49: }
51: /*@C
52: PetscOptionsHelpPrintedCheck - Checks if a particular pre, name pair has previous been entered (meaning the help message was printed)
54: Not collective
56: Input Parameters:
57: + hp - the object used to manage tracking what help messages have been printed
58: . pre - the prefix part of the string, many be NULL
59: - name - the string to look for (cannot be NULL)
61: Output Parameter:
62: . found - PETSC_TRUE if the string was already set
64: Level: intermediate
67: .seealso: PetscOptionsHelpPrintedCreate()
68: @*/
69: PetscErrorCode PetscOptionsHelpPrintedCheck(PetscOptionsHelpPrinted hp,const char *pre,const char* name,PetscBool *found)
70: {
71: size_t l1,l2;
72: #if !defined(PETSC_HAVE_THREADSAFETY)
73: char *both;
74: int newitem;
75: #endif
76: PetscErrorCode ierr;
79: PetscStrlen(pre,&l1);
80: PetscStrlen(name,&l2);
81: if (l1+l2 == 0) {
82: *found = PETSC_FALSE;
83: return(0);
84: }
85: #if !defined(PETSC_HAVE_THREADSAFETY)
86: PetscSegBufferGet(hp->strings,l1+l2+1,&both);
87: PetscStrcpy(both,pre);
88: PetscStrcat(both,name);
89: kh_put(HTPrinted,hp->printed,both,&newitem);
90: if (!newitem) {
91: PetscSegBufferUnuse(hp->strings,l1+l2+1);
92: }
93: *found = newitem ? PETSC_FALSE : PETSC_TRUE;
94: #else
95: *found = PETSC_FALSE;
96: #endif
97: return(0);
98: }
100: static PetscBool noviewer = PETSC_FALSE;
101: static PetscBool noviewers[PETSCVIEWERGETVIEWEROFFPUSHESMAX];
102: static PetscInt inoviewers = 0;
104: /*@
105: PetscOptionsPushGetViewerOff - control whether PetscOptionsGetViewer returns a viewer.
107: Logically Collective
109: Input Parameter:
110: . flg - PETSC_TRUE to turn off viewer creation, PETSC_FALSE to turn it on.
112: Level: developer
114: Notes:
115: Calling XXXViewFromOptions in an inner loop can be very expensive. This can appear, for example, when using
116: many small subsolves. Call this function to control viewer creation in PetscOptionsGetViewer, thus removing the expensive XXXViewFromOptions calls.
118: .seealso: PetscOptionsGetViewer(), PetscOptionsPopGetViewerOff()
119: @*/
120: PetscErrorCode PetscOptionsPushGetViewerOff(PetscBool flg)
121: {
123: if (inoviewers > PETSCVIEWERGETVIEWEROFFPUSHESMAX - 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many PetscOptionsPushGetViewerOff(), perhaps you forgot PetscOptionsPopGetViewerOff()?");
125: noviewers[inoviewers++] = noviewer;
126: noviewer = flg;
127: return(0);
128: }
130: /*@
131: PetscOptionsPopGetViewerOff - reset whether PetscOptionsGetViewer returns a viewer.
133: Logically Collective
135: Level: developer
137: Notes:
138: Calling XXXViewFromOptions in an inner loop can be very expensive. This can appear, for example, when using
139: many small subsolves. Call this function to control viewer creation in PetscOptionsGetViewer, thus removing the expensive XXXViewFromOptions calls.
141: .seealso: PetscOptionsGetViewer(), PetscOptionsPushGetViewerOff()
142: @*/
143: PetscErrorCode PetscOptionsPopGetViewerOff(void)
144: {
146: if (!inoviewers) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many PetscOptionsPopGetViewerOff(), perhaps you forgot PetscOptionsPushGetViewerOff()?");
147: noviewer = noviewers[--inoviewers];
148: return(0);
149: }
151: /*@
152: PetscOptionsGetViewerOff - does PetscOptionsGetViewer return a viewer?
154: Logically Collective
156: Output Parameter:
157: . flg - whether viewers are returned.
159: Level: developer
161: Notes:
162: Calling XXXViewFromOptions in an inner loop can be very expensive. This can appear, for example, when using
163: many small subsolves.
165: .seealso: PetscOptionsGetViewer(), PetscOptionsPushGetViewerOff(), PetscOptionsPopGetViewerOff()
166: @*/
167: PetscErrorCode PetscOptionsGetViewerOff(PetscBool *flg)
168: {
171: *flg = noviewer;
172: return(0);
173: }
175: /*@C
176: PetscOptionsGetViewer - Gets a viewer appropriate for the type indicated by the user
178: Collective
180: Input Parameters:
181: + comm - the communicator to own the viewer
182: . pre - the string to prepend to the name or NULL
183: - name - the option one is seeking
185: Output Parameter:
186: + viewer - the viewer, pass NULL if not needed
187: . format - the PetscViewerFormat requested by the user, pass NULL if not needed
188: - set - PETSC_TRUE if found, else PETSC_FALSE
190: Level: intermediate
192: Notes:
193: If no value is provided ascii:stdout is used
194: $ ascii[:[filename][:[format][:append]]] defaults to stdout - format can be one of ascii_info, ascii_info_detail, or ascii_matlab,
195: for example ascii::ascii_info prints just the information about the object not all details
196: unless :append is given filename opens in write mode, overwriting what was already there
197: $ binary[:[filename][:[format][:append]]] defaults to the file binaryoutput
198: $ draw[:drawtype[:filename]] for example, draw:tikz, draw:tikz:figure.tex or draw:x
199: $ socket[:port] defaults to the standard output port
200: $ saws[:communicatorname] publishes object to the Scientific Application Webserver (SAWs)
202: Use PetscViewerDestroy() after using the viewer, otherwise a memory leak will occur
204: You can control whether calls to this function create a viewer (or return early with *set of PETSC_FALSE) with
205: PetscOptionsPushGetViewerOff. This is useful if calling many small subsolves, in which case XXXViewFromOptions can take
206: an appreciable fraction of the runtime.
208: If PETSc is configured with --with-viewfromoptions=0 this function always returns with *set of PETSC_FALSE
210: .seealso: PetscOptionsGetReal(), PetscOptionsHasName(), PetscOptionsGetString(),
211: PetscOptionsGetIntArray(), PetscOptionsGetRealArray(), PetscOptionsBool()
212: PetscOptionsInt(), PetscOptionsString(), PetscOptionsReal(), PetscOptionsBool(),
213: PetscOptionsName(), PetscOptionsBegin(), PetscOptionsEnd(), PetscOptionsHead(),
214: PetscOptionsStringArray(),PetscOptionsRealArray(), PetscOptionsScalar(),
215: PetscOptionsBoolGroupBegin(), PetscOptionsBoolGroup(), PetscOptionsBoolGroupEnd(),
216: PetscOptionsFList(), PetscOptionsEList(), PetscOptionsPushGetViewerOff(), PetscOptionsPopGetViewerOff(),
217: PetscOptionsGetViewerOff()
218: @*/
219: PetscErrorCode PetscOptionsGetViewer(MPI_Comm comm,PetscOptions options,const char pre[],const char name[],PetscViewer *viewer,PetscViewerFormat *format,PetscBool *set)
220: {
221: const char *value;
222: PetscErrorCode ierr;
223: PetscBool flag,hashelp;
228: if (viewer) *viewer = NULL;
229: if (format) *format = PETSC_VIEWER_DEFAULT;
230: if (set) *set = PETSC_FALSE;
231: PetscOptionsGetViewerOff(&flag);
232: if (flag) return(0);
234: PetscOptionsHasHelp(NULL,&hashelp);
235: if (hashelp) {
236: PetscBool found;
238: if (!PetscOptionsHelpPrintedSingleton) {
239: PetscOptionsHelpPrintedCreate(&PetscOptionsHelpPrintedSingleton);
240: }
241: PetscOptionsHelpPrintedCheck(PetscOptionsHelpPrintedSingleton,pre,name,&found);
242: if (!found) {
243: if (viewer) {
244: (*PetscHelpPrintf)(comm,"\n -%s%s ascii[:[filename][:[format][:append]]]: %s (%s)\n",pre ? pre : "",name+1,"Prints object to stdout or ASCII file","PetscOptionsGetViewer");
245: (*PetscHelpPrintf)(comm," -%s%s binary[:[filename][:[format][:append]]]: %s (%s)\n",pre ? pre : "",name+1,"Saves object to a binary file","PetscOptionsGetViewer");
246: (*PetscHelpPrintf)(comm," -%s%s draw[:[drawtype][:filename|format]] %s (%s)\n",pre ? pre : "",name+1,"Draws object","PetscOptionsGetViewer");
247: (*PetscHelpPrintf)(comm," -%s%s socket[:port]: %s (%s)\n",pre ? pre : "",name+1,"Pushes object to a Unix socket","PetscOptionsGetViewer");
248: (*PetscHelpPrintf)(comm," -%s%s saws[:communicatorname]: %s (%s)\n\n",pre ? pre : "",name+1,"Publishes object to SAWs","PetscOptionsGetViewer");
249: } else {
250: (*PetscHelpPrintf)(comm," -%s%s\n",pre ? pre : "",name+1);
251: }
252: }
253: }
255: if (format) *format = PETSC_VIEWER_DEFAULT;
256: PetscOptionsFindPair(options,pre,name,&value,&flag);
257: if (flag) {
258: if (set) *set = PETSC_TRUE;
259: if (!value) {
260: if (viewer) {
261: PetscViewerASCIIGetStdout(comm,viewer);
262: PetscObjectReference((PetscObject)*viewer);
263: }
264: } else {
265: char *loc0_vtype,*loc1_fname,*loc2_fmt = NULL,*loc3_fmode = NULL;
266: PetscInt cnt;
267: const char *viewers[] = {PETSCVIEWERASCII,PETSCVIEWERBINARY,PETSCVIEWERDRAW,PETSCVIEWERSOCKET,PETSCVIEWERMATLAB,PETSCVIEWERSAWS,PETSCVIEWERVTK,PETSCVIEWERHDF5,PETSCVIEWERGLVIS,0};
269: PetscStrallocpy(value,&loc0_vtype);
270: PetscStrchr(loc0_vtype,':',&loc1_fname);
271: if (loc1_fname) {
272: *loc1_fname++ = 0;
273: PetscStrchr(loc1_fname,':',&loc2_fmt);
274: }
275: if (loc2_fmt) {
276: *loc2_fmt++ = 0;
277: PetscStrchr(loc2_fmt,':',&loc3_fmode);
278: }
279: if (loc3_fmode) *loc3_fmode++ = 0;
280: PetscStrendswithwhich(*loc0_vtype ? loc0_vtype : "ascii",viewers,&cnt);
281: if (cnt > (PetscInt) sizeof(viewers)-1) SETERRQ1(comm,PETSC_ERR_ARG_OUTOFRANGE,"Unknown viewer type: %s",loc0_vtype);
282: if (viewer) {
283: if (!loc1_fname) {
284: switch (cnt) {
285: case 0:
286: PetscViewerASCIIGetStdout(comm,viewer);
287: break;
288: case 1:
289: if (!(*viewer = PETSC_VIEWER_BINARY_(comm))) CHKERRQ(PETSC_ERR_PLIB);
290: break;
291: case 2:
292: if (!(*viewer = PETSC_VIEWER_DRAW_(comm))) CHKERRQ(PETSC_ERR_PLIB);
293: break;
294: #if defined(PETSC_USE_SOCKET_VIEWER)
295: case 3:
296: if (!(*viewer = PETSC_VIEWER_SOCKET_(comm))) CHKERRQ(PETSC_ERR_PLIB);
297: break;
298: #endif
299: #if defined(PETSC_HAVE_MATLAB_ENGINE)
300: case 4:
301: if (!(*viewer = PETSC_VIEWER_MATLAB_(comm))) CHKERRQ(PETSC_ERR_PLIB);
302: break;
303: #endif
304: #if defined(PETSC_HAVE_SAWS)
305: case 5:
306: if (!(*viewer = PETSC_VIEWER_SAWS_(comm))) CHKERRQ(PETSC_ERR_PLIB);
307: break;
308: #endif
309: #if defined(PETSC_HAVE_HDF5)
310: case 7:
311: if (!(*viewer = PETSC_VIEWER_HDF5_(comm))) CHKERRQ(PETSC_ERR_PLIB);
312: break;
313: #endif
314: case 8:
315: if (!(*viewer = PETSC_VIEWER_GLVIS_(comm))) CHKERRQ(PETSC_ERR_PLIB);
316: break;
317: default: SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported viewer %s",loc0_vtype);
318: }
319: PetscObjectReference((PetscObject)*viewer);
320: } else {
321: if (loc2_fmt && !*loc1_fname && (cnt == 0)) { /* ASCII format without file name */
322: PetscViewerASCIIGetStdout(comm,viewer);
323: PetscObjectReference((PetscObject)*viewer);
324: } else {
325: PetscFileMode fmode;
326: PetscViewerCreate(comm,viewer);
327: PetscViewerSetType(*viewer,*loc0_vtype ? loc0_vtype : "ascii");
328: fmode = FILE_MODE_WRITE;
329: if (loc3_fmode && *loc3_fmode) { /* Has non-empty file mode ("write" or "append") */
330: PetscEnumFind(PetscFileModes,loc3_fmode,(PetscEnum*)&fmode,&flag);
331: if (!flag) SETERRQ1(comm,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown file mode: %s",loc3_fmode);
332: }
333: if (loc2_fmt) {
334: PetscBool tk,im;
335: PetscStrcmp(loc1_fname,"tikz",&tk);
336: PetscStrcmp(loc1_fname,"image",&im);
337: if (tk || im) {
338: PetscViewerDrawSetInfo(*viewer,NULL,loc2_fmt,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE);
339: *loc2_fmt = 0;
340: }
341: }
342: PetscViewerFileSetMode(*viewer,flag?fmode:FILE_MODE_WRITE);
343: PetscViewerFileSetName(*viewer,loc1_fname);
344: if (*loc1_fname) {
345: PetscViewerDrawSetDrawType(*viewer,loc1_fname);
346: }
347: PetscViewerSetFromOptions(*viewer);
348: }
349: }
350: }
351: if (viewer) {
352: PetscViewerSetUp(*viewer);
353: }
354: if (loc2_fmt && *loc2_fmt) {
355: PetscViewerFormat tfmt;
357: PetscEnumFind(PetscViewerFormats,loc2_fmt,(PetscEnum*)&tfmt,&flag);
358: if (format) *format = tfmt;
359: if (!flag) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unknown viewer format %s",loc2_fmt);
360: } else if (viewer && (cnt == 6) && format) { /* Get format from VTK viewer */
361: PetscViewerGetFormat(*viewer,format);
362: }
363: PetscFree(loc0_vtype);
364: }
365: }
366: return(0);
367: }
369: /*@
370: PetscViewerCreate - Creates a viewing context
372: Collective
374: Input Parameter:
375: . comm - MPI communicator
377: Output Parameter:
378: . inviewer - location to put the PetscViewer context
380: Level: advanced
382: .seealso: PetscViewerDestroy(), PetscViewerSetType(), PetscViewerType
384: @*/
385: PetscErrorCode PetscViewerCreate(MPI_Comm comm,PetscViewer *inviewer)
386: {
387: PetscViewer viewer;
391: *inviewer = 0;
392: PetscViewerInitializePackage();
393: PetscHeaderCreate(viewer,PETSC_VIEWER_CLASSID,"PetscViewer","PetscViewer","Viewer",comm,PetscViewerDestroy,NULL);
394: *inviewer = viewer;
395: viewer->data = 0;
396: return(0);
397: }
399: /*@C
400: PetscViewerSetType - Builds PetscViewer for a particular implementation.
402: Collective on PetscViewer
404: Input Parameter:
405: + viewer - the PetscViewer context
406: - type - for example, PETSCVIEWERASCII
408: Options Database Command:
409: . -viewer_type <type> - Sets the type; use -help for a list
410: of available methods (for instance, ascii)
412: Level: advanced
414: Notes:
415: See "include/petscviewer.h" for available methods (for instance,
416: PETSCVIEWERSOCKET)
418: .seealso: PetscViewerCreate(), PetscViewerGetType(), PetscViewerType, PetscViewerPushFormat()
419: @*/
420: PetscErrorCode PetscViewerSetType(PetscViewer viewer,PetscViewerType type)
421: {
422: PetscErrorCode ierr,(*r)(PetscViewer);
423: PetscBool match;
428: PetscObjectTypeCompare((PetscObject)viewer,type,&match);
429: if (match) return(0);
431: /* cleanup any old type that may be there */
432: if (viewer->data) {
433: (*viewer->ops->destroy)(viewer);
435: viewer->ops->destroy = NULL;
436: viewer->data = 0;
437: }
438: PetscMemzero(viewer->ops,sizeof(struct _PetscViewerOps));
440: PetscFunctionListFind(PetscViewerList,type,&r);
441: if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown PetscViewer type given: %s",type);
443: PetscObjectChangeTypeName((PetscObject)viewer,type);
444: (*r)(viewer);
445: return(0);
446: }
448: /*@C
449: PetscViewerRegister - Adds a viewer
451: Not Collective
453: Input Parameters:
454: + name_solver - name of a new user-defined viewer
455: - routine_create - routine to create method context
457: Level: developer
458: Notes:
459: PetscViewerRegister() may be called multiple times to add several user-defined viewers.
461: Sample usage:
462: .vb
463: PetscViewerRegister("my_viewer_type",MyViewerCreate);
464: .ve
466: Then, your solver can be chosen with the procedural interface via
467: $ PetscViewerSetType(viewer,"my_viewer_type")
468: or at runtime via the option
469: $ -viewer_type my_viewer_type
471: .seealso: PetscViewerRegisterAll(), PetscViewerRegisterDestroy()
472: @*/
473: PetscErrorCode PetscViewerRegister(const char *sname,PetscErrorCode (*function)(PetscViewer))
474: {
478: PetscViewerInitializePackage();
479: PetscFunctionListAdd(&PetscViewerList,sname,function);
480: return(0);
481: }
483: /*@C
484: PetscViewerSetFromOptions - Sets the graphics type from the options database.
485: Defaults to a PETSc X windows graphics.
487: Collective on PetscViewer
489: Input Parameter:
490: . PetscViewer - the graphics context
492: Level: intermediate
494: Notes:
495: Must be called after PetscViewerCreate() before the PetscViewer is used.
497: .seealso: PetscViewerCreate(), PetscViewerSetType(), PetscViewerType
499: @*/
500: PetscErrorCode PetscViewerSetFromOptions(PetscViewer viewer)
501: {
502: PetscErrorCode ierr;
503: char vtype[256];
504: PetscBool flg;
509: if (!PetscViewerList) {
510: PetscViewerRegisterAll();
511: }
512: PetscObjectOptionsBegin((PetscObject)viewer);
513: PetscOptionsFList("-viewer_type","Type of PetscViewer","None",PetscViewerList,(char*)(((PetscObject)viewer)->type_name ? ((PetscObject)viewer)->type_name : PETSCVIEWERASCII),vtype,256,&flg);
514: if (flg) {
515: PetscViewerSetType(viewer,vtype);
516: }
517: /* type has not been set? */
518: if (!((PetscObject)viewer)->type_name) {
519: PetscViewerSetType(viewer,PETSCVIEWERASCII);
520: }
521: if (viewer->ops->setfromoptions) {
522: (*viewer->ops->setfromoptions)(PetscOptionsObject,viewer);
523: }
525: /* process any options handlers added with PetscObjectAddOptionsHandler() */
526: PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)viewer);
527: PetscViewerViewFromOptions(viewer,NULL,"-viewer_view");
528: PetscOptionsEnd();
529: return(0);
530: }
532: PetscErrorCode PetscViewerFlowControlStart(PetscViewer viewer,PetscInt *mcnt,PetscInt *cnt)
533: {
536: PetscViewerBinaryGetFlowControl(viewer,mcnt);
537: PetscViewerBinaryGetFlowControl(viewer,cnt);
538: return(0);
539: }
541: PetscErrorCode PetscViewerFlowControlStepMaster(PetscViewer viewer,PetscInt i,PetscInt *mcnt,PetscInt cnt)
542: {
544: MPI_Comm comm;
547: PetscObjectGetComm((PetscObject)viewer,&comm);
548: if (i >= *mcnt) {
549: *mcnt += cnt;
550: MPI_Bcast(mcnt,1,MPIU_INT,0,comm);
551: }
552: return(0);
553: }
555: PetscErrorCode PetscViewerFlowControlEndMaster(PetscViewer viewer,PetscInt *mcnt)
556: {
558: MPI_Comm comm;
560: PetscObjectGetComm((PetscObject)viewer,&comm);
561: *mcnt = 0;
562: MPI_Bcast(mcnt,1,MPIU_INT,0,comm);
563: return(0);
564: }
566: PetscErrorCode PetscViewerFlowControlStepWorker(PetscViewer viewer,PetscMPIInt rank,PetscInt *mcnt)
567: {
569: MPI_Comm comm;
571: PetscObjectGetComm((PetscObject)viewer,&comm);
572: while (PETSC_TRUE) {
573: if (rank < *mcnt) break;
574: MPI_Bcast(mcnt,1,MPIU_INT,0,comm);
575: }
576: return(0);
577: }
579: PetscErrorCode PetscViewerFlowControlEndWorker(PetscViewer viewer,PetscInt *mcnt)
580: {
582: MPI_Comm comm;
584: PetscObjectGetComm((PetscObject)viewer,&comm);
585: while (PETSC_TRUE) {
586: MPI_Bcast(mcnt,1,MPIU_INT,0,comm);
587: if (!*mcnt) break;
588: }
589: return(0);
590: }