xref: /petsc/src/sys/classes/draw/interface/drawreg.c (revision 9895aa37ac365bac650f6bd8bf977519f7222510)
1 
2 /*
3        Provides the registration process for PETSc PetscDraw routines
4 */
5 #include <petsc-private/drawimpl.h>  /*I "petscdraw.h" I*/
6 
7 /*
8    Contains the list of registered PetscDraw routines
9 */
10 PetscFunctionList PetscDrawList = 0;
11 
12 #undef __FUNCT__
13 #define __FUNCT__ "PetscDrawCreate"
14 /*@C
15    PetscDrawCreate - Creates a graphics context.
16 
17    Collective on MPI_Comm
18 
19    Input Parameter:
20 +  comm - MPI communicator
21 .  display - X display when using X windows
22 .  title - optional title added to top of window
23 .  x,y - coordinates of lower left corner of window or PETSC_DECIDE
24 -  w, h - width and height of window or PETSC_DECIDE or PETSC_DRAW_HALF_SIZE, PETSC_DRAW_FULL_SIZE,
25           or PETSC_DRAW_THIRD_SIZE or PETSC_DRAW_QUARTER_SIZE
26 
27    Output Parameter:
28 .  draw - location to put the PetscDraw context
29 
30    Level: beginner
31 
32    Concepts: graphics^creating context
33    Concepts: drawing^creating context
34 
35 .seealso: PetscDrawSetFromOptions(), PetscDrawDestroy(), PetscDrawSetType()
36 @*/
37 PetscErrorCode  PetscDrawCreate(MPI_Comm comm,const char display[],const char title[],int x,int y,int w,int h,PetscDraw *indraw)
38 {
39   PetscDraw      draw;
40   PetscErrorCode ierr;
41   PetscReal      dpause;
42   PetscBool      flag;
43 
44   PetscFunctionBegin;
45 #if !defined(PETSC_USE_DYNAMIC_LIBRARIES)
46   ierr = PetscDrawInitializePackage(NULL);CHKERRQ(ierr);
47 #endif
48   *indraw = 0;
49   ierr = PetscHeaderCreate(draw,_p_PetscDraw,struct _PetscDrawOps,PETSC_DRAW_CLASSID,"Draw","Graphics","Draw",comm,PetscDrawDestroy,0);CHKERRQ(ierr);
50 
51   draw->data    = 0;
52   ierr          = PetscStrallocpy(title,&draw->title);CHKERRQ(ierr);
53   ierr          = PetscStrallocpy(display,&draw->display);CHKERRQ(ierr);
54   draw->x       = x;
55   draw->y       = y;
56   draw->w       = w;
57   draw->h       = h;
58   draw->pause   = 0.0;
59   draw->coor_xl = 0.0;
60   draw->coor_xr = 1.0;
61   draw->coor_yl = 0.0;
62   draw->coor_yr = 1.0;
63   draw->port_xl = 0.0;
64   draw->port_xr = 1.0;
65   draw->port_yl = 0.0;
66   draw->port_yr = 1.0;
67   draw->popup   = 0;
68 
69   ierr = PetscOptionsGetReal(NULL,"-draw_pause",&dpause,&flag);CHKERRQ(ierr);
70   if (flag) draw->pause = dpause;
71   draw->savefilename  = NULL;
72   draw->savefilemovie = PETSC_FALSE;
73   draw->savefilecount = -1;
74 
75   ierr = PetscDrawSetCurrentPoint(draw,.5,.9);CHKERRQ(ierr);
76 
77   draw->boundbox_xl  = .5;
78   draw->boundbox_xr  = .5;
79   draw->boundbox_yl  = .9;
80   draw->boundbox_yr  = .9;
81 
82   *indraw = draw;
83   PetscFunctionReturn(0);
84 }
85 
86 #undef __FUNCT__
87 #define __FUNCT__ "PetscDrawSetType"
88 /*@C
89    PetscDrawSetType - Builds graphics object for a particular implementation
90 
91    Collective on PetscDraw
92 
93    Input Parameter:
94 +  draw      - the graphics context
95 -  type      - for example, PETSC_DRAW_X
96 
97    Options Database Command:
98 .  -draw_type  <type> - Sets the type; use -help for a list
99     of available methods (for instance, x)
100 
101    Level: intermediate
102 
103    Notes:
104    See "petsc/include/petscdraw.h" for available methods (for instance,
105    PETSC_DRAW_X)
106 
107    Concepts: drawing^X windows
108    Concepts: X windows^graphics
109    Concepts: drawing^Microsoft Windows
110 
111 .seealso: PetscDrawSetFromOptions(), PetscDrawCreate(), PetscDrawDestroy()
112 @*/
113 PetscErrorCode  PetscDrawSetType(PetscDraw draw,PetscDrawType type)
114 {
115   PetscErrorCode ierr,(*r)(PetscDraw);
116   PetscBool      match;
117   PetscBool      flg=PETSC_FALSE;
118 
119   PetscFunctionBegin;
120   PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
121   PetscValidCharPointer(type,2);
122 
123   ierr = PetscObjectTypeCompare((PetscObject)draw,type,&match);CHKERRQ(ierr);
124   if (match) PetscFunctionReturn(0);
125 
126   /*  User requests no graphics */
127   ierr = PetscOptionsHasName(NULL,"-nox",&flg);CHKERRQ(ierr);
128 
129   /*
130      This is not ideal, but it allows codes to continue to run if X graphics
131    was requested but is not installed on this machine. Mostly this is for
132    testing.
133    */
134 #if !defined(PETSC_HAVE_X)
135   if (!flg) {
136     ierr = PetscStrcmp(type,PETSC_DRAW_X,&match);CHKERRQ(ierr);
137     if (match) {
138       PetscBool dontwarn = PETSC_TRUE;
139       flg  = PETSC_TRUE;
140       ierr = PetscOptionsHasName(NULL,"-nox_warning",&dontwarn);CHKERRQ(ierr);
141       if (!dontwarn) (*PetscErrorPrintf)("PETSc installed without X windows on this machine\nproceeding without graphics\n");
142     }
143   }
144 #endif
145   if (flg) type = PETSC_DRAW_NULL;
146 
147   if (draw->data) {
148     /* destroy the old private PetscDraw context */
149     ierr               = (*draw->ops->destroy)(draw);CHKERRQ(ierr);
150     draw->ops->destroy = NULL;
151     draw->data         = 0;
152   }
153 
154   ierr =  PetscFunctionListFind(PetscObjectComm((PetscObject)draw),PetscDrawList,type,PETSC_TRUE,(void (**)(void)) &r);CHKERRQ(ierr);
155   if (!r) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown PetscDraw type given: %s",type);
156   ierr       = PetscObjectChangeTypeName((PetscObject)draw,type);CHKERRQ(ierr);
157   draw->data = 0;
158   ierr       = (*r)(draw);CHKERRQ(ierr);
159   PetscFunctionReturn(0);
160 }
161 
162 #undef __FUNCT__
163 #define __FUNCT__ "PetscDrawRegisterDestroy"
164 /*@C
165    PetscDrawRegisterDestroy - Frees the list of PetscDraw methods that were
166    registered by PetscDrawRegisterDynamic().
167 
168    Not Collective
169 
170    Level: developer
171 
172 .seealso: PetscDrawRegisterDynamic(), PetscDrawRegisterAll()
173 @*/
174 PetscErrorCode  PetscDrawRegisterDestroy(void)
175 {
176   PetscErrorCode ierr;
177 
178   PetscFunctionBegin;
179   ierr = PetscFunctionListDestroy(&PetscDrawList);CHKERRQ(ierr);
180   PetscFunctionReturn(0);
181 }
182 
183 #undef __FUNCT__
184 #define __FUNCT__ "PetscDrawGetType"
185 /*@C
186    PetscDrawGetType - Gets the PetscDraw type as a string from the PetscDraw object.
187 
188    Not Collective
189 
190    Input Parameter:
191 .  draw - Krylov context
192 
193    Output Parameters:
194 .  name - name of PetscDraw method
195 
196    Level: advanced
197 
198 @*/
199 PetscErrorCode  PetscDrawGetType(PetscDraw draw,PetscDrawType *type)
200 {
201   PetscFunctionBegin;
202   PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
203   PetscValidPointer(type,2);
204   *type = ((PetscObject)draw)->type_name;
205   PetscFunctionReturn(0);
206 }
207 
208 #undef __FUNCT__
209 #define __FUNCT__ "PetscDrawRegister"
210 PetscErrorCode  PetscDrawRegister(const char *sname,const char *path,const char *name,PetscErrorCode (*function)(PetscDraw))
211 {
212   PetscErrorCode ierr;
213   char           fullname[PETSC_MAX_PATH_LEN];
214 
215   PetscFunctionBegin;
216   ierr = PetscFunctionListConcat(path,name,fullname);CHKERRQ(ierr);
217   ierr = PetscFunctionListAdd(PETSC_COMM_WORLD,&PetscDrawList,sname,fullname,(void (*)(void))function);CHKERRQ(ierr);
218   PetscFunctionReturn(0);
219 }
220 
221 #undef __FUNCT__
222 #define __FUNCT__ "PetscDrawSetFromOptions"
223 /*@
224    PetscDrawSetFromOptions - Sets the graphics type from the options database.
225       Defaults to a PETSc X windows graphics.
226 
227    Collective on PetscDraw
228 
229    Input Parameter:
230 .     draw - the graphics context
231 
232    Options Database Keys:
233 +   -nox - do not use X graphics (ignore graphics calls, but run program correctly)
234 .   -nox_warning - when X windows support is not installed this prevents the warning message
235                    from being printed
236 .   -draw_pause <pause amount> -- -1 indicates wait for mouse input, -2 indicates pause when window is to be destroyed
237 .   -draw_save [optional filename] - (X windows only) saves each image before it is cleared to a file
238 -   -draw_save_movie - converts image files to a movie  at the end of the run. See PetscDrawSetSave()
239 
240    Level: intermediate
241 
242    Notes:
243     Must be called after PetscDrawCreate() before the PetscDrawtor is used.
244 
245     Concepts: drawing^setting options
246     Concepts: graphics^setting options
247 
248 .seealso: PetscDrawCreate(), PetscDrawSetType(), PetscDrawSetSave()
249 
250 @*/
251 PetscErrorCode  PetscDrawSetFromOptions(PetscDraw draw)
252 {
253   PetscErrorCode ierr;
254   PetscBool      flg,nox;
255   char           vtype[256];
256   const char     *def;
257   PetscReal      dpause;
258 #if !defined(PETSC_USE_WINDOWS_GRAPHICS) && !defined(PETSC_HAVE_X)
259   PetscBool      warn;
260 #endif
261 
262   PetscFunctionBegin;
263   PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
264 
265   if (!PetscDrawList) {
266     ierr = PetscDrawRegisterAll(NULL);CHKERRQ(ierr);
267   }
268 
269   if (((PetscObject)draw)->type_name) def = ((PetscObject)draw)->type_name;
270   else {
271     ierr = PetscOptionsHasName(NULL,"-nox",&nox);CHKERRQ(ierr);
272     def  = PETSC_DRAW_NULL;
273 #if defined(PETSC_USE_WINDOWS_GRAPHICS)
274     if (!nox) def = PETSC_DRAW_WIN32;
275 #elif defined(PETSC_HAVE_X)
276     if (!nox) def = PETSC_DRAW_X;
277 #elif defined(PETSC_HAVE_GLUT)
278     if (!nox) def = PETSC_DRAW_GLUT;
279 #elif defined(PETSC_HAVE_OPENGLES)
280     if (!nox) def = PETSC_DRAW_OPENGLES;
281 #else
282     ierr = PetscOptionsHasName(NULL,"-nox_warning",&warn);CHKERRQ(ierr);
283     if (!nox && !warn) (*PetscErrorPrintf)("PETSc installed without X windows, Microsoft Graphics, OpenGL ES, or GLUT/OpenGL on this machine\nproceeding without graphics\n");
284 #endif
285   }
286   ierr = PetscObjectOptionsBegin((PetscObject)draw);CHKERRQ(ierr);
287   ierr = PetscOptionsList("-draw_type","Type of graphical output","PetscDrawSetType",PetscDrawList,def,vtype,256,&flg);CHKERRQ(ierr);
288   if (flg) {
289     ierr = PetscDrawSetType(draw,vtype);CHKERRQ(ierr);
290   } else if (!((PetscObject)draw)->type_name) {
291     ierr = PetscDrawSetType(draw,def);CHKERRQ(ierr);
292   }
293   ierr = PetscOptionsName("-nox","Run without graphics","None",&nox);CHKERRQ(ierr);
294 #if defined(PETSC_HAVE_X)
295   {
296     char      filename[PETSC_MAX_PATH_LEN];
297     PetscBool save,movie = PETSC_FALSE;
298     ierr = PetscOptionsBool("-draw_save_movie","Make a movie from the images saved (X Windows only)","PetscDrawSetSave",movie,&movie,NULL);CHKERRQ(ierr);
299     ierr = PetscOptionsString("-draw_save","Save graphics to file (X Windows only)","PetscDrawSetSave",filename,filename,PETSC_MAX_PATH_LEN,&save);CHKERRQ(ierr);
300     if (save) {
301       ierr = PetscDrawSetSave(draw,filename,movie);CHKERRQ(ierr);
302     }
303   }
304 #endif
305   ierr = PetscOptionsGetReal(NULL,"-draw_pause",&dpause,&flg);CHKERRQ(ierr);
306   if (flg) draw->pause = dpause;
307 
308   /* process any options handlers added with PetscObjectAddOptionsHandler() */
309   ierr = PetscObjectProcessOptionsHandlers((PetscObject)draw);CHKERRQ(ierr);
310   ierr = PetscOptionsEnd();CHKERRQ(ierr);
311   PetscFunctionReturn(0);
312 }
313 
314 #undef __FUNCT__
315 #define __FUNCT__ "PetscDrawSetSave"
316 /*@C
317    PetscDrawSave - Saves images produced in a PetscDraw into a file as a Gif file using AfterImage
318 
319    Collective on PetscDraw
320 
321    Input Parameter:
322 +  draw      - the graphics context
323 .  filename  - name of the file, if NULL uses name of draw object
324 -  movie - produce a movie of all the images
325 
326    Options Database Command:
327 +  -draw_save  <filename>
328 -  -draw_save_movie
329 
330    Level: intermediate
331 
332    Concepts: X windows^graphics
333 
334    Notes: You should call this BEFORE calling PetscDrawClear() and creating your image.
335 
336    Requires that PETSc be configured with the option --with-afterimage to save the images and ffmpeg must be in your path to make the movie
337 
338    If X windows generates an error message about X_CreateWindow() failing then Afterimage was installed without X windows. Reinstall Afterimage using the
339    ./configure flags --x-includes=/pathtoXincludes --x-libraries=/pathtoXlibraries   For example under Mac OS X Mountain Lion --x-includes=/opt/X11/include -x-libraries=/opt/X11/lib
340 
341 
342 .seealso: PetscDrawSetFromOptions(), PetscDrawCreate(), PetscDrawDestroy(), PetscDrawSave()
343 @*/
344 PetscErrorCode  PetscDrawSetSave(PetscDraw draw,const char *filename,PetscBool movie)
345 {
346   PetscErrorCode ierr;
347 
348   PetscFunctionBegin;
349   PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
350   ierr = PetscFree(draw->savefilename);CHKERRQ(ierr);
351 
352   draw->savefilemovie = movie;
353   if (filename && filename[0]) {
354     ierr = PetscStrallocpy(filename,&draw->savefilename);CHKERRQ(ierr);
355   } else {
356     const char *name;
357     ierr = PetscObjectGetName((PetscObject)draw,&name);CHKERRQ(ierr);
358     ierr = PetscStrallocpy(name,&draw->savefilename);CHKERRQ(ierr);
359   }
360   if (draw->ops->setsave) {
361     ierr = (*draw->ops->setsave)(draw,filename);CHKERRQ(ierr);
362   }
363   PetscFunctionReturn(0);
364 }
365