xref: /petsc/src/sys/classes/viewer/interface/viewreg.c (revision c86da64b93f6a46990b92b3e5e444e957ea7346b)
1 
2 #include <petsc/private/viewerimpl.h>  /*I "petscviewer.h" I*/
3 #include <petsc/private/hashtable.h>
4 #if defined(PETSC_HAVE_SAWS)
5 #include <petscviewersaws.h>
6 #endif
7 
8 PetscFunctionList PetscViewerList = 0;
9 
10 
11 PetscOptionsHelpPrinted PetscOptionsHelpPrintedSingleton = NULL;
12 KHASH_SET_INIT_STR(HTPrinted)
13 struct  _n_PetscOptionsHelpPrinted{
14   khash_t(HTPrinted) *printed;
15   PetscSegBuffer     strings;
16 };
17 
18 PetscErrorCode PetscOptionsHelpPrintedDestroy(PetscOptionsHelpPrinted *hp)
19 {
20   PetscErrorCode ierr;
21 
22   PetscFunctionBegin;
23   if (!*hp) PetscFunctionReturn(0);
24   kh_destroy(HTPrinted,(*hp)->printed);
25   ierr = PetscSegBufferDestroy(&(*hp)->strings);CHKERRQ(ierr);
26   ierr = PetscFree(*hp);CHKERRQ(ierr);
27   PetscFunctionReturn(0);
28 }
29 
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.
33 
34      Not collective
35 
36     Level: developer
37 
38 .seealso: PetscOptionsHelpPrintedCheck(), PetscOptionsHelpPrintChecked()
39 @*/
40 PetscErrorCode PetscOptionsHelpPrintedCreate(PetscOptionsHelpPrinted *hp)
41 {
42   PetscErrorCode             ierr;
43 
44   PetscFunctionBegin;
45   ierr = PetscNew(hp);CHKERRQ(ierr);
46   (*hp)->printed = kh_init(HTPrinted);
47   ierr = PetscSegBufferCreate(sizeof(char),10000,&(*hp)->strings);CHKERRQ(ierr);
48   PetscFunctionReturn(0);
49 }
50 
51 /*@C
52       PetscOptionsHelpPrintedCheck - Checks if a particular pre, name pair has previous been entered (meaning the help message was printed)
53 
54      Not collective
55 
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)
60 
61     Output Parameter:
62 .     found - PETSC_TRUE if the string was already set
63 
64     Level: intermediate
65 
66 
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;
77 
78   PetscFunctionBegin;
79   ierr = PetscStrlen(pre,&l1);CHKERRQ(ierr);
80   ierr = PetscStrlen(name,&l2);CHKERRQ(ierr);
81   if (l1+l2 == 0) {
82     *found = PETSC_FALSE;
83     PetscFunctionReturn(0);
84   }
85 #if !defined(PETSC_HAVE_THREADSAFETY)
86   ierr = PetscSegBufferGet(hp->strings,l1+l2+1,&both);CHKERRQ(ierr);
87   ierr = PetscStrcpy(both,pre);CHKERRQ(ierr);
88   ierr = PetscStrcat(both,name);CHKERRQ(ierr);
89   kh_put(HTPrinted,hp->printed,both,&newitem);
90   if (!newitem) {
91     ierr = PetscSegBufferUnuse(hp->strings,l1+l2+1);CHKERRQ(ierr);
92   }
93   *found = newitem ? PETSC_FALSE : PETSC_TRUE;
94 #else
95   *found = PETSC_FALSE;
96 #endif
97   PetscFunctionReturn(0);
98 }
99 
100 static PetscBool noviewer = PETSC_FALSE;
101 static PetscBool noviewers[PETSCVIEWERGETVIEWEROFFPUSHESMAX];
102 static PetscInt  inoviewers = 0;
103 
104 /*@
105   PetscOptionsPushGetViewerOff - control whether PetscOptionsGetViewer returns a viewer.
106 
107   Logically Collective
108 
109   Input Parameter:
110 . flg - PETSC_TRUE to turn off viewer creation, PETSC_FALSE to turn it on.
111 
112   Level: developer
113 
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.
117 
118 .seealso: PetscOptionsGetViewer(), PetscOptionsPopGetViewerOff()
119 @*/
120 PetscErrorCode  PetscOptionsPushGetViewerOff(PetscBool flg)
121 {
122   PetscFunctionBegin;
123   if (inoviewers > PETSCVIEWERGETVIEWEROFFPUSHESMAX - 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many PetscOptionsPushGetViewerOff(), perhaps you forgot PetscOptionsPopGetViewerOff()?");
124 
125   noviewers[inoviewers++] = noviewer;
126   noviewer = flg;
127   PetscFunctionReturn(0);
128 }
129 
130 /*@
131   PetscOptionsPopGetViewerOff - reset whether PetscOptionsGetViewer returns a viewer.
132 
133   Logically Collective
134 
135   Level: developer
136 
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.
140 
141 .seealso: PetscOptionsGetViewer(), PetscOptionsPushGetViewerOff()
142 @*/
143 PetscErrorCode  PetscOptionsPopGetViewerOff(void)
144 {
145   PetscFunctionBegin;
146   if (!inoviewers) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many PetscOptionsPopGetViewerOff(), perhaps you forgot PetscOptionsPushGetViewerOff()?");
147   noviewer = noviewers[--inoviewers];
148   PetscFunctionReturn(0);
149 }
150 
151 /*@
152   PetscOptionsGetViewerOff - does PetscOptionsGetViewer return a viewer?
153 
154   Logically Collective
155 
156   Output Parameter:
157 . flg - whether viewers are returned.
158 
159   Level: developer
160 
161   Notes:
162     Calling XXXViewFromOptions in an inner loop can be very expensive.  This can appear, for example, when using
163    many small subsolves.
164 
165 .seealso: PetscOptionsGetViewer(), PetscOptionsPushGetViewerOff(), PetscOptionsPopGetViewerOff()
166 @*/
167 PetscErrorCode  PetscOptionsGetViewerOff(PetscBool *flg)
168 {
169   PetscFunctionBegin;
170   PetscValidPointer(flg,1);
171   *flg = noviewer;
172   PetscFunctionReturn(0);
173 }
174 
175 /*@C
176    PetscOptionsGetViewer - Gets a viewer appropriate for the type indicated by the user
177 
178    Collective on MPI_Comm
179 
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
184 
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
189 
190    Level: intermediate
191 
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)
201 
202    Use PetscViewerDestroy() after using the viewer, otherwise a memory leak will occur
203 
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.
207 
208    If PETSc is configured with --with-viewfromoptions=0 this function always returns with *set of PETSC_FALSE
209 
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,const char pre[],const char name[],PetscViewer *viewer,PetscViewerFormat *format,PetscBool  *set)
220 {
221   const char                     *value;
222   PetscErrorCode                 ierr;
223   PetscBool                      flag,hashelp;
224 
225   PetscFunctionBegin;
226   PetscValidCharPointer(name,3);
227 
228   if (viewer) *viewer = NULL;
229   if (format) *format = PETSC_VIEWER_DEFAULT;
230   if (set)    *set    = PETSC_FALSE;
231   ierr = PetscOptionsGetViewerOff(&flag);CHKERRQ(ierr);
232   if (flag) PetscFunctionReturn(0);
233 
234   ierr = PetscOptionsHasHelp(NULL,&hashelp);CHKERRQ(ierr);
235   if (hashelp) {
236     PetscBool found;
237 
238     if (!PetscOptionsHelpPrintedSingleton) {
239       ierr = PetscOptionsHelpPrintedCreate(&PetscOptionsHelpPrintedSingleton);CHKERRQ(ierr);
240     }
241     ierr = PetscOptionsHelpPrintedCheck(PetscOptionsHelpPrintedSingleton,pre,name,&found);CHKERRQ(ierr);
242     if (!found) {
243       if (viewer) {
244         ierr = (*PetscHelpPrintf)(comm,"\n  -%s%s ascii[:[filename][:[format][:append]]]: %s (%s)\n",pre ? pre : "",name+1,"Prints object to stdout or ASCII file","PetscOptionsGetViewer");CHKERRQ(ierr);
245         ierr = (*PetscHelpPrintf)(comm,"    -%s%s binary[:[filename][:[format][:append]]]: %s (%s)\n",pre ? pre : "",name+1,"Saves object to a binary file","PetscOptionsGetViewer");CHKERRQ(ierr);
246         ierr = (*PetscHelpPrintf)(comm,"    -%s%s draw[:[drawtype][:filename|format]] %s (%s)\n",pre ? pre : "",name+1,"Draws object","PetscOptionsGetViewer");CHKERRQ(ierr);
247         ierr = (*PetscHelpPrintf)(comm,"    -%s%s socket[:port]: %s (%s)\n",pre ? pre : "",name+1,"Pushes object to a Unix socket","PetscOptionsGetViewer");CHKERRQ(ierr);
248         ierr = (*PetscHelpPrintf)(comm,"    -%s%s saws[:communicatorname]: %s (%s)\n\n",pre ? pre : "",name+1,"Publishes object to SAWs","PetscOptionsGetViewer");CHKERRQ(ierr);
249       } else {
250         ierr = (*PetscHelpPrintf)(comm,"    -%s%s\n",pre ? pre : "",name+1);CHKERRQ(ierr);
251       }
252     }
253   }
254 
255   if (format) *format = PETSC_VIEWER_DEFAULT;
256   ierr = PetscOptionsFindPair(NULL,pre,name,&value,&flag);CHKERRQ(ierr);
257   if (flag) {
258     if (set) *set = PETSC_TRUE;
259     if (!value) {
260       if (viewer) {
261         ierr = PetscViewerASCIIGetStdout(comm,viewer);CHKERRQ(ierr);
262         ierr = PetscObjectReference((PetscObject)*viewer);CHKERRQ(ierr);
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};
268 
269       ierr = PetscStrallocpy(value,&loc0_vtype);CHKERRQ(ierr);
270       ierr = PetscStrchr(loc0_vtype,':',&loc1_fname);CHKERRQ(ierr);
271       if (loc1_fname) {
272         *loc1_fname++ = 0;
273         ierr = PetscStrchr(loc1_fname,':',&loc2_fmt);CHKERRQ(ierr);
274       }
275       if (loc2_fmt) {
276         *loc2_fmt++ = 0;
277         ierr = PetscStrchr(loc2_fmt,':',&loc3_fmode);CHKERRQ(ierr);
278       }
279       if (loc3_fmode) *loc3_fmode++ = 0;
280       ierr = PetscStrendswithwhich(*loc0_vtype ? loc0_vtype : "ascii",viewers,&cnt);CHKERRQ(ierr);
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             ierr = PetscViewerASCIIGetStdout(comm,viewer);CHKERRQ(ierr);
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           ierr = PetscObjectReference((PetscObject)*viewer);CHKERRQ(ierr);
320         } else {
321           if (loc2_fmt && !*loc1_fname && (cnt == 0)) { /* ASCII format without file name */
322             ierr = PetscViewerASCIIGetStdout(comm,viewer);CHKERRQ(ierr);
323             ierr = PetscObjectReference((PetscObject)*viewer);CHKERRQ(ierr);
324           } else {
325             PetscFileMode fmode;
326             ierr = PetscViewerCreate(comm,viewer);CHKERRQ(ierr);
327             ierr = PetscViewerSetType(*viewer,*loc0_vtype ? loc0_vtype : "ascii");CHKERRQ(ierr);
328             fmode = FILE_MODE_WRITE;
329             if (loc3_fmode && *loc3_fmode) { /* Has non-empty file mode ("write" or "append") */
330               ierr = PetscEnumFind(PetscFileModes,loc3_fmode,(PetscEnum*)&fmode,&flag);CHKERRQ(ierr);
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               ierr = PetscStrcmp(loc1_fname,"tikz",&tk);CHKERRQ(ierr);
336               ierr = PetscStrcmp(loc1_fname,"image",&im);CHKERRQ(ierr);
337               if (tk || im) {
338                 ierr = PetscViewerDrawSetInfo(*viewer,NULL,loc2_fmt,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
339                 *loc2_fmt = 0;
340               }
341             }
342             ierr = PetscViewerFileSetMode(*viewer,flag?fmode:FILE_MODE_WRITE);CHKERRQ(ierr);
343             ierr = PetscViewerFileSetName(*viewer,loc1_fname);CHKERRQ(ierr);
344             if (*loc1_fname) {
345               ierr = PetscViewerDrawSetDrawType(*viewer,loc1_fname);CHKERRQ(ierr);
346             }
347           }
348         }
349       }
350       if (viewer) {
351         ierr = PetscViewerSetUp(*viewer);CHKERRQ(ierr);
352       }
353       if (loc2_fmt && *loc2_fmt) {
354         PetscViewerFormat tfmt;
355 
356         ierr = PetscEnumFind(PetscViewerFormats,loc2_fmt,(PetscEnum*)&tfmt,&flag);CHKERRQ(ierr);
357         if (format) *format = tfmt;
358         if (!flag) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unknown viewer format %s",loc2_fmt);
359       } else if (viewer && (cnt == 6) && format) { /* Get format from VTK viewer */
360         ierr = PetscViewerGetFormat(*viewer,format);CHKERRQ(ierr);
361       }
362       ierr = PetscFree(loc0_vtype);CHKERRQ(ierr);
363     }
364   }
365   PetscFunctionReturn(0);
366 }
367 
368 /*@
369    PetscViewerCreate - Creates a viewing context
370 
371    Collective on MPI_Comm
372 
373    Input Parameter:
374 .  comm - MPI communicator
375 
376    Output Parameter:
377 .  inviewer - location to put the PetscViewer context
378 
379    Level: advanced
380 
381    Concepts: graphics^creating PetscViewer
382    Concepts: file input/output^creating PetscViewer
383    Concepts: sockets^creating PetscViewer
384 
385 .seealso: PetscViewerDestroy(), PetscViewerSetType(), PetscViewerType
386 
387 @*/
388 PetscErrorCode  PetscViewerCreate(MPI_Comm comm,PetscViewer *inviewer)
389 {
390   PetscViewer    viewer;
391   PetscErrorCode ierr;
392 
393   PetscFunctionBegin;
394   *inviewer = 0;
395   ierr = PetscViewerInitializePackage();CHKERRQ(ierr);
396   ierr         = PetscHeaderCreate(viewer,PETSC_VIEWER_CLASSID,"PetscViewer","PetscViewer","Viewer",comm,PetscViewerDestroy,NULL);CHKERRQ(ierr);
397   *inviewer    = viewer;
398   viewer->data = 0;
399   PetscFunctionReturn(0);
400 }
401 
402 /*@C
403    PetscViewerSetType - Builds PetscViewer for a particular implementation.
404 
405    Collective on PetscViewer
406 
407    Input Parameter:
408 +  viewer      - the PetscViewer context
409 -  type        - for example, PETSCVIEWERASCII
410 
411    Options Database Command:
412 .  -draw_type  <type> - Sets the type; use -help for a list
413     of available methods (for instance, ascii)
414 
415    Level: advanced
416 
417    Notes:
418    See "include/petscviewer.h" for available methods (for instance,
419    PETSCVIEWERSOCKET)
420 
421 .seealso: PetscViewerCreate(), PetscViewerGetType(), PetscViewerType, PetscViewerPushFormat()
422 @*/
423 PetscErrorCode  PetscViewerSetType(PetscViewer viewer,PetscViewerType type)
424 {
425   PetscErrorCode ierr,(*r)(PetscViewer);
426   PetscBool      match;
427 
428   PetscFunctionBegin;
429   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
430   PetscValidCharPointer(type,2);
431   ierr = PetscObjectTypeCompare((PetscObject)viewer,type,&match);CHKERRQ(ierr);
432   if (match) PetscFunctionReturn(0);
433 
434   /* cleanup any old type that may be there */
435   if (viewer->data) {
436     ierr         = (*viewer->ops->destroy)(viewer);CHKERRQ(ierr);
437 
438     viewer->ops->destroy = NULL;
439     viewer->data         = 0;
440   }
441   ierr = PetscMemzero(viewer->ops,sizeof(struct _PetscViewerOps));CHKERRQ(ierr);
442 
443   ierr =  PetscFunctionListFind(PetscViewerList,type,&r);CHKERRQ(ierr);
444   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown PetscViewer type given: %s",type);
445 
446   ierr = PetscObjectChangeTypeName((PetscObject)viewer,type);CHKERRQ(ierr);
447   ierr = (*r)(viewer);CHKERRQ(ierr);
448   PetscFunctionReturn(0);
449 }
450 
451 /*@C
452    PetscViewerRegister - Adds a viewer
453 
454    Not Collective
455 
456    Input Parameters:
457 +  name_solver - name of a new user-defined viewer
458 -  routine_create - routine to create method context
459 
460    Level: developer
461    Notes:
462    PetscViewerRegister() may be called multiple times to add several user-defined viewers.
463 
464    Sample usage:
465 .vb
466    PetscViewerRegister("my_viewer_type",MyViewerCreate);
467 .ve
468 
469    Then, your solver can be chosen with the procedural interface via
470 $     PetscViewerSetType(viewer,"my_viewer_type")
471    or at runtime via the option
472 $     -viewer_type my_viewer_type
473 
474   Concepts: registering^Viewers
475 
476 .seealso: PetscViewerRegisterAll(), PetscViewerRegisterDestroy()
477  @*/
478 PetscErrorCode  PetscViewerRegister(const char *sname,PetscErrorCode (*function)(PetscViewer))
479 {
480   PetscErrorCode ierr;
481 
482   PetscFunctionBegin;
483   ierr = PetscFunctionListAdd(&PetscViewerList,sname,function);CHKERRQ(ierr);
484   PetscFunctionReturn(0);
485 }
486 
487 /*@C
488    PetscViewerSetFromOptions - Sets the graphics type from the options database.
489       Defaults to a PETSc X windows graphics.
490 
491    Collective on PetscViewer
492 
493    Input Parameter:
494 .     PetscViewer - the graphics context
495 
496    Level: intermediate
497 
498    Notes:
499     Must be called after PetscViewerCreate() before the PetscViewer is used.
500 
501   Concepts: PetscViewer^setting options
502 
503 .seealso: PetscViewerCreate(), PetscViewerSetType(), PetscViewerType
504 
505 @*/
506 PetscErrorCode  PetscViewerSetFromOptions(PetscViewer viewer)
507 {
508   PetscErrorCode    ierr;
509   char              vtype[256];
510   PetscBool         flg;
511 
512   PetscFunctionBegin;
513   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,1);
514 
515   if (!PetscViewerList) {
516     ierr = PetscViewerRegisterAll();CHKERRQ(ierr);
517   }
518   ierr = PetscObjectOptionsBegin((PetscObject)viewer);CHKERRQ(ierr);
519   ierr = PetscOptionsFList("-viewer_type","Type of PetscViewer","None",PetscViewerList,(char*)(((PetscObject)viewer)->type_name ? ((PetscObject)viewer)->type_name : PETSCVIEWERASCII),vtype,256,&flg);CHKERRQ(ierr);
520   if (flg) {
521     ierr = PetscViewerSetType(viewer,vtype);CHKERRQ(ierr);
522   }
523   /* type has not been set? */
524   if (!((PetscObject)viewer)->type_name) {
525     ierr = PetscViewerSetType(viewer,PETSCVIEWERASCII);CHKERRQ(ierr);
526   }
527   if (viewer->ops->setfromoptions) {
528     ierr = (*viewer->ops->setfromoptions)(PetscOptionsObject,viewer);CHKERRQ(ierr);
529   }
530 
531   /* process any options handlers added with PetscObjectAddOptionsHandler() */
532   ierr = PetscObjectProcessOptionsHandlers(PetscOptionsObject,(PetscObject)viewer);CHKERRQ(ierr);
533   ierr = PetscViewerViewFromOptions(viewer,NULL,"-viewer_view");CHKERRQ(ierr);
534   ierr = PetscOptionsEnd();CHKERRQ(ierr);
535   PetscFunctionReturn(0);
536 }
537 
538 PetscErrorCode PetscViewerFlowControlStart(PetscViewer viewer,PetscInt *mcnt,PetscInt *cnt)
539 {
540   PetscErrorCode ierr;
541   PetscFunctionBegin;
542   ierr = PetscViewerBinaryGetFlowControl(viewer,mcnt);CHKERRQ(ierr);
543   ierr = PetscViewerBinaryGetFlowControl(viewer,cnt);CHKERRQ(ierr);
544   PetscFunctionReturn(0);
545 }
546 
547 PetscErrorCode PetscViewerFlowControlStepMaster(PetscViewer viewer,PetscInt i,PetscInt *mcnt,PetscInt cnt)
548 {
549   PetscErrorCode ierr;
550   MPI_Comm       comm;
551 
552   PetscFunctionBegin;
553   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
554   if (i >= *mcnt) {
555     *mcnt += cnt;
556     ierr = MPI_Bcast(mcnt,1,MPIU_INT,0,comm);CHKERRQ(ierr);
557   }
558   PetscFunctionReturn(0);
559 }
560 
561 PetscErrorCode PetscViewerFlowControlEndMaster(PetscViewer viewer,PetscInt *mcnt)
562 {
563   PetscErrorCode ierr;
564   MPI_Comm       comm;
565   PetscFunctionBegin;
566   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
567   *mcnt = 0;
568   ierr = MPI_Bcast(mcnt,1,MPIU_INT,0,comm);CHKERRQ(ierr);
569   PetscFunctionReturn(0);
570 }
571 
572 PetscErrorCode PetscViewerFlowControlStepWorker(PetscViewer viewer,PetscMPIInt rank,PetscInt *mcnt)
573 {
574   PetscErrorCode ierr;
575   MPI_Comm       comm;
576   PetscFunctionBegin;
577   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
578   while (PETSC_TRUE) {
579     if (rank < *mcnt) break;
580     ierr = MPI_Bcast(mcnt,1,MPIU_INT,0,comm);CHKERRQ(ierr);
581   }
582   PetscFunctionReturn(0);
583 }
584 
585 PetscErrorCode PetscViewerFlowControlEndWorker(PetscViewer viewer,PetscInt *mcnt)
586 {
587   PetscErrorCode ierr;
588   MPI_Comm       comm;
589   PetscFunctionBegin;
590   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
591   while (PETSC_TRUE) {
592     ierr = MPI_Bcast(mcnt,1,MPIU_INT,0,comm);CHKERRQ(ierr);
593     if (!*mcnt) break;
594   }
595   PetscFunctionReturn(0);
596 }
597