xref: /petsc/src/sys/classes/draw/utils/lgc.c (revision 10699b917c2ca523f62b603029bfcc018e13d799)
15c6c1daeSBarry Smith 
25ccc9f3cSMatthew G Knepley #include <petscviewer.h>
3e118a51fSLisandro Dalcin #include <../src/sys/classes/draw/utils/lgimpl.h>  /*I   "petscdraw.h"  I*/
45c6c1daeSBarry Smith PetscClassId PETSC_DRAWLG_CLASSID = 0;
55c6c1daeSBarry Smith 
65c6c1daeSBarry Smith /*@
75c6c1daeSBarry Smith    PetscDrawLGGetAxis - Gets the axis context associated with a line graph.
85c6c1daeSBarry Smith    This is useful if one wants to change some axis property, such as
95c6c1daeSBarry Smith    labels, color, etc. The axis context should not be destroyed by the
105c6c1daeSBarry Smith    application code.
115c6c1daeSBarry Smith 
125c6c1daeSBarry Smith    Not Collective, if PetscDrawLG is parallel then PetscDrawAxis is parallel
135c6c1daeSBarry Smith 
145c6c1daeSBarry Smith    Input Parameter:
155c6c1daeSBarry Smith .  lg - the line graph context
165c6c1daeSBarry Smith 
175c6c1daeSBarry Smith    Output Parameter:
185c6c1daeSBarry Smith .  axis - the axis context
195c6c1daeSBarry Smith 
205c6c1daeSBarry Smith    Level: advanced
215c6c1daeSBarry Smith 
22ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawAxis
23ba1e01c4SBarry Smith 
245c6c1daeSBarry Smith @*/
255c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGGetAxis(PetscDrawLG lg,PetscDrawAxis *axis)
265c6c1daeSBarry Smith {
275c6c1daeSBarry Smith   PetscFunctionBegin;
28e118a51fSLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
2945f3bb6eSLisandro Dalcin   PetscValidPointer(axis,2);
305c6c1daeSBarry Smith   *axis = lg->axis;
315c6c1daeSBarry Smith   PetscFunctionReturn(0);
325c6c1daeSBarry Smith }
335c6c1daeSBarry Smith 
345c6c1daeSBarry Smith /*@
355c6c1daeSBarry Smith    PetscDrawLGGetDraw - Gets the draw context associated with a line graph.
365c6c1daeSBarry Smith 
375c6c1daeSBarry Smith    Not Collective, if PetscDrawLG is parallel then PetscDraw is parallel
385c6c1daeSBarry Smith 
395c6c1daeSBarry Smith    Input Parameter:
405c6c1daeSBarry Smith .  lg - the line graph context
415c6c1daeSBarry Smith 
425c6c1daeSBarry Smith    Output Parameter:
435c6c1daeSBarry Smith .  draw - the draw context
445c6c1daeSBarry Smith 
455c6c1daeSBarry Smith    Level: intermediate
465c6c1daeSBarry Smith 
47ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDraw
485c6c1daeSBarry Smith @*/
495c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGGetDraw(PetscDrawLG lg,PetscDraw *draw)
505c6c1daeSBarry Smith {
515c6c1daeSBarry Smith   PetscFunctionBegin;
525c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
5345f3bb6eSLisandro Dalcin   PetscValidPointer(draw,2);
545c6c1daeSBarry Smith   *draw = lg->win;
555c6c1daeSBarry Smith   PetscFunctionReturn(0);
565c6c1daeSBarry Smith }
575c6c1daeSBarry Smith 
585c6c1daeSBarry Smith 
595c6c1daeSBarry Smith /*@
605c6c1daeSBarry Smith    PetscDrawLGSPDraw - Redraws a line graph.
615c6c1daeSBarry Smith 
625b399a63SLisandro Dalcin    Collective on PetscDrawLG
635c6c1daeSBarry Smith 
645c6c1daeSBarry Smith    Input Parameter:
655c6c1daeSBarry Smith .  lg - the line graph context
665c6c1daeSBarry Smith 
675c6c1daeSBarry Smith    Level: intermediate
685c6c1daeSBarry Smith 
695c6c1daeSBarry Smith .seealso: PetscDrawLGDraw(), PetscDrawSPDraw()
705c6c1daeSBarry Smith 
7195452b02SPatrick Sanan    Developer Notes:
7295452b02SPatrick Sanan     This code cheats and uses the fact that the LG and SP structs are the same
735c6c1daeSBarry Smith 
745c6c1daeSBarry Smith @*/
755c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGSPDraw(PetscDrawLG lg,PetscDrawSP spin)
765c6c1daeSBarry Smith {
775c6c1daeSBarry Smith   PetscDrawLG    sp = (PetscDrawLG)spin;
785c6c1daeSBarry Smith   PetscReal      xmin,xmax,ymin,ymax;
795c6c1daeSBarry Smith   PetscErrorCode ierr;
80e118a51fSLisandro Dalcin   PetscBool      isnull;
81e118a51fSLisandro Dalcin   PetscMPIInt    rank;
82e118a51fSLisandro Dalcin   PetscDraw      draw;
835c6c1daeSBarry Smith 
845c6c1daeSBarry Smith   PetscFunctionBegin;
855c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
865c6c1daeSBarry Smith   PetscValidHeaderSpecific(sp,PETSC_DRAWSP_CLASSID,2);
875b399a63SLisandro Dalcin   ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr);
88e118a51fSLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
89ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)lg),&rank);CHKERRMPI(ierr);
905c6c1daeSBarry Smith 
915b399a63SLisandro Dalcin   draw = lg->win;
92e118a51fSLisandro Dalcin   ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
935b399a63SLisandro Dalcin   ierr = PetscDrawClear(draw);CHKERRQ(ierr);
94e118a51fSLisandro Dalcin 
95e118a51fSLisandro Dalcin   xmin = PetscMin(lg->xmin,sp->xmin); ymin = PetscMin(lg->ymin,sp->ymin);
96e118a51fSLisandro Dalcin   xmax = PetscMax(lg->xmax,sp->xmax); ymax = PetscMax(lg->ymax,sp->ymax);
975c6c1daeSBarry Smith   ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
985c6c1daeSBarry Smith   ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr);
995c6c1daeSBarry Smith 
1005b399a63SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1015c6c1daeSBarry Smith   if (!rank) {
102e118a51fSLisandro Dalcin     int i,j,dim,nopts;
1035c6c1daeSBarry Smith     dim   = lg->dim;
1045c6c1daeSBarry Smith     nopts = lg->nopts;
1055c6c1daeSBarry Smith     for (i=0; i<dim; i++) {
1065c6c1daeSBarry Smith       for (j=1; j<nopts; j++) {
1075c6c1daeSBarry Smith         ierr = PetscDrawLine(draw,lg->x[(j-1)*dim+i],lg->y[(j-1)*dim+i],lg->x[j*dim+i],lg->y[j*dim+i],PETSC_DRAW_BLACK+i);CHKERRQ(ierr);
108b6fe0379SLisandro Dalcin         if (lg->use_markers) {
10978e9f83eSLisandro Dalcin           ierr = PetscDrawMarker(draw,lg->x[j*dim+i],lg->y[j*dim+i],PETSC_DRAW_RED);CHKERRQ(ierr);
1105c6c1daeSBarry Smith         }
1115c6c1daeSBarry Smith       }
1125c6c1daeSBarry Smith     }
1135c6c1daeSBarry Smith     dim   = sp->dim;
1145c6c1daeSBarry Smith     nopts = sp->nopts;
1155c6c1daeSBarry Smith     for (i=0; i<dim; i++) {
1165c6c1daeSBarry Smith       for (j=0; j<nopts; j++) {
11778e9f83eSLisandro Dalcin         ierr = PetscDrawMarker(draw,sp->x[j*dim+i],sp->y[j*dim+i],PETSC_DRAW_RED);CHKERRQ(ierr);
1185c6c1daeSBarry Smith       }
1195c6c1daeSBarry Smith     }
1205c6c1daeSBarry Smith   }
121e118a51fSLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1225b399a63SLisandro Dalcin 
1235b399a63SLisandro Dalcin   ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
1245b399a63SLisandro Dalcin   ierr = PetscDrawPause(draw);CHKERRQ(ierr);
1255c6c1daeSBarry Smith   PetscFunctionReturn(0);
1265c6c1daeSBarry Smith }
1275c6c1daeSBarry Smith 
1285c6c1daeSBarry Smith 
1295c6c1daeSBarry Smith /*@
1305c6c1daeSBarry Smith     PetscDrawLGCreate - Creates a line graph data structure.
1315c6c1daeSBarry Smith 
1325b399a63SLisandro Dalcin     Collective on PetscDraw
1335c6c1daeSBarry Smith 
1345c6c1daeSBarry Smith     Input Parameters:
1355c6c1daeSBarry Smith +   draw - the window where the graph will be made.
1365c6c1daeSBarry Smith -   dim - the number of curves which will be drawn
1375c6c1daeSBarry Smith 
1385c6c1daeSBarry Smith     Output Parameters:
139e118a51fSLisandro Dalcin .   outlg - the line graph context
1405c6c1daeSBarry Smith 
1415c6c1daeSBarry Smith     Level: intermediate
1425c6c1daeSBarry Smith 
14395452b02SPatrick Sanan     Notes:
14495452b02SPatrick Sanan     The MPI communicator that owns the PetscDraw owns this PetscDrawLG, but the calls to set options and add points are ignored on all processes except the
1457e25d57eSBarry Smith            zeroth MPI process in the communicator. All MPI processes in the communicator must call PetscDrawLGDraw() to display the updated graph.
1467e25d57eSBarry Smith 
147ba1e01c4SBarry Smith .seealso:  PetscDrawLGDestroy(), PetscDrawLGAddPoint(), PetscDrawLGAddCommonPoint(), PetscDrawLGAddPoints(), PetscDrawLGDraw(), PetscDrawLGSave(),
148ba1e01c4SBarry Smith            PetscDrawLGView(), PetscDrawLGReset(), PetscDrawLGSetDimension(), PetscDrawLGGetDimension(), PetscDrawLGSetLegend(), PetscDrawLGGetAxis(),
149ba1e01c4SBarry Smith            PetscDrawLGGetDraw(), PetscDrawLGSetUseMarkers(), PetscDrawLGSetLimits(), PetscDrawLGSetColors(), PetscDrawLGSetOptionsPrefix(), PetscDrawLGSetFromOptions()
1505c6c1daeSBarry Smith @*/
151e118a51fSLisandro Dalcin PetscErrorCode  PetscDrawLGCreate(PetscDraw draw,PetscInt dim,PetscDrawLG *outlg)
1525c6c1daeSBarry Smith {
1535c6c1daeSBarry Smith   PetscDrawLG    lg;
154e118a51fSLisandro Dalcin   PetscErrorCode ierr;
1555c6c1daeSBarry Smith 
1565c6c1daeSBarry Smith   PetscFunctionBegin;
1575c6c1daeSBarry Smith   PetscValidHeaderSpecific(draw,PETSC_DRAW_CLASSID,1);
158e118a51fSLisandro Dalcin   PetscValidLogicalCollectiveInt(draw,dim,2);
159e118a51fSLisandro Dalcin   PetscValidPointer(outlg,3);
160e118a51fSLisandro Dalcin 
16171917b75SLisandro Dalcin   ierr = PetscHeaderCreate(lg,PETSC_DRAWLG_CLASSID,"DrawLG","Line Graph","Draw",PetscObjectComm((PetscObject)draw),PetscDrawLGDestroy,NULL);CHKERRQ(ierr);
162e118a51fSLisandro Dalcin   ierr = PetscLogObjectParent((PetscObject)draw,(PetscObject)lg);CHKERRQ(ierr);
163c2bac407SLisandro Dalcin   ierr = PetscDrawLGSetOptionsPrefix(lg,((PetscObject)draw)->prefix);CHKERRQ(ierr);
164e118a51fSLisandro Dalcin 
165e118a51fSLisandro Dalcin   ierr = PetscObjectReference((PetscObject)draw);CHKERRQ(ierr);
166e118a51fSLisandro Dalcin   lg->win = draw;
167a297a907SKarl Rupp 
16869c83917SLisandro Dalcin   lg->view    = NULL;
16969c83917SLisandro Dalcin   lg->destroy = NULL;
1705c6c1daeSBarry Smith   lg->nopts   = 0;
1715c6c1daeSBarry Smith   lg->dim     = dim;
1725c6c1daeSBarry Smith   lg->xmin    = 1.e20;
1735c6c1daeSBarry Smith   lg->ymin    = 1.e20;
1745c6c1daeSBarry Smith   lg->xmax    = -1.e20;
1755c6c1daeSBarry Smith   lg->ymax    = -1.e20;
176a297a907SKarl Rupp 
177dcca6d9dSJed Brown   ierr = PetscMalloc2(dim*CHUNCKSIZE,&lg->x,dim*CHUNCKSIZE,&lg->y);CHKERRQ(ierr);
1783bb1ff40SBarry Smith   ierr = PetscLogObjectMemory((PetscObject)lg,2*dim*CHUNCKSIZE*sizeof(PetscReal));CHKERRQ(ierr);
179a297a907SKarl Rupp 
1805c6c1daeSBarry Smith   lg->len         = dim*CHUNCKSIZE;
1815c6c1daeSBarry Smith   lg->loc         = 0;
182b6fe0379SLisandro Dalcin   lg->use_markers = PETSC_FALSE;
183a297a907SKarl Rupp 
1845c6c1daeSBarry Smith   ierr = PetscDrawAxisCreate(draw,&lg->axis);CHKERRQ(ierr);
1853bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)lg,(PetscObject)lg->axis);CHKERRQ(ierr);
186a297a907SKarl Rupp 
187e118a51fSLisandro Dalcin   *outlg = lg;
1885c6c1daeSBarry Smith   PetscFunctionReturn(0);
1895c6c1daeSBarry Smith }
1905c6c1daeSBarry Smith 
1915c6c1daeSBarry Smith /*@
1925c6c1daeSBarry Smith    PetscDrawLGSetColors - Sets the color of each line graph drawn
1935c6c1daeSBarry Smith 
1945b399a63SLisandro Dalcin    Logically Collective on PetscDrawLG
1955c6c1daeSBarry Smith 
1965c6c1daeSBarry Smith    Input Parameter:
1975c6c1daeSBarry Smith +  lg - the line graph context.
1985c6c1daeSBarry Smith -  colors - the colors
1995c6c1daeSBarry Smith 
2005c6c1daeSBarry Smith    Level: intermediate
2015c6c1daeSBarry Smith 
202ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate()
203ba1e01c4SBarry Smith 
2045c6c1daeSBarry Smith @*/
20545f3bb6eSLisandro Dalcin PetscErrorCode  PetscDrawLGSetColors(PetscDrawLG lg,const int colors[])
2065c6c1daeSBarry Smith {
2075c6c1daeSBarry Smith   PetscErrorCode ierr;
2085c6c1daeSBarry Smith 
2095c6c1daeSBarry Smith   PetscFunctionBegin;
2105c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
21145f3bb6eSLisandro Dalcin   if (lg->dim) PetscValidIntPointer(colors,2);
212e118a51fSLisandro Dalcin 
2135c6c1daeSBarry Smith   ierr = PetscFree(lg->colors);CHKERRQ(ierr);
214785e854fSJed Brown   ierr = PetscMalloc1(lg->dim,&lg->colors);CHKERRQ(ierr);
215580bdb30SBarry Smith   ierr = PetscArraycpy(lg->colors,colors,lg->dim);CHKERRQ(ierr);
2165c6c1daeSBarry Smith   PetscFunctionReturn(0);
2175c6c1daeSBarry Smith }
2185c6c1daeSBarry Smith 
2195c6c1daeSBarry Smith /*@C
2205c6c1daeSBarry Smith    PetscDrawLGSetLegend - sets the names of each curve plotted
2215c6c1daeSBarry Smith 
2225b399a63SLisandro Dalcin    Logically Collective on PetscDrawLG
2235c6c1daeSBarry Smith 
2245c6c1daeSBarry Smith    Input Parameter:
2255c6c1daeSBarry Smith +  lg - the line graph context.
2265c6c1daeSBarry Smith -  names - the names for each curve
2275c6c1daeSBarry Smith 
2285c6c1daeSBarry Smith    Level: intermediate
2295c6c1daeSBarry Smith 
23095452b02SPatrick Sanan    Notes:
23195452b02SPatrick Sanan     Call PetscDrawLGGetAxis() and then change properties of the PetscDrawAxis for detailed control of the plot
232ba1e01c4SBarry Smith 
233ba1e01c4SBarry Smith .seealso: PetscDrawLGGetAxis(), PetscDrawAxis, PetscDrawAxisSetColors(), PetscDrawAxisSetLabels(), PetscDrawAxisSetHoldLimits()
234ba1e01c4SBarry Smith 
2355c6c1daeSBarry Smith @*/
2365c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGSetLegend(PetscDrawLG lg,const char *const *names)
2375c6c1daeSBarry Smith {
2385c6c1daeSBarry Smith   PetscErrorCode ierr;
2395c6c1daeSBarry Smith   PetscInt       i;
2405c6c1daeSBarry Smith 
2415c6c1daeSBarry Smith   PetscFunctionBegin;
2425c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
24345f3bb6eSLisandro Dalcin   if (names) PetscValidPointer(names,2);
2445c6c1daeSBarry Smith 
2455c6c1daeSBarry Smith   if (lg->legend) {
2465c6c1daeSBarry Smith     for (i=0; i<lg->dim; i++) {
2475c6c1daeSBarry Smith       ierr = PetscFree(lg->legend[i]);CHKERRQ(ierr);
2485c6c1daeSBarry Smith     }
2495c6c1daeSBarry Smith     ierr = PetscFree(lg->legend);CHKERRQ(ierr);
2505c6c1daeSBarry Smith   }
2515c6c1daeSBarry Smith   if (names) {
252785e854fSJed Brown     ierr = PetscMalloc1(lg->dim,&lg->legend);CHKERRQ(ierr);
2535c6c1daeSBarry Smith     for (i=0; i<lg->dim; i++) {
2545c6c1daeSBarry Smith       ierr = PetscStrallocpy(names[i],&lg->legend[i]);CHKERRQ(ierr);
2555c6c1daeSBarry Smith     }
2565c6c1daeSBarry Smith   }
2575c6c1daeSBarry Smith   PetscFunctionReturn(0);
2585c6c1daeSBarry Smith }
2595c6c1daeSBarry Smith 
2605c6c1daeSBarry Smith /*@
2615c6c1daeSBarry Smith    PetscDrawLGGetDimension - Change the number of lines that are to be drawn.
2625c6c1daeSBarry Smith 
2635b399a63SLisandro Dalcin    Not Collective
2645c6c1daeSBarry Smith 
2655c6c1daeSBarry Smith    Input Parameter:
2665c6c1daeSBarry Smith .  lg - the line graph context.
2675c6c1daeSBarry Smith 
2685c6c1daeSBarry Smith    Output Parameter:
2695c6c1daeSBarry Smith .  dim - the number of curves.
2705c6c1daeSBarry Smith 
2715c6c1daeSBarry Smith    Level: intermediate
2725c6c1daeSBarry Smith 
273ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawLGSetDimension()
274ba1e01c4SBarry Smith 
2755c6c1daeSBarry Smith @*/
2765c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGGetDimension(PetscDrawLG lg,PetscInt *dim)
2775c6c1daeSBarry Smith {
2785c6c1daeSBarry Smith   PetscFunctionBegin;
2795c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
28045f3bb6eSLisandro Dalcin   PetscValidIntPointer(dim,2);
2815c6c1daeSBarry Smith   *dim = lg->dim;
2825c6c1daeSBarry Smith   PetscFunctionReturn(0);
2835c6c1daeSBarry Smith }
2845c6c1daeSBarry Smith 
2855c6c1daeSBarry Smith /*@
2865c6c1daeSBarry Smith    PetscDrawLGSetDimension - Change the number of lines that are to be drawn.
2875c6c1daeSBarry Smith 
2885b399a63SLisandro Dalcin    Logically Collective on PetscDrawLG
2895c6c1daeSBarry Smith 
2905c6c1daeSBarry Smith    Input Parameter:
2915c6c1daeSBarry Smith +  lg - the line graph context.
2925c6c1daeSBarry Smith -  dim - the number of curves.
2935c6c1daeSBarry Smith 
2945c6c1daeSBarry Smith    Level: intermediate
2955c6c1daeSBarry Smith 
296ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate(), PetscDrawLGGetDimension()
2975c6c1daeSBarry Smith @*/
2985c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGSetDimension(PetscDrawLG lg,PetscInt dim)
2995c6c1daeSBarry Smith {
3005c6c1daeSBarry Smith   PetscErrorCode ierr;
3015c6c1daeSBarry Smith   PetscInt       i;
3025c6c1daeSBarry Smith 
3035c6c1daeSBarry Smith   PetscFunctionBegin;
3045c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
3055c6c1daeSBarry Smith   PetscValidLogicalCollectiveInt(lg,dim,2);
3065c6c1daeSBarry Smith   if (lg->dim == dim) PetscFunctionReturn(0);
3075c6c1daeSBarry Smith 
3085c6c1daeSBarry Smith   ierr = PetscFree2(lg->x,lg->y);CHKERRQ(ierr);
3095c6c1daeSBarry Smith   if (lg->legend) {
3105c6c1daeSBarry Smith     for (i=0; i<lg->dim; i++) {
3115c6c1daeSBarry Smith       ierr = PetscFree(lg->legend[i]);CHKERRQ(ierr);
3125c6c1daeSBarry Smith     }
3135c6c1daeSBarry Smith     ierr = PetscFree(lg->legend);CHKERRQ(ierr);
3145c6c1daeSBarry Smith   }
3155c6c1daeSBarry Smith   ierr    = PetscFree(lg->colors);CHKERRQ(ierr);
3165c6c1daeSBarry Smith   lg->dim = dim;
317dcca6d9dSJed Brown   ierr    = PetscMalloc2(dim*CHUNCKSIZE,&lg->x,dim*CHUNCKSIZE,&lg->y);CHKERRQ(ierr);
3183bb1ff40SBarry Smith   ierr    = PetscLogObjectMemory((PetscObject)lg,2*dim*CHUNCKSIZE*sizeof(PetscReal));CHKERRQ(ierr);
3195c6c1daeSBarry Smith   lg->len = dim*CHUNCKSIZE;
3205c6c1daeSBarry Smith   PetscFunctionReturn(0);
3215c6c1daeSBarry Smith }
3225c6c1daeSBarry Smith 
32371917b75SLisandro Dalcin 
32471917b75SLisandro Dalcin /*@
32571917b75SLisandro Dalcin    PetscDrawLGSetLimits - Sets the axis limits for a line graph. If more
32671917b75SLisandro Dalcin    points are added after this call, the limits will be adjusted to
32771917b75SLisandro Dalcin    include those additional points.
32871917b75SLisandro Dalcin 
32971917b75SLisandro Dalcin    Logically Collective on PetscDrawLG
33071917b75SLisandro Dalcin 
33171917b75SLisandro Dalcin    Input Parameters:
33271917b75SLisandro Dalcin +  xlg - the line graph context
33371917b75SLisandro Dalcin -  x_min,x_max,y_min,y_max - the limits
33471917b75SLisandro Dalcin 
33571917b75SLisandro Dalcin    Level: intermediate
33671917b75SLisandro Dalcin 
337ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate()
338ba1e01c4SBarry Smith 
33971917b75SLisandro Dalcin @*/
34071917b75SLisandro Dalcin PetscErrorCode  PetscDrawLGSetLimits(PetscDrawLG lg,PetscReal x_min,PetscReal x_max,PetscReal y_min,PetscReal y_max)
34171917b75SLisandro Dalcin {
34271917b75SLisandro Dalcin   PetscFunctionBegin;
34371917b75SLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
34471917b75SLisandro Dalcin 
34571917b75SLisandro Dalcin   (lg)->xmin = x_min;
34671917b75SLisandro Dalcin   (lg)->xmax = x_max;
34771917b75SLisandro Dalcin   (lg)->ymin = y_min;
34871917b75SLisandro Dalcin   (lg)->ymax = y_max;
34971917b75SLisandro Dalcin   PetscFunctionReturn(0);
35071917b75SLisandro Dalcin }
35171917b75SLisandro Dalcin 
3525c6c1daeSBarry Smith /*@
3535c6c1daeSBarry Smith    PetscDrawLGReset - Clears line graph to allow for reuse with new data.
3545c6c1daeSBarry Smith 
3555b399a63SLisandro Dalcin    Logically Collective on PetscDrawLG
3565c6c1daeSBarry Smith 
3575c6c1daeSBarry Smith    Input Parameter:
3585c6c1daeSBarry Smith .  lg - the line graph context.
3595c6c1daeSBarry Smith 
3605c6c1daeSBarry Smith    Level: intermediate
3615c6c1daeSBarry Smith 
362ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate()
3635c6c1daeSBarry Smith @*/
3645c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGReset(PetscDrawLG lg)
3655c6c1daeSBarry Smith {
3665c6c1daeSBarry Smith   PetscFunctionBegin;
3675c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
3685c6c1daeSBarry Smith   lg->xmin  = 1.e20;
3695c6c1daeSBarry Smith   lg->ymin  = 1.e20;
3705c6c1daeSBarry Smith   lg->xmax  = -1.e20;
3715c6c1daeSBarry Smith   lg->ymax  = -1.e20;
3725c6c1daeSBarry Smith   lg->loc   = 0;
3735c6c1daeSBarry Smith   lg->nopts = 0;
3745c6c1daeSBarry Smith   PetscFunctionReturn(0);
3755c6c1daeSBarry Smith }
3765c6c1daeSBarry Smith 
3775c6c1daeSBarry Smith /*@
3785c6c1daeSBarry Smith    PetscDrawLGDestroy - Frees all space taken up by line graph data structure.
3795c6c1daeSBarry Smith 
3805b399a63SLisandro Dalcin    Collective on PetscDrawLG
3815c6c1daeSBarry Smith 
3825c6c1daeSBarry Smith    Input Parameter:
3835c6c1daeSBarry Smith .  lg - the line graph context
3845c6c1daeSBarry Smith 
3855c6c1daeSBarry Smith    Level: intermediate
3865c6c1daeSBarry Smith 
3875c6c1daeSBarry Smith .seealso:  PetscDrawLGCreate()
3885c6c1daeSBarry Smith @*/
3895c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGDestroy(PetscDrawLG *lg)
3905c6c1daeSBarry Smith {
3915c6c1daeSBarry Smith   PetscErrorCode ierr;
3925c6c1daeSBarry Smith   PetscInt       i;
3935c6c1daeSBarry Smith 
3945c6c1daeSBarry Smith   PetscFunctionBegin;
3955c6c1daeSBarry Smith   if (!*lg) PetscFunctionReturn(0);
396e118a51fSLisandro Dalcin   PetscValidHeaderSpecific(*lg,PETSC_DRAWLG_CLASSID,1);
397e118a51fSLisandro Dalcin   if (--((PetscObject)(*lg))->refct > 0) {*lg = NULL; PetscFunctionReturn(0);}
3985c6c1daeSBarry Smith 
3995c6c1daeSBarry Smith   if ((*lg)->legend) {
4005c6c1daeSBarry Smith     for (i=0; i<(*lg)->dim; i++) {
4015c6c1daeSBarry Smith       ierr = PetscFree((*lg)->legend[i]);CHKERRQ(ierr);
4025c6c1daeSBarry Smith     }
4035c6c1daeSBarry Smith     ierr = PetscFree((*lg)->legend);CHKERRQ(ierr);
4045c6c1daeSBarry Smith   }
4055c6c1daeSBarry Smith   ierr = PetscFree((*lg)->colors);CHKERRQ(ierr);
4065c6c1daeSBarry Smith   ierr = PetscFree2((*lg)->x,(*lg)->y);CHKERRQ(ierr);
407e118a51fSLisandro Dalcin   ierr = PetscDrawAxisDestroy(&(*lg)->axis);CHKERRQ(ierr);
408e118a51fSLisandro Dalcin   ierr = PetscDrawDestroy(&(*lg)->win);CHKERRQ(ierr);
4095c6c1daeSBarry Smith   ierr = PetscHeaderDestroy(lg);CHKERRQ(ierr);
4105c6c1daeSBarry Smith   PetscFunctionReturn(0);
4115c6c1daeSBarry Smith }
4125c6c1daeSBarry Smith /*@
413b6fe0379SLisandro Dalcin    PetscDrawLGSetUseMarkers - Causes LG to draw a marker for each data-point.
4145c6c1daeSBarry Smith 
4155b399a63SLisandro Dalcin    Logically Collective on PetscDrawLG
4165c6c1daeSBarry Smith 
4175c6c1daeSBarry Smith    Input Parameters:
418287de1a7SBarry Smith +  lg - the linegraph context
419287de1a7SBarry Smith -  flg - should mark each data point
420287de1a7SBarry Smith 
421287de1a7SBarry Smith    Options Database:
422*10699b91SBarry Smith .  -lg_use_markers  <true,false> - true means the graphPetscDrawLG draws a marker for each point
4235c6c1daeSBarry Smith 
4245c6c1daeSBarry Smith    Level: intermediate
4255c6c1daeSBarry Smith 
426ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate()
4275c6c1daeSBarry Smith @*/
428b6fe0379SLisandro Dalcin PetscErrorCode  PetscDrawLGSetUseMarkers(PetscDrawLG lg,PetscBool flg)
4295c6c1daeSBarry Smith {
4305c6c1daeSBarry Smith   PetscFunctionBegin;
431e118a51fSLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
43245f3bb6eSLisandro Dalcin   PetscValidLogicalCollectiveBool(lg,flg,2);
433b6fe0379SLisandro Dalcin   lg->use_markers = flg;
4345c6c1daeSBarry Smith   PetscFunctionReturn(0);
4355c6c1daeSBarry Smith }
4365c6c1daeSBarry Smith 
4375c6c1daeSBarry Smith /*@
4385c6c1daeSBarry Smith    PetscDrawLGDraw - Redraws a line graph.
4395c6c1daeSBarry Smith 
4405b399a63SLisandro Dalcin    Collective on PetscDrawLG
4415c6c1daeSBarry Smith 
4425c6c1daeSBarry Smith    Input Parameter:
4435c6c1daeSBarry Smith .  lg - the line graph context
4445c6c1daeSBarry Smith 
4455c6c1daeSBarry Smith    Level: intermediate
4465c6c1daeSBarry Smith 
447ba1e01c4SBarry Smith .seealso: PetscDrawSPDraw(), PetscDrawLGSPDraw(), PetscDrawLGReset()
4485c6c1daeSBarry Smith @*/
4495c6c1daeSBarry Smith PetscErrorCode  PetscDrawLGDraw(PetscDrawLG lg)
4505c6c1daeSBarry Smith {
451e118a51fSLisandro Dalcin   PetscReal      xmin,xmax,ymin,ymax;
4525c6c1daeSBarry Smith   PetscErrorCode ierr;
453e118a51fSLisandro Dalcin   PetscMPIInt    rank;
454e118a51fSLisandro Dalcin   PetscDraw      draw;
4555c6c1daeSBarry Smith   PetscBool      isnull;
4565c6c1daeSBarry Smith 
4575c6c1daeSBarry Smith   PetscFunctionBegin;
4585c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
4598f69470aSLisandro Dalcin   ierr = PetscDrawIsNull(lg->win,&isnull);CHKERRQ(ierr);
4608f69470aSLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
461ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)lg),&rank);CHKERRMPI(ierr);
4625c6c1daeSBarry Smith 
4635b399a63SLisandro Dalcin   draw = lg->win;
4645c6c1daeSBarry Smith   ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
4655b399a63SLisandro Dalcin   ierr = PetscDrawClear(draw);CHKERRQ(ierr);
466e118a51fSLisandro Dalcin 
467e118a51fSLisandro Dalcin   xmin = lg->xmin; xmax = lg->xmax; ymin = lg->ymin; ymax = lg->ymax;
4685c6c1daeSBarry Smith   ierr = PetscDrawAxisSetLimits(lg->axis,xmin,xmax,ymin,ymax);CHKERRQ(ierr);
4695c6c1daeSBarry Smith   ierr = PetscDrawAxisDraw(lg->axis);CHKERRQ(ierr);
4705c6c1daeSBarry Smith 
4715b399a63SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
4725c6c1daeSBarry Smith   if (!rank) {
473e118a51fSLisandro Dalcin     int i,j,dim=lg->dim,nopts=lg->nopts,cl;
4745c6c1daeSBarry Smith     for (i=0; i<dim; i++) {
4755c6c1daeSBarry Smith       for (j=1; j<nopts; j++) {
4760ed3bfb6SBarry Smith         cl   = lg->colors ? lg->colors[i] : ((PETSC_DRAW_BLACK + i) % PETSC_DRAW_MAXCOLOR);
4775c6c1daeSBarry Smith         ierr = PetscDrawLine(draw,lg->x[(j-1)*dim+i],lg->y[(j-1)*dim+i],lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);
47871917b75SLisandro Dalcin         if (lg->use_markers) {ierr = PetscDrawMarker(draw,lg->x[j*dim+i],lg->y[j*dim+i],cl);CHKERRQ(ierr);}
4795c6c1daeSBarry Smith       }
4805c6c1daeSBarry Smith     }
4815c6c1daeSBarry Smith   }
48256a02f31SLisandro Dalcin   if (!rank && lg->legend) {
483e118a51fSLisandro Dalcin     int       i,dim=lg->dim,cl;
4845c6c1daeSBarry Smith     PetscReal xl,yl,xr,yr,tw,th;
48571917b75SLisandro Dalcin     size_t    slen,len=0;
48671917b75SLisandro Dalcin     ierr = PetscDrawAxisGetLimits(lg->axis,&xl,&xr,&yl,&yr);CHKERRQ(ierr);
4875c6c1daeSBarry Smith     ierr = PetscDrawStringGetSize(draw,&tw,&th);CHKERRQ(ierr);
4885c6c1daeSBarry Smith     for (i=0; i<dim; i++) {
48971917b75SLisandro Dalcin       ierr = PetscStrlen(lg->legend[i],&slen);CHKERRQ(ierr);
49071917b75SLisandro Dalcin       len = PetscMax(len,slen);
4915c6c1daeSBarry Smith     }
49271917b75SLisandro Dalcin     xr = xr - 1.5*tw; xl = xr - (len + 7)*tw;
49371917b75SLisandro Dalcin     yr = yr - 1.0*th; yl = yr - (dim + 1)*th;
49471917b75SLisandro Dalcin     ierr = PetscDrawLine(draw,xl,yl,xr,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr);
49571917b75SLisandro Dalcin     ierr = PetscDrawLine(draw,xr,yl,xr,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr);
49671917b75SLisandro Dalcin     ierr = PetscDrawLine(draw,xr,yr,xl,yr,PETSC_DRAW_BLACK);CHKERRQ(ierr);
49771917b75SLisandro Dalcin     ierr = PetscDrawLine(draw,xl,yr,xl,yl,PETSC_DRAW_BLACK);CHKERRQ(ierr);
4985c6c1daeSBarry Smith     for  (i=0; i<dim; i++) {
49971917b75SLisandro Dalcin       cl   = lg->colors ? lg->colors[i] : (PETSC_DRAW_BLACK + i);
50071917b75SLisandro Dalcin       ierr = PetscDrawLine(draw,xl + 1*tw,yr - (i + 1)*th,xl + 5*tw,yr - (i + 1)*th,cl);CHKERRQ(ierr);
50171917b75SLisandro Dalcin       ierr = PetscDrawString(draw,xl + 6*tw,yr - (i + 1.5)*th,PETSC_DRAW_BLACK,lg->legend[i]);CHKERRQ(ierr);
5025c6c1daeSBarry Smith     }
5035c6c1daeSBarry Smith   }
504e118a51fSLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
5055b399a63SLisandro Dalcin 
5065b399a63SLisandro Dalcin   ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
507e118a51fSLisandro Dalcin   ierr = PetscDrawPause(draw);CHKERRQ(ierr);
5085c6c1daeSBarry Smith   PetscFunctionReturn(0);
5095c6c1daeSBarry Smith }
5105c6c1daeSBarry Smith 
51157fd6651SLisandro Dalcin /*@
51257fd6651SLisandro Dalcin   PetscDrawLGSave - Saves a drawn image
51357fd6651SLisandro Dalcin 
51457fd6651SLisandro Dalcin   Collective on PetscDrawLG
51557fd6651SLisandro Dalcin 
51657fd6651SLisandro Dalcin   Input Parameter:
51757fd6651SLisandro Dalcin . lg - The line graph context
51857fd6651SLisandro Dalcin 
51957fd6651SLisandro Dalcin   Level: intermediate
52057fd6651SLisandro Dalcin 
52157fd6651SLisandro Dalcin .seealso:  PetscDrawLGCreate(), PetscDrawLGGetDraw(), PetscDrawSetSave(), PetscDrawSave()
52257fd6651SLisandro Dalcin @*/
52357fd6651SLisandro Dalcin PetscErrorCode  PetscDrawLGSave(PetscDrawLG lg)
52457fd6651SLisandro Dalcin {
52557fd6651SLisandro Dalcin   PetscErrorCode ierr;
52657fd6651SLisandro Dalcin 
52757fd6651SLisandro Dalcin   PetscFunctionBegin;
52857fd6651SLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
52957fd6651SLisandro Dalcin   ierr = PetscDrawSave(lg->win);CHKERRQ(ierr);
53057fd6651SLisandro Dalcin   PetscFunctionReturn(0);
53157fd6651SLisandro Dalcin }
53257fd6651SLisandro Dalcin 
5335c6c1daeSBarry Smith /*@
53434a5a0e3SBarry Smith   PetscDrawLGView - Prints a line graph.
5355c6c1daeSBarry Smith 
5365b399a63SLisandro Dalcin   Collective on PetscDrawLG
5375c6c1daeSBarry Smith 
5385c6c1daeSBarry Smith   Input Parameter:
5395c6c1daeSBarry Smith . lg - the line graph context
5405c6c1daeSBarry Smith 
5415c6c1daeSBarry Smith   Level: beginner
5425c6c1daeSBarry Smith 
543ba1e01c4SBarry Smith .seealso: PetscDrawLGCreate()
544ba1e01c4SBarry Smith 
5455c6c1daeSBarry Smith @*/
54634a5a0e3SBarry Smith PetscErrorCode  PetscDrawLGView(PetscDrawLG lg,PetscViewer viewer)
5475c6c1daeSBarry Smith {
5485c6c1daeSBarry Smith   PetscReal      xmin=lg->xmin, xmax=lg->xmax, ymin=lg->ymin, ymax=lg->ymax;
54934a5a0e3SBarry Smith   PetscInt       i, j, dim = lg->dim, nopts = lg->nopts;
55034a5a0e3SBarry Smith   PetscErrorCode ierr;
5515c6c1daeSBarry Smith 
5525c6c1daeSBarry Smith   PetscFunctionBegin;
5535c6c1daeSBarry Smith   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
554e118a51fSLisandro Dalcin 
5555c6c1daeSBarry Smith   if (nopts < 1)                  PetscFunctionReturn(0);
5565c6c1daeSBarry Smith   if (xmin > xmax || ymin > ymax) PetscFunctionReturn(0);
5575c6c1daeSBarry Smith 
55834a5a0e3SBarry Smith   if (!viewer){
55934a5a0e3SBarry Smith     ierr = PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)lg),&viewer);CHKERRQ(ierr);
56034a5a0e3SBarry Smith   }
56198c3331eSBarry Smith   ierr = PetscObjectPrintClassNamePrefixType((PetscObject)lg,viewer);CHKERRQ(ierr);
5625c6c1daeSBarry Smith   for (i = 0; i < dim; i++) {
56334a5a0e3SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer, "Line %D>\n", i);CHKERRQ(ierr);
5645c6c1daeSBarry Smith     for (j = 0; j < nopts; j++) {
56534a5a0e3SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer, "  X: %g Y: %g\n", (double)lg->x[j*dim+i], (double)lg->y[j*dim+i]);CHKERRQ(ierr);
5665c6c1daeSBarry Smith     }
5675c6c1daeSBarry Smith   }
5685c6c1daeSBarry Smith   PetscFunctionReturn(0);
5695c6c1daeSBarry Smith }
570287de1a7SBarry Smith 
571c2bac407SLisandro Dalcin /*@C
572c2bac407SLisandro Dalcin    PetscDrawLGSetOptionsPrefix - Sets the prefix used for searching for all
573c2bac407SLisandro Dalcin    PetscDrawLG options in the database.
574c2bac407SLisandro Dalcin 
575c2bac407SLisandro Dalcin    Logically Collective on PetscDrawLG
576c2bac407SLisandro Dalcin 
577c2bac407SLisandro Dalcin    Input Parameter:
578c2bac407SLisandro Dalcin +  lg - the line graph context
579c2bac407SLisandro Dalcin -  prefix - the prefix to prepend to all option names
580c2bac407SLisandro Dalcin 
581c2bac407SLisandro Dalcin    Level: advanced
582c2bac407SLisandro Dalcin 
583ba1e01c4SBarry Smith .seealso: PetscDrawLGSetFromOptions(), PetscDrawLGCreate()
584c2bac407SLisandro Dalcin @*/
585c2bac407SLisandro Dalcin PetscErrorCode  PetscDrawLGSetOptionsPrefix(PetscDrawLG lg,const char prefix[])
586c2bac407SLisandro Dalcin {
587c2bac407SLisandro Dalcin   PetscErrorCode ierr;
588c2bac407SLisandro Dalcin 
589c2bac407SLisandro Dalcin   PetscFunctionBegin;
590c2bac407SLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
591c2bac407SLisandro Dalcin   ierr = PetscObjectSetOptionsPrefix((PetscObject)lg,prefix);CHKERRQ(ierr);
592c2bac407SLisandro Dalcin   PetscFunctionReturn(0);
593c2bac407SLisandro Dalcin }
594c2bac407SLisandro Dalcin 
595287de1a7SBarry Smith /*@
596287de1a7SBarry Smith     PetscDrawLGSetFromOptions - Sets options related to the PetscDrawLG
597287de1a7SBarry Smith 
5985b399a63SLisandro Dalcin     Collective on PetscDrawLG
599287de1a7SBarry Smith 
600287de1a7SBarry Smith     Options Database:
601287de1a7SBarry Smith 
602287de1a7SBarry Smith     Level: intermediate
603287de1a7SBarry Smith 
604287de1a7SBarry Smith .seealso:  PetscDrawLGDestroy(), PetscDrawLGCreate()
605287de1a7SBarry Smith @*/
606287de1a7SBarry Smith PetscErrorCode  PetscDrawLGSetFromOptions(PetscDrawLG lg)
607287de1a7SBarry Smith {
608287de1a7SBarry Smith   PetscErrorCode      ierr;
60971917b75SLisandro Dalcin   PetscBool           usemarkers,set;
61071917b75SLisandro Dalcin   PetscDrawMarkerType markertype;
611287de1a7SBarry Smith 
612287de1a7SBarry Smith   PetscFunctionBegin;
613e118a51fSLisandro Dalcin   PetscValidHeaderSpecific(lg,PETSC_DRAWLG_CLASSID,1);
614e118a51fSLisandro Dalcin 
61571917b75SLisandro Dalcin   ierr = PetscDrawGetMarkerType(lg->win,&markertype);CHKERRQ(ierr);
61671917b75SLisandro Dalcin   ierr = PetscOptionsGetEnum(((PetscObject)lg)->options,((PetscObject)lg)->prefix,"-lg_marker_type",PetscDrawMarkerTypes,(PetscEnum*)&markertype,&set);CHKERRQ(ierr);
61771917b75SLisandro Dalcin   if (set) {
61871917b75SLisandro Dalcin     ierr = PetscDrawLGSetUseMarkers(lg,PETSC_TRUE);CHKERRQ(ierr);
61971917b75SLisandro Dalcin     ierr = PetscDrawSetMarkerType(lg->win,markertype);CHKERRQ(ierr);
62071917b75SLisandro Dalcin   }
62171917b75SLisandro Dalcin   usemarkers = lg->use_markers;
62271917b75SLisandro Dalcin   ierr = PetscOptionsGetBool(((PetscObject)lg)->options,((PetscObject)lg)->prefix,"-lg_use_markers",&usemarkers,&set);CHKERRQ(ierr);
62371917b75SLisandro Dalcin   if (set) {ierr = PetscDrawLGSetUseMarkers(lg,usemarkers);CHKERRQ(ierr);}
624287de1a7SBarry Smith   PetscFunctionReturn(0);
625287de1a7SBarry Smith }
626